Added support in serializer and deserializer to create arbitrary blocks.

Added detection of end-of-stream in deserializer.

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

diff --git a/include/llvm/Bitcode/Deserialize.h b/include/llvm/Bitcode/Deserialize.h
index 2454dca565dfb5e3dd4ae0207a1a72463126e85c..0398e434ca49ebcd0ba6bc8e8881a09b95892a42 100644 (file)
--- a/include/llvm/Bitcode/Deserialize.h
+++ b/include/llvm/Bitcode/Deserialize.h
@@ -153,6 +153,8 @@ public:
   void RegisterPtr(const void* Ptr) {
     RegisterPtr(ReadInt(),Ptr);
   }
   void RegisterPtr(const void* Ptr) {
     RegisterPtr(ReadInt(),Ptr);
   }

+  
+  bool AtEnd();

 
 private:
   void ReadRecord();  
 
 private:
   void ReadRecord();  

diff --git a/include/llvm/Bitcode/Serialize.h b/include/llvm/Bitcode/Serialize.h
index a465beca9d43c569b3815261e705278a465e5c7f..74637f5d87ff4eb732b0e103c6b9e178ddffba9b 100644 (file)
--- a/include/llvm/Bitcode/Serialize.h
+++ b/include/llvm/Bitcode/Serialize.h
@@ -25,7 +25,7 @@ namespace llvm {
 class Serializer {
   BitstreamWriter& Stream;
   SmallVector<uint64_t,10> Record;
 class Serializer {
   BitstreamWriter& Stream;
   SmallVector<uint64_t,10> Record;

-  bool inBlock;
+  unsigned BlockLevel;

   
   typedef DenseMap<const void*,unsigned> MapTy;
   MapTy PtrMap;
   
   typedef DenseMap<const void*,unsigned> MapTy;
   MapTy PtrMap;


@@ -56,6 +56,9 @@ public:
 
   void Flush() { if (inRecord()) EmitRecord(); }
   
 
   void Flush() { if (inRecord()) EmitRecord(); }
   

+  void EnterBlock(unsigned BlockID = 8, unsigned CodeLen = 3);
+  void ExitBlock();    
+  

 private:
   void EmitRecord();
   inline bool inRecord() { return Record.size() > 0; }
 private:
   void EmitRecord();
   inline bool inRecord() { return Record.size() > 0; }

diff --git a/lib/Bitcode/Reader/Deserialize.cpp b/lib/Bitcode/Reader/Deserialize.cpp
index 31d180ff32e297a6b41273acea4ec7694e763d9f..ef4e4f7e00004d5f4558a1667b7584a07fb15cd1 100644 (file)
--- a/lib/Bitcode/Reader/Deserialize.cpp
+++ b/lib/Bitcode/Reader/Deserialize.cpp
@@ -47,10 +47,15 @@ void Deserializer::ReadRecord() {
   // FIXME: Check if we haven't run off the edge of the stream.
   // FIXME: Handle abbreviations.
 
   // FIXME: Check if we haven't run off the edge of the stream.
   // FIXME: Handle abbreviations.
 

+  assert (Record.size() == 0);
+  

   unsigned Code;
 
   while (true) {
     
   unsigned Code;
 
   while (true) {
     

+    if (Stream.AtEndOfStream())
+      return;
+    

     Code = Stream.ReadCode();
   
     if (Code == bitc::ENTER_SUBBLOCK) {
     Code = Stream.ReadCode();
   
     if (Code == bitc::ENTER_SUBBLOCK) {


@@ -71,7 +76,16 @@ void Deserializer::ReadRecord() {
   
   assert (Record.size() == 0);  
   Stream.ReadRecord(Code,Record);  
   
   assert (Record.size() == 0);  
   Stream.ReadRecord(Code,Record);  

-  assert (Record.size() > 0);
+  assert (Record.size() > 0 || Stream.AtEndOfStream());
+}
+
+bool Deserializer::AtEnd() {
+  if (inRecord())
+    return false;
+  
+  ReadRecord();
+  
+  return Stream.AtEndOfStream();

 }
 
 uint64_t Deserializer::ReadInt() {
 }
 
 uint64_t Deserializer::ReadInt() {

diff --git a/lib/Bitcode/Writer/Serialize.cpp b/lib/Bitcode/Writer/Serialize.cpp
index 58baf104c7a0299abc2e950ea0e983aa2b9df783..f25a1542d375b35d8072bc0c47da3b1f697af309 100644 (file)
--- a/lib/Bitcode/Writer/Serialize.cpp
+++ b/lib/Bitcode/Writer/Serialize.cpp
@@ -17,16 +17,17 @@
 using namespace llvm;
 
 Serializer::Serializer(BitstreamWriter& stream, unsigned BlockID)
 using namespace llvm;
 
 Serializer::Serializer(BitstreamWriter& stream, unsigned BlockID)

-  : Stream(stream), inBlock(BlockID >= 8) {
+  : Stream(stream), BlockLevel(0) {

     
     

-  if (inBlock) Stream.EnterSubblock(8,3);
+  if (BlockID >= 8)
+    EnterBlock(8,3);

 }
 
 Serializer::~Serializer() {
   if (inRecord())
     EmitRecord();
 
 }
 
 Serializer::~Serializer() {
   if (inRecord())
     EmitRecord();
 

-  if (inBlock)
+  while (BlockLevel > 0)

     Stream.ExitBlock();
   
   Stream.FlushToWord();
     Stream.ExitBlock();
   
   Stream.FlushToWord();


@@ -38,7 +39,21 @@ void Serializer::EmitRecord() {
   Record.clear();
 }
 
   Record.clear();
 }
 

+void Serializer::EnterBlock(unsigned BlockID,unsigned CodeLen) {
+  Flush();
+  Stream.EnterSubblock(BlockID,CodeLen);
+  ++BlockLevel;
+}
+
+void Serializer::ExitBlock() {
+  assert (BlockLevel > 0);
+  --BlockLevel;
+  Flush();
+  Stream.ExitBlock();
+}
+

 void Serializer::EmitInt(unsigned X) {
 void Serializer::EmitInt(unsigned X) {

+  assert (BlockLevel > 0);

   Record.push_back(X);
 }
 
   Record.push_back(X);
 }
 


@@ -71,6 +86,7 @@ unsigned Serializer::getPtrId(const void* ptr) {
   else return I->second;
 }
 
   else return I->second;
 }
 

+

 #define INT_EMIT(TYPE)\
 void SerializeTrait<TYPE>::Emit(Serializer&S, TYPE X) { S.EmitInt(X); }
 
 #define INT_EMIT(TYPE)\
 void SerializeTrait<TYPE>::Emit(Serializer&S, TYPE X) { S.EmitInt(X); }