Gecko [ unitednationsplay.com ]

Name	Size	Permission	Date
__init__.py	0 B	-rw-r--r--	2013-05-26 22:44
__init__.pyc	137 B	-rw-r--r--	2018-10-30 22:23
__init__.pyo	137 B	-rw-r--r--	2018-10-30 22:23
data.py	14.85 KB	-rw-r--r--	2018-10-30 22:23
data.pyc	15.25 KB	-rw-r--r--	2018-10-30 22:23
data.pyo	15.21 KB	-rw-r--r--	2018-10-30 22:23
genetlink.py	2 KB	-rw-r--r--	2013-05-26 22:44
genetlink.pyc	3.43 KB	-rw-r--r--	2018-10-30 22:23
genetlink.pyo	3.43 KB	-rw-r--r--	2018-10-30 22:23
ioprio.py	5.43 KB	-rw-r--r--	2018-10-30 22:23
ioprio.pyc	5.5 KB	-rw-r--r--	2018-10-30 22:23
ioprio.pyo	5.5 KB	-rw-r--r--	2018-10-30 22:23
netlink.py	7.71 KB	-rw-r--r--	2013-05-26 22:44
netlink.pyc	12.06 KB	-rw-r--r--	2018-10-30 22:23
netlink.pyo	12.06 KB	-rw-r--r--	2018-10-30 22:23
ui.py	23.17 KB	-rw-r--r--	2013-05-26 22:44
ui.pyc	24.54 KB	-rw-r--r--	2018-10-30 22:23
ui.pyo	24.54 KB	-rw-r--r--	2018-10-30 22:23
version.py	16 B	-rw-r--r--	2013-05-26 22:44
version.pyc	162 B	-rw-r--r--	2018-10-30 22:23
version.pyo	162 B	-rw-r--r--	2018-10-30 22:23
vmstat.py	1.52 KB	-rw-r--r--	2013-05-26 22:44
vmstat.pyc	1.37 KB	-rw-r--r--	2018-10-30 22:23
vmstat.pyo	1.37 KB	-rw-r--r--	2018-10-30 22:23

Rename

'''
Netlink message generation/parsing

GPLv2+; See copying for details.
'''

import struct
from iotop.netlink import NLM_F_REQUEST, NLMSG_MIN_TYPE, Message, parse_attributes
from iotop.netlink import NulStrAttr, Connection, NETLINK_GENERIC

CTRL_CMD_UNSPEC       = 0
CTRL_CMD_NEWFAMILY    = 1
CTRL_CMD_DELFAMILY    = 2
CTRL_CMD_GETFAMILY    = 3
CTRL_CMD_NEWOPS       = 4
CTRL_CMD_DELOPS       = 5
CTRL_CMD_GETOPS       = 6

CTRL_ATTR_UNSPEC      = 0
CTRL_ATTR_FAMILY_ID   = 1
CTRL_ATTR_FAMILY_NAME = 2
CTRL_ATTR_VERSION     = 3
CTRL_ATTR_HDRSIZE     = 4
CTRL_ATTR_MAXATTR     = 5
CTRL_ATTR_OPS         = 6

class GenlHdr:
    def __init__(self, cmd, version = 0):
        self.cmd = cmd
        self.version = version
    def _dump(self):
        return struct.pack("BBxx", self.cmd, self.version)

def _genl_hdr_parse(data):
    return GenlHdr(*struct.unpack("BBxx", data))

GENL_ID_CTRL        = NLMSG_MIN_TYPE

class GeNlMessage(Message):
    def __init__(self, family, cmd, attrs=[], flags=0):
        self.cmd = cmd
        self.attrs = attrs
        self.family = family
        Message.__init__(self, family, flags=flags,
                         payload=[GenlHdr(self.cmd)]+attrs)

@staticmethod
    def recv(conn):
        msg = conn.recv()
        packet = msg.payload
        hdr = _genl_hdr_parse(packet[:4])

genlmsg = GeNlMessage(msg.type, hdr.cmd, [], msg.flags)
        genlmsg.attrs = parse_attributes(packet[4:])
        genlmsg.version = hdr.version

return genlmsg

class Controller:
    def __init__(self, conn):
        self.conn = conn
    def get_family_id(self, family):
        a = NulStrAttr(CTRL_ATTR_FAMILY_NAME, family)
        m = GeNlMessage(GENL_ID_CTRL, CTRL_CMD_GETFAMILY,
                        flags=NLM_F_REQUEST, attrs=[a])
        m.send(self.conn)
        m = GeNlMessage.recv(self.conn)
        return m.attrs[CTRL_ATTR_FAMILY_ID].u16()

connection = Connection(NETLINK_GENERIC)
controller = Controller(connection)