Python: Clean up Ssl.qll

tausbn · web-flow · commit e90ec807ef96 · 2021-06-15T15:04:29.000Z
diff --git a/python/ql/src/Security/CWE-327/Ssl.qll b/python/ql/src/Security/CWE-327/Ssl.qll
@@ -39,11 +39,11 @@ API::Node sslContextInstance() {
   result = API::moduleImport("ssl").getMember(["SSLContext", "create_default_context"]).getReturn()
 }
 
-class WrapSocketCall extends ConnectionCreation, DataFlow::CallCfgNode {
+class WrapSocketCall extends ConnectionCreation, DataFlow::MethodCallNode {
   WrapSocketCall() { this = sslContextInstance().getMember("wrap_socket").getACall() }
 
   override DataFlow::Node getContext() {
-    result = this.getFunction().(DataFlow::AttrRead).getObject()
+    result = this.getReceiver()
   }
 }
 

Original file line number	Diff line number	Diff line change
`@@ -39,11 +39,11 @@ API::Node sslContextInstance() {`
`39`	`39`	`result = API::moduleImport("ssl").getMember(["SSLContext", "create_default_context"]).getReturn()`
`40`	`40`	`}`
`41`	`41`
`42`		`-class WrapSocketCall extends ConnectionCreation, DataFlow::CallCfgNode {`
	`42`	`+class WrapSocketCall extends ConnectionCreation, DataFlow::MethodCallNode {`
`43`	`43`	`WrapSocketCall() { this = sslContextInstance().getMember("wrap_socket").getACall() }`
`44`	`44`
`45`	`45`	`override DataFlow::Node getContext() {`
`46`		`- result = this.getFunction().(DataFlow::AttrRead).getObject()`
	`46`	`+ result = this.getReceiver()`
`47`	`47`	`}`
`48`	`48`	`}`
`49`	`49`