示例#1
0
文件: Packet.py 项目: Zaryob/wubi
    def get_size(self, d):
        """Get total octet count of a body specified by new length header octs.

        :Parameters:
            - `d`: string of "new" length octets, note that this string
              cannot be made up of arbitrary bytes, and must conform
              to the rules of new length values (rfc2440 4.2.2)

        :Returns: XXXXXXX TODO

        Example:

            >>> newlength_octets = [0xff, 0x92, 0xdf, 0x7c, 0xbb]
            >>> newlength_data_string = ''.join(map(chr, length_octet_list))
            >>> newlength = NewLength()
            >>> newlength.get_size(newlength_data_string)
            2464119995
        """
        idx = size = 0
        # the length_list is to keep track of data (d) actually used -
        # it's possible that a string of partials could terminate with
        # leftover octets
        length_list = []
        if 0 < len(d):
            for L in [(x, ord(x)) for x in d]:
                if L[1] < 192:
                    size = size + L[1]
                    length_list.append(L[0])
                    break
                elif 192 <= L[1] <= 223:
                    slice = d[idx:idx + 2]
                    s = STN.doubleoct2int(slice)
                    if 192 <= s <= 8383:  # ?? dunno about this restriction
                        size = size + s
                        length_list.append(slice)
                    else:
                        raise PGPFormatError(
                            "New double octet lengths are confined to the range 192 <= x <= 8383. Received: data->(%s) length->(%s)"
                            % (d, size))
                    break
                elif 255 == L[1]:
                    slice = d[idx:idx + 5]
                    size = size + STN.pentoct2int(slice)
                    length_list.append(slice)
                    break
                elif 224 <= L[1] <= 254:  # partials
                    slice = d[idx:idx + 1]
                    size = size + STN.partial2int(slice)
                    length_list.append(slice)
                    idx = idx + 1
        else:
            raise PGPFormatError(
                "Length must have at least one octet. Received len(d) == 0.")

        return size, length_list
示例#2
0
    def get_size(self, d):
        """Get total octet count of a body specified by new length header octs.

        :Parameters:
            - `d`: string of "new" length octets, note that this string
              cannot be made up of arbitrary bytes, and must conform
              to the rules of new length values (rfc2440 4.2.2)

        :Returns: XXXXXXX TODO
        
        Example:

            >>> newlength_octets = [0xff, 0x92, 0xdf, 0x7c, 0xbb] 
            >>> newlength_data_string = ''.join(map(chr, length_octet_list))
            >>> newlength = NewLength()
            >>> newlength.get_size(newlength_data_string)
            2464119995
        """
        idx = size = 0
        # the length_list is to keep track of data (d) actually used -
        # it's possible that a string of partials could terminate with
        # leftover octets
        length_list = []
        if 0 < len(d):
            for L in [(x, ord(x)) for x in d]:
                if L[1] < 192:
                    size = size + L[1]
                    length_list.append(L[0])
                    break
                elif 192 <= L[1] <= 223:
                    slice = d[idx:idx+2]
                    s = STN.doubleoct2int(slice)
                    if 192 <= s <= 8383: # ?? dunno about this restriction
                        size = size + s
                        length_list.append(slice)
                    else:
                        raise PGPFormatError("New double octet lengths are confined to the range 192 <= x <= 8383. Received: data->(%s) length->(%s)" % (d, size))
                    break
                elif 255 == L[1]:
                    slice = d[idx:idx+5]
                    size = size + STN.pentoct2int(slice)
                    length_list.append(slice)
                    break
                elif 224 <= L[1] <= 254: # partials
                    slice = d[idx:idx+1]
                    size = size + STN.partial2int(slice)
                    length_list.append(slice)
                    idx = idx + 1
        else:
            raise PGPFormatError("Length must have at least one octet. Received len(d) == 0.")

        return size, length_list
 def testD2Sanity(self):
     """strnum: double octet inversion (int2doubleoct/doubleoct2int)"""
     for o in self.good_double_octs:
         self.assertEqual(o[0], doubleoct2int(int2doubleoct(o[0])))
 def testC6OctTranslation(self):
     """strnum: doubleoct2int output"""
     for v in self.good_double_octs:
         self.assertEqual(v[0], doubleoct2int(v[1]))
示例#5
0
 def testD2Sanity(self):
     """strnum: double octet inversion (int2doubleoct/doubleoct2int)"""
     for o in self.good_double_octs:
         self.assertEqual(o[0], doubleoct2int(int2doubleoct(o[0])))
示例#6
0
 def testC6OctTranslation(self):
     """strnum: doubleoct2int output"""
     for v in self.good_double_octs:
         self.assertEqual(v[0], doubleoct2int(v[1]))
示例#7
0
文件: Signature.py 项目: Zaryob/wubi
    def fill(self, d):
        ord_d = ord(d[0])

        if ord_d < 192:
            slice = d[0]
            size = ord_d

        elif 192 <= ord_d < 255:
            slice = d[:2]
            #size = size + STN.doubleoct2int(slice)
            size = STN.doubleoct2int(slice)

        elif 255 == ord_d:
            slice = d[:5]
            #size = size + STN.pentoct2int(slice)
            size = STN.pentoct2int(slice)

        len_slice = len(slice)
        self._d = d[:len_slice + size]
        type_d = d[len_slice:len_slice + 1]
        self.type = 127 & ord(type_d)
        self.critical = 128 & ord(type_d)
        value_d = d[len_slice + 1:len_slice + size]

        if SIGSUB_SIGNERID == self.type:
            self.value = STN.str2hex(value_d[:8])

        elif self.type in [SIGSUB_CREATED, SIGSUB_EXPIRES, SIGSUB_KEYEXPIRES]:
            self.value = STN.str2int(value_d[:4])
            self.value = STN.str2int(value_d)

        elif self.type in [
                SIGSUB_EXPORTABLE, SIGSUB_REVOCABLE, SIGSUB_PRIMARYUID
        ]:
            self.value = STN.str2int(value_d[:1])

            if self.value not in [0, 1]:
                raise SignatureSubpacketValueError, "Subpacket (# %s) value must be 0 or 1." % (
                    str(subtype))

        elif SIGSUB_TRUST == self.type:  # level, amount
            self.value = (STN.str2int(value_d[0]), STN.str2int(value_d[1]))

        elif self.type in [
                SIGSUB_SYMCODE, SIGSUB_HASHCODE, SIGSUB_COMPCODE,
                SIGSUB_KEYSERVPREFS, SIGSUB_KEYFLAGS, SIGSUB_FEATURES
        ]:
            self.value = [ord(x) for x in value_d]

        elif SIGSUB_REVOKER == self.type:
            cls = STN.str2int(value_d[0])
            alg = STN.str2int(value_d[1])
            fprint = STN.str2hex(value_d[2:22])
            self.value = (cls, alg, fprint)

        elif SIGSUB_NOTE == self.type:
            flags = [STN.str2int(x)
                     for x in value_d[:4]]  # first four flag octs
            name_len = STN.str2int(value_d[4:6])
            val_len = STN.str2int(value_d[6:8])
            nam = value_d[8:8 + name_len]
            val = value_d[8 + name_len:8 + name_len + val_len]
            self.value = (flags, nam, val)

        elif self.type in [
                SIGSUB_KEYSERV, SIGSUB_POLICYURL, SIGSUB_SIGNERUID,
                SIGSUB_REGEX
        ]:
            self.value = value_d

        elif SIGSUB_REVOCREASON == self.type:  # code, reason
            self.value = (STN.str2int(value_d[0]), value_d[1:])

        elif SIGSUB_SIGTARGET == self.type:
            raise NotImplementedError, "SIGTARGET not supported"

        else:  # the subpacket has an unknown type, so just pack the data in
            self.value = value_d
示例#8
0
    def fill(self, d):
        ord_d = ord(d[0])

        if ord_d < 192:
            slice = d[0]
            size = ord_d

        elif 192 <= ord_d < 255:
            slice = d[:2]
            #size = size + STN.doubleoct2int(slice)
            size = STN.doubleoct2int(slice)

        elif 255 == ord_d:
            slice = d[:5]
            #size = size + STN.pentoct2int(slice)
            size = STN.pentoct2int(slice)

        len_slice = len(slice)
        self._d = d[:len_slice+size]
        type_d = d[len_slice:len_slice+1]
        self.type = 127 & ord(type_d)
        self.critical = 128 & ord(type_d)
        value_d = d[len_slice+1:len_slice+size]

        if SIGSUB_SIGNERID == self.type:
            self.value = STN.str2hex(value_d[:8])

        elif self.type in [SIGSUB_CREATED, SIGSUB_EXPIRES, SIGSUB_KEYEXPIRES]:
            self.value = STN.str2int(value_d[:4])            
            self.value = STN.str2int(value_d)            

        elif self.type in [SIGSUB_EXPORTABLE, SIGSUB_REVOCABLE, SIGSUB_PRIMARYUID]:
            self.value = STN.str2int(value_d[:1])

            if self.value not in [0, 1]:
                raise SignatureSubpacketValueError, "Subpacket (# %s) value must be 0 or 1." % (str(subtype))

        elif SIGSUB_TRUST == self.type: # level, amount
            self.value = (STN.str2int(value_d[0]), STN.str2int(value_d[1]))

        elif self.type in [SIGSUB_SYMCODE, SIGSUB_HASHCODE, SIGSUB_COMPCODE, SIGSUB_KEYSERVPREFS, SIGSUB_KEYFLAGS, SIGSUB_FEATURES]:
            self.value = [ord(x) for x in value_d]

        elif SIGSUB_REVOKER == self.type:
            cls = STN.str2int(value_d[0])
            alg = STN.str2int(value_d[1])
            fprint = STN.str2hex(value_d[2:22])
            self.value = (cls, alg, fprint)

        elif SIGSUB_NOTE == self.type:
            flags = [STN.str2int(x) for x in value_d[:4]] # first four flag octs
            name_len = STN.str2int(value_d[4:6])
            val_len = STN.str2int(value_d[6:8])
            nam = value_d[8:8+name_len]
            val = value_d[8+name_len:8+name_len+val_len]
            self.value = (flags, nam, val)

        elif self.type in [SIGSUB_KEYSERV, SIGSUB_POLICYURL, SIGSUB_SIGNERUID, SIGSUB_REGEX]:
            self.value = value_d

        elif SIGSUB_REVOCREASON == self.type: # code, reason
            self.value = (STN.str2int(value_d[0]), value_d[1:])

        elif SIGSUB_SIGTARGET == self.type:
            raise NotImplementedError, "SIGTARGET not supported"

        else: # the subpacket has an unknown type, so just pack the data in
            self.value = value_d