pwn 0.5.442 → 0.5.443

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (55) hide show
  1. checksums.yaml +4 -4
  2. data/Gemfile +1 -1
  3. data/README.md +3 -3
  4. data/lib/pwn/sast/amqp_connect_as_guest.rb +12 -89
  5. data/lib/pwn/sast/apache_file_system_util_api.rb +11 -91
  6. data/lib/pwn/sast/aws.rb +13 -93
  7. data/lib/pwn/sast/banned_function_calls_c.rb +138 -218
  8. data/lib/pwn/sast/base64.rb +12 -90
  9. data/lib/pwn/sast/beef_hook.rb +10 -90
  10. data/lib/pwn/sast/cmd_execution_java.rb +12 -92
  11. data/lib/pwn/sast/cmd_execution_python.rb +14 -94
  12. data/lib/pwn/sast/cmd_execution_ruby.rb +22 -102
  13. data/lib/pwn/sast/cmd_execution_scala.rb +12 -92
  14. data/lib/pwn/sast/csrf.rb +10 -90
  15. data/lib/pwn/sast/deserial_java.rb +17 -97
  16. data/lib/pwn/sast/emoticon.rb +17 -98
  17. data/lib/pwn/sast/eval.rb +10 -90
  18. data/lib/pwn/sast/factory.rb +12 -92
  19. data/lib/pwn/sast/http_authorization_header.rb +20 -100
  20. data/lib/pwn/sast/inner_html.rb +10 -90
  21. data/lib/pwn/sast/keystore.rb +10 -90
  22. data/lib/pwn/sast/local_storage.rb +11 -91
  23. data/lib/pwn/sast/location_hash.rb +10 -90
  24. data/lib/pwn/sast/log4j.rb +10 -90
  25. data/lib/pwn/sast/logger.rb +24 -104
  26. data/lib/pwn/sast/md5.rb +10 -90
  27. data/lib/pwn/sast/outer_html.rb +10 -90
  28. data/lib/pwn/sast/padding_oracle.rb +11 -91
  29. data/lib/pwn/sast/password.rb +15 -95
  30. data/lib/pwn/sast/php_input_mechanisms.rb +16 -96
  31. data/lib/pwn/sast/php_type_juggling.rb +14 -94
  32. data/lib/pwn/sast/pom_version.rb +1 -1
  33. data/lib/pwn/sast/port.rb +16 -96
  34. data/lib/pwn/sast/post_message.rb +10 -90
  35. data/lib/pwn/sast/private_key.rb +10 -90
  36. data/lib/pwn/sast/redirect.rb +13 -93
  37. data/lib/pwn/sast/redos.rb +16 -96
  38. data/lib/pwn/sast/shell.rb +18 -98
  39. data/lib/pwn/sast/signature.rb +10 -90
  40. data/lib/pwn/sast/sql.rb +19 -93
  41. data/lib/pwn/sast/ssl.rb +14 -94
  42. data/lib/pwn/sast/sudo.rb +10 -90
  43. data/lib/pwn/sast/task_tag.rb +23 -103
  44. data/lib/pwn/sast/test_case_engine.rb +147 -0
  45. data/lib/pwn/sast/throw_errors.rb +14 -94
  46. data/lib/pwn/sast/token.rb +12 -92
  47. data/lib/pwn/sast/type_script_type_juggling.rb +14 -94
  48. data/lib/pwn/sast/version.rb +12 -92
  49. data/lib/pwn/sast/window_location_hash.rb +10 -90
  50. data/lib/pwn/sast.rb +4 -0
  51. data/lib/pwn/version.rb +1 -1
  52. data/lib/pwn.rb +0 -2
  53. data/spec/lib/pwn/sast/test_case_engine_spec.rb +20 -0
  54. data/third_party/pwn_rdoc.jsonl +1 -1
  55. metadata +5 -3
@@ -19,100 +19,20 @@ module PWN
19
19
  public_class_method def self.scan(opts = {})
20
20
  dir_path = opts[:dir_path]
21
21
  git_repo_root_uri = opts[:git_repo_root_uri].to_s.scrub
22
- result_arr = []
23
- ai_introspection = PWN::Env[:ai][:introspection]
24
- logger_results = "AI Introspection => #{ai_introspection} => "
25
22
 
26
- PWN::Plugins::FileFu.recurse_in_dir(dir_path: dir_path) do |entry|
27
- if File.file?(entry) && File.basename(entry) !~ /^pwn.+(html|json|db)$/ && File.basename(entry) !~ /\.JS-BEAUTIFIED$/ && entry !~ /test/i
28
- line_no_and_contents_arr = []
29
- entry_beautified = false
30
-
31
- if File.extname(entry) == '.js' && (`wc -l #{entry}`.split.first.to_i < 20 || entry.include?('.min.js') || entry.include?('-all.js'))
32
- js_beautify = `js-beautify #{entry} > #{entry}.JS-BEAUTIFIED 2> /dev/null`.to_s.scrub
33
- entry = "#{entry}.JS-BEAUTIFIED"
34
- entry_beautified = true
35
- end
36
-
37
- test_case_filter = %(
38
- grep -Ein \
39
- -e "redirect(\\s=|=)" \
40
- -e "redirect_url(\\s=|=)" \
41
- -e "redirect_uri(\\s=|=)" #{entry} 2> /dev/null
42
- )
43
-
44
- str = `#{test_case_filter}`.to_s.scrub
45
-
46
- if str.to_s.empty?
47
- # If str length is >= 64 KB do not include results. (Due to Mongo Document Size Restrictions)
48
- logger_results = "#{logger_results}~" # Catching bugs is good :)
49
- else
50
- str = "1:Result larger than 64KB -> Size: #{str.to_s.length}. Please click the \"Path\" link for more details." if str.to_s.length >= 64_000
51
-
52
- hash_line = {
53
- timestamp: Time.now.strftime('%Y-%m-%d %H:%M:%S.%9N %z').to_s,
54
- security_references: security_references,
55
- filename: { git_repo_root_uri: git_repo_root_uri, entry: entry },
56
- line_no_and_contents: '',
57
- raw_content: str,
58
- test_case_filter: test_case_filter
59
- }
60
-
61
- # COMMMENT: Must be a better way to implement this (regex is kinda funky)
62
- line_contents_split = str.split(/^(\d{1,}):|\n(\d{1,}):/)[1..-1]
63
- line_no_count = line_contents_split.length # This should always be an even number
64
- current_count = 0
65
- while line_no_count > current_count
66
- line_no = line_contents_split[current_count]
67
- contents = line_contents_split[current_count + 1]
68
- if Dir.exist?('.git')
69
- repo_root = '.'
70
-
71
- author = PWN::Plugins::Git.get_author(
72
- repo_root: repo_root,
73
- from_line: line_no,
74
- to_line: line_no,
75
- target_file: entry,
76
- entry_beautified: entry_beautified
77
- )
78
- end
79
- author ||= 'N/A'
80
-
81
- ai_analysis = nil
82
- if ai_introspection
83
- request = {
84
- scm_uri: "#{hash_line[:filename][:git_repo_root_uri]}/#{hash_line[:filename][:entry]}",
85
- line_no: line_no,
86
- source_code_snippet: contents
87
- }.to_json
88
- response = PWN::AI::Introspection.reflect(request: request)
89
- if response.is_a?(Hash)
90
- ai_analysis = response[:choices].last[:text] if response[:choices].last.keys.include?(:text)
91
- ai_analysis = response[:choices].last[:content] if response[:choices].last.keys.include?(:content)
92
- end
93
- end
94
-
95
- hash_line[:line_no_and_contents] = line_no_and_contents_arr.push(
96
- line_no: line_no,
97
- contents: contents,
98
- author: author,
99
- ai_analysis: ai_analysis
100
- )
101
-
102
- current_count += 2
103
- end
104
- result_arr.push(hash_line)
105
- logger_results = "#{logger_results}x" # Seeing progress is good :)
106
- end
107
- end
108
- end
109
- logger_banner = "http://#{Socket.gethostname}:8808/doc_root/pwn-#{PWN::VERSION.to_s.scrub}/#{to_s.scrub.gsub('::', '/')}.html"
110
- if logger_results.empty?
111
- @@logger.info("#{logger_banner}: No files applicable to this test case.\n")
112
- else
113
- @@logger.info("#{logger_banner} => #{logger_results}complete.\n")
114
- end
115
- result_arr
23
+ test_case_filter = %(
24
+ grep -Ein \
25
+ -e "redirect(\\s=|=)" \
26
+ -e "redirect_url(\\s=|=)" \
27
+ -e "redirect_uri(\\s=|=)" {PWN_SAST_SRC_TARGET} 2> /dev/null
28
+ )
29
+
30
+ PWN::SAST::TestCaseEngine.execute(
31
+ test_case_filter: test_case_filter,
32
+ security_references: security_references,
33
+ dir_path: dir_path,
34
+ git_repo_root_uri: git_repo_root_uri
35
+ )
116
36
  rescue StandardError => e
117
37
  raise e
118
38
  end
@@ -20,104 +20,24 @@ module PWN
20
20
  public_class_method def self.scan(opts = {})
21
21
  dir_path = opts[:dir_path]
22
22
  git_repo_root_uri = opts[:git_repo_root_uri].to_s.scrub
23
- result_arr = []
24
- ai_introspection = PWN::Env[:ai][:introspection]
25
- logger_results = "AI Introspection => #{ai_introspection} => "
26
23
 
27
- PWN::Plugins::FileFu.recurse_in_dir(dir_path: dir_path) do |entry|
28
- if File.file?(entry) && File.basename(entry) !~ /^pwn.+(html|json|db)$/ && File.basename(entry) !~ /\.JS-BEAUTIFIED$/ && entry !~ /test/i
29
- line_no_and_contents_arr = []
30
- entry_beautified = false
31
-
32
- if File.extname(entry) == '.js' && (`wc -l #{entry}`.split.first.to_i < 20 || entry.include?('.min.js') || entry.include?('-all.js'))
33
- js_beautify = `js-beautify #{entry} > #{entry}.JS-BEAUTIFIED 2> /dev/null`.to_s.scrub
34
- entry = "#{entry}.JS-BEAUTIFIED"
35
- entry_beautified = true
36
- end
37
-
38
- test_case_filter = "
39
- grep -Fin \
40
- -e '(a+)+' \
41
- -e '([a-zA-Z]+)*' \
42
- -e '(a|aa)+' \
43
- -e '(a|a?)+' \
44
- -e '(([a-z])' \
45
- -e '([a-zA-Z0-9])' \
46
- -e '(.*a)\{' #{entry} 2> /dev/null
47
- "
48
-
49
- str = `#{test_case_filter}`.to_s.scrub
50
-
51
- if str.to_s.empty?
52
- # If str length is >= 64 KB do not include results. (Due to Mongo Document Size Restrictions)
53
- logger_results = "#{logger_results}~" # Catching bugs is good :)
54
- else
55
- str = "1:Result larger than 64KB -> Size: #{str.to_s.length}. Please click the \"Path\" link for more details." if str.to_s.length >= 64_000
56
-
57
- hash_line = {
58
- timestamp: Time.now.strftime('%Y-%m-%d %H:%M:%S.%9N %z').to_s,
59
- security_references: security_references,
60
- filename: { git_repo_root_uri: git_repo_root_uri, entry: entry },
61
- line_no_and_contents: '',
62
- raw_content: str,
63
- test_case_filter: test_case_filter
64
- }
65
-
66
- # COMMMENT: Must be a better way to implement this (regex is kinda funky)
67
- line_contents_split = str.split(/^(\d{1,}):|\n(\d{1,}):/)[1..-1]
68
- line_no_count = line_contents_split.length # This should always be an even number
69
- current_count = 0
70
- while line_no_count > current_count
71
- line_no = line_contents_split[current_count]
72
- contents = line_contents_split[current_count + 1]
73
- if Dir.exist?('.git')
74
- repo_root = '.'
75
-
76
- author = PWN::Plugins::Git.get_author(
77
- repo_root: repo_root,
78
- from_line: line_no,
79
- to_line: line_no,
80
- target_file: entry,
81
- entry_beautified: entry_beautified
82
- )
83
- end
84
- author ||= 'N/A'
85
-
86
- ai_analysis = nil
87
- if ai_introspection
88
- request = {
89
- scm_uri: "#{hash_line[:filename][:git_repo_root_uri]}/#{hash_line[:filename][:entry]}",
90
- line_no: line_no,
91
- source_code_snippet: contents
92
- }.to_json
93
- response = PWN::AI::Introspection.reflect(request: request)
94
- if response.is_a?(Hash)
95
- ai_analysis = response[:choices].last[:text] if response[:choices].last.keys.include?(:text)
96
- ai_analysis = response[:choices].last[:content] if response[:choices].last.keys.include?(:content)
97
- end
98
- end
99
-
100
- hash_line[:line_no_and_contents] = line_no_and_contents_arr.push(
101
- line_no: line_no,
102
- contents: contents,
103
- author: author,
104
- ai_analysis: ai_analysis
105
- )
24
+ test_case_filter = "
25
+ grep -Fin \
26
+ -e '(a+)+' \
27
+ -e '([a-zA-Z]+)*' \
28
+ -e '(a|aa)+' \
29
+ -e '(a|a?)+' \
30
+ -e '(([a-z])' \
31
+ -e '([a-zA-Z0-9])' \
32
+ -e '(.*a)\{' {PWN_SAST_SRC_TARGET} 2> /dev/null
33
+ "
106
34
 
107
- current_count += 2
108
- end
109
- result_arr.push(hash_line)
110
- logger_results = "#{logger_results}x" # Seeing progress is good :)
111
- end
112
- end
113
- end
114
- logger_banner = "http://#{Socket.gethostname}:8808/doc_root/pwn-#{PWN::VERSION.to_s.scrub}/#{to_s.scrub.gsub('::', '/')}.html"
115
- if logger_results.empty?
116
- @@logger.info("#{logger_banner}: No files applicable to this test case.\n")
117
- else
118
- @@logger.info("#{logger_banner} => #{logger_results}complete.\n")
119
- end
120
- result_arr
35
+ PWN::SAST::TestCaseEngine.execute(
36
+ test_case_filter: test_case_filter,
37
+ security_references: security_references,
38
+ dir_path: dir_path,
39
+ git_repo_root_uri: git_repo_root_uri
40
+ )
121
41
  rescue StandardError => e
122
42
  raise e
123
43
  end
@@ -19,106 +19,26 @@ module PWN
19
19
  public_class_method def self.scan(opts = {})
20
20
  dir_path = opts[:dir_path]
21
21
  git_repo_root_uri = opts[:git_repo_root_uri].to_s.scrub
22
- result_arr = []
23
- ai_introspection = PWN::Env[:ai][:introspection]
24
- logger_results = "AI Introspection => #{ai_introspection} => "
25
22
 
26
- PWN::Plugins::FileFu.recurse_in_dir(dir_path: dir_path) do |entry|
27
- if File.file?(entry) && File.basename(entry) !~ /^pwn.+(html|json|db)$/ && File.basename(entry) !~ /\.JS-BEAUTIFIED$/ && entry !~ /test/i
28
- line_no_and_contents_arr = []
29
- entry_beautified = false
30
-
31
- if File.extname(entry) == '.js' && (`wc -l #{entry}`.split.first.to_i < 20 || entry.include?('.min.js') || entry.include?('-all.js'))
32
- js_beautify = `js-beautify #{entry} > #{entry}.JS-BEAUTIFIED 2> /dev/null`.to_s.scrub
33
- entry = "#{entry}.JS-BEAUTIFIED"
34
- entry_beautified = true
35
- end
36
-
37
- test_case_filter = "
38
- grep -niE \
39
- -e '/bin/bash' \
40
- -e '/bin/csh' \
41
- -e '/bin/dash' \
42
- -e '/bin/ksh' \
43
- -e '/bin/rbash' \
44
- -e '/bin/sh' \
45
- -e '/bin/tcsh' \
46
- -e '/usr/bin/screen' \
47
- -e '/bin/zsh' #{entry} 2> /dev/null
48
- "
49
-
50
- str = `#{test_case_filter}`.to_s.scrub
51
-
52
- if str.to_s.empty?
53
- # If str length is >= 64 KB do not include results. (Due to Mongo Document Size Restrictions)
54
- logger_results = "#{logger_results}~" # Catching bugs is good :)
55
- else
56
- str = "1:Result larger than 64KB -> Size: #{str.to_s.length}. Please click the \"Path\" link for more details." if str.to_s.length >= 64_000
57
-
58
- hash_line = {
59
- timestamp: Time.now.strftime('%Y-%m-%d %H:%M:%S.%9N %z').to_s,
60
- security_references: security_references,
61
- filename: { git_repo_root_uri: git_repo_root_uri, entry: entry },
62
- line_no_and_contents: '',
63
- raw_content: str,
64
- test_case_filter: test_case_filter
65
- }
66
-
67
- # COMMMENT: Must be a better way to implement this (regex is kinda funky)
68
- line_contents_split = str.split(/^(\d{1,}):|\n(\d{1,}):/)[1..-1]
69
- line_no_count = line_contents_split.length # This should always be an even number
70
- current_count = 0
71
- while line_no_count > current_count
72
- line_no = line_contents_split[current_count]
73
- contents = line_contents_split[current_count + 1]
74
- if Dir.exist?('.git')
75
- repo_root = '.'
76
-
77
- author = PWN::Plugins::Git.get_author(
78
- repo_root: repo_root,
79
- from_line: line_no,
80
- to_line: line_no,
81
- target_file: entry,
82
- entry_beautified: entry_beautified
83
- )
84
- end
85
- author ||= 'N/A'
86
-
87
- ai_analysis = nil
88
- if ai_introspection
89
- request = {
90
- scm_uri: "#{hash_line[:filename][:git_repo_root_uri]}/#{hash_line[:filename][:entry]}",
91
- line_no: line_no,
92
- source_code_snippet: contents
93
- }.to_json
94
- response = PWN::AI::Introspection.reflect(request: request)
95
- if response.is_a?(Hash)
96
- ai_analysis = response[:choices].last[:text] if response[:choices].last.keys.include?(:text)
97
- ai_analysis = response[:choices].last[:content] if response[:choices].last.keys.include?(:content)
98
- end
99
- end
100
-
101
- hash_line[:line_no_and_contents] = line_no_and_contents_arr.push(
102
- line_no: line_no,
103
- contents: contents,
104
- author: author,
105
- ai_analysis: ai_analysis
106
- )
23
+ test_case_filter = "
24
+ grep -niE \
25
+ -e '/bin/bash' \
26
+ -e '/bin/csh' \
27
+ -e '/bin/dash' \
28
+ -e '/bin/ksh' \
29
+ -e '/bin/rbash' \
30
+ -e '/bin/sh' \
31
+ -e '/bin/tcsh' \
32
+ -e '/usr/bin/screen' \
33
+ -e '/bin/zsh' {PWN_SAST_SRC_TARGET} 2> /dev/null
34
+ "
107
35
 
108
- current_count += 2
109
- end
110
- result_arr.push(hash_line)
111
- logger_results = "#{logger_results}x" # Seeing progress is good :)
112
- end
113
- end
114
- end
115
- logger_banner = "http://#{Socket.gethostname}:8808/doc_root/pwn-#{PWN::VERSION.to_s.scrub}/#{to_s.scrub.gsub('::', '/')}.html"
116
- if logger_results.empty?
117
- @@logger.info("#{logger_banner}: No files applicable to this test case.\n")
118
- else
119
- @@logger.info("#{logger_banner} => #{logger_results}complete.\n")
120
- end
121
- result_arr
36
+ PWN::SAST::TestCaseEngine.execute(
37
+ test_case_filter: test_case_filter,
38
+ security_references: security_references,
39
+ dir_path: dir_path,
40
+ git_repo_root_uri: git_repo_root_uri
41
+ )
122
42
  rescue StandardError => e
123
43
  raise e
124
44
  end
@@ -19,98 +19,18 @@ module PWN
19
19
  public_class_method def self.scan(opts = {})
20
20
  dir_path = opts[:dir_path]
21
21
  git_repo_root_uri = opts[:git_repo_root_uri].to_s.scrub
22
- result_arr = []
23
- ai_introspection = PWN::Env[:ai][:introspection]
24
- logger_results = "AI Introspection => #{ai_introspection} => "
25
22
 
26
- PWN::Plugins::FileFu.recurse_in_dir(dir_path: dir_path) do |entry|
27
- if File.file?(entry) && File.basename(entry) !~ /^pwn.+(html|json|db)$/ && File.basename(entry) !~ /\.JS-BEAUTIFIED$/ && entry !~ /test/i
28
- line_no_and_contents_arr = []
29
- entry_beautified = false
30
-
31
- if File.extname(entry) == '.js' && (`wc -l #{entry}`.split.first.to_i < 20 || entry.include?('.min.js') || entry.include?('-all.js'))
32
- js_beautify = `js-beautify #{entry} > #{entry}.JS-BEAUTIFIED 2> /dev/null`.to_s.scrub
33
- entry = "#{entry}.JS-BEAUTIFIED"
34
- entry_beautified = true
35
- end
36
-
37
- test_case_filter = "
38
- grep -n \
39
- -e 'Signature' #{entry} 2> /dev/null
40
- "
41
-
42
- str = `#{test_case_filter}`.to_s.scrub
43
-
44
- if str.to_s.empty?
45
- # If str length is >= 64 KB do not include results. (Due to Mongo Document Size Restrictions)
46
- logger_results = "#{logger_results}~" # Catching bugs is good :)
47
- else
48
- str = "1:Result larger than 64KB -> Size: #{str.to_s.length}. Please click the \"Path\" link for more details." if str.to_s.length >= 64_000
49
-
50
- hash_line = {
51
- timestamp: Time.now.strftime('%Y-%m-%d %H:%M:%S.%9N %z').to_s,
52
- security_references: security_references,
53
- filename: { git_repo_root_uri: git_repo_root_uri, entry: entry },
54
- line_no_and_contents: '',
55
- raw_content: str,
56
- test_case_filter: test_case_filter
57
- }
58
-
59
- # COMMMENT: Must be a better way to implement this (regex is kinda funky)
60
- line_contents_split = str.split(/^(\d{1,}):|\n(\d{1,}):/)[1..-1]
61
- line_no_count = line_contents_split.length # This should always be an even number
62
- current_count = 0
63
- while line_no_count > current_count
64
- line_no = line_contents_split[current_count]
65
- contents = line_contents_split[current_count + 1]
66
- if Dir.exist?('.git')
67
- repo_root = '.'
68
-
69
- author = PWN::Plugins::Git.get_author(
70
- repo_root: repo_root,
71
- from_line: line_no,
72
- to_line: line_no,
73
- target_file: entry,
74
- entry_beautified: entry_beautified
75
- )
76
- end
77
- author ||= 'N/A'
78
-
79
- ai_analysis = nil
80
- if ai_introspection
81
- request = {
82
- scm_uri: "#{hash_line[:filename][:git_repo_root_uri]}/#{hash_line[:filename][:entry]}",
83
- line_no: line_no,
84
- source_code_snippet: contents
85
- }.to_json
86
- response = PWN::AI::Introspection.reflect(request: request)
87
- if response.is_a?(Hash)
88
- ai_analysis = response[:choices].last[:text] if response[:choices].last.keys.include?(:text)
89
- ai_analysis = response[:choices].last[:content] if response[:choices].last.keys.include?(:content)
90
- end
91
- end
92
-
93
- hash_line[:line_no_and_contents] = line_no_and_contents_arr.push(
94
- line_no: line_no,
95
- contents: contents,
96
- author: author,
97
- ai_analysis: ai_analysis
98
- )
23
+ test_case_filter = "
24
+ grep -n \
25
+ -e 'Signature' {PWN_SAST_SRC_TARGET} 2> /dev/null
26
+ "
99
27
 
100
- current_count += 2
101
- end
102
- result_arr.push(hash_line)
103
- logger_results = "#{logger_results}x" # Seeing progress is good :)
104
- end
105
- end
106
- end
107
- logger_banner = "http://#{Socket.gethostname}:8808/doc_root/pwn-#{PWN::VERSION.to_s.scrub}/#{to_s.scrub.gsub('::', '/')}.html"
108
- if logger_results.empty?
109
- @@logger.info("#{logger_banner}: No files applicable to this test case.\n")
110
- else
111
- @@logger.info("#{logger_banner} => #{logger_results}complete.\n")
112
- end
113
- result_arr
28
+ PWN::SAST::TestCaseEngine.execute(
29
+ test_case_filter: test_case_filter,
30
+ security_references: security_references,
31
+ dir_path: dir_path,
32
+ git_repo_root_uri: git_repo_root_uri
33
+ )
114
34
  rescue StandardError => e
115
35
  raise e
116
36
  end
data/lib/pwn/sast/sql.rb CHANGED
@@ -20,101 +20,27 @@ module PWN
20
20
  public_class_method def self.scan(opts = {})
21
21
  dir_path = opts[:dir_path]
22
22
  git_repo_root_uri = opts[:git_repo_root_uri].to_s.scrub
23
- result_arr = []
24
- ai_introspection = PWN::Env[:ai][:introspection]
25
- logger_results = "AI Introspection => #{ai_introspection} => "
26
23
 
27
- PWN::Plugins::FileFu.recurse_in_dir(dir_path: dir_path) do |entry|
28
- if File.file?(entry) && File.basename(entry) !~ /^pwn.+(html|json|db)$/ && File.basename(entry) !~ /\.JS-BEAUTIFIED$/ && entry !~ /test/i
29
- line_no_and_contents_arr = []
30
- entry_beautified = false
31
-
32
- if File.extname(entry) == '.js' && (`wc -l #{entry}`.split.first.to_i < 20 || entry.include?('.min.js') || entry.include?('-all.js'))
33
- js_beautify = `js-beautify #{entry} > #{entry}.JS-BEAUTIFIED 2> /dev/null`.to_s.scrub
34
- entry = "#{entry}.JS-BEAUTIFIED"
35
- entry_beautified = true
36
- end
37
-
38
- test_case_filter = "
39
- grep -in \
40
- -e 'select .*from ' \
41
- -e 'insert into .*values' \
42
- -e 'update .*set ' \
43
- -e 'delete from ' #{entry} 2> /dev/null
44
- "
45
-
46
- str = `#{test_case_filter}`.to_s.scrub
47
-
48
- if str.to_s.empty?
49
- # If str length is >= 64 KB do not include results. (Due to Mongo Document Size Restrictions)
50
- logger_results = "#{logger_results}~" # Catching bugs is good :)
51
- else
52
- str = "1:Result larger than 64KB -> Size: #{str.to_s.length}. Please click the \"Path\" link for more details." if str.to_s.length >= 64_000
53
-
54
- hash_line = {
55
- timestamp: Time.now.strftime('%Y-%m-%d %H:%M:%S.%9N %z').to_s,
56
- security_references: security_references,
57
- filename: { git_repo_root_uri: git_repo_root_uri, entry: entry },
58
- line_no_and_contents: '',
59
- raw_content: str,
60
- test_case_filter: test_case_filter
61
- }
62
-
63
- # COMMMENT: Must be a better way to implement this (regex is kinda funky)
64
- line_contents_split = str.split(/^(\d{1,}):|\n(\d{1,}):/)[1..-1]
65
- line_no_count = line_contents_split.length # This should always be an even number
66
- current_count = 0
67
- while line_no_count > current_count
68
- line_no = line_contents_split[current_count]
69
- contents = line_contents_split[current_count + 1]
70
- if Dir.exist?('.git')
71
- repo_root = '.'
72
-
73
- author = PWN::Plugins::Git.get_author(
74
- repo_root: repo_root,
75
- from_line: line_no,
76
- to_line: line_no,
77
- target_file: entry,
78
- entry_beautified: entry_beautified
79
- )
80
- end
81
- author ||= 'N/A'
82
-
83
- ai_analysis = nil
84
- if ai_introspection
85
- request = {
86
- scm_uri: "#{hash_line[:filename][:git_repo_root_uri]}/#{hash_line[:filename][:entry]}",
87
- line_no: line_no,
88
- source_code_snippet: contents
89
- }.to_json
90
- response = PWN::AI::Introspection.reflect(request: request)
91
- if response.is_a?(Hash)
92
- ai_analysis = response[:choices].last[:text] if response[:choices].last.keys.include?(:text)
93
- ai_analysis = response[:choices].last[:content] if response[:choices].last.keys.include?(:content)
94
- end
95
- end
96
-
97
- hash_line[:line_no_and_contents] = line_no_and_contents_arr.push(
98
- line_no: line_no,
99
- contents: contents,
100
- author: author,
101
- ai_analysis: ai_analysis
102
- )
24
+ test_case_filter = "
25
+ grep -in \
26
+ -e 'select .*from ' \
27
+ -e 'insert into .*values' \
28
+ -e 'update .*set ' \
29
+ -e 'delete from ' {PWN_SAST_SRC_TARGET} 2> /dev/null | \
30
+ grep -ivE \
31
+ -e '^[0-9]:.*\\?.*$' \
32
+ -e '^[0-9]:.*@.*$' \
33
+ -e '^[0-9]:.*\\$[0-9].*$' \
34
+ -e '^[0-9]:.*:[0-9].*$' \
35
+ -e '^[0-9]:.*:[a-zA-Z_][a-zA-Z0-9_].*$'
36
+ "
103
37
 
104
- current_count += 2
105
- end
106
- result_arr.push(hash_line)
107
- logger_results = "#{logger_results}x" # Seeing progress is good :)
108
- end
109
- end
110
- end
111
- logger_banner = "http://#{Socket.gethostname}:8808/doc_root/pwn-#{PWN::VERSION.to_s.scrub}/#{to_s.scrub.gsub('::', '/')}.html"
112
- if logger_results.empty?
113
- @@logger.info("#{logger_banner}: No files applicable to this test case.\n")
114
- else
115
- @@logger.info("#{logger_banner} => #{logger_results}complete.\n")
116
- end
117
- result_arr
38
+ PWN::SAST::TestCaseEngine.execute(
39
+ test_case_filter: test_case_filter,
40
+ security_references: security_references,
41
+ dir_path: dir_path,
42
+ git_repo_root_uri: git_repo_root_uri
43
+ )
118
44
  rescue StandardError => e
119
45
  raise e
120
46
  end