Add MemoryBufferRef(MemoryBuffer&) constructor.
authorRafael Espindola <rafael.espindola@gmail.com>
Tue, 17 Nov 2015 05:11:44 +0000 (05:11 +0000)
committerRafael Espindola <rafael.espindola@gmail.com>
Tue, 17 Nov 2015 05:11:44 +0000 (05:11 +0000)
patch by Jonathan Anderson!

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

include/llvm/Support/MemoryBuffer.h
unittests/Support/MemoryBufferTest.cpp

index 4aeca0c..d6ff69d 100644 (file)
@@ -152,6 +152,8 @@ class MemoryBufferRef {
 
 public:
   MemoryBufferRef() {}
 
 public:
   MemoryBufferRef() {}
+  MemoryBufferRef(MemoryBuffer& Buffer)
+      : Buffer(Buffer.getBuffer()), Identifier(Buffer.getBufferIdentifier()) {}
   MemoryBufferRef(StringRef Buffer, StringRef Identifier)
       : Buffer(Buffer), Identifier(Identifier) {}
 
   MemoryBufferRef(StringRef Buffer, StringRef Identifier)
       : Buffer(Buffer), Identifier(Identifier) {}
 
index ffb809a..ffbd74d 100644 (file)
@@ -170,6 +170,15 @@ TEST_F(MemoryBufferTest, getOpenFileReopened) {
 }
 
 
 }
 
 
+TEST_F(MemoryBufferTest, reference) {
+  OwningBuffer MB(MemoryBuffer::getMemBuffer(data));
+  MemoryBufferRef MBR(*MB);
+
+  EXPECT_EQ(MB->getBufferStart(), MBR.getBufferStart());
+  EXPECT_EQ(MB->getBufferIdentifier(), MBR.getBufferIdentifier());
+}
+
+
 TEST_F(MemoryBufferTest, slice) {
   // Create a file that is six pages long with different data on each page.
   int FD;
 TEST_F(MemoryBufferTest, slice) {
   // Create a file that is six pages long with different data on each page.
   int FD;