include an example implementation of netarch.ip, "start-here.py"

This commit is contained in:
pi-rho 2012-06-08 08:02:21 -05:00
parent c7847cf288
commit 1d70f7d6e2
2 changed files with 87 additions and 0 deletions

84
extra/start-here.py Normal file
View File

@ -0,0 +1,84 @@
#!/usr/bin/python
'''Example use of the netarch.ip framework
Searching and replacing "StinkyPinky" with your protocol's name would be a good
start.
'''
import sys
from netarch import ip
from netarch import unpack
class StinkyPinkyPacket(ip.Packet):
''' StinkyPinky Protocol '''
def parse(self, data):
'''Parse Packet Data
This method deals with data on a packet level. Its job in life is to
set parts, payload, and opcode. If any data is passed to this method
is deemed to not be part of the packet's data, it should be returned.
Likewise, if the Packet needs more data, raise ip.NeedsMoreData
self.parts - a magic bag of values. self.parts[:-1] is highlighted when
printed iff the value == length(self.payload)
self.payload - non-header packet data
self.opcode - an integer that triggers additional parsing, or special
display
'''
self.parts = unpack("<BBBB", data)
self.payload = self.parts[-1]
return None
def opcode_0(self): # example - delete me
'''Example opcode parser
Each identified opcode will need to have a method defined for it. They
should be named opcode_ followed by the integer decimal opcode and
retain the same method signature.
It is also important that each opcode method defines a docstring. It
will be used in the packet display.
'''
pass
def opcode_1(self): # example - delete me
''' NOP Command '''
pass
class StinkyPinkySession(ip.HtmlSession):
''' A StinkyPinky Session '''
def __init__(self, frame, packetClass=StinkyPinkyPacket):
ip.HtmlSession.__init__(self, frame, packetClass)
def process(self, packet):
'''Process packet data
This method might be a good spot for special data handling at a session
level. One example would be carving embedded data to a separate file.
'''
packet.show()
# execution harness
if __name__ == '__main__':
if len(sys.argv) > 1:
s = None
reseq = ip.Dispatch(*sys.argv[1:])
for h, d in reseq:
srv, first, chunk = d
if not s:
s = StinkyPinkySession(first)
s.handle(srv, first, chunk, reseq.last)

View File

@ -18,4 +18,7 @@ setup(
"Topic :: System :: Networking"
],
packages=['netarch'],
data_files=[
('share/netarch/examples', ['extra/start-here.py']),
]
)