bugfix: setlock -r

1 files changed, 19 insertions, 3 deletions

diff --git a/src/daemontools-extras/s6-setlock.c b/src/daemontools-extras/s6-setlock.c
index cd938b7..6ba2ec8 100644
--- a/src/daemontools-extras/s6-setlock.c
+++ b/src/daemontools-extras/s6-setlock.c
@@ -42,9 +42,25 @@ int main (int argc, char const *const *argv)
 
   if (nb < 2)
   {
-    int r ;
-    int fd = open_create(argv[0]) ;
-    if (fd < 0) strerr_diefu2sys(111, "open_create ", argv[0]) ;
+    int r, fd ;
+    if (ex)
+    {
+      fd = open_create(argv[0]) ;
+      if (fd < 0) strerr_diefu3sys(111, "open ", argv[0], " for writing") ;
+    }
+    else
+    {
+      fd = open_read(argv[0]) ;
+      if (fd < 0)
+      {
+        if (errno != ENOENT) strerr_diefu3sys(111, "open ", argv[0], " for reading") ;
+        fd = open_create(argv[0]) ;
+        if (fd < 0) strerr_diefu2sys(111, "create ", argv[0]) ;
+        close(fd) ;
+        fd = open_read(argv[0]) ;
+        if (fd < 0) strerr_diefu3sys(111, "open ", argv[0], " for reading") ;
+      }
+    }
     r = fd_lock(fd, ex, nb) ;
     if (!r) errno = EBUSY ;
     if (r < 1) strerr_diefu2sys(1, "lock ", argv[0]) ;