Changeset 3088

Timestamp:

07/05/08 16:15:14 (5 years ago)

Author:

getxsick

Message:

Updated IP class. Extended super class Protocol. Added new exception UMPAAttributeException.

Location:

branch/UMPA/umpa

Files:

• protocols/IP.py (modified) (1 diff)
• protocols/base.py (modified) (3 diffs)
• utils/my_exceptions.py (modified) (1 diff)

branch/UMPA/umpa/protocols/IP.py

@@ -21 +21 @@
 
 import base
+from utils.my_exceptions import UMPAAttributeException
+
+class HVersion(base.Field):
+    def fillout(self):
+        pass
+
+class HIHL(base.Field):
+    def fillout(self):
+        pass
+
+class HTypeOfService(base.Field):
+    def fillout(self):
+        pass
+
+class HTotalLength(base.Field):
+    def fillout(self):
+        pass
+
+class HIdentification(base.Field):
+    def fillout(self):
+        pass
+
+class HFlags(base.Field):
+    def fillout(self):
+        pass
+
+class HFragmentOffset(base.Field):
+    def fillout(self):
+        pass
+
+class HTimeToLive(base.Field):
+    def fillout(self):
+        pass
+
+class HProtocol(base.Field):
+    def fillout(self):
+        pass
+
+class HHeaderChecksum(base.Field):
+    def fillout(self):
+        pass
+
+class HSourceAddress(base.Field):
+    def fillout(self):
+        pass
+
+class HDestinationAddress(base.Field):
+    def fillout(self):
+        pass
+
+class HOptions(base.Field):
+    def fillout(self):
+        pass
+
+class HPadding(base.Field):
+    def fillout(self):
+        pass
+
+# main IP class
 
 class IP(base.Protocol):
+    valid_fields = ['_version', '_ihl', 'type_of_service', '_total_length',
+                    'identification', 'flags', '_fragment_offset', 'time_to_live',
+                    'protocol', '_header_checksum', 'source_address',
+                    'destination_address', 'options', '_padding',]
+
     def __init__(self, **kw):
+        base.Protocol.__init__(kw)
+
         # attributes listed below shouldn't be modifed by user
         # they will be generated automatically
-        self._version = None
-        self._ihl = None
-        self._total_length = None
-        self._fragment_offset = None
-        self._header_checksum = None
-        self._padding = None
+        self._fields = [ HVersion(4, True), HIHL(4, True),
+                        HTypeOfService(8), HTotalLength(16, True),
+                        HIdentification(16, True), HFlags(3, True),
+                        HFragmentOffset(13, True), HTimeToLive(8),
+                        HProtocol(8, True), HHeaderChecksum(16, True),
+                        HSourceAddress(16), HDestinationAddress(16),
+                        HOptions(0), HPadding(0, True), ]
 
         # setting up passed fields
         for field in kw:
-            setattr(self, field, kw[field])
+            self.__setattr__(field, kw[field])
 
-    # fields
 
-    def set_type_of_service(self, val):
-        pass
-    def get_type_of_service(self):
-        return self._type_of_service
-    type_of_service = property(get_type_of_service, set_type_of_service)
+    def _is_valid(self, name):
+        '''Check if attribute is allowed.'''
+        if name in valid_fields:
+            return True
+        return False
 
-    def set_identification(self, val):
-        pass
-    def get_identification(self):
-        return self._identification
-    identification = property(get_identification, set_identification)
+    def __setattr__(self, attr, val):
+        '''Set value of the field.'''
+        if self._is_valid(attr):
+            self._fields[valid_fields.index(attr)].set(val)
+        else:
+            raise UMPAAttributeException, attr + ' not allowed'
 
-    def set_time_to_live(self, val):
-        pass
-    def get_time_to_live(self):
-        return self._time_to_live
-    time_to_live = property(get_time_to_live, set_time_to_live)
-
-    def set_protocol(self, val):
-        pass
-    def get_protocol(self):
-        return self._protocol
-    protocol = property(get_protocol, set_protocol)
-
-    def set_source_address(self, val):
-        pass
-    def get_source_address(self):
-        return self._source_address
-    source_address = property(get_source_address, set_source_address)
-
-    def set_destination_address(self, val):
-        pass
-    def get_destination_address(self):
-        return self._destination_address
-    destination_address = property(get_destination_address,
-            set_destination_address)
-
-    def set_option(self, val):
-        pass
-    def get_option(self):
-        return self._option
-    option = property(get_option, set_option)
+    def __getattr__(self, attr):
+        '''Return value of the field.'''
+        if self._is_valid(attr):
+            return self._fields[valid_fields.index(attr)].get()
+        else:
+            raise UMPAAttributeException, attr + ' not allowed'

branch/UMPA/umpa/protocols/base.py

@@ -22 +22 @@
 import utils
 
-#class Field:
-#    def set(self, val):
-#        if self.is_valid(val):
-#            self.val = val
-#
-#    def is_valid(self, val):
-#        return True
+class Field(object):
+    def __init__(self, bits, auto=False):
+        '''Set auto if you wish to take care about the field
+        by the library. Then you will have to write how
+        to manage the field.
+        '''
+        self._auto = auto
+        self._bits = bits
+        self._value = None   # default value of the field
 
-class Protocol:
-    _valid_fields = []
+    def set(self, value):
+        if self.is_valid(value):
+            self.value = value
+
+    def is_valid(self, val):
+        '''Should be overload by sub-classes.
+
+        Otherwise always return true.
+        '''
+        return True
+
+class Protocol(object):
+    def __init__(self, **kw):
+        self._fields = []
 
     def set_fields(self, *args, **kwargs):
@@ -41 +55 @@
 
         for key in kwargs:
+            if self._is_valid(key):
+                setattr(self, key, kwargs[key])
             self.fields[key].set(kwargs[key])
 
@@ -48 +64 @@
         return False
 
-    def set_flags(self, val):
-        pass
+    def _is_valid(self, field):
+        '''Overload it in subclasses.'''
+        raise NotImplementedError
 
-    def get_flags(self):
-        return self._Flags
-
-    flags = property(get_flags, set_flags)
+    #def set_flags(self, val):
+    #    pass
+    #def get_flags(self):
+    #    return self._Flags
+    #flags = property(get_flags, set_flags)
 
 class Layer4(Protocol):

branch/UMPA/umpa/utils/my_exceptions.py

@@ -25 +25 @@
     def __str__(self):
         return repr(self.msg)
+
+class UMPAAttributeException(UMPAException):
+    pass