diff options
Diffstat (limited to 'discover/file.c')
-rw-r--r-- | discover/file.c | 43 |
1 files changed, 43 insertions, 0 deletions
diff --git a/discover/file.c b/discover/file.c index 269421c..1ebfe43 100644 --- a/discover/file.c +++ b/discover/file.c @@ -17,6 +17,8 @@ #include <fcntl.h> #include <unistd.h> +#include <stdlib.h> +#include <errno.h> #include <sys/stat.h> #include <sys/types.h> @@ -75,3 +77,44 @@ err_close: close(fd); return -1; } + +static int write_fd(int fd, char *buf, int len) +{ + int i, rc; + + for (i = 0; i < len; i += rc) { + rc = write(fd, buf + i, len - i); + if (rc < 0 && errno != -EINTR) + return rc; + } + + return 0; +} + +int replace_file(const char *filename, char *buf, int len) +{ + char *tempfile; + mode_t oldmask; + int rc, fd; + + tempfile = talloc_asprintf(NULL, "%s.XXXXXX", filename); + + oldmask = umask(0644); + fd = mkstemp(tempfile); + umask(oldmask); + if (fd < 0) { + free(tempfile); + return fd; + } + + rc = write_fd(fd, buf, len); + if (rc) { + unlink(tempfile); + } else { + rc = rename(tempfile, filename); + } + + free(tempfile); + close(fd); + return rc; +} |