1import binascii 2import sys 3 4PY26 = sys.version_info[0] == 2 and sys.version_info[1] <= 6 5 6if PY26: 7 import struct 8 9 10class PgpdumpException(Exception): 11 '''Base exception class raised by any parsing errors, etc.''' 12 pass 13 14 15# 256 values corresponding to each possible byte 16CRC24_TABLE = ( 17 0x000000, 0x864cfb, 0x8ad50d, 0x0c99f6, 0x93e6e1, 0x15aa1a, 0x1933ec, 18 0x9f7f17, 0xa18139, 0x27cdc2, 0x2b5434, 0xad18cf, 0x3267d8, 0xb42b23, 19 0xb8b2d5, 0x3efe2e, 0xc54e89, 0x430272, 0x4f9b84, 0xc9d77f, 0x56a868, 20 0xd0e493, 0xdc7d65, 0x5a319e, 0x64cfb0, 0xe2834b, 0xee1abd, 0x685646, 21 0xf72951, 0x7165aa, 0x7dfc5c, 0xfbb0a7, 0x0cd1e9, 0x8a9d12, 0x8604e4, 22 0x00481f, 0x9f3708, 0x197bf3, 0x15e205, 0x93aefe, 0xad50d0, 0x2b1c2b, 23 0x2785dd, 0xa1c926, 0x3eb631, 0xb8faca, 0xb4633c, 0x322fc7, 0xc99f60, 24 0x4fd39b, 0x434a6d, 0xc50696, 0x5a7981, 0xdc357a, 0xd0ac8c, 0x56e077, 25 0x681e59, 0xee52a2, 0xe2cb54, 0x6487af, 0xfbf8b8, 0x7db443, 0x712db5, 26 0xf7614e, 0x19a3d2, 0x9fef29, 0x9376df, 0x153a24, 0x8a4533, 0x0c09c8, 27 0x00903e, 0x86dcc5, 0xb822eb, 0x3e6e10, 0x32f7e6, 0xb4bb1d, 0x2bc40a, 28 0xad88f1, 0xa11107, 0x275dfc, 0xdced5b, 0x5aa1a0, 0x563856, 0xd074ad, 29 0x4f0bba, 0xc94741, 0xc5deb7, 0x43924c, 0x7d6c62, 0xfb2099, 0xf7b96f, 30 0x71f594, 0xee8a83, 0x68c678, 0x645f8e, 0xe21375, 0x15723b, 0x933ec0, 31 0x9fa736, 0x19ebcd, 0x8694da, 0x00d821, 0x0c41d7, 0x8a0d2c, 0xb4f302, 32 0x32bff9, 0x3e260f, 0xb86af4, 0x2715e3, 0xa15918, 0xadc0ee, 0x2b8c15, 33 0xd03cb2, 0x567049, 0x5ae9bf, 0xdca544, 0x43da53, 0xc596a8, 0xc90f5e, 34 0x4f43a5, 0x71bd8b, 0xf7f170, 0xfb6886, 0x7d247d, 0xe25b6a, 0x641791, 35 0x688e67, 0xeec29c, 0x3347a4, 0xb50b5f, 0xb992a9, 0x3fde52, 0xa0a145, 36 0x26edbe, 0x2a7448, 0xac38b3, 0x92c69d, 0x148a66, 0x181390, 0x9e5f6b, 37 0x01207c, 0x876c87, 0x8bf571, 0x0db98a, 0xf6092d, 0x7045d6, 0x7cdc20, 38 0xfa90db, 0x65efcc, 0xe3a337, 0xef3ac1, 0x69763a, 0x578814, 0xd1c4ef, 39 0xdd5d19, 0x5b11e2, 0xc46ef5, 0x42220e, 0x4ebbf8, 0xc8f703, 0x3f964d, 40 0xb9dab6, 0xb54340, 0x330fbb, 0xac70ac, 0x2a3c57, 0x26a5a1, 0xa0e95a, 41 0x9e1774, 0x185b8f, 0x14c279, 0x928e82, 0x0df195, 0x8bbd6e, 0x872498, 42 0x016863, 0xfad8c4, 0x7c943f, 0x700dc9, 0xf64132, 0x693e25, 0xef72de, 43 0xe3eb28, 0x65a7d3, 0x5b59fd, 0xdd1506, 0xd18cf0, 0x57c00b, 0xc8bf1c, 44 0x4ef3e7, 0x426a11, 0xc426ea, 0x2ae476, 0xaca88d, 0xa0317b, 0x267d80, 45 0xb90297, 0x3f4e6c, 0x33d79a, 0xb59b61, 0x8b654f, 0x0d29b4, 0x01b042, 46 0x87fcb9, 0x1883ae, 0x9ecf55, 0x9256a3, 0x141a58, 0xefaaff, 0x69e604, 47 0x657ff2, 0xe33309, 0x7c4c1e, 0xfa00e5, 0xf69913, 0x70d5e8, 0x4e2bc6, 48 0xc8673d, 0xc4fecb, 0x42b230, 0xddcd27, 0x5b81dc, 0x57182a, 0xd154d1, 49 0x26359f, 0xa07964, 0xace092, 0x2aac69, 0xb5d37e, 0x339f85, 0x3f0673, 50 0xb94a88, 0x87b4a6, 0x01f85d, 0x0d61ab, 0x8b2d50, 0x145247, 0x921ebc, 51 0x9e874a, 0x18cbb1, 0xe37b16, 0x6537ed, 0x69ae1b, 0xefe2e0, 0x709df7, 52 0xf6d10c, 0xfa48fa, 0x7c0401, 0x42fa2f, 0xc4b6d4, 0xc82f22, 0x4e63d9, 53 0xd11cce, 0x575035, 0x5bc9c3, 0xdd8538 54) 55 56 57def crc24(data): 58 '''Implementation of the CRC-24 algorithm used by OpenPGP.''' 59 # CRC-24-Radix-64 60 # x24 + x23 + x18 + x17 + x14 + x11 + x10 + x7 + x6 61 # + x5 + x4 + x3 + x + 1 (OpenPGP) 62 # 0x864CFB / 0xDF3261 / 0xC3267D 63 crc = 0x00b704ce 64 # this saves a bunch of slower global accesses 65 crc_table = CRC24_TABLE 66 for byte in data: 67 tbl_idx = ((crc >> 16) ^ byte) & 0xff 68 crc = (crc_table[tbl_idx] ^ (crc << 8)) & 0x00ffffff 69 return crc 70 71 72def get_int2(data, offset): 73 '''Pull two bytes from data at offset and return as an integer.''' 74 return (data[offset] << 8) + data[offset + 1] 75 76 77def get_int4(data, offset): 78 '''Pull four bytes from data at offset and return as an integer.''' 79 return ((data[offset] << 24) + (data[offset + 1] << 16) + 80 (data[offset + 2] << 8) + data[offset + 3]) 81 82 83def get_int8(data, offset): 84 '''Pull eight bytes from data at offset and return as an integer.''' 85 return (get_int4(data, offset) << 32) + get_int4(data, offset + 4) 86 87 88def get_mpi(data, offset): 89 '''Gets a multi-precision integer as per RFC-4880. 90 Returns the MPI and the new offset. 91 See: http://tools.ietf.org/html/rfc4880#section-3.2''' 92 mpi_len = get_int2(data, offset) 93 offset += 2 94 to_process = (mpi_len + 7) // 8 95 mpi = 0 96 i = -4 97 for i in range(0, to_process - 3, 4): 98 mpi <<= 32 99 mpi += get_int4(data, offset + i) 100 for j in range(i + 4, to_process): 101 mpi <<= 8 102 mpi += data[offset + j] 103 # Python 3.2 and later alternative: 104 #mpi = int.from_bytes(data[offset:offset + to_process], byteorder='big') 105 offset += to_process 106 return mpi, offset 107 108 109def get_hex_data(data, offset, byte_count): 110 '''Pull the given number of bytes from data at offset and return as a 111 hex-encoded string.''' 112 key_data = data[offset:offset + byte_count] 113 if PY26: 114 key_data = buffer(key_data) 115 key_id = binascii.hexlify(key_data) 116 return key_id.upper() 117 118 119def get_key_id(data, offset): 120 '''Pull eight bytes from data at offset and return as a 16-byte hex-encoded 121 string.''' 122 return get_hex_data(data, offset, 8) 123 124 125def get_int_bytes(data): 126 '''Get the big-endian byte form of an integer or MPI.''' 127 hexval = '%X' % data 128 new_len = (len(hexval) + 1) // 2 * 2 129 hexval = hexval.zfill(new_len) 130 return binascii.unhexlify(hexval.encode('ascii')) 131 132 133def pack_data(data): 134 '''Pack iterable of binary data into a bytestring if necessary.''' 135 if PY26: 136 return struct.pack('%dB' % len(data), *data) 137 return data 138 139def same_key(key_a, key_b): 140 '''Comparison function for key ID or fingerprint strings, taking into 141 account varying length.''' 142 if len(key_a) == len(key_b): 143 return key_a == key_b 144 elif len(key_a) < len(key_b): 145 return key_b.endswith(key_a) 146 else: 147 return key_a.endswith(key_b) 148