X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;f=lib%2FVMCore%2FAsmWriter.cpp;h=cc8c5a12324fa78dc28f4b6182dad9ef83c3b160;hb=4460f4058fb33e468358f85c7f0925fc0e69e16c;hp=22a0ca001a106ac4051e4aa17049c73e3f812bbc;hpb=3c8f2c66e585f946569fa62cbf26509009d2df93;p=oota-llvm.git diff --git a/lib/VMCore/AsmWriter.cpp b/lib/VMCore/AsmWriter.cpp index 22a0ca001a1..cc8c5a12324 100644 --- a/lib/VMCore/AsmWriter.cpp +++ b/lib/VMCore/AsmWriter.cpp @@ -1,10 +1,10 @@ //===-- AsmWriter.cpp - Printing LLVM as an assembly file -----------------===// -// +// // The LLVM Compiler Infrastructure // // This file was developed by the LLVM research group and is distributed under // the University of Illinois Open Source License. See LICENSE.TXT for details. -// +// //===----------------------------------------------------------------------===// // // This library implements the functionality defined in llvm/Assembly/Writer.h @@ -18,21 +18,26 @@ #include "llvm/Assembly/Writer.h" #include "llvm/Assembly/PrintModulePass.h" #include "llvm/Assembly/AsmAnnotationWriter.h" +#include "llvm/CallingConv.h" #include "llvm/Constants.h" #include "llvm/DerivedTypes.h" +#include "llvm/InlineAsm.h" #include "llvm/Instruction.h" #include "llvm/Instructions.h" #include "llvm/Module.h" #include "llvm/SymbolTable.h" -#include "llvm/Assembly/Writer.h" #include "llvm/Support/CFG.h" #include "llvm/ADT/StringExtras.h" #include "llvm/ADT/STLExtras.h" +#include "llvm/Support/MathExtras.h" #include using namespace llvm; namespace llvm { +// Make virtual table appear in this compilation unit. +AssemblyAnnotationWriter::~AssemblyAnnotationWriter() {} + /// This class provides computation of slot numbers for LLVM Assembly writing. /// @brief LLVM Assembly Writing Slot Computation. class SlotMachine { @@ -46,7 +51,7 @@ public: typedef std::map TypeMap; /// @brief A plane with next slot number and ValueMap - struct ValuePlane { + struct ValuePlane { unsigned next_slot; ///< The next slot number to use ValueMap map; ///< The map of Value* -> unsigned ValuePlane() { next_slot = 0; } ///< Make sure we start at 0 @@ -90,15 +95,15 @@ public: /// @name Mutators /// @{ public: - /// If you'd like to deal with a function instead of just a module, use + /// If you'd like to deal with a function instead of just a module, use /// this method to get its data into the SlotMachine. - void incorporateFunction(const Function *F) { - TheFunction = F; + void incorporateFunction(const Function *F) { + TheFunction = F; FunctionProcessed = false; } - /// After calling incorporateFunction, use this method to remove the - /// most recently incorporated function from the SlotMachine. This + /// After calling incorporateFunction, use this method to remove the + /// most recently incorporated function from the SlotMachine. This /// will reset the state of the machine back to just the module contents. void purgeFunction(); @@ -109,7 +114,7 @@ private: /// This function does the actual initialization. inline void initialize(); - /// Values can be crammed into here at will. If they haven't + /// Values can be crammed into here at will. If they haven't /// been inserted already, they get inserted, otherwise they are ignored. /// Either way, the slot number for the Value* is returned. unsigned createSlot(const Value *V); @@ -117,7 +122,7 @@ private: /// Insert a value into the value table. Return the slot number /// that it now occupies. BadThings(TM) will happen if you insert a - /// Value that's already been inserted. + /// Value that's already been inserted. unsigned insertValue( const Value *V ); unsigned insertValue( const Type* Ty); @@ -162,12 +167,12 @@ X("printm", "Print module to stderr",PassInfo::Analysis|PassInfo::Optimization); static RegisterPass Y("print","Print function to stderr",PassInfo::Analysis|PassInfo::Optimization); -static void WriteAsOperandInternal(std::ostream &Out, const Value *V, +static void WriteAsOperandInternal(std::ostream &Out, const Value *V, bool PrintName, std::map &TypeTable, SlotMachine *Machine); -static void WriteAsOperandInternal(std::ostream &Out, const Type *T, +static void WriteAsOperandInternal(std::ostream &Out, const Type *T, bool PrintName, std::map &TypeTable, SlotMachine *Machine); @@ -203,7 +208,8 @@ static SlotMachine *createSlotMachine(const Value *V) { // getLLVMName - Turn the specified string into an 'LLVM name', which is either // prefixed with % (if the string only contains simple characters) or is // surrounded with ""'s (if it has special chars in it). -static std::string getLLVMName(const std::string &Name) { +static std::string getLLVMName(const std::string &Name, + bool prefixName = true) { assert(!Name.empty() && "Cannot get empty name!"); // First character cannot start with a number... @@ -218,9 +224,12 @@ static std::string getLLVMName(const std::string &Name) { C != '-' && C != '.' && C != '_') return "\"" + Name + "\""; } - + // If we get here, then the identifier is legal to use as a "VarID". - return "%"+Name; + if (prefixName) + return "%"+Name; + else + return Name; } @@ -246,7 +255,7 @@ static void fillTypeNameTable(const Module *M, -static void calcTypeName(const Type *Ty, +static void calcTypeName(const Type *Ty, std::vector &TypeStack, std::map &TypeNames, std::string & Result){ @@ -271,7 +280,7 @@ static void calcTypeName(const Type *Ty, unsigned Slot = 0, CurSize = TypeStack.size(); while (Slot < CurSize && TypeStack[Slot] != Ty) ++Slot; // Scan for type - // This is another base case for the recursion. In this case, we know + // This is another base case for the recursion. In this case, we know // that we have looped back to a type that we have previously visited. // Generate the appropriate upreference to handle this. if (Slot < CurSize) { @@ -280,7 +289,7 @@ static void calcTypeName(const Type *Ty, } TypeStack.push_back(Ty); // Recursive case: Add us to the stack.. - + switch (Ty->getTypeID()) { case Type::FunctionTyID: { const FunctionType *FTy = cast(Ty); @@ -312,7 +321,7 @@ static void calcTypeName(const Type *Ty, break; } case Type::PointerTyID: - calcTypeName(cast(Ty)->getElementType(), + calcTypeName(cast(Ty)->getElementType(), TypeStack, TypeNames, Result); Result += "*"; break; @@ -375,25 +384,42 @@ static std::ostream &printTypeInt(std::ostream &Out, const Type *Ty, /// std::ostream &llvm::WriteTypeSymbolic(std::ostream &Out, const Type *Ty, const Module *M) { - Out << ' '; + Out << ' '; // If they want us to print out a type, attempt to make it symbolic if there // is a symbol table in the module... if (M) { std::map TypeNames; fillTypeNameTable(M, TypeNames); - + return printTypeInt(Out, Ty, TypeNames); } else { return Out << Ty->getDescription(); } } -/// @brief Internal constant writer. -static void WriteConstantInt(std::ostream &Out, const Constant *CV, +// PrintEscapedString - Print each character of the specified string, escaping +// it if it is not printable or if it is an escape char. +static void PrintEscapedString(const std::string &Str, std::ostream &Out) { + for (unsigned i = 0, e = Str.size(); i != e; ++i) { + unsigned char C = Str[i]; + if (isprint(C) && C != '"' && C != '\\') { + Out << C; + } else { + Out << '\\' + << (char) ((C/16 < 10) ? ( C/16 +'0') : ( C/16 -10+'A')) + << (char)(((C&15) < 10) ? ((C&15)+'0') : ((C&15)-10+'A')); + } + } +} + +/// @brief Internal constant writer. +static void WriteConstantInt(std::ostream &Out, const Constant *CV, bool PrintName, std::map &TypeTable, SlotMachine *Machine) { + const int IndentSize = 4; + static std::string Indent = "\n"; if (const ConstantBool *CB = dyn_cast(CV)) { Out << (CB == ConstantBool::True ? "true" : "false"); } else if (const ConstantSInt *CI = dyn_cast(CV)) { @@ -417,51 +443,26 @@ static void WriteConstantInt(std::ostream &Out, const Constant *CV, (StrVal[1] >= '0' && StrVal[1] <= '9'))) // Reparse stringized version! if (atof(StrVal.c_str()) == CFP->getValue()) { - Out << StrVal; return; + Out << StrVal; + return; } - + // Otherwise we could not reparse it to exactly the same value, so we must // output the string in hexadecimal format! - // - // Behave nicely in the face of C TBAA rules... see: - // http://www.nullstone.com/htmls/category/aliastyp.htm - // - double Val = CFP->getValue(); - char *Ptr = (char*)&Val; - assert(sizeof(double) == sizeof(uint64_t) && sizeof(double) == 8 && + assert(sizeof(double) == sizeof(uint64_t) && "assuming that double is 64 bits!"); - Out << "0x" << utohexstr(*(uint64_t*)Ptr); + Out << "0x" << utohexstr(DoubleToBits(CFP->getValue())); } else if (isa(CV)) { Out << "zeroinitializer"; } else if (const ConstantArray *CA = dyn_cast(CV)) { // As a special case, print the array as a string if it is an array of // ubytes or an array of sbytes with positive values. - // + // const Type *ETy = CA->getType()->getElementType(); - bool isString = (ETy == Type::SByteTy || ETy == Type::UByteTy); - - if (ETy == Type::SByteTy) - for (unsigned i = 0; i < CA->getNumOperands(); ++i) - if (cast(CA->getOperand(i))->getValue() < 0) { - isString = false; - break; - } - - if (isString) { + if (CA->isString()) { Out << "c\""; - for (unsigned i = 0; i < CA->getNumOperands(); ++i) { - unsigned char C = - (unsigned char)cast(CA->getOperand(i))->getRawValue(); - - if (isprint(C) && C != '"' && C != '\\') { - Out << C; - } else { - Out << '\\' - << (char) ((C/16 < 10) ? ( C/16 +'0') : ( C/16 -10+'A')) - << (char)(((C&15) < 10) ? ((C&15)+'0') : ((C&15)-10+'A')); - } - } + PrintEscapedString(CA->getAsString(), Out); Out << "\""; } else { // Cannot output in string format... @@ -482,26 +483,34 @@ static void WriteConstantInt(std::ostream &Out, const Constant *CV, } } else if (const ConstantStruct *CS = dyn_cast(CV)) { Out << '{'; - if (CS->getNumOperands()) { - Out << ' '; + unsigned N = CS->getNumOperands(); + if (N) { + if (N > 2) { + Indent += std::string(IndentSize, ' '); + Out << Indent; + } else { + Out << ' '; + } printTypeInt(Out, CS->getOperand(0)->getType(), TypeTable); WriteAsOperandInternal(Out, CS->getOperand(0), PrintName, TypeTable, Machine); - for (unsigned i = 1; i < CS->getNumOperands(); i++) { + for (unsigned i = 1; i < N; i++) { Out << ", "; + if (N > 2) Out << Indent; printTypeInt(Out, CS->getOperand(i)->getType(), TypeTable); WriteAsOperandInternal(Out, CS->getOperand(i), PrintName, TypeTable, Machine); } + if (N > 2) Indent.resize(Indent.size() - IndentSize); } - + Out << " }"; } else if (const ConstantPacked *CP = dyn_cast(CV)) { const Type *ETy = CP->getType()->getElementType(); - assert(CP->getNumOperands() > 0 && + assert(CP->getNumOperands() > 0 && "Number of operands for a PackedConst must be > 0"); Out << '<'; Out << ' '; @@ -523,14 +532,14 @@ static void WriteConstantInt(std::ostream &Out, const Constant *CV, } else if (const ConstantExpr *CE = dyn_cast(CV)) { Out << CE->getOpcodeName() << " ("; - + for (User::const_op_iterator OI=CE->op_begin(); OI != CE->op_end(); ++OI) { printTypeInt(Out, (*OI)->getType(), TypeTable); WriteAsOperandInternal(Out, *OI, PrintName, TypeTable, Machine); if (OI+1 != CE->op_end()) Out << ", "; } - + if (CE->getOpcode() == Instruction::Cast) { Out << " to "; printTypeInt(Out, CE->getType(), TypeTable); @@ -547,7 +556,7 @@ static void WriteConstantInt(std::ostream &Out, const Constant *CV, /// ostream. This can be useful when you just want to print int %reg126, not /// the whole instruction that generated it. /// -static void WriteAsOperandInternal(std::ostream &Out, const Value *V, +static void WriteAsOperandInternal(std::ostream &Out, const Value *V, bool PrintName, std::map &TypeTable, SlotMachine *Machine) { @@ -556,15 +565,24 @@ static void WriteAsOperandInternal(std::ostream &Out, const Value *V, Out << getLLVMName(V->getName()); else { const Constant *CV = dyn_cast(V); - if (CV && !isa(CV)) + if (CV && !isa(CV)) { WriteConstantInt(Out, CV, PrintName, TypeTable, Machine); - else { + } else if (const InlineAsm *IA = dyn_cast(V)) { + Out << "asm "; + if (IA->hasSideEffects()) + Out << "sideeffect "; + Out << '"'; + PrintEscapedString(IA->getAsmString(), Out); + Out << "\", \""; + PrintEscapedString(IA->getConstraintString(), Out); + Out << '"'; + } else { int Slot; if (Machine) { Slot = Machine->getSlot(V); } else { Machine = createSlotMachine(V); - if (Machine == 0) + if (Machine == 0) Slot = Machine->getSlot(V); else Slot = -1; @@ -583,7 +601,7 @@ static void WriteAsOperandInternal(std::ostream &Out, const Value *V, /// the whole instruction that generated it. /// std::ostream &llvm::WriteAsOperand(std::ostream &Out, const Value *V, - bool PrintType, bool PrintName, + bool PrintType, bool PrintName, const Module *Context) { std::map TypeNames; if (Context == 0) Context = getModuleFromVal(V); @@ -593,16 +611,16 @@ std::ostream &llvm::WriteAsOperand(std::ostream &Out, const Value *V, if (PrintType) printTypeInt(Out, V->getType(), TypeNames); - + WriteAsOperandInternal(Out, V, PrintName, TypeNames, 0); return Out; } -/// WriteAsOperandInternal - Write the name of the specified value out to -/// the specified ostream. This can be useful when you just want to print +/// WriteAsOperandInternal - Write the name of the specified value out to +/// the specified ostream. This can be useful when you just want to print /// int %reg126, not the whole instruction that generated it. /// -static void WriteAsOperandInternal(std::ostream &Out, const Type *T, +static void WriteAsOperandInternal(std::ostream &Out, const Type *T, bool PrintName, std::map &TypeTable, SlotMachine *Machine) { @@ -624,7 +642,7 @@ static void WriteAsOperandInternal(std::ostream &Out, const Type *T, /// the whole instruction that generated it. /// std::ostream &llvm::WriteAsOperand(std::ostream &Out, const Type *Ty, - bool PrintType, bool PrintName, + bool PrintType, bool PrintName, const Module *Context) { std::map TypeNames; assert(Context != 0 && "Can't write types as operand without module context"); @@ -633,7 +651,7 @@ std::ostream &llvm::WriteAsOperand(std::ostream &Out, const Type *Ty, // if (PrintType) // printTypeInt(Out, V->getType(), TypeNames); - + printTypeInt(Out, Ty, TypeNames); WriteAsOperandInternal(Out, Ty, PrintName, TypeNames, 0); @@ -671,7 +689,7 @@ public: const Module* getModule() { return TheModule; } -private : +private: void printModule(const Module *M); void printSymbolTable(const SymbolTable &ST); void printConstant(const Constant *CPV); @@ -745,15 +763,24 @@ std::ostream &AssemblyWriter::printTypeAtLeastOneLevel(const Type *Ty) { } -void AssemblyWriter::writeOperand(const Value *Operand, bool PrintType, +void AssemblyWriter::writeOperand(const Value *Operand, bool PrintType, bool PrintName) { - assert(Operand != 0 && "Illegal Operand"); - if (PrintType) { Out << ' '; printType(Operand->getType()); } - WriteAsOperandInternal(Out, Operand, PrintName, TypeNames, &Machine); + if (Operand != 0) { + if (PrintType) { Out << ' '; printType(Operand->getType()); } + WriteAsOperandInternal(Out, Operand, PrintName, TypeNames, &Machine); + } else { + Out << ""; + } } void AssemblyWriter::printModule(const Module *M) { + if (!M->getModuleIdentifier().empty() && + // Don't print the ID if it will start a new line (which would + // require a comment char before it). + M->getModuleIdentifier().find('\n') == std::string::npos) + Out << "; ModuleID = '" << M->getModuleIdentifier() << "'\n"; + switch (M->getEndianness()) { case Module::LittleEndian: Out << "target endian = little\n"; break; case Module::BigEndian: Out << "target endian = big\n"; break; @@ -766,6 +793,26 @@ void AssemblyWriter::printModule(const Module *M) { } if (!M->getTargetTriple().empty()) Out << "target triple = \"" << M->getTargetTriple() << "\"\n"; + + if (!M->getModuleInlineAsm().empty()) { + // Split the string into lines, to make it easier to read the .ll file. + std::string Asm = M->getModuleInlineAsm(); + size_t CurPos = 0; + size_t NewLine = Asm.find_first_of('\n', CurPos); + while (NewLine != std::string::npos) { + // We found a newline, print the portion of the asm string from the + // last newline up to this newline. + Out << "module asm \""; + PrintEscapedString(std::string(Asm.begin()+CurPos, Asm.begin()+NewLine), + Out); + Out << "\"\n"; + CurPos = NewLine+1; + NewLine = Asm.find_first_of('\n', CurPos); + } + Out << "module asm \""; + PrintEscapedString(std::string(Asm.begin()+CurPos, Asm.end()), Out); + Out << "\"\n"; + } // Loop over the dependent libraries and emit them. Module::lib_iterator LI = M->lib_begin(); @@ -783,12 +830,12 @@ void AssemblyWriter::printModule(const Module *M) { // Loop over the symbol table, emitting all named constants. printSymbolTable(M->getSymbolTable()); - - for (Module::const_giterator I = M->gbegin(), E = M->gend(); I != E; ++I) + + for (Module::const_global_iterator I = M->global_begin(), E = M->global_end(); I != E; ++I) printGlobal(I); Out << "\nimplementation ; Functions:\n"; - + // Output all of the functions. for (Module::const_iterator I = M->begin(), E = M->end(); I != E; ++I) printFunction(I); @@ -797,7 +844,7 @@ void AssemblyWriter::printModule(const Module *M) { void AssemblyWriter::printGlobal(const GlobalVariable *GV) { if (GV->hasName()) Out << getLLVMName(GV->getName()) << " = "; - if (!GV->hasInitializer()) + if (!GV->hasInitializer()) Out << "external "; else switch (GV->getLinkage()) { @@ -819,7 +866,12 @@ void AssemblyWriter::printGlobal(const GlobalVariable *GV) { assert(C && "GlobalVar initializer isn't constant?"); writeOperand(GV->getInitializer(), false, isa(C)); } - + + if (GV->hasSection()) + Out << ", section \"" << GV->getSection() << '"'; + if (GV->getAlignment()) + Out << ", align " << GV->getAlignment(); + printInfoComment(*GV); Out << "\n"; } @@ -839,7 +891,7 @@ void AssemblyWriter::printSymbolTable(const SymbolTable &ST) { // printTypeAtLeastOneLevel(TI->second) << "\n"; } - + // Print the constants, in type plane order. for (SymbolTable::plane_const_iterator PI = ST.plane_begin(); PI != ST.plane_end(); ++PI ) { @@ -879,6 +931,9 @@ void AssemblyWriter::printFunction(const Function *F) { // Print out the return type and name... Out << "\n"; + // Ensure that no local symbols conflict with global symbols. + const_cast(F)->renameLocalSymbols(); + if (AnnotationWriter) AnnotationWriter->emitFunctionAnnot(F, Out); if (F->isExternal()) @@ -895,6 +950,14 @@ void AssemblyWriter::printFunction(const Function *F) { abort(); } + // Print the calling convention. + switch (F->getCallingConv()) { + case CallingConv::C: break; // default + case CallingConv::Fast: Out << "fastcc "; break; + case CallingConv::Cold: Out << "coldcc "; break; + default: Out << "cc" << F->getCallingConv() << " "; break; + } + printType(F->getReturnType()) << ' '; if (!F->getName().empty()) Out << getLLVMName(F->getName()); @@ -906,7 +969,7 @@ void AssemblyWriter::printFunction(const Function *F) { // Loop over the arguments, printing them... const FunctionType *FT = F->getFunctionType(); - for(Function::const_aiterator I = F->abegin(), E = F->aend(); I != E; ++I) + for(Function::const_arg_iterator I = F->arg_begin(), E = F->arg_end(); I != E; ++I) printArgument(I); // Finish printing arguments... @@ -916,11 +979,16 @@ void AssemblyWriter::printFunction(const Function *F) { } Out << ')'; + if (F->hasSection()) + Out << " section \"" << F->getSection() << '"'; + if (F->getAlignment()) + Out << " align " << F->getAlignment(); + if (F->isExternal()) { Out << "\n"; } else { Out << " {"; - + // Output all of its basic blocks... for the function for (Function::const_iterator I = F->begin(), E = F->end(); I != E; ++I) printBasicBlock(I); @@ -936,11 +1004,11 @@ void AssemblyWriter::printFunction(const Function *F) { /// void AssemblyWriter::printArgument(const Argument *Arg) { // Insert commas as we go... the first arg doesn't get a comma - if (Arg != &Arg->getParent()->afront()) Out << ", "; + if (Arg != Arg->getParent()->arg_begin()) Out << ", "; // Output type... printType(Arg->getType()); - + // Output name, if available... if (Arg->hasName()) Out << ' ' << getLLVMName(Arg->getName()); @@ -950,7 +1018,7 @@ void AssemblyWriter::printArgument(const Argument *Arg) { /// void AssemblyWriter::printBasicBlock(const BasicBlock *BB) { if (BB->hasName()) { // Print out the label if it exists... - Out << "\n" << BB->getName() << ':'; + Out << "\n" << getLLVMName(BB->getName(), false) << ':'; } else if (!BB->use_empty()) { // Don't print block # of no uses... Out << "\n;