[llvm-commits] CVS: llvm/lib/Support/FileUtilities.cpp

[Chris Lattner]

Changes in directory llvm/lib/Support:

FileUtilities.cpp updated: 1.19 -> 1.20

---
Log message:

Add a pair of functions to hide system specific details of mapping a file in for reading.


---
Diffs of the changes:  (+34 -0)

Index: llvm/lib/Support/FileUtilities.cpp
diff -u llvm/lib/Support/FileUtilities.cpp:1.19 llvm/lib/Support/FileUtilities.cpp:1.20
--- llvm/lib/Support/FileUtilities.cpp:1.19	Wed Dec 31 00:15:37 2003
+++ llvm/lib/Support/FileUtilities.cpp	Thu May 27 19:23:48 2004
@@ -14,8 +14,10 @@
 
 #include "Support/FileUtilities.h"
 #include "Config/unistd.h"
+#include "Config/fcntl.h"
 #include "Config/sys/stat.h"
 #include "Config/sys/types.h"
+#include "Config/sys/mman.h"
 #include <fstream>
 #include <iostream>
 #include <cstdio>
@@ -215,8 +217,40 @@
   return StatBuf.st_mtime;  
 }
 
+/// ReadFileIntoAddressSpace - Attempt to map the specific file into the
+/// address space of the current process for reading.  If this succeeds,
+/// return the address of the buffer and the length of the file mapped.  On
+/// failure, return null.
+void *llvm::ReadFileIntoAddressSpace(const std::string &Filename, 
+                                     unsigned &Length) {
+#ifdef HAVE_MMAP_FILE
+  Length = getFileSize(Filename);
+  if ((int)Length == -1) return 0;
 
+  FDHandle FD(open(Filename.c_str(), O_RDONLY));
+  if (FD == -1) return 0;
 
+  // mmap in the file all at once...
+  void *Buffer = (void*)mmap(0, Length, PROT_READ, MAP_PRIVATE, FD, 0);
+
+  if (Buffer == (void*)MAP_FAILED)
+    return 0;
+  return Buffer;
+#else
+  // FIXME: implement with read/write
+  return 0;
+#endif
+}
+
+/// UnmapFileFromAddressSpace - Remove the specified file from the current
+/// address space.
+void llvm::UnmapFileFromAddressSpace(void *Buffer, unsigned Length) {
+#ifdef HAVE_MMAP_FILE
+  munmap((char*)Buffer, Length);
+#else
+  free(Buffer);
+#endif
+}
 
 //===----------------------------------------------------------------------===//
 // FDHandle class implementation