mirror of
https://github.com/evilsocket/opensnitch.git
synced 2025-03-04 08:34:40 +01:00
misc: small fix or general refactoring i did not bother commenting
This commit is contained in:
parent
08da06acb7
commit
275abb40bb
7 changed files with 45 additions and 17 deletions
8
Makefile
8
Makefile
|
@ -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
1
ui.gtk/.gitignore
vendored
Normal file
|
@ -0,0 +1 @@
|
|||
ui.gtk
|
7
ui.gtk/Makefile
Normal file
7
ui.gtk/Makefile
Normal file
|
@ -0,0 +1,7 @@
|
|||
all: ui.gtk
|
||||
|
||||
ui.gtk:
|
||||
@go build .
|
||||
|
||||
clean:
|
||||
@rm -rf ui.gtk
|
|
@ -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)
|
|
@ -1,4 +1,4 @@
|
|||
package core
|
||||
package main
|
||||
|
||||
const (
|
||||
Name = "opensnitch-test-ui"
|
1
ui.test.service/.gitignore
vendored
1
ui.test.service/.gitignore
vendored
|
@ -1 +0,0 @@
|
|||
ui.test.service
|
|
@ -1,7 +0,0 @@
|
|||
all: daemon
|
||||
|
||||
daemon:
|
||||
@go build .
|
||||
|
||||
clean:
|
||||
@rm -rf daemon
|
Loading…
Add table
Reference in a new issue