mirrors/opensnitch
1
0
Fork 0
mirror of https://github.com/evilsocket/opensnitch.git synced 2025-03-04 08:34:40 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

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

Browse source
This commit is contained in:
evilsocket 2018-04-10 13:15:30 +02:00
parent 01e5134139
commit e3db27dfc3
Failed to generate hash of commit
2 changed files with 8 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
daemon/main.go
View file

@ -208,7 +208,7 @@ func main() {
// prepare the queue
setupWorkers()
queue, err := netfilter.NewQueue(uint16(queueNum), 0xffff, netfilter.NF_DEFAULT_PACKET_SIZE)
queue, err := netfilter.NewQueue(uint16(queueNum))
if err != nil {
log.Fatal("Error while creating queue #%d: %s", queueNum, err)
}

13
daemon/netfilter/queue.go
View file

@ -31,6 +31,7 @@ const (
NF_REPEAT Verdict = 4
NF_STOP Verdict = 5
NF_DEFAULT_QUEUE_SIZE uint32 = 0xffff
NF_DEFAULT_PACKET_SIZE uint32 = 4096
ipv4version = 0x40
@ -54,7 +55,7 @@ type Queue struct {
}
//Create and bind to queue specified by queueId
func NewQueue(queueId uint16, maxPacketsInQueue uint32, packetSize uint32) (*Queue, error) {
func NewQueue(queueId uint16) (*Queue, error) {
var q = Queue{
idx: uint32(time.Now().UnixNano()),
packets: make(chan Packet),
@ -78,17 +79,17 @@ func NewQueue(queueId uint16, maxPacketsInQueue uint32, packetSize uint32) (*Que
queueIndex[q.idx] = &q.packets
queueIndexLock.Unlock()
qLen := C.u_int32_t(maxPacketsInQueue)
bufSize := C.uint(packetSize)
queueSize := C.u_int32_t(NF_DEFAULT_QUEUE_SIZE)
bufferSize := C.uint(NF_DEFAULT_PACKET_SIZE)
if q.qh, err = C.CreateQueue(q.h, C.u_int16_t(queueId), C.u_int32_t(q.idx)); err != nil || q.qh == nil {
C.nfq_close(q.h)
return nil, fmt.Errorf("Error binding to queue: %v", err)
} else if ret, err = C.nfq_set_queue_maxlen(q.qh, qLen); err != nil || ret < 0 {
} else if ret, err = C.nfq_set_queue_maxlen(q.qh, queueSize); err != nil || ret < 0 {
C.nfq_destroy_queue(q.qh)
C.nfq_close(q.h)
return nil, fmt.Errorf("Unable to set max packets in queue: %v", err)
} else if C.nfq_set_mode(q.qh, C.u_int8_t(2), bufSize) < 0 {
} else if C.nfq_set_mode(q.qh, C.u_int8_t(2), bufferSize) < 0 {
C.nfq_destroy_queue(q.qh)
C.nfq_close(q.h)
return nil, fmt.Errorf("Unable to set packets copy mode: %v", err)
@ -96,7 +97,7 @@ func NewQueue(queueId uint16, maxPacketsInQueue uint32, packetSize uint32) (*Que
C.nfq_destroy_queue(q.qh)
C.nfq_close(q.h)
return nil, fmt.Errorf("Unable to get queue file-descriptor. %v", err)
} else if C.nfnl_rcvbufsiz(C.nfq_nfnlh(q.h), qLen*bufSize) < 0 {
} else if C.nfnl_rcvbufsiz(C.nfq_nfnlh(q.h), queueSize*bufferSize) < 0 {
C.nfq_destroy_queue(q.qh)
C.nfq_close(q.h)
return nil, fmt.Errorf("Unable to increase netfilter buffer space size.")