Commit d5bcc01e authored by obscuren's avatar obscuren

Fixed shebang check. Previously it would hang on an unknown shebang

parent 633027d9
......@@ -10,16 +10,21 @@ import (
// General compile function
func Compile(script string) (ret []byte, err error) {
c := strings.Split(script, "\n")[0]
if len(script) > 2 {
line := strings.Split(script, "\n")[0]
if c == "#!serpent" {
if line[0:2] == "#!" {
switch line {
case "#!serpent":
byteCode, err := serpent.Compile(script)
if err != nil {
return nil, err
}
return byteCode, nil
}
} else {
compiler := mutan.NewCompiler(backend.NewEthereumBackend())
byteCode, errors := compiler.Compile(strings.NewReader(script))
if len(errors) > 0 {
......@@ -34,4 +39,7 @@ func Compile(script string) (ret []byte, err error) {
return byteCode, nil
}
}
return nil, nil
}
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment