set getRawParam and setRawParam default argbits to 0

Author: elicn

qiling/cc/__init__.py

@@ -26,7 +26,7 @@ def getNumSlots(argbits: int) -> int:
 
 		raise NotImplementedError
 
-	def getRawParam(self, slot: int, argbits: int = None) -> int:
+	def getRawParam(self, slot: int, argbits: int = 0) -> int:
 		"""Read a value of native size from the specified argument slot.
 
 		Note that argument slots and argument indexes are not the same. Though they often correlate
@@ -41,7 +41,7 @@ def getRawParam(self, slot: int, argbits: int = None) -> int:
 
 		raise NotImplementedError
 
-	def setRawParam(self, slot: int, value: int, argbits: int = None) -> None:
+	def setRawParam(self, slot: int, value: int, argbits: int = 0) -> None:
 		"""Replace the value in the specified argument slot.
 
 		Note that argument slots and argument indexes are not the same. Though they often correlate
@@ -148,17 +148,17 @@ def __access_param(self, index: int, stack_access: Callable, reg_access: Callabl
 		else:
 			return reg_access, reg
 
-	def getRawParam(self, index: int, argbits: int = None) -> int:
+	def getRawParam(self, index: int, argbits: int = 0) -> int:
 		read, loc = self.__access_param(index, self.arch.stack_read, self.arch.regs.read)
 
-		mask = (0 if argbits is None else (1 << argbits)) - 1
+		mask = (argbits and (1 << argbits)) - 1
 
 		return read(loc) & mask
 
-	def setRawParam(self, index: int, value: int, argbits: int = None) -> None:
+	def setRawParam(self, index: int, value: int, argbits: int = 0) -> None:
 		write, loc = self.__access_param(index, self.arch.stack_write, self.arch.regs.write)
 
-		mask = (0 if argbits is None else (1 << argbits)) - 1
+		mask = (argbits and (1 << argbits)) - 1
 
 		write(loc, value & mask)
 

qiling/cc/intel.py

@@ -42,7 +42,7 @@ class QlIntel32(QlIntelBaseCC):
 	def getNumSlots(argbits: int) -> int:
 		return max(argbits, 32) // 32
 
-	def getRawParam(self, slot: int, nbits: int = None) -> int:
+	def getRawParam(self, slot: int, nbits: int = 0) -> int:
 		__super_getparam = super().getRawParam
 
 		if nbits == 64:

qiling/os/fcall.py

@@ -32,8 +32,8 @@ def __init__(self, ql: Qiling, cc: QlCC, accessors: Mapping[int, Accessor] = {})
 		self.cc = cc
 
 		def __make_accessor(nbits: int) -> Accessor:
-			reader = lambda si: cc.getRawParam(si, nbits or None)
-			writer = lambda si, val: cc.setRawParam(si, val, nbits or None)
+			reader = lambda si: cc.getRawParam(si, nbits)
+			writer = lambda si, val: cc.setRawParam(si, val, nbits)
 			nslots = cc.getNumSlots(nbits)
 
 			return (reader, writer, nslots)