Sign-extend array index expressions to work correctly on non-32 bit machines.

llvm-svn: 3496

1 files changed, 9 insertions, 4 deletions

diff --git a/llvm/lib/CWriter/Writer.cpp b/llvm/lib/CWriter/Writer.cpp
index e33622f71f3..f4fb7a24585 100644
--- a/llvm/lib/CWriter/Writer.cpp
+++ b/llvm/lib/CWriter/Writer.cpp
@@ -210,7 +210,8 @@ ostream &CWriter::printType(const Type *Ty, const string &NameSoFar,
 
   case Type::PointerTyID: {
     const PointerType *PTy = cast<PointerType>(Ty);
-    return printType(PTy->getElementType(), "(*" + NameSoFar + ")");
+    std::string ptrName = NameSoFar.length()? "(*"+NameSoFar+")" : string("*");
+    return printType(PTy->getElementType(), ptrName);
   }
 
   case Type::ArrayTyID: {
@@ -844,16 +845,20 @@ void CWriter::printIndexingExpression(Value *Ptr, User::op_iterator I,
       Out << (HasImplicitAddress ? "." : "->");
       Out << "field" << cast<ConstantUInt>(*(I+1))->getValue();
       I += 2;
-    } else {  // Performing array indexing. Just skip the 0
+    } else {  // First array index of 0: Just skip it
       ++I;
     }
   }
 
   for (; I != E; ++I)
     if ((*I)->getType() == Type::UIntTy) {
-      Out << "[";
+      Out << "[((int) (";                 // sign-extend from 32 (to 64) bits
       writeOperand(*I);
-      Out << "]";
+      Out << " * sizeof(";
+      printType(cast<PointerType>(Ptr->getType())->getElementType());
+      Out << "))) / sizeof(";
+      printType(cast<PointerType>(Ptr->getType())->getElementType());
+      Out << ")]";
     } else {
       Out << ".field" << cast<ConstantUInt>(*I)->getValue();
     }