fix FunctionBuiltins: __truffle_source__ (methods allow arbitrary callables as functions)

cosminbasca · cosminbasca · commit 2836b89d2626 · 2018-12-17T12:11:50.000+01:00
diff --git a/graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/objects/function/FunctionBuiltins.java b/graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/objects/function/FunctionBuiltins.java
@@ -249,10 +249,13 @@ Object doFunction(PFunction function) {
         }
 
         @Specialization
-        Object doMethod(PMethod function) {
-            String sourceCode = function.getFunction().getSourceCode();
-            if (sourceCode != null) {
-                return sourceCode;
+        Object doMethod(PMethod method) {
+            Object function = method.getFunction();
+            if (function instanceof PFunction) {
+                String sourceCode = ((PFunction) function).getSourceCode();
+                if (sourceCode != null) {
+                    return sourceCode;
+                }
             }
             return PNone.NONE;
         }

Original file line number	Diff line number	Diff line change
`@@ -249,10 +249,13 @@ Object doFunction(PFunction function) {`
`249`	`249`	`}`
`250`	`250`
`251`	`251`	`@Specialization`
`252`		`- Object doMethod(PMethod function) {`
`253`		`- String sourceCode = function.getFunction().getSourceCode();`
`254`		`- if (sourceCode != null) {`
`255`		`- return sourceCode;`
	`252`	`+ Object doMethod(PMethod method) {`
	`253`	`+ Object function = method.getFunction();`
	`254`	`+ if (function instanceof PFunction) {`
	`255`	`+ String sourceCode = ((PFunction) function).getSourceCode();`
	`256`	`+ if (sourceCode != null) {`
	`257`	`+ return sourceCode;`
	`258`	`+ }`
`256`	`259`	`}`
`257`	`260`	`return PNone.NONE;`
`258`	`261`	`}`