support/lib/Support/NameMangling.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

//===- NameMangling.cpp - Name Mangling for LLVM ----------------------------=//
//
// This file implements a consistent scheme for name mangling symbols.
//
//===----------------------------------------------------------------------===//

#include "llvm/Support/NameMangling.h"
#include "llvm/DerivedTypes.h"
#include "llvm/GlobalValue.h"
using std::string;

// MangleTypeName - Implement a consistent name-mangling scheme for
//                  a given type.
// 
string MangleTypeName(const Type *Ty) {
  string mangledName;
  if (Ty->isPrimitiveType()) {
    const string &longName = Ty->getDescription();
    return string(longName.c_str(), (longName.length() < 2) ? 1 : 2);
  } else if (PointerType *PTy = dyn_cast<PointerType>(Ty)) {
    mangledName = string("P_" + MangleTypeName(PTy->getElementType()));
  } else if (StructType *STy = dyn_cast<StructType>(Ty)) {
    mangledName = string("S_");
    for (unsigned i=0; i < STy->getNumContainedTypes(); ++i)
      mangledName += MangleTypeName(STy->getContainedType(i));
  } else if (ArrayType *ATy = dyn_cast<ArrayType>(Ty)) {
    mangledName = string("A_" +MangleTypeName(ATy->getElementType()));
  } else if (MethodType *MTy = dyn_cast<MethodType>(Ty)) {
    mangledName = string("M_") + MangleTypeName(MTy->getReturnType());
    for (unsigned i = 1; i < MTy->getNumContainedTypes(); ++i)
      mangledName += string(MangleTypeName(MTy->getContainedType(i)));
  }
  
  return mangledName;
}

// mangleName - implement a consistent name-mangling scheme for all
// externally visible (i.e., global) objects.
// privateName should be unique within the module.
// 
string MangleName(const string &privateName, const Value *V) {
  // Lets drop the P_ before every global name since all globals are ptrs
  return privateName + "_" +
    MangleTypeName(isa<GlobalValue>(V)
                   ? cast<GlobalValue>(V)->getType()->getElementType()
                   : V->getType());
}