add RETURN_CONST as an allowed _const_code in safeeval

erikleffler · erikleffler · commit 49bcc148d6a1 · 2024-02-18T17:55:10.000+01:00
diff --git a/pwnlib/util/safeeval.py b/pwnlib/util/safeeval.py
@@ -6,7 +6,7 @@
     'BUILD_CONST_KEY_MAP', 'BUILD_STRING',
     'LOAD_CONST','RETURN_VALUE','STORE_SUBSCR', 'STORE_MAP',
     'LIST_TO_TUPLE', 'LIST_EXTEND', 'SET_UPDATE', 'DICT_UPDATE', 'DICT_MERGE',
-    'COPY', 'RESUME',
+    'COPY', 'RESUME', 'RETURN_CONST'
     ]
 
 _expr_codes = _const_codes + [

Original file line number	Diff line number	Diff line change
`@@ -6,7 +6,7 @@`
`6`	`6`	`'BUILD_CONST_KEY_MAP', 'BUILD_STRING',`
`7`	`7`	`'LOAD_CONST','RETURN_VALUE','STORE_SUBSCR', 'STORE_MAP',`
`8`	`8`	`'LIST_TO_TUPLE', 'LIST_EXTEND', 'SET_UPDATE', 'DICT_UPDATE', 'DICT_MERGE',`
`9`		`- 'COPY', 'RESUME',`
	`9`	`+ 'COPY', 'RESUME', 'RETURN_CONST'`
`10`	`10`	`]`
`11`	`11`
`12`	`12`	`_expr_codes = _const_codes + [`