Implement basic logging of write, rename, rmdir, unlink

1 files changed, 57 insertions, 18 deletions

diff --git a/change_log.cc b/change_log.cc
index d4c3dc4..03c4343 100644
--- a/change_log.cc
+++ b/change_log.cc
@@ -4,11 +4,16 @@
 
 #include <unistd.h>
 #include <dlfcn.h>
+#include <fcntl.h>
 #include <sys/stat.h>
 
 #include <cstring>
 #include <iostream>
+#include <utility>
 #include <functional>
+#include <unordered_set>
+
+static std::unordered_set<std::string> tracked_files;
 
 template <class Result, typename... Arguments>
 std::function<Result(Arguments...)> get_real_function(
@@ -22,6 +27,10 @@ std::function<Result(Arguments...)> get_real_function(
 	return std::function<Result(Arguments...)>(real_function);
 }
 
+void exit(int status) {
+	get_real_function<void, int>("exit")(status);
+}
+
 std::string get_file_name(int fd) {
 	char proc_link[20];
 	char file_name[256];
@@ -45,32 +54,62 @@ bool is_regular_file(int fd) {
 	return S_ISREG(fd_stat.st_mode);
 }
 
-ssize_t read(int fd, void *buffer, size_t count) {
-	static auto real_read = get_real_function<ssize_t, int, void*, size_t>("read");
-
-	const ssize_t result = real_read(fd, buffer, count);
-
-	if ( is_regular_file(fd) ) {
-		std::cerr << "read size "
-		          << count
-		          << " of "
-		          << get_file_name(fd)
-		          << std::endl;
-	}
+bool is_tracked_file(const std::string& file_name) {
+	return tracked_files.find(file_name) != tracked_files.end();
+}
 
-	return result;
+bool track_file(const std::string& file_name) {
+	return tracked_files.emplace(file_name).second;
 }
 
 ssize_t write(int fd, const void* buffer, size_t count) {
 	static auto real_write = get_real_function<ssize_t, int, const void*, size_t>("write");
 
 	if ( is_regular_file(fd) ) {
-		std::cerr << "write size "
-		          << count
-		          << " to "
-		          << get_file_name(fd)
-		          << std::endl;
+		const std::string file_name{ get_file_name(fd) };
+
+		if ( !is_tracked_file(file_name) ) {
+			track_file(file_name);
+
+			std::cerr << "wrote to '" << file_name << "'" << std::endl;
+		}
 	}
 
 	return real_write(fd, buffer, count);
 }
+
+int rename(const char* old_path, const char* new_path) {
+	static auto real_rename = get_real_function<int, const char*, const char*>("rename");
+
+	std::cerr << "renamed '" << old_path << "' to '" << new_path << "'" << std::endl;
+
+	return real_rename(old_path, new_path);
+}
+
+int rmdir(const char* path) {
+	static auto real_rmdir = get_real_function<int, const char*>("rmdir");
+
+	std::cerr << "removed directory '" << path << "'" << std::endl;
+
+	return real_rmdir(path);
+}
+
+int unlink(const char* path) {
+	static auto real_unlink = get_real_function<int, const char*>("unlink");
+
+	std::cerr << "removed '" << path << "'" << std::endl;
+
+	return real_unlink(path);
+}
+
+int unlinkat(int dirfd, const char* path, int flags) {
+	static auto real_unlinkat = get_real_function<int, int, const char*, int>("unlinkat");
+
+	if ( dirfd == AT_FDCWD ) {
+		std::cerr << "removed '" << path << "'" << std::endl;
+	} else {
+		std::cerr << "removed '" << get_file_name(dirfd) << path << "'" << std::endl;
+	}
+
+	return real_unlinkat(dirfd, path, flags);
+}