openstack/deb-ryu
Code Issues Proposed changes

packet lib: make Packet class iterator

Browse Source 
Signed-off-by: FUJITA Tomonori <fujita.tomonori@lab.ntt.co.jp>
This commit is contained in:
FUJITA Tomonori 2012-10-05 19:01:52 +09:00
parent 9c30012955
commit 5fd538ee7f
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
ryu/lib/packet/packet.py
View File

@ -22,6 +22,7 @@ class Packet(object):
super(Packet, self).__init__() super(Packet, self).__init__()
self.data = data self.data = data
self.protocols = [] self.protocols = []
self.protocol_idx = 0
self.parsed_bytes = 0 self.parsed_bytes = 0
if self.data: if self.data:
# Do we need to handle non ethernet? # Do we need to handle non ethernet?
@ -55,3 +56,16 @@ class Packet(object):
for p in self.protocols: for p in self.protocols:
if p.__class__.__name__ == name: if p.__class__.__name__ == name:
return p return p
def next(self):
try:
p = self.protocols[self.protocol_idx]
except:
self.protocol_idx = 0
raise StopIteration
self.protocol_idx += 1
return p
def __iter__(self):
return self