Add a bunch of attributes, patch by Nuno Lopes.

git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@47837 91177308-0d34-0410-b5e6-96231b3b80d8

1 files changed, 30 insertions, 17 deletions

diff --git a/CodeGen/CodeGenModule.cpp b/CodeGen/CodeGenModule.cpp
index 5e63a80c94..1a69796214 100644
--- a/CodeGen/CodeGenModule.cpp
+++ b/CodeGen/CodeGenModule.cpp
@@ -242,25 +242,38 @@ void CodeGenModule::EmitGlobalVar(const FileVarDecl *D) {
   assert(GV->getType()->getElementType() == Init->getType() &&
          "Initializer codegen type mismatch!");
   GV->setInitializer(Init);
+
+  if (const VisibilityAttr *attr = D->getAttr<VisibilityAttr>())
+    GV->setVisibility(attr->getVisibility());
+  // FIXME: else handle -fvisibility
   
   // Set the llvm linkage type as appropriate.
-  // FIXME: This isn't right.  This should handle common linkage and other
-  // stuff.
-  switch (D->getStorageClass()) {
-  case VarDecl::Auto:
-  case VarDecl::Register:
-    assert(0 && "Can't have auto or register globals");
-  case VarDecl::None:
-    if (!D->getInit())
-      GV->setLinkage(llvm::GlobalVariable::WeakLinkage);
-    break;
-  case VarDecl::Extern:
-  case VarDecl::PrivateExtern:
-    // todo: common
-    break;
-  case VarDecl::Static:
-    GV->setLinkage(llvm::GlobalVariable::InternalLinkage);
-    break;
+  if (D->getAttr<DLLImportAttr>())
+    GV->setLinkage(llvm::Function::DLLImportLinkage);
+  else if (D->getAttr<DLLExportAttr>())
+    GV->setLinkage(llvm::Function::DLLExportLinkage);
+  else if (D->getAttr<WeakAttr>()) {
+    GV->setLinkage(llvm::GlobalVariable::WeakLinkage);
+
+  } else {
+    // FIXME: This isn't right.  This should handle common linkage and other
+    // stuff.
+    switch (D->getStorageClass()) {
+    case VarDecl::Auto:
+    case VarDecl::Register:
+      assert(0 && "Can't have auto or register globals");
+    case VarDecl::None:
+      if (!D->getInit())
+        GV->setLinkage(llvm::GlobalVariable::WeakLinkage);
+      break;
+    case VarDecl::Extern:
+    case VarDecl::PrivateExtern:
+      // todo: common
+      break;
+    case VarDecl::Static:
+      GV->setLinkage(llvm::GlobalVariable::InternalLinkage);
+      break;
+    }
   }
 }