[llvm] r197600 - Add a unit test for loading an object file via a file descriptor. Patch by Pete Cooper.

[Owen Anderson]

Author: resistor
Date: Wed Dec 18 13:20:29 2013
New Revision: 197600

URL: http://llvm.org/viewvc/llvm-project?rev=197600&view=rev
Log:
Add a unit test for loading an object file via a file descriptor.  Patch by Pete Cooper.

Added:
    llvm/trunk/unittests/ArchiveFileDescriptor/
    llvm/trunk/unittests/ArchiveFileDescriptor/ArchiveFileDescriptor.cpp
    llvm/trunk/unittests/ArchiveFileDescriptor/CMakeLists.txt
    llvm/trunk/unittests/ArchiveFileDescriptor/Makefile
Modified:
    llvm/trunk/unittests/CMakeLists.txt
    llvm/trunk/unittests/Makefile

Added: llvm/trunk/unittests/ArchiveFileDescriptor/ArchiveFileDescriptor.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/unittests/ArchiveFileDescriptor/ArchiveFileDescriptor.cpp?rev=197600&view=auto
==============================================================================
--- llvm/trunk/unittests/ArchiveFileDescriptor/ArchiveFileDescriptor.cpp (added)
+++ llvm/trunk/unittests/ArchiveFileDescriptor/ArchiveFileDescriptor.cpp Wed Dec 18 13:20:29 2013
@@ -0,0 +1,53 @@
+//===- llvm/unittest/ArchiveFileDescriptor/ArchiveFileDescriptor.cpp ------===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+
+#include "llvm/Object/Archive.h"
+#include "llvm/Object/ObjectFile.h"
+#include "llvm/Support/MemoryBuffer.h"
+#include "gtest/gtest.h"
+
+using namespace llvm;
+using namespace llvm::object;
+
+static void failIfError(error_code EC) {
+  if (!EC)
+    return;
+
+  errs() << "ERROR: " << EC.message() << "\n";
+  exit(1);
+}
+
+TEST(ArchiveFileDescriptor, Test1) {
+  int FD;
+
+  error_code EC = sys::fs::openFileForRead("ArchiveFileDescriptor", FD);
+  failIfError(EC);
+
+  OwningPtr<MemoryBuffer> MemoryBuffer;
+  EC = MemoryBuffer::getOpenFile(FD, "Dummy Filename",
+                                 MemoryBuffer,
+                                 /* FileSize */ -1,
+                                 /* RequiresNullTerminator */ false);
+  failIfError(EC);
+
+  // Attempt to open the binary.
+  OwningPtr<Binary> Binary;
+  EC = createBinary(MemoryBuffer.take(), Binary);
+  failIfError(EC);
+
+  if (Archive *Arc = dyn_cast<Archive>(Binary.get())) {
+    (void)Arc;
+    errs() << "ERROR: Loaded archive, was expecting object file\n";
+  } else if (ObjectFile *Obj = dyn_cast<ObjectFile>(Binary.get())) {
+    (void)Obj;
+  } else {
+    outs() << "ERROR: Unknown file type\n";
+    exit(1);
+  }
+}

Added: llvm/trunk/unittests/ArchiveFileDescriptor/CMakeLists.txt
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/unittests/ArchiveFileDescriptor/CMakeLists.txt?rev=197600&view=auto
==============================================================================
--- llvm/trunk/unittests/ArchiveFileDescriptor/CMakeLists.txt (added)
+++ llvm/trunk/unittests/ArchiveFileDescriptor/CMakeLists.txt Wed Dec 18 13:20:29 2013
@@ -0,0 +1,8 @@
+set(LLVM_LINK_COMPONENTS
+  Object
+  Support
+  )
+
+add_llvm_unittest(ArchiveFileDescriptor
+  ArchiveFileDescriptor.cpp
+  )

Added: llvm/trunk/unittests/ArchiveFileDescriptor/Makefile
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/unittests/ArchiveFileDescriptor/Makefile?rev=197600&view=auto
==============================================================================
--- llvm/trunk/unittests/ArchiveFileDescriptor/Makefile (added)
+++ llvm/trunk/unittests/ArchiveFileDescriptor/Makefile Wed Dec 18 13:20:29 2013
@@ -0,0 +1,15 @@
+##===- unittests/ArchiveFileDescriptor/Makefile ------------*- Makefile -*-===##
+#
+#                     The LLVM Compiler Infrastructure
+#
+# This file is distributed under the University of Illinois Open Source
+# License. See LICENSE.TXT for details.
+#
+##===----------------------------------------------------------------------===##
+
+LEVEL = ../..
+TESTNAME = ArchiveFileDescriptor
+LINK_COMPONENTS := object
+
+include $(LEVEL)/Makefile.config
+include $(LLVM_SRC_ROOT)/unittests/Makefile.unittest

Modified: llvm/trunk/unittests/CMakeLists.txt
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/unittests/CMakeLists.txt?rev=197600&r1=197599&r2=197600&view=diff
==============================================================================
--- llvm/trunk/unittests/CMakeLists.txt (original)
+++ llvm/trunk/unittests/CMakeLists.txt Wed Dec 18 13:20:29 2013
@@ -7,6 +7,7 @@ endfunction()
 
 add_subdirectory(ADT)
 add_subdirectory(Analysis)
+add_subdirectory(ArchiveFileDescriptor)
 add_subdirectory(Bitcode)
 add_subdirectory(CodeGen)
 add_subdirectory(DebugInfo)

Modified: llvm/trunk/unittests/Makefile
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/unittests/Makefile?rev=197600&r1=197599&r2=197600&view=diff
==============================================================================
--- llvm/trunk/unittests/Makefile (original)
+++ llvm/trunk/unittests/Makefile Wed Dec 18 13:20:29 2013
@@ -9,8 +9,8 @@
 
 LEVEL = ..
 
-PARALLEL_DIRS = ADT Analysis Bitcode CodeGen DebugInfo ExecutionEngine IR \
-		MC Object Option Support Transforms
+PARALLEL_DIRS = ADT Analysis ArchiveFileDescriptor Bitcode CodeGen DebugInfo \
+		ExecutionEngine IR MC Object Option Support Transforms
 
 include $(LEVEL)/Makefile.common