replaycli-go/ipcmux/lib_windows.go

117 lines
2.4 KiB
Go
Raw Normal View History

2021-09-30 00:00:49 +00:00
//go:build windows
// +build windows
package ipcmux
import (
"errors"
"github.com/Microsoft/go-winio"
"log"
"net"
"os"
"strconv"
"strings"
"time"
)
func init() {
listenNet = func(path string) (net.Listener, error) {
path = strings.Replace(path, "/", "\\", -1)
log.Printf("Listening to: " + path)
chcon := make(chan bool)
var conn net.Listener
var err error
go func() {
path = "\\\\.\\pipe\\" + path
conn, err = winio.ListenPipe(path, &winio.PipeConfig{
SecurityDescriptor: "",
MessageMode: false,
InputBufferSize: 0,
OutputBufferSize: 0,
})
if err != nil {
log.Printf("ipcmux::listenNet: %s", err.Error())
}
go func() {
go func() {
to := time.Second * time.Duration(3)
fconn, err := winio.DialPipe(path, &to)
if err != nil {
log.Printf("Error on pipe warm up: %s", err.Error())
} else {
fconn.Close()
}
log.Printf("Pipe warm up done")
}()
}()
chcon <- true
}()
select {
case <-chcon:
return conn, err
case <-time.After(time.Second * 5):
err = errors.New("Timeout exceeded for Listening: " + path)
return nil, err
}
return conn, err
}
dialNet = func(path string) (net.Conn, error) {
path = strings.Replace(path, "/", "\\", -1)
path = "\\\\.\\pipe\\" + path
chcon := make(chan bool)
var conn net.Conn
var err error
//Log("Dialing to: " + path)
go func() {
to := time.Second * time.Duration(15)
conn, err = winio.DialPipe(path, &to)
//conn, err = npipe.Dial(path)
chcon <- true
}()
select {
case <-chcon:
return conn, err
case <-time.After(time.Second * 5):
err = errors.New("Timeout exceeded for conn: " + path)
return nil, err
}
return conn, err
}
}
func CheckRoot() {
time.Sleep(time.Second)
rootpid := os.Getenv(ENV_ROOTPID)
rootpidi, err := strconv.Atoi(rootpid)
if err != nil {
log.Printf("No root process found")
}
if rootpidi == 0 || rootpidi == os.Getpid() {
log.Printf("%s is set to zero, no monitoring on root will take place", ENV_ROOTPID)
return
}
p, err := os.FindProcess(rootpidi)
if err != nil {
log.Printf("Error finding process: %s", err.Error())
os.Exit(1)
}
if p == nil {
log.Printf("Process not found.")
os.Exit(1)
}
_, err = p.Wait()
if err != nil {
log.Printf(err.Error())
}
log.Printf("Parent process %s is gone. Finishing here.", os.Getenv(ENV_ROOTPID))
os.Exit(0)
}