Finish s6-rc-repo-init

Signed-off-by: Laurent Bercot <ska@appnovation.com>

3 files changed, 156 insertions, 0 deletions

diff --git a/src/libs6rc/deps-lib/s6rc b/src/libs6rc/deps-lib/s6rc
index 3b0bf75..71a1616 100644
--- a/src/libs6rc/deps-lib/s6rc
+++ b/src/libs6rc/deps-lib/s6rc
@@ -13,6 +13,8 @@ s6rc_livedir_prefix.o
 s6rc_livedir_prefixsize.o
 s6rc_lock.o
 s6rc_read_uint.o
+s6rc_repo_cleanup.o
+s6rc_repo_sync.o
 s6rc_sanitize_dir.o
 s6rc_servicedir_internal.o
 s6rc_servicedir_block.o
diff --git a/src/libs6rc/s6rc_repo_cleanup.c b/src/libs6rc/s6rc_repo_cleanup.c
new file mode 100644
index 0000000..9b42c06
--- /dev/null
+++ b/src/libs6rc/s6rc_repo_cleanup.c
@@ -0,0 +1,76 @@
+/* ISC license. */
+
+#include <skalibs/bsdsnowflake.h>
+
+#include <string.h>
+#include <unistd.h>
+#include <errno.h>
+
+#include <skalibs/stat.h>
+#include <skalibs/direntry.h>
+#include <skalibs/posixplz.h>
+#include <skalibs/djbunix.h>
+
+#include <s6-rc/repo.h>
+
+static inline int sub_cleanup (char const *subdir)
+{
+  size_t dirlen = strlen(subdir) ;
+  DIR *dir = opendir(subdir) ;
+  if (!dir) return 0 ;
+  for (;;)
+  {
+    size_t len ;
+    direntry *d ;
+    errno = 0 ;
+    d = readdir(dir) ;
+    if (!d) break ;
+    if (d->d_name[0] == '.') continue ;
+    len = strlen(d->d_name) ;
+    {
+      char fn[dirlen + len + 2] ;
+      memcpy(fn, dir, dirlen) ;
+      fn[dirlen] = '/' ;
+      memcpy(fn + dirlen + 1, d->d_name, len + 1) ;
+      if (access(fn, F_OK) == -1)
+      {
+        if (errno == ENOENT) unlink_void(fn) ;
+        else break ;
+      }
+      else
+      {
+        struct stat st ;
+        if (lstat(fn, &st) == -1) break ;
+        if (!S_ISLNK(st.st_mode)) rm_rf(fn) ;
+      }
+    }
+  }
+  dir_close(dir) ;
+  return !errno ;
+}
+
+int s6rc_repo_cleanup (char const *repo)
+{
+  size_t repolen = strlen(repo) ;
+  DIR *dir = opendir(repo) ;
+  if (!dir) return 0 ;
+  for (;;)
+  {
+    size_t len ;
+    direntry *d ;
+    errno = 0 ;
+    d = readdir(dir) ;
+    if (!d) break ;
+    if (d->d_name[0] == '.') continue ;
+    len = strlen(d->d_name) ;
+    {
+      char fn[repolen + len + 2] ;
+      memcpy(fn, repo, repolen) ;
+      fn[repolen] = '/' ;
+      memcpy(fn + repolen + 1, d->d_name, len + 1) ;
+      if (!sub_cleanup(fn)) break ;
+    }
+  }
+  dir_close(dir) ;
+  return !errno ;
+}
diff --git a/src/libs6rc/s6rc_repo_sync.c b/src/libs6rc/s6rc_repo_sync.c
new file mode 100644
index 0000000..7ece55f
--- /dev/null
+++ b/src/libs6rc/s6rc_repo_sync.c
@@ -0,0 +1,78 @@
+/* ISC license. */
+
+#include <skalibs/bsdsnowflake.h>  /* stat */
+#include <skalibs/nonposix.h>  /* mkdtemp */
+
+#include <string.h>
+#include <unistd.h>
+#include <stdlib.h>
+#include <errno.h>
+
+#include <skalibs/stat.h>
+#include <skalibs/direntry.h>
+#include <skalibs/posixplz.h>
+#include <skalibs/djbunix.h>
+#include <skalibs/unix-transactional.h>
+
+#include <s6-rc/repo.h>
+
+static inline void cleanup (char const *fn)
+{
+  int e = errno ;
+  rm_rf(fn) ;
+  errno = e ;
+}
+
+int s6rc_repo_sync (char const *repo, char const *const *sources, size_t sourceslen)
+{
+  size_t repolen = strlen(repo) ;
+  char newdir[repolen + 24] ;
+  memcpy(newdir, repo, repolen) ;
+  memcpy(newdir + repolen, "/.everything:cur:XXXXXX", 24) ;
+  if (!mkdtemp(newdir)) return 0 ;
+
+  for (size_t i = 0 ; i < sourceslen ; i++)
+  {
+    size_t srclen = strlen(sources[i]) ;
+    DIR *dir = opendir(sources[i]) ;
+    if (!dir) goto err ;
+    for (;;)
+    {
+      size_t len ;
+      direntry *d ;
+      errno = 0 ;
+      d = readdir(dir) ;
+      if (!d) break ;
+      if (d->d_name[0] == '.') continue ;
+      len = strlen(d->d_name) ;
+      {
+        char dst[repolen + 25 + len] ;
+        char src[srclen + len + 2] ;
+        memcpy(dst, newdir, repolen + 23) ;
+        dst[repolen + 23] = '/' ;
+        memcpy(dst + repolen + 24, d->d_name, len + 1) ;
+        memcpy(src, sources[i], srclen) ;
+        src[srclen] = '/' ;
+        memcpy(src + srclen + 1, d->d_name, len + 1) ;
+        if (symlink(src, dst) == -1) break ;
+      }
+    }
+    dir_close(dir) ;
+    if (errno) goto err ;
+  }
+
+  if (chmod(newdir, 02755) == -1) goto err ;
+
+  {
+    char curdir[repolen + 13] ;
+    memcpy(curdir, repo, repolen) ;
+    memcpy(curdir + repolen, "/.everything", 13) ;
+    if (!atomic_symlink(newdir + repolen + 1, curdir, "tmp")) goto err ;
+  }
+
+  return s6rc_repo_cleanup(repo) ;
+
+ err:
+  cleanup(newdir) ;
+  return 0 ;
+}