misc: small fix or general refactoring i did not bother commenting

This commit is contained in:
evilsocket 2018-04-03 14:04:26 +02:00
parent 08da06acb7
commit 275abb40bb
Failed to generate hash of commit
7 changed files with 45 additions and 17 deletions

View file

@ -7,10 +7,10 @@ osd:
@cd daemon && make && mv daemon ../osd
osui:
@cd ui.test.service && make && mv ui.test.service ../ui
@cd ui.gtk && make && mv ui.gtk ../ui
clean:
@cd ui.proto && make clean
@cd daemon && make clean
@cd ui.test.service && make clean
@rm -rf osd
@cd ui.proto && make clean
@cd ui.gtk && make clean
@rm -rf osd ui

1
ui.gtk/.gitignore vendored Normal file
View file

@ -0,0 +1 @@
ui.gtk

7
ui.gtk/Makefile Normal file
View file

@ -0,0 +1,7 @@
all: ui.gtk
ui.gtk:
@go build .
clean:
@rm -rf ui.gtk

View file

@ -3,11 +3,13 @@ package main
import (
"flag"
"net"
"os"
"os/signal"
"syscall"
"golang.org/x/net/context"
"github.com/evilsocket/opensnitch/ui.test.service/core"
"github.com/evilsocket/opensnitch/daemon/core"
"github.com/evilsocket/opensnitch/daemon/log"
protocol "github.com/evilsocket/opensnitch/ui.proto"
@ -20,6 +22,7 @@ var (
listener = (net.Listener)(nil)
server = (*grpc.Server)(nil)
err = (error)(nil)
sigChan = (chan os.Signal)(nil)
)
type service struct{}
@ -40,6 +43,25 @@ func (s *service) AskRule(ctx context.Context, req *protocol.RuleRequest) (*prot
}, nil
}
func setupSignals() {
sigChan = make(chan os.Signal, 1)
signal.Notify(sigChan,
syscall.SIGHUP,
syscall.SIGINT,
syscall.SIGTERM,
syscall.SIGQUIT)
go func() {
sig := <-sigChan
log.Raw("\n")
log.Important("Got signal: %v", sig)
if listener != nil {
listener.Close()
}
os.Exit(0)
}()
}
func init() {
flag.StringVar(&socketPath, "socket-path", socketPath, "UNIX socket for this gRPC service.")
}
@ -47,9 +69,15 @@ func init() {
func main() {
flag.Parse()
log.Important("Starting %s v%s", core.Name, core.Version)
socketPath, err = core.ExpandPath(socketPath)
if err != nil {
log.Fatal("%s", err)
}
setupSignals()
log.Important("Starting %s v%s on socket %s", Name, Version, socketPath)
log.Info("Creating listener on unix://%s", socketPath)
listener, err = net.Listen("unix", socketPath)
if err != nil {
log.Fatal("%s", err)

View file

@ -1,4 +1,4 @@
package core
package main
const (
Name = "opensnitch-test-ui"

View file

@ -1 +0,0 @@
ui.test.service

View file

@ -1,7 +0,0 @@
all: daemon
daemon:
@go build .
clean:
@rm -rf daemon