Added special treatment of serializing NULL pointers.

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

diff --git a/include/llvm/Bitcode/Deserialize.h b/include/llvm/Bitcode/Deserialize.h
index f5adcffe25c183ec82c3af2b8f6fc50f5329d49f..164c70a9af68992e83214c758cd937e26a349170 100644 (file)
--- a/include/llvm/Bitcode/Deserialize.h
+++ b/include/llvm/Bitcode/Deserialize.h
@@ -86,6 +86,10 @@ public:
   template <typename T>
   inline T* ReadOwnedPtr() {    
     unsigned PtrId = ReadInt();
+
+    if (PtrId == 0)
+      return NULL;
+    
     T* x = SerializeTrait<T>::Materialize(*this);
     RegisterPtr(PtrId,x);
     return x;

diff --git a/include/llvm/Bitcode/Serialize.h b/include/llvm/Bitcode/Serialize.h
index dc1fe646f227df00e6bbab4f05223aa691b2604f..72aac9d0ba4e93f9accd78c4ae968191d613436f 100644 (file)
--- a/include/llvm/Bitcode/Serialize.h
+++ b/include/llvm/Bitcode/Serialize.h
@@ -47,7 +47,7 @@ public:
   template <typename T>
   void EmitOwnedPtr(T* ptr) {
     EmitPtr(ptr);
-    SerializeTrait<T>::Emit(*this,*ptr);    
+    if (ptr) SerializeTrait<T>::Emit(*this,*ptr);
   }
 
   void Flush() { if (inRecord()) EmitRecord(); }

diff --git a/lib/Bitcode/Reader/Deserialize.cpp b/lib/Bitcode/Reader/Deserialize.cpp
index febb2d7cafc93819d1637f34745055ef29186132..d1d6ef098f727647a9c40807da625cbfffecd1b3 100644 (file)
--- a/lib/Bitcode/Reader/Deserialize.cpp
+++ b/lib/Bitcode/Reader/Deserialize.cpp
@@ -103,6 +103,11 @@ void Deserializer::RegisterPtr(unsigned PtrId,void* Ptr) {
 void Deserializer::ReadPtr(void*& PtrRef) {
   unsigned PtrId = ReadInt();
   
+  if (PtrId == 0) {
+    PtrRef = NULL;
+    return;
+  }  
+  
   BPatchEntry& E = BPatchMap[PtrId];
   
   if (E.Ptr == NULL) {

diff --git a/lib/Bitcode/Writer/Serialize.cpp b/lib/Bitcode/Writer/Serialize.cpp
index fbe1d1007c9f054f04dd6b0c745d7e8f272de29e..a1d732047a6c14d438759dd28f4715ab595dc29d 100644 (file)
--- a/lib/Bitcode/Writer/Serialize.cpp
+++ b/lib/Bitcode/Writer/Serialize.cpp
@@ -58,10 +58,13 @@ void Serializer::EmitCStr(const char* s) {
 }
 
 unsigned Serializer::getPtrId(void* ptr) {
+  if (!ptr)
+    return 0;
+  
   MapTy::iterator I = PtrMap.find(ptr);
   
   if (I == PtrMap.end()) {
-    unsigned id = PtrMap.size();
+    unsigned id = PtrMap.size()+1;
     PtrMap[ptr] = id;
     return id;
   }