1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
|
## This file is part of Scapy
## See http://www.secdev.org/projects/scapy for more informations
## Copyright (C) Philippe Biondi <phil@secdev.org>
## Copyright (C) Gabriel Potter <gabriel@potter.fr>
## This program is published under a GPLv2 license
"""
Python 2 and 3 link classes.
"""
from __future__ import absolute_import
import base64
import binascii
import scapy.modules.six as six
###########
# Python3 #
###########
def cmp_to_key(mycmp):
# TODO remove me once all 'key=cmp_to_key(..)' has been fixed in utils6.py, automaton.py
"""Convert a cmp= function into a key= function.
To use with sort()
e.g: def stg_cmp(a, b):
return a == b
list.sort(key=cmp_to_key(stg_cmp))
"""
class K(object):
def __init__(self, obj, *args):
self.obj = obj
def __lt__(self, other):
return mycmp(self.obj, other.obj) < 0
def __gt__(self, other):
return mycmp(self.obj, other.obj) > 0
def __eq__(self, other):
return mycmp(self.obj, other.obj) == 0
def __le__(self, other):
return mycmp(self.obj, other.obj) <= 0
def __ge__(self, other):
return mycmp(self.obj, other.obj) >= 0
def __ne__(self, other):
return mycmp(self.obj, other.obj) != 0
return K
def cmp(a, b):
"""Old Python 2 function"""
return (a > b) - (a < b)
if six.PY2:
def orb(x):
"""Return ord(x) when necessary."""
if isinstance(x, basestring):
return ord(x)
return x
else:
def orb(x):
"""Return ord(x) when necessary."""
if isinstance(x, (bytes, str)):
return ord(x)
return x
if six.PY2:
def raw(x):
"""Convert a str, a packet to bytes"""
if x is None:
return None
if hasattr(x, "__bytes__"):
return x.__bytes__()
try:
return chr(x)
except (ValueError, TypeError):
return str(x)
def plain_str(x):
"""Convert basic byte objects to str"""
return x
def chb(x):
"""Same than chr() but encode as bytes.
"""
if isinstance(x, bytes):
return x
else:
if hasattr(x, "__int__") and not isinstance(x, int):
return bytes(chr(int(x)))
return bytes(chr(x))
else:
def raw(x):
"""Convert a str, an int, a list of ints, a packet to bytes"""
try:
return bytes(x)
except TypeError:
return bytes(x, encoding="utf8")
def plain_str(x):
"""Convert basic byte objects to str"""
if isinstance(x, bytes):
return x.decode('utf8')
return x
def chb(x):
"""Same than chr() but encode as bytes.
"""
if isinstance(x, bytes):
return x
else:
if hasattr(x, "__int__") and not isinstance(x, int):
return bytes([int(x)])
return bytes([x])
def bytes_hex(x):
"""Hexify a str or a bytes object"""
return binascii.b2a_hex(raw(x))
def hex_bytes(x):
"""De-hexify a str or a byte object"""
return binascii.a2b_hex(raw(x))
def base64_bytes(x):
"""Turn base64 into bytes"""
if six.PY2:
return base64.decodestring(x)
return base64.decodebytes(raw(x))
def bytes_base64(x):
"""Turn bytes into base64"""
if six.PY2:
return base64.encodestring(x).replace('\n', '')
return base64.encodebytes(raw(x)).replace(b'\n', b'')
|
