from ctypes import * from ctypes.util import find_library import sys WIN=False if sys.platform.startswith('win'): WIN=True if WIN: SOCKET = c_uint _lib=CDLL('dnet') else: SOCKET = c_int _lib_name = find_library('dnet') if not _lib_name: raise OSError("Cannot find libdnet.so") _lib=CDLL(_lib_name) ETH_ADDR_LEN = 6 INTF_NAME_LEN = 16 INTF_NAME_COUNT = 20 INTF_ALIAS_COUNT = 20 IP6_ADDR_LEN = 16 ADDR_TYPE_NONE = 0 ADDR_TYPE_ETH = 1 ADDR_TYPE_IP = 2 ADDR_TYPE_IP6 = 3 INTF_TYPE_OTHER = 1 INTF_TYPE_ETH = 6 INTF_TYPE_TOKENRING = 9 INTF_TYPE_FDDI = 15 INTF_TYPE_PPP = 23 INTF_TYPE_LOOPBACK = 24 INTF_TYPE_SLIP = 28 INTF_TYPE_TUN = 53 uint8_t = c_ubyte uint16_t = c_ushort uint32_t = c_uint ssize_t = c_long dnet_ip_addr_t = uint32_t dnet_intf_name = c_char * INTF_NAME_LEN class dnet_intf_list(Structure): pass dnet_intf_list._fields_ = [ ('length', c_int), ('interfaces', dnet_intf_name * 20) ] class dnet_eth_addr(Structure): pass dnet_eth_addr._fields_ = [ ('data', uint8_t * ETH_ADDR_LEN) ] dnet_eth_addr_t = dnet_eth_addr class dnet_ip6_addr(Structure): pass dnet_ip6_addr._fields_ = [ ('data', uint8_t * IP6_ADDR_LEN) ] dnet_ip6_addr_t = dnet_ip6_addr class dnet_addr_u(Union): pass dnet_addr_u._fields_ = [ ('eth', dnet_eth_addr_t), ('ip', dnet_ip_addr_t), ('ip6', dnet_ip6_addr_t), ('data8', uint8_t * 16), ('data16', uint16_t * 8), ('data32', uint32_t * 4) ] class dnet_addr(Structure): pass dnet_addr._anonymous_ = ('__addr_u', ) dnet_addr._fields_ = [ ('addr_type', uint16_t), ('addr_bits', uint16_t), ('__addr_u', dnet_addr_u) ] class dnet_intf_entry(Structure): pass dnet_intf_entry._fields_ = [ ('intf_len', c_uint), ('intf_name', c_char * INTF_NAME_LEN), ('intf_type', c_ushort), ('intf_flags', c_ushort), ('intf_mtu', c_uint), ('intf_addr', dnet_addr), ('intf_dst_addr', dnet_addr), ('intf_link_addr', dnet_addr), ('intf_alias_num', c_uint), ('intf_alias_addrs', dnet_addr * INTF_ALIAS_COUNT) ] eth_t = c_void_p intf_t = c_void_p ip_t = c_void_p dnet_intf_handler = CFUNCTYPE(c_int, POINTER(dnet_intf_entry), POINTER(c_void_p)) dnet_eth_open = _lib.eth_open dnet_eth_open.restype = POINTER(eth_t) dnet_eth_open.argtypes = [ POINTER(c_char) ] dnet_eth_get = _lib.eth_get dnet_eth_get.restype = c_int dnet_eth_get.argtypes = [ POINTER(eth_t), POINTER(dnet_eth_addr_t) ] dnet_eth_set = _lib.eth_set dnet_eth_set.restype = c_int dnet_eth_set.argtypes = [ POINTER(eth_t), POINTER(dnet_eth_addr_t) ] dnet_eth_send = _lib.eth_send dnet_eth_send.restype = ssize_t dnet_eth_send.argtypes = [ POINTER(eth_t), c_void_p, c_size_t ] dnet_eth_close = _lib.eth_close dnet_eth_close.restype = POINTER(eth_t) dnet_eth_close.argtypes = [ POINTER(eth_t) ] dnet_intf_open = _lib.intf_open dnet_intf_open.restype = POINTER(intf_t) dnet_intf_open.argtypes = [ ] dnet_intf_get = _lib.intf_get dnet_intf_get.restype = c_int dnet_intf_get.argtypes = [ POINTER(intf_t), POINTER(dnet_intf_entry) ] dnet_intf_get_src = _lib.intf_get_src dnet_intf_get_src.restype = c_int dnet_intf_get_src.argtypes = [ POINTER(intf_t), POINTER(dnet_intf_entry), POINTER(dnet_addr) ] dnet_intf_get_dst = _lib.intf_get_dst dnet_intf_get_dst.restype = c_int dnet_intf_get_dst.argtypes = [ POINTER(intf_t), POINTER(dnet_intf_entry), POINTER(dnet_addr) ] dnet_intf_set = _lib.intf_set dnet_intf_set.restype = c_int dnet_intf_set.argtypes = [ POINTER(intf_t), POINTER(dnet_intf_entry) ] dnet_intf_loop = _lib.intf_loop dnet_intf_loop.restype = POINTER(intf_t) dnet_intf_loop.argtypes = [ POINTER(intf_t), dnet_intf_handler, c_void_p ] dnet_intf_close = _lib.intf_close dnet_intf_close.restype = POINTER(intf_t) dnet_intf_close.argtypes = [ POINTER(intf_t) ] dnet_ip_open = _lib.ip_open dnet_ip_open.restype = POINTER(ip_t) dnet_ip_open.argtypes = [ ] dnet_ip_add_option = _lib.ip_add_option dnet_ip_add_option.restype = ssize_t dnet_ip_add_option.argtypes = [ POINTER(c_void_p), c_size_t, c_int, POINTER(c_void_p), c_size_t ] dnet_ip_checksum = _lib.ip_checksum dnet_ip_checksum.restype = None dnet_ip_checksum.argtypes = [ POINTER(c_void_p), c_size_t ] dnet_ip_send = _lib.ip_send dnet_ip_send.restype = ssize_t dnet_ip_send.argtypes = [ POINTER(ip_t), c_void_p, c_size_t ] dnet_ip_close = _lib.ip_close dnet_ip_close.restype = POINTER(ip_t) dnet_ip_close.argtypes = [ POINTER(ip_t) ] class dnet_eth: def __init__(self, iface): self.iface_b = create_string_buffer(iface.encode('ascii')) self.eth = dnet_eth_open(self.iface_b) def send(self, sx): dnet_eth_send(self.eth, sx, len(sx)) def close(self): return dnet_eth_close(self.eth) class dnet_ip: def __init__(self): self.ip = dnet_ip_open() def send(self, sx): dnet_ip_send(self.ip, sx, len(sx)) def close(self): return dnet_ip_close(self.ip) def dnet_intf_name_loop(entry, intf_list): l = cast(intf_list, POINTER(dnet_intf_list)) if l.contents.length >= INTF_NAME_COUNT: return -1 for i in enumerate(entry.contents.intf_name): l.contents.interfaces[l.contents.length][i[0]] = i[1] l.contents.length += 1 return 0 class dnet_intf: def __init__(self): self.intf = dnet_intf_open() intf_list = dnet_intf_list() intf_list.length = 0 dnet_intf_loop(self.intf, dnet_intf_handler(dnet_intf_name_loop), pointer(intf_list)) self.names = [] for i in range(INTF_NAME_COUNT): if i >= intf_list.length: break self.names.append(intf_list.interfaces[i].value.decode('ascii').strip('\0')) def close(self): return dnet_intf_close(self.intf) def get(self, iface): ret = {} entry = dnet_intf_entry() entry.intf_name = iface.encode('ascii') entry.intf_len = sizeof(entry) r = dnet_intf_get(self.intf, byref(entry)) if r < 0: return {} ret['addr6'] = [] for i in range(entry.intf_alias_num): if entry.intf_alias_addrs[i].addr_type == ADDR_TYPE_IP6: ret['addr6'].append(bytes(entry.intf_alias_addrs[i].data8[:16])) ret['type'] = entry.intf_type ret['addr'] = bytes(entry.intf_addr.data8[:4]) ret['link_addr'] = bytes(entry.intf_link_addr.data8[:6]) return ret