Changeset 5542

Timestamp:

02/18/10 12:40:17 (5 months ago)

Author:

nopper

Message:

Supporting UDP injection in IP decoder

Files:

• pm/trunk/audits/passive/ip/sources/main.py (modified) (4 diffs)

pm/trunk/audits/passive/ip/sources/main.py

@@ -53 +53 @@
 from umit.pm.manager.auditmanager import AuditManager, PassiveAudit
 from umit.pm.manager.sessionmanager import *
-from umit.pm.core.netconst import PROTO_LAYER, NET_LAYER, LL_TYPE_IP, INJ_FORWARD
+from umit.pm.core.netconst import PROTO_LAYER, NET_LAYER, LL_TYPE_IP
 from umit.pm.core.auditutils import checksum
 
@@ -253 +253 @@
 
     return ip
+
+def stateless_ip_injector(context, mpkt, length):
+    ident = IPIdent.create(mpkt)
+    sess = SessionManager().get_session(ident)
+
+    if not sess:
+        return False, length
+
+    mpkt.session = sess
+
+    injector = AuditManager().get_injector(0, LL_TYPE_IP)
+    return injector(context, mpkt, length)
 
 def ip_injector(context, mpkt, length):
@@ -305 +317 @@
         manager.add_decoder(NET_LAYER, LL_TYPE_IP, self.ip_decoder)
         manager.add_injector(0, LL_TYPE_IP, ip_injector)
+        manager.add_injector(0, STATELESS_IP_MAGIC, stateless_ip_injector)
 
     def stop(self):
@@ -310 +323 @@
         manager.remove_decoder(NET_LAYER, LL_TYPE_IP, self.ip_decoder)
         manager.remove_injector(0, LL_TYPE_IP, ip_injector)
+        manager.remove_injector(0, STATELESS_IP_MAGIC, stateless_ip_injector)
 
         self.decoder = None