implement file family functions (7ef5c915) · Commits · hpc / gekkofs

ifs/include/preload/passthrough.hpp

+8 −0

Original line number	Diff line number	Diff line
		@@ -9,8 +9,16 @@
		extern void* libc;

		extern void* libc_open;

		extern void* libc_fopen;
		extern void* libc_fopen64;
		extern void* libc_fread;
		extern void* libc_fwrite;
		extern void* libc_fclose;
		extern void* libc_clearerr;
		extern void* libc_feof;
		extern void* libc_ferror;
		extern void* libc_fileno;

		extern void* libc_mkdir;
		extern void* libc_mkdirat;

ifs/src/preload/intcp_functions.cpp

+106 −1

Original line number	Diff line number	Diff line
		@@ -76,6 +76,8 @@ FILE* fopen(const char* path, const char* fmode) {
		flags = O_RDONLY;
		} else if(str_mode == "r+") {
		flags = O_RDWR;
		} else if(str_mode == "w") {
		flags = (O_WRONLY \| O_CREAT \| O_TRUNC);
		} else {
		CTX->log()->error("{}() stream open flags NOT SUPPORTED: '{}'", __func__, str_mode);
		errno = ENOTSUP;
		@@ -102,7 +104,110 @@ FILE* fopen64(const char* path, const char* fmode) {
		return fopen(path, fmode);
		}
		}
		return (reinterpret_cast<decltype(&fopen)>(libc_fopen64))(path, fmode);
		return (reinterpret_cast<decltype(&fopen64)>(libc_fopen64))(path, fmode);
		}

		size_t fread(void ptr, size_t size, size_t nmemb, FILE stream) {
		init_passthrough_if_needed();
		if(CTX->initialized() && (stream != nullptr)) {
		auto fd = file_to_fd(stream);
		if (CTX->file_map()->exist(fd)) {
		CTX->log()->trace("{}() called with fd {}", __func__, fd);
		auto adafs_fd = CTX->file_map()->get(fd);
		auto pos = adafs_fd->pos(); //retrieve the current offset
		auto ret = adafs_pread_ws(fd, ptr, size*nmemb, pos);
		if (ret > 0) {
		// Update offset in file descriptor in the file map
		adafs_fd->pos(pos + ret);
		return ret % size;
		}
		return ret;
		}
		}
		return (reinterpret_cast<decltype(&fread)>(libc_fread))(ptr, size, nmemb, stream);
		}

		size_t fwrite(const void ptr, size_t size, size_t nmemb, FILE stream) {
		init_passthrough_if_needed();
		if(CTX->initialized() && (stream != nullptr)) {
		auto fd = file_to_fd(stream);
		if (CTX->file_map()->exist(fd)) {
		CTX->log()->trace("{}() called with fd {}", __func__, fd);
		auto adafs_fd = CTX->file_map()->get(fd);
		auto pos = adafs_fd->pos(); //retrieve the current offset
		auto ret = adafs_pwrite_ws(fd, ptr, size*nmemb, pos);
		if (ret > 0) {
		// Update offset in file descriptor in the file map
		adafs_fd->pos(pos + ret);
		return ret % size;
		}
		return ret;
		}
		}
		return (reinterpret_cast<decltype(&fwrite)>(libc_fwrite))(ptr, size, nmemb, stream);
		}

		int fclose(FILE *stream) {
		init_passthrough_if_needed();
		if(CTX->initialized() && (stream != nullptr)) {
		auto fd = file_to_fd(stream);
		if(CTX->file_map()->exist(fd)) {
		// No call to the daemon is required
		CTX->file_map()->remove(fd);
		return 0;
		}
		}
		return (reinterpret_cast<decltype(&fclose)>(libc_fclose))(stream);
		}

		int fileno(FILE *stream) {
		init_passthrough_if_needed();
		if(CTX->initialized() && (stream != nullptr)) {
		auto fd = file_to_fd(stream);
		if(CTX->file_map()->exist(fd)) {
		return fd;
		}
		}
		return (reinterpret_cast<decltype(&fileno)>(libc_fileno))(stream);
		}

		void clearerr(FILE *stream) {
		init_passthrough_if_needed();
		if(CTX->initialized() && (stream != nullptr)) {
		auto fd = file_to_fd(stream);
		if(CTX->file_map()->exist(fd)) {
		CTX->log()->error("{}() NOT SUPPORTED", __func__);
		errno = ENOTSUP;
		return;
		}
		}
		return (reinterpret_cast<decltype(&clearerr)>(libc_clearerr))(stream);
		}

		int feof(FILE *stream) {
		init_passthrough_if_needed();
		if(CTX->initialized() && (stream != nullptr)) {
		auto fd = file_to_fd(stream);
		if(CTX->file_map()->exist(fd)) {
		CTX->log()->error("{}() NOT SUPPORTED", __func__);
		errno = ENOTSUP;
		return -1;
		}
		}
		return (reinterpret_cast<decltype(&feof)>(libc_feof))(stream);
		}

		int ferror(FILE *stream) {
		init_passthrough_if_needed();
		if(CTX->initialized() && (stream != nullptr)) {
		auto fd = file_to_fd(stream);
		if(CTX->file_map()->exist(fd)) {
		CTX->log()->error("{}() NOT SUPPORTED", __func__);
		errno = ENOTSUP;
		return -1;
		}
		}
		return (reinterpret_cast<decltype(&ferror)>(libc_ferror))(stream);
		}

		/**** FILE OPS ****/

ifs/src/preload/passthrough.cpp

+19 −2

Original line number	Diff line number	Diff line
		@@ -13,8 +13,16 @@ static pthread_once_t init_lib_thread = PTHREAD_ONCE_INIT;
		void* libc;

		void* libc_open;
		void* libc_fopen; // XXX Does not work with streaming pointers. If used will block forever
		void* libc_fopen64; // XXX Does not work with streaming pointers. If used will block forever

		void* libc_fopen;
		void* libc_fopen64;
		void* libc_fread;
		void* libc_fwrite;
		void* libc_fclose;
		void* libc_clearerr;
		void* libc_feof;
		void* libc_ferror;
		void* libc_fileno;

		void* libc_mkdir;
		void* libc_mkdirat;
		@@ -76,8 +84,17 @@ void init_passthrough_() {
		}

		libc_open = dlsym(libc, "open");

		libc_fopen = dlsym(libc, "fopen");
		libc_fopen64 = dlsym(libc, "fopen64");
		libc_fread = dlsym(libc, "fread");
		libc_fwrite = dlsym(libc, "fwrite");
		libc_fclose = dlsym(libc, "fclose");
		libc_clearerr = dlsym(libc, "clearerr");
		libc_feof = dlsym(libc, "feof");
		libc_ferror = dlsym(libc, "ferror");
		libc_fileno = dlsym(libc, "fileno");

		libc_mkdir = dlsym(libc, "mkdir");
		libc_mkdirat = dlsym(libc, "mkdirat");