angr 9.2.166__cp310-abi3-manylinux_2_28_x86_64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of angr might be problematic. Click here for more details.

Files changed (1409) hide show
  1. angr/__init__.py +366 -0
  2. angr/__main__.py +152 -0
  3. angr/ailment/__init__.py +81 -0
  4. angr/ailment/block.py +81 -0
  5. angr/ailment/block_walker.py +845 -0
  6. angr/ailment/constant.py +3 -0
  7. angr/ailment/converter_common.py +11 -0
  8. angr/ailment/converter_pcode.py +623 -0
  9. angr/ailment/converter_vex.py +798 -0
  10. angr/ailment/expression.py +1655 -0
  11. angr/ailment/manager.py +33 -0
  12. angr/ailment/statement.py +978 -0
  13. angr/ailment/tagged_object.py +61 -0
  14. angr/ailment/utils.py +114 -0
  15. angr/analyses/__init__.py +113 -0
  16. angr/analyses/analysis.py +429 -0
  17. angr/analyses/backward_slice.py +686 -0
  18. angr/analyses/binary_optimizer.py +670 -0
  19. angr/analyses/bindiff.py +1512 -0
  20. angr/analyses/boyscout.py +76 -0
  21. angr/analyses/callee_cleanup_finder.py +74 -0
  22. angr/analyses/calling_convention/__init__.py +6 -0
  23. angr/analyses/calling_convention/calling_convention.py +1096 -0
  24. angr/analyses/calling_convention/fact_collector.py +636 -0
  25. angr/analyses/calling_convention/utils.py +60 -0
  26. angr/analyses/cdg.py +189 -0
  27. angr/analyses/cfg/__init__.py +23 -0
  28. angr/analyses/cfg/cfb.py +428 -0
  29. angr/analyses/cfg/cfg.py +74 -0
  30. angr/analyses/cfg/cfg_arch_options.py +95 -0
  31. angr/analyses/cfg/cfg_base.py +2909 -0
  32. angr/analyses/cfg/cfg_emulated.py +3451 -0
  33. angr/analyses/cfg/cfg_fast.py +5316 -0
  34. angr/analyses/cfg/cfg_fast_soot.py +662 -0
  35. angr/analyses/cfg/cfg_job_base.py +203 -0
  36. angr/analyses/cfg/indirect_jump_resolvers/__init__.py +28 -0
  37. angr/analyses/cfg/indirect_jump_resolvers/amd64_elf_got.py +62 -0
  38. angr/analyses/cfg/indirect_jump_resolvers/amd64_pe_iat.py +51 -0
  39. angr/analyses/cfg/indirect_jump_resolvers/arm_elf_fast.py +159 -0
  40. angr/analyses/cfg/indirect_jump_resolvers/const_resolver.py +339 -0
  41. angr/analyses/cfg/indirect_jump_resolvers/constant_value_manager.py +107 -0
  42. angr/analyses/cfg/indirect_jump_resolvers/default_resolvers.py +76 -0
  43. angr/analyses/cfg/indirect_jump_resolvers/jumptable.py +2367 -0
  44. angr/analyses/cfg/indirect_jump_resolvers/memload_resolver.py +81 -0
  45. angr/analyses/cfg/indirect_jump_resolvers/mips_elf_fast.py +286 -0
  46. angr/analyses/cfg/indirect_jump_resolvers/mips_elf_got.py +148 -0
  47. angr/analyses/cfg/indirect_jump_resolvers/propagator_utils.py +46 -0
  48. angr/analyses/cfg/indirect_jump_resolvers/resolver.py +74 -0
  49. angr/analyses/cfg/indirect_jump_resolvers/syscall_resolver.py +92 -0
  50. angr/analyses/cfg/indirect_jump_resolvers/x86_elf_pic_plt.py +88 -0
  51. angr/analyses/cfg/indirect_jump_resolvers/x86_pe_iat.py +47 -0
  52. angr/analyses/cfg_slice_to_sink/__init__.py +11 -0
  53. angr/analyses/cfg_slice_to_sink/cfg_slice_to_sink.py +117 -0
  54. angr/analyses/cfg_slice_to_sink/graph.py +87 -0
  55. angr/analyses/cfg_slice_to_sink/transitions.py +27 -0
  56. angr/analyses/class_identifier.py +63 -0
  57. angr/analyses/code_tagging.py +123 -0
  58. angr/analyses/codecave.py +77 -0
  59. angr/analyses/complete_calling_conventions.py +461 -0
  60. angr/analyses/congruency_check.py +377 -0
  61. angr/analyses/data_dep/__init__.py +16 -0
  62. angr/analyses/data_dep/data_dependency_analysis.py +595 -0
  63. angr/analyses/data_dep/dep_nodes.py +171 -0
  64. angr/analyses/data_dep/sim_act_location.py +49 -0
  65. angr/analyses/datagraph_meta.py +105 -0
  66. angr/analyses/ddg.py +1670 -0
  67. angr/analyses/decompiler/__init__.py +41 -0
  68. angr/analyses/decompiler/ail_simplifier.py +2085 -0
  69. angr/analyses/decompiler/ailgraph_walker.py +49 -0
  70. angr/analyses/decompiler/block_io_finder.py +302 -0
  71. angr/analyses/decompiler/block_similarity.py +196 -0
  72. angr/analyses/decompiler/block_simplifier.py +376 -0
  73. angr/analyses/decompiler/callsite_maker.py +571 -0
  74. angr/analyses/decompiler/ccall_rewriters/__init__.py +9 -0
  75. angr/analyses/decompiler/ccall_rewriters/amd64_ccalls.py +580 -0
  76. angr/analyses/decompiler/ccall_rewriters/rewriter_base.py +20 -0
  77. angr/analyses/decompiler/ccall_rewriters/x86_ccalls.py +313 -0
  78. angr/analyses/decompiler/clinic.py +3308 -0
  79. angr/analyses/decompiler/condition_processor.py +1281 -0
  80. angr/analyses/decompiler/counters/__init__.py +16 -0
  81. angr/analyses/decompiler/counters/boolean_counter.py +27 -0
  82. angr/analyses/decompiler/counters/call_counter.py +57 -0
  83. angr/analyses/decompiler/counters/expression_counters.py +77 -0
  84. angr/analyses/decompiler/counters/seq_cf_structure_counter.py +63 -0
  85. angr/analyses/decompiler/decompilation_cache.py +46 -0
  86. angr/analyses/decompiler/decompilation_options.py +275 -0
  87. angr/analyses/decompiler/decompiler.py +710 -0
  88. angr/analyses/decompiler/dephication/__init__.py +6 -0
  89. angr/analyses/decompiler/dephication/dephication_base.py +100 -0
  90. angr/analyses/decompiler/dephication/graph_dephication.py +70 -0
  91. angr/analyses/decompiler/dephication/graph_rewriting.py +112 -0
  92. angr/analyses/decompiler/dephication/graph_vvar_mapping.py +363 -0
  93. angr/analyses/decompiler/dephication/rewriting_engine.py +527 -0
  94. angr/analyses/decompiler/dephication/seqnode_dephication.py +156 -0
  95. angr/analyses/decompiler/empty_node_remover.py +212 -0
  96. angr/analyses/decompiler/expression_narrower.py +287 -0
  97. angr/analyses/decompiler/goto_manager.py +112 -0
  98. angr/analyses/decompiler/graph_region.py +426 -0
  99. angr/analyses/decompiler/jump_target_collector.py +37 -0
  100. angr/analyses/decompiler/jumptable_entry_condition_rewriter.py +67 -0
  101. angr/analyses/decompiler/label_collector.py +32 -0
  102. angr/analyses/decompiler/optimization_passes/__init__.py +151 -0
  103. angr/analyses/decompiler/optimization_passes/base_ptr_save_simplifier.py +157 -0
  104. angr/analyses/decompiler/optimization_passes/call_stmt_rewriter.py +46 -0
  105. angr/analyses/decompiler/optimization_passes/code_motion.py +362 -0
  106. angr/analyses/decompiler/optimization_passes/condition_constprop.py +219 -0
  107. angr/analyses/decompiler/optimization_passes/const_derefs.py +266 -0
  108. angr/analyses/decompiler/optimization_passes/const_prop_reverter.py +365 -0
  109. angr/analyses/decompiler/optimization_passes/cross_jump_reverter.py +106 -0
  110. angr/analyses/decompiler/optimization_passes/deadblock_remover.py +82 -0
  111. angr/analyses/decompiler/optimization_passes/determine_load_sizes.py +64 -0
  112. angr/analyses/decompiler/optimization_passes/div_simplifier.py +425 -0
  113. angr/analyses/decompiler/optimization_passes/duplication_reverter/__init__.py +5 -0
  114. angr/analyses/decompiler/optimization_passes/duplication_reverter/ail_merge_graph.py +503 -0
  115. angr/analyses/decompiler/optimization_passes/duplication_reverter/duplication_reverter.py +1218 -0
  116. angr/analyses/decompiler/optimization_passes/duplication_reverter/errors.py +16 -0
  117. angr/analyses/decompiler/optimization_passes/duplication_reverter/similarity.py +126 -0
  118. angr/analyses/decompiler/optimization_passes/duplication_reverter/utils.py +167 -0
  119. angr/analyses/decompiler/optimization_passes/eager_std_string_concatenation.py +165 -0
  120. angr/analyses/decompiler/optimization_passes/engine_base.py +500 -0
  121. angr/analyses/decompiler/optimization_passes/expr_op_swapper.py +135 -0
  122. angr/analyses/decompiler/optimization_passes/flip_boolean_cmp.py +113 -0
  123. angr/analyses/decompiler/optimization_passes/inlined_string_transformation_simplifier.py +615 -0
  124. angr/analyses/decompiler/optimization_passes/ite_expr_converter.py +224 -0
  125. angr/analyses/decompiler/optimization_passes/ite_region_converter.py +335 -0
  126. angr/analyses/decompiler/optimization_passes/lowered_switch_simplifier.py +923 -0
  127. angr/analyses/decompiler/optimization_passes/mod_simplifier.py +99 -0
  128. angr/analyses/decompiler/optimization_passes/optimization_pass.py +703 -0
  129. angr/analyses/decompiler/optimization_passes/register_save_area_simplifier.py +221 -0
  130. angr/analyses/decompiler/optimization_passes/ret_addr_save_simplifier.py +171 -0
  131. angr/analyses/decompiler/optimization_passes/ret_deduplicator.py +222 -0
  132. angr/analyses/decompiler/optimization_passes/return_duplicator_base.py +640 -0
  133. angr/analyses/decompiler/optimization_passes/return_duplicator_high.py +61 -0
  134. angr/analyses/decompiler/optimization_passes/return_duplicator_low.py +237 -0
  135. angr/analyses/decompiler/optimization_passes/stack_canary_simplifier.py +333 -0
  136. angr/analyses/decompiler/optimization_passes/switch_default_case_duplicator.py +149 -0
  137. angr/analyses/decompiler/optimization_passes/switch_reused_entry_rewriter.py +102 -0
  138. angr/analyses/decompiler/optimization_passes/tag_slicer.py +41 -0
  139. angr/analyses/decompiler/optimization_passes/win_stack_canary_simplifier.py +421 -0
  140. angr/analyses/decompiler/optimization_passes/x86_gcc_getpc_simplifier.py +88 -0
  141. angr/analyses/decompiler/peephole_optimizations/__init__.py +129 -0
  142. angr/analyses/decompiler/peephole_optimizations/a_div_const_add_a_mul_n_div_const.py +42 -0
  143. angr/analyses/decompiler/peephole_optimizations/a_mul_const_div_shr_const.py +38 -0
  144. angr/analyses/decompiler/peephole_optimizations/a_mul_const_sub_a.py +34 -0
  145. angr/analyses/decompiler/peephole_optimizations/a_shl_const_sub_a.py +34 -0
  146. angr/analyses/decompiler/peephole_optimizations/a_sub_a_div.py +25 -0
  147. angr/analyses/decompiler/peephole_optimizations/a_sub_a_shr_const_shr_const.py +37 -0
  148. angr/analyses/decompiler/peephole_optimizations/a_sub_a_sub_n.py +23 -0
  149. angr/analyses/decompiler/peephole_optimizations/arm_cmpf.py +236 -0
  150. angr/analyses/decompiler/peephole_optimizations/base.py +157 -0
  151. angr/analyses/decompiler/peephole_optimizations/basepointeroffset_add_n.py +34 -0
  152. angr/analyses/decompiler/peephole_optimizations/basepointeroffset_and_mask.py +36 -0
  153. angr/analyses/decompiler/peephole_optimizations/bitwise_or_to_logical_or.py +34 -0
  154. angr/analyses/decompiler/peephole_optimizations/bool_expr_xor_1.py +27 -0
  155. angr/analyses/decompiler/peephole_optimizations/bswap.py +142 -0
  156. angr/analyses/decompiler/peephole_optimizations/cas_intrinsics.py +115 -0
  157. angr/analyses/decompiler/peephole_optimizations/cmpord_rewriter.py +71 -0
  158. angr/analyses/decompiler/peephole_optimizations/coalesce_adjacent_shrs.py +39 -0
  159. angr/analyses/decompiler/peephole_optimizations/coalesce_same_cascading_ifs.py +28 -0
  160. angr/analyses/decompiler/peephole_optimizations/constant_derefs.py +44 -0
  161. angr/analyses/decompiler/peephole_optimizations/conv_a_sub0_shr_and.py +69 -0
  162. angr/analyses/decompiler/peephole_optimizations/conv_shl_shr.py +52 -0
  163. angr/analyses/decompiler/peephole_optimizations/eager_eval.py +447 -0
  164. angr/analyses/decompiler/peephole_optimizations/extended_byte_and_mask.py +56 -0
  165. angr/analyses/decompiler/peephole_optimizations/inlined_memcpy.py +78 -0
  166. angr/analyses/decompiler/peephole_optimizations/inlined_strcpy.py +217 -0
  167. angr/analyses/decompiler/peephole_optimizations/inlined_strcpy_consolidation.py +106 -0
  168. angr/analyses/decompiler/peephole_optimizations/inlined_wstrcpy.py +170 -0
  169. angr/analyses/decompiler/peephole_optimizations/invert_negated_logical_conjuction_disjunction.py +50 -0
  170. angr/analyses/decompiler/peephole_optimizations/modulo_simplifier.py +89 -0
  171. angr/analyses/decompiler/peephole_optimizations/one_sub_bool.py +33 -0
  172. angr/analyses/decompiler/peephole_optimizations/optimized_div_simplifier.py +356 -0
  173. angr/analyses/decompiler/peephole_optimizations/remove_cascading_conversions.py +45 -0
  174. angr/analyses/decompiler/peephole_optimizations/remove_cxx_destructor_calls.py +32 -0
  175. angr/analyses/decompiler/peephole_optimizations/remove_empty_if_body.py +46 -0
  176. angr/analyses/decompiler/peephole_optimizations/remove_noop_conversions.py +47 -0
  177. angr/analyses/decompiler/peephole_optimizations/remove_redundant_bitmasks.py +125 -0
  178. angr/analyses/decompiler/peephole_optimizations/remove_redundant_conversions.py +273 -0
  179. angr/analyses/decompiler/peephole_optimizations/remove_redundant_ite_branch.py +30 -0
  180. angr/analyses/decompiler/peephole_optimizations/remove_redundant_ite_comparisons.py +54 -0
  181. angr/analyses/decompiler/peephole_optimizations/remove_redundant_nots.py +36 -0
  182. angr/analyses/decompiler/peephole_optimizations/remove_redundant_reinterprets.py +44 -0
  183. angr/analyses/decompiler/peephole_optimizations/remove_redundant_shifts.py +95 -0
  184. angr/analyses/decompiler/peephole_optimizations/remove_redundant_shifts_around_comparators.py +44 -0
  185. angr/analyses/decompiler/peephole_optimizations/rewrite_bit_extractions.py +85 -0
  186. angr/analyses/decompiler/peephole_optimizations/rewrite_conv_mul.py +40 -0
  187. angr/analyses/decompiler/peephole_optimizations/rewrite_cxx_operator_calls.py +90 -0
  188. angr/analyses/decompiler/peephole_optimizations/rewrite_mips_gp_loads.py +49 -0
  189. angr/analyses/decompiler/peephole_optimizations/rol_ror.py +130 -0
  190. angr/analyses/decompiler/peephole_optimizations/sar_to_signed_div.py +143 -0
  191. angr/analyses/decompiler/peephole_optimizations/shl_to_mul.py +25 -0
  192. angr/analyses/decompiler/peephole_optimizations/simplify_pc_relative_loads.py +51 -0
  193. angr/analyses/decompiler/peephole_optimizations/single_bit_cond_to_boolexpr.py +82 -0
  194. angr/analyses/decompiler/peephole_optimizations/single_bit_xor.py +29 -0
  195. angr/analyses/decompiler/peephole_optimizations/tidy_stack_addr.py +131 -0
  196. angr/analyses/decompiler/peephole_optimizations/utils.py +18 -0
  197. angr/analyses/decompiler/presets/__init__.py +20 -0
  198. angr/analyses/decompiler/presets/basic.py +32 -0
  199. angr/analyses/decompiler/presets/fast.py +58 -0
  200. angr/analyses/decompiler/presets/full.py +68 -0
  201. angr/analyses/decompiler/presets/preset.py +37 -0
  202. angr/analyses/decompiler/redundant_label_remover.py +134 -0
  203. angr/analyses/decompiler/region_identifier.py +1239 -0
  204. angr/analyses/decompiler/region_simplifiers/__init__.py +5 -0
  205. angr/analyses/decompiler/region_simplifiers/cascading_cond_transformer.py +95 -0
  206. angr/analyses/decompiler/region_simplifiers/cascading_ifs.py +82 -0
  207. angr/analyses/decompiler/region_simplifiers/expr_folding.py +818 -0
  208. angr/analyses/decompiler/region_simplifiers/goto.py +178 -0
  209. angr/analyses/decompiler/region_simplifiers/if_.py +135 -0
  210. angr/analyses/decompiler/region_simplifiers/ifelse.py +91 -0
  211. angr/analyses/decompiler/region_simplifiers/loop.py +143 -0
  212. angr/analyses/decompiler/region_simplifiers/node_address_finder.py +24 -0
  213. angr/analyses/decompiler/region_simplifiers/region_simplifier.py +246 -0
  214. angr/analyses/decompiler/region_simplifiers/switch_cluster_simplifier.py +654 -0
  215. angr/analyses/decompiler/region_simplifiers/switch_expr_simplifier.py +87 -0
  216. angr/analyses/decompiler/region_walker.py +24 -0
  217. angr/analyses/decompiler/return_maker.py +72 -0
  218. angr/analyses/decompiler/seq_to_blocks.py +20 -0
  219. angr/analyses/decompiler/sequence_walker.py +257 -0
  220. angr/analyses/decompiler/ssailification/__init__.py +4 -0
  221. angr/analyses/decompiler/ssailification/rewriting.py +379 -0
  222. angr/analyses/decompiler/ssailification/rewriting_engine.py +1053 -0
  223. angr/analyses/decompiler/ssailification/rewriting_state.py +61 -0
  224. angr/analyses/decompiler/ssailification/ssailification.py +276 -0
  225. angr/analyses/decompiler/ssailification/traversal.py +124 -0
  226. angr/analyses/decompiler/ssailification/traversal_engine.py +306 -0
  227. angr/analyses/decompiler/ssailification/traversal_state.py +48 -0
  228. angr/analyses/decompiler/stack_item.py +36 -0
  229. angr/analyses/decompiler/structured_codegen/__init__.py +25 -0
  230. angr/analyses/decompiler/structured_codegen/base.py +132 -0
  231. angr/analyses/decompiler/structured_codegen/c.py +4082 -0
  232. angr/analyses/decompiler/structured_codegen/dummy.py +15 -0
  233. angr/analyses/decompiler/structured_codegen/dwarf_import.py +190 -0
  234. angr/analyses/decompiler/structuring/__init__.py +30 -0
  235. angr/analyses/decompiler/structuring/dream.py +1217 -0
  236. angr/analyses/decompiler/structuring/phoenix.py +3090 -0
  237. angr/analyses/decompiler/structuring/recursive_structurer.py +187 -0
  238. angr/analyses/decompiler/structuring/sailr.py +120 -0
  239. angr/analyses/decompiler/structuring/structurer_base.py +1066 -0
  240. angr/analyses/decompiler/structuring/structurer_nodes.py +440 -0
  241. angr/analyses/decompiler/utils.py +1118 -0
  242. angr/analyses/deobfuscator/__init__.py +18 -0
  243. angr/analyses/deobfuscator/api_obf_finder.py +325 -0
  244. angr/analyses/deobfuscator/api_obf_peephole_optimizer.py +51 -0
  245. angr/analyses/deobfuscator/api_obf_type2_finder.py +166 -0
  246. angr/analyses/deobfuscator/irsb_reg_collector.py +54 -0
  247. angr/analyses/deobfuscator/string_obf_finder.py +959 -0
  248. angr/analyses/deobfuscator/string_obf_opt_passes.py +133 -0
  249. angr/analyses/deobfuscator/string_obf_peephole_optimizer.py +47 -0
  250. angr/analyses/disassembly.py +1295 -0
  251. angr/analyses/disassembly_utils.py +101 -0
  252. angr/analyses/dominance_frontier.py +57 -0
  253. angr/analyses/fcp/__init__.py +4 -0
  254. angr/analyses/fcp/fcp.py +427 -0
  255. angr/analyses/find_objects_static.py +205 -0
  256. angr/analyses/flirt/__init__.py +47 -0
  257. angr/analyses/flirt/consts.py +160 -0
  258. angr/analyses/flirt/flirt.py +244 -0
  259. angr/analyses/flirt/flirt_function.py +20 -0
  260. angr/analyses/flirt/flirt_matcher.py +351 -0
  261. angr/analyses/flirt/flirt_module.py +32 -0
  262. angr/analyses/flirt/flirt_node.py +23 -0
  263. angr/analyses/flirt/flirt_sig.py +359 -0
  264. angr/analyses/flirt/flirt_utils.py +31 -0
  265. angr/analyses/forward_analysis/__init__.py +12 -0
  266. angr/analyses/forward_analysis/forward_analysis.py +530 -0
  267. angr/analyses/forward_analysis/job_info.py +64 -0
  268. angr/analyses/forward_analysis/visitors/__init__.py +14 -0
  269. angr/analyses/forward_analysis/visitors/call_graph.py +29 -0
  270. angr/analyses/forward_analysis/visitors/function_graph.py +86 -0
  271. angr/analyses/forward_analysis/visitors/graph.py +242 -0
  272. angr/analyses/forward_analysis/visitors/loop.py +29 -0
  273. angr/analyses/forward_analysis/visitors/single_node_graph.py +38 -0
  274. angr/analyses/identifier/__init__.py +5 -0
  275. angr/analyses/identifier/custom_callable.py +137 -0
  276. angr/analyses/identifier/errors.py +10 -0
  277. angr/analyses/identifier/func.py +60 -0
  278. angr/analyses/identifier/functions/__init__.py +37 -0
  279. angr/analyses/identifier/functions/atoi.py +73 -0
  280. angr/analyses/identifier/functions/based_atoi.py +125 -0
  281. angr/analyses/identifier/functions/fdprintf.py +123 -0
  282. angr/analyses/identifier/functions/free.py +64 -0
  283. angr/analyses/identifier/functions/int2str.py +287 -0
  284. angr/analyses/identifier/functions/malloc.py +111 -0
  285. angr/analyses/identifier/functions/memcmp.py +67 -0
  286. angr/analyses/identifier/functions/memcpy.py +89 -0
  287. angr/analyses/identifier/functions/memset.py +43 -0
  288. angr/analyses/identifier/functions/printf.py +123 -0
  289. angr/analyses/identifier/functions/recv_until.py +312 -0
  290. angr/analyses/identifier/functions/skip_calloc.py +73 -0
  291. angr/analyses/identifier/functions/skip_realloc.py +97 -0
  292. angr/analyses/identifier/functions/skip_recv_n.py +105 -0
  293. angr/analyses/identifier/functions/snprintf.py +112 -0
  294. angr/analyses/identifier/functions/sprintf.py +116 -0
  295. angr/analyses/identifier/functions/strcasecmp.py +33 -0
  296. angr/analyses/identifier/functions/strcmp.py +113 -0
  297. angr/analyses/identifier/functions/strcpy.py +43 -0
  298. angr/analyses/identifier/functions/strlen.py +27 -0
  299. angr/analyses/identifier/functions/strncmp.py +104 -0
  300. angr/analyses/identifier/functions/strncpy.py +65 -0
  301. angr/analyses/identifier/functions/strtol.py +89 -0
  302. angr/analyses/identifier/identify.py +825 -0
  303. angr/analyses/identifier/runner.py +360 -0
  304. angr/analyses/init_finder.py +289 -0
  305. angr/analyses/loop_analysis.py +349 -0
  306. angr/analyses/loopfinder.py +171 -0
  307. angr/analyses/patchfinder.py +137 -0
  308. angr/analyses/pathfinder.py +282 -0
  309. angr/analyses/propagator/__init__.py +5 -0
  310. angr/analyses/propagator/engine_base.py +62 -0
  311. angr/analyses/propagator/engine_vex.py +297 -0
  312. angr/analyses/propagator/propagator.py +361 -0
  313. angr/analyses/propagator/top_checker_mixin.py +218 -0
  314. angr/analyses/propagator/values.py +117 -0
  315. angr/analyses/propagator/vex_vars.py +68 -0
  316. angr/analyses/proximity_graph.py +444 -0
  317. angr/analyses/reaching_definitions/__init__.py +67 -0
  318. angr/analyses/reaching_definitions/call_trace.py +73 -0
  319. angr/analyses/reaching_definitions/dep_graph.py +433 -0
  320. angr/analyses/reaching_definitions/engine_ail.py +1130 -0
  321. angr/analyses/reaching_definitions/engine_vex.py +1127 -0
  322. angr/analyses/reaching_definitions/external_codeloc.py +0 -0
  323. angr/analyses/reaching_definitions/function_handler.py +638 -0
  324. angr/analyses/reaching_definitions/function_handler_library/__init__.py +12 -0
  325. angr/analyses/reaching_definitions/function_handler_library/stdio.py +269 -0
  326. angr/analyses/reaching_definitions/function_handler_library/stdlib.py +195 -0
  327. angr/analyses/reaching_definitions/function_handler_library/string.py +158 -0
  328. angr/analyses/reaching_definitions/function_handler_library/unistd.py +51 -0
  329. angr/analyses/reaching_definitions/heap_allocator.py +70 -0
  330. angr/analyses/reaching_definitions/rd_initializer.py +237 -0
  331. angr/analyses/reaching_definitions/rd_state.py +579 -0
  332. angr/analyses/reaching_definitions/reaching_definitions.py +581 -0
  333. angr/analyses/reaching_definitions/subject.py +65 -0
  334. angr/analyses/reassembler.py +2900 -0
  335. angr/analyses/s_liveness.py +203 -0
  336. angr/analyses/s_propagator.py +542 -0
  337. angr/analyses/s_reaching_definitions/__init__.py +12 -0
  338. angr/analyses/s_reaching_definitions/s_rda_model.py +136 -0
  339. angr/analyses/s_reaching_definitions/s_rda_view.py +316 -0
  340. angr/analyses/s_reaching_definitions/s_reaching_definitions.py +177 -0
  341. angr/analyses/smc.py +161 -0
  342. angr/analyses/soot_class_hierarchy.py +273 -0
  343. angr/analyses/stack_pointer_tracker.py +953 -0
  344. angr/analyses/static_hooker.py +53 -0
  345. angr/analyses/typehoon/__init__.py +5 -0
  346. angr/analyses/typehoon/dfa.py +118 -0
  347. angr/analyses/typehoon/lifter.py +122 -0
  348. angr/analyses/typehoon/simple_solver.py +1666 -0
  349. angr/analyses/typehoon/translator.py +279 -0
  350. angr/analyses/typehoon/typeconsts.py +338 -0
  351. angr/analyses/typehoon/typehoon.py +319 -0
  352. angr/analyses/typehoon/typevars.py +622 -0
  353. angr/analyses/typehoon/variance.py +11 -0
  354. angr/analyses/unpacker/__init__.py +6 -0
  355. angr/analyses/unpacker/obfuscation_detector.py +103 -0
  356. angr/analyses/unpacker/packing_detector.py +138 -0
  357. angr/analyses/variable_recovery/__init__.py +9 -0
  358. angr/analyses/variable_recovery/annotations.py +58 -0
  359. angr/analyses/variable_recovery/engine_ail.py +885 -0
  360. angr/analyses/variable_recovery/engine_base.py +1197 -0
  361. angr/analyses/variable_recovery/engine_vex.py +593 -0
  362. angr/analyses/variable_recovery/irsb_scanner.py +143 -0
  363. angr/analyses/variable_recovery/variable_recovery.py +574 -0
  364. angr/analyses/variable_recovery/variable_recovery_base.py +489 -0
  365. angr/analyses/variable_recovery/variable_recovery_fast.py +661 -0
  366. angr/analyses/veritesting.py +626 -0
  367. angr/analyses/vfg.py +1898 -0
  368. angr/analyses/vsa_ddg.py +420 -0
  369. angr/analyses/vtable.py +92 -0
  370. angr/analyses/xrefs.py +286 -0
  371. angr/angrdb/__init__.py +14 -0
  372. angr/angrdb/db.py +206 -0
  373. angr/angrdb/models.py +184 -0
  374. angr/angrdb/serializers/__init__.py +10 -0
  375. angr/angrdb/serializers/cfg_model.py +41 -0
  376. angr/angrdb/serializers/comments.py +60 -0
  377. angr/angrdb/serializers/funcs.py +61 -0
  378. angr/angrdb/serializers/kb.py +111 -0
  379. angr/angrdb/serializers/labels.py +59 -0
  380. angr/angrdb/serializers/loader.py +165 -0
  381. angr/angrdb/serializers/structured_code.py +125 -0
  382. angr/angrdb/serializers/variables.py +58 -0
  383. angr/angrdb/serializers/xrefs.py +48 -0
  384. angr/annocfg.py +317 -0
  385. angr/blade.py +431 -0
  386. angr/block.py +509 -0
  387. angr/callable.py +168 -0
  388. angr/calling_conventions.py +2580 -0
  389. angr/code_location.py +163 -0
  390. angr/codenode.py +145 -0
  391. angr/concretization_strategies/__init__.py +32 -0
  392. angr/concretization_strategies/any.py +17 -0
  393. angr/concretization_strategies/any_named.py +35 -0
  394. angr/concretization_strategies/base.py +81 -0
  395. angr/concretization_strategies/controlled_data.py +58 -0
  396. angr/concretization_strategies/eval.py +19 -0
  397. angr/concretization_strategies/logging.py +35 -0
  398. angr/concretization_strategies/max.py +25 -0
  399. angr/concretization_strategies/nonzero.py +16 -0
  400. angr/concretization_strategies/nonzero_range.py +22 -0
  401. angr/concretization_strategies/norepeats.py +37 -0
  402. angr/concretization_strategies/norepeats_range.py +37 -0
  403. angr/concretization_strategies/range.py +19 -0
  404. angr/concretization_strategies/signed_add.py +31 -0
  405. angr/concretization_strategies/single.py +15 -0
  406. angr/concretization_strategies/solutions.py +20 -0
  407. angr/concretization_strategies/unlimited_range.py +17 -0
  408. angr/distributed/__init__.py +9 -0
  409. angr/distributed/server.py +197 -0
  410. angr/distributed/worker.py +185 -0
  411. angr/emulator.py +143 -0
  412. angr/engines/__init__.py +67 -0
  413. angr/engines/concrete.py +66 -0
  414. angr/engines/engine.py +29 -0
  415. angr/engines/failure.py +27 -0
  416. angr/engines/hook.py +68 -0
  417. angr/engines/icicle.py +278 -0
  418. angr/engines/light/__init__.py +23 -0
  419. angr/engines/light/data.py +681 -0
  420. angr/engines/light/engine.py +1285 -0
  421. angr/engines/pcode/__init__.py +9 -0
  422. angr/engines/pcode/behavior.py +994 -0
  423. angr/engines/pcode/cc.py +128 -0
  424. angr/engines/pcode/emulate.py +440 -0
  425. angr/engines/pcode/engine.py +242 -0
  426. angr/engines/pcode/lifter.py +1420 -0
  427. angr/engines/procedure.py +70 -0
  428. angr/engines/soot/__init__.py +5 -0
  429. angr/engines/soot/engine.py +410 -0
  430. angr/engines/soot/exceptions.py +17 -0
  431. angr/engines/soot/expressions/__init__.py +87 -0
  432. angr/engines/soot/expressions/arrayref.py +22 -0
  433. angr/engines/soot/expressions/base.py +21 -0
  434. angr/engines/soot/expressions/binop.py +28 -0
  435. angr/engines/soot/expressions/cast.py +22 -0
  436. angr/engines/soot/expressions/condition.py +35 -0
  437. angr/engines/soot/expressions/constants.py +47 -0
  438. angr/engines/soot/expressions/instanceOf.py +15 -0
  439. angr/engines/soot/expressions/instancefieldref.py +8 -0
  440. angr/engines/soot/expressions/invoke.py +114 -0
  441. angr/engines/soot/expressions/length.py +8 -0
  442. angr/engines/soot/expressions/local.py +8 -0
  443. angr/engines/soot/expressions/new.py +16 -0
  444. angr/engines/soot/expressions/newArray.py +54 -0
  445. angr/engines/soot/expressions/newMultiArray.py +86 -0
  446. angr/engines/soot/expressions/paramref.py +8 -0
  447. angr/engines/soot/expressions/phi.py +30 -0
  448. angr/engines/soot/expressions/staticfieldref.py +8 -0
  449. angr/engines/soot/expressions/thisref.py +7 -0
  450. angr/engines/soot/expressions/unsupported.py +7 -0
  451. angr/engines/soot/field_dispatcher.py +46 -0
  452. angr/engines/soot/method_dispatcher.py +46 -0
  453. angr/engines/soot/statements/__init__.py +44 -0
  454. angr/engines/soot/statements/assign.py +30 -0
  455. angr/engines/soot/statements/base.py +79 -0
  456. angr/engines/soot/statements/goto.py +14 -0
  457. angr/engines/soot/statements/identity.py +15 -0
  458. angr/engines/soot/statements/if_.py +19 -0
  459. angr/engines/soot/statements/invoke.py +12 -0
  460. angr/engines/soot/statements/return_.py +20 -0
  461. angr/engines/soot/statements/switch.py +41 -0
  462. angr/engines/soot/statements/throw.py +15 -0
  463. angr/engines/soot/values/__init__.py +38 -0
  464. angr/engines/soot/values/arrayref.py +122 -0
  465. angr/engines/soot/values/base.py +7 -0
  466. angr/engines/soot/values/constants.py +18 -0
  467. angr/engines/soot/values/instancefieldref.py +44 -0
  468. angr/engines/soot/values/local.py +18 -0
  469. angr/engines/soot/values/paramref.py +18 -0
  470. angr/engines/soot/values/staticfieldref.py +38 -0
  471. angr/engines/soot/values/strref.py +38 -0
  472. angr/engines/soot/values/thisref.py +149 -0
  473. angr/engines/successors.py +654 -0
  474. angr/engines/syscall.py +51 -0
  475. angr/engines/unicorn.py +490 -0
  476. angr/engines/vex/__init__.py +20 -0
  477. angr/engines/vex/claripy/__init__.py +5 -0
  478. angr/engines/vex/claripy/ccall.py +2097 -0
  479. angr/engines/vex/claripy/datalayer.py +141 -0
  480. angr/engines/vex/claripy/irop.py +1276 -0
  481. angr/engines/vex/heavy/__init__.py +16 -0
  482. angr/engines/vex/heavy/actions.py +231 -0
  483. angr/engines/vex/heavy/concretizers.py +403 -0
  484. angr/engines/vex/heavy/dirty.py +466 -0
  485. angr/engines/vex/heavy/heavy.py +370 -0
  486. angr/engines/vex/heavy/inspect.py +52 -0
  487. angr/engines/vex/heavy/resilience.py +85 -0
  488. angr/engines/vex/heavy/super_fastpath.py +34 -0
  489. angr/engines/vex/lifter.py +420 -0
  490. angr/engines/vex/light/__init__.py +11 -0
  491. angr/engines/vex/light/light.py +551 -0
  492. angr/engines/vex/light/resilience.py +74 -0
  493. angr/engines/vex/light/slicing.py +52 -0
  494. angr/errors.py +609 -0
  495. angr/exploration_techniques/__init__.py +53 -0
  496. angr/exploration_techniques/base.py +126 -0
  497. angr/exploration_techniques/bucketizer.py +94 -0
  498. angr/exploration_techniques/common.py +56 -0
  499. angr/exploration_techniques/dfs.py +37 -0
  500. angr/exploration_techniques/director.py +520 -0
  501. angr/exploration_techniques/driller_core.py +100 -0
  502. angr/exploration_techniques/explorer.py +152 -0
  503. angr/exploration_techniques/lengthlimiter.py +22 -0
  504. angr/exploration_techniques/local_loop_seer.py +65 -0
  505. angr/exploration_techniques/loop_seer.py +236 -0
  506. angr/exploration_techniques/manual_mergepoint.py +82 -0
  507. angr/exploration_techniques/memory_watcher.py +43 -0
  508. angr/exploration_techniques/oppologist.py +92 -0
  509. angr/exploration_techniques/slicecutor.py +118 -0
  510. angr/exploration_techniques/spiller.py +280 -0
  511. angr/exploration_techniques/spiller_db.py +27 -0
  512. angr/exploration_techniques/stochastic.py +56 -0
  513. angr/exploration_techniques/stub_stasher.py +19 -0
  514. angr/exploration_techniques/suggestions.py +159 -0
  515. angr/exploration_techniques/tech_builder.py +49 -0
  516. angr/exploration_techniques/threading.py +69 -0
  517. angr/exploration_techniques/timeout.py +34 -0
  518. angr/exploration_techniques/tracer.py +1098 -0
  519. angr/exploration_techniques/unique.py +106 -0
  520. angr/exploration_techniques/veritesting.py +37 -0
  521. angr/factory.py +404 -0
  522. angr/flirt/__init__.py +97 -0
  523. angr/flirt/build_sig.py +305 -0
  524. angr/graph_utils.py +0 -0
  525. angr/keyed_region.py +525 -0
  526. angr/knowledge_base.py +143 -0
  527. angr/knowledge_plugins/__init__.py +43 -0
  528. angr/knowledge_plugins/callsite_prototypes.py +53 -0
  529. angr/knowledge_plugins/cfg/__init__.py +18 -0
  530. angr/knowledge_plugins/cfg/cfg_manager.py +95 -0
  531. angr/knowledge_plugins/cfg/cfg_model.py +1045 -0
  532. angr/knowledge_plugins/cfg/cfg_node.py +536 -0
  533. angr/knowledge_plugins/cfg/indirect_jump.py +65 -0
  534. angr/knowledge_plugins/cfg/memory_data.py +156 -0
  535. angr/knowledge_plugins/comments.py +16 -0
  536. angr/knowledge_plugins/custom_strings.py +38 -0
  537. angr/knowledge_plugins/data.py +22 -0
  538. angr/knowledge_plugins/debug_variables.py +216 -0
  539. angr/knowledge_plugins/functions/__init__.py +9 -0
  540. angr/knowledge_plugins/functions/function.py +1780 -0
  541. angr/knowledge_plugins/functions/function_manager.py +588 -0
  542. angr/knowledge_plugins/functions/function_parser.py +299 -0
  543. angr/knowledge_plugins/functions/soot_function.py +128 -0
  544. angr/knowledge_plugins/indirect_jumps.py +35 -0
  545. angr/knowledge_plugins/key_definitions/__init__.py +17 -0
  546. angr/knowledge_plugins/key_definitions/atoms.py +374 -0
  547. angr/knowledge_plugins/key_definitions/constants.py +29 -0
  548. angr/knowledge_plugins/key_definitions/definition.py +214 -0
  549. angr/knowledge_plugins/key_definitions/environment.py +96 -0
  550. angr/knowledge_plugins/key_definitions/heap_address.py +33 -0
  551. angr/knowledge_plugins/key_definitions/key_definition_manager.py +82 -0
  552. angr/knowledge_plugins/key_definitions/live_definitions.py +1010 -0
  553. angr/knowledge_plugins/key_definitions/liveness.py +165 -0
  554. angr/knowledge_plugins/key_definitions/rd_model.py +171 -0
  555. angr/knowledge_plugins/key_definitions/tag.py +78 -0
  556. angr/knowledge_plugins/key_definitions/undefined.py +70 -0
  557. angr/knowledge_plugins/key_definitions/unknown_size.py +86 -0
  558. angr/knowledge_plugins/key_definitions/uses.py +178 -0
  559. angr/knowledge_plugins/labels.py +110 -0
  560. angr/knowledge_plugins/obfuscations.py +37 -0
  561. angr/knowledge_plugins/patches.py +126 -0
  562. angr/knowledge_plugins/plugin.py +24 -0
  563. angr/knowledge_plugins/propagations/__init__.py +10 -0
  564. angr/knowledge_plugins/propagations/prop_value.py +191 -0
  565. angr/knowledge_plugins/propagations/propagation_manager.py +60 -0
  566. angr/knowledge_plugins/propagations/propagation_model.py +80 -0
  567. angr/knowledge_plugins/propagations/states.py +552 -0
  568. angr/knowledge_plugins/structured_code.py +63 -0
  569. angr/knowledge_plugins/types.py +88 -0
  570. angr/knowledge_plugins/variables/__init__.py +8 -0
  571. angr/knowledge_plugins/variables/variable_access.py +113 -0
  572. angr/knowledge_plugins/variables/variable_manager.py +1380 -0
  573. angr/knowledge_plugins/xrefs/__init__.py +12 -0
  574. angr/knowledge_plugins/xrefs/xref.py +150 -0
  575. angr/knowledge_plugins/xrefs/xref_manager.py +127 -0
  576. angr/knowledge_plugins/xrefs/xref_types.py +16 -0
  577. angr/misc/__init__.py +19 -0
  578. angr/misc/ansi.py +47 -0
  579. angr/misc/autoimport.py +90 -0
  580. angr/misc/bug_report.py +117 -0
  581. angr/misc/hookset.py +106 -0
  582. angr/misc/loggers.py +130 -0
  583. angr/misc/picklable_lock.py +46 -0
  584. angr/misc/plugins.py +289 -0
  585. angr/misc/telemetry.py +54 -0
  586. angr/misc/testing.py +24 -0
  587. angr/misc/ux.py +31 -0
  588. angr/procedures/__init__.py +12 -0
  589. angr/procedures/advapi32/__init__.py +0 -0
  590. angr/procedures/cgc/__init__.py +3 -0
  591. angr/procedures/cgc/_terminate.py +11 -0
  592. angr/procedures/cgc/allocate.py +75 -0
  593. angr/procedures/cgc/deallocate.py +67 -0
  594. angr/procedures/cgc/fdwait.py +65 -0
  595. angr/procedures/cgc/random.py +67 -0
  596. angr/procedures/cgc/receive.py +93 -0
  597. angr/procedures/cgc/transmit.py +65 -0
  598. angr/procedures/definitions/__init__.py +779 -0
  599. angr/procedures/definitions/cgc.py +20 -0
  600. angr/procedures/definitions/glibc.py +8372 -0
  601. angr/procedures/definitions/gnulib.py +32 -0
  602. angr/procedures/definitions/libstdcpp.py +21 -0
  603. angr/procedures/definitions/linux_kernel.py +6171 -0
  604. angr/procedures/definitions/linux_loader.py +7 -0
  605. angr/procedures/definitions/msvcr.py +16 -0
  606. angr/procedures/definitions/parse_syscalls_from_local_system.py +50 -0
  607. angr/procedures/definitions/parse_win32json.py +2553 -0
  608. angr/procedures/definitions/types_stl.py +22 -0
  609. angr/procedures/definitions/types_win32.py +34482 -0
  610. angr/procedures/definitions/wdk_api-ms-win-dx-d3dkmt-l1-1-4.py +30 -0
  611. angr/procedures/definitions/wdk_api-ms-win-dx-d3dkmt-l1-1-6.py +26 -0
  612. angr/procedures/definitions/wdk_clfs.py +140 -0
  613. angr/procedures/definitions/wdk_fltmgr.py +556 -0
  614. angr/procedures/definitions/wdk_fwpkclnt.py +30 -0
  615. angr/procedures/definitions/wdk_fwpuclnt.py +316 -0
  616. angr/procedures/definitions/wdk_gdi32.py +366 -0
  617. angr/procedures/definitions/wdk_hal.py +78 -0
  618. angr/procedures/definitions/wdk_ksecdd.py +62 -0
  619. angr/procedures/definitions/wdk_ndis.py +238 -0
  620. angr/procedures/definitions/wdk_ntoskrnl.py +3451 -0
  621. angr/procedures/definitions/wdk_offreg.py +72 -0
  622. angr/procedures/definitions/wdk_pshed.py +36 -0
  623. angr/procedures/definitions/wdk_secur32.py +40 -0
  624. angr/procedures/definitions/wdk_vhfum.py +34 -0
  625. angr/procedures/definitions/win32_aclui.py +30 -0
  626. angr/procedures/definitions/win32_activeds.py +68 -0
  627. angr/procedures/definitions/win32_advapi32.py +1684 -0
  628. angr/procedures/definitions/win32_advpack.py +124 -0
  629. angr/procedures/definitions/win32_amsi.py +38 -0
  630. angr/procedures/definitions/win32_api-ms-win-appmodel-runtime-l1-1-1.py +44 -0
  631. angr/procedures/definitions/win32_api-ms-win-appmodel-runtime-l1-1-3.py +34 -0
  632. angr/procedures/definitions/win32_api-ms-win-appmodel-runtime-l1-1-6.py +26 -0
  633. angr/procedures/definitions/win32_api-ms-win-core-apiquery-l2-1-0.py +26 -0
  634. angr/procedures/definitions/win32_api-ms-win-core-backgroundtask-l1-1-0.py +26 -0
  635. angr/procedures/definitions/win32_api-ms-win-core-comm-l1-1-1.py +26 -0
  636. angr/procedures/definitions/win32_api-ms-win-core-comm-l1-1-2.py +26 -0
  637. angr/procedures/definitions/win32_api-ms-win-core-enclave-l1-1-1.py +30 -0
  638. angr/procedures/definitions/win32_api-ms-win-core-errorhandling-l1-1-3.py +26 -0
  639. angr/procedures/definitions/win32_api-ms-win-core-featurestaging-l1-1-0.py +34 -0
  640. angr/procedures/definitions/win32_api-ms-win-core-featurestaging-l1-1-1.py +26 -0
  641. angr/procedures/definitions/win32_api-ms-win-core-file-fromapp-l1-1-0.py +46 -0
  642. angr/procedures/definitions/win32_api-ms-win-core-handle-l1-1-0.py +26 -0
  643. angr/procedures/definitions/win32_api-ms-win-core-ioring-l1-1-0.py +48 -0
  644. angr/procedures/definitions/win32_api-ms-win-core-marshal-l1-1-0.py +32 -0
  645. angr/procedures/definitions/win32_api-ms-win-core-memory-l1-1-3.py +32 -0
  646. angr/procedures/definitions/win32_api-ms-win-core-memory-l1-1-4.py +26 -0
  647. angr/procedures/definitions/win32_api-ms-win-core-memory-l1-1-5.py +30 -0
  648. angr/procedures/definitions/win32_api-ms-win-core-memory-l1-1-6.py +32 -0
  649. angr/procedures/definitions/win32_api-ms-win-core-memory-l1-1-7.py +28 -0
  650. angr/procedures/definitions/win32_api-ms-win-core-memory-l1-1-8.py +30 -0
  651. angr/procedures/definitions/win32_api-ms-win-core-path-l1-1-0.py +68 -0
  652. angr/procedures/definitions/win32_api-ms-win-core-psm-appnotify-l1-1-0.py +28 -0
  653. angr/procedures/definitions/win32_api-ms-win-core-psm-appnotify-l1-1-1.py +28 -0
  654. angr/procedures/definitions/win32_api-ms-win-core-realtime-l1-1-1.py +30 -0
  655. angr/procedures/definitions/win32_api-ms-win-core-realtime-l1-1-2.py +30 -0
  656. angr/procedures/definitions/win32_api-ms-win-core-slapi-l1-1-0.py +26 -0
  657. angr/procedures/definitions/win32_api-ms-win-core-state-helpers-l1-1-0.py +26 -0
  658. angr/procedures/definitions/win32_api-ms-win-core-synch-l1-2-0.py +30 -0
  659. angr/procedures/definitions/win32_api-ms-win-core-sysinfo-l1-2-0.py +26 -0
  660. angr/procedures/definitions/win32_api-ms-win-core-sysinfo-l1-2-3.py +28 -0
  661. angr/procedures/definitions/win32_api-ms-win-core-sysinfo-l1-2-4.py +28 -0
  662. angr/procedures/definitions/win32_api-ms-win-core-sysinfo-l1-2-6.py +26 -0
  663. angr/procedures/definitions/win32_api-ms-win-core-util-l1-1-1.py +28 -0
  664. angr/procedures/definitions/win32_api-ms-win-core-winrt-error-l1-1-0.py +44 -0
  665. angr/procedures/definitions/win32_api-ms-win-core-winrt-error-l1-1-1.py +38 -0
  666. angr/procedures/definitions/win32_api-ms-win-core-winrt-l1-1-0.py +40 -0
  667. angr/procedures/definitions/win32_api-ms-win-core-winrt-registration-l1-1-0.py +24 -0
  668. angr/procedures/definitions/win32_api-ms-win-core-winrt-robuffer-l1-1-0.py +24 -0
  669. angr/procedures/definitions/win32_api-ms-win-core-winrt-roparameterizediid-l1-1-0.py +28 -0
  670. angr/procedures/definitions/win32_api-ms-win-core-winrt-string-l1-1-0.py +76 -0
  671. angr/procedures/definitions/win32_api-ms-win-core-winrt-string-l1-1-1.py +24 -0
  672. angr/procedures/definitions/win32_api-ms-win-core-wow64-l1-1-1.py +30 -0
  673. angr/procedures/definitions/win32_api-ms-win-devices-query-l1-1-0.py +42 -0
  674. angr/procedures/definitions/win32_api-ms-win-devices-query-l1-1-1.py +34 -0
  675. angr/procedures/definitions/win32_api-ms-win-dx-d3dkmt-l1-1-0.py +26 -0
  676. angr/procedures/definitions/win32_api-ms-win-gaming-deviceinformation-l1-1-0.py +26 -0
  677. angr/procedures/definitions/win32_api-ms-win-gaming-expandedresources-l1-1-0.py +30 -0
  678. angr/procedures/definitions/win32_api-ms-win-gaming-tcui-l1-1-0.py +38 -0
  679. angr/procedures/definitions/win32_api-ms-win-gaming-tcui-l1-1-1.py +28 -0
  680. angr/procedures/definitions/win32_api-ms-win-gaming-tcui-l1-1-2.py +38 -0
  681. angr/procedures/definitions/win32_api-ms-win-gaming-tcui-l1-1-3.py +28 -0
  682. angr/procedures/definitions/win32_api-ms-win-gaming-tcui-l1-1-4.py +40 -0
  683. angr/procedures/definitions/win32_api-ms-win-mm-misc-l1-1-1.py +26 -0
  684. angr/procedures/definitions/win32_api-ms-win-net-isolation-l1-1-0.py +40 -0
  685. angr/procedures/definitions/win32_api-ms-win-security-base-l1-2-2.py +26 -0
  686. angr/procedures/definitions/win32_api-ms-win-security-isolatedcontainer-l1-1-0.py +26 -0
  687. angr/procedures/definitions/win32_api-ms-win-security-isolatedcontainer-l1-1-1.py +26 -0
  688. angr/procedures/definitions/win32_api-ms-win-service-core-l1-1-3.py +26 -0
  689. angr/procedures/definitions/win32_api-ms-win-service-core-l1-1-4.py +26 -0
  690. angr/procedures/definitions/win32_api-ms-win-service-core-l1-1-5.py +28 -0
  691. angr/procedures/definitions/win32_api-ms-win-shcore-scaling-l1-1-0.py +30 -0
  692. angr/procedures/definitions/win32_api-ms-win-shcore-scaling-l1-1-1.py +36 -0
  693. angr/procedures/definitions/win32_api-ms-win-shcore-scaling-l1-1-2.py +26 -0
  694. angr/procedures/definitions/win32_api-ms-win-shcore-stream-winrt-l1-1-0.py +28 -0
  695. angr/procedures/definitions/win32_api-ms-win-wsl-api-l1-1-0.py +38 -0
  696. angr/procedures/definitions/win32_apphelp.py +26 -0
  697. angr/procedures/definitions/win32_authz.py +90 -0
  698. angr/procedures/definitions/win32_avicap32.py +32 -0
  699. angr/procedures/definitions/win32_avifil32.py +144 -0
  700. angr/procedures/definitions/win32_avrt.py +52 -0
  701. angr/procedures/definitions/win32_bcp47mrm.py +28 -0
  702. angr/procedures/definitions/win32_bcrypt.py +130 -0
  703. angr/procedures/definitions/win32_bcryptprimitives.py +28 -0
  704. angr/procedures/definitions/win32_bluetoothapis.py +106 -0
  705. angr/procedures/definitions/win32_bthprops.py +34 -0
  706. angr/procedures/definitions/win32_bthprops_cpl.py +36 -0
  707. angr/procedures/definitions/win32_cabinet.py +68 -0
  708. angr/procedures/definitions/win32_certadm.py +60 -0
  709. angr/procedures/definitions/win32_certpoleng.py +40 -0
  710. angr/procedures/definitions/win32_cfgmgr32.py +502 -0
  711. angr/procedures/definitions/win32_chakra.py +198 -0
  712. angr/procedures/definitions/win32_cldapi.py +96 -0
  713. angr/procedures/definitions/win32_clfsw32.py +142 -0
  714. angr/procedures/definitions/win32_clusapi.py +584 -0
  715. angr/procedures/definitions/win32_comctl32.py +254 -0
  716. angr/procedures/definitions/win32_comdlg32.py +66 -0
  717. angr/procedures/definitions/win32_compstui.py +32 -0
  718. angr/procedures/definitions/win32_computecore.py +132 -0
  719. angr/procedures/definitions/win32_computenetwork.py +110 -0
  720. angr/procedures/definitions/win32_computestorage.py +48 -0
  721. angr/procedures/definitions/win32_comsvcs.py +38 -0
  722. angr/procedures/definitions/win32_coremessaging.py +24 -0
  723. angr/procedures/definitions/win32_credui.py +62 -0
  724. angr/procedures/definitions/win32_crypt32.py +482 -0
  725. angr/procedures/definitions/win32_cryptnet.py +34 -0
  726. angr/procedures/definitions/win32_cryptui.py +44 -0
  727. angr/procedures/definitions/win32_cryptxml.py +62 -0
  728. angr/procedures/definitions/win32_cscapi.py +32 -0
  729. angr/procedures/definitions/win32_d2d1.py +50 -0
  730. angr/procedures/definitions/win32_d3d10.py +78 -0
  731. angr/procedures/definitions/win32_d3d10_1.py +28 -0
  732. angr/procedures/definitions/win32_d3d11.py +30 -0
  733. angr/procedures/definitions/win32_d3d12.py +40 -0
  734. angr/procedures/definitions/win32_d3d9.py +46 -0
  735. angr/procedures/definitions/win32_d3dcompiler_47.py +76 -0
  736. angr/procedures/definitions/win32_d3dcsx.py +42 -0
  737. angr/procedures/definitions/win32_davclnt.py +60 -0
  738. angr/procedures/definitions/win32_dbgeng.py +32 -0
  739. angr/procedures/definitions/win32_dbghelp.py +462 -0
  740. angr/procedures/definitions/win32_dbgmodel.py +26 -0
  741. angr/procedures/definitions/win32_dciman32.py +64 -0
  742. angr/procedures/definitions/win32_dcomp.py +48 -0
  743. angr/procedures/definitions/win32_ddraw.py +38 -0
  744. angr/procedures/definitions/win32_deviceaccess.py +26 -0
  745. angr/procedures/definitions/win32_dflayout.py +26 -0
  746. angr/procedures/definitions/win32_dhcpcsvc.py +54 -0
  747. angr/procedures/definitions/win32_dhcpcsvc6.py +36 -0
  748. angr/procedures/definitions/win32_dhcpsapi.py +416 -0
  749. angr/procedures/definitions/win32_diagnosticdataquery.py +94 -0
  750. angr/procedures/definitions/win32_dinput8.py +26 -0
  751. angr/procedures/definitions/win32_directml.py +28 -0
  752. angr/procedures/definitions/win32_dmprocessxmlfiltered.py +26 -0
  753. angr/procedures/definitions/win32_dnsapi.py +152 -0
  754. angr/procedures/definitions/win32_drt.py +56 -0
  755. angr/procedures/definitions/win32_drtprov.py +42 -0
  756. angr/procedures/definitions/win32_drttransport.py +28 -0
  757. angr/procedures/definitions/win32_dsound.py +44 -0
  758. angr/procedures/definitions/win32_dsparse.py +62 -0
  759. angr/procedures/definitions/win32_dsprop.py +38 -0
  760. angr/procedures/definitions/win32_dssec.py +32 -0
  761. angr/procedures/definitions/win32_dsuiext.py +32 -0
  762. angr/procedures/definitions/win32_dwmapi.py +86 -0
  763. angr/procedures/definitions/win32_dwrite.py +26 -0
  764. angr/procedures/definitions/win32_dxcompiler.py +28 -0
  765. angr/procedures/definitions/win32_dxcore.py +26 -0
  766. angr/procedures/definitions/win32_dxgi.py +36 -0
  767. angr/procedures/definitions/win32_dxva2.py +100 -0
  768. angr/procedures/definitions/win32_eappcfg.py +52 -0
  769. angr/procedures/definitions/win32_eappprxy.py +60 -0
  770. angr/procedures/definitions/win32_efswrt.py +28 -0
  771. angr/procedures/definitions/win32_elscore.py +34 -0
  772. angr/procedures/definitions/win32_esent.py +482 -0
  773. angr/procedures/definitions/win32_evr.py +38 -0
  774. angr/procedures/definitions/win32_faultrep.py +32 -0
  775. angr/procedures/definitions/win32_fhsvcctl.py +38 -0
  776. angr/procedures/definitions/win32_firewallapi.py +30 -0
  777. angr/procedures/definitions/win32_fltlib.py +80 -0
  778. angr/procedures/definitions/win32_fontsub.py +28 -0
  779. angr/procedures/definitions/win32_forceinline.py +30 -0
  780. angr/procedures/definitions/win32_fwpuclnt.py +408 -0
  781. angr/procedures/definitions/win32_fxsutility.py +28 -0
  782. angr/procedures/definitions/win32_gdi32.py +886 -0
  783. angr/procedures/definitions/win32_gdiplus.py +1282 -0
  784. angr/procedures/definitions/win32_glu32.py +128 -0
  785. angr/procedures/definitions/win32_gpedit.py +36 -0
  786. angr/procedures/definitions/win32_hhctrl_ocx.py +28 -0
  787. angr/procedures/definitions/win32_hid.py +114 -0
  788. angr/procedures/definitions/win32_hlink.py +80 -0
  789. angr/procedures/definitions/win32_hrtfapo.py +26 -0
  790. angr/procedures/definitions/win32_httpapi.py +110 -0
  791. angr/procedures/definitions/win32_icm32.py +66 -0
  792. angr/procedures/definitions/win32_icmui.py +28 -0
  793. angr/procedures/definitions/win32_icu.py +2074 -0
  794. angr/procedures/definitions/win32_ieframe.py +82 -0
  795. angr/procedures/definitions/win32_imagehlp.py +76 -0
  796. angr/procedures/definitions/win32_imgutil.py +42 -0
  797. angr/procedures/definitions/win32_imm32.py +188 -0
  798. angr/procedures/definitions/win32_infocardapi.py +58 -0
  799. angr/procedures/definitions/win32_inkobjcore.py +78 -0
  800. angr/procedures/definitions/win32_iphlpapi.py +426 -0
  801. angr/procedures/definitions/win32_iscsidsc.py +182 -0
  802. angr/procedures/definitions/win32_isolatedwindowsenvironmentutils.py +28 -0
  803. angr/procedures/definitions/win32_kernel32.py +3185 -0
  804. angr/procedures/definitions/win32_kernelbase.py +36 -0
  805. angr/procedures/definitions/win32_keycredmgr.py +32 -0
  806. angr/procedures/definitions/win32_ksproxy_ax.py +36 -0
  807. angr/procedures/definitions/win32_ksuser.py +40 -0
  808. angr/procedures/definitions/win32_ktmw32.py +102 -0
  809. angr/procedures/definitions/win32_licenseprotection.py +28 -0
  810. angr/procedures/definitions/win32_loadperf.py +48 -0
  811. angr/procedures/definitions/win32_magnification.py +62 -0
  812. angr/procedures/definitions/win32_mapi32.py +156 -0
  813. angr/procedures/definitions/win32_mdmlocalmanagement.py +30 -0
  814. angr/procedures/definitions/win32_mdmregistration.py +54 -0
  815. angr/procedures/definitions/win32_mf.py +148 -0
  816. angr/procedures/definitions/win32_mfcore.py +28 -0
  817. angr/procedures/definitions/win32_mfplat.py +314 -0
  818. angr/procedures/definitions/win32_mfplay.py +26 -0
  819. angr/procedures/definitions/win32_mfreadwrite.py +34 -0
  820. angr/procedures/definitions/win32_mfsensorgroup.py +44 -0
  821. angr/procedures/definitions/win32_mfsrcsnk.py +28 -0
  822. angr/procedures/definitions/win32_mgmtapi.py +42 -0
  823. angr/procedures/definitions/win32_mi.py +26 -0
  824. angr/procedures/definitions/win32_mmdevapi.py +26 -0
  825. angr/procedures/definitions/win32_mpr.py +118 -0
  826. angr/procedures/definitions/win32_mprapi.py +248 -0
  827. angr/procedures/definitions/win32_mqrt.py +92 -0
  828. angr/procedures/definitions/win32_mrmsupport.py +78 -0
  829. angr/procedures/definitions/win32_msacm32.py +108 -0
  830. angr/procedures/definitions/win32_msajapi.py +1118 -0
  831. angr/procedures/definitions/win32_mscms.py +182 -0
  832. angr/procedures/definitions/win32_mscoree.py +78 -0
  833. angr/procedures/definitions/win32_msctfmonitor.py +30 -0
  834. angr/procedures/definitions/win32_msdelta.py +56 -0
  835. angr/procedures/definitions/win32_msdmo.py +46 -0
  836. angr/procedures/definitions/win32_msdrm.py +192 -0
  837. angr/procedures/definitions/win32_msi.py +552 -0
  838. angr/procedures/definitions/win32_msimg32.py +30 -0
  839. angr/procedures/definitions/win32_mspatcha.py +56 -0
  840. angr/procedures/definitions/win32_mspatchc.py +42 -0
  841. angr/procedures/definitions/win32_msports.py +38 -0
  842. angr/procedures/definitions/win32_msrating.py +62 -0
  843. angr/procedures/definitions/win32_mssign32.py +44 -0
  844. angr/procedures/definitions/win32_mstask.py +28 -0
  845. angr/procedures/definitions/win32_msvfw32.py +110 -0
  846. angr/procedures/definitions/win32_mswsock.py +56 -0
  847. angr/procedures/definitions/win32_mtxdm.py +26 -0
  848. angr/procedures/definitions/win32_ncrypt.py +102 -0
  849. angr/procedures/definitions/win32_ndfapi.py +56 -0
  850. angr/procedures/definitions/win32_netapi32.py +436 -0
  851. angr/procedures/definitions/win32_netsh.py +40 -0
  852. angr/procedures/definitions/win32_netshell.py +28 -0
  853. angr/procedures/definitions/win32_newdev.py +46 -0
  854. angr/procedures/definitions/win32_ninput.py +84 -0
  855. angr/procedures/definitions/win32_normaliz.py +28 -0
  856. angr/procedures/definitions/win32_ntdll.py +171 -0
  857. angr/procedures/definitions/win32_ntdllk.py +26 -0
  858. angr/procedures/definitions/win32_ntdsapi.py +186 -0
  859. angr/procedures/definitions/win32_ntlanman.py +44 -0
  860. angr/procedures/definitions/win32_odbc32.py +392 -0
  861. angr/procedures/definitions/win32_odbcbcp.py +78 -0
  862. angr/procedures/definitions/win32_ole32.py +658 -0
  863. angr/procedures/definitions/win32_oleacc.py +58 -0
  864. angr/procedures/definitions/win32_oleaut32.py +834 -0
  865. angr/procedures/definitions/win32_oledlg.py +70 -0
  866. angr/procedures/definitions/win32_ondemandconnroutehelper.py +34 -0
  867. angr/procedures/definitions/win32_opengl32.py +734 -0
  868. angr/procedures/definitions/win32_opmxbox.py +30 -0
  869. angr/procedures/definitions/win32_p2p.py +240 -0
  870. angr/procedures/definitions/win32_p2pgraph.py +98 -0
  871. angr/procedures/definitions/win32_pdh.py +220 -0
  872. angr/procedures/definitions/win32_peerdist.py +80 -0
  873. angr/procedures/definitions/win32_powrprof.py +192 -0
  874. angr/procedures/definitions/win32_prntvpt.py +46 -0
  875. angr/procedures/definitions/win32_projectedfslib.py +62 -0
  876. angr/procedures/definitions/win32_propsys.py +460 -0
  877. angr/procedures/definitions/win32_psapi.py +78 -0
  878. angr/procedures/definitions/win32_quartz.py +28 -0
  879. angr/procedures/definitions/win32_query.py +32 -0
  880. angr/procedures/definitions/win32_qwave.py +46 -0
  881. angr/procedures/definitions/win32_rasapi32.py +192 -0
  882. angr/procedures/definitions/win32_rasdlg.py +36 -0
  883. angr/procedures/definitions/win32_resutils.py +264 -0
  884. angr/procedures/definitions/win32_rometadata.py +24 -0
  885. angr/procedures/definitions/win32_rpcns4.py +146 -0
  886. angr/procedures/definitions/win32_rpcproxy.py +32 -0
  887. angr/procedures/definitions/win32_rpcrt4.py +918 -0
  888. angr/procedures/definitions/win32_rstrtmgr.py +46 -0
  889. angr/procedures/definitions/win32_rtm.py +176 -0
  890. angr/procedures/definitions/win32_rtutils.py +106 -0
  891. angr/procedures/definitions/win32_rtworkq.py +90 -0
  892. angr/procedures/definitions/win32_sas.py +26 -0
  893. angr/procedures/definitions/win32_scarddlg.py +34 -0
  894. angr/procedures/definitions/win32_schannel.py +42 -0
  895. angr/procedures/definitions/win32_sechost.py +28 -0
  896. angr/procedures/definitions/win32_secur32.py +202 -0
  897. angr/procedures/definitions/win32_sensapi.py +30 -0
  898. angr/procedures/definitions/win32_sensorsutilsv2.py +104 -0
  899. angr/procedures/definitions/win32_setupapi.py +692 -0
  900. angr/procedures/definitions/win32_sfc.py +36 -0
  901. angr/procedures/definitions/win32_shdocvw.py +30 -0
  902. angr/procedures/definitions/win32_shell32.py +512 -0
  903. angr/procedures/definitions/win32_shlwapi.py +744 -0
  904. angr/procedures/definitions/win32_slc.py +88 -0
  905. angr/procedures/definitions/win32_slcext.py +32 -0
  906. angr/procedures/definitions/win32_slwga.py +26 -0
  907. angr/procedures/definitions/win32_snmpapi.py +76 -0
  908. angr/procedures/definitions/win32_spoolss.py +76 -0
  909. angr/procedures/definitions/win32_srclient.py +26 -0
  910. angr/procedures/definitions/win32_srpapi.py +46 -0
  911. angr/procedures/definitions/win32_sspicli.py +38 -0
  912. angr/procedures/definitions/win32_sti.py +26 -0
  913. angr/procedures/definitions/win32_t2embed.py +52 -0
  914. angr/procedures/definitions/win32_tapi32.py +522 -0
  915. angr/procedures/definitions/win32_tbs.py +52 -0
  916. angr/procedures/definitions/win32_tdh.py +78 -0
  917. angr/procedures/definitions/win32_tokenbinding.py +44 -0
  918. angr/procedures/definitions/win32_traffic.py +64 -0
  919. angr/procedures/definitions/win32_txfw32.py +42 -0
  920. angr/procedures/definitions/win32_ualapi.py +32 -0
  921. angr/procedures/definitions/win32_uiautomationcore.py +220 -0
  922. angr/procedures/definitions/win32_urlmon.py +178 -0
  923. angr/procedures/definitions/win32_user32.py +1551 -0
  924. angr/procedures/definitions/win32_userenv.py +112 -0
  925. angr/procedures/definitions/win32_usp10.py +104 -0
  926. angr/procedures/definitions/win32_uxtheme.py +178 -0
  927. angr/procedures/definitions/win32_verifier.py +26 -0
  928. angr/procedures/definitions/win32_version.py +52 -0
  929. angr/procedures/definitions/win32_vertdll.py +38 -0
  930. angr/procedures/definitions/win32_virtdisk.py +82 -0
  931. angr/procedures/definitions/win32_vmdevicehost.py +50 -0
  932. angr/procedures/definitions/win32_vmsavedstatedumpprovider.py +110 -0
  933. angr/procedures/definitions/win32_vssapi.py +26 -0
  934. angr/procedures/definitions/win32_wcmapi.py +34 -0
  935. angr/procedures/definitions/win32_wdsbp.py +38 -0
  936. angr/procedures/definitions/win32_wdsclientapi.py +98 -0
  937. angr/procedures/definitions/win32_wdsmc.py +36 -0
  938. angr/procedures/definitions/win32_wdspxe.py +86 -0
  939. angr/procedures/definitions/win32_wdstptc.py +50 -0
  940. angr/procedures/definitions/win32_webauthn.py +50 -0
  941. angr/procedures/definitions/win32_webservices.py +410 -0
  942. angr/procedures/definitions/win32_websocket.py +50 -0
  943. angr/procedures/definitions/win32_wecapi.py +54 -0
  944. angr/procedures/definitions/win32_wer.py +66 -0
  945. angr/procedures/definitions/win32_wevtapi.py +94 -0
  946. angr/procedures/definitions/win32_winbio.py +132 -0
  947. angr/procedures/definitions/win32_windows_ai_machinelearning.py +26 -0
  948. angr/procedures/definitions/win32_windows_data_pdf.py +24 -0
  949. angr/procedures/definitions/win32_windows_media_mediacontrol.py +40 -0
  950. angr/procedures/definitions/win32_windows_networking.py +26 -0
  951. angr/procedures/definitions/win32_windows_ui_xaml.py +28 -0
  952. angr/procedures/definitions/win32_windowscodecs.py +42 -0
  953. angr/procedures/definitions/win32_winfax.py +136 -0
  954. angr/procedures/definitions/win32_winhttp.py +136 -0
  955. angr/procedures/definitions/win32_winhvemulation.py +32 -0
  956. angr/procedures/definitions/win32_winhvplatform.py +156 -0
  957. angr/procedures/definitions/win32_wininet.py +616 -0
  958. angr/procedures/definitions/win32_winml.py +26 -0
  959. angr/procedures/definitions/win32_winmm.py +376 -0
  960. angr/procedures/definitions/win32_winscard.py +164 -0
  961. angr/procedures/definitions/win32_winspool.py +364 -0
  962. angr/procedures/definitions/win32_winspool_drv.py +368 -0
  963. angr/procedures/definitions/win32_wintrust.py +144 -0
  964. angr/procedures/definitions/win32_winusb.py +92 -0
  965. angr/procedures/definitions/win32_wlanapi.py +144 -0
  966. angr/procedures/definitions/win32_wlanui.py +26 -0
  967. angr/procedures/definitions/win32_wldap32.py +510 -0
  968. angr/procedures/definitions/win32_wldp.py +42 -0
  969. angr/procedures/definitions/win32_wmvcore.py +46 -0
  970. angr/procedures/definitions/win32_wnvapi.py +28 -0
  971. angr/procedures/definitions/win32_wofutil.py +46 -0
  972. angr/procedures/definitions/win32_ws2_32.py +344 -0
  973. angr/procedures/definitions/win32_wscapi.py +36 -0
  974. angr/procedures/definitions/win32_wsclient.py +30 -0
  975. angr/procedures/definitions/win32_wsdapi.py +88 -0
  976. angr/procedures/definitions/win32_wsmsvc.py +90 -0
  977. angr/procedures/definitions/win32_wsnmp32.py +122 -0
  978. angr/procedures/definitions/win32_wtsapi32.py +150 -0
  979. angr/procedures/definitions/win32_xaudio2_8.py +32 -0
  980. angr/procedures/definitions/win32_xinput1_4.py +38 -0
  981. angr/procedures/definitions/win32_xinputuap.py +36 -0
  982. angr/procedures/definitions/win32_xmllite.py +36 -0
  983. angr/procedures/definitions/win32_xolehlp.py +32 -0
  984. angr/procedures/definitions/win32_xpsprint.py +28 -0
  985. angr/procedures/glibc/__ctype_b_loc.py +21 -0
  986. angr/procedures/glibc/__ctype_tolower_loc.py +21 -0
  987. angr/procedures/glibc/__ctype_toupper_loc.py +21 -0
  988. angr/procedures/glibc/__errno_location.py +7 -0
  989. angr/procedures/glibc/__init__.py +3 -0
  990. angr/procedures/glibc/__libc_init.py +37 -0
  991. angr/procedures/glibc/__libc_start_main.py +301 -0
  992. angr/procedures/glibc/dynamic_loading.py +20 -0
  993. angr/procedures/glibc/scanf.py +11 -0
  994. angr/procedures/glibc/sscanf.py +6 -0
  995. angr/procedures/gnulib/__init__.py +3 -0
  996. angr/procedures/gnulib/xalloc_die.py +14 -0
  997. angr/procedures/gnulib/xstrtol_fatal.py +14 -0
  998. angr/procedures/java/__init__.py +42 -0
  999. angr/procedures/java/unconstrained.py +65 -0
  1000. angr/procedures/java_io/__init__.py +0 -0
  1001. angr/procedures/java_io/read.py +12 -0
  1002. angr/procedures/java_io/write.py +17 -0
  1003. angr/procedures/java_jni/__init__.py +482 -0
  1004. angr/procedures/java_jni/array_operations.py +312 -0
  1005. angr/procedures/java_jni/class_and_interface_operations.py +31 -0
  1006. angr/procedures/java_jni/field_access.py +173 -0
  1007. angr/procedures/java_jni/global_and_local_refs.py +57 -0
  1008. angr/procedures/java_jni/method_calls.py +365 -0
  1009. angr/procedures/java_jni/not_implemented.py +26 -0
  1010. angr/procedures/java_jni/object_operations.py +94 -0
  1011. angr/procedures/java_jni/string_operations.py +87 -0
  1012. angr/procedures/java_jni/version_information.py +12 -0
  1013. angr/procedures/java_lang/__init__.py +0 -0
  1014. angr/procedures/java_lang/character.py +30 -0
  1015. angr/procedures/java_lang/double.py +24 -0
  1016. angr/procedures/java_lang/exit.py +13 -0
  1017. angr/procedures/java_lang/getsimplename.py +18 -0
  1018. angr/procedures/java_lang/integer.py +43 -0
  1019. angr/procedures/java_lang/load_library.py +9 -0
  1020. angr/procedures/java_lang/math.py +15 -0
  1021. angr/procedures/java_lang/string.py +78 -0
  1022. angr/procedures/java_lang/stringbuilder.py +44 -0
  1023. angr/procedures/java_lang/system.py +18 -0
  1024. angr/procedures/java_util/__init__.py +0 -0
  1025. angr/procedures/java_util/collection.py +35 -0
  1026. angr/procedures/java_util/iterator.py +46 -0
  1027. angr/procedures/java_util/list.py +99 -0
  1028. angr/procedures/java_util/map.py +131 -0
  1029. angr/procedures/java_util/random.py +14 -0
  1030. angr/procedures/java_util/scanner_nextline.py +23 -0
  1031. angr/procedures/libc/__init__.py +3 -0
  1032. angr/procedures/libc/abort.py +9 -0
  1033. angr/procedures/libc/access.py +13 -0
  1034. angr/procedures/libc/atoi.py +14 -0
  1035. angr/procedures/libc/atol.py +13 -0
  1036. angr/procedures/libc/calloc.py +8 -0
  1037. angr/procedures/libc/closelog.py +10 -0
  1038. angr/procedures/libc/err.py +14 -0
  1039. angr/procedures/libc/error.py +54 -0
  1040. angr/procedures/libc/exit.py +11 -0
  1041. angr/procedures/libc/fclose.py +19 -0
  1042. angr/procedures/libc/feof.py +21 -0
  1043. angr/procedures/libc/fflush.py +16 -0
  1044. angr/procedures/libc/fgetc.py +27 -0
  1045. angr/procedures/libc/fgets.py +68 -0
  1046. angr/procedures/libc/fopen.py +63 -0
  1047. angr/procedures/libc/fprintf.py +25 -0
  1048. angr/procedures/libc/fputc.py +23 -0
  1049. angr/procedures/libc/fputs.py +24 -0
  1050. angr/procedures/libc/fread.py +24 -0
  1051. angr/procedures/libc/free.py +9 -0
  1052. angr/procedures/libc/fscanf.py +20 -0
  1053. angr/procedures/libc/fseek.py +34 -0
  1054. angr/procedures/libc/ftell.py +22 -0
  1055. angr/procedures/libc/fwrite.py +19 -0
  1056. angr/procedures/libc/getchar.py +13 -0
  1057. angr/procedures/libc/getdelim.py +99 -0
  1058. angr/procedures/libc/getegid.py +8 -0
  1059. angr/procedures/libc/geteuid.py +8 -0
  1060. angr/procedures/libc/getgid.py +8 -0
  1061. angr/procedures/libc/gets.py +68 -0
  1062. angr/procedures/libc/getuid.py +8 -0
  1063. angr/procedures/libc/malloc.py +12 -0
  1064. angr/procedures/libc/memcmp.py +69 -0
  1065. angr/procedures/libc/memcpy.py +38 -0
  1066. angr/procedures/libc/memset.py +72 -0
  1067. angr/procedures/libc/openlog.py +10 -0
  1068. angr/procedures/libc/perror.py +13 -0
  1069. angr/procedures/libc/printf.py +34 -0
  1070. angr/procedures/libc/putchar.py +13 -0
  1071. angr/procedures/libc/puts.py +19 -0
  1072. angr/procedures/libc/rand.py +8 -0
  1073. angr/procedures/libc/realloc.py +8 -0
  1074. angr/procedures/libc/rewind.py +12 -0
  1075. angr/procedures/libc/scanf.py +20 -0
  1076. angr/procedures/libc/setbuf.py +9 -0
  1077. angr/procedures/libc/setvbuf.py +7 -0
  1078. angr/procedures/libc/snprintf.py +36 -0
  1079. angr/procedures/libc/sprintf.py +25 -0
  1080. angr/procedures/libc/srand.py +7 -0
  1081. angr/procedures/libc/sscanf.py +13 -0
  1082. angr/procedures/libc/stpcpy.py +18 -0
  1083. angr/procedures/libc/strcat.py +14 -0
  1084. angr/procedures/libc/strchr.py +48 -0
  1085. angr/procedures/libc/strcmp.py +31 -0
  1086. angr/procedures/libc/strcpy.py +13 -0
  1087. angr/procedures/libc/strlen.py +114 -0
  1088. angr/procedures/libc/strncat.py +19 -0
  1089. angr/procedures/libc/strncmp.py +183 -0
  1090. angr/procedures/libc/strncpy.py +22 -0
  1091. angr/procedures/libc/strnlen.py +13 -0
  1092. angr/procedures/libc/strstr.py +101 -0
  1093. angr/procedures/libc/strtol.py +261 -0
  1094. angr/procedures/libc/strtoul.py +9 -0
  1095. angr/procedures/libc/system.py +13 -0
  1096. angr/procedures/libc/time.py +9 -0
  1097. angr/procedures/libc/tmpnam.py +20 -0
  1098. angr/procedures/libc/tolower.py +10 -0
  1099. angr/procedures/libc/toupper.py +10 -0
  1100. angr/procedures/libc/ungetc.py +20 -0
  1101. angr/procedures/libc/vsnprintf.py +17 -0
  1102. angr/procedures/libc/wchar.py +16 -0
  1103. angr/procedures/libstdcpp/__init__.py +0 -0
  1104. angr/procedures/libstdcpp/_unwind_resume.py +11 -0
  1105. angr/procedures/libstdcpp/std____throw_bad_alloc.py +13 -0
  1106. angr/procedures/libstdcpp/std____throw_bad_cast.py +13 -0
  1107. angr/procedures/libstdcpp/std____throw_length_error.py +13 -0
  1108. angr/procedures/libstdcpp/std____throw_logic_error.py +13 -0
  1109. angr/procedures/libstdcpp/std__terminate.py +13 -0
  1110. angr/procedures/linux_kernel/__init__.py +3 -0
  1111. angr/procedures/linux_kernel/access.py +18 -0
  1112. angr/procedures/linux_kernel/arch_prctl.py +34 -0
  1113. angr/procedures/linux_kernel/arm_user_helpers.py +59 -0
  1114. angr/procedures/linux_kernel/brk.py +18 -0
  1115. angr/procedures/linux_kernel/cwd.py +28 -0
  1116. angr/procedures/linux_kernel/fstat.py +138 -0
  1117. angr/procedures/linux_kernel/fstat64.py +170 -0
  1118. angr/procedures/linux_kernel/futex.py +17 -0
  1119. angr/procedures/linux_kernel/getegid.py +17 -0
  1120. angr/procedures/linux_kernel/geteuid.py +17 -0
  1121. angr/procedures/linux_kernel/getgid.py +17 -0
  1122. angr/procedures/linux_kernel/getpid.py +14 -0
  1123. angr/procedures/linux_kernel/getrlimit.py +24 -0
  1124. angr/procedures/linux_kernel/gettid.py +9 -0
  1125. angr/procedures/linux_kernel/getuid.py +17 -0
  1126. angr/procedures/linux_kernel/iovec.py +47 -0
  1127. angr/procedures/linux_kernel/lseek.py +42 -0
  1128. angr/procedures/linux_kernel/mmap.py +16 -0
  1129. angr/procedures/linux_kernel/mprotect.py +42 -0
  1130. angr/procedures/linux_kernel/munmap.py +8 -0
  1131. angr/procedures/linux_kernel/openat.py +26 -0
  1132. angr/procedures/linux_kernel/set_tid_address.py +8 -0
  1133. angr/procedures/linux_kernel/sigaction.py +19 -0
  1134. angr/procedures/linux_kernel/sigprocmask.py +23 -0
  1135. angr/procedures/linux_kernel/stat.py +23 -0
  1136. angr/procedures/linux_kernel/sysinfo.py +59 -0
  1137. angr/procedures/linux_kernel/tgkill.py +10 -0
  1138. angr/procedures/linux_kernel/time.py +34 -0
  1139. angr/procedures/linux_kernel/uid.py +30 -0
  1140. angr/procedures/linux_kernel/uname.py +29 -0
  1141. angr/procedures/linux_kernel/unlink.py +22 -0
  1142. angr/procedures/linux_kernel/vsyscall.py +16 -0
  1143. angr/procedures/linux_loader/__init__.py +3 -0
  1144. angr/procedures/linux_loader/_dl_initial_error_catch_tsd.py +7 -0
  1145. angr/procedures/linux_loader/_dl_rtld_lock.py +15 -0
  1146. angr/procedures/linux_loader/sim_loader.py +54 -0
  1147. angr/procedures/linux_loader/tls.py +40 -0
  1148. angr/procedures/msvcr/__getmainargs.py +16 -0
  1149. angr/procedures/msvcr/__init__.py +4 -0
  1150. angr/procedures/msvcr/_initterm.py +38 -0
  1151. angr/procedures/msvcr/fmode.py +31 -0
  1152. angr/procedures/ntdll/__init__.py +0 -0
  1153. angr/procedures/ntdll/exceptions.py +60 -0
  1154. angr/procedures/posix/__init__.py +3 -0
  1155. angr/procedures/posix/accept.py +29 -0
  1156. angr/procedures/posix/bind.py +13 -0
  1157. angr/procedures/posix/bzero.py +9 -0
  1158. angr/procedures/posix/chroot.py +27 -0
  1159. angr/procedures/posix/close.py +9 -0
  1160. angr/procedures/posix/closedir.py +7 -0
  1161. angr/procedures/posix/dup.py +56 -0
  1162. angr/procedures/posix/fcntl.py +10 -0
  1163. angr/procedures/posix/fdopen.py +76 -0
  1164. angr/procedures/posix/fileno.py +18 -0
  1165. angr/procedures/posix/fork.py +13 -0
  1166. angr/procedures/posix/getenv.py +35 -0
  1167. angr/procedures/posix/gethostbyname.py +43 -0
  1168. angr/procedures/posix/getpass.py +19 -0
  1169. angr/procedures/posix/getsockopt.py +11 -0
  1170. angr/procedures/posix/htonl.py +11 -0
  1171. angr/procedures/posix/htons.py +11 -0
  1172. angr/procedures/posix/inet_ntoa.py +59 -0
  1173. angr/procedures/posix/listen.py +13 -0
  1174. angr/procedures/posix/mmap.py +144 -0
  1175. angr/procedures/posix/open.py +18 -0
  1176. angr/procedures/posix/opendir.py +10 -0
  1177. angr/procedures/posix/poll.py +55 -0
  1178. angr/procedures/posix/pread64.py +46 -0
  1179. angr/procedures/posix/pthread.py +87 -0
  1180. angr/procedures/posix/pwrite64.py +46 -0
  1181. angr/procedures/posix/read.py +13 -0
  1182. angr/procedures/posix/readdir.py +62 -0
  1183. angr/procedures/posix/recv.py +13 -0
  1184. angr/procedures/posix/recvfrom.py +13 -0
  1185. angr/procedures/posix/select.py +48 -0
  1186. angr/procedures/posix/send.py +23 -0
  1187. angr/procedures/posix/setsockopt.py +9 -0
  1188. angr/procedures/posix/sigaction.py +23 -0
  1189. angr/procedures/posix/sim_time.py +48 -0
  1190. angr/procedures/posix/sleep.py +8 -0
  1191. angr/procedures/posix/socket.py +18 -0
  1192. angr/procedures/posix/strcasecmp.py +26 -0
  1193. angr/procedures/posix/strdup.py +18 -0
  1194. angr/procedures/posix/strtok_r.py +64 -0
  1195. angr/procedures/posix/syslog.py +15 -0
  1196. angr/procedures/posix/tz.py +9 -0
  1197. angr/procedures/posix/unlink.py +11 -0
  1198. angr/procedures/posix/usleep.py +8 -0
  1199. angr/procedures/posix/write.py +13 -0
  1200. angr/procedures/procedure_dict.py +50 -0
  1201. angr/procedures/stubs/CallReturn.py +13 -0
  1202. angr/procedures/stubs/NoReturnUnconstrained.py +13 -0
  1203. angr/procedures/stubs/Nop.py +7 -0
  1204. angr/procedures/stubs/PathTerminator.py +9 -0
  1205. angr/procedures/stubs/Redirect.py +18 -0
  1206. angr/procedures/stubs/ReturnChar.py +11 -0
  1207. angr/procedures/stubs/ReturnUnconstrained.py +24 -0
  1208. angr/procedures/stubs/UnresolvableCallTarget.py +9 -0
  1209. angr/procedures/stubs/UnresolvableJumpTarget.py +9 -0
  1210. angr/procedures/stubs/UserHook.py +18 -0
  1211. angr/procedures/stubs/__init__.py +3 -0
  1212. angr/procedures/stubs/b64_decode.py +15 -0
  1213. angr/procedures/stubs/caller.py +14 -0
  1214. angr/procedures/stubs/crazy_scanf.py +20 -0
  1215. angr/procedures/stubs/format_parser.py +669 -0
  1216. angr/procedures/stubs/syscall_stub.py +24 -0
  1217. angr/procedures/testing/__init__.py +3 -0
  1218. angr/procedures/testing/manyargs.py +9 -0
  1219. angr/procedures/testing/retreg.py +8 -0
  1220. angr/procedures/tracer/__init__.py +4 -0
  1221. angr/procedures/tracer/random.py +9 -0
  1222. angr/procedures/tracer/receive.py +23 -0
  1223. angr/procedures/tracer/transmit.py +26 -0
  1224. angr/procedures/uclibc/__init__.py +3 -0
  1225. angr/procedures/uclibc/__uClibc_main.py +10 -0
  1226. angr/procedures/win32/EncodePointer.py +7 -0
  1227. angr/procedures/win32/ExitProcess.py +9 -0
  1228. angr/procedures/win32/GetCommandLine.py +12 -0
  1229. angr/procedures/win32/GetCurrentProcessId.py +7 -0
  1230. angr/procedures/win32/GetCurrentThreadId.py +7 -0
  1231. angr/procedures/win32/GetLastInputInfo.py +40 -0
  1232. angr/procedures/win32/GetModuleHandle.py +29 -0
  1233. angr/procedures/win32/GetProcessAffinityMask.py +37 -0
  1234. angr/procedures/win32/InterlockedExchange.py +15 -0
  1235. angr/procedures/win32/IsProcessorFeaturePresent.py +7 -0
  1236. angr/procedures/win32/VirtualAlloc.py +114 -0
  1237. angr/procedures/win32/VirtualProtect.py +60 -0
  1238. angr/procedures/win32/__init__.py +3 -0
  1239. angr/procedures/win32/critical_section.py +12 -0
  1240. angr/procedures/win32/dynamic_loading.py +104 -0
  1241. angr/procedures/win32/file_handles.py +47 -0
  1242. angr/procedures/win32/gethostbyname.py +12 -0
  1243. angr/procedures/win32/heap.py +45 -0
  1244. angr/procedures/win32/is_bad_ptr.py +26 -0
  1245. angr/procedures/win32/local_storage.py +88 -0
  1246. angr/procedures/win32/mutex.py +11 -0
  1247. angr/procedures/win32/sim_time.py +135 -0
  1248. angr/procedures/win32/system_paths.py +35 -0
  1249. angr/procedures/win32_kernel/ExAllocatePool.py +13 -0
  1250. angr/procedures/win32_kernel/ExFreePoolWithTag.py +8 -0
  1251. angr/procedures/win32_kernel/__fastfail.py +15 -0
  1252. angr/procedures/win32_kernel/__init__.py +3 -0
  1253. angr/procedures/win_user32/__init__.py +0 -0
  1254. angr/procedures/win_user32/chars.py +15 -0
  1255. angr/procedures/win_user32/keyboard.py +14 -0
  1256. angr/procedures/win_user32/messagebox.py +49 -0
  1257. angr/project.py +847 -0
  1258. angr/protos/__init__.py +19 -0
  1259. angr/protos/cfg_pb2.py +31 -0
  1260. angr/protos/function_pb2.py +27 -0
  1261. angr/protos/primitives_pb2.py +52 -0
  1262. angr/protos/variables_pb2.py +44 -0
  1263. angr/protos/xrefs_pb2.py +25 -0
  1264. angr/py.typed +1 -0
  1265. angr/rustylib.abi3.so +0 -0
  1266. angr/serializable.py +66 -0
  1267. angr/sim_manager.py +971 -0
  1268. angr/sim_options.py +438 -0
  1269. angr/sim_procedure.py +606 -0
  1270. angr/sim_state.py +901 -0
  1271. angr/sim_state_options.py +403 -0
  1272. angr/sim_type.py +3702 -0
  1273. angr/sim_variable.py +465 -0
  1274. angr/simos/__init__.py +47 -0
  1275. angr/simos/cgc.py +153 -0
  1276. angr/simos/javavm.py +458 -0
  1277. angr/simos/linux.py +509 -0
  1278. angr/simos/simos.py +444 -0
  1279. angr/simos/snimmuc_nxp.py +149 -0
  1280. angr/simos/userland.py +163 -0
  1281. angr/simos/windows.py +601 -0
  1282. angr/simos/xbox.py +32 -0
  1283. angr/slicer.py +352 -0
  1284. angr/state_hierarchy.py +262 -0
  1285. angr/state_plugins/__init__.py +84 -0
  1286. angr/state_plugins/callstack.py +398 -0
  1287. angr/state_plugins/cgc.py +155 -0
  1288. angr/state_plugins/debug_variables.py +192 -0
  1289. angr/state_plugins/filesystem.py +463 -0
  1290. angr/state_plugins/gdb.py +148 -0
  1291. angr/state_plugins/globals.py +65 -0
  1292. angr/state_plugins/heap/__init__.py +15 -0
  1293. angr/state_plugins/heap/heap_base.py +128 -0
  1294. angr/state_plugins/heap/heap_brk.py +136 -0
  1295. angr/state_plugins/heap/heap_freelist.py +213 -0
  1296. angr/state_plugins/heap/heap_libc.py +46 -0
  1297. angr/state_plugins/heap/heap_ptmalloc.py +620 -0
  1298. angr/state_plugins/heap/utils.py +22 -0
  1299. angr/state_plugins/history.py +564 -0
  1300. angr/state_plugins/inspect.py +375 -0
  1301. angr/state_plugins/javavm_classloader.py +134 -0
  1302. angr/state_plugins/jni_references.py +95 -0
  1303. angr/state_plugins/libc.py +1263 -0
  1304. angr/state_plugins/light_registers.py +168 -0
  1305. angr/state_plugins/log.py +84 -0
  1306. angr/state_plugins/loop_data.py +92 -0
  1307. angr/state_plugins/plugin.py +170 -0
  1308. angr/state_plugins/posix.py +703 -0
  1309. angr/state_plugins/preconstrainer.py +196 -0
  1310. angr/state_plugins/scratch.py +173 -0
  1311. angr/state_plugins/sim_action.py +326 -0
  1312. angr/state_plugins/sim_action_object.py +271 -0
  1313. angr/state_plugins/sim_event.py +59 -0
  1314. angr/state_plugins/solver.py +1127 -0
  1315. angr/state_plugins/symbolizer.py +291 -0
  1316. angr/state_plugins/trace_additions.py +738 -0
  1317. angr/state_plugins/uc_manager.py +94 -0
  1318. angr/state_plugins/unicorn_engine.py +1886 -0
  1319. angr/state_plugins/view.py +340 -0
  1320. angr/storage/__init__.py +15 -0
  1321. angr/storage/file.py +1210 -0
  1322. angr/storage/memory_mixins/__init__.py +317 -0
  1323. angr/storage/memory_mixins/actions_mixin.py +72 -0
  1324. angr/storage/memory_mixins/address_concretization_mixin.py +384 -0
  1325. angr/storage/memory_mixins/bvv_conversion_mixin.py +73 -0
  1326. angr/storage/memory_mixins/clouseau_mixin.py +137 -0
  1327. angr/storage/memory_mixins/conditional_store_mixin.py +25 -0
  1328. angr/storage/memory_mixins/convenient_mappings_mixin.py +256 -0
  1329. angr/storage/memory_mixins/default_filler_mixin.py +144 -0
  1330. angr/storage/memory_mixins/dirty_addrs_mixin.py +11 -0
  1331. angr/storage/memory_mixins/hex_dumper_mixin.py +82 -0
  1332. angr/storage/memory_mixins/javavm_memory_mixin.py +392 -0
  1333. angr/storage/memory_mixins/keyvalue_memory_mixin.py +42 -0
  1334. angr/storage/memory_mixins/label_merger_mixin.py +31 -0
  1335. angr/storage/memory_mixins/memory_mixin.py +174 -0
  1336. angr/storage/memory_mixins/multi_value_merger_mixin.py +79 -0
  1337. angr/storage/memory_mixins/name_resolution_mixin.py +67 -0
  1338. angr/storage/memory_mixins/paged_memory/__init__.py +0 -0
  1339. angr/storage/memory_mixins/paged_memory/page_backer_mixins.py +266 -0
  1340. angr/storage/memory_mixins/paged_memory/paged_memory_mixin.py +743 -0
  1341. angr/storage/memory_mixins/paged_memory/paged_memory_multivalue_mixin.py +65 -0
  1342. angr/storage/memory_mixins/paged_memory/pages/__init__.py +26 -0
  1343. angr/storage/memory_mixins/paged_memory/pages/base.py +31 -0
  1344. angr/storage/memory_mixins/paged_memory/pages/cooperation.py +341 -0
  1345. angr/storage/memory_mixins/paged_memory/pages/history_tracking_mixin.py +92 -0
  1346. angr/storage/memory_mixins/paged_memory/pages/ispo_mixin.py +55 -0
  1347. angr/storage/memory_mixins/paged_memory/pages/list_page.py +338 -0
  1348. angr/storage/memory_mixins/paged_memory/pages/multi_values.py +324 -0
  1349. angr/storage/memory_mixins/paged_memory/pages/mv_list_page.py +419 -0
  1350. angr/storage/memory_mixins/paged_memory/pages/permissions_mixin.py +36 -0
  1351. angr/storage/memory_mixins/paged_memory/pages/refcount_mixin.py +52 -0
  1352. angr/storage/memory_mixins/paged_memory/pages/ultra_page.py +503 -0
  1353. angr/storage/memory_mixins/paged_memory/privileged_mixin.py +36 -0
  1354. angr/storage/memory_mixins/paged_memory/stack_allocation_mixin.py +74 -0
  1355. angr/storage/memory_mixins/regioned_memory/__init__.py +17 -0
  1356. angr/storage/memory_mixins/regioned_memory/abstract_address_descriptor.py +36 -0
  1357. angr/storage/memory_mixins/regioned_memory/abstract_merger_mixin.py +31 -0
  1358. angr/storage/memory_mixins/regioned_memory/region_category_mixin.py +9 -0
  1359. angr/storage/memory_mixins/regioned_memory/region_data.py +246 -0
  1360. angr/storage/memory_mixins/regioned_memory/region_meta_mixin.py +241 -0
  1361. angr/storage/memory_mixins/regioned_memory/regioned_address_concretization_mixin.py +119 -0
  1362. angr/storage/memory_mixins/regioned_memory/regioned_memory_mixin.py +441 -0
  1363. angr/storage/memory_mixins/regioned_memory/static_find_mixin.py +69 -0
  1364. angr/storage/memory_mixins/simple_interface_mixin.py +71 -0
  1365. angr/storage/memory_mixins/simplification_mixin.py +15 -0
  1366. angr/storage/memory_mixins/size_resolution_mixin.py +143 -0
  1367. angr/storage/memory_mixins/slotted_memory.py +140 -0
  1368. angr/storage/memory_mixins/smart_find_mixin.py +161 -0
  1369. angr/storage/memory_mixins/symbolic_merger_mixin.py +16 -0
  1370. angr/storage/memory_mixins/top_merger_mixin.py +25 -0
  1371. angr/storage/memory_mixins/underconstrained_mixin.py +67 -0
  1372. angr/storage/memory_mixins/unwrapper_mixin.py +26 -0
  1373. angr/storage/memory_object.py +195 -0
  1374. angr/tablespecs.py +91 -0
  1375. angr/unicornlib.so +0 -0
  1376. angr/utils/__init__.py +46 -0
  1377. angr/utils/ail.py +70 -0
  1378. angr/utils/algo.py +34 -0
  1379. angr/utils/bits.py +46 -0
  1380. angr/utils/constants.py +9 -0
  1381. angr/utils/cowdict.py +63 -0
  1382. angr/utils/cpp.py +17 -0
  1383. angr/utils/doms.py +149 -0
  1384. angr/utils/dynamic_dictlist.py +89 -0
  1385. angr/utils/endness.py +18 -0
  1386. angr/utils/enums_conv.py +97 -0
  1387. angr/utils/env.py +12 -0
  1388. angr/utils/formatting.py +128 -0
  1389. angr/utils/funcid.py +159 -0
  1390. angr/utils/graph.py +933 -0
  1391. angr/utils/lazy_import.py +13 -0
  1392. angr/utils/library.py +212 -0
  1393. angr/utils/loader.py +55 -0
  1394. angr/utils/mp.py +66 -0
  1395. angr/utils/orderedset.py +74 -0
  1396. angr/utils/ssa/__init__.py +457 -0
  1397. angr/utils/ssa/tmp_uses_collector.py +23 -0
  1398. angr/utils/ssa/vvar_uses_collector.py +37 -0
  1399. angr/utils/tagged_interval_map.py +112 -0
  1400. angr/utils/timing.py +74 -0
  1401. angr/utils/types.py +151 -0
  1402. angr/utils/vex.py +11 -0
  1403. angr/vaults.py +367 -0
  1404. angr-9.2.166.dist-info/METADATA +110 -0
  1405. angr-9.2.166.dist-info/RECORD +1409 -0
  1406. angr-9.2.166.dist-info/WHEEL +5 -0
  1407. angr-9.2.166.dist-info/entry_points.txt +2 -0
  1408. angr-9.2.166.dist-info/licenses/LICENSE +27 -0
  1409. angr-9.2.166.dist-info/top_level.txt +1 -0
@@ -0,0 +1,2580 @@
1
+ # pylint:disable=line-too-long,missing-class-docstring,no-self-use
2
+ from __future__ import annotations
3
+ import logging
4
+ from typing import Generic, cast, TypeVar
5
+
6
+ from collections.abc import Iterable
7
+ from collections import defaultdict
8
+ import contextlib
9
+
10
+ import claripy
11
+ import archinfo
12
+ from archinfo import RegisterName
13
+ from unique_log_filter import UniqueLogFilter
14
+
15
+ import angr
16
+ from .errors import AngrTypeError
17
+ from .sim_type import (
18
+ NamedTypeMixin,
19
+ SimType,
20
+ SimTypeChar,
21
+ SimTypePointer,
22
+ SimTypeFixedSizeArray,
23
+ SimTypeArray,
24
+ SimTypeString,
25
+ SimTypeFunction,
26
+ SimTypeFloat,
27
+ SimTypeDouble,
28
+ SimTypeReg,
29
+ SimStruct,
30
+ SimStructValue,
31
+ SimTypeInt,
32
+ SimTypeNum,
33
+ SimUnion,
34
+ SimTypeBottom,
35
+ parse_signature,
36
+ SimTypeReference,
37
+ )
38
+ from .state_plugins.sim_action_object import SimActionObject
39
+
40
+ l = logging.getLogger(name=__name__)
41
+ l.addFilter(UniqueLogFilter())
42
+
43
+ T = TypeVar("T", bound="SimFunctionArgument")
44
+
45
+
46
+ class PointerWrapper:
47
+ def __init__(self, value, buffer=False):
48
+ self.value = value
49
+ self.buffer = buffer
50
+
51
+
52
+ class AllocHelper:
53
+ def __init__(self, ptrsize):
54
+ self.base = claripy.BVS("alloc_base", ptrsize)
55
+ self.ptr = self.base
56
+ self.stores = {}
57
+ self.store_asts = {}
58
+
59
+ def alloc(self, size):
60
+ out = self.ptr
61
+ self.ptr += size
62
+ return out
63
+
64
+ def dump(self, val, state, loc=None):
65
+ if loc is None:
66
+ loc = self.stack_loc(val, state.arch)
67
+ self.stores[self.ptr] = (val, loc)
68
+ return self.alloc(self.calc_size(val, state.arch))
69
+
70
+ def translate(self, val, base):
71
+ if type(val) is SimStructValue:
72
+ return SimStructValue(
73
+ val.struct, {field: self.translate(subval, base) for field, subval in val._values.items()}
74
+ )
75
+ if isinstance(val, claripy.ast.Bits):
76
+ return claripy.replace(val, self.base, base)
77
+ if type(val) is list:
78
+ return [self.translate(subval, base) for subval in val]
79
+ raise TypeError(type(val))
80
+
81
+ def apply(self, state, base):
82
+ for ptr, (val, loc) in self.stores.items():
83
+ translated_val = self.translate(val, base)
84
+ translated_ptr = self.translate(ptr, base)
85
+ loc.set_value(state, translated_val, stack_base=translated_ptr)
86
+
87
+ def size(self):
88
+ val = self.translate(self.ptr, claripy.BVV(0, len(self.ptr)))
89
+ assert isinstance(val, claripy.ast.Base) and val.op == "BVV"
90
+ assert isinstance(val.args[0], int)
91
+ return abs(val.args[0])
92
+
93
+ @classmethod
94
+ def calc_size(cls, val, arch):
95
+ if type(val) is SimStructValue:
96
+ return val.struct.size // arch.byte_width
97
+ if isinstance(val, claripy.ast.Bits):
98
+ return len(val) // arch.byte_width
99
+ if type(val) is list:
100
+ # TODO real strides
101
+ if len(val) == 0:
102
+ return 0
103
+ return cls.calc_size(val[0], arch) * len(val)
104
+ raise TypeError(type(val))
105
+
106
+ @classmethod
107
+ def stack_loc(cls, val, arch, offset=0):
108
+ if isinstance(val, claripy.ast.Bits):
109
+ return SimStackArg(offset, len(val) // arch.byte_width)
110
+ if type(val) is list:
111
+ # TODO real strides
112
+ if len(val) == 0:
113
+ return SimArrayArg([])
114
+ stride = cls.calc_size(val[0], arch)
115
+ return SimArrayArg([cls.stack_loc(subval, arch, offset + i * stride) for i, subval in enumerate(val)])
116
+ if type(val) is SimStructValue:
117
+ return SimStructArg(
118
+ val.struct,
119
+ {
120
+ field: cls.stack_loc(subval, arch, offset + val.struct.offsets[field])
121
+ for field, subval in val._values.items()
122
+ },
123
+ )
124
+ raise TypeError(type(val))
125
+
126
+
127
+ def refine_locs_with_struct_type(
128
+ arch: archinfo.Arch,
129
+ locs: list,
130
+ arg_type: SimType,
131
+ offset: int = 0,
132
+ treat_bot_as_int=True,
133
+ treat_unsupported_as_int=True,
134
+ ):
135
+ # CONTRACT FOR USING THIS METHOD: locs must be a list of locs which are all wordsize
136
+ # ADDITIONAL NUANCE: this will not respect the need for big-endian integers to be stored at the end of words.
137
+ # that's why this is named with_struct_type, because it will blindly trust the offsets given to it.
138
+
139
+ if treat_bot_as_int and isinstance(arg_type, SimTypeBottom):
140
+ arg_type = SimTypeInt(label=arg_type.label).with_arch(arch)
141
+
142
+ if isinstance(arg_type, (SimTypeReg, SimTypeNum, SimTypeFloat)):
143
+ assert arg_type.size is not None
144
+ seen_bytes = 0
145
+ pieces = []
146
+ while seen_bytes < arg_type.size // arch.byte_width:
147
+ start_offset = offset + seen_bytes
148
+ chunk = start_offset // arch.bytes
149
+ chunk_offset = start_offset % arch.bytes
150
+ chunk_remaining = arch.bytes - chunk_offset
151
+ type_remaining = arg_type.size // arch.byte_width - seen_bytes
152
+ use_bytes = min(chunk_remaining, type_remaining)
153
+ pieces.append(locs[chunk].refine(size=use_bytes, offset=chunk_offset))
154
+ seen_bytes += use_bytes
155
+
156
+ piece = pieces[0] if len(pieces) == 1 else SimComboArg(pieces)
157
+ if isinstance(arg_type, SimTypeFloat):
158
+ piece.is_fp = True
159
+ return piece
160
+ if isinstance(arg_type, SimTypeFixedSizeArray):
161
+ assert arg_type.elem_type.size is not None and arg_type.length is not None
162
+ # TODO explicit stride
163
+ locs_list = [
164
+ refine_locs_with_struct_type(
165
+ arch, locs, arg_type.elem_type, offset=offset + i * arg_type.elem_type.size // arch.byte_width
166
+ )
167
+ for i in range(arg_type.length)
168
+ ]
169
+ return SimArrayArg(locs_list)
170
+ if isinstance(arg_type, SimStruct):
171
+ locs_dict = {
172
+ field: refine_locs_with_struct_type(arch, locs, field_ty, offset=offset + arg_type.offsets[field])
173
+ for field, field_ty in arg_type.fields.items()
174
+ }
175
+ return SimStructArg(arg_type, locs_dict)
176
+ if isinstance(arg_type, SimUnion):
177
+ # Treat a SimUnion as functionality equivalent to its longest member
178
+ for member in arg_type.members.values():
179
+ if member.size == arg_type.size:
180
+ return refine_locs_with_struct_type(arch, locs, member, offset)
181
+
182
+ # for all other types, we basically treat them as integers until someone implements proper layouting logic
183
+ if treat_unsupported_as_int:
184
+ arg_type = SimTypeInt().with_arch(arch)
185
+ return refine_locs_with_struct_type(
186
+ arch,
187
+ locs,
188
+ arg_type,
189
+ offset=offset,
190
+ treat_bot_as_int=treat_bot_as_int,
191
+ treat_unsupported_as_int=treat_unsupported_as_int,
192
+ )
193
+
194
+ raise TypeError(f"I don't know how to lay out a {arg_type}")
195
+
196
+
197
+ class SerializableIterator:
198
+ def __iter__(self):
199
+ return self
200
+
201
+ def __next__(self):
202
+ raise NotImplementedError
203
+
204
+ def getstate(self):
205
+ raise NotImplementedError
206
+
207
+ def setstate(self, state):
208
+ raise NotImplementedError
209
+
210
+
211
+ class SerializableListIterator(SerializableIterator):
212
+ def __init__(self, lst):
213
+ self._lst = lst
214
+ self._index = 0
215
+
216
+ def __next__(self):
217
+ if self._index >= len(self._lst):
218
+ raise StopIteration
219
+ result = self._lst[self._index]
220
+ self._index += 1
221
+ return result
222
+
223
+ def getstate(self):
224
+ return self._index
225
+
226
+ def setstate(self, state):
227
+ self._index = state
228
+
229
+
230
+ class SerializableCounter(SerializableIterator):
231
+ def __init__(self, start, stride, mapping=lambda x: x):
232
+ self._next = start
233
+ self._stride = stride
234
+ self._mapping = mapping
235
+
236
+ def __next__(self):
237
+ result = self._mapping(self._next)
238
+ self._next += self._stride
239
+ return result
240
+
241
+ def getstate(self):
242
+ return self._next
243
+
244
+ def setstate(self, state):
245
+ self._next = state
246
+
247
+
248
+ class SimFunctionArgument:
249
+ """
250
+ Represent a generic function argument.
251
+
252
+ :ivar int size: The size of the argument, in number of bytes.
253
+ :ivar bool is_fp: Whether loads from this location should return a floating point bitvector
254
+ """
255
+
256
+ def __init__(self, size: int, is_fp: bool = False):
257
+ self.size = size
258
+ self.is_fp = is_fp
259
+
260
+ def __ne__(self, other):
261
+ return not self == other
262
+
263
+ def __hash__(self):
264
+ return hash(("function_argument", self.size))
265
+
266
+ def check_value_set(self, value, arch):
267
+ if not isinstance(value, claripy.ast.Base) and self.size is None:
268
+ raise TypeError("Only claripy objects may be stored through SimFunctionArgument when size is not provided")
269
+ if self.size is not None and isinstance(value, claripy.ast.Base) and self.size * arch.byte_width < value.length:
270
+ raise TypeError(f"{value} doesn't fit in an argument of size {self.size}")
271
+ if isinstance(value, int):
272
+ value = claripy.BVV(value, self.size * arch.byte_width)
273
+ if isinstance(value, float):
274
+ if self.size not in (4, 8):
275
+ raise ValueError(f"What do I do with a float {self.size} bytes long")
276
+ value = claripy.FPV(value, claripy.FSORT_FLOAT if self.size == 4 else claripy.FSORT_DOUBLE)
277
+ return value.raw_to_bv() # type:ignore
278
+
279
+ def check_value_get(self, value):
280
+ if self.is_fp:
281
+ return value.raw_to_fp()
282
+ return value
283
+
284
+ def set_value(self, state, value, **kwargs):
285
+ raise NotImplementedError
286
+
287
+ def get_value(self, state, **kwargs):
288
+ raise NotImplementedError
289
+
290
+ def refine(self, size, arch=None, offset=None, is_fp=None):
291
+ raise NotImplementedError
292
+
293
+ def get_footprint(self) -> Iterable[SimRegArg | SimStackArg]:
294
+ """
295
+ Return a list of SimRegArg and SimStackArgs that are the base components used for this location
296
+ """
297
+ raise NotImplementedError
298
+
299
+
300
+ class SimRegArg(SimFunctionArgument):
301
+ """
302
+ Represents a function argument that has been passed in a register.
303
+
304
+ :ivar string reg_name: The name of the represented register.
305
+ :ivar int size: The size of the data to store, in number of bytes.
306
+ :ivar reg_offset: The offset into the register to start storing data.
307
+ :ivar clear_entire_reg: Whether a store to this register should zero the unused parts of the register.
308
+ :ivar bool is_fp: Whether loads from this location should return a floating point bitvector
309
+ """
310
+
311
+ def __init__(self, reg_name: RegisterName, size: int, reg_offset=0, is_fp=False, clear_entire_reg=False):
312
+ super().__init__(size, is_fp)
313
+ self.reg_name = reg_name
314
+ self.reg_offset = reg_offset
315
+ self.clear_entire_reg = clear_entire_reg
316
+
317
+ def get_footprint(self):
318
+ return {self}
319
+
320
+ def __repr__(self):
321
+ return f"<{self.reg_name}>"
322
+
323
+ def __eq__(self, other):
324
+ return (
325
+ type(other) is SimRegArg
326
+ and self.reg_name == other.reg_name
327
+ and self.reg_offset == other.reg_offset
328
+ and self.size == other.size
329
+ )
330
+
331
+ def __hash__(self):
332
+ return hash((self.size, self.reg_name, self.reg_offset))
333
+
334
+ def check_offset(self, arch) -> int:
335
+ return arch.registers[self.reg_name][0] + self.reg_offset
336
+
337
+ def set_value(self, state, value, **kwargs): # pylint: disable=unused-argument,arguments-differ
338
+ value = self.check_value_set(value, state.arch)
339
+ offset = self.check_offset(state.arch)
340
+ if self.clear_entire_reg:
341
+ state.registers.store(self.reg_name, 0)
342
+ state.registers.store(offset, value, size=self.size)
343
+
344
+ def get_value(self, state, **kwargs): # pylint: disable=unused-argument,arguments-differ
345
+ offset = self.check_offset(state.arch)
346
+ return self.check_value_get(state.registers.load(offset, size=self.size))
347
+
348
+ def refine(self, size, arch=None, offset=None, is_fp=None):
349
+ passed_offset_none = offset is None
350
+ if offset is None:
351
+ if arch is None:
352
+ raise ValueError("Need to specify either offset or arch in order to refine a register argument")
353
+ offset = 0 if arch.register_endness == "Iend_LE" else self.size - size
354
+ if is_fp is None:
355
+ is_fp = self.is_fp
356
+ return SimRegArg(self.reg_name, size, self.reg_offset + offset, is_fp, clear_entire_reg=passed_offset_none)
357
+
358
+ def sse_extend(self):
359
+ return SimRegArg(self.reg_name, self.size, self.reg_offset + self.size, is_fp=self.is_fp)
360
+
361
+
362
+ class SimStackArg(SimFunctionArgument):
363
+ """
364
+ Represents a function argument that has been passed on the stack.
365
+
366
+ :var int stack_offset: The position of the argument relative to the stack pointer after the function prelude.
367
+ :ivar int size: The size of the argument, in number of bytes.
368
+ :ivar bool is_fp: Whether loads from this location should return a floating point bitvector
369
+ """
370
+
371
+ def __init__(self, stack_offset: int, size: int, is_fp: bool = False):
372
+ SimFunctionArgument.__init__(self, size, is_fp)
373
+ self.stack_offset: int = stack_offset
374
+
375
+ def get_footprint(self):
376
+ return {self}
377
+
378
+ def __repr__(self):
379
+ return f"[{self.stack_offset:#x}]"
380
+
381
+ def __eq__(self, other):
382
+ return type(other) is SimStackArg and self.stack_offset == other.stack_offset
383
+
384
+ def __hash__(self):
385
+ return hash((self.size, self.stack_offset))
386
+
387
+ def set_value(self, state, value, stack_base=None, **kwargs): # pylint: disable=arguments-differ
388
+ value = self.check_value_set(value, state.arch)
389
+ if stack_base is None:
390
+ stack_base = state.regs.sp
391
+ state.memory.store(stack_base + self.stack_offset, value, endness=state.arch.memory_endness)
392
+
393
+ def get_value(self, state, stack_base=None, **kwargs): # pylint: disable=arguments-differ
394
+ if stack_base is None:
395
+ stack_base = state.regs.sp
396
+ value = state.memory.load(stack_base + self.stack_offset, endness=state.arch.memory_endness, size=self.size)
397
+ return self.check_value_get(value)
398
+
399
+ def refine(self, size, arch=None, offset=None, is_fp=None):
400
+ if offset is None:
401
+ if arch is None:
402
+ raise ValueError("Need to specify either offset or arch in order to refine a stack argument")
403
+ offset = 0 if arch.register_endness == "Iend_LE" else self.size - size
404
+ if is_fp is None:
405
+ is_fp = self.is_fp
406
+ return SimStackArg(self.stack_offset + offset, size, is_fp)
407
+
408
+
409
+ class SimComboArg(SimFunctionArgument, Generic[T]):
410
+ """
411
+ An argument which spans multiple storage locations. Locations should be given least-significant first.
412
+ """
413
+
414
+ def __init__(self, locations: list[T], is_fp=False):
415
+ super().__init__(sum(x.size for x in locations), is_fp=is_fp)
416
+ self.locations = locations
417
+
418
+ def get_footprint(self):
419
+ return {y for x in self.locations for y in x.get_footprint()}
420
+
421
+ def __repr__(self):
422
+ return f"SimComboArg({self.locations!r})"
423
+
424
+ def __eq__(self, other):
425
+ return type(other) is SimComboArg and all(a == b for a, b in zip(self.locations, other.locations))
426
+
427
+ def set_value(self, state, value, **kwargs): # pylint:disable=arguments-differ
428
+ value = self.check_value_set(value, state.arch)
429
+ cur = 0
430
+ for loc in self.locations:
431
+ size_bits = loc.size * state.arch.byte_width
432
+ loc.set_value(state, value[cur + size_bits - 1 : cur], **kwargs)
433
+ cur += size_bits
434
+
435
+ def get_value(self, state, **kwargs): # pylint:disable=arguments-differ
436
+ vals = []
437
+ for loc in reversed(self.locations):
438
+ vals.append(loc.get_value(state, **kwargs))
439
+ return self.check_value_get(claripy.Concat(*vals))
440
+
441
+
442
+ class SimStructArg(SimFunctionArgument):
443
+ """
444
+ An argument which de/serializes a struct from a list of storage locations
445
+
446
+ :ivar struct: The simtype describing the structure
447
+ :ivar locs: The storage locations to use
448
+ """
449
+
450
+ def __init__(self, struct: SimStruct, locs: dict[str, SimFunctionArgument]):
451
+ super().__init__(sum(loc.size for loc in locs.values()))
452
+ self.struct = struct
453
+ self.locs = locs
454
+
455
+ def get_footprint(self):
456
+ regs: defaultdict[str, set[SimRegArg]] = defaultdict(set)
457
+ others: set[SimRegArg | SimStackArg] = set()
458
+ for loc in self.locs.values():
459
+ for footloc in loc.get_footprint():
460
+ if isinstance(footloc, SimRegArg):
461
+ regs[footloc.reg_name].add(footloc)
462
+ else:
463
+ others.add(footloc)
464
+
465
+ for reg, locset in regs.items():
466
+ min_offset = min(loc.reg_offset for loc in locset)
467
+ max_offset = max(loc.reg_offset + loc.size for loc in locset)
468
+ others.add(SimRegArg(reg, max_offset - min_offset, min_offset))
469
+
470
+ return others
471
+
472
+ def get_single_footprint(self) -> SimStackArg | SimRegArg | SimComboArg:
473
+ if self.struct._arch is None:
474
+ raise TypeError("Can't tell the size of a struct without an arch")
475
+ stack_min = None
476
+ stack_max = None
477
+ regs = []
478
+ for field in self.struct.fields:
479
+ loc = self.locs[field]
480
+ if isinstance(loc, SimStackArg):
481
+ if stack_min is None or stack_max is None:
482
+ stack_min = loc.stack_offset
483
+ stack_max = loc.stack_offset
484
+ else:
485
+ # sanity check that arguments are laid out in order...
486
+ assert loc.stack_offset >= stack_max
487
+ stack_max = loc.stack_offset + loc.size
488
+ elif isinstance(loc, SimRegArg):
489
+ regs.append(loc)
490
+ else:
491
+ assert False, "Why would a struct have layout elements other than stack and reg?"
492
+
493
+ # things to consider...
494
+ # what happens if we return the concat of two registers but there's slack space missing?
495
+ # an example of this would be big-endian struct { long a; int b; }
496
+ # do any CCs do this??
497
+ # for now assume no
498
+
499
+ if stack_min is not None:
500
+ if regs:
501
+ assert (
502
+ False
503
+ ), "Unknown CC argument passing structure - why are we passing both regs and stack at the same time?"
504
+ return SimStackArg(stack_min, self.struct.size // self.struct._arch.byte_width)
505
+ if not regs:
506
+ assert False, "huh??????"
507
+ if len(regs) == 1:
508
+ return regs[0]
509
+ return SimComboArg(regs)
510
+
511
+ def get_value(self, state, **kwargs):
512
+ return SimStructValue(
513
+ self.struct, {field: getter.get_value(state, **kwargs) for field, getter in self.locs.items()}
514
+ )
515
+
516
+ def set_value(self, state, value, **kwargs):
517
+ for field, setter in self.locs.items():
518
+ setter.set_value(state, value[field], **kwargs)
519
+
520
+
521
+ class SimArrayArg(SimFunctionArgument):
522
+ def __init__(self, locs):
523
+ super().__init__(sum(loc.size for loc in locs))
524
+ self.locs = locs
525
+
526
+ def get_footprint(self):
527
+ return {y for x in self.locs for y in x.get_footprint()}
528
+
529
+ def get_value(self, state, **kwargs):
530
+ return [getter.get_value(state, **kwargs) for getter in self.locs]
531
+
532
+ def set_value(self, state, value, **kwargs):
533
+ if len(value) != len(self.locs):
534
+ raise TypeError(f"Expected {len(self.locs)} elements, got {len(value)}")
535
+ for subvalue, setter in zip(value, self.locs):
536
+ setter.set_value(state, subvalue, **kwargs)
537
+
538
+
539
+ class SimReferenceArgument(SimFunctionArgument):
540
+ """
541
+ A function argument which is passed by reference.
542
+
543
+ :ivar ptr_loc: The location the reference's pointer is stored
544
+ :ivar main_loc: A SimStackArgument describing how to load the argument's value as if it were stored at offset
545
+ zero on the stack. It will be passed ``stack_base=ptr_loc.get_value(state)``
546
+ """
547
+
548
+ def __init__(self, ptr_loc: SimFunctionArgument, main_loc: SimFunctionArgument):
549
+ super().__init__(ptr_loc.size) # ???
550
+ self.ptr_loc = ptr_loc
551
+ self.main_loc = main_loc
552
+
553
+ def get_footprint(self):
554
+ return self.main_loc.get_footprint()
555
+
556
+ def get_value(self, state, **kwargs):
557
+ ptr_val = self.ptr_loc.get_value(state, **kwargs)
558
+ return self.main_loc.get_value(state, stack_base=ptr_val, **kwargs)
559
+
560
+ def set_value(self, state, value, **kwargs):
561
+ ptr_val = self.ptr_loc.get_value(state, **kwargs)
562
+ self.main_loc.set_value(state, value, stack_base=ptr_val, **kwargs)
563
+
564
+
565
+ class ArgSession:
566
+ """
567
+ A class to keep track of the state accumulated in laying parameters out into memory
568
+ """
569
+
570
+ __slots__ = (
571
+ "both_iter",
572
+ "cc",
573
+ "fp_iter",
574
+ "int_iter",
575
+ )
576
+
577
+ def __init__(self, cc):
578
+ self.cc = cc
579
+ self.fp_iter = cc.fp_args
580
+ self.int_iter = cc.int_args
581
+ self.both_iter = cc.memory_args
582
+
583
+ def getstate(self):
584
+ return (self.fp_iter.getstate(), self.int_iter.getstate(), self.both_iter.getstate())
585
+
586
+ def setstate(self, state):
587
+ fp, int_, both = state
588
+ self.fp_iter.setstate(fp)
589
+ self.int_iter.setstate(int_)
590
+ self.both_iter.setstate(both)
591
+
592
+
593
+ class UsercallArgSession:
594
+ """
595
+ An argsession for use with SimCCUsercall
596
+ """
597
+
598
+ __slots__ = (
599
+ "cc",
600
+ "real_args",
601
+ )
602
+
603
+ def __init__(self, cc):
604
+ self.cc = cc
605
+ self.real_args = SerializableListIterator(self.cc.arg_locs)
606
+
607
+ def getstate(self):
608
+ return self.real_args.getstate()
609
+
610
+ def setstate(self, state):
611
+ self.real_args.setstate(state)
612
+
613
+
614
+ class SimCC:
615
+ """
616
+ A calling convention allows you to extract from a state the data passed from function to
617
+ function by calls and returns. Most of the methods provided by SimCC that operate on a state
618
+ assume that the program is just after a call but just before stack frame allocation, though
619
+ this may be overridden with the `stack_base` parameter to each individual method.
620
+
621
+ This is the base class for all calling conventions.
622
+ """
623
+
624
+ def __init__(self, arch: archinfo.Arch):
625
+ """
626
+ :param arch: The Archinfo arch for this CC
627
+ """
628
+ self.arch = arch
629
+
630
+ #
631
+ # Here are all the things a subclass needs to specify!
632
+ #
633
+
634
+ ARG_REGS: list[str] = [] # A list of all the registers used for integral args, in order (names or offsets)
635
+ FP_ARG_REGS: list[str] = [] # A list of all the registers used for floating point args, in order
636
+ STACKARG_SP_BUFF = 0 # The amount of stack space reserved between the saved return address
637
+ # (if applicable) and the arguments. Probably zero.
638
+ STACKARG_SP_DIFF = 0 # The amount of stack space reserved for the return address
639
+ CALLER_SAVED_REGS: list[str] = [] # Caller-saved registers
640
+ RETURN_ADDR: SimFunctionArgument | None = (
641
+ None # The location where the return address is stored, as a SimFunctionArgument
642
+ )
643
+ RETURN_VAL: SimFunctionArgument | None = (
644
+ None # The location where the return value is stored, as a SimFunctionArgument
645
+ )
646
+ OVERFLOW_RETURN_VAL: SimFunctionArgument | None = (
647
+ None # The second half of the location where a double-length return value is stored
648
+ )
649
+ FP_RETURN_VAL: SimFunctionArgument | None = (
650
+ None # The location where floating-point argument return values are stored
651
+ )
652
+ ARCH: type[archinfo.Arch] | None = (
653
+ None # The archinfo.Arch class for which this CC is most likely relevant, if related
654
+ )
655
+ # archinfo.Arch classes for which this CC is relevant, in addition to self.ARCH.
656
+ # you should access cls.arches() to get a list of all arches for which this CC is relevant
657
+ EXTRA_ARCHES: tuple[type[archinfo.Arch], ...] = ()
658
+ CALLEE_CLEANUP = False # Whether the callee has to deallocate the stack space for the arguments
659
+
660
+ STACK_ALIGNMENT = 1 # the alignment requirement of the stack pointer at function start BEFORE call
661
+
662
+ #
663
+ # Here are several things you MAY want to override to change your cc's convention
664
+ #
665
+
666
+ @property
667
+ def int_args(self):
668
+ """
669
+ Iterate through all the possible arg positions that can only be used to store integer or pointer values.
670
+
671
+ Returns an iterator of SimFunctionArguments
672
+ """
673
+ if self.ARG_REGS is None:
674
+ raise NotImplementedError
675
+ return SerializableListIterator([SimRegArg(reg, self.arch.bytes) for reg in self.ARG_REGS])
676
+
677
+ @property
678
+ def memory_args(self):
679
+ """
680
+ Iterate through all the possible arg positions that can be used to store any kind of argument.
681
+
682
+ Returns an iterator of SimFunctionArguments
683
+ """
684
+ start = self.STACKARG_SP_BUFF + self.STACKARG_SP_DIFF
685
+ return SerializableCounter(start, self.arch.bytes, lambda offset: SimStackArg(offset, self.arch.bytes))
686
+
687
+ @property
688
+ def fp_args(self):
689
+ """
690
+ Iterate through all the possible arg positions that can only be used to store floating point values.
691
+
692
+ Returns an iterator of SimFunctionArguments
693
+ """
694
+ if self.FP_ARG_REGS is None:
695
+ raise NotImplementedError
696
+ return SerializableListIterator([SimRegArg(reg, self.arch.bytes) for reg in self.FP_ARG_REGS])
697
+
698
+ def is_fp_arg(self, arg):
699
+ """
700
+ This should take a SimFunctionArgument instance and return whether or not that argument is a floating-point
701
+ argument.
702
+
703
+ Returns True for MUST be a floating point arg,
704
+ False for MUST NOT be a floating point arg,
705
+ None for when it can be either.
706
+ """
707
+ if arg in self.int_args:
708
+ return False
709
+ if arg in self.fp_args or arg == self.FP_RETURN_VAL:
710
+ return True
711
+ return None
712
+
713
+ ArgSession = ArgSession # import this from global scope so SimCC subclasses can subclass it if they like
714
+
715
+ def arg_session(self, ret_ty: SimType | None):
716
+ """
717
+ Return an arg session.
718
+
719
+ A session provides the control interface necessary to describe how integral and floating-point arguments are
720
+ laid out into memory. The default behavior is that there are a finite list of int-only and fp-only argument
721
+ slots, and an infinite number of generic slots, and when an argument of a given type is requested, the most
722
+ slot available is used. If you need different behavior, subclass ArgSession.
723
+
724
+ You need to provide the return type of the function in order to kick off an arg layout session.
725
+ """
726
+ session = self.ArgSession(self)
727
+ if self.return_in_implicit_outparam(ret_ty):
728
+ self.next_arg(session, SimTypePointer(SimTypeBottom()).with_arch(self.arch))
729
+ return session
730
+
731
+ def return_in_implicit_outparam(self, ty) -> bool: # pylint:disable=unused-argument
732
+ return False
733
+
734
+ def stack_space(self, args):
735
+ """
736
+ :param args: A list of SimFunctionArguments
737
+
738
+ :returns: The number of bytes that should be allocated on the stack to store all these args,
739
+ NOT INCLUDING the return address.
740
+ """
741
+ out = self.STACKARG_SP_DIFF
742
+ for arg in args:
743
+ if isinstance(arg, SimStackArg):
744
+ out = max(out, arg.stack_offset + self.arch.bytes)
745
+
746
+ out += self.STACKARG_SP_BUFF
747
+ return out
748
+
749
+ def return_val(self, ty, perspective_returned=False):
750
+ """
751
+ The location the return value is stored, based on its type.
752
+ """
753
+ if ty._arch is None:
754
+ ty = ty.with_arch(self.arch)
755
+ if isinstance(ty, (SimStruct, SimUnion, SimTypeFixedSizeArray)):
756
+ raise AngrTypeError(
757
+ f"{self} doesn't know how to return aggregate types ({type(ty)}). Consider overriding return_val to "
758
+ "implement its ABI logic"
759
+ )
760
+ if self.return_in_implicit_outparam(ty):
761
+ if perspective_returned:
762
+ assert self.RETURN_VAL is not None
763
+ ptr_loc = self.RETURN_VAL
764
+ else:
765
+ ptr_loc = self.next_arg(self.ArgSession(self), SimTypePointer(SimTypeBottom()).with_arch(self.arch))
766
+ return SimReferenceArgument(
767
+ ptr_loc, SimStackArg(0, ty.size // self.arch.byte_width, is_fp=isinstance(ty, SimTypeFloat))
768
+ )
769
+
770
+ if isinstance(ty, SimTypeFloat) and self.FP_RETURN_VAL is not None:
771
+ return self.FP_RETURN_VAL.refine(size=ty.size // self.arch.byte_width, arch=self.arch, is_fp=True)
772
+
773
+ if self.RETURN_VAL is None or isinstance(ty, SimTypeBottom):
774
+ return None
775
+ if ty.size > self.RETURN_VAL.size * self.arch.byte_width:
776
+ assert self.OVERFLOW_RETURN_VAL is not None
777
+ return SimComboArg([self.RETURN_VAL, self.OVERFLOW_RETURN_VAL])
778
+ return self.RETURN_VAL.refine(size=ty.size // self.arch.byte_width, arch=self.arch, is_fp=False)
779
+
780
+ @property
781
+ def return_addr(self):
782
+ """
783
+ The location the return address is stored.
784
+ """
785
+ return self.RETURN_ADDR
786
+
787
+ def next_arg(self, session: ArgSession, arg_type: SimType):
788
+ if isinstance(arg_type, (SimTypeArray, SimTypeFixedSizeArray)): # hack
789
+ arg_type = SimTypePointer(arg_type.elem_type).with_arch(self.arch)
790
+ if isinstance(arg_type, (SimStruct, SimUnion, SimTypeFixedSizeArray)):
791
+ raise TypeError(
792
+ f"{self} doesn't know how to store aggregate type {type(arg_type)}. Consider overriding next_arg to "
793
+ "implement its ABI logic"
794
+ )
795
+ if isinstance(arg_type, SimTypeBottom):
796
+ # This is usually caused by failures or mistakes during type inference
797
+ l.warning("Function argument type cannot be BOT. Treating it as a 32-bit int.")
798
+ arg_type = SimTypeInt().with_arch(self.arch)
799
+ is_fp = isinstance(arg_type, SimTypeFloat)
800
+ assert arg_type.size is not None
801
+ size = arg_type.size // self.arch.byte_width
802
+ try:
803
+ arg = next(session.fp_iter) if is_fp else next(session.int_iter)
804
+ except StopIteration:
805
+ try:
806
+ arg = next(session.both_iter)
807
+ except StopIteration as err:
808
+ raise TypeError("Accessed too many arguments - exhausted all positions?") from err
809
+
810
+ if size > arg.size:
811
+ if isinstance(arg, SimStackArg):
812
+ arg_size = arg.size
813
+ locations = [arg]
814
+ while arg_size < size:
815
+ next_arg = next(session.both_iter)
816
+ arg_size += next_arg.size
817
+ locations.append(next_arg)
818
+ return SimComboArg(locations, is_fp=is_fp)
819
+ raise ValueError(
820
+ f"{self} doesn't know how to store large types. Consider overriding"
821
+ " next_arg to implement its ABI logic"
822
+ )
823
+ return arg.refine(size, is_fp=is_fp, arch=self.arch)
824
+
825
+ #
826
+ # Useful functions!
827
+ #
828
+
829
+ @staticmethod
830
+ def is_fp_value(val):
831
+ return (
832
+ isinstance(val, (float, claripy.ast.FP))
833
+ or (isinstance(val, claripy.ast.Base) and val.op.startswith("fp")) # type: ignore
834
+ or (
835
+ isinstance(val, claripy.ast.Base)
836
+ and val.op == "Reverse" # type:ignore
837
+ and val.args[0].op.startswith("fp") # type:ignore
838
+ )
839
+ )
840
+
841
+ @staticmethod
842
+ def guess_prototype(args, prototype=None):
843
+ """
844
+ Come up with a plausible SimTypeFunction for the given args (as would be passed to e.g. setup_callsite).
845
+
846
+ You can pass a variadic function prototype in the `base_type` parameter and all its arguments will be used,
847
+ only guessing types for the variadic arguments.
848
+ """
849
+ if type(prototype) is str:
850
+ prototype = parse_signature(prototype)
851
+ elif prototype is None:
852
+ l.warning("Guessing call prototype. Please specify prototype.")
853
+
854
+ charp = SimTypePointer(SimTypeChar())
855
+ result = prototype if prototype is not None else SimTypeFunction([], charp)
856
+ for arg in args[len(result.args) :]:
857
+ if type(arg) in (int, bytes, PointerWrapper):
858
+ result.args += (charp,)
859
+ elif type(arg) is float:
860
+ result.args += (SimTypeDouble(),)
861
+ elif isinstance(arg, claripy.ast.BV):
862
+ result.args += (SimTypeNum(len(arg), False),)
863
+ elif isinstance(arg, claripy.ast.FP):
864
+ if arg.sort == claripy.FSORT_FLOAT:
865
+ result.args += (SimTypeFloat(),)
866
+ elif arg.sort == claripy.FSORT_DOUBLE:
867
+ result.args += (SimTypeDouble(),)
868
+ else:
869
+ raise TypeError("WHAT kind of floating point is this")
870
+ else:
871
+ raise TypeError(f"Cannot guess FFI type for {type(arg)}")
872
+
873
+ return result
874
+
875
+ def arg_locs(self, prototype) -> list[SimFunctionArgument]:
876
+ if prototype._arch is None:
877
+ prototype = prototype.with_arch(self.arch)
878
+ session = self.arg_session(prototype.returnty)
879
+ return [self.next_arg(session, arg_ty) for arg_ty in prototype.args]
880
+
881
+ def get_args(self, state, prototype, stack_base=None):
882
+ arg_locs = self.arg_locs(prototype)
883
+ return [loc.get_value(state, stack_base=stack_base) for loc in arg_locs]
884
+
885
+ def set_return_val(self, state, val, ty, stack_base=None, perspective_returned=False):
886
+ loc = self.return_val(ty, perspective_returned=perspective_returned)
887
+ if loc is None:
888
+ raise ValueError("Cannot set return value - there is no return value location")
889
+ loc.set_value(state, val, stack_base=stack_base)
890
+
891
+ def setup_callsite(self, state, ret_addr, args, prototype, stack_base=None, alloc_base=None, grow_like_stack=True):
892
+ """
893
+ This function performs the actions of the caller getting ready to jump into a function.
894
+
895
+ :param state: The SimState to operate on
896
+ :param ret_addr: The address to return to when the called function finishes
897
+ :param args: The list of arguments that that the called function will see
898
+ :param prototype: The signature of the call you're making. Should include variadic args concretely.
899
+ :param stack_base: An optional pointer to use as the top of the stack, circa the function entry point
900
+ :param alloc_base: An optional pointer to use as the place to put excess argument data
901
+ :param grow_like_stack: When allocating data at alloc_base, whether to allocate at decreasing addresses
902
+
903
+ The idea here is that you can provide almost any kind of python type in `args` and it'll be translated to a
904
+ binary format to be placed into simulated memory. Lists (representing arrays) must be entirely elements of the
905
+ same type and size, while tuples (representing structs) can be elements of any type and size.
906
+ If you'd like there to be a pointer to a given value, wrap the value in a `PointerWrapper`.
907
+
908
+ If stack_base is not provided, the current stack pointer will be used, and it will be updated.
909
+ If alloc_base is not provided, the stack base will be used and grow_like_stack will implicitly be True.
910
+
911
+ grow_like_stack controls the behavior of allocating data at alloc_base. When data from args needs to be wrapped
912
+ in a pointer, the pointer needs to point somewhere, so that data is dumped into memory at alloc_base. If you
913
+ set alloc_base to point to somewhere other than the stack, set grow_like_stack to False so that sequential
914
+ allocations happen at increasing addresses.
915
+ """
916
+
917
+ # STEP 0: clerical work
918
+
919
+ allocator = AllocHelper(self.arch.bits)
920
+ if type(prototype) is str:
921
+ prototype = parse_signature(prototype, arch=self.arch)
922
+ elif prototype._arch is None:
923
+ prototype = prototype.with_arch(self.arch)
924
+
925
+ #
926
+ # STEP 1: convert all values into serialized form
927
+ # this entails creating the vals list of simple values to store and also populating the allocator's
928
+ # understanding of what aux data needs to be stored
929
+ # This is also where we compute arg locations (arg_locs)
930
+ #
931
+
932
+ vals = [self._standardize_value(arg, ty, state, allocator.dump) for arg, ty in zip(args, prototype.args)]
933
+ arg_locs = self.arg_locs(prototype)
934
+
935
+ # step 1.5, gotta handle the SimReferenceArguments correctly
936
+ for i, (loc, val) in enumerate(zip(arg_locs, vals)):
937
+ if not isinstance(loc, SimReferenceArgument):
938
+ continue
939
+ dumped = allocator.dump(val, state, loc=val.main_loc)
940
+ vals[i] = dumped
941
+ arg_locs[i] = val.ptr_loc
942
+
943
+ # step 1.75 allocate implicit outparam stuff
944
+ if self.return_in_implicit_outparam(prototype.returnty):
945
+ loc = self.return_val(prototype.returnty)
946
+ assert isinstance(loc, SimReferenceArgument)
947
+ # hack: because the allocator gives us a pointer that needs to be translated, we need to shove it into
948
+ # the args list so it'll be translated and stored once everything is laid out
949
+ vals.append(allocator.alloc(loc.main_loc.size))
950
+ arg_locs.append(loc.ptr_loc)
951
+
952
+ #
953
+ # STEP 2: decide on memory storage locations
954
+ # implement the contract for stack_base/alloc_base/grow_like_stack
955
+ # after this, stack_base should be the final stack pointer, alloc_base should be the final aux storage location,
956
+ # and the stack pointer should be updated
957
+ #
958
+
959
+ if stack_base is None:
960
+ if alloc_base is None:
961
+ alloc_size = allocator.size()
962
+ state.regs.sp -= alloc_size
963
+ alloc_base = state.regs.sp
964
+ grow_like_stack = False
965
+
966
+ state.regs.sp -= self.stack_space(arg_locs)
967
+
968
+ # handle alignment
969
+ alignment = (state.regs.sp + self.STACKARG_SP_DIFF) % self.STACK_ALIGNMENT
970
+ state.regs.sp -= alignment
971
+
972
+ else:
973
+ state.regs.sp = stack_base
974
+
975
+ if alloc_base is None:
976
+ alloc_base = stack_base + self.stack_space(arg_locs)
977
+ grow_like_stack = False
978
+
979
+ if grow_like_stack:
980
+ alloc_base -= allocator.size()
981
+ if type(alloc_base) is int:
982
+ alloc_base = claripy.BVV(alloc_base, state.arch.bits)
983
+
984
+ for i, val in enumerate(vals):
985
+ vals[i] = allocator.translate(val, alloc_base)
986
+
987
+ #
988
+ # STEP 3: store everything!
989
+ #
990
+
991
+ allocator.apply(state, alloc_base)
992
+
993
+ for loc, val in zip(arg_locs, vals):
994
+ assert loc is not None
995
+ loc.set_value(state, val, stack_base=stack_base)
996
+ if self.return_addr is not None:
997
+ self.return_addr.set_value(state, ret_addr, stack_base=stack_base)
998
+
999
+ def teardown_callsite(self, state, return_val=None, prototype=None, force_callee_cleanup=False):
1000
+ """
1001
+ This function performs the actions of the callee as it's getting ready to return.
1002
+ It returns the address to return to.
1003
+
1004
+ :param state: The state to mutate
1005
+ :param return_val: The value to return
1006
+ :param prototype: The prototype of the given function
1007
+ :param force_callee_cleanup: If we should clean up the stack allocation for the arguments even if it's not
1008
+ the callee's job to do so
1009
+
1010
+ TODO: support the stack_base parameter from setup_callsite...? Does that make sense in this context?
1011
+ Maybe it could make sense by saying that you pass it in as something like the "saved base pointer" value?
1012
+ """
1013
+ if return_val is not None and prototype is not None and not isinstance(prototype.returnty, SimTypeBottom):
1014
+ self.set_return_val(state, return_val, prototype.returnty)
1015
+ # ummmmmmmm hack
1016
+ loc = self.return_val(prototype.returnty)
1017
+ if self.RETURN_VAL is not None and isinstance(loc, SimReferenceArgument):
1018
+ self.RETURN_VAL.set_value(state, loc.ptr_loc.get_value(state))
1019
+
1020
+ ret_addr = self.return_addr.get_value(state) if self.return_addr is not None else None
1021
+
1022
+ if state.arch.sp_offset is not None and prototype is not None:
1023
+ if force_callee_cleanup or self.CALLEE_CLEANUP:
1024
+ session = self.arg_session(prototype.returnty)
1025
+ if self.return_in_implicit_outparam(prototype.returnty):
1026
+ extra = [cast(SimReferenceArgument, self.return_val(prototype.returnty)).ptr_loc]
1027
+ else:
1028
+ extra = []
1029
+ state.regs.sp += self.stack_space(extra + [self.next_arg(session, x) for x in prototype.args])
1030
+ else:
1031
+ state.regs.sp += self.STACKARG_SP_DIFF
1032
+
1033
+ return ret_addr
1034
+
1035
+ #
1036
+ # Helper functions
1037
+ #
1038
+
1039
+ @staticmethod
1040
+ def _standardize_value(arg, ty, state, alloc):
1041
+ if isinstance(arg, SimActionObject):
1042
+ return SimCC._standardize_value(arg.ast, ty, state, alloc)
1043
+ if isinstance(arg, PointerWrapper):
1044
+ if not isinstance(ty, (SimTypePointer, SimTypeReference)):
1045
+ raise TypeError(f"Type mismatch: expected {ty}, got pointer-wrapper")
1046
+
1047
+ if arg.buffer:
1048
+ if isinstance(arg.value, claripy.ast.Bits):
1049
+ real_value = arg.value.chop(state.arch.byte_width) # type:ignore
1050
+ elif type(arg.value) in (bytes, str):
1051
+ real_value = claripy.BVV(arg.value).chop(8)
1052
+ else:
1053
+ raise TypeError("PointerWrapper(buffer=True) can only be used with a bitvector or a bytestring")
1054
+ else:
1055
+ sub = ty.pts_to if isinstance(ty, SimTypePointer) else ty.refs
1056
+ child_type = SimTypeArray(sub) if isinstance(arg.value, (str, bytes, list)) else sub
1057
+ try:
1058
+ real_value = SimCC._standardize_value(arg.value, child_type, state, alloc)
1059
+ except TypeError as e: # this is a dangerous catch...
1060
+ raise TypeError(
1061
+ f"Failed to store pointer-wrapped data ({e.args[0]}). "
1062
+ "Do you want a PointerWrapper(buffer=True)?"
1063
+ ) from None
1064
+ return alloc(real_value, state)
1065
+
1066
+ if isinstance(arg, (str, bytes)):
1067
+ # sanitize the argument and request standardization again with SimTypeArray
1068
+ if isinstance(arg, str):
1069
+ arg = arg.encode()
1070
+ arg += b"\0"
1071
+ if isinstance(ty, SimTypePointer) and isinstance(ty.pts_to, SimTypeChar):
1072
+ pass
1073
+ elif (isinstance(ty, SimTypeFixedSizeArray) and isinstance(ty.elem_type, SimTypeChar)) or (
1074
+ isinstance(ty, SimTypeArray) and isinstance(ty.elem_type, SimTypeChar)
1075
+ ):
1076
+ if ty.length is not None:
1077
+ if len(arg) > ty.length:
1078
+ raise TypeError(f"String {arg!r} is too long for {ty}")
1079
+ arg = arg.ljust(ty.length, b"\0")
1080
+ elif isinstance(ty, SimTypeString):
1081
+ if ty.length is not None:
1082
+ if len(arg) > ty.length + 1:
1083
+ raise TypeError(f"String {arg!r} is too long for {ty}")
1084
+ arg = arg.ljust(ty.length + 1, b"\0")
1085
+ else:
1086
+ raise TypeError(f"Type mismatch: Expected {ty}, got char*")
1087
+ return SimCC._standardize_value(list(arg), SimTypeArray(SimTypeChar(), len(arg)), state, alloc)
1088
+
1089
+ if isinstance(arg, list):
1090
+ if isinstance(ty, SimTypePointer):
1091
+ ref = True
1092
+ subty = ty.pts_to
1093
+ elif isinstance(ty, SimTypeReference):
1094
+ ref = True
1095
+ subty = ty.refs
1096
+ elif isinstance(ty, SimTypeArray):
1097
+ ref = True
1098
+ subty = ty.elem_type
1099
+ if ty.length is not None and len(arg) != ty.length:
1100
+ raise TypeError(f"Array {arg!r} is the wrong length for {ty}")
1101
+ else:
1102
+ raise TypeError(f"Type mismatch: Expected {ty}, got char*")
1103
+
1104
+ val = [SimCC._standardize_value(sarg, subty, state, alloc) for sarg in arg]
1105
+ if ref:
1106
+ val = alloc(val, state)
1107
+ return val
1108
+
1109
+ if isinstance(arg, (tuple, dict, SimStructValue)):
1110
+ if not isinstance(ty, SimStruct):
1111
+ raise TypeError(f"Type mismatch: Expected {ty}, got {type(arg)} (i.e. struct)")
1112
+ if not isinstance(arg, SimStructValue):
1113
+ if len(arg) != len(ty.fields):
1114
+ raise TypeError(f"Wrong number of fields in struct, expected {len(ty.fields)} got {len(arg)}")
1115
+ arg = SimStructValue(ty, arg)
1116
+ return SimStructValue(
1117
+ ty, [SimCC._standardize_value(arg[field], ty.fields[field], state, alloc) for field in ty.fields]
1118
+ )
1119
+
1120
+ if isinstance(arg, int):
1121
+ if isinstance(ty, SimTypeFloat):
1122
+ return SimCC._standardize_value(float(arg), ty, state, alloc)
1123
+
1124
+ return claripy.BVV(arg, ty.size)
1125
+
1126
+ if isinstance(arg, float):
1127
+ if isinstance(ty, SimTypeDouble):
1128
+ sort = claripy.FSORT_DOUBLE
1129
+ elif isinstance(ty, SimTypeFloat):
1130
+ sort = claripy.FSORT_FLOAT
1131
+ else:
1132
+ raise TypeError(f"Type mismatch: expected {ty}, got float")
1133
+
1134
+ return claripy.FPV(arg, sort)
1135
+
1136
+ if isinstance(arg, claripy.ast.FP):
1137
+ if isinstance(ty, SimTypeFloat):
1138
+ if len(arg) != ty.size:
1139
+ raise TypeError(f"Type mismatch: expected {ty}, got {arg.sort}")
1140
+ return arg
1141
+ if isinstance(ty, (SimTypeReg, SimTypeNum)):
1142
+ return arg.val_to_bv(ty.size, ty.signed if isinstance(ty, SimTypeNum) else False)
1143
+ raise TypeError(f"Type mismatch: expected {ty}, got {arg.sort}")
1144
+
1145
+ if isinstance(arg, claripy.ast.BV):
1146
+ if isinstance(ty, (SimTypeReg, SimTypeNum)):
1147
+ if len(arg) != ty.size:
1148
+ if arg.concrete:
1149
+ size = ty.size
1150
+ assert size is not None
1151
+ return claripy.BVV(arg.concrete_value, size)
1152
+ raise TypeError(f"Type mismatch of symbolic data: expected {ty}, got {len(arg)} bits")
1153
+ return arg
1154
+ if isinstance(ty, (SimTypeFloat)):
1155
+ raise TypeError(
1156
+ "It's unclear how to coerce a bitvector to %s. "
1157
+ "Do you want .raw_to_fp or .val_to_fp, and signed or unsigned?"
1158
+ )
1159
+ raise TypeError(f"Type mismatch: expected {ty}, got bitvector")
1160
+
1161
+ raise TypeError(f"I don't know how to serialize {arg!r}.")
1162
+
1163
+ def __repr__(self):
1164
+ return f"<{self.__class__.__name__}>"
1165
+
1166
+ def __eq__(self, other):
1167
+ return isinstance(other, self.__class__)
1168
+
1169
+ @classmethod
1170
+ def _match(cls, arch, args: list[SimRegArg | SimStackArg], sp_delta):
1171
+ if (
1172
+ cls.arches() is not None and ":" not in arch.name and not isinstance(arch, cls.arches())
1173
+ ): # pylint:disable=isinstance-second-argument-not-valid-type
1174
+ return False
1175
+ if sp_delta != cls.STACKARG_SP_DIFF:
1176
+ return False
1177
+
1178
+ def _arg_ident(a: SimRegArg | SimStackArg) -> int | str:
1179
+ if isinstance(a, SimRegArg):
1180
+ return a.reg_name
1181
+ return a.stack_offset
1182
+
1183
+ sample_inst = cls(arch)
1184
+ all_fp_args: set[int | str] = {_arg_ident(a) for a in sample_inst.fp_args}
1185
+ all_int_args: set[int | str] = {_arg_ident(a) for a in sample_inst.int_args}
1186
+ both_iter = sample_inst.memory_args
1187
+ max_args = cls._guess_arg_count(args)
1188
+ some_both_args: set[int | str] = {_arg_ident(next(both_iter)) for _ in range(max_args)}
1189
+
1190
+ new_args = []
1191
+ for arg in args:
1192
+ arg_ident = _arg_ident(arg)
1193
+ if arg_ident not in all_fp_args and arg_ident not in all_int_args and arg_ident not in some_both_args:
1194
+ if isinstance(arg, SimRegArg) and arg.reg_name in sample_inst.CALLER_SAVED_REGS:
1195
+ continue
1196
+ return False
1197
+ new_args.append(arg)
1198
+
1199
+ # update args (e.g., drop caller-saved register arguments)
1200
+ args.clear()
1201
+ args.extend(new_args)
1202
+
1203
+ return True
1204
+
1205
+ @classmethod
1206
+ def _guess_arg_count(cls, args, limit: int = 64) -> int:
1207
+ # pylint:disable=not-callable
1208
+ assert cls.ARCH is not None
1209
+ stack_args = [a for a in args if isinstance(a, SimStackArg)]
1210
+ stack_arg_count = (
1211
+ (max(a.stack_offset for a in stack_args) // cls.ARCH(archinfo.Endness.LE).bytes + 1) if stack_args else 0
1212
+ )
1213
+ return min(limit, max(len(args), stack_arg_count))
1214
+
1215
+ @staticmethod
1216
+ def find_cc(
1217
+ arch: archinfo.Arch, args: list[SimRegArg | SimStackArg], sp_delta: int, platform: str | None = "Linux"
1218
+ ) -> SimCC | None:
1219
+ """
1220
+ Pinpoint the best-fit calling convention and return the corresponding SimCC instance, or None if no fit is
1221
+ found.
1222
+
1223
+ :param arch: An ArchX instance. Can be obtained from archinfo.
1224
+ :param args: A list of arguments. It may be updated by the first matched calling convention to
1225
+ remove non-argument arguments.
1226
+ :param sp_delta: The change of stack pointer before and after the call is made.
1227
+ :return: A calling convention instance, or None if none of the SimCC subclasses seems to fit the
1228
+ arguments provided.
1229
+ """
1230
+ if arch.name not in CC:
1231
+ return None
1232
+ if platform not in CC[arch.name]:
1233
+ # fallback to default
1234
+ platform = "default"
1235
+ possible_cc_classes = CC[arch.name][platform]
1236
+ for cc_cls in possible_cc_classes:
1237
+ if cc_cls._match(arch, args, sp_delta):
1238
+ return cc_cls(arch)
1239
+ return None
1240
+
1241
+ @classmethod
1242
+ def arches(cls) -> tuple[type[archinfo.Arch], ...]:
1243
+ if cls.ARCH is not None:
1244
+ return (cls.ARCH, *cls.EXTRA_ARCHES)
1245
+ return cls.EXTRA_ARCHES
1246
+
1247
+ def get_arg_info(self, state, prototype):
1248
+ """
1249
+ This is just a simple wrapper that collects the information from various locations
1250
+ prototype is as passed to self.arg_locs and self.get_args
1251
+ :param angr.SimState state: The state to evaluate and extract the values from
1252
+ :return: A list of tuples, where the nth tuple is (type, name, location, value) of the nth argument
1253
+ """
1254
+
1255
+ argument_locations = self.arg_locs(prototype)
1256
+ argument_values = self.get_args(state, prototype)
1257
+
1258
+ argument_types = prototype.args
1259
+ argument_names = prototype.arg_names if prototype.arg_names else ["unknown"] * len(prototype.args)
1260
+ return list(zip(argument_types, argument_names, argument_locations, argument_values))
1261
+
1262
+
1263
+ class SimLyingRegArg(SimRegArg):
1264
+ """
1265
+ A register that LIES about the types it holds
1266
+ """
1267
+
1268
+ def __init__(self, name, size=8):
1269
+ super().__init__(name, 8)
1270
+ self._real_size = size
1271
+
1272
+ def get_value(self, state, **kwargs): # pylint:disable=arguments-differ
1273
+ # val = super(SimLyingRegArg, self).get_value(state, **kwargs)
1274
+ val = state.registers.load(self.reg_name).raw_to_fp()
1275
+ if self._real_size == 4:
1276
+ val = claripy.fpToFP(claripy.fp.RM.RM_NearestTiesEven, val.raw_to_fp(), claripy.FSORT_FLOAT)
1277
+ return val
1278
+
1279
+ def set_value(self, state, value, **kwargs): # pylint:disable=arguments-differ,unused-argument
1280
+ value = self.check_value_set(value, state.arch)
1281
+ if self._real_size == 4:
1282
+ value = claripy.fpToFP(claripy.fp.RM.RM_NearestTiesEven, value.raw_to_fp(), claripy.FSORT_DOUBLE)
1283
+ state.registers.store(self.reg_name, value)
1284
+ # super(SimLyingRegArg, self).set_value(state, value, endness=endness, **kwargs)
1285
+
1286
+ def refine(self, size, arch=None, offset=None, is_fp=None):
1287
+ return SimLyingRegArg(self.reg_name, size)
1288
+
1289
+
1290
+ class SimCCUsercall(SimCC):
1291
+ def __init__(self, arch, args, ret_loc):
1292
+ super().__init__(arch)
1293
+ self.args = args
1294
+ self.ret_loc = ret_loc
1295
+
1296
+ ArgSession = UsercallArgSession
1297
+
1298
+ def next_arg(self, session: UsercallArgSession, arg_type): # type:ignore[reportIncompatibleMethodOverride]
1299
+ return next(session.real_args)
1300
+
1301
+ def return_val(self, ty, **kwargs): # type:ignore # pylint: disable=unused-argument
1302
+ return self.ret_loc
1303
+
1304
+
1305
+ class SimCCCdecl(SimCC):
1306
+ ARG_REGS = [] # All arguments are passed in stack
1307
+ FP_ARG_REGS = []
1308
+ STACKARG_SP_DIFF = 4 # Return address is pushed on to stack by call
1309
+ CALLER_SAVED_REGS = ["eax", "ecx", "edx"]
1310
+ RETURN_VAL = SimRegArg("eax", 4)
1311
+ OVERFLOW_RETURN_VAL = SimRegArg("edx", 4)
1312
+ FP_RETURN_VAL = SimLyingRegArg("st0")
1313
+ RETURN_ADDR = SimStackArg(0, 4)
1314
+ ARCH = archinfo.ArchX86
1315
+
1316
+ def next_arg(self, session, arg_type):
1317
+ if isinstance(arg_type, (SimTypeArray, SimTypeFixedSizeArray)): # hack
1318
+ arg_type = SimTypePointer(arg_type.elem_type).with_arch(self.arch)
1319
+ locs_size = 0
1320
+ byte_size = arg_type.size // self.arch.byte_width if arg_type.size is not None else self.arch.bytes
1321
+ locs = []
1322
+ while locs_size < byte_size:
1323
+ locs.append(next(session.both_iter))
1324
+ locs_size += locs[-1].size
1325
+
1326
+ return refine_locs_with_struct_type(self.arch, locs, arg_type)
1327
+
1328
+ STRUCT_RETURN_THRESHOLD = 32
1329
+
1330
+ def return_val(self, ty, perspective_returned=False):
1331
+ if ty._arch is None:
1332
+ ty = ty.with_arch(self.arch)
1333
+ if not isinstance(ty, SimStruct):
1334
+ return super().return_val(ty, perspective_returned)
1335
+
1336
+ if ty.size > self.STRUCT_RETURN_THRESHOLD:
1337
+ # TODO this code is duplicated a ton of places. how should it be a function?
1338
+ byte_size = ty.size // self.arch.byte_width
1339
+ referenced_locs = [SimStackArg(offset, self.arch.bytes) for offset in range(0, byte_size, self.arch.bytes)]
1340
+ referenced_loc = refine_locs_with_struct_type(self.arch, referenced_locs, ty)
1341
+ ptr_loc = self.RETURN_VAL if perspective_returned else SimStackArg(0, 4)
1342
+ assert ptr_loc is not None
1343
+ return SimReferenceArgument(ptr_loc, referenced_loc)
1344
+
1345
+ return refine_locs_with_struct_type(self.arch, [self.RETURN_VAL, self.OVERFLOW_RETURN_VAL], ty)
1346
+
1347
+ def return_in_implicit_outparam(self, ty):
1348
+ if isinstance(ty, SimTypeBottom):
1349
+ return False
1350
+ return isinstance(ty, SimStruct) and ty.size > self.STRUCT_RETURN_THRESHOLD
1351
+
1352
+
1353
+ class SimCCMicrosoftCdecl(SimCCCdecl):
1354
+ STRUCT_RETURN_THRESHOLD = 64
1355
+
1356
+
1357
+ class SimCCMicrosoftThiscall(SimCCCdecl):
1358
+ CALLEE_CLEANUP = True
1359
+ ARG_REGS = ["ecx"]
1360
+ CALLER_SAVED_REGS = ["eax", "ecx", "edx"]
1361
+ STRUCT_RETURN_THRESHOLD = 64
1362
+
1363
+ def arg_locs(self, prototype) -> list[SimFunctionArgument]:
1364
+ if prototype._arch is None:
1365
+ prototype = prototype.with_arch(self.arch)
1366
+ session = self.arg_session(prototype.returnty)
1367
+ if not prototype.args:
1368
+ return []
1369
+ return [SimRegArg("ecx", self.arch.bytes)] + [self.next_arg(session, arg_ty) for arg_ty in prototype.args[1:]]
1370
+
1371
+
1372
+ class SimCCStdcall(SimCCMicrosoftCdecl):
1373
+ CALLEE_CLEANUP = True
1374
+
1375
+
1376
+ class SimCCMicrosoftFastcall(SimCC):
1377
+ ARG_REGS = ["ecx", "edx"] # Remaining arguments are passed in stack
1378
+ STACKARG_SP_DIFF = 4 # Return address is pushed on to stack by call
1379
+ RETURN_VAL = SimRegArg("eax", 4)
1380
+ RETURN_ADDR = SimStackArg(0, 4)
1381
+ ARCH = archinfo.ArchX86
1382
+
1383
+
1384
+ class MicrosoftAMD64ArgSession:
1385
+ def __init__(self, cc):
1386
+ self.cc = cc
1387
+ self.int_iter = cc.int_args
1388
+ self.fp_iter = cc.fp_args
1389
+ self.both_iter = cc.memory_args
1390
+
1391
+
1392
+ class SimCCMicrosoftAMD64(SimCC):
1393
+ ARG_REGS = ["rcx", "rdx", "r8", "r9"]
1394
+ FP_ARG_REGS = ["xmm0", "xmm1", "xmm2", "xmm3"]
1395
+ STACKARG_SP_DIFF = 8 # Return address is pushed on to stack by call
1396
+ STACKARG_SP_BUFF = 32 # 32 bytes of shadow stack space
1397
+ RETURN_VAL = SimRegArg("rax", 8)
1398
+ OVERFLOW_RETURN_VAL = SimRegArg("rdx", 8)
1399
+ FP_RETURN_VAL = SimRegArg("xmm0", 32)
1400
+ RETURN_ADDR = SimStackArg(0, 8)
1401
+ ARCH = archinfo.ArchAMD64
1402
+ STACK_ALIGNMENT = 16
1403
+
1404
+ ArgSession = MicrosoftAMD64ArgSession
1405
+
1406
+ STRUCT_RETURN_THRESHOLD = 64
1407
+
1408
+ def next_arg(self, session, arg_type):
1409
+ if isinstance(arg_type, (SimTypeArray, SimTypeFixedSizeArray)): # hack
1410
+ arg_type = SimTypePointer(arg_type.elem_type).with_arch(self.arch)
1411
+ try:
1412
+ int_loc = next(session.int_iter)
1413
+ fp_loc = next(session.fp_iter)
1414
+ except StopIteration:
1415
+ int_loc = fp_loc = next(session.both_iter)
1416
+
1417
+ byte_size = arg_type.size // self.arch.byte_width if arg_type.size is not None else self.arch.bytes
1418
+
1419
+ if isinstance(arg_type, SimTypeFloat):
1420
+ return fp_loc.refine(size=byte_size, is_fp=True, arch=self.arch)
1421
+
1422
+ if byte_size <= int_loc.size:
1423
+ return int_loc.refine(size=byte_size, is_fp=False, arch=self.arch)
1424
+
1425
+ referenced_locs = [SimStackArg(offset, self.arch.bytes) for offset in range(0, byte_size, self.arch.bytes)]
1426
+ referenced_loc = refine_locs_with_struct_type(self.arch, referenced_locs, arg_type)
1427
+ return SimReferenceArgument(int_loc, referenced_loc)
1428
+
1429
+ def return_in_implicit_outparam(self, ty):
1430
+ if isinstance(ty, SimTypeBottom):
1431
+ return False
1432
+ return not isinstance(ty, SimTypeFloat) and ty.size > self.STRUCT_RETURN_THRESHOLD
1433
+
1434
+ def return_val(self, ty, perspective_returned=False):
1435
+ if ty._arch is None:
1436
+ ty = ty.with_arch(self.arch)
1437
+
1438
+ # Unions are allocated according to the layout of the largest member
1439
+ if isinstance(ty, SimUnion):
1440
+ chosen = None
1441
+ size = None
1442
+ for subty in ty.members.values():
1443
+ if subty.size is not None and (size is None or size < subty.size):
1444
+ chosen = subty
1445
+ size = subty.size
1446
+ if chosen is None:
1447
+ # fallback to void*
1448
+ chosen = SimTypePointer(SimTypeBottom()).with_arch(self.arch)
1449
+ return self.return_val(chosen, perspective_returned=perspective_returned)
1450
+
1451
+ if not isinstance(ty, SimStruct):
1452
+ return super().return_val(ty, perspective_returned)
1453
+
1454
+ if ty.size > self.STRUCT_RETURN_THRESHOLD:
1455
+ # TODO this code is duplicated a ton of places. how should it be a function?
1456
+ byte_size = ty.size // self.arch.byte_width
1457
+ referenced_locs = [SimStackArg(offset, self.arch.bytes) for offset in range(0, byte_size, self.arch.bytes)]
1458
+ referenced_loc = refine_locs_with_struct_type(self.arch, referenced_locs, ty)
1459
+ if perspective_returned:
1460
+ ptr_loc = self.RETURN_VAL
1461
+ else:
1462
+ ptr_loc = self.next_arg(self.ArgSession(self), SimTypePointer(SimTypeBottom()).with_arch(self.arch))
1463
+ return SimReferenceArgument(ptr_loc, referenced_loc)
1464
+
1465
+ return refine_locs_with_struct_type(self.arch, [self.RETURN_VAL], ty)
1466
+
1467
+
1468
+ class SimCCSyscall(SimCC):
1469
+ """
1470
+ The base class of all syscall CCs.
1471
+ """
1472
+
1473
+ ERROR_REG: SimRegArg = None
1474
+ SYSCALL_ERRNO_START = None
1475
+
1476
+ @staticmethod
1477
+ def syscall_num(state) -> int:
1478
+ raise NotImplementedError
1479
+
1480
+ def linux_syscall_update_error_reg(self, state, expr):
1481
+ # special handling for Linux syscalls: on some architectures (mips/a3, powerpc/cr0_0) a bool indicating success
1482
+ # or failure of a system call is used as an error flag (0 for success, 1 for error). we have to set this
1483
+ if state.project is None or state.project.simos is None or state.project.simos.name != "Linux":
1484
+ return expr
1485
+ if type(expr) is int:
1486
+ expr = claripy.BVV(expr, state.arch.bits)
1487
+ with contextlib.suppress(AttributeError):
1488
+ expr = expr.ast
1489
+ nbits = self.ERROR_REG.size * state.arch.byte_width
1490
+ error_cond = claripy.UGE(expr, self.SYSCALL_ERRNO_START)
1491
+ if state.solver.is_false(error_cond):
1492
+ # guaranteed no error
1493
+ error_reg_val = claripy.BVV(0, nbits)
1494
+ elif state.solver.is_true(error_cond):
1495
+ # guaranteed error
1496
+ error_reg_val = claripy.BVV(-1, nbits)
1497
+ expr = -expr
1498
+ else:
1499
+ # both are satisfied, handle gracefully
1500
+ error_reg_val = claripy.If(error_cond, claripy.BVV(-1, nbits), 0)
1501
+ expr = claripy.If(error_cond, -expr, expr)
1502
+
1503
+ self.ERROR_REG.set_value(state, error_reg_val)
1504
+ return expr
1505
+
1506
+ def set_return_val(self, state, val, ty, **kwargs): # type:ignore # pylint:disable=arguments-differ
1507
+ if self.ERROR_REG is not None:
1508
+ val = self.linux_syscall_update_error_reg(state, val)
1509
+ super().set_return_val(state, val, ty, **kwargs)
1510
+
1511
+
1512
+ class SimCCX86LinuxSyscall(SimCCSyscall):
1513
+ ARG_REGS = ["ebx", "ecx", "edx", "esi", "edi", "ebp"]
1514
+ FP_ARG_REGS = []
1515
+ RETURN_VAL = SimRegArg("eax", 4)
1516
+ RETURN_ADDR = SimRegArg("ip_at_syscall", 4)
1517
+ ARCH = archinfo.ArchX86
1518
+
1519
+ @classmethod
1520
+ def _match(cls, arch, args, sp_delta): # pylint: disable=unused-argument
1521
+ # never appears anywhere except syscalls
1522
+ return False
1523
+
1524
+ @staticmethod
1525
+ def syscall_num(state):
1526
+ return state.regs.eax
1527
+
1528
+
1529
+ class SimCCX86WindowsSyscall(SimCCSyscall):
1530
+ # TODO: Make sure the information is correct
1531
+ ARG_REGS = ["ecx"]
1532
+ FP_ARG_REGS = []
1533
+ RETURN_VAL = SimRegArg("eax", 4)
1534
+ RETURN_ADDR = SimRegArg("ip_at_syscall", 4)
1535
+ ARCH = archinfo.ArchX86
1536
+
1537
+ @classmethod
1538
+ def _match(cls, arch, args, sp_delta): # pylint: disable=unused-argument
1539
+ # never appears anywhere except syscalls
1540
+ return False
1541
+
1542
+ @staticmethod
1543
+ def syscall_num(state):
1544
+ return state.regs.eax
1545
+
1546
+
1547
+ class SimCCSystemVAMD64(SimCC):
1548
+ ARG_REGS = ["rdi", "rsi", "rdx", "rcx", "r8", "r9"]
1549
+ FP_ARG_REGS = ["xmm0", "xmm1", "xmm2", "xmm3", "xmm4", "xmm5", "xmm6", "xmm7"]
1550
+ STACKARG_SP_DIFF = 8 # Return address is pushed on to stack by call
1551
+ CALLER_SAVED_REGS = [
1552
+ "rdi",
1553
+ "rsi",
1554
+ "rdx",
1555
+ "rcx",
1556
+ "r8",
1557
+ "r9",
1558
+ "r10",
1559
+ "r11",
1560
+ "rax",
1561
+ ]
1562
+ RETURN_ADDR = SimStackArg(0, 8)
1563
+ RETURN_VAL = SimRegArg("rax", 8)
1564
+ OVERFLOW_RETURN_VAL = SimRegArg("rdx", 8)
1565
+ FP_RETURN_VAL = SimRegArg("xmm0", 128)
1566
+ OVERFLOW_FP_RETURN_VAL = SimRegArg("xmm1", 128)
1567
+ ARCH = archinfo.ArchAMD64
1568
+ STACK_ALIGNMENT = 16
1569
+
1570
+ @classmethod
1571
+ def _match(cls, arch, args, sp_delta):
1572
+ if cls.ARCH is not None and ":" not in arch.name and not isinstance(arch, cls.ARCH):
1573
+ return False
1574
+ # if sp_delta != cls.STACKARG_SP_DIFF:
1575
+ # return False
1576
+
1577
+ sample_inst = cls(arch)
1578
+ all_fp_args = list(sample_inst.fp_args)
1579
+ all_int_args = list(sample_inst.int_args)
1580
+ both_iter = sample_inst.memory_args
1581
+ some_both_args = [next(both_iter) for _ in range(len(args))]
1582
+
1583
+ for arg in args:
1584
+ ex_arg = arg
1585
+ # attempt to coerce the argument into a form that might show up in these lists
1586
+ if type(ex_arg) is SimRegArg:
1587
+ if ex_arg.reg_name not in arch.registers:
1588
+ # danger!
1589
+ # if the register name is a digit-only string, we use it as an offset
1590
+ try:
1591
+ regfile_offset = int(ex_arg.reg_name)
1592
+ except ValueError:
1593
+ return False
1594
+ else:
1595
+ regfile_offset = arch.registers[ex_arg.reg_name][0]
1596
+ while regfile_offset not in arch.register_names:
1597
+ regfile_offset -= 1
1598
+ ex_arg.reg_name = arch.register_names[regfile_offset]
1599
+ ex_arg.reg_offset = 0
1600
+
1601
+ if ex_arg not in all_fp_args and ex_arg not in all_int_args and ex_arg not in some_both_args:
1602
+ if isinstance(arg, SimStackArg) and arg.stack_offset == 0:
1603
+ continue # ignore return address?
1604
+ return False
1605
+
1606
+ return True
1607
+
1608
+ # https://raw.githubusercontent.com/wiki/hjl-tools/x86-psABI/x86-64-psABI-1.0.pdf
1609
+ # section 3.2.3
1610
+ def next_arg(self, session, arg_type):
1611
+ if isinstance(arg_type, (SimTypeArray, SimTypeFixedSizeArray)): # hack
1612
+ arg_type = SimTypePointer(arg_type.elem_type).with_arch(self.arch)
1613
+ state = session.getstate()
1614
+ classification = self._classify(arg_type)
1615
+ try:
1616
+ mapped_classes = []
1617
+ for cls in classification:
1618
+ if cls == "SSEUP":
1619
+ mapped_classes.append(mapped_classes[-1].sse_extend(self.arch.bytes))
1620
+ elif cls == "NO_CLASS":
1621
+ raise NotImplementedError("Bug. Report to @rhelmot")
1622
+ elif cls == "MEMORY":
1623
+ mapped_classes.append(next(session.both_iter))
1624
+ elif cls == "INTEGER":
1625
+ mapped_classes.append(next(session.int_iter))
1626
+ elif cls == "SSE":
1627
+ mapped_classes.append(next(session.fp_iter))
1628
+ else:
1629
+ raise NotImplementedError("Bug. Report to @rhelmot")
1630
+ except StopIteration:
1631
+ session.setstate(state)
1632
+ mapped_classes = [next(session.both_iter) for _ in classification]
1633
+
1634
+ return refine_locs_with_struct_type(self.arch, mapped_classes, arg_type)
1635
+
1636
+ def return_val(self, ty: SimType | None, perspective_returned=False):
1637
+ if ty is None:
1638
+ return None
1639
+ if ty._arch is None:
1640
+ ty = ty.with_arch(self.arch)
1641
+ classification = self._classify(ty)
1642
+ if any(cls == "MEMORY" for cls in classification):
1643
+ assert all(cls == "MEMORY" for cls in classification)
1644
+ assert ty.size is not None
1645
+ byte_size = ty.size // self.arch.byte_width
1646
+ referenced_locs = [SimStackArg(offset, self.arch.bytes) for offset in range(0, byte_size, self.arch.bytes)]
1647
+ referenced_loc = refine_locs_with_struct_type(self.arch, referenced_locs, ty)
1648
+ ptr_loc = self.RETURN_VAL if perspective_returned else SimRegArg("rdi", 8)
1649
+ assert ptr_loc is not None
1650
+ return SimReferenceArgument(ptr_loc, referenced_loc)
1651
+ mapped_classes = []
1652
+ int_iter = iter([self.RETURN_VAL, self.OVERFLOW_RETURN_VAL])
1653
+ fp_iter = iter([self.FP_RETURN_VAL, self.OVERFLOW_FP_RETURN_VAL])
1654
+ for cls in classification:
1655
+ if cls == "SSEUP":
1656
+ mapped_classes.append(mapped_classes[-1].sse_extend(self.arch.bytes))
1657
+ elif cls == "NO_CLASS":
1658
+ raise NotImplementedError("Bug. Report to @rhelmot")
1659
+ elif cls == "INTEGER":
1660
+ mapped_classes.append(next(int_iter))
1661
+ elif cls == "SSE":
1662
+ mapped_classes.append(next(fp_iter))
1663
+ else:
1664
+ raise NotImplementedError("Bug. Report to @rhelmot")
1665
+
1666
+ return refine_locs_with_struct_type(self.arch, mapped_classes, ty)
1667
+
1668
+ def return_in_implicit_outparam(self, ty):
1669
+ if isinstance(ty, SimTypeBottom):
1670
+ return False
1671
+ # :P
1672
+ return isinstance(self.return_val(ty), SimReferenceArgument)
1673
+
1674
+ def _classify(self, ty, chunksize=None):
1675
+ if chunksize is None:
1676
+ chunksize = self.arch.bytes
1677
+ # treat BOT as INTEGER
1678
+ nchunks = 1 if isinstance(ty, SimTypeBottom) else (ty.size // self.arch.byte_width + chunksize - 1) // chunksize
1679
+ if isinstance(ty, (SimTypeFloat,)):
1680
+ return ["SSE"] + ["SSEUP"] * (nchunks - 1)
1681
+ if isinstance(ty, (SimTypeReg, SimTypeNum, SimTypeBottom)):
1682
+ return ["INTEGER"] * nchunks
1683
+ if isinstance(ty, SimTypeArray) or (isinstance(ty, SimType) and isinstance(ty, NamedTypeMixin)):
1684
+ # NamedTypeMixin covers SimUnion, SimStruct, SimTypeString, and other struct-like classes
1685
+ assert ty.size is not None
1686
+ if ty.size > 512:
1687
+ return ["MEMORY"] * nchunks
1688
+ flattened = self._flatten(ty)
1689
+ if flattened is None:
1690
+ return ["MEMORY"] * nchunks
1691
+ result = ["NO_CLASS"] * nchunks
1692
+ for offset, subty_list in flattened.items():
1693
+ for subty in subty_list:
1694
+ # is the smaller chunk size necessary? Genuinely unsure
1695
+ subresult = self._classify(subty, chunksize=1)
1696
+ idx_start = offset // chunksize
1697
+ idx_end = (offset + (subty.size // self.arch.byte_width) - 1) // chunksize
1698
+ for i, idx in enumerate(range(idx_start, idx_end + 1)):
1699
+ subclass = subresult[i * chunksize]
1700
+ result[idx] = self._combine_classes(result[idx], subclass)
1701
+ if any(subresult == "MEMORY" for subresult in result):
1702
+ return ["MEMORY"] * nchunks
1703
+ if nchunks > 2 and (result[0] != "SSE" or any(subresult != "SSEUP" for subresult in result[1:])):
1704
+ return ["MEMORY"] * nchunks
1705
+ for i in range(1, nchunks):
1706
+ if result[i] == "SSEUP" and result[i - 1] not in ("SSE", "SSEUP"):
1707
+ result[i] = "SSE"
1708
+ return result
1709
+ raise NotImplementedError("Ummmmm... not sure what goes here. report bug to @rhelmot")
1710
+
1711
+ def _flatten(self, ty) -> dict[int, list[SimType]] | None:
1712
+ result: dict[int, list[SimType]] = defaultdict(list)
1713
+ if isinstance(ty, SimStruct):
1714
+ if ty.packed:
1715
+ return None
1716
+ for field, subty in ty.fields.items():
1717
+ offset = ty.offsets[field]
1718
+ subresult = self._flatten(subty)
1719
+ if subresult is None:
1720
+ return None
1721
+ for suboffset, subsubty_list in subresult.items():
1722
+ result[offset + suboffset] += subsubty_list
1723
+ elif isinstance(ty, SimTypeFixedSizeArray):
1724
+ assert ty.length is not None and ty.elem_type.size is not None
1725
+ subresult = self._flatten(ty.elem_type)
1726
+ if subresult is None:
1727
+ return None
1728
+ for suboffset, subsubty_list in subresult.items():
1729
+ for idx in range(ty.length):
1730
+ # TODO I think we need an explicit stride field on array types
1731
+ result[idx * ty.elem_type.size // self.arch.byte_width + suboffset] += subsubty_list
1732
+ elif isinstance(ty, SimUnion):
1733
+ for subty in ty.members.values():
1734
+ subresult = self._flatten(subty)
1735
+ if subresult is None:
1736
+ return None
1737
+ for suboffset, subsubty_list in subresult.items():
1738
+ result[suboffset] += subsubty_list
1739
+ else:
1740
+ result[0].append(ty)
1741
+ return result
1742
+
1743
+ def _combine_classes(self, cls1, cls2):
1744
+ if cls1 == cls2:
1745
+ return cls1
1746
+ if cls1 == "NO_CLASS":
1747
+ return cls2
1748
+ if cls2 == "NO_CLASS":
1749
+ return cls1
1750
+ if cls1 == "MEMORY" or cls2 == "MEMORY":
1751
+ return "MEMORY"
1752
+ if cls1 == "INTEGER" or cls2 == "INTEGER":
1753
+ return "INTEGER"
1754
+ return "SSE"
1755
+
1756
+
1757
+ class SimCCAMD64LinuxSyscall(SimCCSyscall):
1758
+ ARG_REGS = ["rdi", "rsi", "rdx", "r10", "r8", "r9"]
1759
+ RETURN_VAL = SimRegArg("rax", 8)
1760
+ RETURN_ADDR = SimRegArg("ip_at_syscall", 8)
1761
+ ARCH = archinfo.ArchAMD64
1762
+ CALLER_SAVED_REGS = ["rax", "rcx", "r11"]
1763
+
1764
+ @staticmethod
1765
+ def _match(arch, args, sp_delta): # type:ignore # pylint: disable=unused-argument
1766
+ # doesn't appear anywhere but syscalls
1767
+ return False
1768
+
1769
+ @staticmethod
1770
+ def syscall_num(state):
1771
+ return state.regs.rax
1772
+
1773
+
1774
+ class SimCCAMD64WindowsSyscall(SimCCSyscall):
1775
+ # TODO: Make sure the information is correct
1776
+ ARG_REGS = ["rcx"]
1777
+ FP_ARG_REGS = []
1778
+ RETURN_VAL = SimRegArg("rax", 8)
1779
+ RETURN_ADDR = SimRegArg("ip_at_syscall", 8)
1780
+ ARCH = archinfo.ArchAMD64
1781
+
1782
+ @classmethod
1783
+ def _match(cls, arch, args, sp_delta): # pylint: disable=unused-argument
1784
+ # never appears anywhere except syscalls
1785
+ return False
1786
+
1787
+ @staticmethod
1788
+ def syscall_num(state):
1789
+ return state.regs.rax
1790
+
1791
+
1792
+ class SimCCARM(SimCC):
1793
+ ARG_REGS = ["r0", "r1", "r2", "r3"]
1794
+ FP_ARG_REGS = [] # regular arg regs are used as fp arg regs
1795
+ CALLER_SAVED_REGS = ["r0", "r1", "r2", "r3"]
1796
+ RETURN_ADDR = SimRegArg("lr", 4)
1797
+ RETURN_VAL = SimRegArg("r0", 4)
1798
+ OVERFLOW_RETURN_VAL = SimRegArg("r1", 4)
1799
+ ARCH = archinfo.ArchARM
1800
+
1801
+ # https://github.com/ARM-software/abi-aa/blob/60a8eb8c55e999d74dac5e368fc9d7e36e38dda4/aapcs32/aapcs32.rst#parameter-passing
1802
+ def next_arg(self, session, arg_type):
1803
+ if isinstance(arg_type, (SimTypeArray, SimTypeFixedSizeArray)): # hack
1804
+ arg_type = SimTypePointer(arg_type.elem_type).with_arch(self.arch)
1805
+ state = session.getstate()
1806
+ classification = self._classify(arg_type)
1807
+ try:
1808
+ mapped_classes = []
1809
+ for cls in classification:
1810
+ if cls == "DOUBLEP":
1811
+ if session.getstate()[1] % 2 == 1: # doubles must start on an even register
1812
+ next(session.int_iter)
1813
+
1814
+ if session.getstate()[1] == len(self.ARG_REGS) - 2:
1815
+ mapped_classes.append(next(session.int_iter))
1816
+ mapped_classes.append(next(session.both_iter))
1817
+ else:
1818
+ try:
1819
+ mapped_classes.append(next(session.int_iter))
1820
+ mapped_classes.append(next(session.int_iter))
1821
+ except StopIteration:
1822
+ mapped_classes.append(next(session.both_iter))
1823
+ mapped_classes.append(next(session.both_iter))
1824
+ elif cls == "NO_CLASS":
1825
+ raise NotImplementedError("Bug. Report to @rhelmot")
1826
+ elif cls == "MEMORY":
1827
+ mapped_classes.append(next(session.both_iter))
1828
+ elif cls in {"INTEGER", "SINGLEP"}:
1829
+ try:
1830
+ mapped_classes.append(next(session.int_iter))
1831
+ except StopIteration:
1832
+ mapped_classes.append(next(session.both_iter))
1833
+ else:
1834
+ raise NotImplementedError("Bug. Report to @rhelmot")
1835
+ except StopIteration:
1836
+ session.setstate(state)
1837
+ mapped_classes = [next(session.both_iter) for _ in classification]
1838
+
1839
+ return refine_locs_with_struct_type(self.arch, mapped_classes, arg_type)
1840
+
1841
+ def _classify(self, ty, chunksize=None):
1842
+ if chunksize is None:
1843
+ chunksize = self.arch.bytes
1844
+ # treat BOT as INTEGER
1845
+ nchunks = 1 if isinstance(ty, SimTypeBottom) else (ty.size // self.arch.byte_width + chunksize - 1) // chunksize
1846
+ if isinstance(ty, (SimTypeInt, SimTypeChar, SimTypePointer, SimTypeNum, SimTypeBottom, SimTypeReference)):
1847
+ return ["INTEGER"] * nchunks
1848
+ if isinstance(ty, (SimTypeFloat,)):
1849
+ if ty.size == 64:
1850
+ return ["DOUBLEP"]
1851
+ if ty.size == 32:
1852
+ return ["SINGLEP"]
1853
+ return ["NO_CLASS"]
1854
+ if isinstance(ty, (SimStruct, SimTypeFixedSizeArray, SimUnion)):
1855
+ flattened = self._flatten(ty)
1856
+ if flattened is None:
1857
+ return ["MEMORY"] * nchunks
1858
+ result = ["NO_CLASS"] * nchunks
1859
+ for offset, subty_list in flattened.items():
1860
+ for subty in subty_list:
1861
+ # is the smaller chunk size necessary? Genuinely unsure
1862
+ subresult = self._classify(subty, chunksize=1)
1863
+ idx_start = offset // chunksize
1864
+ idx_end = (offset + (subty.size // self.arch.byte_width) - 1) // chunksize
1865
+ for i, idx in enumerate(range(idx_start, idx_end + 1)):
1866
+ subclass = subresult[i * chunksize]
1867
+ result[idx] = self._combine_classes(result[idx], subclass)
1868
+ return result
1869
+ raise NotImplementedError("Ummmmm... not sure what goes here. report bug to @rhelmot")
1870
+
1871
+ def _combine_classes(self, cls1, cls2):
1872
+ if cls1 == cls2:
1873
+ return cls1
1874
+ if cls1 == "NO_CLASS":
1875
+ return cls2
1876
+ if cls2 == "NO_CLASS":
1877
+ return cls1
1878
+ if cls1 == "MEMORY" or cls2 == "MEMORY":
1879
+ return "MEMORY"
1880
+ if cls1 == "INTEGER" or cls2 == "INTEGER":
1881
+ return "INTEGER"
1882
+ return "SSE"
1883
+
1884
+ def _flatten(self, ty) -> dict[int, list[SimType]] | None:
1885
+ result: dict[int, list[SimType]] = defaultdict(list)
1886
+ if isinstance(ty, SimStruct):
1887
+ if ty.packed:
1888
+ return None
1889
+ for field, subty in ty.fields.items():
1890
+ offset = ty.offsets[field]
1891
+ subresult = self._flatten(subty)
1892
+ if subresult is None:
1893
+ return None
1894
+ for suboffset, subsubty_list in subresult.items():
1895
+ result[offset + suboffset] += subsubty_list
1896
+ elif isinstance(ty, SimTypeFixedSizeArray):
1897
+ assert ty.length is not None and ty.elem_type.size is not None
1898
+ subresult = self._flatten(ty.elem_type)
1899
+ if subresult is None:
1900
+ return None
1901
+ for suboffset, subsubty_list in subresult.items():
1902
+ for idx in range(ty.length):
1903
+ # TODO I think we need an explicit stride field on array types
1904
+ result[idx * ty.elem_type.size // self.arch.byte_width + suboffset] += subsubty_list
1905
+ elif isinstance(ty, SimUnion):
1906
+ for subty in ty.members.values():
1907
+ subresult = self._flatten(subty)
1908
+ if subresult is None:
1909
+ return None
1910
+ for suboffset, subsubty_list in subresult.items():
1911
+ result[suboffset] += subsubty_list
1912
+ else:
1913
+ result[0].append(ty)
1914
+ return result
1915
+
1916
+
1917
+ class SimCCARMHF(SimCCARM):
1918
+ ARG_REGS = ["r0", "r1", "r2", "r3"]
1919
+ FP_ARG_REGS = [f"s{i}" for i in range(16)] # regular arg regs are used as fp arg regs
1920
+ FP_RETURN_VAL = SimRegArg("s0", 32)
1921
+ CALLER_SAVED_REGS = []
1922
+ RETURN_ADDR = SimRegArg("lr", 4)
1923
+ RETURN_VAL = SimRegArg("r0", 4)
1924
+ OVERFLOW_RETURN_VAL = SimRegArg("r1", 4)
1925
+ ARCH = archinfo.ArchARMHF
1926
+ EXTRA_ARCHES = (archinfo.ArchARMCortexM,)
1927
+
1928
+ def next_arg(self, session, arg_type):
1929
+ if isinstance(arg_type, (SimTypeArray, SimTypeFixedSizeArray)): # hack
1930
+ arg_type = SimTypePointer(arg_type.elem_type).with_arch(self.arch)
1931
+ state = session.getstate()
1932
+ classification = self._classify(arg_type)
1933
+ try:
1934
+ mapped_classes = []
1935
+ for cls in classification:
1936
+ if cls == "DOUBLEP":
1937
+ if session.getstate()[1] % 2 == 1: # doubles must start on an even register
1938
+ next(session.int_iter)
1939
+
1940
+ if session.getstate()[1] == len(self.ARG_REGS) - 2:
1941
+ mapped_classes.append(next(session.int_iter))
1942
+ mapped_classes.append(next(session.both_iter))
1943
+ else:
1944
+ try:
1945
+ mapped_classes.append(next(session.int_iter))
1946
+ mapped_classes.append(next(session.int_iter))
1947
+ except StopIteration:
1948
+ mapped_classes.append(next(session.both_iter))
1949
+ mapped_classes.append(next(session.both_iter))
1950
+ elif cls == "NO_CLASS":
1951
+ raise NotImplementedError("Bug. Report to @rhelmot")
1952
+ elif cls == "MEMORY":
1953
+ mapped_classes.append(next(session.both_iter))
1954
+ elif cls == "INTEGER":
1955
+ try:
1956
+ mapped_classes.append(next(session.int_iter))
1957
+ except StopIteration:
1958
+ mapped_classes.append(next(session.both_iter))
1959
+ elif cls == "SINGLEP":
1960
+ try:
1961
+ mapped_classes.append(next(session.fp_iter))
1962
+ except StopIteration:
1963
+ mapped_classes.append(next(session.both_iter))
1964
+ else:
1965
+ raise NotImplementedError("Bug. Report to @rhelmot")
1966
+ except StopIteration:
1967
+ session.setstate(state)
1968
+ mapped_classes = [next(session.both_iter) for _ in classification]
1969
+
1970
+ return refine_locs_with_struct_type(self.arch, mapped_classes, arg_type)
1971
+
1972
+
1973
+ class SimCCARMLinuxSyscall(SimCCSyscall):
1974
+ # TODO: Make sure all the information is correct
1975
+ ARG_REGS = ["r0", "r1", "r2", "r3"]
1976
+ FP_ARG_REGS = [] # TODO: ???
1977
+ RETURN_ADDR = SimRegArg("ip_at_syscall", 4)
1978
+ RETURN_VAL = SimRegArg("r0", 4)
1979
+ ARCH = archinfo.ArchARM
1980
+
1981
+ @classmethod
1982
+ def _match(cls, arch, args, sp_delta): # pylint: disable=unused-argument
1983
+ # never appears anywhere except syscalls
1984
+ return False
1985
+
1986
+ @staticmethod
1987
+ def syscall_num(state):
1988
+ if ((state.regs.ip_at_syscall & 1) == 1).is_true():
1989
+ insn = state.mem[state.regs.ip_at_syscall - 3].short.resolved
1990
+ is_svc = ((insn & 0xFF00) == 0xDF00).is_true()
1991
+ svc_num = insn & 0xFF
1992
+ else:
1993
+ insn = state.mem[state.regs.ip_at_syscall - 4].dword.resolved
1994
+ is_svc = ((insn & 0x0F000000) == 0x0F000000).is_true()
1995
+ svc_num = insn & 0xFFFFFF
1996
+ if not is_svc:
1997
+ l.error("ARM syscall number being queried at an address which is not an SVC")
1998
+ return claripy.BVV(0, 32)
1999
+
2000
+ if len(svc_num) == 32 and (svc_num > 0x900000).is_true() and (svc_num < 0x90FFFF).is_true():
2001
+ return svc_num - 0x900000
2002
+ return state.regs.r7
2003
+
2004
+
2005
+ class SimCCAArch64(SimCC):
2006
+ ARG_REGS = ["x0", "x1", "x2", "x3", "x4", "x5", "x6", "x7"]
2007
+ FP_ARG_REGS = [] # TODO: ???
2008
+ RETURN_ADDR = SimRegArg("lr", 8)
2009
+ RETURN_VAL = SimRegArg("x0", 8)
2010
+ ARCH = archinfo.ArchAArch64
2011
+
2012
+
2013
+ class SimCCAArch64LinuxSyscall(SimCCSyscall):
2014
+ # TODO: Make sure all the information is correct
2015
+ ARG_REGS = ["x0", "x1", "x2", "x3", "x4", "x5", "x6", "x7"]
2016
+ FP_ARG_REGS = [] # TODO: ???
2017
+ RETURN_VAL = SimRegArg("x0", 8)
2018
+ RETURN_ADDR = SimRegArg("ip_at_syscall", 8)
2019
+ ARCH = archinfo.ArchAArch64
2020
+
2021
+ @classmethod
2022
+ def _match(cls, arch, args, sp_delta): # pylint: disable=unused-argument
2023
+ # never appears anywhere except syscalls
2024
+ return False
2025
+
2026
+ @staticmethod
2027
+ def syscall_num(state):
2028
+ return state.regs.x8
2029
+
2030
+
2031
+ class SimCCRISCV64LinuxSyscall(SimCCSyscall):
2032
+ # TODO: Make sure all the information is correct
2033
+ ARG_REGS = ["a0", "a1", "a2", "a3", "a4", "a5", "a6", "a7"]
2034
+ FP_ARG_REGS = [] # TODO: ???
2035
+ RETURN_VAL = SimRegArg("a0", 8)
2036
+ RETURN_ADDR = SimRegArg("ip_at_syscall", 4)
2037
+ ARCH = archinfo.ArchRISCV64
2038
+
2039
+ @classmethod
2040
+ def _match(cls, arch, args, sp_delta): # pylint: disable=unused-argument
2041
+ # never appears anywhere except syscalls
2042
+ return False
2043
+
2044
+ @staticmethod
2045
+ def syscall_num(state):
2046
+ return state.regs.a0
2047
+
2048
+
2049
+ class SimCCO32(SimCC):
2050
+ ARG_REGS = ["a0", "a1", "a2", "a3"]
2051
+ FP_ARG_REGS = [
2052
+ "f12",
2053
+ "f13",
2054
+ "f14",
2055
+ "f15",
2056
+ ] # Note double precision args are split between f12-f13, f14-f15 and single precision only use f12 and f14
2057
+ STACKARG_SP_BUFF = 16
2058
+ CALLER_SAVED_REGS = ["t9", "gp"]
2059
+ RETURN_ADDR = SimRegArg("ra", 4)
2060
+ RETURN_VAL = SimRegArg("v0", 4)
2061
+ OVERFLOW_RETURN_VAL = SimRegArg("v1", 4)
2062
+ ARCH = archinfo.ArchMIPS32
2063
+
2064
+ # http://math-atlas.sourceforge.net/devel/assembly/mipsabi32.pdf Section 3-17
2065
+ def next_arg(self, session, arg_type):
2066
+ if isinstance(arg_type, (SimTypeArray, SimTypeFixedSizeArray)): # hack
2067
+ arg_type = SimTypePointer(arg_type.elem_type).with_arch(self.arch)
2068
+ state = session.getstate()
2069
+ classification = self._classify(arg_type)
2070
+ try:
2071
+ mapped_classes = []
2072
+ can_use_fp = True
2073
+ for idx, cls in enumerate(classification):
2074
+ if cls == "DOUBLEP":
2075
+ mapped_classes.append(next(session.fp_iter))
2076
+ mapped_classes.append(next(session.fp_iter))
2077
+ if isinstance(arg_type, SimStruct) and idx < 2 and can_use_fp:
2078
+ next(session.fp_iter) # consume next two fp regs since it's double precision
2079
+ next(session.fp_iter)
2080
+ elif cls == "NO_CLASS":
2081
+ raise NotImplementedError("Bug. Report to @rhelmot")
2082
+ elif cls == "MEMORY":
2083
+ mapped_classes.append(next(session.both_iter))
2084
+ can_use_fp = False
2085
+ elif cls == "INTEGER":
2086
+ mapped_classes.append(next(session.int_iter))
2087
+ can_use_fp = False
2088
+ elif cls == "SINGLEP":
2089
+ if isinstance(arg_type, SimStruct):
2090
+ if idx < 2 and can_use_fp:
2091
+ mapped_classes.append(next(session.fp_iter))
2092
+ next(session.int_iter) # Need to take up the arg slot
2093
+ else:
2094
+ mapped_classes.append(next(session.both_iter))
2095
+ else:
2096
+ mapped_classes.append(next(session.fp_iter))
2097
+ next(session.fp_iter) # consume f13 or f15 since it's single precision
2098
+
2099
+ else:
2100
+ raise NotImplementedError("Bug. Report to @rhelmot")
2101
+ except StopIteration:
2102
+ session.setstate(state)
2103
+ mapped_classes = [next(session.both_iter) for _ in classification]
2104
+
2105
+ return refine_locs_with_struct_type(self.arch, mapped_classes, arg_type)
2106
+
2107
+ def _classify(self, ty, chunksize=None):
2108
+ if chunksize is None:
2109
+ chunksize = self.arch.bytes
2110
+ # treat BOT as INTEGER
2111
+ nchunks = 1 if isinstance(ty, SimTypeBottom) else (ty.size // self.arch.byte_width + chunksize - 1) // chunksize
2112
+ if isinstance(ty, (SimTypeInt, SimTypeChar, SimTypePointer, SimTypeNum, SimTypeBottom, SimTypeReference)):
2113
+ return ["INTEGER"] * nchunks
2114
+ if isinstance(ty, (SimTypeFloat,)):
2115
+ if ty.size == 64:
2116
+ return ["DOUBLEP"]
2117
+ if ty.size == 32:
2118
+ return ["SINGLEP"]
2119
+ return ["NO_CLASS"]
2120
+ if isinstance(ty, (SimStruct, SimTypeFixedSizeArray, SimUnion)):
2121
+ flattened = self._flatten(ty)
2122
+ if flattened is None:
2123
+ return ["MEMORY"] * nchunks
2124
+ result = ["NO_CLASS"] * nchunks
2125
+ for offset, subty_list in flattened.items():
2126
+ for subty in subty_list:
2127
+ # is the smaller chunk size necessary? Genuinely unsure
2128
+ subresult = self._classify(subty, chunksize=1)
2129
+ idx_start = offset // chunksize
2130
+ idx_end = (offset + (subty.size // self.arch.byte_width) - 1) // chunksize
2131
+ for i, idx in enumerate(range(idx_start, idx_end + 1)):
2132
+ subclass = subresult[i * chunksize]
2133
+ result[idx] = self._combine_classes(result[idx], subclass)
2134
+ return result
2135
+ raise NotImplementedError("Ummmmm... not sure what goes here. report bug to @rhelmot")
2136
+
2137
+ def _combine_classes(self, cls1, cls2):
2138
+ if cls1 == cls2:
2139
+ return cls1
2140
+ if cls1 == "NO_CLASS":
2141
+ return cls2
2142
+ if cls2 == "NO_CLASS":
2143
+ return cls1
2144
+ if cls1 == "MEMORY" or cls2 == "MEMORY":
2145
+ return "MEMORY"
2146
+ if cls1 == "INTEGER" or cls2 == "INTEGER":
2147
+ return "INTEGER"
2148
+ return "SSE"
2149
+
2150
+ def _flatten(self, ty) -> dict[int, list[SimType]] | None:
2151
+ result: dict[int, list[SimType]] = defaultdict(list)
2152
+ if isinstance(ty, SimStruct):
2153
+ if ty.packed:
2154
+ return None
2155
+ for field, subty in ty.fields.items():
2156
+ offset = ty.offsets[field]
2157
+ subresult = self._flatten(subty)
2158
+ if subresult is None:
2159
+ return None
2160
+ for suboffset, subsubty_list in subresult.items():
2161
+ result[offset + suboffset] += subsubty_list
2162
+ elif isinstance(ty, SimTypeFixedSizeArray):
2163
+ assert ty.length is not None and ty.elem_type.size is not None
2164
+ subresult = self._flatten(ty.elem_type)
2165
+ if subresult is None:
2166
+ return None
2167
+ for suboffset, subsubty_list in subresult.items():
2168
+ for idx in range(ty.length):
2169
+ # TODO I think we need an explicit stride field on array types
2170
+ result[idx * ty.elem_type.size // self.arch.byte_width + suboffset] += subsubty_list
2171
+ elif isinstance(ty, SimUnion):
2172
+ for subty in ty.members.values():
2173
+ subresult = self._flatten(subty)
2174
+ if subresult is None:
2175
+ return None
2176
+ for suboffset, subsubty_list in subresult.items():
2177
+ result[suboffset] += subsubty_list
2178
+ else:
2179
+ result[0].append(ty)
2180
+ return result
2181
+
2182
+
2183
+ class SimCCO32LinuxSyscall(SimCCSyscall):
2184
+ # TODO: Make sure all the information is correct
2185
+ ARG_REGS = ["a0", "a1", "a2", "a3"]
2186
+ FP_ARG_REGS = [] # TODO: ???
2187
+ RETURN_VAL = SimRegArg("v0", 4)
2188
+ RETURN_ADDR = SimRegArg("ip_at_syscall", 4)
2189
+ ARCH = archinfo.ArchMIPS32
2190
+
2191
+ ERROR_REG = SimRegArg("a3", 4)
2192
+ SYSCALL_ERRNO_START = -1133
2193
+
2194
+ @classmethod
2195
+ def _match(cls, arch, args, sp_delta): # pylint: disable=unused-argument
2196
+ # never appears anywhere except syscalls
2197
+ return False
2198
+
2199
+ @staticmethod
2200
+ def syscall_num(state):
2201
+ return state.regs.v0
2202
+
2203
+
2204
+ class SimCCN64(SimCC): # TODO: add n32
2205
+ ARG_REGS = ["a0", "a1", "a2", "a3", "a4", "a5", "a6", "a7"]
2206
+ CALLER_SAVED_REGS = ["t9", "gp"]
2207
+ FP_ARG_REGS = [] # TODO: ???
2208
+ STACKARG_SP_BUFF = 32
2209
+ RETURN_ADDR = SimRegArg("ra", 8)
2210
+ RETURN_VAL = SimRegArg("v0", 8)
2211
+ ARCH = archinfo.ArchMIPS64
2212
+
2213
+
2214
+ SimCCO64 = SimCCN64 # compatibility
2215
+
2216
+
2217
+ class SimCCN64LinuxSyscall(SimCCSyscall):
2218
+ ARG_REGS = ["a0", "a1", "a2", "a3", "a4", "a5", "a6", "a7"]
2219
+ FP_ARG_REGS = [] # TODO: ???
2220
+ RETURN_VAL = SimRegArg("v0", 8)
2221
+ RETURN_ADDR = SimRegArg("ip_at_syscall", 8)
2222
+ ARCH = archinfo.ArchMIPS64
2223
+
2224
+ ERROR_REG = SimRegArg("a3", 8)
2225
+ SYSCALL_ERRNO_START = -1133
2226
+
2227
+ @classmethod
2228
+ def _match(cls, arch, args, sp_delta): # pylint: disable=unused-argument
2229
+ # never appears anywhere except syscalls
2230
+ return False
2231
+
2232
+ @staticmethod
2233
+ def syscall_num(state):
2234
+ return state.regs.v0
2235
+
2236
+
2237
+ class SimCCPowerPC(SimCC):
2238
+ ARG_REGS = ["r3", "r4", "r5", "r6", "r7", "r8", "r9", "r10"]
2239
+ FP_ARG_REGS = [] # TODO: ???
2240
+ STACKARG_SP_BUFF = 8
2241
+ RETURN_ADDR = SimRegArg("lr", 4)
2242
+ RETURN_VAL = SimRegArg("r3", 4)
2243
+ OVERFLOW_RETURN_VAL = SimRegArg("r4", 4)
2244
+ ARCH = archinfo.ArchPPC32
2245
+
2246
+
2247
+ class SimCCPowerPCLinuxSyscall(SimCCSyscall):
2248
+ # TODO: Make sure all the information is correct
2249
+ ARG_REGS = ["r3", "r4", "r5", "r6", "r7", "r8", "r9", "r10"]
2250
+ FP_ARG_REGS = []
2251
+ RETURN_VAL = SimRegArg("r3", 4)
2252
+ RETURN_ADDR = SimRegArg("ip_at_syscall", 4)
2253
+ ARCH = archinfo.ArchPPC32
2254
+
2255
+ ERROR_REG = SimRegArg("cr0_0", 1)
2256
+ SYSCALL_ERRNO_START = -515
2257
+
2258
+ @classmethod
2259
+ def _match(cls, arch, args, sp_delta): # pylint: disable=unused-argument
2260
+ # never appears anywhere except syscalls
2261
+ return False
2262
+
2263
+ @staticmethod
2264
+ def syscall_num(state):
2265
+ return state.regs.r0
2266
+
2267
+
2268
+ class SimCCPowerPC64(SimCC):
2269
+ ARG_REGS = ["r3", "r4", "r5", "r6", "r7", "r8", "r9", "r10"]
2270
+ FP_ARG_REGS = [] # TODO: ???
2271
+ STACKARG_SP_BUFF = 0x70
2272
+ RETURN_ADDR = SimRegArg("lr", 8)
2273
+ RETURN_VAL = SimRegArg("r3", 8)
2274
+ ARCH = archinfo.ArchPPC64
2275
+
2276
+
2277
+ class SimCCPowerPC64LinuxSyscall(SimCCSyscall):
2278
+ # TODO: Make sure all the information is correct
2279
+ ARG_REGS = ["r3", "r4", "r5", "r6", "r7", "r8", "r9", "r10"]
2280
+ FP_ARG_REGS = []
2281
+ RETURN_VAL = SimRegArg("r3", 8)
2282
+ RETURN_ADDR = SimRegArg("ip_at_syscall", 8)
2283
+ ARCH = archinfo.ArchPPC64
2284
+
2285
+ ERROR_REG = SimRegArg("cr0_0", 1)
2286
+ SYSCALL_ERRNO_START = -515
2287
+
2288
+ @classmethod
2289
+ def _match(cls, arch, args, sp_delta): # pylint: disable=unused-argument
2290
+ # never appears anywhere except syscalls
2291
+ return False
2292
+
2293
+ @staticmethod
2294
+ def syscall_num(state):
2295
+ return state.regs.r0
2296
+
2297
+
2298
+ class SimCCSoot(SimCC):
2299
+ ARCH = archinfo.ArchSoot
2300
+ ARG_REGS = []
2301
+
2302
+ def setup_callsite(self, state, ret_addr, args, prototype, stack_base=None, alloc_base=None, grow_like_stack=True):
2303
+ angr.engines.SootMixin.setup_callsite(state, args, ret_addr)
2304
+
2305
+ @staticmethod
2306
+ def guess_prototype(args, prototype=None):
2307
+ # uhhhhhhhhhhhhhhhh
2308
+ return None
2309
+
2310
+
2311
+ class SimCCUnknown(SimCC):
2312
+ """
2313
+ Represent an unknown calling convention.
2314
+ """
2315
+
2316
+ @staticmethod
2317
+ def _match(arch, args, sp_delta): # type:ignore # pylint: disable=unused-argument
2318
+ # It always returns True
2319
+ return True
2320
+
2321
+ def __repr__(self):
2322
+ return f"<SimCCUnknown - {self.arch.name}>"
2323
+
2324
+
2325
+ class SimCCS390X(SimCC):
2326
+ ARG_REGS = ["r2", "r3", "r4", "r5", "r6"]
2327
+ FP_ARG_REGS = ["f0", "f2", "f4", "f6"]
2328
+ STACKARG_SP_BUFF = 0xA0
2329
+ RETURN_ADDR = SimRegArg("r14", 8)
2330
+ RETURN_VAL = SimRegArg("r2", 8)
2331
+ ARCH = archinfo.ArchS390X
2332
+
2333
+
2334
+ class SimCCS390XLinuxSyscall(SimCCSyscall):
2335
+ ARG_REGS = ["r2", "r3", "r4", "r5", "r6", "r7"]
2336
+ FP_ARG_REGS = []
2337
+ RETURN_VAL = SimRegArg("r2", 8)
2338
+ RETURN_ADDR = SimRegArg("ip_at_syscall", 8)
2339
+ ARCH = archinfo.ArchS390X
2340
+
2341
+ @classmethod
2342
+ def _match(cls, arch, args, sp_delta): # pylint: disable=unused-argument
2343
+ # never appears anywhere except syscalls
2344
+ return False
2345
+
2346
+ @staticmethod
2347
+ def syscall_num(state):
2348
+ return state.regs.r1
2349
+
2350
+
2351
+ CC: dict[str, dict[str, list[type[SimCC]]]] = {
2352
+ "AMD64": {
2353
+ "default": [SimCCSystemVAMD64],
2354
+ "Linux": [SimCCSystemVAMD64],
2355
+ "Win32": [SimCCMicrosoftAMD64],
2356
+ },
2357
+ "X86": {
2358
+ "default": [SimCCCdecl],
2359
+ "Linux": [SimCCCdecl],
2360
+ "CGC": [SimCCCdecl],
2361
+ "Win32": [SimCCMicrosoftCdecl, SimCCMicrosoftFastcall, SimCCMicrosoftThiscall],
2362
+ },
2363
+ "ARMEL": {
2364
+ "default": [SimCCARM],
2365
+ "Linux": [SimCCARM],
2366
+ },
2367
+ "ARMHF": {
2368
+ "default": [SimCCARMHF, SimCCARM],
2369
+ "Linux": [SimCCARMHF, SimCCARM],
2370
+ },
2371
+ "ARMCortexM": {
2372
+ "default": [SimCCARMHF, SimCCARM],
2373
+ "Linux": [SimCCARMHF, SimCCARM],
2374
+ },
2375
+ "MIPS32": {
2376
+ "default": [SimCCO32],
2377
+ "Linux": [SimCCO32],
2378
+ },
2379
+ "MIPS64": {
2380
+ "default": [SimCCN64],
2381
+ "Linux": [SimCCN64],
2382
+ },
2383
+ "PPC32": {
2384
+ "default": [SimCCPowerPC],
2385
+ "Linux": [SimCCPowerPC],
2386
+ },
2387
+ "PPC64": {
2388
+ "default": [SimCCPowerPC64],
2389
+ "Linux": [SimCCPowerPC64],
2390
+ },
2391
+ "AARCH64": {
2392
+ "default": [SimCCAArch64],
2393
+ "Linux": [SimCCAArch64],
2394
+ },
2395
+ "S390X": {
2396
+ "default": [SimCCS390X],
2397
+ "Linux": [SimCCS390X],
2398
+ },
2399
+ }
2400
+
2401
+
2402
+ DEFAULT_CC: dict[str, dict[str, type[SimCC]]] = {
2403
+ "AMD64": {"Linux": SimCCSystemVAMD64, "Win32": SimCCMicrosoftAMD64},
2404
+ "X86": {"Linux": SimCCCdecl, "CGC": SimCCCdecl, "Win32": SimCCMicrosoftCdecl},
2405
+ "ARMEL": {"Linux": SimCCARM},
2406
+ "ARMHF": {"Linux": SimCCARMHF},
2407
+ "ARMCortexM": {"Linux": SimCCARMHF},
2408
+ "MIPS32": {"Linux": SimCCO32},
2409
+ "MIPS64": {"Linux": SimCCN64},
2410
+ "PPC32": {"Linux": SimCCPowerPC},
2411
+ "PPC64": {"Linux": SimCCPowerPC64},
2412
+ "AARCH64": {"Linux": SimCCAArch64},
2413
+ "Soot": {"Linux": SimCCSoot},
2414
+ "AVR8": {"Linux": SimCCUnknown},
2415
+ "MSP": {"Linux": SimCCUnknown},
2416
+ "S390X": {"Linux": SimCCS390X},
2417
+ }
2418
+
2419
+
2420
+ def register_default_cc(arch: str, cc: type[SimCC], platform: str = "Linux"):
2421
+ DEFAULT_CC[arch] = {platform: cc}
2422
+ if arch not in CC:
2423
+ CC[arch] = {}
2424
+ if platform not in CC[arch]:
2425
+ CC[arch][platform] = [cc]
2426
+ if platform != "default":
2427
+ CC[arch]["default"] = [cc]
2428
+ else:
2429
+ if cc not in CC[arch][platform]:
2430
+ CC[arch][platform].append(cc)
2431
+
2432
+
2433
+ ARCH_NAME_ALIASES = {
2434
+ "X86": ["x8632"],
2435
+ "AMD64": ["x86-64", "x86_64", "x8664"],
2436
+ "ARMEL": [],
2437
+ "ARMHF": [],
2438
+ "ARMCortexM": [],
2439
+ "AARCH64": ["arm64", "aarch64"],
2440
+ "MIPS32": [],
2441
+ "MIPS64": [],
2442
+ "PPC32": ["powerpc32"],
2443
+ "PPC64": ["powerpc64"],
2444
+ "Soot": [],
2445
+ "AVR8": ["avr8"],
2446
+ "MSP": [],
2447
+ "S390X": [],
2448
+ }
2449
+
2450
+ ALIAS_TO_ARCH_NAME = {}
2451
+ for k, vs in ARCH_NAME_ALIASES.items():
2452
+ for v in vs:
2453
+ ALIAS_TO_ARCH_NAME[v] = k
2454
+
2455
+
2456
+ def default_cc( # pylint:disable=unused-argument
2457
+ arch: str,
2458
+ platform: str | None = "Linux",
2459
+ language: str | None = None,
2460
+ syscall: bool = False,
2461
+ default: type[SimCC] | None = None,
2462
+ ) -> type[SimCC] | None:
2463
+ """
2464
+ Return the default calling convention for a given architecture, platform, and language combination.
2465
+
2466
+ :param arch: The architecture name.
2467
+ :param platform: The platform name (e.g., "Linux" or "Win32").
2468
+ :param language: The programming language name (e.g., "go").
2469
+ :param syscall: Return syscall convention (True), or normal calling convention (False, default).
2470
+ :param default: The default calling convention to return if nothing fits.
2471
+ :return: A default calling convention class if we can find one for the architecture, platform, and
2472
+ language combination, or the default if nothing fits.
2473
+ """
2474
+
2475
+ if platform is None:
2476
+ platform = "Linux"
2477
+
2478
+ cc_map = SYSCALL_CC if syscall else DEFAULT_CC
2479
+
2480
+ if arch in cc_map:
2481
+ if platform not in cc_map[arch]:
2482
+ if default is not None:
2483
+ return default
2484
+ if "Linux" in cc_map[arch]:
2485
+ return cc_map[arch]["Linux"]
2486
+ return cc_map[arch][platform]
2487
+
2488
+ alias = unify_arch_name(arch)
2489
+ if alias not in cc_map or platform not in cc_map[alias]:
2490
+ return default
2491
+ return cc_map[alias][platform]
2492
+
2493
+
2494
+ def unify_arch_name(arch: str) -> str:
2495
+ """
2496
+ Return the unified architecture name.
2497
+
2498
+ :param arch: The architecture name.
2499
+ :return: A unified architecture name.
2500
+ """
2501
+
2502
+ if ":" in arch:
2503
+ # Sleigh architecture names
2504
+ chunks = arch.lower().split(":")
2505
+ if len(chunks) >= 3:
2506
+ arch_base, _, bits = chunks[:3]
2507
+
2508
+ if arch_base in ALIAS_TO_ARCH_NAME:
2509
+ return ALIAS_TO_ARCH_NAME[arch_base]
2510
+
2511
+ base_with_bits = f"{arch_base}{bits}"
2512
+ if base_with_bits in ALIAS_TO_ARCH_NAME:
2513
+ return ALIAS_TO_ARCH_NAME[base_with_bits]
2514
+
2515
+ return arch
2516
+
2517
+
2518
+ SYSCALL_CC: dict[str, dict[str, type[SimCCSyscall]]] = {
2519
+ "X86": {
2520
+ "default": SimCCX86LinuxSyscall,
2521
+ "Linux": SimCCX86LinuxSyscall,
2522
+ "Win32": SimCCX86WindowsSyscall,
2523
+ "CGC": SimCCX86LinuxSyscall,
2524
+ },
2525
+ "AMD64": {
2526
+ "default": SimCCAMD64LinuxSyscall,
2527
+ "Linux": SimCCAMD64LinuxSyscall,
2528
+ "Win32": SimCCAMD64WindowsSyscall,
2529
+ },
2530
+ "ARMEL": {
2531
+ "default": SimCCARMLinuxSyscall,
2532
+ "Linux": SimCCARMLinuxSyscall,
2533
+ },
2534
+ "ARMCortexM": {
2535
+ # FIXME: TODO: This is wrong. Fill in with a real CC when we support CM syscalls
2536
+ "default": SimCCARMLinuxSyscall,
2537
+ },
2538
+ "ARMHF": {
2539
+ "default": SimCCARMLinuxSyscall,
2540
+ "Linux": SimCCARMLinuxSyscall,
2541
+ },
2542
+ "AARCH64": {
2543
+ "default": SimCCAArch64LinuxSyscall,
2544
+ "Linux": SimCCAArch64LinuxSyscall,
2545
+ },
2546
+ "MIPS32": {
2547
+ "default": SimCCO32LinuxSyscall,
2548
+ "Linux": SimCCO32LinuxSyscall,
2549
+ },
2550
+ "MIPS64": {
2551
+ "default": SimCCN64LinuxSyscall,
2552
+ "Linux": SimCCN64LinuxSyscall,
2553
+ },
2554
+ "PPC32": {
2555
+ "default": SimCCPowerPCLinuxSyscall,
2556
+ "Linux": SimCCPowerPCLinuxSyscall,
2557
+ },
2558
+ "PPC64": {
2559
+ "default": SimCCPowerPC64LinuxSyscall,
2560
+ "Linux": SimCCPowerPC64LinuxSyscall,
2561
+ },
2562
+ "S390X": {
2563
+ "default": SimCCS390XLinuxSyscall,
2564
+ "Linux": SimCCS390XLinuxSyscall,
2565
+ },
2566
+ "RISCV64": {
2567
+ "default": SimCCRISCV64LinuxSyscall,
2568
+ "Linux": SimCCRISCV64LinuxSyscall,
2569
+ },
2570
+ }
2571
+
2572
+
2573
+ def register_syscall_cc(arch, os, cc):
2574
+ if arch not in SYSCALL_CC:
2575
+ SYSCALL_CC[arch] = {}
2576
+ SYSCALL_CC[arch][os] = cc
2577
+
2578
+
2579
+ SyscallCC = SYSCALL_CC
2580
+ DefaultCC = DEFAULT_CC