| skipped 6 lines |
7 | 7 | | "github.com/bearer/bearer/new/language/tree" |
8 | 8 | | "github.com/bearer/bearer/pkg/util/stringutil" |
9 | 9 | | |
| 10 | + | "github.com/bearer/bearer/new/detector/implementation/generic" |
10 | 11 | | generictypes "github.com/bearer/bearer/new/detector/implementation/generic/types" |
11 | 12 | | languagetypes "github.com/bearer/bearer/new/language/types" |
12 | 13 | | ) |
13 | 14 | | |
14 | 15 | | type objectDetector struct { |
15 | 16 | | types.DetectorBase |
16 | | - | // Gathering properties |
| 17 | + | // Base |
17 | 18 | | objectPairQuery *tree.Query |
| 19 | + | classQuery *tree.Query |
18 | 20 | | // Naming |
19 | 21 | | assignmentQuery *tree.Query |
20 | | - | parentPairQuery *tree.Query |
21 | | - | // Variables |
22 | | - | variableDeclarationQuery *tree.Query |
23 | | - | objectDeconstructionQuery *tree.Query |
24 | | - | // class |
25 | | - | classNameQuery *tree.Query |
26 | | - | constructorQuery *tree.Query |
27 | | - | // properties |
| 22 | + | // Projection |
28 | 23 | | memberExpressionQuery *tree.Query |
29 | 24 | | subscriptExpressionQuery *tree.Query |
| 25 | + | callQuery *tree.Query |
| 26 | + | // FIXME: what to do with this? |
| 27 | + | objectDeconstructionQuery *tree.Query |
30 | 28 | | } |
31 | 29 | | |
32 | 30 | | func New(lang languagetypes.Language) (types.Detector, error) { |
33 | 31 | | // { first_name: ..., ... } |
34 | | - | objectPairQuery, err := lang.CompileQuery(`(object (pair) @pair) @root`) |
| 32 | + | objectPairQuery, err := lang.CompileQuery(`(object (pair key: (_) @key value: (_) @value) @pair) @root`) |
35 | 33 | | if err != nil { |
36 | 34 | | return nil, fmt.Errorf("error compiling object pair query: %s", err) |
37 | 35 | | } |
38 | 36 | | |
| 37 | + | // user = <object> |
39 | 38 | | // const user = <object> |
40 | 39 | | // var user = <object> |
41 | 40 | | // let user = <object> |
42 | | - | variableDeclarationQuery, err := lang.CompileQuery(`(variable_declarator name: (identifier) @name value: (_) @value) @root`) |
| 41 | + | assignmentQuery, err := lang.CompileQuery(`[ |
| 42 | + | (assignment_expression left: (identifier) @name right: (_) @value) |
| 43 | + | (variable_declarator name: (identifier) @name value: (_) @value) |
| 44 | + | ] @root`) |
43 | 45 | | if err != nil { |
44 | 46 | | return nil, fmt.Errorf("error compiling assignment query: %s", err) |
45 | 47 | | } |
| skipped 1 lines |
47 | 49 | | // const { user } = <object> |
48 | 50 | | // let { user } = <object> |
49 | 51 | | // var { user } = <object> |
50 | | - | objectDeconstructionQuery, err := lang.CompileQuery(`(variable_declarator name:(object_pattern (shorthand_property_identifier_pattern) @match ) value: (_) @value) @root`) |
| 52 | + | objectDeconstructionQuery, err := lang.CompileQuery(`(variable_declarator name: (object_pattern (shorthand_property_identifier_pattern) @match) value: (_) @value) @root`) |
51 | 53 | | if err != nil { |
52 | 54 | | return nil, fmt.Errorf("error compiling object deconstruction query: %s", err) |
53 | 55 | | } |
54 | 56 | | |
55 | | - | // user = <object> |
56 | | - | assignmentQuery, err := lang.CompileQuery(`(assignment_expression left: (identifier) @left right: (_) @right) @root`) |
57 | | - | if err != nil { |
58 | | - | return nil, fmt.Errorf("error compiling assignment query: %s", err) |
59 | | - | } |
60 | | - | |
61 | | - | // { user: <object> } |
62 | | - | parentPairQuery, err := lang.CompileQuery(`(pair key: (_) @key value: (_) @value) @root`) |
63 | | - | if err != nil { |
64 | | - | return nil, fmt.Errorf("error compiling parent pair query: %s", err) |
65 | | - | } |
66 | | - | |
67 | | - | // new User() |
68 | | - | constructorQuery, err := lang.CompileQuery(`(new_expression constructor: (identifier) @name) @root`) |
69 | | - | if err != nil { |
70 | | - | return nil, fmt.Errorf("error compiling class name query: %s", err) |
71 | | - | } |
72 | | - | |
73 | | - | classNameQuery, err := lang.CompileQuery(`(class_declaration name: (type_identifier) @name) @root`) |
| 57 | + | // class User { |
| 58 | + | // constructor(name, surname) {} |
| 59 | + | // GetName() {} |
| 60 | + | // } |
| 61 | + | classQuery, err := lang.CompileQuery(` |
| 62 | + | (class_declaration |
| 63 | + | name: (type_identifier) @class_name |
| 64 | + | body: (class_body |
| 65 | + | (method_definition name: (property_identifier) @method_name (formal_parameters) @params) |
| 66 | + | ) |
| 67 | + | ) @root`) |
74 | 68 | | if err != nil { |
75 | | - | return nil, fmt.Errorf("error compiling class name query: %s", err) |
| 69 | + | return nil, fmt.Errorf("error compiling class query: %s", err) |
76 | 70 | | } |
77 | 71 | | |
78 | 72 | | // user.name |
| skipped 8 lines |
87 | 81 | | return nil, fmt.Errorf("error compiling subscript expression query %s", err) |
88 | 82 | | } |
89 | 83 | | |
| 84 | + | callQuery, err := lang.CompileQuery(`(call_expression function: (_) @function) @root`) |
| 85 | + | if err != nil { |
| 86 | + | return nil, fmt.Errorf("error compiling call query: %s", err) |
| 87 | + | } |
| 88 | + | |
90 | 89 | | return &objectDetector{ |
91 | 90 | | objectPairQuery: objectPairQuery, |
92 | 91 | | assignmentQuery: assignmentQuery, |
93 | | - | variableDeclarationQuery: variableDeclarationQuery, |
94 | 92 | | objectDeconstructionQuery: objectDeconstructionQuery, |
95 | | - | parentPairQuery: parentPairQuery, |
96 | | - | constructorQuery: constructorQuery, |
97 | | - | classNameQuery: classNameQuery, |
| 93 | + | classQuery: classQuery, |
98 | 94 | | memberExpressionQuery: memberExpressionQuery, |
99 | 95 | | subscriptExpressionQuery: subscriptExpressionQuery, |
| 96 | + | callQuery: callQuery, |
100 | 97 | | }, nil |
101 | 98 | | } |
102 | 99 | | |
| skipped 9 lines |
112 | 109 | | node *tree.Node, |
113 | 110 | | evaluator types.Evaluator, |
114 | 111 | | ) ([]interface{}, error) { |
115 | | - | detections, err := detector.getobject(node, evaluator) |
| 112 | + | detections, err := detector.getObject(node, evaluator) |
116 | 113 | | if len(detections) != 0 || err != nil { |
117 | 114 | | return detections, err |
118 | 115 | | } |
119 | 116 | | |
120 | | - | detections, err = detector.getAssigment(node, evaluator) |
121 | | - | if len(detections) != 0 || err != nil { |
122 | | - | return detections, err |
123 | | - | } |
124 | | - | |
125 | | - | detections, err = detector.getVariableDeclaration(node, evaluator) |
126 | | - | if len(detections) != 0 || err != nil { |
127 | | - | return detections, err |
128 | | - | } |
129 | | - | |
130 | | - | detections, err = detector.getObjectDeconstruction(node, evaluator) |
| 117 | + | detections, err = detector.getAssignment(node, evaluator) |
131 | 118 | | if len(detections) != 0 || err != nil { |
132 | 119 | | return detections, err |
133 | 120 | | } |
| skipped 3 lines |
137 | 124 | | return detections, err |
138 | 125 | | } |
139 | 126 | | |
140 | | - | detections, err = detector.getConstructor(node, evaluator) |
141 | | - | if len(detections) != 0 || err != nil { |
142 | | - | return detections, err |
143 | | - | } |
144 | | - | |
145 | | - | detections, err = detector.getProperties(node, evaluator) |
146 | | - | if len(detections) != 0 || err != nil { |
147 | | - | return detections, err |
148 | | - | } |
149 | | - | |
150 | | - | return detector.nameParentPairObject(node, evaluator) |
| 127 | + | return detector.getProjections(node, evaluator) |
151 | 128 | | } |
152 | 129 | | |
153 | | - | func (detector *objectDetector) getobject( |
| 130 | + | func (detector *objectDetector) getObject( |
154 | 131 | | node *tree.Node, |
155 | 132 | | evaluator types.Evaluator, |
156 | 133 | | ) ([]interface{}, error) { |
157 | 134 | | results, err := detector.objectPairQuery.MatchAt(node) |
158 | | - | if err != nil { |
| 135 | + | if len(results) == 0 || err != nil { |
159 | 136 | | return nil, err |
160 | 137 | | } |
161 | 138 | | |
162 | | - | if len(results) == 0 { |
163 | | - | return nil, nil |
164 | | - | } |
165 | | - | |
166 | | - | var properties []*types.Detection |
| 139 | + | var properties []generictypes.Property |
167 | 140 | | for _, result := range results { |
168 | | - | nodeProperties, err := evaluator.ForNode(result["pair"], "property", false) |
| 141 | + | var name string |
| 142 | + | key := result["key"] |
| 143 | + | |
| 144 | + | switch key.Type() { |
| 145 | + | case "string": // {"user": "admin_user"} |
| 146 | + | name = stringutil.StripQuotes(key.Content()) |
| 147 | + | case "property_identifier": // { user: "admin_user"} |
| 148 | + | name = key.Content() |
| 149 | + | } |
| 150 | + | |
| 151 | + | if name == "" { |
| 152 | + | continue |
| 153 | + | } |
| 154 | + | |
| 155 | + | propertyObjects, err := evaluator.ForTree(result["value"], "object", true) |
169 | 156 | | if err != nil { |
170 | 157 | | return nil, err |
171 | 158 | | } |
172 | 159 | | |
173 | | - | properties = append(properties, nodeProperties...) |
| 160 | + | pairNode := result["pair"] |
| 161 | + | |
| 162 | + | if len(propertyObjects) == 0 { |
| 163 | + | properties = append(properties, generictypes.Property{ |
| 164 | + | Name: name, |
| 165 | + | Node: pairNode, |
| 166 | + | }) |
| 167 | + | |
| 168 | + | continue |
| 169 | + | } |
| 170 | + | |
| 171 | + | for _, propertyObject := range propertyObjects { |
| 172 | + | properties = append(properties, generictypes.Property{ |
| 173 | + | Name: name, |
| 174 | + | Node: pairNode, |
| 175 | + | Object: propertyObject, |
| 176 | + | }) |
| 177 | + | } |
174 | 178 | | } |
175 | 179 | | |
176 | 180 | | return []interface{}{generictypes.Object{Properties: properties}}, nil |
177 | 181 | | } |
178 | 182 | | |
179 | | - | func (detector *objectDetector) getAssigment( |
| 183 | + | func (detector *objectDetector) getAssignment( |
180 | 184 | | node *tree.Node, |
181 | 185 | | evaluator types.Evaluator, |
182 | 186 | | ) ([]interface{}, error) { |
| skipped 2 lines |
185 | 189 | | return nil, err |
186 | 190 | | } |
187 | 191 | | |
188 | | - | objects, err := evaluator.ForNode(result["right"], "object", true) |
| 192 | + | valueObjects, err := generic.GetNonVirtualObjects(evaluator, result["value"]) |
189 | 193 | | if err != nil { |
190 | 194 | | return nil, err |
191 | 195 | | } |
192 | 196 | | |
193 | | - | var detections []interface{} |
194 | | - | for _, object := range objects { |
195 | | - | objectData := object.Data.(generictypes.Object) |
196 | | - | |
197 | | - | if objectData.Name == "" { |
198 | | - | detections = append(detections, generictypes.Object{ |
199 | | - | Name: result["left"].Content(), |
200 | | - | Properties: objectData.Properties, |
201 | | - | }) |
202 | | - | } |
| 197 | + | var objects []interface{} |
| 198 | + | for _, object := range valueObjects { |
| 199 | + | objects = append(objects, generictypes.Object{ |
| 200 | + | IsVirtual: true, |
| 201 | + | Properties: []generictypes.Property{{ |
| 202 | + | Name: result["name"].Content(), |
| 203 | + | Node: node, |
| 204 | + | Object: object, |
| 205 | + | }}, |
| 206 | + | }) |
203 | 207 | | } |
204 | 208 | | |
205 | | - | return detections, nil |
| 209 | + | return objects, nil |
206 | 210 | | } |
207 | 211 | | |
208 | 212 | | func (detector *objectDetector) getClass(node *tree.Node, evaluator types.Evaluator) ([]interface{}, error) { |
209 | | - | result, err := detector.classNameQuery.MatchOnceAt(node) |
210 | | - | if result == nil || err != nil { |
| 213 | + | results, err := detector.classQuery.MatchAt(node) |
| 214 | + | if len(results) == 0 || err != nil { |
211 | 215 | | return nil, err |
212 | 216 | | } |
213 | 217 | | |
214 | | - | data := generictypes.Object{ |
215 | | - | Name: result["name"].Content(), |
216 | | - | Properties: []*types.Detection{}, |
217 | | - | } |
| 218 | + | className := results[0]["class_name"].Content() |
218 | 219 | | |
219 | | - | body := node.ChildByFieldName("body") |
| 220 | + | var properties []generictypes.Property |
| 221 | + | for _, result := range results { |
| 222 | + | methodName := result["method_name"].Content() |
| 223 | + | if methodName == "constructor" { |
| 224 | + | params := result["params"] |
220 | 225 | | |
221 | | - | for i := 0; i < body.ChildCount(); i++ { |
222 | | - | detections, err := evaluator.ForNode(body.Child(i), "property", true) |
223 | | - | if err != nil { |
224 | | - | return nil, err |
225 | | - | } |
226 | | - | data.Properties = append(data.Properties, detections...) |
227 | | - | } |
| 226 | + | for i := 0; i < params.ChildCount(); i++ { |
| 227 | + | param := params.Child(i) |
| 228 | + | if param.Type() != "identifier" { |
| 229 | + | continue |
| 230 | + | } |
228 | 231 | | |
229 | | - | return []interface{}{data}, nil |
230 | | - | } |
231 | | - | |
232 | | - | func (detector *objectDetector) getConstructor(node *tree.Node, evaluator types.Evaluator) ([]interface{}, error) { |
233 | | - | result, err := detector.constructorQuery.MatchOnceAt(node) |
234 | | - | if result == nil || err != nil { |
235 | | - | return nil, err |
236 | | - | } |
237 | | - | |
238 | | - | data := generictypes.Object{ |
239 | | - | Name: result["name"].Content(), |
240 | | - | } |
241 | | - | |
242 | | - | return []interface{}{data}, nil |
243 | | - | } |
244 | | - | |
245 | | - | func (detector *objectDetector) nameParentPairObject( |
246 | | - | node *tree.Node, |
247 | | - | evaluator types.Evaluator, |
248 | | - | ) ([]interface{}, error) { |
249 | | - | result, err := detector.parentPairQuery.MatchOnceAt(node) |
250 | | - | if result == nil || err != nil { |
251 | | - | return nil, err |
252 | | - | } |
253 | | - | |
254 | | - | objects, err := evaluator.ForNode(result["value"], "object", true) |
255 | | - | if err != nil { |
256 | | - | return nil, err |
257 | | - | } |
258 | | - | |
259 | | - | var detections []interface{} |
260 | | - | for _, object := range objects { |
261 | | - | objectData := object.Data.(generictypes.Object) |
262 | | - | |
263 | | - | objectName := result["key"].Content() |
264 | | - | objectNameNode := result["key"] |
265 | | - | if objectNameNode.Type() == "string" { |
266 | | - | objectName = stringutil.StripQuotes(objectName) |
| 232 | + | properties = append(properties, generictypes.Property{Name: param.Content()}) |
| 233 | + | } |
| 234 | + | } else { |
| 235 | + | properties = append(properties, generictypes.Property{Name: methodName}) |
267 | 236 | | } |
268 | | - | |
269 | | - | detections = append(detections, generictypes.Object{ |
270 | | - | Name: objectName, |
271 | | - | Properties: objectData.Properties, |
272 | | - | }, |
273 | | - | ) |
274 | 237 | | } |
275 | 238 | | |
276 | | - | return detections, nil |
| 239 | + | return []interface{}{generictypes.Object{ |
| 240 | + | Properties: []generictypes.Property{{ |
| 241 | + | Name: className, |
| 242 | + | Object: &types.Detection{ |
| 243 | + | DetectorType: "object", |
| 244 | + | MatchNode: node, |
| 245 | + | Data: generictypes.Object{ |
| 246 | + | Properties: properties, |
| 247 | + | }, |
| 248 | + | }, |
| 249 | + | }}, |
| 250 | + | }}, nil |
277 | 251 | | } |
278 | 252 | | |
279 | 253 | | func (detector *objectDetector) Close() { |
280 | 254 | | detector.objectPairQuery.Close() |
281 | 255 | | detector.assignmentQuery.Close() |
282 | | - | detector.variableDeclarationQuery.Close() |
283 | 256 | | detector.objectDeconstructionQuery.Close() |
284 | | - | detector.parentPairQuery.Close() |
285 | | - | detector.classNameQuery.Close() |
| 257 | + | detector.classQuery.Close() |
286 | 258 | | detector.memberExpressionQuery.Close() |
287 | 259 | | detector.subscriptExpressionQuery.Close() |
| 260 | + | detector.callQuery.Close() |
288 | 261 | | } |
289 | 262 | | |