summaryrefslogtreecommitdiff
path: root/src/lib/rib.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/lib/rib.c')
-rw-r--r--src/lib/rib.c389
1 files changed, 389 insertions, 0 deletions
diff --git a/src/lib/rib.c b/src/lib/rib.c
new file mode 100644
index 00000000..b6c8e140
--- /dev/null
+++ b/src/lib/rib.c
@@ -0,0 +1,389 @@
+/*
+ * Ouroboros - Copyright (C) 2016 - 2017
+ *
+ * RIB export using FUSE
+ *
+ * Dimitri Staessens <[email protected]>
+ * Sander Vrijders <[email protected]>
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public License
+ * version 2.1 as published by the Free Software Foundation.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., http://www.fsf.org/about/contact/.
+ */
+
+#define _POSIX_C_SOURCE 200112L
+
+#include "config.h"
+
+#include <ouroboros/errno.h>
+#include <ouroboros/list.h>
+#include <ouroboros/rib.h>
+#include <ouroboros/utils.h>
+
+#include <pthread.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+
+#ifdef HAVE_FUSE
+#define _FILE_OFFSET_BITS 64
+#define FUSE_USE_VERSION 26
+#include <fuse.h>
+#endif
+
+#ifdef HAVE_FUSE
+
+#ifndef CLOCK_REALTIME_COARSE
+#define CLOCK_REALTIME_COARSE CLOCK_REALTIME
+#endif
+
+#define RT "/"
+
+struct reg_comp {
+ struct list_head next;
+
+ char path[RIB_PATH_LEN + 1];
+ struct rib_ops * ops;
+};
+
+struct {
+ struct list_head reg_comps;
+
+ char mnt[RIB_PATH_LEN + 1];
+
+ struct fuse * fuse;
+ struct fuse_chan * ch;
+
+ pthread_rwlock_t lock;
+
+ pthread_t fuse_thr;
+} rib;
+
+static int rib_open(const char * path,
+ struct fuse_file_info * info)
+{
+ (void) path;
+
+ info->nonseekable = 1;
+
+ return 0;
+}
+
+static int rib_opendir(const char * path,
+ struct fuse_file_info * info)
+{
+ (void) path;
+ (void) info;
+
+ return 0;
+}
+
+static int rib_read(const char * path,
+ char * buf,
+ size_t size,
+ off_t offset,
+ struct fuse_file_info * info)
+{
+ struct list_head * p;
+ char comp[RIB_PATH_LEN + 1];
+ char * c;
+
+ strcpy(comp, path + 1);
+
+ c = strstr(comp, "/");
+
+ if (c != NULL)
+ *c = '\0';
+
+ (void) info;
+ (void) offset;
+
+ pthread_rwlock_wrlock(&rib.lock);
+
+ list_for_each(p, &rib.reg_comps) {
+ struct reg_comp * r = list_entry(p, struct reg_comp, next);
+ if (strcmp(comp, r->path) == 0) {
+ int ret = r->ops->read(c + 1, buf, size);
+ pthread_rwlock_unlock(&rib.lock);
+ return ret;
+ }
+ }
+
+ pthread_rwlock_unlock(&rib.lock);
+
+ return -1;
+}
+
+static int rib_readdir(const char * path,
+ void * buf,
+ fuse_fill_dir_t filler,
+ off_t offset,
+ struct fuse_file_info * info)
+{
+ struct list_head * p;
+
+ (void) offset;
+ (void) info;
+
+ filler(buf, ".", NULL, 0);
+ filler(buf, "..", NULL, 0);
+
+ pthread_rwlock_rdlock(&rib.lock);
+
+ if (strcmp(path, RT) == 0) {
+ list_for_each(p, &rib.reg_comps) {
+ struct reg_comp * c;
+ c = list_entry(p, struct reg_comp, next);
+ filler(buf, c->path, NULL, 0);
+ }
+ } else {
+ list_for_each(p, &rib.reg_comps) {
+ char ** dir_entries;
+ ssize_t len;
+ ssize_t i;
+ struct reg_comp * c;
+ c = list_entry(p, struct reg_comp, next);
+ if (strcmp(path + 1, c->path) == 0)
+ if (c->ops->readdir == NULL)
+ break;
+
+ len = c->ops->readdir(&dir_entries);
+ if (len < 0)
+ break;
+ for (i = 0; i < len; ++i)
+ filler(buf, dir_entries[i], NULL, 0);
+ freepp(char, dir_entries, len);
+ }
+ }
+
+ pthread_rwlock_unlock(&rib.lock);
+
+ return 0;
+}
+
+static int rib_getattr(const char * path,
+ struct stat * st)
+{
+ struct list_head * p;
+ struct timespec now;
+
+ clock_gettime(CLOCK_REALTIME_COARSE, &now);
+
+ memset(st, 0, sizeof(*st));
+
+ if (strcmp(path, RT) == 0) {
+ st->st_mode = __S_IFDIR | 0755;
+ st->st_nlink = 2;
+ st->st_uid = getuid();
+ st->st_gid = getgid();
+ st->st_mtime = now.tv_sec;
+ return 0;
+ }
+
+ pthread_rwlock_rdlock(&rib.lock);
+
+ list_for_each(p, &rib.reg_comps) {
+ struct reg_comp * rc = list_entry(p, struct reg_comp, next);
+ if (strcmp(path + 1, rc->path) == 0) {
+ st->st_mode = __S_IFDIR | 0755;
+ st->st_nlink = 2;
+ break;
+ }
+ }
+
+ pthread_rwlock_unlock(&rib.lock);
+
+ if (st->st_mode == 0) {
+ char buf[4096];
+ st->st_nlink = 2;
+ st->st_mode = __S_IFREG | 0755;
+ st->st_size = rib_read(path, buf, 4096, 0, NULL);
+ }
+
+ st->st_uid = getuid();
+ st->st_gid = getgid();
+ st->st_mtime = now.tv_sec;
+
+ return 0;
+}
+
+static struct fuse_operations r_ops = {
+ .getattr = rib_getattr,
+ .open = rib_open,
+ .opendir = rib_opendir,
+ .read = rib_read,
+ .readdir = rib_readdir
+};
+
+static void * fuse_thr(void * o)
+{
+ (void) o;
+
+ if (fuse_loop(rib.fuse) < 0)
+ return (void *) -1;
+
+ return (void *) 0;
+}
+#endif /* HAVE_FUSE */
+
+int rib_init(const char * prefix)
+{
+#ifdef HAVE_FUSE
+ struct stat st;
+ char * argv[] = {"ignored",
+ NULL,
+ "-f",
+ "-o",
+ "ro,",
+ "allow_other,",
+ "default_permissions,",
+ "fsname=rib",
+ NULL};
+ struct fuse_args args = FUSE_ARGS_INIT(0, NULL);
+
+ if (stat(FUSE_PREFIX, &st) == -1)
+ return -1;
+
+ sprintf(rib.mnt, FUSE_PREFIX "/%s.%d", prefix, getpid());
+
+ if (stat(rib.mnt, &st) == -1)
+ mkdir(rib.mnt, 0777);
+
+ argv[1] = rib.mnt;
+
+ fuse_opt_parse(&args, argv, NULL, NULL);
+
+ list_head_init(&rib.reg_comps);
+
+ rib.ch = fuse_mount(rib.mnt, &args);
+ if (rib.ch == NULL)
+ goto fail_mount;
+
+ rib.fuse = fuse_new(rib.ch, &args, &r_ops, sizeof(r_ops), NULL);
+ if (rib.fuse == NULL)
+ goto fail_fuse;
+
+ if (pthread_rwlock_init(&rib.lock, NULL))
+ goto fail_rwlock_init;
+
+ if (pthread_create(&rib.fuse_thr, NULL, fuse_thr, NULL))
+ goto fail_fuse_thr;
+
+ fuse_opt_free_args(&args);
+
+ return 0;
+
+ fail_fuse_thr:
+ pthread_rwlock_destroy(&rib.lock);
+ fail_rwlock_init:
+ fuse_destroy(rib.fuse);
+ fail_fuse:
+ fuse_unmount(rib.mnt, rib.ch);
+ fail_mount:
+ fuse_opt_free_args(&args);
+ rmdir(rib.mnt);
+ return -1;
+#else
+ (void) prefix;
+ return 0;
+#endif
+}
+
+void rib_fini(void)
+{
+#ifdef HAVE_FUSE
+ struct list_head * p;
+ struct list_head * h;
+
+ fuse_unmount(rib.mnt, rib.ch);
+ pthread_join(rib.fuse_thr, NULL);
+
+ fuse_destroy(rib.fuse);
+
+ rmdir(rib.mnt);
+
+ pthread_rwlock_wrlock(&rib.lock);
+
+ list_for_each_safe(p, h, &rib.reg_comps) {
+ struct reg_comp * c = list_entry(p, struct reg_comp, next);
+ list_del(&c->next);
+ free(c);
+ }
+
+ pthread_rwlock_unlock(&rib.lock);
+
+ pthread_rwlock_destroy(&rib.lock);
+#endif
+}
+
+int rib_reg(const char * path,
+ struct rib_ops * ops)
+{
+#ifdef HAVE_FUSE
+ struct reg_comp * rc;
+ struct list_head * p;
+
+ pthread_rwlock_wrlock(&rib.lock);
+
+ list_for_each(p, &rib.reg_comps) {
+ struct reg_comp * r = list_entry(p, struct reg_comp, next);
+ if (strcmp(r->path, path) == 0) {
+ pthread_rwlock_unlock(&rib.lock);
+ return -EPERM;
+ }
+
+ if (strcmp(r->path, path) > 0)
+ break;
+ }
+
+ rc = malloc(sizeof(*rc));
+ if (rc == NULL) {
+ pthread_rwlock_unlock(&rib.lock);
+ return -ENOMEM;
+ }
+
+ strcpy(rc->path, path);
+ rc->ops = ops;
+
+ list_add_tail(&rc->next, p);
+
+ pthread_rwlock_unlock(&rib.lock);
+#else
+ (void) path;
+ (void) ops;
+#endif
+ return 0;
+}
+
+void rib_unreg(const char * path)
+{
+#ifdef HAVE_FUSE
+ struct list_head * p;
+ struct list_head * h;
+
+ pthread_rwlock_wrlock(&rib.lock);
+
+ list_for_each_safe(p, h, &rib.reg_comps) {
+ struct reg_comp * r = list_entry(p, struct reg_comp, next);
+ if (strcmp(r->path, path) == 0) {
+ list_del(&r->next);
+ free(r);
+ break;
+ }
+ }
+
+ pthread_rwlock_unlock(&rib.lock);
+#else
+ (void) path;
+#endif
+}
generated by cgit v1.2.3 (git 2.39.1) at 2024-11-28 21:52:10 +0000