src/afb-api-so.c

   1 /*
   2  * Copyright (C) 2016, 2017 "IoT.bzh"
   3  * Author José Bollo <jose.bollo@iot.bzh>
   4  *
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *   http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  */
  17
  18 #define _GNU_SOURCE
  19
  20 #include <stdio.h>
  21 #include <dlfcn.h>
  22 #include <string.h>
  23 #include <dirent.h>
  24 #include <errno.h>
  25 #include <sys/stat.h>
  26
  27 #include "afb-api-so.h"
  28 #include "afb-api-so-v1.h"
  29 #include "afb-api-so-v2.h"
  30 #include "afb-api-so-vdyn.h"
  31 #include "verbose.h"
  32 #include "sig-monitor.h"
  33
  34 struct safe_dlopen
  35 {
  36         const char *path;
  37         void *handle;
  38         int flags;
  39 };
  40
  41 static void safe_dlopen_cb(int sig, void *closure)
  42 {
  43         struct safe_dlopen *sd = closure;
  44         if (!sig)
  45                 sd->handle = dlopen(sd->path, sd->flags);
  46         else {
  47                 ERROR("dlopen of %s raised signal %s", sd->path, strsignal(sig));
  48                 sd->handle = NULL;
  49         }
  50 }
  51
  52 static void *safe_dlopen(const char *filename, int flags)
  53 {
  54         struct safe_dlopen sd;
  55         sd.path = filename;
  56         sd.flags = flags;
  57         sd.handle = NULL;
  58         sig_monitor(0, safe_dlopen_cb, &sd);
  59         return sd.handle;
  60 }
  61
  62 static int load_binding(const char *path, int force, struct afb_apiset *apiset)
  63 {
  64         int rc;
  65         void *handle;
  66
  67         // This is a loadable library let's check if it's a binding
  68         rc = -!!force;
  69         handle = safe_dlopen(path, RTLD_NOW | RTLD_LOCAL | RTLD_DEEPBIND);
  70         if (handle == NULL) {
  71                 if (force)
  72                         ERROR("binding [%s] not loadable: %s", path, dlerror());
  73                 else
  74                         INFO("binding [%s] not loadable: %s", path, dlerror());
  75                 goto error;
  76         }
  77
  78         /* try the version 2 */
  79         rc = afb_api_so_v2_add(path, handle, apiset);
  80         if (rc < 0) {
  81                 /* error when loading a valid v2 binding */
  82                 goto error2;
  83         }
  84         if (rc)
  85                 return 0; /* yes version 2 */
  86
  87         /* try the version dyn */
  88         rc = afb_api_so_vdyn_add(path, handle, apiset);
  89         if (rc < 0) {
  90                 /* error when loading a valid dyn binding */
  91                 goto error2;
  92         }
  93         if (rc)
  94                 return 0; /* yes version dyn */
  95
  96         /* try the version 1 */
  97         rc = afb_api_so_v1_add(path, handle, apiset);
  98         if (rc < 0) {
  99                 /* error when loading a valid v1 binding */
 100                 goto error2;
 101         }
 102         if (rc)
 103                 return 0; /* yes version 1 */
 104
 105         /* not a valid binding */
 106         if (force)
 107                 ERROR("binding [%s] is not an AFB binding", path);
 108         else
 109                 INFO("binding [%s] is not an AFB binding", path);
 110
 111 error2:
 112         dlclose(handle);
 113 error:
 114         return rc;
 115 }
 116
 117
 118 int afb_api_so_add_binding(const char *path, struct afb_apiset *apiset)
 119 {
 120         return load_binding(path, 1, apiset);
 121 }
 122
 123 static int adddirs(char path[PATH_MAX], size_t end, struct afb_apiset *apiset, int failstops)
 124 {
 125         DIR *dir;
 126         struct dirent *dent;
 127         size_t len;
 128         int rc = 0;
 129
 130         /* open the DIR now */
 131         dir = opendir(path);
 132         if (dir == NULL) {
 133                 ERROR("can't scan binding directory %s, %m", path);
 134                 return -1;
 135         }
 136         INFO("Scanning dir=[%s] for bindings", path);
 137
 138         /* scan each entry */
 139         if (end)
 140                 path[end++] = '/';
 141         for (;;) {
 142                 errno = 0;
 143                 dent = readdir(dir);
 144                 if (dent == NULL) {
 145                         if (errno != 0)
 146                                 ERROR("read error while scanning directory %.*s: %m", (int)(end - 1), path);
 147                         break;
 148                 }
 149
 150                 len = strlen(dent->d_name);
 151                 if (len + end >= PATH_MAX) {
 152                         ERROR("path too long while scanning bindings for %s", dent->d_name);
 153                         continue;
 154                 }
 155                 if (dent->d_type == DT_DIR) {
 156                         /* case of directories */
 157                         if (dent->d_name[0] == '.') {
 158                                 if (len == 1)
 159                                         continue; /* . */
 160                                 if (dent->d_name[1] == '.' && len == 2)
 161                                         continue; /* .. */
 162                         }
 163                         memcpy(&path[end], dent->d_name, len+1);
 164                         rc = adddirs(path, end+len, apiset, failstops);
 165                 } else if (dent->d_type == DT_REG) {
 166                         /* case of files */
 167                         if (memcmp(&dent->d_name[len - 3], ".so", 4))
 168                                 continue;
 169                         memcpy(&path[end], dent->d_name, len+1);
 170                         rc = load_binding(path, 0, apiset);
 171                 }
 172                 if (rc < 0 && failstops) {
 173                         closedir(dir);
 174                         return rc;
 175                 }
 176         }
 177         closedir(dir);
 178         return 0;
 179 }
 180
 181 int afb_api_so_add_directory(const char *path, struct afb_apiset *apiset, int failstops)
 182 {
 183         size_t length;
 184         char buffer[PATH_MAX];
 185
 186         length = strlen(path);
 187         if (length >= sizeof(buffer)) {
 188                 ERROR("path too long %lu [%.99s...]", (unsigned long)length, path);
 189                 return -1;
 190         }
 191
 192         memcpy(buffer, path, length + 1);
 193         return adddirs(buffer, length, apiset, failstops);
 194 }
 195
 196 int afb_api_so_add_path(const char *path, struct afb_apiset *apiset, int failstops)
 197 {
 198         struct stat st;
 199         int rc;
 200
 201         rc = stat(path, &st);
 202         if (rc < 0)
 203                 ERROR("Invalid binding path [%s]: %m", path);
 204         else if (S_ISDIR(st.st_mode))
 205                 rc = afb_api_so_add_directory(path, apiset, failstops);
 206         else if (strstr(path, ".so"))
 207                 rc = load_binding(path, 0, apiset);
 208         else
 209                 INFO("not a binding [%s], skipped", path);
 210         return rc;
 211 }
 212
 213 int afb_api_so_add_pathset(const char *pathset, struct afb_apiset *apiset, int failstops)
 214 {
 215         static char sep[] = ":";
 216         char *ps, *p;
 217         int rc;
 218
 219         ps = strdupa(pathset);
 220         for (;;) {
 221                 p = strsep(&ps, sep);
 222                 if (!p)
 223                         return 0;
 224                 rc = afb_api_so_add_path(p, apiset, failstops);
 225                 if (rc < 0)
 226                         return rc;
 227         }
 228 }
 229
 230 int afb_api_so_add_pathset_fails(const char *pathset, struct afb_apiset *apiset)
 231 {
 232         return afb_api_so_add_pathset(pathset, apiset, 1);
 233 }
 234
 235 int afb_api_so_add_pathset_nofails(const char *pathset, struct afb_apiset *apiset)
 236 {
 237         return afb_api_so_add_pathset(pathset, apiset, 0);
 238 }
 239