contrast-agent 3.8.4

Sign up to get free protection for your applications and to get access to all the features.
Files changed (500) hide show
  1. checksums.yaml +7 -0
  2. data/.clang-format +5 -0
  3. data/.dockerignore +10 -0
  4. data/.gitignore +58 -0
  5. data/.gitmodules +6 -0
  6. data/.rspec +6 -0
  7. data/.simplecov +4 -0
  8. data/Gemfile +7 -0
  9. data/LICENSE.txt +12 -0
  10. data/Rakefile +15 -0
  11. data/exe/contrast_service +29 -0
  12. data/ext/build_funchook.rb +48 -0
  13. data/ext/cs__assess_active_record_named/cs__active_record_named.c +47 -0
  14. data/ext/cs__assess_active_record_named/cs__active_record_named.h +10 -0
  15. data/ext/cs__assess_active_record_named/extconf.rb +2 -0
  16. data/ext/cs__assess_array/cs__assess_array.c +38 -0
  17. data/ext/cs__assess_array/cs__assess_array.h +9 -0
  18. data/ext/cs__assess_array/extconf.rb +2 -0
  19. data/ext/cs__assess_basic_object/cs__assess_basic_object.c +50 -0
  20. data/ext/cs__assess_basic_object/cs__assess_basic_object.h +17 -0
  21. data/ext/cs__assess_basic_object/extconf.rb +2 -0
  22. data/ext/cs__assess_fiber_track/cs__assess_fiber_track.c +86 -0
  23. data/ext/cs__assess_fiber_track/cs__assess_fiber_track.h +34 -0
  24. data/ext/cs__assess_fiber_track/extconf.rb +2 -0
  25. data/ext/cs__assess_hash/cs__assess_hash.c +64 -0
  26. data/ext/cs__assess_hash/cs__assess_hash.h +24 -0
  27. data/ext/cs__assess_hash/extconf.rb +2 -0
  28. data/ext/cs__assess_kernel/cs__assess_kernel.c +36 -0
  29. data/ext/cs__assess_kernel/cs__assess_kernel.h +10 -0
  30. data/ext/cs__assess_kernel/extconf.rb +2 -0
  31. data/ext/cs__assess_marshal_module/cs__assess_marshal_module.c +47 -0
  32. data/ext/cs__assess_marshal_module/cs__assess_marshal_module.h +18 -0
  33. data/ext/cs__assess_marshal_module/extconf.rb +2 -0
  34. data/ext/cs__assess_module/cs__assess_module.c +78 -0
  35. data/ext/cs__assess_module/cs__assess_module.h +25 -0
  36. data/ext/cs__assess_module/extconf.rb +2 -0
  37. data/ext/cs__assess_regexp/cs__assess_regexp.c +48 -0
  38. data/ext/cs__assess_regexp/cs__assess_regexp.h +22 -0
  39. data/ext/cs__assess_regexp/extconf.rb +2 -0
  40. data/ext/cs__assess_regexp_track/cs__assess_regexp_track.c +63 -0
  41. data/ext/cs__assess_regexp_track/cs__assess_regexp_track.h +29 -0
  42. data/ext/cs__assess_regexp_track/extconf.rb +2 -0
  43. data/ext/cs__assess_string/cs__assess_string.c +38 -0
  44. data/ext/cs__assess_string/cs__assess_string.h +19 -0
  45. data/ext/cs__assess_string/extconf.rb +2 -0
  46. data/ext/cs__assess_string_interpolation26/cs__assess_string_interpolation26.c +31 -0
  47. data/ext/cs__assess_string_interpolation26/cs__assess_string_interpolation26.h +13 -0
  48. data/ext/cs__assess_string_interpolation26/extconf.rb +2 -0
  49. data/ext/cs__common/cs__common.c +60 -0
  50. data/ext/cs__common/cs__common.h +28 -0
  51. data/ext/cs__common/extconf.rb +20 -0
  52. data/ext/cs__contrast_patch/cs__contrast_patch.c +445 -0
  53. data/ext/cs__contrast_patch/cs__contrast_patch.h +196 -0
  54. data/ext/cs__contrast_patch/extconf.rb +2 -0
  55. data/ext/cs__protect_kernel/cs__protect_kernel.c +37 -0
  56. data/ext/cs__protect_kernel/cs__protect_kernel.h +11 -0
  57. data/ext/cs__protect_kernel/extconf.rb +2 -0
  58. data/ext/cs__scope/cs__scope.c +96 -0
  59. data/ext/cs__scope/cs__scope.h +33 -0
  60. data/ext/cs__scope/extconf.rb +2 -0
  61. data/ext/extconf_common.rb +49 -0
  62. data/funchook/LICENSE +360 -0
  63. data/funchook/Makefile +29 -0
  64. data/funchook/Makefile.in +29 -0
  65. data/funchook/README.md +121 -0
  66. data/funchook/appveyor.yml +42 -0
  67. data/funchook/autogen.sh +3 -0
  68. data/funchook/autom4te.cache/output.0 +4976 -0
  69. data/funchook/autom4te.cache/requests +78 -0
  70. data/funchook/autom4te.cache/traces.0 +364 -0
  71. data/funchook/config.guess +1530 -0
  72. data/funchook/config.log +490 -0
  73. data/funchook/config.status +1016 -0
  74. data/funchook/config.sub +1773 -0
  75. data/funchook/configure +4976 -0
  76. data/funchook/configure.ac +59 -0
  77. data/funchook/distorm/COPYING +26 -0
  78. data/funchook/distorm/MANIFEST +25 -0
  79. data/funchook/distorm/MANIFEST.in +4 -0
  80. data/funchook/distorm/README.md +12 -0
  81. data/funchook/distorm/disOps/disOps.py +795 -0
  82. data/funchook/distorm/disOps/x86db.py +404 -0
  83. data/funchook/distorm/disOps/x86header.py +247 -0
  84. data/funchook/distorm/disOps/x86sets.py +1664 -0
  85. data/funchook/distorm/examples/cs/TestdiStorm/Program.cs +79 -0
  86. data/funchook/distorm/examples/cs/TestdiStorm/Properties/AssemblyInfo.cs +36 -0
  87. data/funchook/distorm/examples/cs/TestdiStorm/TestdiStorm.csproj +69 -0
  88. data/funchook/distorm/examples/cs/distorm-net.sln +26 -0
  89. data/funchook/distorm/examples/cs/distorm-net/CodeInfo.cs +23 -0
  90. data/funchook/distorm/examples/cs/distorm-net/DecodedInst.cs +15 -0
  91. data/funchook/distorm/examples/cs/distorm-net/DecodedResult.cs +14 -0
  92. data/funchook/distorm/examples/cs/distorm-net/DecomposedInst.cs +36 -0
  93. data/funchook/distorm/examples/cs/distorm-net/DecomposedResult.cs +14 -0
  94. data/funchook/distorm/examples/cs/distorm-net/Opcodes.cs +1268 -0
  95. data/funchook/distorm/examples/cs/distorm-net/Opcodes.tt +37 -0
  96. data/funchook/distorm/examples/cs/distorm-net/Operand.cs +25 -0
  97. data/funchook/distorm/examples/cs/distorm-net/Properties/AssemblyInfo.cs +36 -0
  98. data/funchook/distorm/examples/cs/distorm-net/diStorm3.cs +411 -0
  99. data/funchook/distorm/examples/cs/distorm-net/distorm-net.csproj +80 -0
  100. data/funchook/distorm/examples/cs/readme +3 -0
  101. data/funchook/distorm/examples/ddk/README +48 -0
  102. data/funchook/distorm/examples/ddk/distorm.ini +11 -0
  103. data/funchook/distorm/examples/ddk/dummy.c +15 -0
  104. data/funchook/distorm/examples/ddk/main.c +91 -0
  105. data/funchook/distorm/examples/ddk/makefile +1 -0
  106. data/funchook/distorm/examples/ddk/sources +10 -0
  107. data/funchook/distorm/examples/java/Makefile +23 -0
  108. data/funchook/distorm/examples/java/distorm/src/Main.java +43 -0
  109. data/funchook/distorm/examples/java/distorm/src/diStorm3/CodeInfo.java +27 -0
  110. data/funchook/distorm/examples/java/distorm/src/diStorm3/DecodedInst.java +32 -0
  111. data/funchook/distorm/examples/java/distorm/src/diStorm3/DecodedResult.java +11 -0
  112. data/funchook/distorm/examples/java/distorm/src/diStorm3/DecomposedInst.java +89 -0
  113. data/funchook/distorm/examples/java/distorm/src/diStorm3/DecomposedResult.java +11 -0
  114. data/funchook/distorm/examples/java/distorm/src/diStorm3/OpcodeEnum.java +131 -0
  115. data/funchook/distorm/examples/java/distorm/src/diStorm3/Opcodes.java +1123 -0
  116. data/funchook/distorm/examples/java/distorm/src/diStorm3/Operand.java +24 -0
  117. data/funchook/distorm/examples/java/distorm/src/diStorm3/distorm3.java +41 -0
  118. data/funchook/distorm/examples/java/jdistorm.c +405 -0
  119. data/funchook/distorm/examples/java/jdistorm.h +40 -0
  120. data/funchook/distorm/examples/java/jdistorm.sln +20 -0
  121. data/funchook/distorm/examples/java/jdistorm.vcproj +208 -0
  122. data/funchook/distorm/examples/linux/Makefile +15 -0
  123. data/funchook/distorm/examples/linux/main.c +181 -0
  124. data/funchook/distorm/examples/tests/Makefile +15 -0
  125. data/funchook/distorm/examples/tests/main.cpp +42 -0
  126. data/funchook/distorm/examples/tests/main.py +66 -0
  127. data/funchook/distorm/examples/tests/test_distorm3.py +1672 -0
  128. data/funchook/distorm/examples/tests/tests.sln +20 -0
  129. data/funchook/distorm/examples/tests/tests.vcxproj +82 -0
  130. data/funchook/distorm/examples/tests/tests.vcxproj.filters +22 -0
  131. data/funchook/distorm/examples/win32/disasm.sln +25 -0
  132. data/funchook/distorm/examples/win32/disasm.vcxproj +201 -0
  133. data/funchook/distorm/examples/win32/disasm.vcxproj.filters +14 -0
  134. data/funchook/distorm/examples/win32/main.cpp +163 -0
  135. data/funchook/distorm/include/distorm.h +482 -0
  136. data/funchook/distorm/include/mnemonics.h +301 -0
  137. data/funchook/distorm/make/linux/Makefile +28 -0
  138. data/funchook/distorm/make/mac/Makefile +24 -0
  139. data/funchook/distorm/make/win32/cdistorm.vcxproj +239 -0
  140. data/funchook/distorm/make/win32/cdistorm.vcxproj.filters +80 -0
  141. data/funchook/distorm/make/win32/distorm.sln +25 -0
  142. data/funchook/distorm/make/win32/resource.h +14 -0
  143. data/funchook/distorm/make/win32/resource.rc +99 -0
  144. data/funchook/distorm/python/distorm3/__init__.py +957 -0
  145. data/funchook/distorm/python/distorm3/sample.py +51 -0
  146. data/funchook/distorm/setup.cfg +10 -0
  147. data/funchook/distorm/setup.py +266 -0
  148. data/funchook/distorm/src/config.h +169 -0
  149. data/funchook/distorm/src/decoder.c +641 -0
  150. data/funchook/distorm/src/decoder.h +33 -0
  151. data/funchook/distorm/src/distorm.c +413 -0
  152. data/funchook/distorm/src/instructions.c +597 -0
  153. data/funchook/distorm/src/instructions.h +463 -0
  154. data/funchook/distorm/src/insts.c +7939 -0
  155. data/funchook/distorm/src/insts.h +64 -0
  156. data/funchook/distorm/src/mnemonics.c +284 -0
  157. data/funchook/distorm/src/operands.c +1290 -0
  158. data/funchook/distorm/src/operands.h +28 -0
  159. data/funchook/distorm/src/prefix.c +368 -0
  160. data/funchook/distorm/src/prefix.h +64 -0
  161. data/funchook/distorm/src/textdefs.c +172 -0
  162. data/funchook/distorm/src/textdefs.h +57 -0
  163. data/funchook/distorm/src/wstring.c +47 -0
  164. data/funchook/distorm/src/wstring.h +35 -0
  165. data/funchook/distorm/src/x86defs.h +82 -0
  166. data/funchook/include/funchook.h +123 -0
  167. data/funchook/install-sh +527 -0
  168. data/funchook/src/Makefile +70 -0
  169. data/funchook/src/Makefile.in +70 -0
  170. data/funchook/src/__strerror.h +109 -0
  171. data/funchook/src/config.h +101 -0
  172. data/funchook/src/config.h.in +100 -0
  173. data/funchook/src/decoder.o +0 -0
  174. data/funchook/src/distorm.o +0 -0
  175. data/funchook/src/funchook.c +440 -0
  176. data/funchook/src/funchook.o +0 -0
  177. data/funchook/src/funchook_internal.h +155 -0
  178. data/funchook/src/funchook_io.c +182 -0
  179. data/funchook/src/funchook_io.h +64 -0
  180. data/funchook/src/funchook_io.o +0 -0
  181. data/funchook/src/funchook_syscall.S +134 -0
  182. data/funchook/src/funchook_syscall.o +0 -0
  183. data/funchook/src/funchook_unix.c +480 -0
  184. data/funchook/src/funchook_unix.o +0 -0
  185. data/funchook/src/funchook_windows.c +397 -0
  186. data/funchook/src/funchook_x86.c +622 -0
  187. data/funchook/src/funchook_x86.o +0 -0
  188. data/funchook/src/instructions.o +0 -0
  189. data/funchook/src/insts.o +0 -0
  190. data/funchook/src/libfunchook.so +0 -0
  191. data/funchook/src/mnemonics.o +0 -0
  192. data/funchook/src/operands.o +0 -0
  193. data/funchook/src/os_func.c +115 -0
  194. data/funchook/src/os_func.h +75 -0
  195. data/funchook/src/os_func.o +0 -0
  196. data/funchook/src/os_func_unix.c +94 -0
  197. data/funchook/src/os_func_unix.o +0 -0
  198. data/funchook/src/os_func_windows.c +32 -0
  199. data/funchook/src/prefix.o +0 -0
  200. data/funchook/src/printf_base.c +1688 -0
  201. data/funchook/src/printf_base.h +46 -0
  202. data/funchook/src/printf_base.o +0 -0
  203. data/funchook/src/textdefs.o +0 -0
  204. data/funchook/src/wstring.o +0 -0
  205. data/funchook/test/Makefile +43 -0
  206. data/funchook/test/Makefile.in +43 -0
  207. data/funchook/test/funchook_test +0 -0
  208. data/funchook/test/libfunchook_test.c +25 -0
  209. data/funchook/test/libfunchook_test.so +0 -0
  210. data/funchook/test/libfunchook_test2.c +18 -0
  211. data/funchook/test/suffix.list +600 -0
  212. data/funchook/test/test_main.c +430 -0
  213. data/funchook/test/test_main.o +0 -0
  214. data/funchook/test/x86_64_test.S +10 -0
  215. data/funchook/test/x86_64_test.o +0 -0
  216. data/funchook/test/x86_test.S +339 -0
  217. data/funchook/win32/config.h +1 -0
  218. data/funchook/win32/funchook.sln +52 -0
  219. data/funchook/win32/funchook.vcxproj +188 -0
  220. data/funchook/win32/funchook.vcxproj.filters +84 -0
  221. data/funchook/win32/funchook_test.vcxproj +170 -0
  222. data/funchook/win32/funchook_test.vcxproj.filters +22 -0
  223. data/funchook/win32/funchook_test_dll.vcxproj +184 -0
  224. data/funchook/win32/funchook_test_dll.vcxproj.filters +30 -0
  225. data/funchook/win32/funchook_test_exe.def +3 -0
  226. data/lib/contrast-agent.rb +8 -0
  227. data/lib/contrast.rb +57 -0
  228. data/lib/contrast/agent.rb +80 -0
  229. data/lib/contrast/agent/assess.rb +45 -0
  230. data/lib/contrast/agent/assess/adjusted_span.rb +25 -0
  231. data/lib/contrast/agent/assess/class_reverter.rb +82 -0
  232. data/lib/contrast/agent/assess/contrast_event.rb +398 -0
  233. data/lib/contrast/agent/assess/frozen_properties.rb +41 -0
  234. data/lib/contrast/agent/assess/insulator.rb +53 -0
  235. data/lib/contrast/agent/assess/policy/dynamic_source_factory.rb +78 -0
  236. data/lib/contrast/agent/assess/policy/patcher.rb +85 -0
  237. data/lib/contrast/agent/assess/policy/policy.rb +116 -0
  238. data/lib/contrast/agent/assess/policy/policy_node.rb +289 -0
  239. data/lib/contrast/agent/assess/policy/policy_scanner.rb +44 -0
  240. data/lib/contrast/agent/assess/policy/preshift.rb +94 -0
  241. data/lib/contrast/agent/assess/policy/propagation_method.rb +260 -0
  242. data/lib/contrast/agent/assess/policy/propagation_node.rb +127 -0
  243. data/lib/contrast/agent/assess/policy/propagator.rb +35 -0
  244. data/lib/contrast/agent/assess/policy/propagator/append.rb +54 -0
  245. data/lib/contrast/agent/assess/policy/propagator/base.rb +37 -0
  246. data/lib/contrast/agent/assess/policy/propagator/center.rb +73 -0
  247. data/lib/contrast/agent/assess/policy/propagator/custom.rb +36 -0
  248. data/lib/contrast/agent/assess/policy/propagator/database_write.rb +62 -0
  249. data/lib/contrast/agent/assess/policy/propagator/insert.rb +55 -0
  250. data/lib/contrast/agent/assess/policy/propagator/keep.rb +26 -0
  251. data/lib/contrast/agent/assess/policy/propagator/next.rb +42 -0
  252. data/lib/contrast/agent/assess/policy/propagator/prepend.rb +50 -0
  253. data/lib/contrast/agent/assess/policy/propagator/remove.rb +76 -0
  254. data/lib/contrast/agent/assess/policy/propagator/replace.rb +27 -0
  255. data/lib/contrast/agent/assess/policy/propagator/reverse.rb +38 -0
  256. data/lib/contrast/agent/assess/policy/propagator/select.rb +86 -0
  257. data/lib/contrast/agent/assess/policy/propagator/splat.rb +60 -0
  258. data/lib/contrast/agent/assess/policy/propagator/split.rb +49 -0
  259. data/lib/contrast/agent/assess/policy/propagator/substitution.rb +169 -0
  260. data/lib/contrast/agent/assess/policy/propagator/trim.rb +81 -0
  261. data/lib/contrast/agent/assess/policy/rewriter_patch.rb +79 -0
  262. data/lib/contrast/agent/assess/policy/source_method.rb +209 -0
  263. data/lib/contrast/agent/assess/policy/source_node.rb +62 -0
  264. data/lib/contrast/agent/assess/policy/trigger_method.rb +209 -0
  265. data/lib/contrast/agent/assess/policy/trigger_node.rb +198 -0
  266. data/lib/contrast/agent/assess/policy/trigger_validation/ssrf_validator.rb +77 -0
  267. data/lib/contrast/agent/assess/policy/trigger_validation/trigger_validation.rb +31 -0
  268. data/lib/contrast/agent/assess/policy/trigger_validation/xss_validator.rb +40 -0
  269. data/lib/contrast/agent/assess/properties.rb +392 -0
  270. data/lib/contrast/agent/assess/rule.rb +18 -0
  271. data/lib/contrast/agent/assess/rule/base.rb +72 -0
  272. data/lib/contrast/agent/assess/rule/csrf.rb +66 -0
  273. data/lib/contrast/agent/assess/rule/csrf/csrf_action.rb +28 -0
  274. data/lib/contrast/agent/assess/rule/csrf/csrf_applicator.rb +69 -0
  275. data/lib/contrast/agent/assess/rule/csrf/csrf_watcher.rb +132 -0
  276. data/lib/contrast/agent/assess/rule/provider.rb +21 -0
  277. data/lib/contrast/agent/assess/rule/provider/hardcoded_key.rb +62 -0
  278. data/lib/contrast/agent/assess/rule/provider/hardcoded_password.rb +73 -0
  279. data/lib/contrast/agent/assess/rule/provider/hardcoded_value_rule.rb +121 -0
  280. data/lib/contrast/agent/assess/rule/redos.rb +68 -0
  281. data/lib/contrast/agent/assess/rule/response_scanning_rule.rb +47 -0
  282. data/lib/contrast/agent/assess/rule/response_watcher.rb +36 -0
  283. data/lib/contrast/agent/assess/rule/watcher.rb +36 -0
  284. data/lib/contrast/agent/assess/tag.rb +151 -0
  285. data/lib/contrast/agent/at_exit_hook.rb +33 -0
  286. data/lib/contrast/agent/class_reopener.rb +195 -0
  287. data/lib/contrast/agent/deadzone/policy/deadzone_node.rb +26 -0
  288. data/lib/contrast/agent/deadzone/policy/policy.rb +57 -0
  289. data/lib/contrast/agent/disable_reaction.rb +24 -0
  290. data/lib/contrast/agent/exclusion_matcher.rb +190 -0
  291. data/lib/contrast/agent/feature_state.rb +379 -0
  292. data/lib/contrast/agent/inventory/policy/policy.rb +32 -0
  293. data/lib/contrast/agent/inventory/policy/trigger_node.rb +22 -0
  294. data/lib/contrast/agent/logger_manager.rb +116 -0
  295. data/lib/contrast/agent/middleware.rb +352 -0
  296. data/lib/contrast/agent/module_data.rb +16 -0
  297. data/lib/contrast/agent/patching/policy/after_load_patch.rb +37 -0
  298. data/lib/contrast/agent/patching/policy/after_load_patcher.rb +58 -0
  299. data/lib/contrast/agent/patching/policy/method_policy.rb +94 -0
  300. data/lib/contrast/agent/patching/policy/module_policy.rb +116 -0
  301. data/lib/contrast/agent/patching/policy/patch.rb +312 -0
  302. data/lib/contrast/agent/patching/policy/patch_status.rb +192 -0
  303. data/lib/contrast/agent/patching/policy/patcher.rb +310 -0
  304. data/lib/contrast/agent/patching/policy/policy.rb +138 -0
  305. data/lib/contrast/agent/patching/policy/policy_node.rb +80 -0
  306. data/lib/contrast/agent/patching/policy/policy_unpatcher.rb +28 -0
  307. data/lib/contrast/agent/patching/policy/trigger_node.rb +81 -0
  308. data/lib/contrast/agent/protect/policy/policy.rb +37 -0
  309. data/lib/contrast/agent/protect/policy/trigger_node.rb +23 -0
  310. data/lib/contrast/agent/protect/rule.rb +58 -0
  311. data/lib/contrast/agent/protect/rule/base.rb +300 -0
  312. data/lib/contrast/agent/protect/rule/base_service.rb +88 -0
  313. data/lib/contrast/agent/protect/rule/cmd_injection.rb +156 -0
  314. data/lib/contrast/agent/protect/rule/csrf.rb +118 -0
  315. data/lib/contrast/agent/protect/rule/csrf/csrf_evaluator.rb +103 -0
  316. data/lib/contrast/agent/protect/rule/csrf/csrf_token_injector.rb +85 -0
  317. data/lib/contrast/agent/protect/rule/default_scanner.rb +300 -0
  318. data/lib/contrast/agent/protect/rule/deserialization.rb +193 -0
  319. data/lib/contrast/agent/protect/rule/http_method_tampering.rb +80 -0
  320. data/lib/contrast/agent/protect/rule/no_sqli.rb +101 -0
  321. data/lib/contrast/agent/protect/rule/no_sqli/mongo_no_sql_scanner.rb +40 -0
  322. data/lib/contrast/agent/protect/rule/path_traversal.rb +143 -0
  323. data/lib/contrast/agent/protect/rule/sqli.rb +101 -0
  324. data/lib/contrast/agent/protect/rule/sqli/default_sql_scanner.rb +16 -0
  325. data/lib/contrast/agent/protect/rule/sqli/mysql_sql_scanner.rb +38 -0
  326. data/lib/contrast/agent/protect/rule/sqli/postgres_sql_scanner.rb +22 -0
  327. data/lib/contrast/agent/protect/rule/sqli/sqlite_sql_scanner.rb +19 -0
  328. data/lib/contrast/agent/protect/rule/unsafe_file_upload.rb +20 -0
  329. data/lib/contrast/agent/protect/rule/xss.rb +24 -0
  330. data/lib/contrast/agent/protect/rule/xxe.rb +120 -0
  331. data/lib/contrast/agent/protect/rule/xxe/entity_wrapper.rb +82 -0
  332. data/lib/contrast/agent/railtie.rb +30 -0
  333. data/lib/contrast/agent/reaction_processor.rb +47 -0
  334. data/lib/contrast/agent/request.rb +493 -0
  335. data/lib/contrast/agent/request_context.rb +225 -0
  336. data/lib/contrast/agent/require_state.rb +61 -0
  337. data/lib/contrast/agent/response.rb +215 -0
  338. data/lib/contrast/agent/rewriter.rb +244 -0
  339. data/lib/contrast/agent/scope.rb +28 -0
  340. data/lib/contrast/agent/service_heartbeat.rb +37 -0
  341. data/lib/contrast/agent/settings_state.rb +148 -0
  342. data/lib/contrast/agent/socket_client.rb +125 -0
  343. data/lib/contrast/agent/thread.rb +26 -0
  344. data/lib/contrast/agent/tracepoint_hook.rb +51 -0
  345. data/lib/contrast/agent/version.rb +8 -0
  346. data/lib/contrast/api.rb +17 -0
  347. data/lib/contrast/api/.gitkeep +0 -0
  348. data/lib/contrast/api/connection_status.rb +49 -0
  349. data/lib/contrast/api/socket.rb +43 -0
  350. data/lib/contrast/api/speedracer.rb +206 -0
  351. data/lib/contrast/api/tcp_socket.rb +31 -0
  352. data/lib/contrast/api/unix_socket.rb +25 -0
  353. data/lib/contrast/common_agent_configuration.rb +86 -0
  354. data/lib/contrast/components/agent.rb +85 -0
  355. data/lib/contrast/components/app_context.rb +188 -0
  356. data/lib/contrast/components/assess.rb +67 -0
  357. data/lib/contrast/components/config.rb +135 -0
  358. data/lib/contrast/components/contrast_service.rb +113 -0
  359. data/lib/contrast/components/heap_dump.rb +34 -0
  360. data/lib/contrast/components/interface.rb +178 -0
  361. data/lib/contrast/components/inventory.rb +23 -0
  362. data/lib/contrast/components/logger.rb +92 -0
  363. data/lib/contrast/components/protect.rb +38 -0
  364. data/lib/contrast/components/sampling.rb +41 -0
  365. data/lib/contrast/components/scope.rb +106 -0
  366. data/lib/contrast/components/settings.rb +140 -0
  367. data/lib/contrast/config.rb +33 -0
  368. data/lib/contrast/config/agent_configuration.rb +24 -0
  369. data/lib/contrast/config/application_configuration.rb +27 -0
  370. data/lib/contrast/config/assess_configuration.rb +22 -0
  371. data/lib/contrast/config/assess_rules_configuration.rb +18 -0
  372. data/lib/contrast/config/base_configuration.rb +105 -0
  373. data/lib/contrast/config/default_value.rb +16 -0
  374. data/lib/contrast/config/exception_configuration.rb +21 -0
  375. data/lib/contrast/config/heap_dump_configuration.rb +23 -0
  376. data/lib/contrast/config/inventory_configuration.rb +20 -0
  377. data/lib/contrast/config/logger_configuration.rb +20 -0
  378. data/lib/contrast/config/protect_configuration.rb +20 -0
  379. data/lib/contrast/config/protect_rule_configuration.rb +37 -0
  380. data/lib/contrast/config/protect_rules_configuration.rb +30 -0
  381. data/lib/contrast/config/root_configuration.rb +26 -0
  382. data/lib/contrast/config/ruby_configuration.rb +39 -0
  383. data/lib/contrast/config/sampling_configuration.rb +22 -0
  384. data/lib/contrast/config/server_configuration.rb +23 -0
  385. data/lib/contrast/config/service_configuration.rb +22 -0
  386. data/lib/contrast/configuration.rb +214 -0
  387. data/lib/contrast/core_extensions/assess.rb +51 -0
  388. data/lib/contrast/core_extensions/assess/array.rb +58 -0
  389. data/lib/contrast/core_extensions/assess/assess_extension.rb +145 -0
  390. data/lib/contrast/core_extensions/assess/basic_object.rb +15 -0
  391. data/lib/contrast/core_extensions/assess/erb.rb +42 -0
  392. data/lib/contrast/core_extensions/assess/exec_trigger.rb +48 -0
  393. data/lib/contrast/core_extensions/assess/fiber.rb +125 -0
  394. data/lib/contrast/core_extensions/assess/hash.rb +22 -0
  395. data/lib/contrast/core_extensions/assess/kernel.rb +95 -0
  396. data/lib/contrast/core_extensions/assess/module.rb +14 -0
  397. data/lib/contrast/core_extensions/assess/regexp.rb +206 -0
  398. data/lib/contrast/core_extensions/assess/string.rb +75 -0
  399. data/lib/contrast/core_extensions/assess/tilt_template_trigger.rb +73 -0
  400. data/lib/contrast/core_extensions/delegator.rb +14 -0
  401. data/lib/contrast/core_extensions/eval_trigger.rb +52 -0
  402. data/lib/contrast/core_extensions/inventory.rb +22 -0
  403. data/lib/contrast/core_extensions/inventory/datastores.rb +37 -0
  404. data/lib/contrast/core_extensions/module.rb +42 -0
  405. data/lib/contrast/core_extensions/object.rb +27 -0
  406. data/lib/contrast/core_extensions/protect.rb +20 -0
  407. data/lib/contrast/core_extensions/protect/applies_command_injection_rule.rb +70 -0
  408. data/lib/contrast/core_extensions/protect/applies_deserialization_rule.rb +58 -0
  409. data/lib/contrast/core_extensions/protect/applies_no_sqli_rule.rb +81 -0
  410. data/lib/contrast/core_extensions/protect/applies_path_traversal_rule.rb +119 -0
  411. data/lib/contrast/core_extensions/protect/applies_sqli_rule.rb +63 -0
  412. data/lib/contrast/core_extensions/protect/applies_xxe_rule.rb +141 -0
  413. data/lib/contrast/core_extensions/protect/kernel.rb +30 -0
  414. data/lib/contrast/core_extensions/protect/psych.rb +7 -0
  415. data/lib/contrast/core_extensions/thread.rb +31 -0
  416. data/lib/contrast/internal_exception.rb +8 -0
  417. data/lib/contrast/rails_extensions/assess/action_controller_inheritance.rb +48 -0
  418. data/lib/contrast/rails_extensions/assess/active_record.rb +32 -0
  419. data/lib/contrast/rails_extensions/assess/active_record_named.rb +61 -0
  420. data/lib/contrast/rails_extensions/assess/configuration.rb +26 -0
  421. data/lib/contrast/rails_extensions/buffer.rb +30 -0
  422. data/lib/contrast/rails_extensions/rack.rb +45 -0
  423. data/lib/contrast/security_exception.rb +14 -0
  424. data/lib/contrast/sinatra_extensions/assess/cookie.rb +26 -0
  425. data/lib/contrast/sinatra_extensions/inventory/sinatra_base.rb +59 -0
  426. data/lib/contrast/tasks/service.rb +95 -0
  427. data/lib/contrast/utils/assess/sampling_util.rb +96 -0
  428. data/lib/contrast/utils/assess/tracking_util.rb +39 -0
  429. data/lib/contrast/utils/boolean_util.rb +33 -0
  430. data/lib/contrast/utils/cache.rb +69 -0
  431. data/lib/contrast/utils/class_util.rb +58 -0
  432. data/lib/contrast/utils/comment_range.rb +19 -0
  433. data/lib/contrast/utils/data_store_util.rb +23 -0
  434. data/lib/contrast/utils/duck_utils.rb +58 -0
  435. data/lib/contrast/utils/env_configuration_item.rb +52 -0
  436. data/lib/contrast/utils/environment_util.rb +152 -0
  437. data/lib/contrast/utils/freeze_util.rb +36 -0
  438. data/lib/contrast/utils/gemfile_reader.rb +191 -0
  439. data/lib/contrast/utils/hash_digest.rb +148 -0
  440. data/lib/contrast/utils/heap_dump_util.rb +113 -0
  441. data/lib/contrast/utils/invalid_configuration_util.rb +88 -0
  442. data/lib/contrast/utils/inventory_util.rb +126 -0
  443. data/lib/contrast/utils/io_util.rb +61 -0
  444. data/lib/contrast/utils/object_share.rb +117 -0
  445. data/lib/contrast/utils/operating_environment.rb +38 -0
  446. data/lib/contrast/utils/os.rb +49 -0
  447. data/lib/contrast/utils/path_util.rb +151 -0
  448. data/lib/contrast/utils/performs_logging.rb +152 -0
  449. data/lib/contrast/utils/preflight_util.rb +13 -0
  450. data/lib/contrast/utils/prevent_serialization.rb +52 -0
  451. data/lib/contrast/utils/rack_assess_session_cookie.rb +104 -0
  452. data/lib/contrast/utils/rails_assess_configuration.rb +95 -0
  453. data/lib/contrast/utils/random_util.rb +22 -0
  454. data/lib/contrast/utils/resource_loader.rb +23 -0
  455. data/lib/contrast/utils/ruby_ast_rewriter.rb +74 -0
  456. data/lib/contrast/utils/scope_util.rb +99 -0
  457. data/lib/contrast/utils/service_response_util.rb +116 -0
  458. data/lib/contrast/utils/service_sender_util.rb +98 -0
  459. data/lib/contrast/utils/sha256_builder.rb +69 -0
  460. data/lib/contrast/utils/sinatra_helper.rb +49 -0
  461. data/lib/contrast/utils/stack_trace_utils.rb +209 -0
  462. data/lib/contrast/utils/string_utils.rb +72 -0
  463. data/lib/contrast/utils/tag_util.rb +139 -0
  464. data/lib/contrast/utils/thread_tracker.rb +54 -0
  465. data/lib/contrast/utils/timer.rb +78 -0
  466. data/resources/assess/policy.json +1673 -0
  467. data/resources/csrf/inject.js +44 -0
  468. data/resources/deadzone/policy.json +55 -0
  469. data/resources/factory-bot-spec/spec_helper.rb +30 -0
  470. data/resources/inventory/policy.json +110 -0
  471. data/resources/protect/policy.json +417 -0
  472. data/resources/rubocops/kernel/catch_cop.rb +37 -0
  473. data/resources/rubocops/kernel/require_cop.rb +37 -0
  474. data/resources/rubocops/kernel/require_relative_cop.rb +33 -0
  475. data/resources/rubocops/module/autoload_cop.rb +37 -0
  476. data/resources/rubocops/module/const_defined_cop.rb +37 -0
  477. data/resources/rubocops/module/const_get_cop.rb +37 -0
  478. data/resources/rubocops/module/const_set_cop.rb +37 -0
  479. data/resources/rubocops/module/constants_cop.rb +37 -0
  480. data/resources/rubocops/module/name_cop.rb +37 -0
  481. data/resources/rubocops/object/class_cop.rb +37 -0
  482. data/resources/rubocops/object/freeze_cop.rb +37 -0
  483. data/resources/rubocops/object/frozen_cop.rb +37 -0
  484. data/resources/rubocops/object/is_a_cop.rb +37 -0
  485. data/resources/rubocops/object/method_cop.rb +37 -0
  486. data/resources/rubocops/object/respond_to_cop.rb +37 -0
  487. data/resources/rubocops/object/singleton_class_cop.rb +37 -0
  488. data/resources/rubocops/regexp/spelling_cop.rb +44 -0
  489. data/resources/rubocops/thread/new_cop.rb +39 -0
  490. data/resources/ruby-spec/ancestors_spec.rb +70 -0
  491. data/resources/ruby-spec/modulo_spec.rb +831 -0
  492. data/resources/ruby-spec/parameters_spec.rb +261 -0
  493. data/resources/ruby-spec/ruby_spec_spec_helper.rb +35 -0
  494. data/resources/test_marker.txt +1 -0
  495. data/ruby-agent.gemspec +129 -0
  496. data/service_executables/.gitkeep +0 -0
  497. data/service_executables/VERSION +1 -0
  498. data/service_executables/linux/contrast-service +0 -0
  499. data/service_executables/mac/contrast-service +0 -0
  500. metadata +945 -0
@@ -0,0 +1,641 @@
1
+ /*
2
+ decoder.c
3
+
4
+ diStorm3 - Powerful disassembler for X86/AMD64
5
+ http://ragestorm.net/distorm/
6
+ distorm at gmail dot com
7
+ Copyright (C) 2003-2016 Gil Dabah
8
+ This library is licensed under the BSD license. See the file COPYING.
9
+ */
10
+
11
+
12
+ #include "decoder.h"
13
+ #include "instructions.h"
14
+ #include "insts.h"
15
+ #include "prefix.h"
16
+ #include "x86defs.h"
17
+ #include "operands.h"
18
+ #include "insts.h"
19
+ #include "../include/mnemonics.h"
20
+
21
+
22
+ /* Instruction Prefixes - Opcode - ModR/M - SIB - Displacement - Immediate */
23
+
24
+ static _DecodeType decode_get_effective_addr_size(_DecodeType dt, _iflags decodedPrefixes)
25
+ {
26
+ /*
27
+ * This table is to map from the current decoding mode to an effective address size:
28
+ * Decode16 -> Decode32
29
+ * Decode32 -> Decode16
30
+ * Decode64 -> Decode32
31
+ */
32
+ static _DecodeType AddrSizeTable[] = {Decode32Bits, Decode16Bits, Decode32Bits};
33
+
34
+ /* Switch to non default mode if prefix exists, only for ADDRESS SIZE. */
35
+ if (decodedPrefixes & INST_PRE_ADDR_SIZE) dt = AddrSizeTable[dt];
36
+ return dt;
37
+ }
38
+
39
+ static _DecodeType decode_get_effective_op_size(_DecodeType dt, _iflags decodedPrefixes, unsigned int rex, _iflags instFlags)
40
+ {
41
+ /*
42
+ * This table is to map from the current decoding mode to an effective operand size:
43
+ * Decode16 -> Decode32
44
+ * Decode32 -> Decode16
45
+ * Decode64 -> Decode16
46
+ * Not that in 64bits it's a bit more complicated, because of REX and promoted instructions.
47
+ */
48
+ static _DecodeType OpSizeTable[] = {Decode32Bits, Decode16Bits, Decode16Bits};
49
+
50
+ if (decodedPrefixes & INST_PRE_OP_SIZE) return OpSizeTable[dt];
51
+
52
+ if (dt == Decode64Bits) {
53
+ /*
54
+ * REX Prefix toggles data size to 64 bits.
55
+ * Operand size prefix toggles data size to 16.
56
+ * Default data size is 32 bits.
57
+ * Promoted instructions are 64 bits if they don't require a REX perfix.
58
+ * Non promoted instructions are 64 bits if the REX prefix exists.
59
+ */
60
+ /* Automatically promoted instructions have only INST_64BITS SET! */
61
+ if (((instFlags & (INST_64BITS | INST_PRE_REX)) == INST_64BITS) ||
62
+ /* Other instructions in 64 bits can be promoted only with a REX prefix. */
63
+ ((decodedPrefixes & INST_PRE_REX) && (rex & PREFIX_EX_W))) dt = Decode64Bits;
64
+ else dt = Decode32Bits; /* Default. */
65
+ }
66
+ return dt;
67
+ }
68
+
69
+ /* A helper macro to convert from diStorm's CPU flags to EFLAGS. */
70
+ #define CONVERT_FLAGS_TO_EFLAGS(dst, src, field) dst->field = ((src->field & D_COMPACT_SAME_FLAGS) | \
71
+ ((src->field & D_COMPACT_IF) ? D_IF : 0) | \
72
+ ((src->field & D_COMPACT_DF) ? D_DF : 0) | \
73
+ ((src->field & D_COMPACT_OF) ? D_OF : 0));
74
+
75
+ static _DecodeResult decode_inst(_CodeInfo* ci, _PrefixState* ps, _DInst* di)
76
+ {
77
+ /* Remember whether the instruction is privileged. */
78
+ uint16_t privilegedFlag = 0;
79
+
80
+ /* The ModR/M byte of the current instruction. */
81
+ unsigned int modrm = 0;
82
+
83
+ /* The REX/VEX prefix byte value. */
84
+ unsigned int vrex = ps->vrex;
85
+
86
+ /*
87
+ * Backup original input, so we can use it later if a problem occurs
88
+ * (like not enough data for decoding, invalid opcode, etc).
89
+ */
90
+ const uint8_t* startCode = ci->code;
91
+
92
+ /* Holds the info about the current found instruction. */
93
+ _InstInfo* ii = NULL;
94
+ _InstSharedInfo* isi = NULL;
95
+
96
+ /* Used only for special CMP instructions which have pseudo opcodes suffix. */
97
+ unsigned char cmpType = 0;
98
+
99
+ /*
100
+ * Indicates whether it is right to LOCK the instruction by decoding its first operand.
101
+ * Only then you know if it's ok to output the LOCK prefix's text...
102
+ * Used for first operand only.
103
+ */
104
+ int lockable = FALSE;
105
+
106
+ /* Calculate (and cache) effective-operand-size and effective-address-size only once. */
107
+ _DecodeType effOpSz, effAdrSz;
108
+ _iflags instFlags;
109
+
110
+ ii = inst_lookup(ci, ps);
111
+ if (ii == NULL) goto _Undecodable;
112
+ isi = &InstSharedInfoTable[ii->sharedIndex];
113
+ instFlags = FlagsTable[isi->flagsIndex];
114
+
115
+ /* Copy the privileged bit and remove it from the opcodeId field ASAP. */
116
+ privilegedFlag = ii->opcodeId & OPCODE_ID_PRIVILEGED;
117
+ ii->opcodeId &= ~OPCODE_ID_PRIVILEGED;
118
+
119
+ /*
120
+ * If both REX and OpSize are available we will have to disable the OpSize, because REX has precedence.
121
+ * However, only if REX.W is set !
122
+ * We had to wait with this test, since the operand size may be a mandatory prefix,
123
+ * and we know it only after prefetching.
124
+ */
125
+ if ((ps->prefixExtType == PET_REX) &&
126
+ (ps->decodedPrefixes & INST_PRE_OP_SIZE) &&
127
+ (!ps->isOpSizeMandatory) &&
128
+ (vrex & PREFIX_EX_W)) {
129
+ ps->decodedPrefixes &= ~INST_PRE_OP_SIZE;
130
+ prefixes_ignore(ps, PFXIDX_OP_SIZE);
131
+ }
132
+
133
+ /*
134
+ * In this point we know the instruction we are about to decode and its operands (unless, it's an invalid one!),
135
+ * so it makes it the right time for decoding-type suitability testing.
136
+ * Which practically means, don't allow 32 bits instructions in 16 bits decoding mode, but do allow
137
+ * 16 bits instructions in 32 bits decoding mode, of course...
138
+
139
+ * NOTE: Make sure the instruction set for 32 bits has explicitly this specific flag set.
140
+ * NOTE2: Make sure the instruction set for 64 bits has explicitly this specific flag set.
141
+
142
+ * If this is the case, drop what we've got and restart all over after DB'ing that byte.
143
+
144
+ * Though, don't drop an instruction which is also supported in 16 and 32 bits.
145
+ */
146
+
147
+ /* ! ! ! DISABLED UNTIL FURTHER NOTICE ! ! ! Decode16Bits CAN NOW DECODE 32 BITS INSTRUCTIONS ! ! !*/
148
+ /* if (ii && (dt == Decode16Bits) && (instFlags & INST_32BITS) && (~instFlags & INST_16BITS)) ii = NULL; */
149
+
150
+ /* Drop instructions which are invalid in 64 bits. */
151
+ if ((ci->dt == Decode64Bits) && (instFlags & INST_INVALID_64BITS)) goto _Undecodable;
152
+
153
+ /* If it's only a 64 bits instruction drop it in other decoding modes. */
154
+ if ((ci->dt != Decode64Bits) && (instFlags & INST_64BITS_FETCH)) goto _Undecodable;
155
+
156
+ if (instFlags & INST_MODRM_REQUIRED) {
157
+ /* If the ModRM byte is not part of the opcode, skip the last byte code, so code points now to ModRM. */
158
+ if (~instFlags & INST_MODRM_INCLUDED) {
159
+ ci->code++;
160
+ if (--ci->codeLen < 0) goto _Undecodable;
161
+ }
162
+ modrm = *ci->code;
163
+
164
+ /* Some instructions enforce that reg=000, so validate that. (Specifically EXTRQ). */
165
+ if ((instFlags & INST_FORCE_REG0) && (((modrm >> 3) & 7) != 0)) goto _Undecodable;
166
+ /* Some instructions enforce that mod=11, so validate that. */
167
+ if ((instFlags & INST_MODRR_REQUIRED) && (modrm < INST_DIVIDED_MODRM)) goto _Undecodable;
168
+ }
169
+
170
+ ci->code++; /* Skip the last byte we just read (either last opcode's byte code or a ModRM). */
171
+
172
+ /* Cache the effective operand-size and address-size. */
173
+ effOpSz = decode_get_effective_op_size(ci->dt, ps->decodedPrefixes, vrex, instFlags);
174
+ effAdrSz = decode_get_effective_addr_size(ci->dt, ps->decodedPrefixes);
175
+
176
+ memset(di, 0, sizeof(_DInst));
177
+ di->base = R_NONE;
178
+
179
+ /*
180
+ * Try to extract the next operand only if the latter exists.
181
+ * For example, if there is not first operand, no reason to try to extract second operand...
182
+ * I decided that a for-break is better for readability in this specific case than goto.
183
+ * Note: do-while with a constant 0 makes the compiler warning about it.
184
+ */
185
+ for (;;) {
186
+ if (isi->d != OT_NONE) {
187
+ if (!operands_extract(ci, di, ii, instFlags, (_OpType)isi->d, ONT_1, modrm, ps, effOpSz, effAdrSz, &lockable)) goto _Undecodable;
188
+ } else break;
189
+
190
+ if (isi->s != OT_NONE) {
191
+ if (!operands_extract(ci, di, ii, instFlags, (_OpType)isi->s, ONT_2, modrm, ps, effOpSz, effAdrSz, NULL)) goto _Undecodable;
192
+ } else break;
193
+
194
+ /* Use third operand, only if the flags says this InstInfo requires it. */
195
+ if (instFlags & INST_USE_OP3) {
196
+ if (!operands_extract(ci, di, ii, instFlags, (_OpType)((_InstInfoEx*)ii)->op3, ONT_3, modrm, ps, effOpSz, effAdrSz, NULL)) goto _Undecodable;
197
+ } else break;
198
+
199
+ /* Support for a fourth operand is added for (i.e:) INSERTQ instruction. */
200
+ if (instFlags & INST_USE_OP4) {
201
+ if (!operands_extract(ci, di, ii, instFlags, (_OpType)((_InstInfoEx*)ii)->op4, ONT_4, modrm, ps, effOpSz, effAdrSz, NULL)) goto _Undecodable;
202
+ }
203
+ break;
204
+ } /* Continue here after all operands were extracted. */
205
+
206
+ /* If it were a 3DNow! instruction, we will have to find the instruction itself now that we got its operands extracted. */
207
+ if (instFlags & INST_3DNOW_FETCH) {
208
+ ii = inst_lookup_3dnow(ci);
209
+ if (ii == NULL) goto _Undecodable;
210
+ isi = &InstSharedInfoTable[ii->sharedIndex];
211
+ instFlags = FlagsTable[isi->flagsIndex];
212
+ }
213
+
214
+ /* Check whether pseudo opcode is needed, only for CMP instructions: */
215
+ if (instFlags & INST_PSEUDO_OPCODE) {
216
+ if (--ci->codeLen < 0) goto _Undecodable;
217
+ cmpType = *ci->code;
218
+ ci->code++;
219
+ if (instFlags & INST_PRE_VEX) {
220
+ /* AVX Comparison type must be between 0 to 32, otherwise Reserved. */
221
+ if (cmpType >= INST_VCMP_MAX_RANGE) goto _Undecodable;
222
+ } else {
223
+ /* SSE Comparison type must be between 0 to 8, otherwise Reserved. */
224
+ if (cmpType >= INST_CMP_MAX_RANGE) goto _Undecodable;
225
+ }
226
+ }
227
+
228
+ /*
229
+ * There's a limit of 15 bytes on instruction length. The only way to violate
230
+ * this limit is by putting redundant prefixes before an instruction.
231
+ * start points to first prefix if any, otherwise it points to instruction first byte.
232
+ */
233
+ if ((ci->code - ps->start) > INST_MAXIMUM_SIZE) goto _Undecodable; /* Drop instruction. */
234
+
235
+ /*
236
+ * If we reached here the instruction was fully decoded, we located the instruction in the DB and extracted operands.
237
+ * Use the correct mnemonic according to the DT.
238
+ * If we are in 32 bits decoding mode it doesn't necessarily mean we will choose mnemonic2, alas,
239
+ * it means that if there is a mnemonic2, it will be used.
240
+ */
241
+
242
+ /* Start with prefix LOCK. */
243
+ if ((lockable == TRUE) && (instFlags & INST_PRE_LOCK)) {
244
+ ps->usedPrefixes |= INST_PRE_LOCK;
245
+ di->flags |= FLAG_LOCK;
246
+ } else if ((instFlags & INST_PRE_REPNZ) && (ps->decodedPrefixes & INST_PRE_REPNZ)) {
247
+ ps->usedPrefixes |= INST_PRE_REPNZ;
248
+ di->flags |= FLAG_REPNZ;
249
+ } else if ((instFlags & INST_PRE_REP) && (ps->decodedPrefixes & INST_PRE_REP)) {
250
+ ps->usedPrefixes |= INST_PRE_REP;
251
+ di->flags |= FLAG_REP;
252
+ }
253
+
254
+ /* If it's JeCXZ the ADDR_SIZE prefix affects them. */
255
+ if ((instFlags & (INST_PRE_ADDR_SIZE | INST_USE_EXMNEMONIC)) == (INST_PRE_ADDR_SIZE | INST_USE_EXMNEMONIC)) {
256
+ ps->usedPrefixes |= INST_PRE_ADDR_SIZE;
257
+ if (effAdrSz == Decode16Bits) di->opcode = ii->opcodeId;
258
+ else if (effAdrSz == Decode32Bits) di->opcode = ((_InstInfoEx*)ii)->opcodeId2;
259
+ /* Ignore REX.W in 64bits, JECXZ is promoted. */
260
+ else /* Decode64Bits */ di->opcode = ((_InstInfoEx*)ii)->opcodeId3;
261
+ }
262
+
263
+ /* LOOPxx instructions are also native instruction, but they are special case ones, ADDR_SIZE prefix affects them. */
264
+ else if ((instFlags & (INST_PRE_ADDR_SIZE | INST_NATIVE)) == (INST_PRE_ADDR_SIZE | INST_NATIVE)) {
265
+ di->opcode = ii->opcodeId;
266
+
267
+ /* If LOOPxx gets here from 64bits, it must be Decode32Bits because Address Size prefix is set. */
268
+ ps->usedPrefixes |= INST_PRE_ADDR_SIZE;
269
+ }
270
+ /*
271
+ * Note:
272
+ * If the instruction is prefixed by operand size we will format it in the non-default decoding mode!
273
+ * So there might be a situation that an instruction of 32 bit gets formatted in 16 bits decoding mode.
274
+ * Both ways should end up with a correct and expected formatting of the text.
275
+ */
276
+ else if (effOpSz == Decode16Bits) { /* Decode16Bits */
277
+
278
+ /* Set operand size. */
279
+ FLAG_SET_OPSIZE(di, Decode16Bits);
280
+
281
+ /*
282
+ * If it's a special instruction which has two mnemonics, then use the 16 bits one + update usedPrefixes.
283
+ * Note: use 16 bits mnemonic if that instruction supports 32 bit or 64 bit explicitly.
284
+ */
285
+ if ((instFlags & INST_USE_EXMNEMONIC) && ((instFlags & (INST_32BITS | INST_64BITS)) == 0)) ps->usedPrefixes |= INST_PRE_OP_SIZE;
286
+ di->opcode = ii->opcodeId;
287
+ } else if (effOpSz == Decode32Bits) { /* Decode32Bits */
288
+
289
+ /* Set operand size. */
290
+ FLAG_SET_OPSIZE(di, Decode32Bits);
291
+
292
+ /* Give a chance for special mnemonic instruction in 32 bits decoding. */
293
+ if (instFlags & INST_USE_EXMNEMONIC) {
294
+ ps->usedPrefixes |= INST_PRE_OP_SIZE;
295
+ /* Is it a special instruction which has another mnemonic for mod=11 ? */
296
+ if (instFlags & INST_MNEMONIC_MODRM_BASED) {
297
+ if (modrm >= INST_DIVIDED_MODRM) di->opcode = ii->opcodeId;
298
+ else di->opcode = ((_InstInfoEx*)ii)->opcodeId2;
299
+ } else di->opcode = ((_InstInfoEx*)ii)->opcodeId2;
300
+ } else di->opcode = ii->opcodeId;
301
+ } else { /* Decode64Bits, note that some instructions might be decoded in Decode32Bits above. */
302
+
303
+ /* Set operand size. */
304
+ FLAG_SET_OPSIZE(di, Decode64Bits);
305
+
306
+ if (instFlags & (INST_USE_EXMNEMONIC | INST_USE_EXMNEMONIC2)) {
307
+ /*
308
+ * We shouldn't be here for MODRM based mnemonics with a MOD=11,
309
+ * because they must not use REX (otherwise it will get to the wrong instruction which share same opcode).
310
+ * See XRSTOR and XSAVEOPT.
311
+ */
312
+ if ((instFlags & INST_MNEMONIC_MODRM_BASED) && (modrm >= INST_DIVIDED_MODRM)) goto _Undecodable;
313
+
314
+ /* Use third mnemonic, for 64 bits. */
315
+ if ((instFlags & INST_USE_EXMNEMONIC2) && (vrex & PREFIX_EX_W)) {
316
+ ps->usedPrefixes |= INST_PRE_REX;
317
+ di->opcode = ((_InstInfoEx*)ii)->opcodeId3;
318
+ } else di->opcode = ((_InstInfoEx*)ii)->opcodeId2; /* Use second mnemonic. */
319
+ } else di->opcode = ii->opcodeId;
320
+ }
321
+
322
+ /* If it's a native instruction use OpSize Prefix. */
323
+ if ((instFlags & INST_NATIVE) && (ps->decodedPrefixes & INST_PRE_OP_SIZE)) ps->usedPrefixes |= INST_PRE_OP_SIZE;
324
+
325
+ /* Check VEX mnemonics: */
326
+ if ((instFlags & INST_PRE_VEX) &&
327
+ (((((_InstInfoEx*)ii)->flagsEx & INST_MNEMONIC_VEXW_BASED) && (vrex & PREFIX_EX_W)) ||
328
+ ((((_InstInfoEx*)ii)->flagsEx & INST_MNEMONIC_VEXL_BASED) && (vrex & PREFIX_EX_L)))) {
329
+ di->opcode = ((_InstInfoEx*)ii)->opcodeId2;
330
+ }
331
+
332
+ /* Or is it a special CMP instruction which needs a pseudo opcode suffix ? */
333
+ if (instFlags & INST_PSEUDO_OPCODE) {
334
+ /*
335
+ * The opcodeId is the offset to the FIRST pseudo compare mnemonic,
336
+ * we will have to fix it so it offsets into the corrected mnemonic.
337
+ * Therefore, we use another table to fix the offset.
338
+ */
339
+ if (instFlags & INST_PRE_VEX) {
340
+ /* Use the AVX pseudo compare mnemonics table. */
341
+ di->opcode = ii->opcodeId + VCmpMnemonicOffsets[cmpType];
342
+ } else {
343
+ /* Use the SSE pseudo compare mnemonics table. */
344
+ di->opcode = ii->opcodeId + CmpMnemonicOffsets[cmpType];
345
+ }
346
+ }
347
+
348
+ /*
349
+ * Store the address size inside the flags.
350
+ * This is necessary for the caller to know the size of rSP when using PUSHA for example.
351
+ */
352
+ FLAG_SET_ADDRSIZE(di, effAdrSz);
353
+
354
+ /* Copy DST_WR flag. */
355
+ if (instFlags & INST_DST_WR) di->flags |= FLAG_DST_WR;
356
+
357
+ /* Set the unused prefixes mask. */
358
+ di->unusedPrefixesMask = prefixes_set_unused_mask(ps);
359
+
360
+ /* Fix privileged. Assumes the privilegedFlag is 0x8000 only. */
361
+ di->flags |= privilegedFlag;
362
+
363
+ /* Copy instruction meta. */
364
+ di->meta = isi->meta;
365
+ if (di->segment == 0) di->segment = R_NONE;
366
+
367
+ /* Take into account the O_MEM base register for the mask. */
368
+ if (di->base != R_NONE) di->usedRegistersMask |= _REGISTERTORCLASS[di->base];
369
+
370
+ /* Copy CPU affected flags. */
371
+ CONVERT_FLAGS_TO_EFLAGS(di, isi, modifiedFlagsMask);
372
+ CONVERT_FLAGS_TO_EFLAGS(di, isi, testedFlagsMask);
373
+ CONVERT_FLAGS_TO_EFLAGS(di, isi, undefinedFlagsMask);
374
+
375
+ /* Calculate the size of the instruction we've just decoded. */
376
+ di->size = (uint8_t)((ci->code - startCode) & 0xff);
377
+ return DECRES_SUCCESS;
378
+
379
+ _Undecodable: /* If the instruction couldn't be decoded for some reason, drop the first byte. */
380
+ memset(di, 0, sizeof(_DInst));
381
+ di->base = R_NONE;
382
+
383
+ di->size = 1;
384
+ /* Clean prefixes just in case... */
385
+ ps->usedPrefixes = 0;
386
+
387
+ /* Special case for WAIT instruction: If it's dropped, you have to return a valid instruction! */
388
+ if (*startCode == INST_WAIT_INDEX) {
389
+ di->opcode = I_WAIT;
390
+ META_SET_ISC(di, ISC_INTEGER);
391
+ return DECRES_SUCCESS;
392
+ }
393
+
394
+ /* Mark that we didn't manage to decode the instruction well, caller will drop it. */
395
+ return DECRES_INPUTERR;
396
+ }
397
+
398
+ /*
399
+ * decode_internal
400
+ *
401
+ * supportOldIntr - Since now we work with new structure instead of the old _DecodedInst, we are still interested in backward compatibility.
402
+ * So although, the array is now of type _DInst, we want to read it in jumps of the old array element's size.
403
+ * This is in order to save memory allocation for conversion between the new and the old structures.
404
+ * It really means we can do the conversion in-place now.
405
+ */
406
+ _DecodeResult decode_internal(_CodeInfo* _ci, int supportOldIntr, _DInst result[], unsigned int maxResultCount, unsigned int* usedInstructionsCount)
407
+ {
408
+ _PrefixState ps;
409
+ unsigned int prefixSize;
410
+ _CodeInfo ci;
411
+ unsigned int features;
412
+ unsigned int mfc;
413
+
414
+ _OffsetType codeOffset = _ci->codeOffset;
415
+ const uint8_t* code = _ci->code;
416
+ int codeLen = _ci->codeLen;
417
+
418
+ /*
419
+ * This is used for printing only, it is the real offset of where the whole instruction begins.
420
+ * We need this variable in addition to codeOffset, because prefixes might change the real offset an instruction begins at.
421
+ * So we keep track of both.
422
+ */
423
+ _OffsetType startInstOffset = 0;
424
+
425
+ const uint8_t* p;
426
+
427
+ /* Current working decoded instruction in results. */
428
+ unsigned int nextPos = 0;
429
+ _DInst *pdi = NULL;
430
+
431
+ _OffsetType addrMask = (_OffsetType)-1;
432
+
433
+ _DecodeResult decodeResult;
434
+
435
+ #ifdef DISTORM_LIGHT
436
+ supportOldIntr; /* Unreferenced. */
437
+
438
+ /*
439
+ * Only truncate address if we are using the decompose interface.
440
+ * Otherwise, we use the textual interface which needs full addresses for formatting bytes output.
441
+ * So distorm_format will truncate later.
442
+ */
443
+ if (_ci->features & DF_MAXIMUM_ADDR32) addrMask = 0xffffffff;
444
+ else if (_ci->features & DF_MAXIMUM_ADDR16) addrMask = 0xffff;
445
+ #endif
446
+
447
+ /* No entries are used yet. */
448
+ *usedInstructionsCount = 0;
449
+ ci.dt = _ci->dt;
450
+ _ci->nextOffset = codeOffset;
451
+
452
+ /* Decode instructions as long as we have what to decode/enough room in entries. */
453
+ while (codeLen > 0) {
454
+
455
+ /* startInstOffset holds the displayed offset of current instruction. */
456
+ startInstOffset = codeOffset;
457
+
458
+ memset(&ps, 0, (size_t)((char*)&ps.pfxIndexer[0] - (char*)&ps));
459
+ memset(ps.pfxIndexer, PFXIDX_NONE, sizeof(int) * PFXIDX_MAX);
460
+ ps.start = code;
461
+ ps.last = code;
462
+ prefixSize = 0;
463
+
464
+ if (prefixes_is_valid(*code, ci.dt)) {
465
+ prefixes_decode(code, codeLen, &ps, ci.dt);
466
+ /* Count prefixes, start points to first prefix. */
467
+ prefixSize = (unsigned int)(ps.last - ps.start);
468
+ /*
469
+ * It might be that we will just notice that we ran out of bytes, or only prefixes
470
+ * so we will have to drop everything and halt.
471
+ * Also take into consideration of flow control instruction filter.
472
+ */
473
+ codeLen -= prefixSize;
474
+ if ((codeLen == 0) || (prefixSize == INST_MAXIMUM_SIZE)) {
475
+ if (~_ci->features & DF_RETURN_FC_ONLY) {
476
+ /* Make sure there is enough room. */
477
+ if (nextPos + (ps.last - code) > maxResultCount) return DECRES_MEMORYERR;
478
+
479
+ for (p = code; p < ps.last; p++, startInstOffset++) {
480
+ /* Use next entry. */
481
+ #ifndef DISTORM_LIGHT
482
+ if (supportOldIntr) {
483
+ pdi = (_DInst*)((char*)result + nextPos * sizeof(_DecodedInst));
484
+ }
485
+ else
486
+ #endif /* DISTORM_LIGHT */
487
+ {
488
+ pdi = &result[nextPos];
489
+ }
490
+ nextPos++;
491
+ memset(pdi, 0, sizeof(_DInst));
492
+
493
+ pdi->flags = FLAG_NOT_DECODABLE;
494
+ pdi->imm.byte = *p;
495
+ pdi->size = 1;
496
+ pdi->addr = startInstOffset & addrMask;
497
+ }
498
+ *usedInstructionsCount = nextPos; /* Include them all. */
499
+ }
500
+ if (codeLen == 0) break; /* Bye bye, out of bytes. */
501
+ }
502
+ code += prefixSize;
503
+ codeOffset += prefixSize;
504
+
505
+ /* If we got only prefixes continue to next instruction. */
506
+ if (prefixSize == INST_MAXIMUM_SIZE) continue;
507
+ }
508
+
509
+ /*
510
+ * Now we decode the instruction and only then we do further prefixes handling.
511
+ * This is because the instruction could not be decoded at all, or an instruction requires
512
+ * a mandatory prefix, or some of the prefixes were useless, etc...
513
+
514
+ * Even if there were a mandatory prefix, we already took into account its size as a normal prefix.
515
+ * so prefixSize includes that, and the returned size in pdi is simply the size of the real(=without prefixes) instruction.
516
+ */
517
+ if (ci.dt == Decode64Bits) {
518
+ if (ps.decodedPrefixes & INST_PRE_REX) {
519
+ /* REX prefix must precede first byte of instruction. */
520
+ if (ps.rexPos != (code - 1)) {
521
+ ps.decodedPrefixes &= ~INST_PRE_REX;
522
+ ps.prefixExtType = PET_NONE;
523
+ prefixes_ignore(&ps, PFXIDX_REX);
524
+ }
525
+ /*
526
+ * We will disable operand size prefix,
527
+ * if it exists only after decoding the instruction, since it might be a mandatory prefix.
528
+ * This will be done after calling inst_lookup in decode_inst.
529
+ */
530
+ }
531
+ /* In 64 bits, segment overrides of CS, DS, ES and SS are ignored. So don't take'em into account. */
532
+ if (ps.decodedPrefixes & INST_PRE_SEGOVRD_MASK32) {
533
+ ps.decodedPrefixes &= ~INST_PRE_SEGOVRD_MASK32;
534
+ prefixes_ignore(&ps, PFXIDX_SEG);
535
+ }
536
+ }
537
+
538
+ /* Make sure there is at least one more entry to use, for the upcoming instruction. */
539
+ if (nextPos + 1 > maxResultCount) return DECRES_MEMORYERR;
540
+ #ifndef DISTORM_LIGHT
541
+ if (supportOldIntr) {
542
+ pdi = (_DInst*)((char*)result + nextPos * sizeof(_DecodedInst));
543
+ }
544
+ else
545
+ #endif /* DISTORM_LIGHT */
546
+ {
547
+ pdi = &result[nextPos];
548
+ }
549
+ nextPos++;
550
+
551
+ /*
552
+ * The reason we copy these two again is because we have to keep track on the input ourselves.
553
+ * There might be a case when an instruction is invalid, and then it will be counted as one byte only.
554
+ * But that instruction already read a byte or two from the stream and only then returned the error.
555
+ * Thus, we end up unsynchronized on the stream.
556
+ * This way, we are totally safe, because we keep track after the call to decode_inst, using the returned size.
557
+ */
558
+ ci.code = code;
559
+ ci.codeLen = codeLen;
560
+ /* Nobody uses codeOffset in the decoder itself, so spare it. */
561
+
562
+ decodeResult = decode_inst(&ci, &ps, pdi);
563
+
564
+ /* See if we need to filter this instruction. */
565
+ if ((_ci->features & DF_RETURN_FC_ONLY) && (META_GET_FC(pdi->meta) == FC_NONE)) decodeResult = DECRES_FILTERED;
566
+
567
+ /* Set address to the beginning of the instruction. */
568
+ pdi->addr = startInstOffset & addrMask;
569
+ /* pdi->disp &= addrMask; */
570
+
571
+ if ((decodeResult == DECRES_INPUTERR) && (ps.decodedPrefixes & INST_PRE_VEX)) {
572
+ if (ps.prefixExtType == PET_VEX3BYTES) {
573
+ prefixSize -= 2;
574
+ codeLen += 2;
575
+ } else if (ps.prefixExtType == PET_VEX2BYTES) {
576
+ prefixSize -= 1;
577
+ codeLen += 1;
578
+ }
579
+ ps.last = ps.start + prefixSize - 1;
580
+ code = ps.last + 1;
581
+ codeOffset = startInstOffset + prefixSize;
582
+ } else {
583
+ /* Advance to next instruction. */
584
+ codeLen -= pdi->size;
585
+ codeOffset += pdi->size;
586
+ code += pdi->size;
587
+
588
+ /* Instruction's size should include prefixes. */
589
+ pdi->size += (uint8_t)prefixSize;
590
+ }
591
+
592
+ /* Drop all prefixes and the instruction itself, because the instruction wasn't successfully decoded. */
593
+ if ((decodeResult == DECRES_INPUTERR) && (~_ci->features & DF_RETURN_FC_ONLY)) {
594
+ nextPos--; /* Undo last result. */
595
+ if ((prefixSize + 1) > 0) { /* 1 for the first instruction's byte. */
596
+ if ((nextPos + prefixSize + 1) > maxResultCount) return DECRES_MEMORYERR;
597
+
598
+ for (p = ps.start; p < ps.last + 1; p++, startInstOffset++) {
599
+ /* Use next entry. */
600
+ #ifndef DISTORM_LIGHT
601
+ if (supportOldIntr) {
602
+ pdi = (_DInst*)((char*)result + nextPos * sizeof(_DecodedInst));
603
+ }
604
+ else
605
+ #endif /* DISTORM_LIGHT */
606
+ {
607
+ pdi = &result[nextPos];
608
+ }
609
+ nextPos++;
610
+
611
+ memset(pdi, 0, sizeof(_DInst));
612
+ pdi->flags = FLAG_NOT_DECODABLE;
613
+ pdi->imm.byte = *p;
614
+ pdi->size = 1;
615
+ pdi->addr = startInstOffset & addrMask;
616
+ }
617
+ }
618
+ } else if (decodeResult == DECRES_FILTERED) nextPos--; /* Return it to pool, since it was filtered. */
619
+
620
+ /* Alright, the caller can read, at least, up to this one. */
621
+ *usedInstructionsCount = nextPos;
622
+ /* Fix next offset. */
623
+ _ci->nextOffset = codeOffset;
624
+
625
+ /* Check whether we need to stop on any flow control instruction. */
626
+ features = _ci->features;
627
+ mfc = META_GET_FC(pdi->meta);
628
+ if ((decodeResult == DECRES_SUCCESS) && (features & DF_STOP_ON_FLOW_CONTROL)) {
629
+ if (((features & DF_STOP_ON_CALL) && (mfc == FC_CALL)) ||
630
+ ((features & DF_STOP_ON_RET) && (mfc == FC_RET)) ||
631
+ ((features & DF_STOP_ON_SYS) && (mfc == FC_SYS)) ||
632
+ ((features & DF_STOP_ON_UNC_BRANCH) && (mfc == FC_UNC_BRANCH)) ||
633
+ ((features & DF_STOP_ON_CND_BRANCH) && (mfc == FC_CND_BRANCH)) ||
634
+ ((features & DF_STOP_ON_INT) && (mfc == FC_INT)) ||
635
+ ((features & DF_STOP_ON_CMOV) && (mfc == FC_CMOV)))
636
+ return DECRES_SUCCESS;
637
+ }
638
+ }
639
+
640
+ return DECRES_SUCCESS;
641
+ }