117 lines
2.4 KiB
Go
117 lines
2.4 KiB
Go
|
//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)
|
||
|
|
|
||
|
|
}
|