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.
- checksums.yaml +4 -4
- data/Gemfile +1 -1
- data/README.md +3 -3
- data/lib/pwn/sast/amqp_connect_as_guest.rb +12 -89
- data/lib/pwn/sast/apache_file_system_util_api.rb +11 -91
- data/lib/pwn/sast/aws.rb +13 -93
- data/lib/pwn/sast/banned_function_calls_c.rb +138 -218
- data/lib/pwn/sast/base64.rb +12 -90
- data/lib/pwn/sast/beef_hook.rb +10 -90
- data/lib/pwn/sast/cmd_execution_java.rb +12 -92
- data/lib/pwn/sast/cmd_execution_python.rb +14 -94
- data/lib/pwn/sast/cmd_execution_ruby.rb +22 -102
- data/lib/pwn/sast/cmd_execution_scala.rb +12 -92
- data/lib/pwn/sast/csrf.rb +10 -90
- data/lib/pwn/sast/deserial_java.rb +17 -97
- data/lib/pwn/sast/emoticon.rb +17 -98
- data/lib/pwn/sast/eval.rb +10 -90
- data/lib/pwn/sast/factory.rb +12 -92
- data/lib/pwn/sast/http_authorization_header.rb +20 -100
- data/lib/pwn/sast/inner_html.rb +10 -90
- data/lib/pwn/sast/keystore.rb +10 -90
- data/lib/pwn/sast/local_storage.rb +11 -91
- data/lib/pwn/sast/location_hash.rb +10 -90
- data/lib/pwn/sast/log4j.rb +10 -90
- data/lib/pwn/sast/logger.rb +24 -104
- data/lib/pwn/sast/md5.rb +10 -90
- data/lib/pwn/sast/outer_html.rb +10 -90
- data/lib/pwn/sast/padding_oracle.rb +11 -91
- data/lib/pwn/sast/password.rb +15 -95
- data/lib/pwn/sast/php_input_mechanisms.rb +16 -96
- data/lib/pwn/sast/php_type_juggling.rb +14 -94
- data/lib/pwn/sast/pom_version.rb +1 -1
- data/lib/pwn/sast/port.rb +16 -96
- data/lib/pwn/sast/post_message.rb +10 -90
- data/lib/pwn/sast/private_key.rb +10 -90
- data/lib/pwn/sast/redirect.rb +13 -93
- data/lib/pwn/sast/redos.rb +16 -96
- data/lib/pwn/sast/shell.rb +18 -98
- data/lib/pwn/sast/signature.rb +10 -90
- data/lib/pwn/sast/sql.rb +19 -93
- data/lib/pwn/sast/ssl.rb +14 -94
- data/lib/pwn/sast/sudo.rb +10 -90
- data/lib/pwn/sast/task_tag.rb +23 -103
- data/lib/pwn/sast/test_case_engine.rb +147 -0
- data/lib/pwn/sast/throw_errors.rb +14 -94
- data/lib/pwn/sast/token.rb +12 -92
- data/lib/pwn/sast/type_script_type_juggling.rb +14 -94
- data/lib/pwn/sast/version.rb +12 -92
- data/lib/pwn/sast/window_location_hash.rb +10 -90
- data/lib/pwn/sast.rb +4 -0
- data/lib/pwn/version.rb +1 -1
- data/lib/pwn.rb +0 -2
- data/spec/lib/pwn/sast/test_case_engine_spec.rb +20 -0
- data/third_party/pwn_rdoc.jsonl +1 -1
- metadata +5 -3
data/lib/pwn/sast/redirect.rb
CHANGED
@@ -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
|
-
|
27
|
-
|
28
|
-
|
29
|
-
|
30
|
-
|
31
|
-
|
32
|
-
|
33
|
-
|
34
|
-
|
35
|
-
|
36
|
-
|
37
|
-
|
38
|
-
|
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
|
data/lib/pwn/sast/redos.rb
CHANGED
@@ -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
|
-
|
28
|
-
|
29
|
-
|
30
|
-
|
31
|
-
|
32
|
-
|
33
|
-
|
34
|
-
|
35
|
-
|
36
|
-
|
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
|
-
|
108
|
-
|
109
|
-
|
110
|
-
|
111
|
-
|
112
|
-
|
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
|
data/lib/pwn/sast/shell.rb
CHANGED
@@ -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
|
-
|
27
|
-
|
28
|
-
|
29
|
-
|
30
|
-
|
31
|
-
|
32
|
-
|
33
|
-
|
34
|
-
|
35
|
-
|
36
|
-
|
37
|
-
|
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
|
-
|
109
|
-
|
110
|
-
|
111
|
-
|
112
|
-
|
113
|
-
|
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
|
data/lib/pwn/sast/signature.rb
CHANGED
@@ -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
|
-
|
27
|
-
|
28
|
-
|
29
|
-
|
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
|
-
|
101
|
-
|
102
|
-
|
103
|
-
|
104
|
-
|
105
|
-
|
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
|
-
|
28
|
-
|
29
|
-
|
30
|
-
|
31
|
-
|
32
|
-
|
33
|
-
|
34
|
-
|
35
|
-
|
36
|
-
|
37
|
-
|
38
|
-
|
39
|
-
|
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
|
-
|
105
|
-
|
106
|
-
|
107
|
-
|
108
|
-
|
109
|
-
|
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
|