Cb (C flat) compiler. Cb is simplified C..zip

package net.loveruby.cflat.sysdep.x86; import net.loveruby.cflat.sysdep.CodeGeneratorOptions; import net.loveruby.cflat.ir.*; import net.loveruby.cflat.entity.*; import net.loveruby.cflat.asm.*; import net.loveruby.cflat.ast.Location; import net.loveruby.cflat.utils.AsmUtils; import net.loveruby.cflat.utils.ListUtils; import net.loveruby.cflat.utils.ErrorHandler; import java.util.*; public class CodeGenerator implements net.loveruby.cflat.sysdep.CodeGenerator, IRVisitor<Void,Void>, ELFConstants { // #@@range/ctor{ final CodeGeneratorOptions options; final Type naturalType; final ErrorHandler errorHandler; public CodeGenerator(CodeGeneratorOptions options, Type naturalType, ErrorHandler errorHandler) { this.options = options; this.naturalType = naturalType; this.errorHandler = errorHandler; } // #@@} /** Compiles IR and generates assembly code. */ // #@@range/generate{ public AssemblyCode generate(IR ir) { locateSymbols(ir); return generateAssemblyCode(ir); } // #@@} static final String LABEL_SYMBOL_BASE = ".L"; static final String CONST_SYMBOL_BASE = ".LC"; // // locateSymbols // // #@@range/locateSymbols{ private void locateSymbols(IR ir) { SymbolTable constSymbols = new SymbolTable(CONST_SYMBOL_BASE); for (ConstantEntry ent : ir.constantTable().entries()) { locateStringLiteral(ent, constSymbols); } for (Variable var : ir.allGlobalVariables()) { locateGlobalVariable(var); } for (Function func : ir.allFunctions()) { locateFunction(func); } } // #@@} // #@@range/locateStringLiteral{ private void locateStringLiteral(ConstantEntry ent, SymbolTable syms) { ent.setSymbol(syms.newSymbol()); if (options.isPositionIndependent()) { Symbol offset = localGOTSymbol(ent.symbol()); ent.setMemref(mem(offset, GOTBaseReg())); } else { ent.setMemref(mem(ent.symbol())); ent.setAddress(imm(ent.symbol())); } } // #@@} // #@@range/locateGlobalVariable{ private void locateGlobalVariable(Entity ent) { Symbol sym = symbol(ent.symbolString(), ent.isPrivate()); if (options.isPositionIndependent()) { if (ent.isPrivate() || optimizeGvarAccess(ent)) { ent.setMemref(mem(localGOTSymbol(sym), GOTBaseReg())); } else { ent.setAddress(mem(globalGOTSymbol(sym), GOTBaseReg())); } } else { ent.setMemref(mem(sym)); ent.setAddress(imm(sym)); } } // #@@} // #@@range/locateFunction{ private void locateFunction(Function func) { func.setCallingSymbol(callingSymbol(func)); locateGlobalVariable(func); } // #@@} // #@@range/symbol{ private Symbol symbol(String sym, boolean isPrivate) { return isPrivate ? privateSymbol(sym) : globalSymbol(sym); } // #@@} // #@@range/globalSymbol{ private Symbol globalSymbol(String sym) { return new NamedSymbol(sym); } // #@@} // #@@range/privateSymbol{ private Symbol privateSymbol(String sym) { return new NamedSymbol(sym); } // #@@} // #@@range/callingSymbol{ private Symbol callingSymbol(Function func) { if (func.isPrivate()) { return privateSymbol(func.symbolString()); } else { Symbol sym = globalSymbol(func.symbolString()); return shouldUsePLT(func) ? PLTSymbol(sym) : sym; } } // #@@} // #@@range/shouldUsePLT{ private boolean shouldUsePLT(Entity ent) { return options.isPositionIndependent() && !optimizeGvarAccess(ent); } // #@@} // #@@range/optimizeGvarAccess{ private boolean optimizeGvarAccess(Entity ent) { return options.isPIERequired() && ent.isDefined(); } // #@@} // // generateAssemblyCode // // #@@range/generateAssemblyCode{ private AssemblyCode generateAssemblyCode(IR ir) { AssemblyCode file = newAssemblyCode(); file._file(ir.fileName()); if (ir.isGlobalVariableDefined()) { generateDataSection(file, ir.definedGlobalVariables()); } if (ir.isStringLiteralDefined()) { generateReadOnlyDataSection(file, ir.constantTable()); } if (ir.isFunctionDefined()) { generateTextSection(file, ir.definedFunctions()); } // #@@range/generateAssemblyCode_last{ if (ir.isCommonSymbolDefined()) { generateCommonSymbols(file, ir.definedCommonSymbols()); } if (options.isPositionIndependent()) { PICThunk(file, GOTBaseReg()); } return file; // #@@} } // #@@} // #@@range/newAssemblyCode{ private AssemblyCode newAssemblyCode() { return new AssemblyCode( naturalType, STACK_WORD_SIZE, new SymbolTable(LABEL_SYMBOL_BASE), options.isVerboseAsm()); } // #@@} /** Generates initialized entries */ // #@@range/generateDataSection{ private void generateDataSection(AssemblyCode file, List<DefinedVariable> gvars) { file._data(); for (DefinedVariable var : gvars) { Symbol sym = globalSymbol(var.symbolString()); if (!var.isPrivate()) { file._globl(sym); } file._align(var.alignment()); file._type(sym, "@object"); file._size(sym, var.allocSize()); file.label(sym); generateImmediate(file, var.type().allocSize(), var.ir()); } } // #@@} /** Generates immediate values for .data section */ // #@@range/generateImmediate{ private void generateImmediate(AssemblyCode file, long size, Expr node) { if (node instanceof Int) { Int expr = (Int)node; switch ((int)size) { case 1: file._byte(expr.value()); break; case 2: file._value(expr.value()); break; case 4: file._long(expr.value()); break; case 8: file._quad(expr.value()); break; default: throw new Error("entry size must be 1,2,4,8"); } } else if (node instanceof Str) { Str expr = (Str)node; switch ((int)size) { case 4: file._long(expr.symbol()); break; case 8: file._quad(expr.symbol()); break; default: throw new Error("pointer size must be 4,8"); } } else { throw new Error("unknown literal node type" + node.getClass()); } } // #@@} /** Generates .rodata entries (constant strings) */ // #@@range/generateReadOnlyDataSection{ private void generateReadOnlyDataSection(AssemblyCode file, ConstantTable constants) { file._section(".rodata"); for (ConstantEntry ent : constants) { file.label(ent.symbol()); file._string(ent.value()); } } // #@@} // #@@range/generateTextSection{ private void generateTextSection(AssemblyCode file, List<DefinedFunction> functions) { file._text(); for (DefinedFunction func : functions) { Symbol sym = globalSymbol(func.name()); if (! func.isPrivate()) { file._globl(sym); } file._type(sym, "@function"); file.label(sym); compileFunctionBody(file, func); file._size(sym, ".-" + sym.toSource()); } } // #@@} /** Generates BSS entries */ // #@@range/generateCommonSymbols{