78 lines
1.2 KiB
Go
78 lines
1.2 KiB
Go
|
package bootstrap
|
||
|
|
|
||
|
|
import (
|
||
|
|
log "github.com/cihub/seelog"
|
||
|
|
"net"
|
||
|
|
"os"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
const FILE_SERVER string = "/var/run/of_server"
|
||
|
|
|
||
|
|
var fd net.Conn
|
||
|
|
|
||
|
|
func HandleShell(stop chan bool) {
|
||
|
|
go func() {
|
||
|
|
//os.Stdout.Close()
|
||
|
|
//os.Stderr.Close()
|
||
|
|
//os.Stdin.Close()
|
||
|
|
|
||
|
|
if fi, _ := os.Stat(FILE_SERVER); fi != nil {
|
||
|
|
os.Remove(FILE_SERVER)
|
||
|
|
}
|
||
|
|
|
||
|
|
l, err := net.ListenUnix("unix", &net.UnixAddr{Name: FILE_SERVER, Net: "unix"})
|
||
|
|
if err != nil {
|
||
|
|
log.Error(err.Error())
|
||
|
|
return
|
||
|
|
}
|
||
|
|
defer func() {
|
||
|
|
l.Close()
|
||
|
|
os.Remove(FILE_SERVER)
|
||
|
|
}()
|
||
|
|
|
||
|
|
for {
|
||
|
|
fd, err = l.Accept()
|
||
|
|
if err != nil {
|
||
|
|
log.Error(err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
buf := make([]byte, 1024)
|
||
|
|
nr, err := fd.Read(buf)
|
||
|
|
if err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
data := string(buf[0:nr])
|
||
|
|
switch strings.ToUpper(data) {
|
||
|
|
case "STOP":
|
||
|
|
if fi, _ := os.Stat(FILE_SERVER); fi != nil {
|
||
|
|
os.Remove(FILE_SERVER)
|
||
|
|
}
|
||
|
|
stop <- true
|
||
|
|
case "STATUS":
|
||
|
|
fd.Write(status())
|
||
|
|
default:
|
||
|
|
}
|
||
|
|
|
||
|
|
fd.Close()
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
}
|
||
|
|
|
||
|
|
func StopHandleShell() {
|
||
|
|
log.Info("Shell Handler stopped.")
|
||
|
|
|
||
|
|
if fi, _ := os.Stat(FILE_SERVER); fi != nil {
|
||
|
|
os.Remove(FILE_SERVER)
|
||
|
|
}
|
||
|
|
if fd != nil {
|
||
|
|
fd.Close()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func status() []byte {
|
||
|
|
return []byte("STATUS OK\n")
|
||
|
|
}
|