contrast-agent 4.13.1 → 5.1.0

Sign up to get free protection for your applications and to get access to all the features.
Files changed (399) hide show
  1. checksums.yaml +4 -4
  2. data/.simplecov +2 -2
  3. data/Gemfile +1 -1
  4. data/LICENSE.txt +1 -1
  5. data/Rakefile +1 -1
  6. data/exe/contrast_service +1 -1
  7. data/ext/build_funchook.rb +1 -1
  8. data/ext/cs__assess_array/cs__assess_array.c +1 -1
  9. data/ext/cs__assess_array/extconf.rb +1 -1
  10. data/ext/cs__assess_basic_object/cs__assess_basic_object.c +1 -1
  11. data/ext/cs__assess_basic_object/extconf.rb +1 -1
  12. data/ext/cs__assess_fiber_track/cs__assess_fiber_track.c +1 -1
  13. data/ext/cs__assess_fiber_track/extconf.rb +1 -1
  14. data/ext/cs__assess_hash/cs__assess_hash.c +1 -1
  15. data/ext/cs__assess_hash/extconf.rb +1 -1
  16. data/ext/cs__assess_kernel/cs__assess_kernel.c +1 -1
  17. data/ext/cs__assess_kernel/extconf.rb +1 -1
  18. data/ext/cs__assess_marshal_module/cs__assess_marshal_module.c +1 -1
  19. data/ext/cs__assess_marshal_module/extconf.rb +1 -1
  20. data/ext/cs__assess_module/cs__assess_module.c +42 -32
  21. data/ext/cs__assess_module/extconf.rb +1 -1
  22. data/ext/cs__assess_regexp/cs__assess_regexp.c +1 -1
  23. data/ext/cs__assess_regexp/extconf.rb +1 -1
  24. data/ext/cs__assess_string/cs__assess_string.c +1 -1
  25. data/ext/cs__assess_string/extconf.rb +1 -1
  26. data/ext/cs__assess_string_interpolation26/cs__assess_string_interpolation26.c +1 -1
  27. data/ext/cs__assess_string_interpolation26/extconf.rb +1 -1
  28. data/ext/cs__assess_yield_track/cs__assess_yield_track.c +1 -1
  29. data/ext/cs__assess_yield_track/extconf.rb +1 -1
  30. data/ext/cs__common/cs__common.c +1 -1
  31. data/ext/cs__common/extconf.rb +1 -1
  32. data/ext/cs__contrast_patch/cs__contrast_patch.c +1 -1
  33. data/ext/cs__contrast_patch/extconf.rb +1 -1
  34. data/ext/cs__os_information/cs__os_information.c +1 -1
  35. data/ext/cs__os_information/extconf.rb +1 -1
  36. data/ext/extconf_common.rb +1 -1
  37. data/lib/contrast/agent/assess/contrast_event.rb +7 -11
  38. data/lib/contrast/agent/assess/contrast_object.rb +1 -1
  39. data/lib/contrast/agent/assess/events/event_data.rb +30 -0
  40. data/lib/contrast/agent/assess/events/event_factory.rb +14 -6
  41. data/lib/contrast/agent/assess/events/source_event.rb +22 -3
  42. data/lib/contrast/agent/assess/finalizers/freeze.rb +1 -1
  43. data/lib/contrast/agent/assess/finalizers/hash.rb +1 -1
  44. data/lib/contrast/agent/assess/policy/dynamic_source_factory.rb +7 -7
  45. data/lib/contrast/agent/assess/policy/patcher.rb +1 -35
  46. data/lib/contrast/agent/assess/policy/policy.rb +1 -1
  47. data/lib/contrast/agent/assess/policy/policy_node.rb +7 -7
  48. data/lib/contrast/agent/assess/policy/policy_scanner.rb +6 -1
  49. data/lib/contrast/agent/assess/policy/preshift.rb +1 -1
  50. data/lib/contrast/agent/assess/policy/propagation_method.rb +55 -28
  51. data/lib/contrast/agent/assess/policy/propagation_node.rb +1 -1
  52. data/lib/contrast/agent/assess/policy/propagator/append.rb +1 -1
  53. data/lib/contrast/agent/assess/policy/propagator/base.rb +1 -1
  54. data/lib/contrast/agent/assess/policy/propagator/center.rb +2 -2
  55. data/lib/contrast/agent/assess/policy/propagator/custom.rb +1 -1
  56. data/lib/contrast/agent/assess/policy/propagator/database_write.rb +6 -2
  57. data/lib/contrast/agent/assess/policy/propagator/insert.rb +5 -2
  58. data/lib/contrast/agent/assess/policy/propagator/keep.rb +1 -1
  59. data/lib/contrast/agent/assess/policy/propagator/match_data.rb +11 -3
  60. data/lib/contrast/agent/assess/policy/propagator/next.rb +1 -1
  61. data/lib/contrast/agent/assess/policy/propagator/prepend.rb +1 -1
  62. data/lib/contrast/agent/assess/policy/propagator/rack_protection.rb +1 -1
  63. data/lib/contrast/agent/assess/policy/propagator/remove.rb +1 -1
  64. data/lib/contrast/agent/assess/policy/propagator/replace.rb +1 -1
  65. data/lib/contrast/agent/assess/policy/propagator/reverse.rb +1 -1
  66. data/lib/contrast/agent/assess/policy/propagator/select.rb +3 -2
  67. data/lib/contrast/agent/assess/policy/propagator/splat.rb +1 -1
  68. data/lib/contrast/agent/assess/policy/propagator/split.rb +27 -25
  69. data/lib/contrast/agent/assess/policy/propagator/substitution.rb +3 -157
  70. data/lib/contrast/agent/assess/policy/propagator/substitution_utils.rb +190 -0
  71. data/lib/contrast/agent/assess/policy/propagator/trim.rb +9 -4
  72. data/lib/contrast/agent/assess/policy/propagator.rb +1 -1
  73. data/lib/contrast/agent/assess/policy/source_method.rb +39 -26
  74. data/lib/contrast/agent/assess/policy/source_node.rb +1 -1
  75. data/lib/contrast/agent/assess/policy/source_validation/cross_site_validator.rb +1 -1
  76. data/lib/contrast/agent/assess/policy/source_validation/source_validation.rb +1 -1
  77. data/lib/contrast/agent/assess/policy/trigger/reflected_xss.rb +7 -2
  78. data/lib/contrast/agent/assess/policy/trigger/xpath.rb +1 -1
  79. data/lib/contrast/agent/assess/policy/trigger_method.rb +68 -18
  80. data/lib/contrast/agent/assess/policy/trigger_node.rb +15 -7
  81. data/lib/contrast/agent/assess/policy/trigger_validation/redos_validator.rb +1 -1
  82. data/lib/contrast/agent/assess/policy/trigger_validation/ssrf_validator.rb +1 -1
  83. data/lib/contrast/agent/assess/policy/trigger_validation/trigger_validation.rb +1 -1
  84. data/lib/contrast/agent/assess/policy/trigger_validation/xss_validator.rb +2 -2
  85. data/lib/contrast/agent/assess/properties.rb +1 -1
  86. data/lib/contrast/agent/assess/property/evented.rb +25 -12
  87. data/lib/contrast/agent/assess/property/tagged.rb +52 -58
  88. data/lib/contrast/agent/assess/property/updated.rb +1 -1
  89. data/lib/contrast/agent/assess/rule/provider/hardcoded_key.rb +1 -19
  90. data/lib/contrast/agent/assess/rule/provider/hardcoded_password.rb +1 -13
  91. data/lib/contrast/agent/assess/rule/provider/hardcoded_value_rule.rb +42 -60
  92. data/lib/contrast/agent/assess/rule/provider.rb +1 -1
  93. data/lib/contrast/agent/assess/rule/response/autocomplete_rule.rb +68 -0
  94. data/lib/contrast/agent/assess/rule/response/base_rule.rb +197 -0
  95. data/lib/contrast/agent/assess/rule/response/cachecontrol_rule.rb +184 -0
  96. data/lib/contrast/agent/assess/rule/response/clickjacking_rule.rb +66 -0
  97. data/lib/contrast/agent/assess/rule/response/csp_header_insecure_rule.rb +101 -0
  98. data/lib/contrast/agent/assess/rule/response/csp_header_missing_rule.rb +46 -0
  99. data/lib/contrast/agent/assess/rule/response/hsts_header_rule.rb +60 -0
  100. data/lib/contrast/agent/assess/rule/response/parameters_pollution_rule.rb +60 -0
  101. data/lib/contrast/agent/assess/rule/response/x_content_type_rule.rb +52 -0
  102. data/lib/contrast/agent/assess/rule/response/x_xss_protection_rule.rb +53 -0
  103. data/lib/contrast/agent/assess/tag.rb +1 -1
  104. data/lib/contrast/agent/assess/tracker.rb +1 -1
  105. data/lib/contrast/agent/assess.rb +1 -2
  106. data/lib/contrast/agent/at_exit_hook.rb +1 -1
  107. data/lib/contrast/agent/deadzone/policy/deadzone_node.rb +13 -8
  108. data/lib/contrast/agent/deadzone/policy/policy.rb +1 -1
  109. data/lib/contrast/agent/disable_reaction.rb +1 -1
  110. data/lib/contrast/agent/exclusion_matcher.rb +1 -1
  111. data/lib/contrast/agent/inventory/database_config.rb +115 -77
  112. data/lib/contrast/agent/inventory/dependencies.rb +1 -1
  113. data/lib/contrast/agent/inventory/dependency_analysis.rb +1 -1
  114. data/lib/contrast/agent/inventory/dependency_usage_analysis.rb +3 -3
  115. data/lib/contrast/agent/inventory/policy/datastores.rb +1 -1
  116. data/lib/contrast/agent/inventory/policy/policy.rb +1 -1
  117. data/lib/contrast/agent/inventory/policy/trigger_node.rb +1 -1
  118. data/lib/contrast/agent/inventory.rb +1 -1
  119. data/lib/contrast/agent/metric_telemetry_event.rb +3 -3
  120. data/lib/contrast/agent/middleware.rb +8 -77
  121. data/lib/contrast/agent/module_data.rb +1 -1
  122. data/lib/contrast/agent/patching/policy/after_load_patch.rb +1 -1
  123. data/lib/contrast/agent/patching/policy/after_load_patcher.rb +1 -1
  124. data/lib/contrast/agent/patching/policy/method_policy.rb +31 -104
  125. data/lib/contrast/agent/patching/policy/method_policy_extend.rb +113 -0
  126. data/lib/contrast/agent/patching/policy/module_policy.rb +1 -1
  127. data/lib/contrast/agent/patching/policy/patch.rb +1 -1
  128. data/lib/contrast/agent/patching/policy/patch_status.rb +2 -26
  129. data/lib/contrast/agent/patching/policy/patcher.rb +13 -13
  130. data/lib/contrast/agent/patching/policy/policy.rb +1 -1
  131. data/lib/contrast/agent/patching/policy/policy_node.rb +1 -1
  132. data/lib/contrast/agent/patching/policy/trigger_node.rb +1 -1
  133. data/lib/contrast/agent/protect/policy/applies_command_injection_rule.rb +1 -1
  134. data/lib/contrast/agent/protect/policy/applies_deserialization_rule.rb +1 -1
  135. data/lib/contrast/agent/protect/policy/applies_no_sqli_rule.rb +1 -1
  136. data/lib/contrast/agent/protect/policy/applies_path_traversal_rule.rb +1 -1
  137. data/lib/contrast/agent/protect/policy/applies_sqli_rule.rb +1 -1
  138. data/lib/contrast/agent/protect/policy/applies_xxe_rule.rb +1 -1
  139. data/lib/contrast/agent/protect/policy/policy.rb +1 -1
  140. data/lib/contrast/agent/protect/policy/rule_applicator.rb +1 -1
  141. data/lib/contrast/agent/protect/policy/trigger_node.rb +1 -1
  142. data/lib/contrast/agent/protect/rule/base.rb +1 -1
  143. data/lib/contrast/agent/protect/rule/base_service.rb +1 -1
  144. data/lib/contrast/agent/protect/rule/cmd_injection.rb +1 -1
  145. data/lib/contrast/agent/protect/rule/default_scanner.rb +1 -1
  146. data/lib/contrast/agent/protect/rule/deserialization.rb +1 -1
  147. data/lib/contrast/agent/protect/rule/http_method_tampering.rb +1 -1
  148. data/lib/contrast/agent/protect/rule/no_sqli/mongo_no_sql_scanner.rb +1 -1
  149. data/lib/contrast/agent/protect/rule/no_sqli.rb +1 -1
  150. data/lib/contrast/agent/protect/rule/path_traversal.rb +1 -1
  151. data/lib/contrast/agent/protect/rule/sql_sample_builder.rb +1 -1
  152. data/lib/contrast/agent/protect/rule/sqli/default_sql_scanner.rb +1 -1
  153. data/lib/contrast/agent/protect/rule/sqli/mysql_sql_scanner.rb +1 -1
  154. data/lib/contrast/agent/protect/rule/sqli/postgres_sql_scanner.rb +2 -2
  155. data/lib/contrast/agent/protect/rule/sqli/sqlite_sql_scanner.rb +1 -1
  156. data/lib/contrast/agent/protect/rule/sqli.rb +1 -1
  157. data/lib/contrast/agent/protect/rule/unsafe_file_upload.rb +1 -1
  158. data/lib/contrast/agent/protect/rule/xss.rb +1 -1
  159. data/lib/contrast/agent/protect/rule/xxe/entity_wrapper.rb +1 -1
  160. data/lib/contrast/agent/protect/rule/xxe.rb +1 -1
  161. data/lib/contrast/agent/protect/rule.rb +1 -1
  162. data/lib/contrast/agent/reaction_processor.rb +1 -1
  163. data/lib/contrast/agent/reporting/report.rb +27 -0
  164. data/lib/contrast/agent/reporting/reporter.rb +115 -0
  165. data/lib/contrast/agent/reporting/reporting_events/application_inventory.rb +49 -0
  166. data/lib/contrast/agent/reporting/reporting_events/application_update.rb +82 -0
  167. data/lib/contrast/agent/reporting/reporting_events/architecture_component.rb +80 -0
  168. data/lib/contrast/agent/reporting/reporting_events/discovered_route.rb +59 -0
  169. data/lib/contrast/agent/reporting/reporting_events/finding.rb +195 -0
  170. data/lib/contrast/agent/reporting/reporting_events/finding_event.rb +264 -0
  171. data/lib/contrast/agent/reporting/reporting_events/finding_event_source.rb +57 -0
  172. data/lib/contrast/agent/reporting/reporting_events/finding_object.rb +90 -0
  173. data/lib/contrast/agent/reporting/reporting_events/finding_request.rb +121 -0
  174. data/lib/contrast/agent/reporting/reporting_events/finding_signature.rb +105 -0
  175. data/lib/contrast/agent/reporting/reporting_events/finding_stack.rb +67 -0
  176. data/lib/contrast/agent/reporting/reporting_events/finding_taint_range.rb +58 -0
  177. data/lib/contrast/agent/reporting/reporting_events/library_discovery.rb +93 -0
  178. data/lib/contrast/agent/reporting/reporting_events/library_usage_observation.rb +50 -0
  179. data/lib/contrast/agent/reporting/reporting_events/observed_library_usage.rb +54 -0
  180. data/lib/contrast/agent/reporting/reporting_events/observed_route.rb +66 -0
  181. data/lib/contrast/agent/reporting/reporting_events/preflight.rb +39 -0
  182. data/lib/contrast/agent/reporting/reporting_events/preflight_message.rb +75 -0
  183. data/lib/contrast/agent/reporting/reporting_events/reporting_event.rb +47 -0
  184. data/lib/contrast/agent/reporting/reporting_events/route_coverage.rb +76 -0
  185. data/lib/contrast/agent/reporting/reporting_events/route_discovery.rb +66 -0
  186. data/lib/contrast/agent/reporting/reporting_events/route_discovery_observation.rb +65 -0
  187. data/lib/contrast/agent/reporting/reporting_events/server_activity.rb +52 -0
  188. data/lib/contrast/agent/reporting/reporting_events/trace_event_source.rb +30 -0
  189. data/lib/contrast/agent/reporting/reporting_utilities/audit.rb +137 -0
  190. data/lib/contrast/agent/reporting/reporting_utilities/dtm_message.rb +67 -0
  191. data/lib/contrast/agent/reporting/reporting_utilities/endpoints.rb +165 -0
  192. data/lib/contrast/agent/reporting/reporting_utilities/headers.rb +55 -0
  193. data/lib/contrast/agent/reporting/reporting_utilities/reporter_client.rb +86 -0
  194. data/lib/contrast/agent/reporting/reporting_utilities/reporter_client_utils.rb +154 -0
  195. data/lib/contrast/agent/reporting/reporting_utilities/reporting_storage.rb +66 -0
  196. data/lib/contrast/agent/reporting/reporting_utilities/response.rb +30 -0
  197. data/lib/contrast/agent/reporting/reporting_utilities/response_handler.rb +57 -0
  198. data/lib/contrast/agent/reporting/reporting_utilities/response_handler_utils.rb +196 -0
  199. data/lib/contrast/agent/reporting/settings/application_settings.rb +67 -0
  200. data/lib/contrast/agent/reporting/settings/assess.rb +45 -0
  201. data/lib/contrast/agent/reporting/settings/assess_server_feature.rb +136 -0
  202. data/lib/contrast/agent/reporting/settings/exclusions.rb +123 -0
  203. data/lib/contrast/agent/reporting/settings/protect.rb +89 -0
  204. data/lib/contrast/agent/reporting/settings/protect_server_feature.rb +243 -0
  205. data/lib/contrast/agent/reporting/settings/reaction.rb +30 -0
  206. data/lib/contrast/agent/reporting/settings/server_features.rb +78 -0
  207. data/lib/contrast/agent/request.rb +46 -84
  208. data/lib/contrast/agent/request_context.rb +20 -131
  209. data/lib/contrast/agent/request_context_extend.rb +176 -0
  210. data/lib/contrast/agent/request_handler.rb +41 -5
  211. data/lib/contrast/agent/response.rb +25 -88
  212. data/lib/contrast/agent/rule_set.rb +1 -1
  213. data/lib/contrast/agent/scope.rb +1 -1
  214. data/lib/contrast/agent/service_heartbeat.rb +1 -1
  215. data/lib/contrast/agent/startup_metrics_telemetry_event.rb +73 -23
  216. data/lib/contrast/agent/static_analysis.rb +14 -4
  217. data/lib/contrast/agent/telemetry.rb +16 -8
  218. data/lib/contrast/agent/telemetry_event.rb +9 -10
  219. data/lib/contrast/agent/thread.rb +1 -1
  220. data/lib/contrast/agent/thread_watcher.rb +32 -6
  221. data/lib/contrast/agent/tracepoint_hook.rb +1 -4
  222. data/lib/contrast/agent/version.rb +2 -2
  223. data/lib/contrast/agent/worker_thread.rb +1 -1
  224. data/lib/contrast/agent.rb +16 -1
  225. data/lib/contrast/api/communication/connection_status.rb +11 -8
  226. data/lib/contrast/api/communication/messaging_queue.rb +38 -4
  227. data/lib/contrast/api/communication/response_processor.rb +22 -11
  228. data/lib/contrast/api/communication/service_lifecycle.rb +14 -4
  229. data/lib/contrast/api/communication/socket.rb +7 -9
  230. data/lib/contrast/api/communication/socket_client.rb +33 -13
  231. data/lib/contrast/api/communication/speedracer.rb +38 -2
  232. data/lib/contrast/api/communication/tcp_socket.rb +5 -4
  233. data/lib/contrast/api/communication/unix_socket.rb +2 -1
  234. data/lib/contrast/api/communication.rb +1 -1
  235. data/lib/contrast/api/decorators/address.rb +1 -1
  236. data/lib/contrast/api/decorators/agent_startup.rb +1 -1
  237. data/lib/contrast/api/decorators/application_settings.rb +1 -1
  238. data/lib/contrast/api/decorators/application_startup.rb +1 -1
  239. data/lib/contrast/api/decorators/application_update.rb +1 -1
  240. data/lib/contrast/api/decorators/architecture_component.rb +36 -0
  241. data/lib/contrast/api/decorators/finding.rb +29 -0
  242. data/lib/contrast/api/decorators/http_request.rb +3 -2
  243. data/lib/contrast/api/decorators/input_analysis.rb +1 -1
  244. data/lib/contrast/api/decorators/instrumentation_mode.rb +1 -1
  245. data/lib/contrast/api/decorators/library.rb +1 -1
  246. data/lib/contrast/api/decorators/library_usage_update.rb +1 -1
  247. data/lib/contrast/api/decorators/message.rb +1 -1
  248. data/lib/contrast/api/decorators/rasp_rule_sample.rb +1 -1
  249. data/lib/contrast/api/decorators/route_coverage.rb +1 -1
  250. data/lib/contrast/api/decorators/server_features.rb +1 -1
  251. data/lib/contrast/api/decorators/trace_event.rb +1 -1
  252. data/lib/contrast/api/decorators/trace_event_object.rb +1 -1
  253. data/lib/contrast/api/decorators/trace_event_signature.rb +1 -1
  254. data/lib/contrast/api/decorators/trace_taint_range.rb +1 -1
  255. data/lib/contrast/api/decorators/trace_taint_range_tags.rb +1 -1
  256. data/lib/contrast/api/decorators/user_input.rb +1 -1
  257. data/lib/contrast/api/decorators.rb +2 -1
  258. data/lib/contrast/api.rb +1 -1
  259. data/lib/contrast/components/agent.rb +5 -24
  260. data/lib/contrast/components/api.rb +67 -2
  261. data/lib/contrast/components/app_context.rb +12 -66
  262. data/lib/contrast/components/app_context_extend.rb +78 -0
  263. data/lib/contrast/components/assess.rb +12 -8
  264. data/lib/contrast/components/base.rb +24 -1
  265. data/lib/contrast/components/config.rb +38 -23
  266. data/lib/contrast/components/contrast_service.rb +6 -1
  267. data/lib/contrast/components/heap_dump.rb +1 -1
  268. data/lib/contrast/components/inventory.rb +5 -1
  269. data/lib/contrast/components/logger.rb +1 -1
  270. data/lib/contrast/components/protect.rb +6 -2
  271. data/lib/contrast/components/sampling.rb +3 -3
  272. data/lib/contrast/components/scope.rb +1 -1
  273. data/lib/contrast/components/settings.rb +25 -12
  274. data/lib/contrast/config/agent_configuration.rb +2 -2
  275. data/lib/contrast/config/api_configuration.rb +9 -4
  276. data/lib/contrast/config/api_proxy_configuration.rb +14 -0
  277. data/lib/contrast/config/application_configuration.rb +3 -4
  278. data/lib/contrast/config/assess_configuration.rb +4 -4
  279. data/lib/contrast/config/assess_rules_configuration.rb +1 -1
  280. data/lib/contrast/config/base_configuration.rb +18 -29
  281. data/lib/contrast/config/certification_configuration.rb +15 -0
  282. data/lib/contrast/config/env_variables.rb +3 -10
  283. data/lib/contrast/config/exception_configuration.rb +1 -1
  284. data/lib/contrast/config/heap_dump_configuration.rb +7 -7
  285. data/lib/contrast/config/inventory_configuration.rb +2 -6
  286. data/lib/contrast/config/logger_configuration.rb +1 -1
  287. data/lib/contrast/config/protect_configuration.rb +1 -1
  288. data/lib/contrast/config/protect_rule_configuration.rb +2 -2
  289. data/lib/contrast/config/protect_rules_configuration.rb +1 -1
  290. data/lib/contrast/config/request_audit_configuration.rb +18 -0
  291. data/lib/contrast/config/root_configuration.rb +1 -1
  292. data/lib/contrast/config/ruby_configuration.rb +10 -7
  293. data/lib/contrast/config/sampling_configuration.rb +1 -1
  294. data/lib/contrast/config/server_configuration.rb +1 -1
  295. data/lib/contrast/config/service_configuration.rb +2 -3
  296. data/lib/contrast/config.rb +1 -2
  297. data/lib/contrast/configuration.rb +2 -3
  298. data/lib/contrast/extension/assess/array.rb +9 -9
  299. data/lib/contrast/extension/assess/erb.rb +32 -5
  300. data/lib/contrast/extension/assess/eval_trigger.rb +3 -1
  301. data/lib/contrast/extension/assess/exec_trigger.rb +2 -1
  302. data/lib/contrast/extension/assess/fiber.rb +6 -3
  303. data/lib/contrast/extension/assess/hash.rb +1 -1
  304. data/lib/contrast/extension/assess/kernel.rb +8 -3
  305. data/lib/contrast/extension/assess/marshal.rb +6 -2
  306. data/lib/contrast/extension/assess/regexp.rb +8 -2
  307. data/lib/contrast/extension/assess/string.rb +8 -2
  308. data/lib/contrast/extension/assess.rb +1 -1
  309. data/lib/contrast/extension/delegator.rb +1 -1
  310. data/lib/contrast/extension/extension.rb +2 -4
  311. data/lib/contrast/extension/inventory.rb +1 -1
  312. data/lib/contrast/extension/kernel.rb +1 -1
  313. data/lib/contrast/extension/module.rb +1 -1
  314. data/lib/contrast/extension/protect/psych.rb +1 -1
  315. data/lib/contrast/extension/protect.rb +1 -1
  316. data/lib/contrast/extension/thread.rb +32 -13
  317. data/lib/contrast/framework/base_support.rb +5 -1
  318. data/lib/contrast/framework/grape/support.rb +25 -1
  319. data/lib/contrast/framework/manager.rb +33 -36
  320. data/lib/contrast/framework/manager_extend.rb +50 -0
  321. data/lib/contrast/framework/platform_version.rb +1 -1
  322. data/lib/contrast/framework/rack/patch/session_cookie.rb +1 -1
  323. data/lib/contrast/framework/rack/patch/support.rb +1 -1
  324. data/lib/contrast/framework/rack/support.rb +1 -1
  325. data/lib/contrast/framework/rails/patch/action_controller_live_buffer.rb +1 -1
  326. data/lib/contrast/framework/rails/patch/assess_configuration.rb +1 -1
  327. data/lib/contrast/framework/rails/patch/rails_application_configuration.rb +1 -1
  328. data/lib/contrast/framework/rails/patch/support.rb +1 -1
  329. data/lib/contrast/framework/rails/railtie.rb +2 -2
  330. data/lib/contrast/framework/rails/support.rb +46 -2
  331. data/lib/contrast/framework/sinatra/support.rb +26 -3
  332. data/lib/contrast/funchook/funchook.rb +21 -18
  333. data/lib/contrast/logger/application.rb +1 -1
  334. data/lib/contrast/logger/format.rb +1 -1
  335. data/lib/contrast/logger/log.rb +9 -104
  336. data/lib/contrast/logger/request.rb +1 -1
  337. data/lib/contrast/logger/time.rb +1 -1
  338. data/lib/contrast/security_exception.rb +1 -1
  339. data/lib/contrast/tasks/config.rb +1 -1
  340. data/lib/contrast/tasks/service.rb +1 -1
  341. data/lib/contrast/utils/assess/propagation_method_utils.rb +1 -1
  342. data/lib/contrast/utils/assess/property/tagged_utils.rb +24 -1
  343. data/lib/contrast/utils/assess/sampling_util.rb +4 -4
  344. data/lib/contrast/utils/assess/source_method_utils.rb +1 -1
  345. data/lib/contrast/utils/assess/split_utils.rb +23 -0
  346. data/lib/contrast/utils/assess/tracking_util.rb +21 -16
  347. data/lib/contrast/utils/assess/trigger_method_utils.rb +4 -3
  348. data/lib/contrast/utils/class_util.rb +19 -15
  349. data/lib/contrast/utils/duck_utils.rb +1 -1
  350. data/lib/contrast/utils/env_configuration_item.rb +2 -2
  351. data/lib/contrast/utils/exclude_key.rb +1 -1
  352. data/lib/contrast/utils/findings.rb +65 -0
  353. data/lib/contrast/utils/hash_digest.rb +46 -79
  354. data/lib/contrast/utils/hash_digest_extend.rb +129 -0
  355. data/lib/contrast/utils/head_dump_utils_extend.rb +74 -0
  356. data/lib/contrast/utils/heap_dump_util.rb +3 -66
  357. data/lib/contrast/utils/invalid_configuration_util.rb +32 -2
  358. data/lib/contrast/utils/io_util.rb +2 -2
  359. data/lib/contrast/utils/job_servers_running.rb +1 -1
  360. data/lib/contrast/utils/log_utils.rb +108 -0
  361. data/lib/contrast/utils/lru_cache.rb +1 -1
  362. data/lib/contrast/utils/metrics_hash.rb +2 -2
  363. data/lib/contrast/utils/middleware_utils.rb +87 -0
  364. data/lib/contrast/utils/net_http_base.rb +165 -0
  365. data/lib/contrast/utils/object_share.rb +2 -6
  366. data/lib/contrast/utils/os.rb +9 -5
  367. data/lib/contrast/utils/patching/policy/patch_utils.rb +63 -99
  368. data/lib/contrast/utils/patching/policy/patcher_utils.rb +1 -1
  369. data/lib/contrast/utils/preflight_util.rb +1 -1
  370. data/lib/contrast/utils/request_utils.rb +96 -0
  371. data/lib/contrast/utils/resource_loader.rb +1 -1
  372. data/lib/contrast/utils/response_utils.rb +97 -0
  373. data/lib/contrast/utils/sha256_builder.rb +1 -1
  374. data/lib/contrast/utils/stack_trace_utils.rb +1 -1
  375. data/lib/contrast/utils/string_utils.rb +69 -67
  376. data/lib/contrast/utils/tag_util.rb +11 -10
  377. data/lib/contrast/utils/telemetry.rb +5 -3
  378. data/lib/contrast/utils/telemetry_client.rb +90 -0
  379. data/lib/contrast/utils/telemetry_identifier.rb +18 -25
  380. data/lib/contrast/utils/thread_tracker.rb +1 -1
  381. data/lib/contrast/utils/timer.rb +1 -1
  382. data/lib/contrast-agent.rb +1 -1
  383. data/lib/contrast.rb +1 -1
  384. data/resources/assess/policy.json +10 -0
  385. data/ruby-agent.gemspec +9 -10
  386. metadata +111 -50
  387. data/ext/cs__assess_active_record_named/cs__active_record_named.c +0 -46
  388. data/ext/cs__assess_active_record_named/cs__active_record_named.h +0 -11
  389. data/ext/cs__assess_active_record_named/extconf.rb +0 -5
  390. data/lib/contrast/agent/assess/policy/rewriter_patch.rb +0 -95
  391. data/lib/contrast/agent/class_reopener.rb +0 -258
  392. data/lib/contrast/agent/rewriter.rb +0 -259
  393. data/lib/contrast/config/default_value.rb +0 -17
  394. data/lib/contrast/framework/rails/rewrite/action_controller_railties_helper_inherited.rb +0 -37
  395. data/lib/contrast/framework/rails/rewrite/active_record_attribute_methods_read.rb +0 -41
  396. data/lib/contrast/framework/rails/rewrite/active_record_named.rb +0 -75
  397. data/lib/contrast/framework/rails/rewrite/active_record_time_zone_inherited.rb +0 -35
  398. data/lib/contrast/utils/requests_client.rb +0 -150
  399. data/lib/contrast/utils/ruby_ast_rewriter.rb +0 -82
@@ -0,0 +1,78 @@
1
+ # Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
2
+ # frozen_string_literal: true
3
+
4
+ require 'contrast/utils/object_share'
5
+ require 'contrast/agent/reporting/settings/assess_server_feature'
6
+ require 'contrast/agent/reporting/settings/protect_server_feature'
7
+
8
+ module Contrast
9
+ module Agent
10
+ module Reporting
11
+ # This module will hold all the settings from the TS responce
12
+ module Settings
13
+ # All of the settings from TeamServer that apply at the server level.
14
+ # At least one, but not necessarily all, setting will differ from the agent's current set.
15
+ # agents are able to replace all server settings with those in this message.
16
+ class ServerFeatures
17
+ # The level at which the agent should log. Overridden by agent.logger.level
18
+ # if set in a local configuration
19
+ #
20
+ # @return log_level [String] [ ERROR, WARN, INFO, DEBUG, TRACE ]
21
+ def log_level
22
+ @_log_level ||= Contrast::Utils::ObjectShare::EMPTY_STRING
23
+ end
24
+
25
+ # set the log level
26
+ #
27
+ # @param log_level [String]
28
+ # @return log_level [String] [ ERROR, WARN, INFO, DEBUG, TRACE ]
29
+ def log_level= log_level
30
+ @_log_level = log_level if log_level.is_a? String
31
+ end
32
+
33
+ # Where to log the agent's log file, if set by the user. Overridden by agent.logger.path
34
+ # if set in a local configuration.
35
+ #
36
+ # @return log_file [String] path
37
+ def log_file
38
+ @_log_file ||= Contrast::Utils::ObjectShare::EMPTY_STRING
39
+ end
40
+
41
+ # Set the log file
42
+ #
43
+ # @param log_file [String] path
44
+ # @return log_file [String] path
45
+ def log_file= log_file
46
+ @_log_file = log_file if log_file.is_a? String
47
+ end
48
+
49
+ # Controls for the reporting of telemetry events from the agent to TeamServer.
50
+ # This is NOT for the agent telemetry feature collecting metrics sent to other services.
51
+ #
52
+ # @return telemetry [Boolean]
53
+ def telemetry
54
+ @_telemetry
55
+ end
56
+
57
+ # sets the telemetry value
58
+ #
59
+ # @param telemetry [Boolean]
60
+ # @return telemetry [Boolean]
61
+ def telemetry= telemetry
62
+ @_telemetry = telemetry if !!telemetry == telemetry
63
+ end
64
+
65
+ # @return assess [Contrast::Agent::Reporting::Settings::AssessServerFeature]
66
+ def assess
67
+ @_assess ||= Contrast::Agent::Reporting::Settings::AssessServerFeature.new
68
+ end
69
+
70
+ # @return protect [Contrast::Agent::Reporting::Settings::ProtectServerFeature]
71
+ def protect
72
+ @_protect ||= Contrast::Agent::Reporting::Settings::ProtectServerFeature.new
73
+ end
74
+ end
75
+ end
76
+ end
77
+ end
78
+ end
@@ -1,4 +1,4 @@
1
- # Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
1
+ # Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
2
2
  # frozen_string_literal: true
3
3
 
4
4
  require 'resolv'
@@ -9,6 +9,7 @@ require 'contrast/utils/string_utils'
9
9
  require 'contrast/utils/hash_digest'
10
10
  require 'contrast/components/logger'
11
11
  require 'contrast/components/scope'
12
+ require 'contrast/utils/request_utils'
12
13
 
13
14
  module Contrast
14
15
  module Agent
@@ -16,7 +17,14 @@ module Contrast
16
17
  # provides access to the original Rack::Request object as well as extracts
17
18
  # data in a format that the Agent expects, caching those transformations in
18
19
  # order to avoid repeatedly creating Strings & thrashing GC.
20
+ #
21
+ # @attr_reader rack_request [Rack::Request] The passed to the Agent RackRequest to be wrapped.
22
+ # @attr_accessor route [Contrast::Api::Dtm::RouteCoverage] the route, used for findings, of this request
23
+ # @attr_accessor observed_route [Contrast::Api::Dtm::ObservedRoute] the route, used for coverage of this request
24
+ # @attr_accessor new_observed_route [Contrast::Agent::Reporting::ObservedRoute] the route, used for coverage, of
25
+ # this request
19
26
  class Request
27
+ include Contrast::Utils::RequestUtils
20
28
  include Contrast::Components::Logger::InstanceMethods
21
29
  include Contrast::Components::Scope::InstanceMethods
22
30
 
@@ -28,16 +36,31 @@ module Contrast
28
36
  LAST_NUMBER_MARKER = '/{n}'
29
37
 
30
38
  attr_reader :rack_request
31
- attr_accessor :route, :observed_route
39
+ attr_accessor :route, :observed_route, :new_observed_route
32
40
 
33
41
  # Delegate calls to the following methods to the attribute @rack_request
34
42
  def_delegators :@rack_request, :base_url, :content_type, :cookies, :env, :ip, :path, :port, :query_string,
35
43
  :request_method, :scheme, :url, :user_agent
36
44
 
45
+ # Initialize new Contrast Request
46
+ #
47
+ # @param rack_request [Rack::Request] The passed to the Agent RackRequest to be wrapped.
37
48
  def initialize rack_request
38
49
  @rack_request = rack_request
39
50
  end
40
51
 
52
+ # The HTTP version of this request
53
+ # @return [String]
54
+ def version
55
+ env = rack_request.env
56
+ return '1.1' unless env
57
+
58
+ version = env['HTTP_VERSION']
59
+ return '1.1' unless version
60
+
61
+ version.split('/')[-1]
62
+ end
63
+
41
64
  # Returns a normalized form of the URI. In "normal" URIs
42
65
  # this will return an unchanged String, but in REST-y
43
66
  # URIs this will normalize the digit path tokens, e.g.:
@@ -47,9 +70,11 @@ module Contrast
47
70
  # /accounts/{n}/view
48
71
  #
49
72
  # Should also handle the ;jsessionid.
73
+ #
74
+ # @return uri [String]
50
75
  def normalized_uri
51
76
  @_normalized_uri ||= begin
52
- path = rack_request.path
77
+ path = rack_request.path_info
53
78
  uri = path.split(Contrast::Utils::ObjectShare::SEMICOLON)[0] # remove ;jsessionid
54
79
  uri = uri.split(Contrast::Utils::ObjectShare::QUESTION_MARK)[0] # remove ?query_string=
55
80
  uri.gsub(INNER_REST_TOKEN, INNER_NUMBER_MARKER) # replace interior tokens
@@ -57,6 +82,9 @@ module Contrast
57
82
  end
58
83
  end
59
84
 
85
+ # Returns the request file type
86
+ #
87
+ # @return type [Symbol<:XML, :JSON, :NORMAL>]
60
88
  def document_type
61
89
  @_document_type ||= if /xml/i.match?(content_type) || body&.start_with?('<?xml')
62
90
  :XML
@@ -68,6 +96,8 @@ module Contrast
68
96
  end
69
97
 
70
98
  # Header keys upcased and any underscores replaced with dashes
99
+ #
100
+ # @return headers [Hash]
71
101
  def headers
72
102
  @_headers ||= with_contrast_scope do
73
103
  hash = {}
@@ -83,6 +113,10 @@ module Contrast
83
113
  end
84
114
  end
85
115
 
116
+ # Try and read the body and return memorized object of the body.
117
+ # If body contains file, do not parse it, otherwiseBody might be nil
118
+ #
119
+ # @return @_body [String, nil] The body of the Rack::Request
86
120
  def body
87
121
  # Memoize a flag indicating whether we've tried to read the body or not
88
122
  # (can't use body because it might be nil)
@@ -119,10 +153,16 @@ module Contrast
119
153
  @_dtm ||= Contrast::Api::Dtm::HttpRequest.build(self)
120
154
  end
121
155
 
156
+ # flattened hash of request params
157
+ #
158
+ # @return parameters [Hash]
122
159
  def parameters
123
160
  @_parameters ||= with_contrast_scope { normalize_params(rack_request.params) }
124
161
  end
125
162
 
163
+ # returns multipart filenames
164
+ #
165
+ # @return file_names [Hash]
126
166
  def file_names
127
167
  @_file_names ||= begin
128
168
  names = {}
@@ -134,6 +174,9 @@ module Contrast
134
174
  end
135
175
  end
136
176
 
177
+ # returns or fenerates the hash checksum for the request
178
+ #
179
+ # @return @_hash_id [String] Contrast::Utils::HashDigest generated string checksum
137
180
  def hash_id
138
181
  @_hash_id ||= Contrast::Utils::HashDigest.generate_request_hash(self)
139
182
  end
@@ -152,87 +195,6 @@ module Contrast
152
195
 
153
196
  accepts.start_with?(*MEDIA_TYPE_MARKERS)
154
197
  end
155
-
156
- private
157
-
158
- # Return a flattened hash of params with realized paths for keys, in
159
- # addition to a separate, valueless, entry for each nest key.
160
- # See RUBY-621 for more details.
161
- # { key : { nested_key : ['x','y','z' ] } }
162
- # becomes
163
- # {
164
- # key[nested_key][0] : 'x'
165
- # key[nested_key][1] : 'y'
166
- # key[nested_key][2] : 'z'
167
- # key : ''
168
- # nested_key : ''
169
- # }
170
- def normalize_params val, prefix: nil
171
- # In non-recursive invocations, val should always be a Hash
172
- # (rather than breaking this out into two methods)
173
- case val
174
- when Tempfile
175
- # Skip if it's the auto-generated value from rails when it handles
176
- # file uploads. The file name will still be sent to SR for analysis.
177
- {}
178
- when Hash
179
- res = val.each_with_object({}) do |(k, v), hash|
180
- k = Contrast::Utils::StringUtils.force_utf8(k)
181
- nested_prefix = prefix.nil? ? k : "#{ prefix }[#{ k }]"
182
- hash[k] = Contrast::Utils::ObjectShare::EMPTY_STRING
183
- hash.merge! normalize_params(v, prefix: nested_prefix)
184
- end
185
- res[prefix] = Contrast::Utils::ObjectShare::EMPTY_STRING if prefix
186
- res
187
- when Enumerable
188
- idx = 0
189
- res = {}
190
- while idx < val.length
191
- res.merge! normalize_params(val[idx], prefix: "#{ prefix }[#{ idx }]")
192
- idx += 1
193
- end
194
- res[prefix] = Contrast::Utils::ObjectShare::EMPTY_STRING if prefix
195
- res
196
- else
197
- { prefix => Contrast::Utils::StringUtils.force_utf8(val) }
198
- end
199
- end
200
-
201
- def read_body body
202
- return body if body.is_a?(String)
203
-
204
- begin
205
- can_rewind = Contrast::Utils::DuckUtils.quacks_to?(body, :rewind)
206
- # if we are after a middleware that failed to rewind
207
- body.rewind if can_rewind
208
- body.read
209
- rescue StandardError => e
210
- logger.error('Error in attempt to read body', message: e.message)
211
- logger.trace('With Stack', e)
212
- body.to_s
213
- ensure
214
- # be a good citizen and rewind
215
- body.rewind if can_rewind
216
- end
217
- end
218
-
219
- def traverse_parsed_multipart multipart_data, current_names
220
- return current_names unless multipart_data
221
-
222
- multipart_data.each_value do |data_value|
223
- next unless data_value.is_a?(Hash)
224
-
225
- tempfile = data_value[:tempfile]
226
- if tempfile.nil?
227
- traverse_parsed_multipart(data_value, current_names)
228
- else
229
- name = data_value[:name].to_s
230
- file_name = data_value[:filename].to_s
231
- current_names[name] = file_name
232
- end
233
- end
234
- current_names
235
- end
236
198
  end
237
199
  end
238
200
  end
@@ -1,4 +1,4 @@
1
- # Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
1
+ # Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
2
2
  # frozen_string_literal: true
3
3
 
4
4
  require 'contrast/utils/timer'
@@ -7,6 +7,9 @@ require 'contrast/agent/response'
7
7
  require 'contrast/agent/inventory/database_config'
8
8
  require 'contrast/components/logger'
9
9
  require 'contrast/components/scope'
10
+ require 'contrast/utils/request_utils'
11
+ require 'contrast/agent/request_context_extend'
12
+ require 'contrast/agent/reporting/reporting_events/observed_route'
10
13
 
11
14
  module Contrast
12
15
  module Agent
@@ -24,14 +27,18 @@ module Contrast
24
27
  # @attr_reader server_activity [Contrast::Api::Dtm::ServerActivity] the server activity found in this request
25
28
  # @attr_reader route [Contrast::Api::Dtm::RouteCoverage] the route, used for findings, of this request
26
29
  # @attr_reader observed_route [Contrast::Api::Dtm::ObservedRoute] the route, used for coverage, of this request
30
+ # @attr_reader new_observed_route [Contrast::Agent::Reporting::ObservedRoute] the route, used for coverage, of this
31
+ # request
27
32
  class RequestContext
28
33
  include Contrast::Components::Logger::InstanceMethods
29
34
  include Contrast::Components::Scope::InstanceMethods
35
+ include Contrast::Utils::RequestUtils
36
+ include Contrast::Agent::RequestContextExtend
30
37
 
31
38
  EMPTY_INPUT_ANALYSIS_PB = Contrast::Api::Settings::InputAnalysis.new
32
39
 
33
- attr_reader :activity, :logging_hash, :observed_route, :request, :response, :route, :speedracer_input_analysis,
34
- :server_activity, :timer
40
+ attr_reader :activity, :logging_hash, :observed_route, :new_observed_route, :request, :response, :route,
41
+ :speedracer_input_analysis, :server_activity, :timer
35
42
 
36
43
  def initialize rack_request, app_loaded = true
37
44
  with_contrast_scope do
@@ -47,8 +54,6 @@ module Contrast
47
54
 
48
55
  @server_activity = Contrast::Api::Dtm::ServerActivity.new
49
56
 
50
- @observed_route = Contrast::Api::Dtm::ObservedRoute.new
51
-
52
57
  # build analyzer
53
58
  @do_not_track = false
54
59
  @speedracer_input_analysis = EMPTY_INPUT_ANALYSIS_PB
@@ -64,7 +69,7 @@ module Contrast
64
69
  @sample_req, @sample_res = Contrast::Utils::Assess::SamplingUtil.instance.sample?(@request)
65
70
  end
66
71
 
67
- append_route_coverage(Contrast::Agent.framework_manager.get_route_dtm(@request))
72
+ handle_routes
68
73
  end
69
74
  end
70
75
 
@@ -100,98 +105,6 @@ module Contrast
100
105
  ::Contrast::ASSESS.enabled?
101
106
  end
102
107
 
103
- # Convert the discovered route for this request to appropriate forms and disseminate it to those locations
104
- # where it is necessary for our route coverage and finding vulnerability discovery features to function.
105
- #
106
- # @param route [Contrast::Api::Dtm::RouteCoverage, nil] the route of the current request, as determined from the
107
- # framework
108
- def append_route_coverage route
109
- return unless route
110
-
111
- # For our findings
112
- @route = route
113
-
114
- # For SR findings
115
- @activity.routes << route
116
-
117
- # For TS routes
118
- @observed_route.signature = route.route
119
- @observed_route.verb = route.verb
120
- @observed_route.url = route.url if route.url
121
- @request.route = route
122
- @request.observed_route = @observed_route
123
- end
124
-
125
- # Collect the results for the given rule with the given action
126
- #
127
- # @param rule [String] the id of the rule to which the results apply
128
- # @param response_type [Symbol] the result of the response, matching a value of
129
- # Contrast::Api::Dtm::AttackResult::ResponseType
130
- # @return [Array<Contrast::Api::Dtm::AttackResult>]
131
- def results_for rule, response_type = nil
132
- if response_type.nil?
133
- activity.results.select { |r| r.rule_id == rule }
134
- else
135
- activity.results.select { |r| r.rule_id == rule && r.response == response_type }
136
- end
137
- end
138
-
139
- def service_extract_request
140
- return false unless ::Contrast::AGENT.enabled?
141
- return false unless ::Contrast::PROTECT.enabled?
142
- return false if @do_not_track
143
-
144
- service_response = Contrast::Agent.messaging_queue.send_event_immediately(@activity.http_request)
145
- return false unless service_response
146
-
147
- handle_protect_state(service_response)
148
- ia = service_response.input_analysis
149
- if ia
150
- if logger.trace?
151
- logger.trace('Analysis from Contrast Service', evaluations: ia.results.length)
152
- logger.trace('Results', input_analysis: ia.inspect)
153
- end
154
- @speedracer_input_analysis = ia
155
- speedracer_input_analysis.request = request
156
- else
157
- logger.trace('Analysis from Contrast Service was empty.')
158
- false
159
- end
160
- rescue Contrast::SecurityException => e
161
- raise e
162
- rescue StandardError => e
163
- logger.warn('Unable to extract Contrast Service information from request', e)
164
- false
165
- end
166
-
167
- # NOTE: this method is only used as a backstop if Speedracer sends Input Evaluations when the protect state
168
- # indicates a security exception should be thrown. This method ensures that the attack reports are generated.
169
- # Normally these should be generated on Speedracer for any attacks detected during prefilter.
170
- #
171
- # @param agent_settings [Contrast::Api::Settings::AgentSettings]
172
- def handle_protect_state agent_settings
173
- return unless agent_settings&.protect_state
174
-
175
- state = agent_settings.protect_state
176
- @uuid = state.uuid
177
- @do_not_track = true unless state.track_request
178
- return unless state.security_exception
179
-
180
- # If Contrast Service has NOT handled the input analysis, handle them here
181
- build_attack_results(agent_settings)
182
- logger.debug('Contrast Service said to block this request')
183
- raise Contrast::SecurityException.new(nil, (state.security_message || 'Blocking suspicious behavior'))
184
- end
185
-
186
- # append anything we've learned to the request seen message this is the sum-total of all inventory information
187
- # that has been accumulated since the last request
188
- def extract_after rack_response
189
- @response = Contrast::Agent::Response.new(rack_response)
190
- activity.http_response = @response.dtm if @sample_res
191
- rescue StandardError => e
192
- logger.error('Unable to extract information after request', e)
193
- end
194
-
195
108
  def add_property key, value
196
109
  @_properties[key] = value
197
110
  end
@@ -203,43 +116,19 @@ module Contrast
203
116
  def reset_activity
204
117
  @activity = Contrast::Api::Dtm::Activity.new(http_request: request.dtm)
205
118
  @server_activity = Contrast::Api::Dtm::ServerActivity.new
206
- @observed_route = Contrast::Api::Dtm::ObservedRoute.new
119
+ @observed_route = Contrast::Api::Dtm::ObservedRoute.new # TODO: RUBY-1438 -- remove
120
+ @new_observed_route = Contrast::Agent::Reporting::ObservedRoute.new
207
121
  end
208
122
 
209
123
  private
210
124
 
211
- # Generate attack results directly from any evaluations on the agent settings object.
212
- #
213
- # @param agent_settings [Contrast::Api::Settings::AgentSettings]
214
- def build_attack_results agent_settings
215
- return unless agent_settings&.input_analysis&.results&.any?
216
-
217
- attack_results_by_rule = {}
218
- agent_settings.input_analysis.results.each do |ia_result|
219
- rule_id = ia_result.rule_id
220
- rule = ::Contrast::PROTECT.rule(rule_id)
221
- next unless rule
222
-
223
- if logger.debug?
224
- logger.debug('Building attack result from Contrast Service input analysis result',
225
- result: ia_result.inspect)
226
- end
227
-
228
- attack_result = if rule.mode == :BLOCK
229
- # special case for rules (like reflected xss) that used to have an infilter / block mode
230
- # but now are just block at perimeter
231
- rule.build_attack_with_match(self, ia_result, attack_results_by_rule[rule_id],
232
- ia_result.value)
233
- else
234
- rule.build_attack_without_match(self, ia_result, attack_results_by_rule[rule_id])
235
- end
236
- attack_results_by_rule[rule_id] = attack_result
237
- end
238
-
239
- attack_results_by_rule.each_pair do |_, attack_result|
240
- logger.info('Blocking attack result', rule: attack_result.rule_id)
241
- activity.results << attack_result
242
- end
125
+ def handle_routes
126
+ @observed_route = Contrast::Api::Dtm::ObservedRoute.new # TODO: RUBY-1438 -- remove
127
+ @new_observed_route = Contrast::Agent::Reporting::ObservedRoute.new
128
+ route_dtm = Contrast::Agent.framework_manager.get_route_dtm(@request)
129
+ new_route_coverage_dtm = Contrast::Agent.framework_manager.get_route_information(@request)
130
+ append_route_coverage(route_dtm)
131
+ append_to_new_observed_route(new_route_coverage_dtm)
243
132
  end
244
133
  end
245
134
  end
@@ -0,0 +1,176 @@
1
+ # Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
2
+ # frozen_string_literal: true
3
+
4
+ require 'contrast/agent/assess/rule/response/autocomplete_rule'
5
+ require 'contrast/agent/assess/rule/response/hsts_header_rule'
6
+ require 'contrast/agent/assess/rule/response/cachecontrol_rule'
7
+ require 'contrast/agent/assess/rule/response/clickjacking_rule'
8
+ require 'contrast/agent/assess/rule/response/x_content_type_rule'
9
+ require 'contrast/agent/assess/rule/response/parameters_pollution_rule'
10
+
11
+ module Contrast
12
+ module Agent
13
+ # This class extends RequestContexts: this class acts to encapsulate information about the currently
14
+ # executed request, making it available to the Agent for the duration of the request in a standardized
15
+ # and normalized format which the Agent understands.
16
+ module RequestContextExtend
17
+ BUILD_ATTACK_LOGGER_MESSAGE = 'Building attack result from Contrast Service input analysis result'
18
+ # Convert the discovered route for this request to appropriate forms and disseminate it to those locations
19
+ # where it is necessary for our route coverage and finding vulnerability discovery features to function.
20
+ #
21
+ # @param route [Contrast::Api::Dtm::RouteCoverage, nil] the route of the current request, as determined from the
22
+ # framework
23
+ def append_route_coverage route
24
+ return unless route
25
+
26
+ # For our findings
27
+ @route = route
28
+
29
+ # For SR findings
30
+ @activity.routes << route
31
+
32
+ # For TS routes
33
+ @observed_route.signature = route.route
34
+ @observed_route.verb = route.verb
35
+ @observed_route.url = route.url if route.url
36
+ @request.route = route
37
+ @request.observed_route = @observed_route
38
+ end
39
+
40
+ # Convert the discovered route for this request to appropriate forms and disseminate it to those locations
41
+ # where it is necessary for our route coverage and finding vulnerability discovery features to function.
42
+ #
43
+ def append_to_new_observed_route route
44
+ return unless route
45
+
46
+ @new_observed_route.signature = route.route
47
+ @new_observed_route.verb = route.verb
48
+ @new_observed_route.url = route.url if route.url
49
+ @request.new_observed_route = @new_observed_route
50
+ end
51
+
52
+ # Collect the results for the given rule with the given action
53
+ #
54
+ # @param rule [String] the id of the rule to which the results apply
55
+ # @param response_type [Symbol] the result of the response, matching a value of
56
+ # Contrast::Api::Dtm::AttackResult::ResponseType
57
+ # @return [Array<Contrast::Api::Dtm::AttackResult>]
58
+ def results_for rule, response_type = nil
59
+ if response_type.nil?
60
+ activity.results.select { |r| r.rule_id == rule }
61
+ else
62
+ activity.results.select { |r| r.rule_id == rule && r.response == response_type }
63
+ end
64
+ end
65
+
66
+ def service_extract_request
67
+ return false unless ::Contrast::AGENT.enabled?
68
+ return false unless ::Contrast::PROTECT.enabled?
69
+ return false if @do_not_track
70
+
71
+ service_response = Contrast::Agent.messaging_queue.send_event_immediately(@activity.http_request)
72
+ return false unless service_response
73
+
74
+ handle_protect_state(service_response)
75
+ ia = service_response.input_analysis
76
+ if ia
77
+ if logger.trace?
78
+ logger.trace('Analysis from Contrast Service', evaluations: ia.results.length)
79
+ logger.trace('Results', input_analysis: ia.inspect)
80
+ end
81
+ @speedracer_input_analysis = ia
82
+ speedracer_input_analysis.request = request
83
+ else
84
+ logger.trace('Analysis from Contrast Service was empty.')
85
+ false
86
+ end
87
+ rescue Contrast::SecurityException => e
88
+ raise e
89
+ rescue StandardError => e
90
+ logger.warn('Unable to extract Contrast Service information from request', e)
91
+ false
92
+ end
93
+
94
+ # NOTE: this method is only used as a backstop if Speedracer sends Input Evaluations when the protect state
95
+ # indicates a security exception should be thrown. This method ensures that the attack reports are generated.
96
+ # Normally these should be generated on Speedracer for any attacks detected during prefilter.
97
+ #
98
+ # @param agent_settings [Contrast::Api::Settings::AgentSettings]
99
+ def handle_protect_state agent_settings
100
+ return unless agent_settings&.protect_state
101
+
102
+ state = agent_settings.protect_state
103
+ @uuid = state.uuid
104
+ @do_not_track = true unless state.track_request
105
+ return unless state.security_exception
106
+
107
+ # If Contrast Service has NOT handled the input analysis, handle them here
108
+ build_attack_results(agent_settings)
109
+ logger.debug('Contrast Service said to block this request')
110
+ raise Contrast::SecurityException.new(nil, (state.security_message || 'Blocking suspicious behavior'))
111
+ end
112
+
113
+ # append anything we've learned to the request seen message this is the sum-total of all inventory information
114
+ # that has been accumulated since the last request
115
+ def extract_after rack_response
116
+ @response = Contrast::Agent::Response.new(rack_response)
117
+ activity.http_response = @response.dtm if @sample_res
118
+ return unless Contrast::Agent::Reporter.enabled?
119
+
120
+ Contrast::Agent::Assess::Rule::Response::Autocomplete.new.analyze(@response)
121
+ Contrast::Agent::Assess::Rule::Response::HSTSHeader.new.analyze(@response)
122
+ Contrast::Agent::Assess::Rule::Response::Cachecontrol.new.analyze(@response)
123
+ Contrast::Agent::Assess::Rule::Response::XXssProtection.new.analyze(@response)
124
+ Contrast::Agent::Assess::Rule::Response::CspHeaderMissing.new.analyze(@response)
125
+ Contrast::Agent::Assess::Rule::Response::CspHeaderInsecure.new.analyze(@response)
126
+ Contrast::Agent::Assess::Rule::Response::Clickjacking.new.analyze(@response)
127
+ Contrast::Agent::Assess::Rule::Response::XContentType.new.analyze(@response)
128
+ Contrast::Agent::Assess::Rule::Response::ParametersPollution.new.analyze(@response)
129
+ rescue StandardError => e
130
+ logger.error('Unable to extract information after request', e)
131
+ end
132
+
133
+ private
134
+
135
+ # Generate attack results directly from any evaluations on the agent settings object.
136
+ #
137
+ # @param agent_settings [Contrast::Api::Settings::AgentSettings]
138
+ def build_attack_results agent_settings
139
+ return unless agent_settings&.input_analysis&.results&.any?
140
+
141
+ results_by_rule = {}
142
+ agent_settings.input_analysis.results.each do |ia_result|
143
+ rule_id = ia_result.rule_id
144
+ rule = ::Contrast::PROTECT.rule(rule_id)
145
+ next unless rule
146
+
147
+ logger.debug(BUILD_ATTACK_LOGGER_MESSAGE, result: ia_result.inspect) if logger.debug?
148
+ results_by_rule[rule_id] = attack_result rule, rule_id, ia_result, results_by_rule
149
+ end
150
+
151
+ results_by_rule.each_pair do |_, attack_result|
152
+ logger.info('Blocking attack result', rule: attack_result.rule_id)
153
+ activity.results << attack_result
154
+ end
155
+ end
156
+
157
+ # Generates the attack result
158
+ #
159
+ # @param rule [Contrast::Agent::Protect::Rule, Contrast::Agent::Assess::Rule]
160
+ # @param rule_id [String] String name of the rule
161
+ # @param ia_result [Contrast::Api::Settings::InputAnalysisResult] the
162
+ # analysis of the input that was determined to be an attack
163
+ # @param results_by_rule [Hash] attack results from any evaluations on the agent settings object.
164
+ # @return [Contrast::Api::Dtm::AttackResult] the attack result from this input
165
+ def attack_result rule, rule_id, ia_result, results_by_rule
166
+ @_attack_result = if rule.mode == :BLOCK
167
+ # special case for rules (like reflected xss) that used to have an infilter / block mode
168
+ # but now are just block at perimeter
169
+ rule.build_attack_with_match(self, ia_result, results_by_rule[rule_id], ia_result.value)
170
+ else
171
+ rule.build_attack_without_match(self, ia_result, results_by_rule[rule_id])
172
+ end
173
+ end
174
+ end
175
+ end
176
+ end