cpp/funstraw
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Full decode up the stack.

Browse Source
This commit is contained in:
Ian Gulliver
2015-01-09 06:38:56 +00:00
parent f4c6a7bff2
commit 0a1f6f04dc
1 changed files with 30 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
nl80211.py
View File

@@ -98,10 +98,6 @@ class EmptyParser(object):
pass pass
nlmsghdr = StructParser('LHHLL', ('length', 'type', 'flags', 'seq', 'pid'))
genlmsghdr = StructParser('BBH', ('cmd', 'version', 'reserved'))
class Attribute(object): class Attribute(object):
_nlattr = StructParser('HH', ('len', 'type')) _nlattr = StructParser('HH', ('len', 'type'))
@@ -265,7 +261,9 @@ STA_FLAG_TDLS_PEER = 1 << 5
STA_FLAG_ASSOCIATED = 1 << 6 STA_FLAG_ASSOCIATED = 1 << 6
class Connection(object): class Netlink(object):
_nlmsghdr = StructParser('LHHLL', ('length', 'type', 'flags', 'seq', 'pid'))
def __init__(self): def __init__(self):
self._sock = socket.socket(socket.AF_NETLINK, socket.SOCK_DGRAM, 16) self._sock = socket.socket(socket.AF_NETLINK, socket.SOCK_DGRAM, 16)
self._sock.bind((0, 0)) self._sock.bind((0, 0))
@@ -273,18 +271,37 @@ class Connection(object):
def Send(self, msg): def Send(self, msg):
self._sock.send(msg) self._sock.send(msg)
def RecvAndUnpack(self): def Recv(self):
ret = []
while True: while True:
data = self._sock.recv(4096) data = self._sock.recv(4096)
iterator = Iterator(data) iterator = Iterator(data)
while not iterator.AtEnd(): while not iterator.AtEnd():
myhdr = nlmsghdr.Unpack(iterator) myhdr = self._nlmsghdr.Unpack(iterator)
if myhdr['type'] == NLMSG_DONE: if myhdr['type'] == NLMSG_DONE:
return ret return
ret.append(iterator.ExtractIterator(myhdr['length'] - nlmsghdr.size)) yield (myhdr['type'], iterator.ExtractIterator(myhdr['length'] - self._nlmsghdr.size))
if not myhdr['flags'] & NLMSG_F_MULTI: if not myhdr['flags'] & NLMSG_F_MULTI:
return ret return
class GenericNetlink(object):
_genlmsghdr = StructParser('BBH', ('cmd', 'version', 'reserved'))
_parser = {
0x10: ctrl_attr,
}
def __init__(self):
self._netlink = Netlink()
def Send(self, msg):
self._netlink.Send(msg)
def Recv(self):
for msgtype, iterator in self._netlink.Recv():
genlhdr = self._genlmsghdr.Unpack(iterator)
parser = self._parser[msgtype]
yield (genlhdr['cmd'], parser.Unpack(iterator))
#int_genquery = Accumulator() #int_genquery = Accumulator()
@@ -306,15 +323,7 @@ class Connection(object):
# pid=os.getpid()) # pid=os.getpid())
#genquery.Append(str(int_genquery)) #genquery.Append(str(int_genquery))
#sock.send(str(genquery))
#data = sock.recv(4096)
#iterator = Iterator(data)
#print 'nlmsghdr: %s' % nlmsghdr.Unpack(iterator)
#print 'genlmsghdr: %s' % genlmsghdr.Unpack(iterator)
#print 'nl80211_attr: %s' % nl80211_attr.Unpack(iterator)
query = '\24\0\0\0\20\0\5\3a\6\256T\v\17\0\0\3\1\0\0' query = '\24\0\0\0\20\0\5\3a\6\256T\v\17\0\0\3\1\0\0'
conn = Connection() conn = GenericNetlink()
conn.Send(query) conn.Send(query)
print ['%s\n' % s for s in conn.RecvAndUnpack()] print ['%s\n' % s for cmd, s in conn.Recv()]