dprevot/netflow
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

converted hardcoded host/port to arg driven, switched int.from_bytes to Python2 friendly routine

Browse source
This commit is contained in:
Randy Gelhausen 2016-11-29 22:50:09 -05:00
parent 8fa999b877
commit bd22551669
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
collector-v9.py
View file

@ -13,9 +13,10 @@ Licensed under MIT License. See LICENSE.
import socket import socket
import struct import struct
import sys
HOST = '0.0.0.0' HOST = sys.argv[1]
PORT = 9001 PORT = int(sys.argv[2])
field_types = { field_types = {
1: 'IN_BYTES', 1: 'IN_BYTES',
@ -151,7 +152,9 @@ class DataFlowSet:
dataslice = data[offset:offset+flen] dataslice = data[offset:offset+flen]
# Better solution than struct.unpack with variable field length # Better solution than struct.unpack with variable field length
fdata = int.from_bytes(dataslice, byteorder='big') fdata = 0
for idx, byte in enumerate(reversed(bytearray(dataslice))):
fdata += byte << (idx * 8)
new_record.data[fkey] = fdata new_record.data[fkey] = fdata