151 lines
2.7 KiB
Go
151 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/dballard/warren/lib/warren"
|
|
"log"
|
|
"net"
|
|
"os"
|
|
"os/signal"
|
|
"strings"
|
|
"syscall"
|
|
)
|
|
|
|
var cmdSock net.Listener
|
|
|
|
type SockCommand struct {
|
|
warren.Command
|
|
Run func(cmd *SockCommand, c net.Conn, args ...string)
|
|
}
|
|
|
|
func (cmd *SockCommand) GetName() string {
|
|
return cmd.Name
|
|
}
|
|
|
|
func (cmd *SockCommand) GetSummary() string {
|
|
return cmd.Summary
|
|
}
|
|
|
|
func (cmd *SockCommand) Exec(c net.Conn, args []string) {
|
|
cmd.Flag.Usage = func() {
|
|
// helpFunc(c, c.Name)
|
|
}
|
|
cmd.Flag.Parse(args)
|
|
cmd.Run(cmd, c, cmd.Flag.Args()...)
|
|
}
|
|
|
|
func sockCmdHandle(c net.Conn) {
|
|
defer c.Close()
|
|
fmt.Println("cmdHandle")
|
|
data, err := warren.ReadStringz(c)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
warren.WriteStringz(c, "read error:"+err.Error())
|
|
return
|
|
}
|
|
fmt.Println("cmdHandle got command:", string(data))
|
|
|
|
// BAD: TODO: handle quotes "
|
|
var args = strings.Split(data, " ")
|
|
|
|
// parse args
|
|
var icmd = warren.GetCommand(args[0])
|
|
// get and run cmd
|
|
if icmd == nil {
|
|
warren.WriteStringz(c, "Unknown command "+args[0])
|
|
// usage ?
|
|
|
|
return
|
|
}
|
|
icmd.(*SockCommand).Exec(c, args[1:])
|
|
}
|
|
|
|
// determine if warrend is already running
|
|
func checkRunning() bool {
|
|
if pfile, err := os.Open(warren.WarrendPidFile); err == nil {
|
|
// pid file exists
|
|
var pid int
|
|
n, err := fmt.Fscanf(pfile, "%d", &pid)
|
|
if n == 0 || err != nil {
|
|
// read error
|
|
return false
|
|
}
|
|
// is there a runnign process with the pid?
|
|
err = syscall.Kill(pid, 0)
|
|
return err == nil
|
|
|
|
} else {
|
|
// pid file does not exist
|
|
return false
|
|
}
|
|
}
|
|
|
|
func initRunning() {
|
|
// rm old sock
|
|
os.Remove(warren.CmdSockAddr)
|
|
// drop down pid
|
|
pidfile, err := os.Create(warren.WarrendPidFile)
|
|
if err != nil {
|
|
log.Fatal("Could not create pidfile: ", warren.WarrendPidFile, " : ", err)
|
|
}
|
|
pid := os.Getpid()
|
|
fmt.Fprintf(pidfile, "%d", pid)
|
|
pidfile.Close()
|
|
|
|
cmdSock, err = net.Listen(warren.CmdSockType, warren.CmdSockAddr)
|
|
if err != nil {
|
|
fmt.Println(cmdSock)
|
|
Shutdown()
|
|
log.Fatal("Listen on sock error:", err)
|
|
}
|
|
|
|
}
|
|
|
|
func Shutdown() {
|
|
os.Remove(warren.WarrendPidFile)
|
|
if cmdSock != nil {
|
|
cmdSock.Close()
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
if checkRunning() {
|
|
log.Fatal("warrend instance already running")
|
|
}
|
|
|
|
LoadConf()
|
|
if conf == nil {
|
|
log.Fatal("failed to load conf")
|
|
}
|
|
|
|
initRunning()
|
|
|
|
warren.RegisterCommand(statusCmd)
|
|
warren.RegisterCommand(regCmd)
|
|
|
|
// Handle signals
|
|
c := make(chan os.Signal, 1)
|
|
signal.Notify(c,
|
|
syscall.SIGINT,
|
|
syscall.SIGTERM,
|
|
syscall.SIGQUIT,
|
|
syscall.SIGHUP,
|
|
)
|
|
go func() {
|
|
_ = <-c
|
|
Shutdown()
|
|
}()
|
|
|
|
// Listen
|
|
for {
|
|
fd, err := cmdSock.Accept()
|
|
fmt.Println("accept")
|
|
if err != nil {
|
|
Shutdown()
|
|
log.Fatal("accpet error:", err)
|
|
}
|
|
go sockCmdHandle(fd)
|
|
}
|
|
Shutdown()
|
|
}
|