contrast-agent 6.6.4 → 6.8.0

Sign up to get free protection for your applications and to get access to all the features.
Files changed (340) hide show
  1. checksums.yaml +4 -4
  2. data/.gitignore +3 -2
  3. data/.gitmodules +0 -3
  4. data/.simplecov +0 -1
  5. data/Rakefile +0 -1
  6. data/ext/cs__assess_array/cs__assess_array.c +41 -10
  7. data/ext/cs__assess_array/cs__assess_array.h +4 -1
  8. data/ext/cs__scope/cs__scope.c +1 -1
  9. data/lib/contrast/agent/assess/contrast_event.rb +2 -24
  10. data/lib/contrast/agent/assess/events/source_event.rb +7 -61
  11. data/lib/contrast/agent/assess/finalizers/hash.rb +11 -0
  12. data/lib/contrast/agent/assess/policy/dynamic_source_factory.rb +0 -55
  13. data/lib/contrast/agent/assess/policy/policy_node.rb +3 -3
  14. data/lib/contrast/agent/assess/policy/policy_node_utils.rb +0 -1
  15. data/lib/contrast/agent/assess/policy/propagation_node.rb +4 -4
  16. data/lib/contrast/agent/assess/policy/source_method.rb +24 -1
  17. data/lib/contrast/agent/assess/policy/trigger/reflected_xss.rb +7 -5
  18. data/lib/contrast/agent/assess/policy/trigger/xpath.rb +6 -1
  19. data/lib/contrast/agent/assess/policy/trigger_method.rb +40 -121
  20. data/lib/contrast/agent/assess/policy/trigger_node.rb +3 -3
  21. data/lib/contrast/agent/assess/policy/trigger_validation/redos_validator.rb +1 -1
  22. data/lib/contrast/agent/assess/policy/trigger_validation/ssrf_validator.rb +1 -1
  23. data/lib/contrast/agent/assess/policy/trigger_validation/xss_validator.rb +1 -1
  24. data/lib/contrast/agent/assess/property/evented.rb +2 -12
  25. data/lib/contrast/agent/assess/rule/provider/hardcoded_value_rule.rb +42 -82
  26. data/lib/contrast/agent/assess/rule/response/base_rule.rb +11 -27
  27. data/lib/contrast/agent/assess/rule/response/body_rule.rb +1 -3
  28. data/lib/contrast/agent/assess/rule/response/cache_control_header_rule.rb +77 -62
  29. data/lib/contrast/agent/assess/rule/response/csp_header_insecure_rule.rb +1 -1
  30. data/lib/contrast/agent/assess/rule/response/framework/rails_support.rb +6 -1
  31. data/lib/contrast/agent/assess/rule/response/header_rule.rb +5 -5
  32. data/lib/contrast/agent/assess/rule/response/hsts_header_rule.rb +1 -1
  33. data/lib/contrast/agent/assess/rule/response/x_xss_protection_header_rule.rb +1 -1
  34. data/lib/contrast/agent/assess/tracker.rb +1 -7
  35. data/lib/contrast/agent/at_exit_hook.rb +1 -7
  36. data/lib/contrast/agent/excluder.rb +224 -0
  37. data/lib/contrast/agent/exclusion_matcher.rb +25 -7
  38. data/lib/contrast/agent/inventory/database_config.rb +18 -23
  39. data/lib/contrast/agent/middleware.rb +4 -5
  40. data/lib/contrast/agent/patching/policy/after_load_patcher.rb +6 -0
  41. data/lib/contrast/agent/protect/input_analyzer/input_analyzer.rb +146 -127
  42. data/lib/contrast/agent/protect/policy/applies_command_injection_rule.rb +4 -0
  43. data/lib/contrast/agent/protect/policy/applies_path_traversal_rule.rb +20 -0
  44. data/lib/contrast/agent/protect/policy/applies_sqli_rule.rb +1 -0
  45. data/lib/contrast/agent/protect/policy/rule_applicator.rb +1 -1
  46. data/lib/contrast/agent/protect/rule/base.rb +98 -66
  47. data/lib/contrast/agent/protect/rule/base_service.rb +49 -24
  48. data/lib/contrast/agent/protect/rule/bot_blocker/bot_blocker_input_classification.rb +98 -0
  49. data/lib/contrast/agent/protect/rule/bot_blocker.rb +81 -0
  50. data/lib/contrast/agent/protect/rule/cmd_injection.rb +30 -99
  51. data/lib/contrast/agent/protect/rule/cmdi/cmdi_backdoors.rb +132 -0
  52. data/lib/contrast/agent/protect/rule/cmdi/cmdi_base_rule.rb +169 -0
  53. data/lib/contrast/agent/protect/rule/cmdi/cmdi_chained_command.rb +69 -0
  54. data/lib/contrast/agent/protect/rule/cmdi/cmdi_dangerous_path.rb +68 -0
  55. data/lib/contrast/agent/protect/rule/cmdi/cmdi_input_classification.rb +2 -58
  56. data/lib/contrast/agent/protect/rule/default_scanner.rb +1 -1
  57. data/lib/contrast/agent/protect/rule/deserialization.rb +10 -19
  58. data/lib/contrast/agent/protect/rule/http_method_tampering/http_method_tampering_input_classification.rb +2 -2
  59. data/lib/contrast/agent/protect/rule/http_method_tampering.rb +0 -11
  60. data/lib/contrast/agent/protect/rule/no_sqli/no_sqli_input_classification.rb +29 -34
  61. data/lib/contrast/agent/protect/rule/no_sqli.rb +25 -18
  62. data/lib/contrast/agent/protect/rule/path_traversal/path_traversal_input_classification.rb +61 -0
  63. data/lib/contrast/agent/protect/rule/path_traversal/path_traversal_semantic_security_bypass.rb +114 -0
  64. data/lib/contrast/agent/protect/rule/path_traversal.rb +46 -18
  65. data/lib/contrast/agent/protect/rule/sql_sample_builder.rb +49 -29
  66. data/lib/contrast/agent/protect/rule/sqli/sqli_base_rule.rb +37 -0
  67. data/lib/contrast/agent/protect/rule/sqli/sqli_input_classification.rb +2 -62
  68. data/lib/contrast/agent/protect/rule/sqli/sqli_semantic/sqli_dangerous_functions.rb +67 -0
  69. data/lib/contrast/agent/protect/rule/sqli.rb +67 -22
  70. data/lib/contrast/agent/protect/rule/unsafe_file_upload/unsafe_file_upload_input_classification.rb +39 -63
  71. data/lib/contrast/agent/protect/rule/unsafe_file_upload.rb +6 -33
  72. data/lib/contrast/agent/protect/rule/xss/reflected_xss_input_classification.rb +58 -0
  73. data/lib/contrast/agent/protect/rule/xss.rb +14 -20
  74. data/lib/contrast/agent/protect/rule/xxe.rb +15 -30
  75. data/lib/contrast/agent/protect/rule.rb +3 -1
  76. data/lib/contrast/agent/reporting/attack_result/attack_result.rb +8 -0
  77. data/lib/contrast/agent/reporting/attack_result/rasp_rule_sample.rb +70 -36
  78. data/lib/contrast/agent/reporting/attack_result/response_type.rb +9 -9
  79. data/lib/contrast/agent/reporting/attack_result/user_input.rb +11 -0
  80. data/lib/contrast/agent/reporting/details/bot_blocker_details.rb +29 -0
  81. data/lib/contrast/agent/reporting/details/cmd_injection_details.rb +30 -0
  82. data/lib/contrast/agent/reporting/details/details.rb +18 -0
  83. data/lib/contrast/agent/reporting/details/http_method_tempering_details.rb +27 -0
  84. data/lib/contrast/agent/reporting/details/ip_denylist_details.rb +35 -0
  85. data/lib/contrast/agent/reporting/details/no_sqli_details.rb +36 -0
  86. data/lib/contrast/agent/reporting/details/path_traversal_details.rb +24 -0
  87. data/lib/contrast/agent/reporting/details/path_traversal_semantic_analysis_details.rb +32 -0
  88. data/lib/contrast/agent/reporting/details/protect_rule_details.rb +17 -0
  89. data/lib/contrast/agent/reporting/details/sqli_dangerous_functions.rb +22 -0
  90. data/lib/contrast/agent/reporting/details/sqli_details.rb +36 -0
  91. data/lib/contrast/agent/reporting/details/untrusted_deserialization_details.rb +27 -0
  92. data/lib/contrast/agent/reporting/details/virtual_patch_details.rb +30 -0
  93. data/lib/contrast/agent/reporting/details/xss_details.rb +33 -0
  94. data/lib/contrast/agent/reporting/details/xss_match.rb +30 -0
  95. data/lib/contrast/agent/reporting/details/xxe_details.rb +36 -0
  96. data/lib/contrast/agent/reporting/details/xxe_match.rb +25 -0
  97. data/lib/contrast/agent/reporting/details/xxe_wrapper.rb +25 -0
  98. data/lib/contrast/agent/reporting/input_analysis/details/bot_blocker_details.rb +27 -0
  99. data/lib/contrast/agent/reporting/input_analysis/details/protect_rule_details.rb +15 -0
  100. data/lib/contrast/agent/reporting/input_analysis/input_analysis.rb +1 -2
  101. data/lib/contrast/agent/reporting/input_analysis/input_analysis_result.rb +17 -3
  102. data/lib/contrast/agent/reporting/masker/masker.rb +80 -65
  103. data/lib/contrast/agent/reporting/masker/masker_utils.rb +1 -30
  104. data/lib/contrast/agent/reporting/reporter.rb +1 -15
  105. data/lib/contrast/agent/reporting/reporting_events/agent_startup.rb +2 -2
  106. data/lib/contrast/agent/reporting/reporting_events/application_activity.rb +84 -15
  107. data/lib/contrast/agent/reporting/reporting_events/application_defend_activity.rb +13 -25
  108. data/lib/contrast/agent/reporting/reporting_events/application_defend_attack_activity.rb +19 -24
  109. data/lib/contrast/agent/reporting/reporting_events/application_defend_attack_sample.rb +46 -126
  110. data/lib/contrast/agent/reporting/reporting_events/application_defend_attack_sample_activity.rb +5 -16
  111. data/lib/contrast/agent/reporting/reporting_events/application_defend_attacker_activity.rb +10 -18
  112. data/lib/contrast/agent/reporting/reporting_events/application_inventory_activity.rb +6 -14
  113. data/lib/contrast/agent/reporting/reporting_events/application_startup.rb +1 -1
  114. data/lib/contrast/agent/reporting/reporting_events/application_update.rb +0 -2
  115. data/lib/contrast/agent/reporting/reporting_events/architecture_component.rb +7 -22
  116. data/lib/contrast/agent/reporting/reporting_events/finding.rb +23 -53
  117. data/lib/contrast/agent/reporting/reporting_events/finding_event.rb +12 -9
  118. data/lib/contrast/agent/reporting/reporting_events/finding_event_signature.rb +1 -1
  119. data/lib/contrast/agent/reporting/reporting_events/finding_event_source.rb +23 -21
  120. data/lib/contrast/agent/reporting/reporting_events/finding_event_stack.rb +5 -18
  121. data/lib/contrast/agent/reporting/reporting_events/finding_event_taint_range.rb +1 -0
  122. data/lib/contrast/{api/decorators/trace_taint_range_tags.rb → agent/reporting/reporting_events/finding_event_taint_range_tags.rb} +7 -6
  123. data/lib/contrast/agent/reporting/reporting_events/finding_request.rb +40 -10
  124. data/lib/contrast/agent/reporting/reporting_events/library_discovery.rb +0 -1
  125. data/lib/contrast/agent/reporting/reporting_events/library_usage_observation.rb +1 -1
  126. data/lib/contrast/agent/reporting/reporting_events/observed_route.rb +2 -2
  127. data/lib/contrast/agent/reporting/reporting_events/poll.rb +1 -11
  128. data/lib/contrast/agent/reporting/reporting_events/preflight_message.rb +10 -14
  129. data/lib/contrast/agent/reporting/reporting_events/reporting_event.rb +11 -0
  130. data/lib/contrast/agent/reporting/reporting_events/route_coverage.rb +3 -1
  131. data/lib/contrast/agent/reporting/reporting_events/route_discovery.rb +12 -25
  132. data/lib/contrast/agent/reporting/reporting_events/route_discovery_observation.rb +8 -27
  133. data/lib/contrast/agent/reporting/reporting_utilities/audit.rb +3 -3
  134. data/lib/contrast/agent/reporting/reporting_utilities/build_preflight.rb +4 -7
  135. data/lib/contrast/agent/reporting/reporting_utilities/headers.rb +1 -1
  136. data/lib/contrast/agent/reporting/reporting_utilities/reporter_client.rb +2 -1
  137. data/lib/contrast/agent/reporting/reporting_utilities/reporter_client_utils.rb +3 -3
  138. data/lib/contrast/agent/reporting/reporting_utilities/response.rb +1 -1
  139. data/lib/contrast/agent/reporting/reporting_utilities/response_handler.rb +0 -3
  140. data/lib/contrast/agent/reporting/reporting_utilities/response_handler_utils.rb +1 -0
  141. data/lib/contrast/agent/reporting/settings/code_exclusion.rb +6 -1
  142. data/lib/contrast/agent/reporting/settings/exclusion_base.rb +18 -0
  143. data/lib/contrast/agent/reporting/settings/exclusions.rb +2 -1
  144. data/lib/contrast/agent/reporting/settings/input_exclusion.rb +9 -3
  145. data/lib/contrast/agent/reporting/settings/protect.rb +15 -15
  146. data/lib/contrast/agent/request.rb +4 -14
  147. data/lib/contrast/agent/request_context.rb +18 -24
  148. data/lib/contrast/agent/request_context_extend.rb +23 -164
  149. data/lib/contrast/agent/request_handler.rb +1 -11
  150. data/lib/contrast/agent/response.rb +0 -18
  151. data/lib/contrast/agent/telemetry/events/event.rb +1 -1
  152. data/lib/contrast/agent/telemetry/events/metric_event.rb +1 -1
  153. data/lib/contrast/agent/telemetry/events/startup_metrics_event.rb +3 -3
  154. data/lib/contrast/agent/thread_watcher.rb +3 -18
  155. data/lib/contrast/agent/version.rb +1 -1
  156. data/lib/contrast/agent.rb +0 -11
  157. data/lib/contrast/agent_lib/api/command_injection.rb +46 -0
  158. data/lib/contrast/agent_lib/api/init.rb +101 -0
  159. data/lib/contrast/agent_lib/api/input_tracing.rb +267 -0
  160. data/lib/contrast/agent_lib/api/method_tempering.rb +29 -0
  161. data/lib/contrast/agent_lib/api/panic.rb +87 -0
  162. data/lib/contrast/agent_lib/api/path_semantic_file_security_bypass.rb +40 -0
  163. data/lib/contrast/agent_lib/interface.rb +260 -0
  164. data/lib/contrast/agent_lib/interface_base.rb +118 -0
  165. data/lib/contrast/agent_lib/return_types/eval_result.rb +44 -0
  166. data/lib/contrast/agent_lib/test.rb +29 -0
  167. data/lib/contrast/api/communication/connection_status.rb +5 -5
  168. data/lib/contrast/components/agent.rb +13 -23
  169. data/lib/contrast/components/api.rb +10 -10
  170. data/lib/contrast/components/app_context.rb +9 -11
  171. data/lib/contrast/components/app_context_extend.rb +1 -26
  172. data/lib/contrast/components/assess.rb +92 -38
  173. data/lib/contrast/components/assess_rules.rb +36 -0
  174. data/lib/contrast/components/config.rb +49 -24
  175. data/lib/contrast/components/heap_dump.rb +1 -1
  176. data/lib/contrast/components/protect.rb +9 -6
  177. data/lib/contrast/components/ruby_component.rb +81 -0
  178. data/lib/contrast/components/sampling.rb +1 -1
  179. data/lib/contrast/components/security_logger.rb +23 -0
  180. data/lib/contrast/components/settings.rb +41 -85
  181. data/lib/contrast/config/base_configuration.rb +1 -1
  182. data/lib/contrast/config/protect_rule_configuration.rb +7 -7
  183. data/lib/contrast/config/protect_rules_configuration.rb +24 -48
  184. data/lib/contrast/config/server_configuration.rb +1 -1
  185. data/lib/contrast/config.rb +0 -6
  186. data/lib/contrast/configuration.rb +73 -18
  187. data/lib/contrast/extension/assess/array.rb +9 -0
  188. data/lib/contrast/extension/assess/exec_trigger.rb +3 -1
  189. data/lib/contrast/extension/assess/marshal.rb +3 -2
  190. data/lib/contrast/extension/assess/string.rb +0 -1
  191. data/lib/contrast/extension/delegator.rb +2 -0
  192. data/lib/contrast/extension/extension.rb +1 -1
  193. data/lib/contrast/framework/base_support.rb +0 -5
  194. data/lib/contrast/framework/grape/support.rb +1 -23
  195. data/lib/contrast/framework/manager.rb +3 -11
  196. data/lib/contrast/framework/rails/patch/action_controller_live_buffer.rb +1 -6
  197. data/lib/contrast/framework/rails/railtie.rb +0 -1
  198. data/lib/contrast/framework/rails/support.rb +5 -59
  199. data/lib/contrast/framework/sinatra/support.rb +2 -21
  200. data/lib/contrast/logger/cef_log.rb +21 -3
  201. data/lib/contrast/logger/log.rb +1 -11
  202. data/lib/contrast/tasks/config.rb +5 -10
  203. data/lib/contrast/utils/assess/event_limit_utils.rb +28 -12
  204. data/lib/contrast/utils/assess/trigger_method_utils.rb +10 -18
  205. data/lib/contrast/utils/duck_utils.rb +1 -0
  206. data/lib/contrast/utils/findings.rb +6 -5
  207. data/lib/contrast/utils/hash_digest.rb +9 -24
  208. data/lib/contrast/utils/hash_digest_extend.rb +6 -6
  209. data/lib/contrast/utils/input_classification_base.rb +156 -0
  210. data/lib/contrast/utils/invalid_configuration_util.rb +21 -58
  211. data/lib/contrast/utils/log_utils.rb +47 -17
  212. data/lib/contrast/utils/net_http_base.rb +2 -2
  213. data/lib/contrast/utils/os.rb +0 -20
  214. data/lib/contrast/utils/patching/policy/patch_utils.rb +3 -2
  215. data/lib/contrast/utils/response_utils.rb +0 -16
  216. data/lib/contrast/utils/stack_trace_utils.rb +3 -40
  217. data/lib/contrast/utils/string_utils.rb +19 -7
  218. data/lib/contrast/utils/telemetry_client.rb +13 -7
  219. data/lib/contrast.rb +7 -13
  220. data/resources/protect/policy.json +1 -2
  221. data/ruby-agent.gemspec +2 -5
  222. metadata +78 -137
  223. data/exe/contrast_service +0 -23
  224. data/lib/contrast/agent/protect/rule/cmdi/cmdi_worth_watching.rb +0 -64
  225. data/lib/contrast/agent/protect/rule/sqli/sqli_worth_watching.rb +0 -118
  226. data/lib/contrast/agent/protect/rule/unsafe_file_upload/unsafe_file_upload_matcher.rb +0 -45
  227. data/lib/contrast/agent/reaction_processor.rb +0 -47
  228. data/lib/contrast/agent/reporting/reporting_events/trace_event_source.rb +0 -30
  229. data/lib/contrast/agent/reporting/reporting_utilities/dtm_message.rb +0 -43
  230. data/lib/contrast/agent/service_heartbeat.rb +0 -35
  231. data/lib/contrast/api/communication/messaging_queue.rb +0 -129
  232. data/lib/contrast/api/communication/response_processor.rb +0 -90
  233. data/lib/contrast/api/communication/service_lifecycle.rb +0 -77
  234. data/lib/contrast/api/communication/socket.rb +0 -44
  235. data/lib/contrast/api/communication/socket_client.rb +0 -130
  236. data/lib/contrast/api/communication/speedracer.rb +0 -142
  237. data/lib/contrast/api/communication/tcp_socket.rb +0 -32
  238. data/lib/contrast/api/communication/unix_socket.rb +0 -28
  239. data/lib/contrast/api/communication.rb +0 -20
  240. data/lib/contrast/api/decorators/activity.rb +0 -33
  241. data/lib/contrast/api/decorators/address.rb +0 -59
  242. data/lib/contrast/api/decorators/agent_startup.rb +0 -57
  243. data/lib/contrast/api/decorators/application_settings.rb +0 -42
  244. data/lib/contrast/api/decorators/application_startup.rb +0 -56
  245. data/lib/contrast/api/decorators/architecture_component.rb +0 -36
  246. data/lib/contrast/api/decorators/bot_blocker.rb +0 -37
  247. data/lib/contrast/api/decorators/finding.rb +0 -29
  248. data/lib/contrast/api/decorators/http_request.rb +0 -137
  249. data/lib/contrast/api/decorators/input_analysis.rb +0 -18
  250. data/lib/contrast/api/decorators/instrumentation_mode.rb +0 -35
  251. data/lib/contrast/api/decorators/ip_denylist.rb +0 -37
  252. data/lib/contrast/api/decorators/message.rb +0 -71
  253. data/lib/contrast/api/decorators/rasp_rule_sample.rb +0 -58
  254. data/lib/contrast/api/decorators/response_type.rb +0 -17
  255. data/lib/contrast/api/decorators/route_coverage.rb +0 -91
  256. data/lib/contrast/api/decorators/server_features.rb +0 -25
  257. data/lib/contrast/api/decorators/trace_event.rb +0 -120
  258. data/lib/contrast/api/decorators/trace_event_object.rb +0 -63
  259. data/lib/contrast/api/decorators/trace_event_signature.rb +0 -69
  260. data/lib/contrast/api/decorators/trace_taint_range.rb +0 -52
  261. data/lib/contrast/api/decorators/user_input.rb +0 -51
  262. data/lib/contrast/api/decorators/virtual_patch.rb +0 -34
  263. data/lib/contrast/api/decorators.rb +0 -28
  264. data/lib/contrast/api/dtm.pb.rb +0 -852
  265. data/lib/contrast/api/settings.pb.rb +0 -500
  266. data/lib/contrast/api.rb +0 -16
  267. data/lib/contrast/components/contrast_service.rb +0 -88
  268. data/lib/contrast/config/assess_configuration.rb +0 -93
  269. data/lib/contrast/config/assess_rules_configuration.rb +0 -32
  270. data/lib/contrast/config/root_configuration.rb +0 -90
  271. data/lib/contrast/config/ruby_configuration.rb +0 -81
  272. data/lib/contrast/config/service_configuration.rb +0 -49
  273. data/lib/contrast/tasks/service.rb +0 -84
  274. data/lib/contrast/utils/input_classification.rb +0 -73
  275. data/lib/contrast/utils/preflight_util.rb +0 -13
  276. data/lib/protobuf/code_generator.rb +0 -129
  277. data/lib/protobuf/decoder.rb +0 -28
  278. data/lib/protobuf/deprecation.rb +0 -117
  279. data/lib/protobuf/descriptors/google/protobuf/compiler/plugin.pb.rb +0 -79
  280. data/lib/protobuf/descriptors/google/protobuf/descriptor.pb.rb +0 -360
  281. data/lib/protobuf/descriptors.rb +0 -3
  282. data/lib/protobuf/encoder.rb +0 -11
  283. data/lib/protobuf/enum.rb +0 -365
  284. data/lib/protobuf/exceptions.rb +0 -9
  285. data/lib/protobuf/field/base_field.rb +0 -380
  286. data/lib/protobuf/field/base_field_object_definitions.rb +0 -504
  287. data/lib/protobuf/field/bool_field.rb +0 -64
  288. data/lib/protobuf/field/bytes_field.rb +0 -67
  289. data/lib/protobuf/field/double_field.rb +0 -25
  290. data/lib/protobuf/field/enum_field.rb +0 -56
  291. data/lib/protobuf/field/field_array.rb +0 -102
  292. data/lib/protobuf/field/field_hash.rb +0 -122
  293. data/lib/protobuf/field/fixed32_field.rb +0 -25
  294. data/lib/protobuf/field/fixed64_field.rb +0 -28
  295. data/lib/protobuf/field/float_field.rb +0 -43
  296. data/lib/protobuf/field/int32_field.rb +0 -21
  297. data/lib/protobuf/field/int64_field.rb +0 -34
  298. data/lib/protobuf/field/integer_field.rb +0 -23
  299. data/lib/protobuf/field/message_field.rb +0 -51
  300. data/lib/protobuf/field/sfixed32_field.rb +0 -27
  301. data/lib/protobuf/field/sfixed64_field.rb +0 -28
  302. data/lib/protobuf/field/signed_integer_field.rb +0 -29
  303. data/lib/protobuf/field/sint32_field.rb +0 -21
  304. data/lib/protobuf/field/sint64_field.rb +0 -21
  305. data/lib/protobuf/field/string_field.rb +0 -51
  306. data/lib/protobuf/field/uint32_field.rb +0 -21
  307. data/lib/protobuf/field/uint64_field.rb +0 -21
  308. data/lib/protobuf/field/varint_field.rb +0 -77
  309. data/lib/protobuf/field.rb +0 -74
  310. data/lib/protobuf/generators/base.rb +0 -85
  311. data/lib/protobuf/generators/enum_generator.rb +0 -39
  312. data/lib/protobuf/generators/extension_generator.rb +0 -27
  313. data/lib/protobuf/generators/field_generator.rb +0 -193
  314. data/lib/protobuf/generators/file_generator.rb +0 -262
  315. data/lib/protobuf/generators/group_generator.rb +0 -122
  316. data/lib/protobuf/generators/message_generator.rb +0 -104
  317. data/lib/protobuf/generators/option_generator.rb +0 -17
  318. data/lib/protobuf/generators/printable.rb +0 -160
  319. data/lib/protobuf/generators/service_generator.rb +0 -50
  320. data/lib/protobuf/lifecycle.rb +0 -33
  321. data/lib/protobuf/logging.rb +0 -39
  322. data/lib/protobuf/message/fields.rb +0 -233
  323. data/lib/protobuf/message/serialization.rb +0 -85
  324. data/lib/protobuf/message.rb +0 -241
  325. data/lib/protobuf/optionable.rb +0 -72
  326. data/lib/protobuf/tasks/compile.rake +0 -80
  327. data/lib/protobuf/tasks.rb +0 -1
  328. data/lib/protobuf/varint.rb +0 -20
  329. data/lib/protobuf/varint_pure.rb +0 -31
  330. data/lib/protobuf/version.rb +0 -3
  331. data/lib/protobuf/wire_type.rb +0 -10
  332. data/lib/protobuf.rb +0 -91
  333. data/proto/dynamic_discovery.proto +0 -46
  334. data/proto/google/protobuf/compiler/plugin.proto +0 -183
  335. data/proto/google/protobuf/descriptor.proto +0 -911
  336. data/proto/rpc.proto +0 -71
  337. data/service_executables/.gitkeep +0 -0
  338. data/service_executables/VERSION +0 -1
  339. data/service_executables/linux/contrast-service +0 -0
  340. data/service_executables/mac/contrast-service +0 -0
@@ -1,137 +0,0 @@
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/api/dtm.pb'
5
- require 'contrast/api/decorators/address'
6
- require 'contrast/components/scope'
7
- require 'contrast/utils/string_utils'
8
- require 'contrast/utils/timer'
9
-
10
- module Contrast
11
- module Api
12
- module Decorators
13
- # Used to decorate the {Contrast::Api::Dtm::HttpRequest} protobuf model
14
- # so it can own some of the data massaging required for Request dtm. Only
15
- # works as an extension of that class.
16
- module HttpRequest
17
- OMITTED_BODY = '{{body-omitted-by-contrast}}'
18
-
19
- def self.included klass
20
- klass.extend(ClassMethods)
21
- end
22
-
23
- # Append the sender and receiver to self.
24
- #
25
- # @param request [Contrast::Agent::Request]
26
- def append_addresses request
27
- self.sender = Contrast::Api::Dtm::Address.build_sender(request)
28
- self.receiver = Contrast::Api::Dtm::Address.build_receiver
29
- end
30
-
31
- # Append the connection to self.
32
- #
33
- # @param request [Contrast::Agent::Request]
34
- def append_connection request
35
- self.protocol = Contrast::Utils::StringUtils.force_utf8(request.scheme)
36
- self.version = Contrast::Utils::StringUtils.force_utf8(request.version)
37
- self.method = Contrast::Utils::StringUtils.force_utf8(request.request_method)
38
- self.raw = Contrast::Utils::StringUtils.force_utf8(request.rack_request.path_info)
39
- self.uri = Contrast::Utils::StringUtils.force_utf8(request.rack_request.path_info)
40
- end
41
-
42
- # Append the parameters to self, using the preferred
43
- # normalized_request_params field.
44
- #
45
- # @param request [Contrast::Agent::Request]
46
- def append_params request
47
- request.parameters.each do |k, v|
48
- append_pair(normalized_request_params, k, v)
49
- end
50
- end
51
-
52
- # Append the normalized headers to self, using the preferred
53
- # request_headers field.
54
- #
55
- # @param request [Contrast::Agent::Request]
56
- def append_headers request
57
- request.headers.each do |k, v|
58
- next unless k && v
59
- next if v.is_a?(Hash)
60
-
61
- key = Contrast::Utils::StringUtils.force_utf8(k)
62
- val = Contrast::Utils::StringUtils.force_utf8(v)
63
- request_headers[key] = val
64
- end
65
-
66
- request.cookies.each do |k, v|
67
- append_pair(normalized_cookies, k, v)
68
- end
69
- end
70
-
71
- # Append the body to self, using the preferred
72
- # request_body_binary field.
73
- #
74
- # @param request [Contrast::Agent::Request]
75
- def append_body request
76
- self.document_type = Contrast::Utils::StringUtils.force_utf8(request.document_type)
77
-
78
- self.request_body_binary = if omit_body?(request)
79
- OMITTED_BODY
80
- else
81
- Contrast::Utils::StringUtils.force_utf8(request.body)
82
- end
83
- return if request.file_names.empty?
84
-
85
- request.file_names.each do |name, filename|
86
- pair = Contrast::Api::Dtm::SimplePair.new
87
- pair.key = Contrast::Utils::StringUtils.force_utf8(name)
88
- pair.value = Contrast::Utils::StringUtils.force_utf8(filename)
89
- multipart_headers << pair
90
- end
91
- end
92
-
93
- def omit_body? request
94
- return true if ::Contrast::AGENT.omit_body?
95
- return false if request.document_type != :NORMAL
96
-
97
- request.media_type&.include?('multipart/form-data')
98
- end
99
-
100
- def append_pair map, key, value
101
- return unless key && value
102
- return if value.is_a?(Hash)
103
-
104
- safe_key = Contrast::Utils::StringUtils.force_utf8(key)
105
- map[safe_key] ||= Contrast::Api::Dtm::Pair.new
106
- map[safe_key].key = safe_key
107
- map[safe_key].values << Contrast::Utils::StringUtils.force_utf8(value)
108
- end
109
-
110
- # Used to add class methods to the ApplicationUpdate class on inclusion of the decorator
111
- module ClassMethods
112
- include Contrast::Components::Scope::InstanceMethods
113
-
114
- # Convert our Request into a DTM
115
- # @param request [Contrast::Agent::Request]
116
- # @return [Contrast::Api::Dtm::HttpRequest]
117
- def build request
118
- with_contrast_scope do
119
- msg = new
120
- msg.uuid = Contrast::Utils::StringUtils.force_utf8(__id__)
121
- msg.timestamp_ms = Contrast::Utils::Timer.now_ms.to_i
122
-
123
- msg.append_addresses(request)
124
- msg.append_connection(request)
125
- msg.append_params(request)
126
- msg.append_headers(request)
127
- msg.append_body(request)
128
- msg
129
- end
130
- end
131
- end
132
- end
133
- end
134
- end
135
- end
136
-
137
- Contrast::Api::Dtm::HttpRequest.include(Contrast::Api::Decorators::HttpRequest)
@@ -1,18 +0,0 @@
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/string_utils'
5
-
6
- module Contrast
7
- module Api
8
- module Decorators
9
- # Used to decorate the {Contrast::Api::Settings::InputAnalysis} protobuf
10
- # model so it can own the request which its data is for.
11
- module InputAnalysis
12
- attr_accessor :request
13
- end
14
- end
15
- end
16
- end
17
-
18
- Contrast::Api::Settings::InputAnalysis.include(Contrast::Api::Decorators::InputAnalysis)
@@ -1,35 +0,0 @@
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/api/dtm.pb'
5
- require 'contrast/components/base'
6
-
7
- module Contrast
8
- module Api
9
- module Decorators
10
- # Used to decorate the InstrumentationMode protobuf model to handle reporting Agent process start
11
- module InstrumentationMode
12
- include Contrast::Components::ComponentBase
13
-
14
- def self.included klass
15
- klass.extend(ClassMethods)
16
- end
17
-
18
- # Used to add class methods to the AgentStartup class on inclusion of the decorator
19
- module ClassMethods
20
- # Return a new DTM with the values from the configuration
21
- #
22
- # @return [Contrast::Api::Dtm::InstrumentationMode]
23
- def build
24
- msg = new
25
- msg.assess = ::Contrast::ASSESS.enabled?
26
- msg.protect = ::Contrast::PROTECT.enabled?
27
- msg
28
- end
29
- end
30
- end
31
- end
32
- end
33
- end
34
-
35
- Contrast::Api::Dtm::InstrumentationMode.include(Contrast::Api::Decorators::InstrumentationMode)
@@ -1,37 +0,0 @@
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/api/dtm.pb'
5
- require 'contrast/utils/string_utils'
6
- require 'contrast/components/base'
7
-
8
- module Contrast
9
- module Api
10
- module Decorators
11
- # Used to decorate the {Contrast::Api::Dtm::IpDenylistDetails} protobuf
12
- # model so it can own the request which its data is for.
13
- module IpDenylistDetails
14
- def self.included klass
15
- klass.extend(ClassMethods)
16
- end
17
-
18
- # Used to add class methods to the AgentStartup class on inclusion of the decorator
19
- module ClassMethods
20
- def build
21
- new
22
- end
23
-
24
- # @param result [Contrast::Api::Dtm::IpDenylistDetails]
25
- def to_controlled_hash result
26
- {
27
- ip: result.ip,
28
- uuid: result.uuid
29
- }
30
- end
31
- end
32
- end
33
- end
34
- end
35
- end
36
-
37
- Contrast::Api::Dtm::IpDenylistDetails.include(Contrast::Api::Decorators::IpDenylistDetails)
@@ -1,71 +0,0 @@
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/utils/string_utils'
6
- require 'contrast/components/logger'
7
-
8
- module Contrast
9
- module Api
10
- module Decorators
11
- # Used to decorate the {Contrast::Api::Dtm::Message} protobuf model so it
12
- # can handle instance data massaging.
13
- module Message
14
- include Contrast::Components::Logger::InstanceMethods
15
-
16
- def self.included klass
17
- klass.extend(ClassMethods)
18
- end
19
-
20
- def append_event event
21
- case event
22
- when Contrast::Api::Dtm::AgentStartup
23
- self.agent_startup = event
24
- when Contrast::Api::Dtm::ApplicationCreate
25
- self.application_create = event
26
- when Contrast::Api::Dtm::Activity
27
- self.activity = event
28
- when Contrast::Api::Dtm::HttpRequest
29
- self.prefilter = event
30
- when Contrast::Api::Dtm::HttpResponse
31
- self.postfilter = event
32
- when Contrast::Api::Dtm::Poll
33
- self.poll = event
34
- else
35
- logger.error('Unknown event type received. Unsure how to send.', event_type: event.cs__class.cs__name)
36
- return
37
- end
38
- logger.debug('Wrapping event in message',
39
- msg_id: __id__,
40
- p_id: pid,
41
- msg_count: message_count,
42
- event_id: event.__id__,
43
- event_type: event.cs__class.cs__name)
44
- end
45
-
46
- # Used to add class methods to the ApplicationUpdate class on inclusion of the decorator
47
- module ClassMethods
48
- @@message_total = 0 # rubocop:disable Style/ClassVars
49
- def message_count
50
- @@message_total += 1 # rubocop:disable Style/ClassVars
51
- end
52
-
53
- def build event
54
- msg = new
55
- msg.app_name = ::Contrast::APP_CONTEXT.name # rubocop:disable Security/Module/Name
56
- msg.app_path = ::Contrast::APP_CONTEXT.path
57
- msg.app_language = Contrast::Utils::ObjectShare::RUBY
58
- msg.client_id = ::Contrast::APP_CONTEXT.client_id
59
- msg.message_count = message_count
60
- msg.pid = ::Contrast::APP_CONTEXT.pid
61
- msg.ppid = ::Contrast::APP_CONTEXT.ppid
62
- msg.append_event(event)
63
- msg
64
- end
65
- end
66
- end
67
- end
68
- end
69
- end
70
-
71
- Contrast::Api::Dtm::Message.include(Contrast::Api::Decorators::Message)
@@ -1,58 +0,0 @@
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
- module Contrast
5
- module Api
6
- module Decorators
7
- # Used to decorate the {Contrast::Api::Dtm::RaspRuleSample} protobuf
8
- # model.
9
- module RaspRuleSample
10
- def self.included klass
11
- klass.extend(ClassMethods)
12
- end
13
-
14
- # Used to add class methods to RaspRuleSample
15
- module ClassMethods
16
- def build context, ia_result
17
- sample = new
18
- sample.timestamp_ms = context.timer.start_ms
19
- sample.user_input = Contrast::Api::Dtm::UserInput.build_from_ia_result(ia_result)
20
- sample.user_input.document_type = Contrast::Utils::StringUtils.force_utf8(context.request.document_type)
21
- sample
22
- end
23
-
24
- # @param result [Contrast::Api::Dtm::RaspRuleSample]
25
- def to_controlled_hash result
26
- {
27
- timestamp: Time.at(result.timestamp_ms).iso8601,
28
- user_input: result.user_input,
29
- brute_force: result.brute_force,
30
- bot_blocker: result.bot_blocker,
31
- cmdi: result.cmdi,
32
- csrf: result.csrf,
33
- cve: result.cve,
34
- untrusted_deserialization: result.untrusted_deserialization,
35
- el_injection: result.el_injection,
36
- mark_of_the_beast: result.mark_of_the_beast,
37
- padding_oracle: result.padding_oracle,
38
- path_traversal: result.path_traversal,
39
- re_dos: result.re_dos,
40
- sqli: result.sqli,
41
- ssrf: result.ssrf,
42
- virtual_patch: result.virtual_patch,
43
- xss: result.xss,
44
- xxe: result.xxe,
45
- no_sqli: result.no_sqli,
46
- method_tampering: result.method_tampering,
47
- path_traversal_semantic: result.path_traversal_semantic,
48
- ssjs: result.ssjs,
49
- ip_denylist: result.ip_denylist
50
- }
51
- end
52
- end
53
- end
54
- end
55
- end
56
- end
57
-
58
- Contrast::Api::Dtm::RaspRuleSample.include(Contrast::Api::Decorators::RaspRuleSample)
@@ -1,17 +0,0 @@
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/api/dtm.pb'
5
- require 'protobuf/message'
6
-
7
- module Contrast
8
- module Api
9
- module Dtm
10
- class AttackResult < ::Protobuf::Message
11
- class ResponseType < ::Protobuf::Enum
12
- define :SUSPICIOUS, 6
13
- end
14
- end
15
- end
16
- end
17
- end
@@ -1,91 +0,0 @@
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/string_utils'
5
-
6
- module Contrast
7
- module Api
8
- module Decorators
9
- # Used to decorate the {Contrast::Api::Dtm::RouteCoverage} protobuf model
10
- # to handle conversion between framework route classes and the dtm.
11
- module RouteCoverage
12
- def self.included klass
13
- klass.extend(ClassMethods)
14
- end
15
-
16
- # Class methods for RouteCoverage
17
- module ClassMethods
18
- def source_or_string obj
19
- if obj.cs__is_a?(Regexp)
20
- obj.source
21
- elsif obj.cs__respond_to?(:safe_string)
22
- obj.safe_string
23
- else
24
- obj.to_s
25
- end
26
- end
27
-
28
- # Convert ActionDispatch::Journey::Route to Contrast::Api::Dtm::RouteCoverage
29
- #
30
- # @param journey_obj [ActionDispatch::Journey::Route] a rails route
31
- # @param url [String, nil] use url from string instead of journey object.
32
- # @return [Contrast::Api::Dtm::RouteCoverage]
33
- def from_action_dispatch_journey journey_obj, url = nil
34
- msg = new
35
- msg.route = "#{ journey_obj.defaults[:controller] }##{ journey_obj.defaults[:action] }"
36
-
37
- verb = source_or_string(journey_obj.verb)
38
- msg.verb = Contrast::Utils::StringUtils.force_utf8(verb)
39
-
40
- url ||= source_or_string(journey_obj.path.spec)
41
- msg.url = Contrast::Utils::StringUtils.force_utf8(url)
42
- msg
43
- end
44
-
45
- # Convert Sinatra route data to dtm message.
46
- #
47
- # @param controller [::Sinatra::Base] the route's final controller.
48
- # @param method [String] GET, PUT, POST, etc...
49
- # @param pattern [::Mustermann::Sinatra] the pattern that was matched in routing.
50
- # @param url [String, nil] use url from string instead matched pattern.
51
- # @return [Contrast::Api::Dtm::RouteCoverage]
52
- def from_sinatra_route controller, method, pattern, url = nil
53
- safe_pattern = source_or_string(pattern)
54
- safe_url = source_or_string(url || pattern)
55
-
56
- msg = new
57
- msg.route = "#{ controller }##{ method } #{ safe_pattern }"
58
- msg.verb = Contrast::Utils::StringUtils.force_utf8(method)
59
- msg.url = Contrast::Utils::StringUtils.force_utf8(safe_url)
60
- msg
61
- end
62
-
63
- # Convert Grape route data to dtm message.
64
- #
65
- # @param controller [::Grape::API] the route's final controller.
66
- # @param method [String] GET, PUT, POST, etc...
67
- # @param url [String, nil] use url from string instead matched pattern.
68
- # @param pattern [String, Grape::Router::Route] the pattern that was matched in routing.
69
- # @return [Contrast::Api::Dtm::RouteCoverage]
70
- def from_grape_controller controller, method, pattern, url = nil
71
- if pattern.cs__is_a?(Grape::Router::Route)
72
- safe_pattern = pattern.pattern&.path&.to_s
73
- safe_url = source_or_string(url || safe_pattern)
74
- else
75
- safe_pattern = source_or_string(pattern)
76
- safe_url = source_or_string(url || pattern)
77
- end
78
-
79
- msg = new
80
- msg.route = "#{ controller }##{ method } #{ safe_pattern }"
81
- msg.verb = Contrast::Utils::StringUtils.force_utf8(method)
82
- msg.url = Contrast::Utils::StringUtils.force_utf8(safe_url)
83
- msg
84
- end
85
- end
86
- end
87
- end
88
- end
89
- end
90
-
91
- Contrast::Api::Dtm::RouteCoverage.include(Contrast::Api::Decorators::RouteCoverage)
@@ -1,25 +0,0 @@
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/string_utils'
5
-
6
- module Contrast
7
- module Api
8
- module Decorators
9
- # Used to decorate the {Contrast::Api::Settings::ServerFeatures} protobuf
10
- # model to handle setting translation to boolean values.
11
- module ServerFeatures
12
- # 'defend' is legacy for 'protect' in the dtm.
13
- def protect_enabled?
14
- !!defend.enabled
15
- end
16
-
17
- def assess_enabled?
18
- !!assess.enabled
19
- end
20
- end
21
- end
22
- end
23
- end
24
-
25
- Contrast::Api::Settings::ServerFeatures.include(Contrast::Api::Decorators::ServerFeatures)
@@ -1,120 +0,0 @@
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/string_utils'
5
-
6
- module Contrast
7
- module Api
8
- module Decorators
9
- # Used to decorate the {Contrast::Api::Dtm::TraceEvent} protobuf model to
10
- # convert our {Contrast::Agent::AssessContrastEvent} to the dtm.
11
- module TraceEvent
12
- def self.included klass
13
- klass.extend(ClassMethods)
14
- end
15
-
16
- # The TeamServer uses the Event's type and action to render it in the Details page. These values control the
17
- # left-hand "What happened" column and the data shown in the right-hand data
18
- #
19
- # @param contrast_event [Contrast::Agent::Assess::ContrastEvent]
20
- # @return [Contrast::Api::Dtm::TraceEvent]
21
- def build_display_params! contrast_event
22
- self.type = contrast_event.policy_node.node_type
23
- self.action = contrast_event.policy_node.build_action
24
- self
25
- end
26
-
27
- # The TeamServer uses the Event's representation of the data to render the actual data used in the dataflow on
28
- # the Details page.
29
- #
30
- # @param contrast_event [Contrast::Agent::Assess::ContrastEvent]
31
- # @return [Contrast::Api::Dtm::TraceEvent]
32
- def build_dataflow! contrast_event
33
- # Figure out what the target of this event was. This can't be pulled into the decorator because SourceEvent
34
- # has a custom impl :/
35
- taint_target = contrast_event.determine_taint_target(self)
36
- truncate_obj = Contrast::Utils::ObjectShare::OBJECT_KEY != taint_target
37
- self.object = Contrast::Api::Dtm::TraceEventObject.build(contrast_event.object, truncate_obj)
38
- truncate_ret = Contrast::Utils::ObjectShare::RETURN_KEY != taint_target
39
- self.ret = Contrast::Api::Dtm::TraceEventObject.build(contrast_event.ret, truncate_ret)
40
- build_event_args!(contrast_event, taint_target)
41
- build_taint_ranges!(contrast_event)
42
- self
43
- end
44
-
45
- # Wrapper around build_event_object for the args array. Handles
46
- # tainting the correct argument.
47
- # @return [Contrast::Api::Dtm::TraceEvent]
48
- def build_event_args! contrast_event, taint_target
49
- contrast_event.args.each_index do |idx|
50
- truncate_arg = taint_target != idx
51
- event_arg = Contrast::Api::Dtm::TraceEventObject.build(contrast_event.args[idx], truncate_arg)
52
- args << event_arg
53
- end
54
- self
55
- end
56
-
57
- # TeamServer only supports one object's taint ranges at a time. We keep
58
- # those tags on the event directly, so we just need to convert them to
59
- # their DTM form in order to report this.
60
- #
61
- # @param contrast_event [Contrast::Agent::AssessContrastEvent]
62
- # @return [Contrast::Api::Dtm::TraceEvent]
63
- def build_taint_ranges! contrast_event
64
- # If there's no taint_target, this isn't a dataflow trace, but a
65
- # trigger one
66
- return self unless contrast_event&.tags
67
-
68
- self.taint_ranges += Contrast::Api::Dtm::TraceTaintRange.build_for_event(contrast_event.tags)
69
- self
70
- end
71
-
72
- # For each Parent in the ContrastEvent, capture its id and report it to TeamServer.
73
- #
74
- # @param contrast_event [Contrast::Agent::AssessContrastEvent]
75
- # @return [Contrast::Api::Dtm::TraceEvent]
76
- def build_parent_ids! contrast_event
77
- contrast_event&.parent_events&.each do |event|
78
- next unless event
79
-
80
- parent = Contrast::Api::Dtm::ParentObjectId.new
81
- parent.id = event.event_id.to_i
82
- parent_object_ids << parent
83
- end
84
- self
85
- end
86
-
87
- # Convert the caller into the Stack DTM TeamServer consumes
88
- #
89
- # @param contrast_event [Contrast::Agent::AssessContrastEvent]
90
- # @return [Contrast::Api::Dtm::TraceEvent]
91
- def build_stack! contrast_event
92
- # We delayed doing this as long as possible b/c it's expensive
93
- stack_dtms = Contrast::Utils::StackTraceUtils.build_assess_stack_array(contrast_event.stack_trace)
94
- self.stack += stack_dtms
95
- self
96
- end
97
-
98
- # Class methods for TraceEvent
99
- module ClassMethods
100
- def build contrast_event
101
- event_dtm = new
102
- event_dtm.build_display_params!(contrast_event)
103
- event_dtm.build_dataflow!(contrast_event)
104
- event_dtm.build_stack!(contrast_event)
105
- event_dtm.timestamp_ms = contrast_event.time.to_i
106
- event_dtm.thread = Contrast::Utils::StringUtils.force_utf8(contrast_event.thread)
107
- event_dtm.build_parent_ids!(contrast_event)
108
- event_dtm.object_id = contrast_event.event_id.to_i
109
- event_dtm.signature = Contrast::Api::Dtm::TraceEventSignature.build(contrast_event.ret,
110
- contrast_event.policy_node,
111
- contrast_event.args)
112
- event_dtm
113
- end
114
- end
115
- end
116
- end
117
- end
118
- end
119
-
120
- Contrast::Api::Dtm::TraceEvent.include(Contrast::Api::Decorators::TraceEvent)
@@ -1,63 +0,0 @@
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/string_utils'
5
- require 'contrast/utils/assess/tracking_util'
6
- require 'base64'
7
- require 'contrast/components/scope'
8
-
9
- module Contrast
10
- module Api
11
- module Decorators
12
- # Used to decorate the {Contrast::Api::Dtm::TraceEventObject} protobuf
13
- # model.
14
- module TraceEventObject
15
- def self.included klass
16
- klass.extend(ClassMethods)
17
- end
18
-
19
- # Class methods for TraceEventObject
20
- module ClassMethods
21
- include Contrast::Components::Scope::InstanceMethods
22
- # Build the event object. We were originally going to include taint on
23
- # each one, but TS doesn't accept / use that, so it is a waste of time.
24
- #
25
- # We'll truncate any object that isn't important to the taint ranges of
26
- # this event, so that we don't murder TeamServer by, for instance,
27
- # hypothetically sending the entire rendered HTML page >_> <_< >_>
28
- ELLIPSIS = '...'
29
- UNTRUNCATED_PORTION_LENGTH = 25
30
- TRUNCATION_LENGTH = (UNTRUNCATED_PORTION_LENGTH * 2) + ELLIPSIS.length
31
-
32
- # Convert the given Object into a Contrast::Api::Dtm::TraceEventObject
33
- #
34
- # @param contrast_object [Contrast::Agent::Assess::ContrastObject, nil]
35
- # the thing to convert, if any
36
- # @param truncate [Boolean] if the converted object can/should be
37
- # truncated.
38
- # @return [Contrast::Api::Dtm::TraceEventObject]
39
- def build contrast_object, truncate
40
- event_object = new
41
- with_contrast_scope do
42
- obj_string = Contrast::Utils::StringUtils.force_utf8(contrast_object&.object)
43
- obj_string = truncate(obj_string) if truncate && obj_string.length > TRUNCATION_LENGTH
44
- event_object.value = Base64.encode64(obj_string)
45
- event_object.tracked = contrast_object&.tracked?
46
- end
47
- event_object
48
- end
49
-
50
- def truncate obj_string
51
- tmp = []
52
- tmp << obj_string[0, UNTRUNCATED_PORTION_LENGTH]
53
- tmp << ELLIPSIS
54
- tmp << obj_string[obj_string.length - UNTRUNCATED_PORTION_LENGTH, UNTRUNCATED_PORTION_LENGTH]
55
- tmp.join
56
- end
57
- end
58
- end
59
- end
60
- end
61
- end
62
-
63
- Contrast::Api::Dtm::TraceEventObject.include(Contrast::Api::Decorators::TraceEventObject)