Merge pull request #218 from franklinsch/selector-public-funcs

Don't include non-public functions in function selector
flintlang · May 17, 2018 · 2ebf275 · 2ebf275
2 parents 8f71c3c + 41378d2
commit 2ebf275
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/Sources/IRGen/IULIAContract.swift b/Sources/IRGen/IULIAContract.swift
@@ -34,8 +34,10 @@ struct IULIAContract {
 
     let functionsCode = functions.map({ $0.rendered() }).joined(separator: "\n\n").indented(by: 6)
 
+    let publicFunctions = functions.filter { $0.functionDeclaration.isPublic }
+
     // Create a function selector, to determine which function is called in the Ethereum transaction.
-    let functionSelector = IULIAFunctionSelector(functions: functions)
+    let functionSelector = IULIAFunctionSelector(functions: publicFunctions)
     let selectorCode = functionSelector.rendered().indented(by: 6)
 
     // Generate code for each function in the structs.