aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/actual.h3
-rw-r--r--src/bootstrap.cc4
-rw-r--r--src/main.cc41
-rw-r--r--src/tracking/change_tracker.cc2
-rw-r--r--src/utility/logger.h3
5 files changed, 47 insertions, 6 deletions
diff --git a/src/actual.h b/src/actual.h
index a860590..ae951b9 100644
--- a/src/actual.h
+++ b/src/actual.h
@@ -6,9 +6,6 @@
#endif
#include <dlfcn.h>
-#include <sys/mman.h>
-#include <sys/uio.h>
-
#include <memory>
#include <cstring>
diff --git a/src/bootstrap.cc b/src/bootstrap.cc
index 309ff20..1d509d8 100644
--- a/src/bootstrap.cc
+++ b/src/bootstrap.cc
@@ -2,6 +2,8 @@
#include "init/alloc.h"
+extern "C" {
+
void free(void* ptr) {
static actual::ptr<void, void*> actual_free{};
@@ -41,3 +43,5 @@ void* calloc(size_t block, size_t size) {
return actual_calloc(block, size);
}
}
+
+}
diff --git a/src/main.cc b/src/main.cc
index 8b6ca23..7e849fd 100644
--- a/src/main.cc
+++ b/src/main.cc
@@ -1,3 +1,7 @@
+#include <stdlib.h>
+#include <sys/mman.h>
+#include <sys/uio.h>
+
#include "actual.h"
#include "utility/io.h"
@@ -61,6 +65,14 @@ inline void track_write(const int fd) {
}
}
+inline void track_write(const std::string& path) {
+ if ( enabled && utility::is_regular_file(path.c_str()) ) {
+ if ( !matcher->isMatching(path) ) {
+ tracker->track(path);
+ }
+ }
+}
+
inline void track_rename(
const std::string& old_path, const std::string& new_path) {
if ( enabled ) {
@@ -82,6 +94,28 @@ inline void track_remove(const std::string& path) {
}
}
+extern "C" {
+
+int open(const char* path, int flags, mode_t mode) {
+ static actual::ptr<int, const char*, int, mode_t> actual_open{};
+
+ if ( !actual_open ) {
+ actual_open = actual::get_ptr<decltype(actual_open)>("open");
+ }
+
+ // `open` may reset the file contents when used with the `O_TRUNC` flag.
+ // e.g. this is how _emacs_ clears the file content prior to writing the
+ // new content.
+ // Normally `O_TRUNC` is defined in `fcntl.h` which we can not include
+ // as it defines the very c-function we are currently _overriding_.
+ //
+ if ( flags & 01000 ) {
+ track_write(path);
+ }
+
+ return actual_open(path, flags, mode);
+}
+
ssize_t write(int fd, const void* buffer, size_t count) {
static actual::ptr<ssize_t, int, const void*, size_t> actual_write{};
@@ -163,7 +197,10 @@ int unlinkat(int dirfd, const char* path, int flags) {
actual_unlinkat = actual::get_ptr<decltype(actual_unlinkat)>("unlinkat");
}
- if ( dirfd == AT_FDCWD ) {
+ // Normally `AT_FDCWD` is defined in `fcntl.h` which we can not include
+ // as it defines `open` which this library also aims to override.
+ //
+ if ( dirfd == -100 ) {
track_remove(path);
} else {
track_remove(utility::get_file_path(dirfd) + path);
@@ -171,3 +208,5 @@ int unlinkat(int dirfd, const char* path, int flags) {
return actual_unlinkat(dirfd, path, flags);
}
+
+}
diff --git a/src/tracking/change_tracker.cc b/src/tracking/change_tracker.cc
index bc8465e..c966c04 100644
--- a/src/tracking/change_tracker.cc
+++ b/src/tracking/change_tracker.cc
@@ -1,7 +1,7 @@
#include "change_tracker.h"
+#include <boost/process.hpp>
#include <boost/filesystem.hpp>
-#include <boost/filesystem/fstream.hpp>
namespace {
diff --git a/src/utility/logger.h b/src/utility/logger.h
index 0dd88a6..229032c 100644
--- a/src/utility/logger.h
+++ b/src/utility/logger.h
@@ -5,7 +5,8 @@
#include <ext/stdio_filebuf.h>
-#include <boost/process.hpp>
+#include <boost/filesystem.hpp>
+#include <boost/process/pistream.hpp>
namespace utility {