src/afb-apiset.c

   1 /*
   2  * Copyright (C) 2016, 2017 "IoT.bzh"
   3  * Author "Fulup Ar Foll"
   4  * Author José Bollo <jose.bollo@iot.bzh>
   5  *
   6  * Licensed under the Apache License, Version 2.0 (the "License");
   7  * you may not use this file except in compliance with the License.
   8  * You may obtain a copy of the License at
   9  *
  10  *   http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software
  13  * distributed under the License is distributed on an "AS IS" BASIS,
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  * See the License for the specific language governing permissions and
  16  * limitations under the License.
  17  */
  18
  19 #define _GNU_SOURCE
  20
  21 #include <stdlib.h>
  22 #include <stdio.h>
  23 #include <string.h>
  24 #include <errno.h>
  25
  26 #include "afb-session.h"
  27 #include "verbose.h"
  28 #include "afb-api.h"
  29 #include "afb-apiset.h"
  30 #include "afb-context.h"
  31 #include "afb-xreq.h"
  32 #include "jobs.h"
  33
  34 #define INCR 8          /* CAUTION: must be a power of 2 */
  35
  36 /**
  37  * Internal description of an api
  38  */
  39 struct api_desc
  40 {
  41         int status;
  42         const char *name;       /**< name of the api */
  43         struct afb_api api;     /**< handler of the api */
  44 };
  45
  46 /**
  47  * Data structure for apiset
  48  */
  49 struct afb_apiset
  50 {
  51         struct api_desc *apis;          /**< description of apis */
  52         struct afb_apiset *subset;      /**< subset if any */
  53         int count;                      /**< count of apis in the set */
  54         int timeout;                    /**< the timeout in second for the apiset */
  55         int refcount;                   /**< reference count for freeing resources */
  56         char name[1];                   /**< name of the apiset */
  57 };
  58
  59 /**
  60  * Search the api of 'name'.
  61  * @param set the api set
  62  * @param name the api name to search
  63  * @return the descriptor if found or NULL otherwise
  64  */
  65 static struct api_desc *search(struct afb_apiset *set, const char *name)
  66 {
  67         int i, c, up, lo;
  68         struct api_desc *a;
  69
  70         /* dichotomic search of the api */
  71         /* initial slice */
  72         lo = 0;
  73         up = set->count;
  74         for (;;) {
  75                 /* check remaining slice */
  76                 if (lo >= up) {
  77                         /* not found */
  78                         return NULL;
  79                 }
  80                 /* check the mid of the slice */
  81                 i = (lo + up) >> 1;
  82                 a = &set->apis[i];
  83                 c = strcasecmp(a->name, name);
  84                 if (c == 0) {
  85                         /* found */
  86                         return a;
  87                 }
  88                 /* update the slice */
  89                 if (c < 0)
  90                         lo = i + 1;
  91                 else
  92                         up = i;
  93         }
  94 }
  95
  96 /**
  97  * Increases the count of references to the apiset and return its address
  98  * @param set the set whose reference count is to be increased
  99  * @return the given apiset
 100  */
 101 struct afb_apiset *afb_apiset_addref(struct afb_apiset *set)
 102 {
 103         if (set)
 104                 __atomic_add_fetch(&set->refcount, 1, __ATOMIC_RELAXED);
 105         return set;
 106 }
 107
 108 /**
 109  * Decreases the count of references to the apiset and frees its
 110  * resources when no more references exists.
 111  * @param set the set to unrefrence
 112  */
 113 void afb_apiset_unref(struct afb_apiset *set)
 114 {
 115         if (set && !__atomic_sub_fetch(&set->refcount, 1, __ATOMIC_RELAXED)) {
 116                 afb_apiset_unref(set->subset);
 117                 free(set->apis);
 118                 free(set);
 119         }
 120 }
 121
 122 /**
 123  * Create an apiset
 124  * @param name the name of the apiset
 125  * @param timeout the default timeout in seconds for the apiset
 126  * @return the created apiset or NULL in case of error
 127  */
 128 struct afb_apiset *afb_apiset_create(const char *name, int timeout)
 129 {
 130         struct afb_apiset *set;
 131
 132         set = malloc((name ? strlen(name) : 0) + sizeof *set);
 133         if (set) {
 134                 set->apis = malloc(INCR * sizeof *set->apis);
 135                 set->count = 0;
 136                 set->timeout = timeout;
 137                 set->refcount = 1;
 138                 set->subset = NULL;
 139                 if (name)
 140                         strcpy(set->name, name);
 141                 else
 142                         set->name[0] = 0;
 143         }
 144         return set;
 145 }
 146
 147 /**
 148  * the name of the apiset
 149  * @param set the api set
 150  * @return the name of the set
 151  */
 152 const char *afb_apiset_name(struct afb_apiset *set)
 153 {
 154         return set->name;
 155 }
 156
 157 /**
 158  * Get the API timeout of the set
 159  * @param set the api set
 160  * @return the timeout in seconds
 161  */
 162 int afb_apiset_timeout_get(struct afb_apiset *set)
 163 {
 164         return set->timeout;
 165 }
 166
 167 /**
 168  * Set the API timeout of the set
 169  * @param set the api set
 170  * @param to the timeout in seconds
 171  */
 172 void afb_apiset_timeout_set(struct afb_apiset *set, int to)
 173 {
 174         set->timeout = to;
 175 }
 176
 177 /**
 178  * Get the subset of the set
 179  * @param set the api set
 180  * @return the subset of set
 181  */
 182 struct afb_apiset *afb_apiset_subset_get(struct afb_apiset *set)
 183 {
 184         return set->subset;
 185 }
 186
 187 /**
 188  * Set the subset of the set
 189  * @param set the api set
 190  * @param subset the subset to set
 191  */
 192 void afb_apiset_subset_set(struct afb_apiset *set, struct afb_apiset *subset)
 193 {
 194         struct afb_apiset *tmp;
 195         if (subset == set) {
 196                 /* avoid infinite loop */
 197                 subset = NULL;
 198         }
 199         tmp = set->subset;
 200         set->subset = afb_apiset_addref(subset);
 201         afb_apiset_unref(tmp);
 202 }
 203
 204 /**
 205  * Adds the api of 'name' described by 'api'.
 206  * @param set the api set
 207  * @param name the name of the api to add (have to survive, not copied!)
 208  * @param api the api
 209  * @returns 0 in case of success or -1 in case
 210  * of error with errno set:
 211  *   - EEXIST if name already registered
 212  *   - ENOMEM when out of memory
 213  */
 214 int afb_apiset_add(struct afb_apiset *set, const char *name, struct afb_api api)
 215 {
 216         struct api_desc *apis;
 217         int i, c;
 218
 219         /* check previously existing plugin */
 220         for (i = 0 ; i < set->count ; i++) {
 221                 c = strcasecmp(set->apis[i].name, name);
 222                 if (c == 0) {
 223                         ERROR("api of name %s already exists", name);
 224                         errno = EEXIST;
 225                         goto error;
 226                 }
 227                 if (c > 0)
 228                         break;
 229         }
 230
 231         /* allocates enough memory */
 232         c = (set->count + INCR) & ~(INCR - 1);
 233         apis = realloc(set->apis, ((unsigned)c) * sizeof * apis);
 234         if (apis == NULL) {
 235                 ERROR("out of memory");
 236                 errno = ENOMEM;
 237                 goto error;
 238         }
 239         set->apis = apis;
 240
 241         /* copy higher part of the array */
 242         apis += i;
 243         if (i != set->count)
 244                 memmove(apis + 1, apis, ((unsigned)(set->count - i)) * sizeof *apis);
 245
 246         /* record the plugin */
 247         apis->status = -1;
 248         apis->api = api;
 249         apis->name = name;
 250         set->count++;
 251
 252         NOTICE("API %s added", name);
 253
 254         return 0;
 255
 256 error:
 257         return -1;
 258 }
 259
 260 /**
 261  * Delete from the 'set' the api of 'name'.
 262  * @param set the set to be changed
 263  * @param name the name of the API to remove
 264  * @return 0 in case of success or -1 in case where the API doesn't exist.
 265  */
 266 int afb_apiset_del(struct afb_apiset *set, const char *name)
 267 {
 268         int i, c;
 269
 270         /* search the api */
 271         for (i = 0 ; i < set->count ; i++) {
 272                 c = strcasecmp(set->apis[i].name, name);
 273                 if (c == 0) {
 274                         set->count--;
 275                         while(i < set->count) {
 276                                 set->apis[i] = set->apis[i + 1];
 277                                 i++;
 278                         }
 279                         return 0;
 280                 }
 281                 if (c > 0)
 282                         break;
 283         }
 284         errno = ENOENT;
 285         return -1;
 286 }
 287
 288 /**
 289  * Get from the 'set' the API of 'name' in 'api'
 290  * @param set the set of API
 291  * @param name the name of the API to get
 292  * @param api the structure where to store data about the API of name
 293  * @return 0 in case of success or -1 in case of error
 294  */
 295 int afb_apiset_lookup(struct afb_apiset *set, const char *name, struct afb_api *api)
 296 {
 297         const struct api_desc *i;
 298
 299         i = search(set, name);
 300         if (i) {
 301                 *api = i->api;
 302                 return 0;
 303         }
 304
 305         errno = ENOENT;
 306         return -1;
 307 }
 308
 309 /**
 310  * Get from the 'set' the API of 'name' in 'api' with fallback to subset or default api
 311  * @param set the set of API
 312  * @param name the name of the API to get
 313  * @param api the structure where to store data about the API of name
 314  * @return 0 in case of success or -1 in case of error
 315  */
 316 static struct api_desc *get_api(struct afb_apiset *set, const char *name)
 317 {
 318         struct api_desc *i = search(set, name);
 319         return i || !set->subset ? i : get_api(set->subset, name);
 320 }
 321
 322 /**
 323  * Get from the 'set' the API of 'name' in 'api' with fallback to subset or default api
 324  * @param set the set of API
 325  * @param name the name of the API to get
 326  * @param api the structure where to store data about the API of name
 327  * @return 0 in case of success or -1 in case of error
 328  */
 329 int afb_apiset_get(struct afb_apiset *set, const char *name, struct afb_api *api)
 330 {
 331         const struct api_desc *i;
 332
 333         i = get_api(set, name);
 334         if (!i) {
 335                 errno = ENOENT;
 336                 return -1;
 337         }
 338         *api = i->api;
 339         return 0;
 340 }
 341
 342 /**
 343  * Starts the service 'api'.
 344  * @param api the api
 345  * @param share_session if true start the servic"e in a shared session
 346  *                      if false start it in its own session
 347  * @param onneed if true start the service if possible, if false the api
 348  *               must be a service
 349  * @return a positive number on success
 350  */
 351 static int start_api(struct afb_apiset *set, struct api_desc *api, int share_session, int onneed)
 352 {
 353         int rc;
 354
 355         if (api->status == 0)
 356                 return 0;
 357         else if (api->status > 0) {
 358                 errno = api->status;
 359                 return -1;
 360         }
 361
 362         NOTICE("API %s starting...", api->name);
 363         if (api->api.itf->service_start) {
 364                 api->status = EBUSY;
 365                 rc = api->api.itf->service_start(api->api.closure, share_session, onneed, set);
 366                 if (rc < 0) {
 367                         api->status = errno ?: ECANCELED;
 368                         ERROR("The api %s failed to start (%d)", api->name, rc);
 369                         return -1;
 370                 }
 371         } else if (!onneed) {
 372                 /* already started: it is an error */
 373                 ERROR("The api %s is not a startable service", api->name);
 374                 api->status = EINVAL;
 375                 return -1;
 376         }
 377         NOTICE("API %s started", api->name);
 378         api->status = 0;
 379         return 0;
 380 }
 381
 382 /**
 383  * Get from the 'set' the API of 'name' in 'api' with fallback to subset or default api
 384  * @param set the set of API
 385  * @param name the name of the API to get
 386  * @param api the structure where to store data about the API of name
 387  * @return 0 in case of success or -1 in case of error
 388  */
 389 int afb_apiset_get_started(struct afb_apiset *set, const char *name, struct afb_api *api)
 390 {
 391         struct api_desc *i;
 392
 393         i = get_api(set, name);
 394         if (!i) {
 395                 errno = ENOENT;
 396                 return -1;
 397         }
 398         *api = i->api;
 399         return i->status ? start_api(set, i, 1, 1) : 0;
 400 }
 401
 402 /**
 403  * Starts a service by its 'api' name.
 404  * @param set the api set
 405  * @param name name of the service to start
 406  * @param share_session if true start the servic"e in a shared session
 407  *                      if false start it in its own session
 408  * @param onneed if true start the service if possible, if false the api
 409  *               must be a service
 410  * @return a positive number on success
 411  */
 412 int afb_apiset_start_service(struct afb_apiset *set, const char *name, int share_session, int onneed)
 413 {
 414         struct api_desc *a;
 415
 416         a = search(set, name);
 417         if (!a) {
 418                 ERROR("can't find service %s", name);
 419                 errno = ENOENT;
 420                 return -1;
 421         }
 422
 423         return start_api(set, a, share_session, onneed);
 424 }
 425
 426 /**
 427  * Starts all possible services but stops at first error.
 428  * @param set the api set
 429  * @param share_session if true start the servic"e in a shared session
 430  *                      if false start it in its own session
 431  * @return 0 on success or a negative number when an error is found
 432  */
 433 int afb_apiset_start_all_services(struct afb_apiset *set, int share_session)
 434 {
 435         int rc;
 436         struct api_desc *i, *e;
 437
 438         i = set->apis;
 439         e = &set->apis[set->count];
 440         while (i != e) {
 441                 rc = start_api(set, i, share_session, 1);
 442                 if (rc < 0)
 443                         return rc;
 444                 i++;
 445         }
 446
 447         return set->subset ? afb_apiset_start_all_services(set->subset, share_session) : 0;
 448 }
 449
 450 /**
 451  * Ask to update the hook flags of the 'api'
 452  * @param set the api set
 453  * @param name the api to update (NULL updates all)
 454  */
 455 void afb_apiset_update_hooks(struct afb_apiset *set, const char *name)
 456 {
 457         const struct api_desc *i, *e;
 458
 459         if (!name) {
 460                 i = set->apis;
 461                 e = &set->apis[set->count];
 462         } else {
 463                 i = search(set, name);
 464                 e = &i[!!i];
 465         }
 466         while (i != e) {
 467                 if (i->api.itf->update_hooks)
 468                         i->api.itf->update_hooks(i->api.closure);
 469                 i++;
 470         }
 471 }
 472
 473 /**
 474  * Set the verbosity level of the 'api'
 475  * @param set the api set
 476  * @param name the api to set (NULL set all)
 477  */
 478 void afb_apiset_set_verbosity(struct afb_apiset *set, const char *name, int level)
 479 {
 480         const struct api_desc *i, *e;
 481
 482         if (!name) {
 483                 i = set->apis;
 484                 e = &set->apis[set->count];
 485         } else {
 486                 i = search(set, name);
 487                 e = &i[!!i];
 488         }
 489         while (i != e) {
 490                 if (i->api.itf->set_verbosity)
 491                         i->api.itf->set_verbosity(i->api.closure, level);
 492                 i++;
 493         }
 494 }
 495
 496 /**
 497  * Set the verbosity level of the 'api'
 498  * @param set the api set
 499  * @param name the api to set (NULL set all)
 500  */
 501 int afb_apiset_get_verbosity(struct afb_apiset *set, const char *name)
 502 {
 503         const struct api_desc *i;
 504
 505         i = name ? search(set, name) : NULL;
 506         if (!i) {
 507                 errno = ENOENT;
 508                 return -1;
 509         }
 510         if (!i->api.itf->get_verbosity)
 511                 return verbosity;
 512
 513         return i->api.itf->get_verbosity(i->api.closure);
 514 }
 515
 516 /**
 517  * Get the list of api names
 518  * @param set the api set
 519  * @return a NULL terminated array of api names. Must be freed.
 520  */
 521 const char **afb_apiset_get_names(struct afb_apiset *set)
 522 {
 523         size_t size;
 524         char *dest;
 525         const char **names;
 526         int i;
 527
 528         size = set->count * (1 + sizeof(*names)) + sizeof(*names);
 529         for (i = 0 ; i < set->count ; i++)
 530                 size += strlen(set->apis[i].name);
 531
 532         names = malloc(size);
 533         if (!names)
 534                 errno = ENOMEM;
 535         else {
 536                 dest = (void*)&names[set->count+1];
 537                 for (i = 0 ; i < set->count ; i++) {
 538                         names[i] = dest;
 539                         dest = stpcpy(dest, set->apis[i].name) + 1;
 540                 }
 541                 names[i] = NULL;
 542         }
 543         return names;
 544 }
 545
 546 /**
 547  * Enumerate the api names to a callback.
 548  * @param set the api set
 549  * @param callback the function to call for each name
 550  * @param closure the closure for the callback
 551  */
 552 void afb_apiset_enum(struct afb_apiset *set, void (*callback)(struct afb_apiset *set, const char *name, void *closure), void *closure)
 553 {
 554         int i;
 555
 556         for (i = 0 ; i < set->count ; i++)
 557                 callback(set, set->apis[i].name, closure);
 558 }
 559