Project

General

Profile

ThreatScript Definitions » History » Version 21

Luke Murphey, 11/03/2010 11:17 PM

1 1 Luke Murphey
h1. ThreatScript Definitions
2 1 Luke Murphey
3 1 Luke Murphey
ThreatScript Definitions are written in ECMAScript (basically the same as JavaScript). The ThreatScript definitions return a Result object which indicates whether a match was observed.
4 1 Luke Murphey
5 1 Luke Murphey
h2. ThreatScript Example
6 1 Luke Murphey
7 1 Luke Murphey
Below is an example of a ThreatScript that triggers if the web-page has a form element.
8 1 Luke Murphey
9 1 Luke Murphey
<pre><code class="javascript">
10 1 Luke Murphey
/*
11 1 Luke Murphey
 * Name: Example.General.Has_Form_Tag
12 1 Luke Murphey
 * Version: 1
13 1 Luke Murphey
 * ID: 1000000
14 1 Luke Murphey
 * Message: Indicates if the page has as a form tag
15 1 Luke Murphey
 * Severity: Low
16 11 Luke Murphey
 * Reference: url,threatfactor.com
17 1 Luke Murphey
 */
18 1 Luke Murphey
19 1 Luke Murphey
importPackage(Packages.ThreatScript);
20 1 Luke Murphey
importPackage(Packages.HTTP);
21 1 Luke Murphey
22 8 Luke Murphey
function analyze( httpResponse, variables, environment ){
23 1 Luke Murphey
24 1 Luke Murphey
	var parser = httpResponse.getDocumentParser();
25 1 Luke Murphey
	var location = new URL( httpResponse.getLocation() );
26 1 Luke Murphey
27 1 Luke Murphey
	//Get a list of all script tags
28 1 Luke Murphey
	var tagNameFilter = new TagNameFilter("form");
29 1 Luke Murphey
	var nodesList = parser.extractAllNodesThatMatch(tagNameFilter); 
30 1 Luke Murphey
        if( nodesList.size() > 0 ){
31 1 Luke Murphey
	     return new Result( true, "A form was detected" );
32 1 Luke Murphey
	}
33 1 Luke Murphey
        
34 1 Luke Murphey
	return new Result( false, "No forms detected" );
35 1 Luke Murphey
}
36 1 Luke Murphey
</code>
37 1 Luke Murphey
</pre>
38 2 Luke Murphey
39 3 Luke Murphey
h2. Analysis Function
40 3 Luke Murphey
41 12 Luke Murphey
ThreatScripts must provide an analyze function that takes 3 arguments:
42 3 Luke Murphey
43 7 Luke Murphey
| *Name*       | *Type*            | *Note*                                                                           |
44 7 Luke Murphey
| httpResponse | HttpResponseData  | See source:trunk/src/net/lukemurphey/nsia/scan/HttpResponseData.java             |
45 7 Luke Murphey
| variables    | Variables         | See source:trunk/src/net/lukemurphey/nsia/scan/scriptenvironment/Variables.java  |
46 7 Luke Murphey
| environment  | Environment       | See source:trunk/src/net/lukemurphey/nsia/scan/ScriptDefinition.java#L605        |
47 1 Luke Murphey
48 16 Luke Murphey
h2. Saving / Loading Saved Data (using the environment object)
49 16 Luke Murphey
50 16 Luke Murphey
ThreatScripts can save and load data using the environment object. The environment allows the script to recall information between runs of the definition for a given URL and rule.
51 16 Luke Murphey
52 16 Luke Murphey
h3. Retrieving Values
53 16 Luke Murphey
54 16 Luke Murphey
Values can be retrieved using the _get_ function of the environment object. Call _getValue_ on the returned object to get the value. Below is an example:
55 16 Luke Murphey
56 16 Luke Murphey
<pre><code class="javascript">
57 16 Luke Murphey
var saved = environment.get("ValueName");
58 16 Luke Murphey
var value = saved.getValue();
59 16 Luke Murphey
</code></pre>
60 16 Luke Murphey
61 16 Luke Murphey
h3. Saving Values
62 16 Luke Murphey
63 16 Luke Murphey
Values can be saved using the _set_ function of the environment object. Optionally, the set argument accepts a boolean indicating if the provided value should only be returned when the definition is running against the same URL. Below is an example:
64 16 Luke Murphey
65 16 Luke Murphey
<pre><code class="javascript">
66 16 Luke Murphey
var test = "1234ABCD";
67 17 Luke Murphey
68 17 Luke Murphey
//Will only be returned when the given definition is executed against the current URL:
69 17 Luke Murphey
environment.set("ValueName", test);
70 17 Luke Murphey
71 17 Luke Murphey
//Will be returned when the given definition is executed against the any URL (within the given rule):
72 17 Luke Murphey
environment.set("ValueName", test, false);
73 16 Luke Murphey
</code></pre>
74 16 Luke Murphey
75 20 Luke Murphey
h3. Saving / Loading Collections
76 20 Luke Murphey
77 20 Luke Murphey
If your script needs to load collections then you ought to use the Vector class since it will perform the necessary conversions that are required to make sure that the contents are the same when loaded.
78 20 Luke Murphey
79 20 Luke Murphey
<pre><code class="javascript">
80 20 Luke Murphey
v = new Vector();
81 20 Luke Murphey
v.push("one");
82 20 Luke Murphey
v.get(0);
83 20 Luke Murphey
v.pop();
84 20 Luke Murphey
v.remove();
85 20 Luke Murphey
environment.set("Vector", test);
86 20 Luke Murphey
</code></pre>
87 20 Luke Murphey
88 1 Luke Murphey
h2. Baseline Function
89 1 Luke Murphey
90 4 Luke Murphey
ThreatScripts may declare a baseline function that will allow the definition to be configured to baseline itself against the previous set of scan results. The baseline function is called by NSIA when a user presses the baseline method for a rule. The objective of the baseline function is to view the provided scan results and ignore the particular finding for the given resource in the future. For example, a definition that triggers when the hash of the web-page changes may define a baseline function that causes it to not trigger unless the web-page hashes changes to yet another value.
91 4 Luke Murphey
92 4 Luke Murphey
Below is an example:
93 4 Luke Murphey
94 4 Luke Murphey
<pre>
95 4 Luke Murphey
<code class="javascript">
96 4 Luke Murphey
function baseline( environment ){
97 4 Luke Murphey
	var previousValue = environment.get("LastObservedHash");
98 4 Luke Murphey
99 4 Luke Murphey
	if( previousValue != null && previousValue.getValue() != null ){
100 4 Luke Murphey
		environment.set("Hash", previousValue.getValue() );
101 4 Luke Murphey
	}
102 4 Luke Murphey
103 4 Luke Murphey
	return true;
104 4 Luke Murphey
}
105 4 Luke Murphey
</code>
106 4 Luke Murphey
</pre>
107 4 Luke Murphey
108 21 Luke Murphey
h2. Link Extraction
109 21 Luke Murphey
110 21 Luke Murphey
ThreatScript definitions can add URLs to the list that will be scanned. This allows ThreatScripts to parse content and enumerate other content that is linked. URLs can be added by calling addURL on the Result object or by specifying the URLs in the Result constructor. See below for an example:
111 21 Luke Murphey
112 21 Luke Murphey
<pre>
113 21 Luke Murphey
<code class="javascript">
114 21 Luke Murphey
importPackage(Packages.ThreatScript);
115 21 Luke Murphey
importPackage(Packages.HTTP);
116 21 Luke Murphey
117 21 Luke Murphey
function analyze( httpResponse, operation, environment ){
118 21 Luke Murphey
    a = new Array();
119 21 Luke Murphey
    a[0] = new URL("http://mydomain.com/");
120 21 Luke Murphey
    a[1] = new URL("http://someotherdomain.com/");
121 21 Luke Murphey
122 21 Luke Murphey
    // You can add URLs via the results constructor (or add them manually via the "addURL" methods) 
123 21 Luke Murphey
    result = new Result( false, "Two URLs extracted", a);
124 21 Luke Murphey
125 21 Luke Murphey
    // This URL will not be scanned unless it matches the domain restriction in the rule
126 21 Luke Murphey
    result.addURL( new URL("http://mydomain.com/") );
127 21 Luke Murphey
128 21 Luke Murphey
    // This URL will be scanned regardless of whether it matches the domain restriction in the rule
129 21 Luke Murphey
    result.addURL( new URL("http://someotherdomain.com"), true);
130 21 Luke Murphey
    return result;
131 21 Luke Murphey
}
132 21 Luke Murphey
</code>
133 21 Luke Murphey
</pre>
134 21 Luke Murphey
135 21 Luke Murphey
136 14 Luke Murphey
h2. Terminate Function
137 3 Luke Murphey
138 14 Luke Murphey
ThreatScripts can also declare a terminate function that allows the scan engine to terminate the definition it execution exceeds the maximum execution time.
139 14 Luke Murphey
140 14 Luke Murphey
Below is an example of a (useless) rule that uses the terminate function to stop execution:
141 14 Luke Murphey
142 14 Luke Murphey
<pre>
143 14 Luke Murphey
<code class="javascript">
144 14 Luke Murphey
importPackage(Packages.ThreatScript);
145 14 Luke Murphey
146 14 Luke Murphey
var keep_going = true;
147 14 Luke Murphey
148 14 Luke Murphey
function analyze( httpResponse, operation, environment ){
149 14 Luke Murphey
150 14 Luke Murphey
    while(keep_going ){
151 14 Luke Murphey
        //Infinite loop
152 14 Luke Murphey
    }
153 14 Luke Murphey
154 14 Luke Murphey
    return new Result( false, "Definition did not match the input");
155 14 Luke Murphey
}
156 14 Luke Murphey
157 14 Luke Murphey
function terminate(){
158 14 Luke Murphey
    keep_going = false; //Will cause the analysis to stop
159 14 Luke Murphey
}
160 14 Luke Murphey
</code>
161 14 Luke Murphey
</pre>
162 3 Luke Murphey
163 1 Luke Murphey
h2. Meta-Data
164 1 Luke Murphey
165 1 Luke Murphey
ThreatScripts must provide a meta-data that indicates the following information:
166 1 Luke Murphey
167 3 Luke Murphey
| *Name*  | *Valid Input*                                      | *Notes* |
168 3 Luke Murphey
| Name    | <category>.<sub_category>.<definition_name>        |         |
169 3 Luke Murphey
| Version | integer                                            | Should be incremented each time the definition is updated |
170 3 Luke Murphey
| ID      | integer                                            | Must be 1000000 or greater (only official definitions can be less than 1000000)        |
171 3 Luke Murphey
| Message | message to be displayed when definition matches    |         |
172 3 Luke Murphey
| Severity| Either: Low, Medium or High                        |         |
173 3 Luke Murphey
| Invasive| Either: True or False (this argument is optional)  |         |
174 1 Luke Murphey
175 1 Luke Murphey
This meta-data is provided in a comment as name-value pairs (see above ThreatScript example).
176 1 Luke Murphey
177 9 Luke Murphey
h3. Definition Name
178 8 Luke Murphey
179 1 Luke Murphey
{{include(Definition_Naming_Convention)}}
180 1 Luke Murphey
181 9 Luke Murphey
h3. Definition Severity
182 9 Luke Murphey
183 9 Luke Murphey
{{include(Definition_Severity)}}
184 11 Luke Murphey
185 11 Luke Murphey
h3. Definition References 
186 11 Luke Murphey
187 11 Luke Murphey
{{include(Definition_References)}}
188 11 Luke Murphey
189 11 Luke Murphey
Note that definition references are defined a comment block with the "Reference:" as a prefix (Example: "// Reference: url,threatfactor.com").
190 9 Luke Murphey
191 3 Luke Murphey
h2. Available Packages
192 1 Luke Murphey
193 3 Luke Murphey
A series of packages are available to ThreatScripts in order to perform analysis.
194 3 Luke Murphey
195 1 Luke Murphey
| *Package*            | *Class*            | *Description*                                     |
196 13 Luke Murphey
|/9.HTTP               | URL                | Same as java.net.URL                              |
197 1 Luke Murphey
                       | TagNameFilter      | See http://htmlparser.sourceforge.net/javadoc/org/htmlparser/filters/TagNameFilter.html |
198 15 Luke Murphey
                       | GetRequest         | [[ThreatScript Web client|Web-client]] for performing HTTP Get requests  |
199 15 Luke Murphey
                       | PostRequest        | [[ThreatScript Web client|Web-client]] for performing HTTP Post requests |
200 15 Luke Murphey
                       | DeleteRequest      | [[ThreatScript Web client|Web-client]] for performing HTTP Delete requests |
201 15 Luke Murphey
                       | PutRequest         | [[ThreatScript Web client|Web-client]] for performing HTTP Put requests |
202 15 Luke Murphey
                       | TraceRequest       | [[ThreatScript Web client|Web-client]] for performing HTTP Trace requests |
203 15 Luke Murphey
                       | HeadRequest        | [[ThreatScript Web client|Web-client]] for performing HTTP Head requests |
204 15 Luke Murphey
                       | OptionsRequest     | [[ThreatScript Web client|Web-client]] for performing HTTP Option requests |
205 13 Luke Murphey
|/2.<default>          | StringUtils        | Provides a trim function for Strings, see source:/trunk/src/net/lukemurphey/nsia/scan/scriptenvironment/StringUtils.java              |
206 13 Luke Murphey
| Debug                | Provides method that allows scripts to create log messages, see source:/trunk/src/net/lukemurphey/nsia/scan/scriptenvironment/Debug.java              |
207 19 Luke Murphey
|/3.ThreatScript       | Result             | Indicates the results of analysis, see source:/trunk/src/net/lukemurphey/nsia/scan/scriptenvironment/Result.java                 |
208 19 Luke Murphey
                       | DataAnalysis       | Provides functions useful for analysis, see source:/trunk/src/net/lukemurphey/nsia/scan/ScriptSignatureUtils.java            |
209 18 Luke Murphey
                       | Vector             | Provides a storage class for scripts, see source:/trunk/src/net/lukemurphey/nsia/scan/scriptenvironment/Vector.java            |
210 1 Luke Murphey
211 1 Luke Murphey
212 1 Luke Murphey
h2. Debugging ThreatScripts
213 8 Luke Murphey
214 8 Luke Murphey
ThreatScripts can create event log messages by using the sendMessage() function in the Debug class. Simply call _Debug.sendMessage_ with a string as an argument to create an event log message. The event log messages can be viewed in the event log for NSIA.
215 8 Luke Murphey
216 8 Luke Murphey
Generally, script created log messages are used only for debugging and should be disabled on rules you want to use in production.
217 8 Luke Murphey
218 8 Luke Murphey
h2. General Notes When Writing Definitions
219 8 Luke Murphey
220 8 Luke Murphey
h3. ThreatScript Maximum Runtime
221 8 Luke Murphey
222 8 Luke Murphey
ThreatScript  definitions are forceably terminated by the scan engine if the script runs for longer than 10 seconds. Thus, it is important to write definitions that can complete within the timeframe alloted; otherwise, the definition will be flagged as having an error.
223 8 Luke Murphey
224 8 Luke Murphey
h3. Maximum Data Size
225 8 Luke Murphey
226 8 Luke Murphey
The scan engine only provides the first 1 MB of the data observed to the scan engine. Therefore, do not design ThreatScripts that won't work if only the first 1 MB of a larger file is provided.