Changeset 3672

Timestamp:

08/29/08 19:55:26 (5 years ago)

Author:

nopper

Message:

Implementing save and load in SequenceContext? (Scapy)

Location:

branch/PacketManipulator/PM/Backend/Scapy

Files:

• Context/Sequence.py (modified) (3 diffs)
• serialize.py (modified) (7 diffs)

branch/PacketManipulator/PM/Backend/Scapy/Context/Sequence.py

@@ -23 +23 @@
 
 from PM.Core.I18N import _
+from PM.Backend.Scapy.serialize import load_sequence, save_sequence
 from PM.Backend.Scapy.utils import execute_sequence
 
@@ -28 +29 @@
     
     class SequenceContext(BaseSequenceContext):
+        file_types = [(_('Scapy sequence'), '*.pms')]
+
         def __init__(self, seq, count, inter, iface, strict, report_recv, report_sent,
                      scallback, rcallback, sudata=None, rudata=None):
@@ -40 +43 @@
             self.title = _('Unsaved')
             self.summary = _('Executing sequence (%d packets %d times)') % (self.tot_packet_count, count)
+        
+        def save(self):
+            if self.cap_file:
+                self.seq = self.load_sequence(self.cap_file)
+
+                if self.seq is not None:
+                    self.state = self.SAVED
+                    return True
+
+            self.state = self.NOT_SAVED
+            return False
+
+        def load(self):
+            if self.cap_file and self.seq is not None and \
+               save_sequence(self.cap_file, self.seq):
+
+                self.state = self.SAVED
+                return True
+
+            self.state = self.NOT_SAVED
+            return False
 
         def get_all_data(self):

branch/PacketManipulator/PM/Backend/Scapy/serialize.py

@@ -24 +24 @@
 
 from PM.Core.Atoms import Node
-
-from PM import Backend
+from PM.Core.Logger import log
+
+from PM.Backend.Scapy import *
 from PM.Backend import SequencePacket
 
@@ -62 +63 @@
             self.protocol.time = self.current_protocol.time
 
-            self.current_node.data.packet = Backend.MetaPacket(self.protocol)
+            self.current_node.data.packet = MetaPacket(self.protocol)
             self.protocol = self.current_protocol = None
 
@@ -92 +93 @@
                 proto_time = eval(attrs['time'])
 
-                protocol = Backend.get_proto(proto_id)()
+                protocol = get_proto(proto_id)()
                 protocol.time = proto_time
 
@@ -149 +150 @@
                 value = self.field_value
 
-            field = Backend.get_proto_field(self.current_protocol,
+            field = get_proto_field(self.current_protocol,
                                             self.field_id)
 
-            Backend.set_field_value(self.current_protocol, field, value)
+            set_field_value(self.current_protocol, field, value)
 
 class SequenceWriter(object):
@@ -237 +238 @@
 
         for proto in protocols:
-            attr_vals = {(None, u'id') : Backend.get_proto_name(proto),
+            attr_vals = {(None, u'id') : get_proto_name(proto),
                          (None, u'time') : "%.6f" % proto.time}
             attr_qnames = {(None, u'id') : u'id', (None, u'time') : u'time'}
@@ -245 +246 @@
 
 
-            for field in Backend.get_proto_fields(proto):
-                name = Backend.get_field_name(field)
-                value = Backend.get_field_value(proto, field)
+            for field in get_proto_fields(proto):
+                name = get_field_name(field)
+                value = get_field_value(proto, field)
 
                 attr_vals = {(None, u'id') : name}
@@ -262 +263 @@
             self.endElementNS((None, 'proto'), 'proto')
 
+def save_sequence(fname, sequence):
+    assert isinstance(sequence, Node)
+
+    try:
+        SequenceWriter(fname, sequence)
+    except Exception, err:
+        log.debug("Cannot save sequence to file %s (%s)" % (fname, str(err)))
+        return False
+    else:
+        return True
+
+def load_sequence(fname):
+    try:
+        tree = SequenceLoader("test.xml").tree
+        return tree
+    except Exception, err:
+        log.debug("Cannot load sequence from file %s (%s)" % (fname, str(err)))
+        return None
+
 if __name__ == "__main__":
     tree = Node()
-    first = Node(SequencePacket(Backend.MetaPacket(Backend.Ether() / Backend.IP() /  Backend.TCP())))
-    first.append_node(Node(SequencePacket(Backend.MetaPacket(Backend.IP()))))
+    first = Node(SequencePacket(MetaPacket(Ether() / IP() /  TCP())))
+    first.append_node(Node(SequencePacket(MetaPacket(IP()))))
 
     tree.append_node(first)