diff --git a/src/device.go b/src/device.go index d32d648..1185d60 100644 --- a/src/device.go +++ b/src/device.go @@ -186,6 +186,6 @@ func (device *Device) Close() { close(device.signal.stop) } -func (device *Device) Wait() { - <-device.signal.stop +func (device *Device) WaitChannel() chan struct{} { + return device.signal.stop } diff --git a/src/logger.go b/src/logger.go index 827f9e9..9fe73b4 100644 --- a/src/logger.go +++ b/src/logger.go @@ -40,11 +40,11 @@ func NewLogger(level int) *Logger { logger.Info = log.New(logInfo, "INFO: ", - log.Ldate|log.Ltime|log.Lshortfile, + log.Ldate|log.Ltime, ) logger.Error = log.New(logErr, "ERROR: ", - log.Ldate|log.Ltime|log.Lshortfile, + log.Ldate|log.Ltime, ) return logger } diff --git a/src/main.go b/src/main.go index 0857999..dde21fb 100644 --- a/src/main.go +++ b/src/main.go @@ -4,6 +4,7 @@ import ( "fmt" "log" "os" + "os/signal" "runtime" ) @@ -78,17 +79,38 @@ func main() { if err != nil { logError.Fatal("UAPI listen error:", err) } - defer uapi.Close() + + errs := make(chan error) + term := make(chan os.Signal) + wait := device.WaitChannel() go func() { for { conn, err := uapi.Accept() if err != nil { - logError.Fatal("UAPI accept error:", err) + errs <- err + return } go ipcHandle(device, conn) } }() - device.Wait() + logInfo.Println("UAPI listener started") + + // wait for program to terminate + + signal.Notify(term, os.Kill) + signal.Notify(term, os.Interrupt) + + select { + case <-wait: + case <-term: + case <-errs: + } + + // clean up UAPI bind + + uapi.Close() + + logInfo.Println("Closing") }