aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--include/Support/FileUtilities.h12
-rw-r--r--include/llvm/Support/FileUtilities.h12
-rw-r--r--lib/Support/FileUtilities.cpp34
3 files changed, 56 insertions, 2 deletions
diff --git a/include/Support/FileUtilities.h b/include/Support/FileUtilities.h
index 0543d76f5a..d5db63f6d7 100644
--- a/include/Support/FileUtilities.h
+++ b/include/Support/FileUtilities.h
@@ -42,7 +42,7 @@ bool IsSharedObject(const std::string &FN);
/// FileOpenable - Returns true IFF Filename names an existing regular file
/// which we can successfully open.
///
-bool FileOpenable (const std::string &Filename);
+bool FileOpenable(const std::string &Filename);
/// DiffFiles - Compare the two files specified, returning true if they are
/// different or if there is a file error. If you specify a string to fill in
@@ -93,6 +93,16 @@ long long getFileSize(const std::string &Filename);
/// not exist or there is an error getting the time-stamp, zero is returned.
unsigned long long getFileTimestamp(const std::string &Filename);
+/// 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 *ReadFileIntoAddressSpace(const std::string &Filename, unsigned &Length);
+
+/// UnmapFileFromAddressSpace - Remove the specified file from the current
+/// address space.
+void UnmapFileFromAddressSpace(void *Buffer, unsigned Length);
+
/// FDHandle - Simple handle class to make sure a file descriptor gets closed
/// when the object is destroyed. This handle acts similarly to an
diff --git a/include/llvm/Support/FileUtilities.h b/include/llvm/Support/FileUtilities.h
index 0543d76f5a..d5db63f6d7 100644
--- a/include/llvm/Support/FileUtilities.h
+++ b/include/llvm/Support/FileUtilities.h
@@ -42,7 +42,7 @@ bool IsSharedObject(const std::string &FN);
/// FileOpenable - Returns true IFF Filename names an existing regular file
/// which we can successfully open.
///
-bool FileOpenable (const std::string &Filename);
+bool FileOpenable(const std::string &Filename);
/// DiffFiles - Compare the two files specified, returning true if they are
/// different or if there is a file error. If you specify a string to fill in
@@ -93,6 +93,16 @@ long long getFileSize(const std::string &Filename);
/// not exist or there is an error getting the time-stamp, zero is returned.
unsigned long long getFileTimestamp(const std::string &Filename);
+/// 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 *ReadFileIntoAddressSpace(const std::string &Filename, unsigned &Length);
+
+/// UnmapFileFromAddressSpace - Remove the specified file from the current
+/// address space.
+void UnmapFileFromAddressSpace(void *Buffer, unsigned Length);
+
/// FDHandle - Simple handle class to make sure a file descriptor gets closed
/// when the object is destroyed. This handle acts similarly to an
diff --git a/lib/Support/FileUtilities.cpp b/lib/Support/FileUtilities.cpp
index 1b8df43629..74df8b253c 100644
--- a/lib/Support/FileUtilities.cpp
+++ b/lib/Support/FileUtilities.cpp
@@ -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 @@ unsigned long long llvm::getFileTimestamp(const std::string &Filename) {
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