diff options
author | Adrian Kummerlaender | 2016-02-20 22:30:11 +0100 |
---|---|---|
committer | Adrian Kummerlaender | 2016-02-20 22:30:11 +0100 |
commit | b3ef0fc8daa41e433f1919a4933a1cc047f59341 (patch) | |
tree | b551c34e381d6df29334848d6aff242c14b69817 /src | |
parent | ee51dc19c859e718280bebae1c089be058737b1c (diff) | |
download | change-b3ef0fc8daa41e433f1919a4933a1cc047f59341.tar change-b3ef0fc8daa41e433f1919a4933a1cc047f59341.tar.gz change-b3ef0fc8daa41e433f1919a4933a1cc047f59341.tar.bz2 change-b3ef0fc8daa41e433f1919a4933a1cc047f59341.tar.lz change-b3ef0fc8daa41e433f1919a4933a1cc047f59341.tar.xz change-b3ef0fc8daa41e433f1919a4933a1cc047f59341.tar.zst change-b3ef0fc8daa41e433f1919a4933a1cc047f59341.zip |
Interpose `open` library function
`open` is not as side effect free as I had imagined - i.e. if the flag `O_TRUNC` is passed it truncates the file contents alongside opening the file descriptor. In practice this is done by _emacs_ prior to writing the new file content and as such needs to be intercepted so we can start tracking the file before it is changed.
Interposing `open` required some changes to make the library work without including `fcntl.h`. This header not only defines some of the flags we require to check if a library call actually is able to change files but also defines the `open` library function.
While implementing this change I noticed that the function interpositions implemented in C++ actually need to be declared as `external "C"` so their names do not get wrangled during compilation. I suspect that this was previously implicitly done for e.g. `mmap` and `write` by the included C standard library headers. However this did not work for `open` which is why all function interpositions are now explicitly declared external.
End result: _emacs_ file changes are now tracked correctly.
Diffstat (limited to 'src')
-rw-r--r-- | src/actual.h | 3 | ||||
-rw-r--r-- | src/bootstrap.cc | 4 | ||||
-rw-r--r-- | src/main.cc | 41 | ||||
-rw-r--r-- | src/tracking/change_tracker.cc | 2 | ||||
-rw-r--r-- | src/utility/logger.h | 3 |
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 { |