packing: Do use extra arguments in p* and u*

Author: tesuji

pwnlib/util/packing.py

@@ -332,7 +332,7 @@ def routine(data, stacklevel=None):
 #
 # Make normal user-oriented packers, e.g. p8
 #
-def _do_packing(op, size, number):
+def _do_packing(op, size, number, endianness=None, sign=None):
 
     name = "%s%s" % (op,size)
     mod = sys.modules[__name__]
@@ -342,8 +342,8 @@ def _do_packing(op, size, number):
     bs = getattr(mod, "_%sbs" % (name))
     bu = getattr(mod, "_%sbu" % (name))
 
-    endian = context.endian
-    signed = context.signed
+    endian = endianness or context.endian
+    signed = sign or context.signed
     return {("little", True ):  ls,
             ("little", False):  lu,
             ("big",    True ):  bs,
@@ -365,7 +365,7 @@ def p8(number, endianness = None, sign = None, **kwargs):
     Returns:
         The packed number as a byte string
     """
-    return _do_packing('p', 8, number)
+    return _do_packing('p', 8, number, endianness, sign)
 
 @LocalNoarchContext
 def p16(number, endianness = None, sign = None, **kwargs):
@@ -382,8 +382,15 @@ def p16(number, endianness = None, sign = None, **kwargs):
 
     Returns:
         The packed number as a byte string
+
+    Examples:
+
+        >>> p16(0x4142, 'big')
+        b'AB'
+        >>> p16(0x4142, endianness='big')
+        b'AB'
     """
-    return _do_packing('p', 16, number)
+    return _do_packing('p', 16, number, endianness, sign)
 
 @LocalNoarchContext
 def p32(number, endianness = None, sign = None, **kwargs):
@@ -400,8 +407,15 @@ def p32(number, endianness = None, sign = None, **kwargs):
 
     Returns:
         The packed number as a byte string
+
+    Examples:
+
+        >>> p32(0x41424344, 'big')
+        b'ABCD'
+        >>> p32(0x41424344, endianness='big')
+        b'ABCD'
     """
-    return _do_packing('p', 32, number)
+    return _do_packing('p', 32, number, endianness, sign)
 
 @LocalNoarchContext
 def p64(number, endianness = None, sign = None, **kwargs):
@@ -418,8 +432,15 @@ def p64(number, endianness = None, sign = None, **kwargs):
 
     Returns:
         The packed number as a byte string
+
+    Examples:
+
+        >>> p64(0x4142434445464748, 'big')
+        b'ABCDEFGH'
+        >>> p64(0x4142434445464748, endianness='big')
+        b'ABCDEFGH'
     """
-    return _do_packing('p', 64, number)
+    return _do_packing('p', 64, number, endianness, sign)
 
 @LocalNoarchContext
 def u8(data, endianness = None, sign = None, **kwargs):
@@ -437,7 +458,7 @@ def u8(data, endianness = None, sign = None, **kwargs):
     Returns:
         The unpacked number
     """
-    return _do_packing('u', 8, data)
+    return _do_packing('u', 8, data, endianness, sign)
 
 @LocalNoarchContext
 def u16(data, endianness = None, sign = None, **kwargs):
@@ -455,7 +476,7 @@ def u16(data, endianness = None, sign = None, **kwargs):
     Returns:
         The unpacked number
     """
-    return _do_packing('u', 16, data)
+    return _do_packing('u', 16, data, endianness, sign)
 
 @LocalNoarchContext
 def u32(data, endianness = None, sign = None, **kwargs):
@@ -473,7 +494,7 @@ def u32(data, endianness = None, sign = None, **kwargs):
     Returns:
         The unpacked number
     """
-    return _do_packing('u', 32, data)
+    return _do_packing('u', 32, data, endianness, sign)
 
 @LocalNoarchContext
 def u64(data, endianness = None, sign = None, **kwargs):
@@ -491,7 +512,7 @@ def u64(data, endianness = None, sign = None, **kwargs):
     Returns:
         The unpacked number
     """
-    return _do_packing('u', 64, data)
+    return _do_packing('u', 64, data, endianness, sign)
 
 def make_packer(word_size = None, sign = None, **kwargs):
     """make_packer(word_size = None, endianness = None, sign = None) -> number → str
@@ -775,7 +796,7 @@ def flat(*args, **kwargs):
     Examples:
 
         (Test setup, please ignore)
-    
+
         >>> context.clear()
 
         Basic usage of :meth:`flat` works similar to the pack() routines.
@@ -822,7 +843,7 @@ def flat(*args, **kwargs):
 
         Dictionary usage permits directly using values derived from :func:`.cyclic`.
         See :func:`.cyclic`, :function:`pwnlib.context.context.cyclic_alphabet`, and :data:`.context.cyclic_size`
-        for more options.  
+        for more options.
 
         The cyclic pattern can be provided as either the text or hexadecimal offset.
 
@@ -873,7 +894,7 @@ def flat(*args, **kwargs):
 
         Negative indices are also supported, though this only works for integer
         keys.
-    
+
         >>> flat({-4: b'x', -1: b'A', 0: b'0', 4: b'y'})
         b'xaaA0aaay'
     """