Example #1
0
def test_typecheck(filename="Object.java"):
    import typechecker
    import environment
    environment.cached_environments = {}
    environment.cached_trees = {}
    # stdlib = all_files_in_dir("test_cases/stdlib/6.0")
    trees = typechecker.check_files([filename])
    to_file(trees[-1])
Example #2
0
def check_and_generate(files):
    init_globals()  # reset global state
    try:
        out = open("output/out.s", "w")

        trees = typechecker.check_files(files)
        initializer_list = []

        mega_vtable_offset = 1
        type_tag = 0
        string_literal_index = 0
        for tree in trees:
            clazz = find_type_decl(tree)
            clazz.type_tag = type_tag
            method_list.class_list += [clazz]
            type_tag += 1
            methods = (
                tree.findall(".//method")
                + tree.findall(".//constructor_declaration")
                + tree.findall(".//abstract_method")
            )
            for method in methods:
                method.mega_vtable_offset = mega_vtable_offset
                mega_vtable_offset += 1
                method_list.method_list += [method]
                if "static" in modifiers(method):
                    slot = -2
                else:
                    slot = -3
                for param in method.findall(".//param"):
                    param.slot = slot
                    slot -= 1

            counters.mega_vtable_members = mega_vtable_offset

            for string_literal in tree.findall(".//string_literal"):
                string_literal.index = string_literal_index
                string_literal_index += 1

        out.write(instanceof_table(trees))

        for tree in trees:
            CurrentFile.name = tree.filename
            CurrentFile.mangled_name = CurrentFile.name.replace(".", "_").replace("$", "_").replace("/", "_")
            CurrentFile.static_slot = 0

            for this in tree.findall(".//tok_this"):
                this.slot = -2

            generate(tree)
            to_file(tree, tree.filename + ".xml")

            out.write("section .data\n")
            for clazz in tree.findall(".//class"):
                out.write(mega_vtable_for_class(clazz))

            out.write("section .text\n")
            initializer_name = "static_init_" + CurrentFile.mangled_name
            out.write(
                """
; global {initializer_name}
{initializer_name}:
push ebp
mov ebp, esp
sub esp, {frame_size}
""".format(
                    initializer_name=initializer_name, frame_size=CurrentFile.static_slot * 4
                )
            )
            initializer_list.append(initializer_name)

            for field in tree.findall(".//field"):
                if "static" in modifiers(field):
                    mangled_name = mangle_field_name(field)
                    out.write(
                        """
; initializing {mangled_name}
{field_assembly}
mov eax, {mangled_name}
mov ebx, {assigned_value}
mov DWORD [eax], ebx
""".format(
                            mangled_name=mangled_name,
                            field_assembly=field.assembly,
                            assigned_value=stack_slot_to_operand(field.slot),
                        )
                    )

            out.write(
                """
leave
ret
; done global static initialization
"""
            )

            for constructor in tree.findall(".//constructor_declaration"):
                out.write(constructor.assembly + "\n")

            for method in tree.findall(".//method"):
                if method.find("block") is not None:
                    out.write(method.assembly + "\n")

            out.write("\nsection .data\n")
            for string_literal in tree.findall(".//string_literal"):
                string_value = string_literal.get("value")
                expanded_value = expand_string_literal(string_value)
                out.write("\n; string literal " + string_value + "\n")
                out.write("__string_literal_" + str(string_literal.index) + ":\n")
                out.write("dd _vtable_java_lang_$Array_ ; vtable pointer\n")
                out.write("dd -3 ; type tag for char\n")
                out.write("dd " + str(len(expanded_value)) + " ; string length\n")
                for character in expanded_value:
                    out.write("dd " + str(hex(ord(character))) + "\n")
                out.write("")

            out.write("\nsection .bss\n")
            for field in tree.findall(".//field"):
                if "static" in modifiers(field):
                    mangled_name = mangle_field_name(field)
                    out.write("""{mangled_name} resb 4""".format(mangled_name=mangled_name) + "\n")

        out.write("section .text\n")
        out.write("; prelude\n")
        out.write("extern __malloc\n")
        out.write("extern __debexit\n")
        out.write("extern __exception\n")
        out.write("extern NATIVEjava.io.OutputStream.nativeWrite\n")
        out.write("global _start\n")
        out.write("_start:\n")
        out.write("call install_segv_handler\n")

        for initializer in initializer_list:
            out.write("call " + initializer + "\n")

        class_name = mangle_class_name(find_type_decl(trees[1]).get("canonical_name"))
        out.write("call " + class_name + "test_\n")
        out.write("mov ebx, eax\n")
        out.write("mov eax, 1\n")
        out.write("int 0x80 ; invoke exit(1)\n")
        out.write("; end prelude\n")
        out.write(
            """
%define __NR_signal 48
%define SIGSEGV     11
global install_segv_handler
install_segv_handler:
  mov eax, __NR_signal
  mov ebx, SIGSEGV
  mov ecx, __exception
  int 0x80
  ret
"""
        )

        return 0
    except JoosSyntaxException, e:
        if Testing.testing:
            return 42
        else:
            out.write("\n" + e.msg + "\n")
            raise