1
0
Fork 0
This repository has been archived on 2023-05-11. You can view files and clone it, but cannot push or open issues or pull requests.
lesson-python_ip_tunnel/main.py

55 lines
1.1 KiB
Python
Executable File

#!/usr/bin/env python3
import pytun
import struct
import ipaddress
class IpHeader:
def __init__(self, raw):
self.ver_ihl, \
self.tos, \
self.total_length, \
self.ident, \
self.flags_fragoffset, \
self.ttl, \
self.proto, \
self.chksum, \
src, \
dst, \
self.opt1, \
self.opt2, \
self.pad = struct.unpack('>BBHHHBBHIIHBB', raw)
self.src = ipaddress.IPv4Address(src)
self.dst = ipaddress.IPv4Address(dst)
def main():
tun_iface = pytun.TunTapDevice()
print(tun_iface.name)
tun_iface.addr = '10.0.0.1'
tun_iface.dstaddr = '10.0.0.2'
tun_iface.netmask = '255.255.255.0'
tun_iface.mtu = 1500
tun_iface.up()
while True:
buf = tun_iface.read(tun_iface.mtu)
flags = buf[:2]
proto = buf[2:4]
if proto != b'\x08\x00':
continue
ip_header = IpHeader(buf[4:28])
body = buf[28:]
print(ip_header.src, '', ip_header.dst, 'len: %d' % ip_header.total_length)
if __name__ == '__main__':
main()