For classes, add a field to the script object just like functions.

Then, (external) classes can be found by the PA when they are imported. Otherwise, they cannot be found.
ponder-lab · Feb 14, 2024 · 4fe8cc9 · 4fe8cc9
1 parent 1dc720a
commit 4fe8cc9
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/com.ibm.wala.cast.python/source/com/ibm/wala/cast/python/ir/PythonCAstToIRTranslator.java b/com.ibm.wala.cast.python/source/com/ibm/wala/cast/python/ir/PythonCAstToIRTranslator.java
@@ -613,6 +613,20 @@ protected void leaveTypeEntity(
                             .PutInstruction(code.cfg().getCurrentInstruction(), v, val, fr));
               });
     }
+
+    code.cfg()
+        .unknownInstructions(
+            () -> {
+              FieldReference fnField =
+                  FieldReference.findOrCreate(
+                      PythonTypes.Root,
+                      Atom.findOrCreateUnicodeAtom(n.getName()),
+                      PythonTypes.Root);
+              code.cfg()
+                  .addInstruction(
+                      Python.instructionFactory()
+                          .PutInstruction(code.cfg().getCurrentInstruction(), 1, v, fnField));
+            });
   }
 
   @Override