Add windowskext integration, update related packages
This commit is contained in:
110
firewall/interception/windowskext/test/main.go
Normal file
110
firewall/interception/windowskext/test/main.go
Normal file
@@ -0,0 +1,110 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
|
||||
"github.com/Safing/portbase/log"
|
||||
"github.com/Safing/portmaster/firewall/interception/windowskext"
|
||||
"github.com/Safing/portmaster/network/packet"
|
||||
)
|
||||
|
||||
var (
|
||||
packets chan packet.Packet
|
||||
)
|
||||
|
||||
func main() {
|
||||
|
||||
// check parameter count
|
||||
if len(os.Args) < 3 {
|
||||
fmt.Printf("usage: %s <dll> <sys>", os.Args[0])
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// check parameters
|
||||
for i := 1; i < 3; i++ {
|
||||
if _, err := os.Stat(os.Args[i]); err != nil {
|
||||
fmt.Printf("could not access %s: %s", os.Args[i], err)
|
||||
os.Exit(2)
|
||||
}
|
||||
}
|
||||
|
||||
// logging
|
||||
log.Start()
|
||||
log.Info("starting Portmaster Windows Kext Test Program")
|
||||
|
||||
// init
|
||||
err := windowskext.Init(os.Args[1], os.Args[2])
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
// start
|
||||
err = windowskext.Start()
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
packets = make(chan packet.Packet, 1000)
|
||||
go windowskext.Handler(packets)
|
||||
go handlePackets()
|
||||
|
||||
// catch interrupt for clean shutdown
|
||||
signalCh := make(chan os.Signal)
|
||||
signal.Notify(
|
||||
signalCh,
|
||||
os.Interrupt,
|
||||
syscall.SIGHUP,
|
||||
syscall.SIGINT,
|
||||
syscall.SIGTERM,
|
||||
syscall.SIGQUIT,
|
||||
)
|
||||
<-signalCh
|
||||
fmt.Println(" <INTERRUPT>")
|
||||
log.Warning("program was interrupted, shutting down")
|
||||
|
||||
// stop
|
||||
err = windowskext.Stop()
|
||||
if err != nil {
|
||||
fmt.Printf("error stopping: %s\n", err)
|
||||
}
|
||||
|
||||
log.Info("shutdown complete")
|
||||
log.Shutdown()
|
||||
|
||||
os.Exit(0)
|
||||
}
|
||||
|
||||
func handlePackets() {
|
||||
for {
|
||||
pkt := <-packets
|
||||
|
||||
if pkt == nil {
|
||||
log.Infof("stopped handling packets")
|
||||
return
|
||||
}
|
||||
|
||||
log.Infof("received packet: %s", pkt)
|
||||
|
||||
data, err := pkt.GetPayload()
|
||||
if err != nil {
|
||||
log.Errorf("failed to get payload: %s", err)
|
||||
} else {
|
||||
log.Infof("payload is: %x", data)
|
||||
}
|
||||
|
||||
// reroute dns requests to nameserver
|
||||
if pkt.IsOutbound() && !pkt.Info().Src.Equal(pkt.Info().Dst) && pkt.Info().DstPort == 53 {
|
||||
log.Infof("rerouting %s", pkt)
|
||||
pkt.RerouteToNameserver()
|
||||
continue
|
||||
}
|
||||
|
||||
// accept all
|
||||
log.Infof("accepting %s", pkt)
|
||||
pkt.PermanentAccept()
|
||||
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user