You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
68 lines
1.8 KiB
68 lines
1.8 KiB
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import signal
|
|
from scapy.all import *
|
|
from netfilterqueue import NetfilterQueue
|
|
import argparse
|
|
|
|
window_size = 20
|
|
|
|
def modify_window(pkt):
|
|
try:
|
|
ip = IP(pkt.get_payload())
|
|
if ip.haslayer(TCP) and ip[TCP].flags == "SA":
|
|
ip[TCP].window = window_size
|
|
del ip[IP].chksum
|
|
del ip[TCP].chksum
|
|
pkt.set_payload(bytes(ip))
|
|
elif ip.haslayer(TCP) and ip[TCP].flags == "FA":
|
|
ip[TCP].window = window_size
|
|
del ip[IP].chksum
|
|
del ip[TCP].chksum
|
|
pkt.set_payload(bytes(ip))
|
|
elif ip.haslayer(TCP) and ip[TCP].flags == "PA":
|
|
ip[TCP].window = window_size
|
|
del ip[IP].chksum
|
|
del ip[TCP].chksum
|
|
pkt.set_payload(bytes(ip))
|
|
elif ip.haslayer(TCP) and ip[TCP].flags == "A":
|
|
ip[TCP].window = window_size
|
|
del ip[IP].chksum
|
|
del ip[TCP].chksum
|
|
pkt.set_payload(bytes(ip))
|
|
except:
|
|
pass
|
|
|
|
pkt.accept()
|
|
|
|
def parsearg():
|
|
parser = argparse.ArgumentParser(description='Description of your program')
|
|
|
|
parser.add_argument('-q', '--queue', type=int, help='iptables Queue Num')
|
|
parser.add_argument('-w', '--window_size', type=int, help='Tcp Window Size')
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.queue is None or args.window_size is None:
|
|
exit(1)
|
|
|
|
window_size = args.window_size
|
|
|
|
return args.queue
|
|
|
|
def main():
|
|
queue_num = parsearg()
|
|
nfqueue = NetfilterQueue()
|
|
nfqueue.bind(queue_num, modify_window)
|
|
|
|
try:
|
|
print("Starting netfilter_queue process...")
|
|
nfqueue.run()
|
|
except KeyboardInterrupt:
|
|
pass
|
|
|
|
if __name__ == "__main__":
|
|
#sys.stdout = os.fdopen(sys.stdout.fileno(), 'w', 0)
|
|
signal.signal(signal.SIGINT, lambda signal, frame: sys.exit(0))
|
|
main()
|