Changeset 3077

Timestamp:

07/04/08 16:17:53 (5 years ago)

Author:

getxsick

Message:

Updated class Socket. Added description of ARP issue.
Updated some names of modules to the pattern.

Location:

branch/UMPA

Files:

• example.py (modified) (1 diff)
• umpa/__init__.py (modified) (1 diff)
• umpa/sockets.py (modified) (2 diffs)
• umpa/utils.py (modified) (1 diff)

branch/UMPA/example.py

@@ -38 +38 @@
 tcp1.set_flags(syn=True)
 
-# create a new packet
+# create a new packet and include one protocol (ip1)
 packet = umpa.Packet(ip1)
-# packing those protocols into our packet
+# packing another protocol into our packet
 packet.include(tcp1)
-
-# XXX: other solutions would be
-# ip1.include(tcp1)
-# packet = umpa.Packet(ip1)
 
 # creating new socket connection

branch/UMPA/umpa/__init__.py

@@ -20 +20 @@
 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 
-from packet import *
+from packets import *
 from sockets import *

branch/UMPA/umpa/sockets.py

@@ -36 +36 @@
         Just use it in any doubts.
         '''
-        #try:
+        # XXX: if non-root EUID, then Python will exit the application
+        # and report the error
+        # TODO: try/except section + trying to switch EUID into root and recall socket()
         self._sock = socket(AF_INET, SOCK_RAW, IPPROTO_RAW)
-        utils.leave_priviliges()    # dropping root-priviliges
-        #except error, msg:
-        #    sys.stderr.write('Error while opening new socket:\n')
-        #    sys.stderr.write(str(msg) + '\n')
-        #    sys.stderr.write('Leaving...')
-        #    sys.exit(1)
+        # dropping root-priviliges
+        utils.drop_priviliges()    
+        # to build own headers of IP
+        self._sock.setsockopt(IPPROTO_IP, IP_HDRINCL, 1)
 
     def send(self, *packets):
@@ -49 +49 @@
 
         for packet in packets:
-            self._sock.send(packet.get_raw())
+            # XXX: this connects to a remote socket, so destination address has to be known.
+            # the better solution would be to use ARP (the lowest software layer from OSI).
+            # it will be implemented when ARP protocol will be implemented ;)
+            # so now we have to parse the packet for destination address
+            dst_addr = self._get_address(packet)
+            self._sock.sendto(packet.get_raw(), (dst_addr,0) )
+
+    def _get_address(self, packet):
+        '''Because of ARP issue (described in send() method,
+        we have to parse packets for destination addresses.
+        '''
+        print "Not implemented yet."
+        return False

branch/UMPA/umpa/utils.py

@@ -23 +23 @@
 import pwd
 
-def leave_priviliges():
+def drop_priviliges():
     '''Some functions require root-privilegies and after we done them,
     we don't really need this privilegies. So, we can simple leave them.