feat[venom]: add venom parser

setup.py

@@ -94,6 +94,7 @@ def _global_version(version):
         "asttokens>=2.0.5,<3",
         "pycryptodome>=3.5.1,<4",
         "packaging>=23.1,<24",
+        "lark>=1.0.0,<2",
         "importlib-metadata",
         "wheel",
     ],
@@ -105,6 +106,7 @@ def _global_version(version):
             "vyper=vyper.cli.vyper_compile:_parse_cli_args",
             "fang=vyper.cli.vyper_ir:_parse_cli_args",
             "vyper-json=vyper.cli.vyper_json:_parse_cli_args",
+            "venom=vyper.cli.vyper_venom:_parse_cli_args"
         ]
     },
     classifiers=[

vyper/cli/vyper_venom.py

@@ -0,0 +1,54 @@
+#!/usr/bin/env python3
+import argparse
+import sys
+import vyper
+from vyper.venom.parser import parse_venom
+from vyper.venom import generate_assembly_experimental, run_passes_on
+from vyper.compiler.settings import OptimizationLevel, set_global_settings, Settings
+from vyper.compiler.phases import generate_bytecode
+import vyper.evm.opcodes as evm
+
+
+def _parse_cli_args():
+    return _parse_args(sys.argv[1:])
+
+
+def _parse_args(argv: list[str]):
+    parser = argparse.ArgumentParser(
+        description="Venom EVM IR parser & compiler",
+        formatter_class=argparse.RawTextHelpFormatter,
+    )
+    parser.add_argument("input_file", help="Venom sourcefile", nargs='?')
+    parser.add_argument("--version", action="version", version=vyper.__long_version__)
+    parser.add_argument(
+        "--evm-version",
+        help=f"Select desired EVM version (default {evm.DEFAULT_EVM_VERSION})",
+        choices=list(evm.EVM_VERSIONS),
+        dest="evm_version",
+    )
+
+    args = parser.parse_args(argv)
+
+    if args.evm_version is not None:
+        set_global_settings(Settings(evm_version=args.evm_version))
+
+    if args.input_file is None:
+        if not sys.stdin.isatty():
+            venom_source = sys.stdin.read()
+        else:
+            # No input provided
+            print("Error: No input provided")
+            sys.exit(1)
+    else:
+        with open(args.input_file, 'r') as f:
+            venom_source = f.read()
+
+    ctx = parse_venom(venom_source)
+    run_passes_on(ctx, OptimizationLevel.default())
+    asm = generate_assembly_experimental(ctx)
+    bytecode = generate_bytecode(asm, compiler_metadata=None)
+    print(bytecode.hex())
+
+
+if __name__ == "__main__":
+    _parse_args(sys.argv[1:])

vyper/venom/README.md

@@ -29,59 +29,39 @@ Venom employs two scopes: global and function level.
 ### Example code
 
 ```llvm
-IRFunction: global
-
-global:
-    %1 = calldataload 0
-    %2 = shr 224, %1
-    jmp label %selector_bucket_0
-
-selector_bucket_0:
-    %3 = xor %2, 1579456981
-    %4 = iszero %3
-    jnz label %1, label %2, %4
-
-1:  IN=[selector_bucket_0] OUT=[9]
-    jmp label %fallback
-
-2:
-    %5 = callvalue
-    %6 = calldatasize
-    %7 = lt %6, 164
-    %8 = or %5, %7
-    %9 = iszero %8
-    assert %9
-    stop
-
-fallback:
-    revert 0, 0
+fn global => {
+    global:
+        %1 = calldataload 0
+        %2 = shr 224, %1
+        jmp @selector_bucket_0
+
+    selector_bucket_0:
+        %3 = xor %2, 1579456981
+        %4 = iszero %3
+        jnz @1, @2, %4
+
+    1:
+        jmp @fallback
+
+    2:
+        %5 = callvalue
+        %6 = calldatasize
+        %7 = lt %6, 164
+        %8 = or %5, %7
+        %9 = iszero %8
+        assert %9
+        stop
+
+    fallback:
+        revert 0, 0
+}
+
+data:
 ```
 
 ### Grammar
 
-Below is a (not-so-complete) grammar to describe the text format of Venom IR:
-
-```llvm
-program              ::= function_declaration*
-
-function_declaration ::= "IRFunction:" identifier input_list? output_list? "=>" block
-
-input_list           ::= "IN=" "[" (identifier ("," identifier)*)? "]"
-output_list          ::= "OUT=" "[" (identifier ("," identifier)*)? "]"
-
-block                ::= label ":" input_list? output_list? "=>{" operation* "}"
-
-operation            ::= "%" identifier "=" opcode operand ("," operand)*
-                     |  opcode operand ("," operand)*
-
-opcode               ::= "calldataload" | "shr" | "shl" | "and" |  "add" | "codecopy" | "mload" | "jmp" | "xor" | "iszero" |  "jnz" | "label" | "lt" | "or" | "assert" | "callvalue" | "calldatasize" | "alloca" | "calldatacopy" |  "invoke" | "gt" | ...
-
-operand              ::= "%" identifier | label | integer | "label" "%" identifier
-label                ::= "%" identifier
-
-identifier           ::= [a-zA-Z_][a-zA-Z0-9_]*
-integer              ::= [0-9]+
-```
+To see a definition of grammar see the [venom parser](./parser.py)
 
 ## Implementation
 

vyper/venom/__init__.py

@@ -72,10 +72,14 @@ def _run_passes(fn: IRFunction, optimize: OptimizationLevel) -> None:
     DFTPass(ac, fn).run_pass()
 
 
+def run_passes_on(ctx: IRContext, optimize: OptimizationLevel):
+    for fn in ctx.functions.values():
+        _run_passes(fn, optimize)
+
+
 def generate_ir(ir: IRnode, optimize: OptimizationLevel) -> IRContext:
     # Convert "old" IR to "new" IR
     ctx = ir_node_to_venom(ir)
-    for fn in ctx.functions.values():
-        _run_passes(fn, optimize)
+    run_passes_on(ctx, optimize)
 
     return ctx

vyper/venom/function.py

@@ -12,7 +12,6 @@ class IRFunction:
     name: IRLabel  # symbol name
     ctx: "IRContext"  # type: ignore # noqa: F821
     args: list
-    last_label: int
     last_variable: int
     _basic_block_dict: dict[str, IRBasicBlock]
 
@@ -182,7 +181,6 @@ def chain_basic_blocks(self) -> None:
     def copy(self):
         new = IRFunction(self.name)
         new._basic_block_dict = self._basic_block_dict.copy()
-        new.last_label = self.last_label
         new.last_variable = self.last_variable
         return new
 

vyper/venom/parser.py

@@ -0,0 +1,139 @@
+from vyper.venom.context import IRContext
+from vyper.venom.basicblock import (
+    IRLabel, IRVariable, IRLiteral, IROperand, IRInstruction, IRBasicBlock
+)
+from vyper.venom.function import IRFunction
+from lark import Lark, Transformer
+from functools import reduce
+
+
+VENOM_PARSER = Lark(
+    '''
+    %import common.CNAME
+    %import common.DIGIT
+    %import common.LETTER
+    %import common.WS
+    %import common.INT
+
+    start: function* data_section
+    function: "fn" NAME "=>" "{" block* "}"
+    data_section: "data:" call*
+
+    block: NAME ":" (call | assignment)*
+
+    assignment: VAR_IDENT "=" expr
+    expr: call | CONST
+    call: CNAME operands_list
+
+    operands_list: (operand ("," operand)*)?
+
+    operand: VAR_IDENT | CONST | LABEL
+    CONST: INT
+
+    VAR_IDENT: "%" INT (":" INT)?
+    LABEL: "@" NAME
+    NAME: (DIGIT|LETTER|"_")+
+
+    %ignore WS
+    '''
+)
+
+
+def set_last_var(fn: IRFunction):
+    for block in fn.get_basic_blocks():
+        output_vars = (
+            instruction.output
+            for instruction in block.instructions
+            if instruction.output is not None
+        )
+        for output_var in output_vars:
+            assert isinstance(output_var, IRVariable)
+            value = output_var.value
+            assert value.startswith('%')
+            value = value.replace('%', '', 1)
+            if not value.isdigit():
+                continue
+            fn.last_variable = max(fn.last_variable, int(value))
+
+
+class VenomTransformer(Transformer):
+    def start(self, children) -> IRContext:
+        ctx = IRContext()
+        funcs = children[:-1]
+        data_section = children[-1]
+        for fn_name, blocks in funcs:
+            fn = ctx.create_function(fn_name)
+            for block_name, instructions in blocks:
+                bb = IRBasicBlock(IRLabel(block_name), fn)
+
+                for instruction in instructions:
+                    bb.insert_instruction(instruction)
+
+                # Manually insert because we need to override function entry
+                fn._basic_block_dict[block_name] = bb
+
+            set_last_var(fn)
+
+        ctx.data_segment = data_section
+
+        ctx.chain_basic_blocks()
+
+        return ctx
+
+    def function(self, children) -> tuple[str, list[tuple[str, list[IRInstruction]]]]:
+        name, *blocks = children
+        return name, blocks
+
+    def data_section(self, children):
+        return children
+
+    def block(self, children) -> tuple[str, list[IRInstruction]]:
+        label, *instructions = children
+        return label, instructions
+
+    def assignment(self, children) -> IRInstruction:
+        to, value = children
+        if isinstance(value, IRInstruction):
+            value.output = to
+            return value
+        elif isinstance(value, IRLiteral):
+            return IRInstruction('store', [value], output=to)
+        else:
+            raise TypeError(f'Unexpected value {value} of type {type(value)}')
+
+    def expr(self, children):
+        return children[0]
+
+    def call(self, children) -> IRInstruction:
+        name, operands = children
+        # reverse operands because top->bottom is more intuitive but Venom does bottom->top
+        return IRInstruction(name, reversed(operands))
+
+    def operands_list(self, children) -> list[IROperand]:
+        return children
+
+    def operand(self, children) -> IROperand:
+        return children[0]
+
+    def LABEL(self, label) -> IRLabel:
+        return IRLabel(label[1:])
+
+    def VAR_IDENT(self, var_ident) -> IRVariable:
+        parts = var_ident[1:].split(':', maxsplit=1)
+        return IRVariable(*parts)
+
+    def CONST(self, val) -> IRLiteral:
+        return IRLiteral(int(val))
+
+    def CNAME(self, val) -> str:
+        return val.value
+
+    def NAME(self, val) -> str:
+        return val.value
+
+
+def parse_venom(source: str) -> IRContext:
+    tree = VENOM_PARSER.parse(source)
+    ctx = VenomTransformer().transform(tree)
+    assert isinstance(ctx, IRContext)
+    return ctx

vyper/venom/passes/sccp/sccp.py

@@ -143,7 +143,7 @@ def _handle_SSA_work_item(self, work_item: SSAWorkListItem):
             self._visit_expr(work_item.inst)
 
     def _lookup_from_lattice(self, op: IROperand) -> LatticeItem:
-        assert isinstance(op, IRVariable), "Can't get lattice for non-variable"
+        assert isinstance(op, IRVariable), f"Can't get lattice for non-variable ({op})"
         lat = self.lattice[op]
         assert lat is not None, f"Got undefined var {op}"
         return lat