(wrong string) ünnich

From: <git_AT_suckless.org>
Date: Wed, 17 May 2017 20:26:19 +0200 (CEST)

commit a20173a42df64515c0a5d1c5fba0c056a633a441
Author: Bert Münnich <ber.t_AT_posteo.de>
AuthorDate: Wed May 17 20:16:16 2017 +0200
Commit: Bert Münnich <ber.t_AT_posteo.de>
CommitDate: Wed May 17 20:20:39 2017 +0200

    Detect all file overwrites in autoreload_inotify
    
    mv(1) inside the same filesystem was not detected.
    
    Supporting this case made it necessary to always watch the directory. Turns out
    the logic and state keeping between arl_setup() and arl_handle() is easier,
    when using different watch descriptors for the file and the directory and not
    using a oneshot descriptor for the file.
    
    Requiring an absolute canonical path for arl_setup() simplifies dir and base
    name splitting. No need for dirname(3) and basename(3) anymore.

diff --git a/autoreload.h b/autoreload.h
index 6a3053c..bb30eb6 100644
--- a/autoreload.h
+++ b/autoreload.h
_AT_@ -23,13 +23,14 @@
 
 typedef struct {
         int fd;
- int wd;
- bool watching_dir;
+ int wd_dir;
+ int wd_file;
+ char *filename;
 } arl_t;
 
-void arl_cleanup(arl_t*);
-bool arl_handle(arl_t*, const char*);
 void arl_init(arl_t*);
-void arl_setup(arl_t*, const char*);
+void arl_cleanup(arl_t*);
+void arl_setup(arl_t*, const char* /* result of realpath(3) */);
+bool arl_handle(arl_t*);
 
 #endif /* AUTORELOAD_H */
diff --git a/autoreload_inotify.c b/autoreload_inotify.c
index 473b8cb..fe05483 100644
--- a/autoreload_inotify.c
+++ b/autoreload_inotify.c
_AT_@ -21,39 +21,60 @@
 #include <string.h>
 #include <unistd.h>
 #include <sys/inotify.h>
-#include <libgen.h>
 
 #include "util.h"
 #include "autoreload.h"
 
+void arl_init(arl_t *arl)
+{
+ arl->fd = inotify_init1(IN_CLOEXEC | IN_NONBLOCK);
+ arl->wd_dir = arl->wd_file = -1;
+ if (arl->fd == -1)
+ error(0, 0, "Could not initialize inotify, no automatic image reloading");
+}
+
 CLEANUP void arl_cleanup(arl_t *arl)
 {
         if (arl->fd != -1)
                 close(arl->fd);
+ free(arl->filename);
 }
 
-static void arl_setup_dir(arl_t *arl, const char *filepath)
+static void rm_watch(int fd, int *wd)
 {
- char *dntmp, *dn;
+ if (*wd != -1) {
+ inotify_rm_watch(fd, *wd);
+ *wd = -1;
+ }
+}
+
+static void add_watch(int fd, int *wd, const char *path, uint32_t mask)
+{
+ *wd = inotify_add_watch(fd, path, mask);
+ if (*wd == -1)
+ error(0, errno, "inotify: %s", path);
+}
+
+void arl_setup(arl_t *arl, const char *filepath)
+{
+ char *base = strrchr(filepath, '/');
 
         if (arl->fd == -1)
                 return;
 
- /* get dirname */
- dntmp = (char*) strdup(filepath);
- dn = (char*) dirname(dntmp);
-
- /* this is not one-shot as other stuff may be created too
- * note: we won't handle deletion of the directory itself,
- * this is a design decision
- */
- arl->wd = inotify_add_watch(arl->fd, dn, IN_CREATE);
- if (arl->wd == -1)
- error(0, 0, "%s: Error watching directory", dn);
- else
- arl->watching_dir = true;
-
- free(dntmp);
+ rm_watch(arl->fd, &arl->wd_dir);
+ rm_watch(arl->fd, &arl->wd_file);
+
+ add_watch(arl->fd, &arl->wd_file, filepath, IN_CLOSE_WRITE | IN_DELETE_SELF);
+
+ free(arl->filename);
+ arl->filename = estrdup(filepath);
+
+ if (base != NULL) {
+ arl->filename[++base - filepath] = '\0';
+ add_watch(arl->fd, &arl->wd_dir, arl->filename, IN_CREATE | IN_MOVED_TO);
+ strcpy(arl->filename, base);
+ }
 }
 
 union {
_AT_@ -61,7 +82,7 @@ union {
         struct inotify_event e;
 } buf;
 
-bool arl_handle(arl_t *arl, const char *filepath)
+bool arl_handle(arl_t *arl)
 {
         bool reload = false;
         char *ptr;
_AT_@ -77,59 +98,16 @@ bool arl_handle(arl_t *arl, const char *filepath)
                 }
                 for (ptr = buf.d; ptr < buf.d + len; ptr += sizeof(*event) + event->len) {
                         event = (const struct inotify_event*) ptr;
-
- /* events from watching the file itself */
                         if (event->mask & IN_CLOSE_WRITE) {
                                 reload = true;
- }
- if (event->mask & IN_DELETE_SELF)
- arl_setup_dir(arl, filepath);
-
- /* events from watching the file's directory */
- if (event->mask & IN_CREATE) {
- char *fntmp = strdup(filepath);
- char *fn = basename(fntmp);
-
- if (STREQ(event->name, fn)) {
- /* this is the file we're looking for */
-
- /* cleanup, this has not been one-shot */
- if (arl->watching_dir) {
- inotify_rm_watch(arl->fd, arl->wd);
- arl->watching_dir = false;
- }
+ } else if (event->mask & IN_DELETE_SELF) {
+ rm_watch(arl->fd, &arl->wd_file);
+ } else if (event->mask & (IN_CREATE | IN_MOVED_TO)) {
+ if (STREQ(event->name, arl->filename))
                                         reload = true;
- }
- free(fntmp);
                         }
                 }
         }
         return reload;
 }
 
-void arl_init(arl_t *arl)
-{
- /* this needs to be done only once */
- arl->fd = inotify_init1(IN_CLOEXEC | IN_NONBLOCK);
- arl->watching_dir = false;
- if (arl->fd == -1)
- error(0, 0, "Could not initialize inotify, no automatic image reloading");
-}
-
-void arl_setup(arl_t *arl, const char *filepath)
-{
- if (arl->fd == -1)
- return;
-
- /* may have switched from a deleted to another image */
- if (arl->watching_dir) {
- inotify_rm_watch(arl->fd, arl->wd);
- arl->watching_dir = false;
- }
-
- arl->wd = inotify_add_watch(arl->fd, filepath,
- IN_ONESHOT | IN_CLOSE_WRITE | IN_DELETE_SELF);
- if (arl->wd == -1)
- error(0, 0, "%s: Error watching file", filepath);
-}
-
diff --git a/main.c b/main.c
index 70f7521..aa11616 100644
--- a/main.c
+++ b/main.c
_AT_@ -723,7 +723,7 @@ void run(void)
                                 if (info.fd != -1 && FD_ISSET(info.fd, &fds))
                                         read_info();
                                 if (arl.fd != -1 && FD_ISSET(arl.fd, &fds)) {
- if (arl_handle(&arl, files[fileidx].path)) {
+ if (arl_handle(&arl)) {
                                                 /* when too fast, imlib2 can't load the image */
                                                 nanosleep(&ten_ms, NULL);
                                                 load_image(fileidx);
Received on Wed May 17 2017 - 20:26:19 CEST

This archive was generated by hypermail 2.3.0 : Wed May 17 2017 - 20:37:01 CEST