src/afb-xreq.c

   1 /*
   2  * Copyright (C) 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 <stdlib.h>
  21 #include <stdio.h>
  22 #include <string.h>
  23 #include <errno.h>
  24
  25 #include <json-c/json.h>
  26 #include <afb/afb-binding-v1.h>
  27 #include <afb/afb-binding-v2.h>
  28 #include <afb/afb-request.h>
  29
  30 #include "afb-context.h"
  31 #include "afb-xreq.h"
  32 #include "afb-evt.h"
  33 #include "afb-msg-json.h"
  34 #include "afb-cred.h"
  35 #include "afb-hook.h"
  36 #include "afb-api.h"
  37 #include "afb-apiset.h"
  38 #include "afb-auth.h"
  39 #include "jobs.h"
  40 #include "verbose.h"
  41
  42 /******************************************************************************/
  43
  44 static void xreq_finalize(struct afb_xreq *xreq)
  45 {
  46         if (!xreq->replied)
  47                 afb_xreq_fail(xreq, "error", "no reply");
  48         if (xreq->hookflags)
  49                 afb_hook_xreq_end(xreq);
  50         if (xreq->caller)
  51                 afb_xreq_unhooked_unref(xreq->caller);
  52         xreq->queryitf->unref(xreq);
  53 }
  54
  55 inline void afb_xreq_unhooked_addref(struct afb_xreq *xreq)
  56 {
  57         __atomic_add_fetch(&xreq->refcount, 1, __ATOMIC_RELAXED);
  58 }
  59
  60 inline void afb_xreq_unhooked_unref(struct afb_xreq *xreq)
  61 {
  62         if (!__atomic_sub_fetch(&xreq->refcount, 1, __ATOMIC_RELAXED))
  63                 xreq_finalize(xreq);
  64 }
  65
  66 /******************************************************************************/
  67
  68 static inline struct afb_request *to_request(struct afb_xreq *xreq)
  69 {
  70         return &xreq->request;
  71 }
  72
  73 static inline struct afb_req to_req(struct afb_xreq *xreq)
  74 {
  75         return (struct afb_req){ .itf = xreq->request.itf, .closure = &xreq->request };
  76 }
  77
  78 static inline struct afb_xreq *from_request(struct afb_request *request)
  79 {
  80         return CONTAINER_OF(struct afb_xreq, request, request);
  81 }
  82
  83 /******************************************************************************/
  84
  85 struct subcall
  86 {
  87         struct afb_xreq xreq;
  88
  89         void (*completion)(struct subcall*, int, struct json_object*);
  90
  91         union {
  92                 struct {
  93                         struct jobloop *jobloop;
  94                         struct json_object *result;
  95                         int status;
  96                 };
  97                 struct {
  98                         union {
  99                                 void (*callback)(void*, int, struct json_object*);
 100                                 void (*callback2)(void*, int, struct json_object*, struct afb_req);
 101                         };
 102                         void *closure;
 103                 };
 104         };
 105 };
 106
 107 static int subcall_subscribe_cb(struct afb_xreq *xreq, struct afb_eventid *eventid)
 108 {
 109         struct subcall *subcall = CONTAINER_OF_XREQ(struct subcall, xreq);
 110
 111         return afb_xreq_subscribe(subcall->xreq.caller, eventid);
 112 }
 113
 114 static int subcall_unsubscribe_cb(struct afb_xreq *xreq, struct afb_eventid *eventid)
 115 {
 116         struct subcall *subcall = CONTAINER_OF_XREQ(struct subcall, xreq);
 117
 118         return afb_xreq_unsubscribe(subcall->xreq.caller, eventid);
 119 }
 120
 121 static void subcall_reply_cb(struct afb_xreq *xreq, int status, struct json_object *result)
 122 {
 123         struct subcall *subcall = CONTAINER_OF_XREQ(struct subcall, xreq);
 124
 125         subcall->completion(subcall, status, result);
 126         json_object_put(result);
 127         afb_xreq_unhooked_unref(&subcall->xreq);
 128 }
 129
 130 static void subcall_destroy_cb(struct afb_xreq *xreq)
 131 {
 132         struct subcall *subcall = CONTAINER_OF_XREQ(struct subcall, xreq);
 133
 134         json_object_put(subcall->xreq.json);
 135         afb_cred_unref(subcall->xreq.cred);
 136         free(subcall);
 137 }
 138
 139 const struct afb_xreq_query_itf afb_xreq_subcall_itf = {
 140         .reply = subcall_reply_cb,
 141         .unref = subcall_destroy_cb,
 142         .subscribe = subcall_subscribe_cb,
 143         .unsubscribe = subcall_unsubscribe_cb
 144 };
 145
 146 static struct subcall *subcall_alloc(
 147                 struct afb_xreq *caller,
 148                 const char *api,
 149                 const char *verb,
 150                 struct json_object *args
 151 )
 152 {
 153         struct subcall *subcall;
 154         size_t lenapi, lenverb;
 155         char *copy;
 156
 157         lenapi = 1 + strlen(api);
 158         lenverb = 1 + strlen(verb);
 159         subcall = malloc(lenapi + lenverb + sizeof *subcall);
 160         if (!subcall)
 161                 ERROR("out of memory");
 162         else {
 163                 copy = (char*)&subcall[1];
 164                 memcpy(copy, api, lenapi);
 165                 api = copy;
 166                 copy = &copy[lenapi];
 167                 memcpy(copy, verb, lenverb);
 168                 verb = copy;
 169
 170                 afb_xreq_init(&subcall->xreq, &afb_xreq_subcall_itf);
 171                 afb_context_subinit(&subcall->xreq.context, &caller->context);
 172                 subcall->xreq.cred = afb_cred_addref(caller->cred);
 173                 subcall->xreq.json = args;
 174                 subcall->xreq.api = api;
 175                 subcall->xreq.verb = verb;
 176                 subcall->xreq.caller = caller;
 177                 afb_xreq_unhooked_addref(caller);
 178         }
 179         return subcall;
 180 }
 181
 182
 183 static void subcall_on_reply(struct subcall *subcall, int status, struct json_object *result)
 184 {
 185         subcall->callback(subcall->closure, status, result);
 186 }
 187
 188 static void subcall_req_on_reply(struct subcall *subcall, int status, struct json_object *result)
 189 {
 190         subcall->callback2(subcall->closure, status, result, to_req(subcall->xreq.caller));
 191 }
 192
 193 static void subcall_hooked_on_reply(struct subcall *subcall, int status, struct json_object *result)
 194 {
 195         afb_hook_xreq_subcall_result(subcall->xreq.caller, status, result);
 196         subcall_on_reply(subcall, status, result);
 197 }
 198
 199 static void subcall_req_hooked_on_reply(struct subcall *subcall, int status, struct json_object *result)
 200 {
 201         afb_hook_xreq_subcall_req_result(subcall->xreq.caller, status, result);
 202         subcall_req_on_reply(subcall, status, result);
 203 }
 204
 205 static void subcall_reply_direct_cb(void *closure, int status, struct json_object *result)
 206 {
 207         struct afb_xreq *xreq = closure;
 208
 209         if (xreq->replied) {
 210                 ERROR("subcall replied more than one time!!");
 211                 json_object_put(result);
 212         } else {
 213                 xreq->replied = 1;
 214                 subcall_reply_cb(xreq, status, result);
 215         }
 216 }
 217
 218 static void subcall_process(struct subcall *subcall, void (*completion)(struct subcall*, int, struct json_object*))
 219 {
 220         subcall->completion = completion;
 221         if (subcall->xreq.caller->queryitf->subcall) {
 222                 subcall->xreq.caller->queryitf->subcall(
 223                         subcall->xreq.caller, subcall->xreq.api, subcall->xreq.verb,
 224                         subcall->xreq.json, subcall_reply_direct_cb, &subcall->xreq);
 225         } else {
 226                 afb_xreq_unhooked_addref(&subcall->xreq);
 227                 afb_xreq_process(&subcall->xreq, subcall->xreq.caller->apiset);
 228         }
 229 }
 230
 231 static void subcall(struct subcall *subcall, void (*callback)(void*, int, struct json_object*), void *cb_closure)
 232 {
 233         subcall->callback = callback;
 234         subcall->closure = cb_closure;
 235         subcall_process(subcall, subcall_on_reply);
 236 }
 237
 238 static void subcall_req(struct subcall *subcall, void (*callback)(void*, int, struct json_object*, struct afb_req), void *cb_closure)
 239 {
 240         subcall->callback2 = callback;
 241         subcall->closure = cb_closure;
 242         subcall_process(subcall, subcall_req_on_reply);
 243 }
 244
 245 static void subcall_hooked(struct subcall *subcall, void (*callback)(void*, int, struct json_object*), void *cb_closure)
 246 {
 247         subcall->callback = callback;
 248         subcall->closure = cb_closure;
 249         subcall_process(subcall, subcall_hooked_on_reply);
 250 }
 251
 252 static void subcall_req_hooked(struct subcall *subcall, void (*callback)(void*, int, struct json_object*, struct afb_req), void *cb_closure)
 253 {
 254         subcall->callback2 = callback;
 255         subcall->closure = cb_closure;
 256         subcall_process(subcall, subcall_req_hooked_on_reply);
 257 }
 258
 259 static void subcall_sync_leave(struct subcall *subcall)
 260 {
 261         struct jobloop *jobloop = __atomic_exchange_n(&subcall->jobloop, NULL, __ATOMIC_RELAXED);
 262         if (jobloop)
 263                 jobs_leave(jobloop);
 264 }
 265
 266 static void subcall_sync_reply(struct subcall *subcall, int status, struct json_object *result)
 267 {
 268         subcall->status = status;
 269         subcall->result = json_object_get(result);
 270         subcall_sync_leave(subcall);
 271 }
 272
 273 static void subcall_sync_enter(int signum, void *closure, struct jobloop *jobloop)
 274 {
 275         struct subcall *subcall = closure;
 276
 277         if (!signum) {
 278                 subcall->jobloop = jobloop;
 279                 subcall->result = NULL;
 280                 subcall->status = 0;
 281                 subcall_process(subcall, subcall_sync_reply);
 282         } else {
 283                 subcall->status = -1;
 284                 subcall_sync_leave(subcall);
 285         }
 286 }
 287
 288 static int subcallsync(struct subcall *subcall, struct json_object **result)
 289 {
 290         int rc;
 291
 292         afb_xreq_unhooked_addref(&subcall->xreq);
 293         rc = jobs_enter(NULL, 0, subcall_sync_enter, subcall);
 294         *result = subcall->result;
 295         if (rc < 0 || subcall->status < 0) {
 296                 *result = *result ?: afb_msg_json_internal_error();
 297                 rc = -1;
 298         }
 299         afb_xreq_unhooked_unref(&subcall->xreq);
 300         return rc;
 301 }
 302
 303 /******************************************************************************/
 304
 305 static void vinfo(void *first, void *second, const char *fmt, va_list args, void (*fun)(void*,void*,const char*))
 306 {
 307         char *info;
 308         if (fmt == NULL || vasprintf(&info, fmt, args) < 0)
 309                 info = NULL;
 310         fun(first, second, info);
 311         free(info);
 312 }
 313
 314 /******************************************************************************/
 315
 316 static struct json_object *xreq_json_cb(struct afb_request *closure)
 317 {
 318         struct afb_xreq *xreq = from_request(closure);
 319         if (!xreq->json && xreq->queryitf->json)
 320                 xreq->json = xreq->queryitf->json(xreq);
 321         return xreq->json;
 322 }
 323
 324 static struct afb_arg xreq_get_cb(struct afb_request *closure, const char *name)
 325 {
 326         struct afb_xreq *xreq = from_request(closure);
 327         struct afb_arg arg;
 328         struct json_object *object, *value;
 329
 330         if (xreq->queryitf->get)
 331                 arg = xreq->queryitf->get(xreq, name);
 332         else {
 333                 object = xreq_json_cb(closure);
 334                 if (json_object_object_get_ex(object, name, &value)) {
 335                         arg.name = name;
 336                         arg.value = json_object_get_string(value);
 337                 } else {
 338                         arg.name = NULL;
 339                         arg.value = NULL;
 340                 }
 341                 arg.path = NULL;
 342         }
 343         return arg;
 344 }
 345
 346 static void xreq_success_cb(struct afb_request *closure, struct json_object *obj, const char *info)
 347 {
 348         struct afb_xreq *xreq = from_request(closure);
 349
 350         if (xreq->replied) {
 351                 ERROR("reply called more than one time!!");
 352                 json_object_put(obj);
 353         } else {
 354                 xreq->replied = 1;
 355                 if (xreq->queryitf->success)
 356                         xreq->queryitf->success(xreq, obj, info);
 357                 else
 358                         xreq->queryitf->reply(xreq, 0, afb_msg_json_reply_ok(info, obj, &xreq->context, NULL));
 359         }
 360 }
 361
 362 static void xreq_fail_cb(struct afb_request *closure, const char *status, const char *info)
 363 {
 364         struct afb_xreq *xreq = from_request(closure);
 365
 366         if (xreq->replied) {
 367                 ERROR("reply called more than one time!!");
 368         } else {
 369                 xreq->replied = 1;
 370                 if (xreq->queryitf->fail)
 371                         xreq->queryitf->fail(xreq, status, info);
 372                 else
 373                         xreq->queryitf->reply(xreq, -1, afb_msg_json_reply_error(status, info, &xreq->context, NULL));
 374         }
 375 }
 376
 377 static void xreq_vsuccess_cb(struct afb_request *closure, struct json_object *obj, const char *fmt, va_list args)
 378 {
 379         vinfo(closure, obj, fmt, args, (void*)xreq_success_cb);
 380 }
 381
 382 static void xreq_vfail_cb(struct afb_request *closure, const char *status, const char *fmt, va_list args)
 383 {
 384         vinfo(closure, (void*)status, fmt, args, (void*)xreq_fail_cb);
 385 }
 386
 387 static void *xreq_context_get_cb(struct afb_request *closure)
 388 {
 389         struct afb_xreq *xreq = from_request(closure);
 390         return afb_context_get(&xreq->context);
 391 }
 392
 393 static void xreq_context_set_cb(struct afb_request *closure, void *value, void (*free_value)(void*))
 394 {
 395         struct afb_xreq *xreq = from_request(closure);
 396         afb_context_set(&xreq->context, value, free_value);
 397 }
 398
 399 static struct afb_request *xreq_addref_cb(struct afb_request *closure)
 400 {
 401         struct afb_xreq *xreq = from_request(closure);
 402         afb_xreq_unhooked_addref(xreq);
 403         return closure;
 404 }
 405
 406 static void xreq_unref_cb(struct afb_request *closure)
 407 {
 408         struct afb_xreq *xreq = from_request(closure);
 409         afb_xreq_unhooked_unref(xreq);
 410 }
 411
 412 static void xreq_session_close_cb(struct afb_request *closure)
 413 {
 414         struct afb_xreq *xreq = from_request(closure);
 415         afb_context_close(&xreq->context);
 416 }
 417
 418 static int xreq_session_set_LOA_cb(struct afb_request *closure, unsigned level)
 419 {
 420         struct afb_xreq *xreq = from_request(closure);
 421         return afb_context_change_loa(&xreq->context, level);
 422 }
 423
 424 static int xreq_subscribe_eventid_cb(struct afb_request *closure, struct afb_eventid *eventid);
 425 static int xreq_subscribe_cb(struct afb_request *closure, struct afb_event event)
 426 {
 427         return xreq_subscribe_eventid_cb(closure, event.closure);
 428 }
 429
 430 static int xreq_subscribe_eventid_cb(struct afb_request *closure, struct afb_eventid *eventid)
 431 {
 432         struct afb_xreq *xreq = from_request(closure);
 433         return afb_xreq_subscribe(xreq, eventid);
 434 }
 435
 436 int afb_xreq_subscribe(struct afb_xreq *xreq, struct afb_eventid *eventid)
 437 {
 438         if (xreq->listener)
 439                 return afb_evt_add_watch(xreq->listener, eventid);
 440         if (xreq->queryitf->subscribe)
 441                 return xreq->queryitf->subscribe(xreq, eventid);
 442         ERROR("no event listener, subscription impossible");
 443         errno = EINVAL;
 444         return -1;
 445 }
 446
 447 static int xreq_unsubscribe_eventid_cb(struct afb_request *closure, struct afb_eventid *eventid);
 448 static int xreq_unsubscribe_cb(struct afb_request *closure, struct afb_event event)
 449 {
 450         return xreq_unsubscribe_eventid_cb(closure, event.closure);
 451 }
 452
 453 static int xreq_unsubscribe_eventid_cb(struct afb_request *closure, struct afb_eventid *eventid)
 454 {
 455         struct afb_xreq *xreq = from_request(closure);
 456         return afb_xreq_unsubscribe(xreq, eventid);
 457 }
 458
 459 int afb_xreq_unsubscribe(struct afb_xreq *xreq, struct afb_eventid *eventid)
 460 {
 461         if (xreq->listener)
 462                 return afb_evt_remove_watch(xreq->listener, eventid);
 463         if (xreq->queryitf->unsubscribe)
 464                 return xreq->queryitf->unsubscribe(xreq, eventid);
 465         ERROR("no event listener, unsubscription impossible");
 466         errno = EINVAL;
 467         return -1;
 468 }
 469
 470 static void xreq_subcall_cb(struct afb_request *closure, const char *api, const char *verb, struct json_object *args, void (*callback)(void*, int, struct json_object*), void *cb_closure)
 471 {
 472         struct afb_xreq *xreq = from_request(closure);
 473         struct subcall *sc;
 474
 475         sc = subcall_alloc(xreq, api, verb, args);
 476         if (sc == NULL) {
 477                 if (callback)
 478                         callback(cb_closure, 1, afb_msg_json_internal_error());
 479                 json_object_put(args);
 480         } else {
 481                 subcall(sc, callback, cb_closure);
 482         }
 483 }
 484
 485 static void xreq_subcall_req_cb(struct afb_request *closure, const char *api, const char *verb, struct json_object *args, void (*callback)(void*, int, struct json_object*, struct afb_req), void *cb_closure)
 486 {
 487         struct afb_xreq *xreq = from_request(closure);
 488         struct subcall *sc;
 489
 490         sc = subcall_alloc(xreq, api, verb, args);
 491         if (sc == NULL) {
 492                 if (callback)
 493                         callback(cb_closure, 1, afb_msg_json_internal_error(), to_req(xreq));
 494                 json_object_put(args);
 495         } else {
 496                 subcall_req(sc, callback, cb_closure);
 497         }
 498 }
 499
 500
 501 static int xreq_subcallsync_cb(struct afb_request *closure, const char *api, const char *verb, struct json_object *args, struct json_object **result)
 502 {
 503         int rc;
 504         struct subcall *sc;
 505         struct afb_xreq *xreq = from_request(closure);
 506         struct json_object *resu;
 507
 508         sc = subcall_alloc(xreq, api, verb, args);
 509         if (!sc) {
 510                 rc = -1;
 511                 resu = afb_msg_json_internal_error();
 512                 json_object_put(args);
 513         } else {
 514                 rc = subcallsync(sc, &resu);
 515         }
 516         if (result)
 517                 *result = resu;
 518         else
 519                 json_object_put(resu);
 520         return rc;
 521 }
 522
 523 static void xreq_vverbose_cb(struct afb_request *closure, int level, const char *file, int line, const char *func, const char *fmt, va_list args)
 524 {
 525         char *p;
 526         struct afb_xreq *xreq = from_request(closure);
 527
 528         if (!fmt || vasprintf(&p, fmt, args) < 0)
 529                 vverbose(level, file, line, func, fmt, args);
 530         else {
 531                 verbose(level, file, line, func, "[REQ/API %s] %s", xreq->api, p);
 532                 free(p);
 533         }
 534 }
 535
 536 static struct afb_stored_req *xreq_store_cb(struct afb_request *closure)
 537 {
 538         xreq_addref_cb(closure);
 539         return (struct afb_stored_req*)closure;
 540 }
 541
 542 static int xreq_has_permission_cb(struct afb_request *closure, const char *permission)
 543 {
 544         struct afb_xreq *xreq = from_request(closure);
 545         return afb_auth_has_permission(xreq, permission);
 546 }
 547
 548 static char *xreq_get_application_id_cb(struct afb_request *closure)
 549 {
 550         struct afb_xreq *xreq = from_request(closure);
 551         return xreq->cred && xreq->cred->id ? strdup(xreq->cred->id) : NULL;
 552 }
 553
 554 static void *xreq_context_make_cb(struct afb_request *closure, int replace, void *(*create_value)(void*), void (*free_value)(void*), void *create_closure)
 555 {
 556         struct afb_xreq *xreq = from_request(closure);
 557         return afb_context_make(&xreq->context, replace, create_value, free_value, create_closure);
 558 }
 559
 560 /******************************************************************************/
 561
 562 static struct json_object *xreq_hooked_json_cb(struct afb_request *closure)
 563 {
 564         struct json_object *r = xreq_json_cb(closure);
 565         struct afb_xreq *xreq = from_request(closure);
 566         return afb_hook_xreq_json(xreq, r);
 567 }
 568
 569 static struct afb_arg xreq_hooked_get_cb(struct afb_request *closure, const char *name)
 570 {
 571         struct afb_arg r = xreq_get_cb(closure, name);
 572         struct afb_xreq *xreq = from_request(closure);
 573         return afb_hook_xreq_get(xreq, name, r);
 574 }
 575
 576 static void xreq_hooked_success_cb(struct afb_request *closure, struct json_object *obj, const char *info)
 577 {
 578         struct afb_xreq *xreq = from_request(closure);
 579         afb_hook_xreq_success(xreq, obj, info);
 580         xreq_success_cb(closure, obj, info);
 581 }
 582
 583 static void xreq_hooked_fail_cb(struct afb_request *closure, const char *status, const char *info)
 584 {
 585         struct afb_xreq *xreq = from_request(closure);
 586         afb_hook_xreq_fail(xreq, status, info);
 587         xreq_fail_cb(closure, status, info);
 588 }
 589
 590 static void xreq_hooked_vsuccess_cb(struct afb_request *closure, struct json_object *obj, const char *fmt, va_list args)
 591 {
 592         vinfo(closure, obj, fmt, args, (void*)xreq_hooked_success_cb);
 593 }
 594
 595 static void xreq_hooked_vfail_cb(struct afb_request *closure, const char *status, const char *fmt, va_list args)
 596 {
 597         vinfo(closure, (void*)status, fmt, args, (void*)xreq_hooked_fail_cb);
 598 }
 599
 600 static void *xreq_hooked_context_get_cb(struct afb_request *closure)
 601 {
 602         void *r = xreq_context_get_cb(closure);
 603         struct afb_xreq *xreq = from_request(closure);
 604         return afb_hook_xreq_context_get(xreq, r);
 605 }
 606
 607 static void xreq_hooked_context_set_cb(struct afb_request *closure, void *value, void (*free_value)(void*))
 608 {
 609         struct afb_xreq *xreq = from_request(closure);
 610         afb_hook_xreq_context_set(xreq, value, free_value);
 611         xreq_context_set_cb(closure, value, free_value);
 612 }
 613
 614 static struct afb_request *xreq_hooked_addref_cb(struct afb_request *closure)
 615 {
 616         struct afb_xreq *xreq = from_request(closure);
 617         afb_hook_xreq_addref(xreq);
 618         return xreq_addref_cb(closure);
 619 }
 620
 621 static void xreq_hooked_unref_cb(struct afb_request *closure)
 622 {
 623         struct afb_xreq *xreq = from_request(closure);
 624         afb_hook_xreq_unref(xreq);
 625         xreq_unref_cb(closure);
 626 }
 627
 628 static void xreq_hooked_session_close_cb(struct afb_request *closure)
 629 {
 630         struct afb_xreq *xreq = from_request(closure);
 631         afb_hook_xreq_session_close(xreq);
 632         xreq_session_close_cb(closure);
 633 }
 634
 635 static int xreq_hooked_session_set_LOA_cb(struct afb_request *closure, unsigned level)
 636 {
 637         int r = xreq_session_set_LOA_cb(closure, level);
 638         struct afb_xreq *xreq = from_request(closure);
 639         return afb_hook_xreq_session_set_LOA(xreq, level, r);
 640 }
 641
 642 static int xreq_hooked_subscribe_eventid_cb(struct afb_request *closure, struct afb_eventid *eventid);
 643 static int xreq_hooked_subscribe_cb(struct afb_request *closure, struct afb_event event)
 644 {
 645         return xreq_hooked_subscribe_eventid_cb(closure, event.closure);
 646 }
 647
 648 static int xreq_hooked_subscribe_eventid_cb(struct afb_request *closure, struct afb_eventid *eventid)
 649 {
 650         int r = xreq_subscribe_eventid_cb(closure, eventid);
 651         struct afb_xreq *xreq = from_request(closure);
 652         return afb_hook_xreq_subscribe(xreq, eventid, r);
 653 }
 654
 655 static int xreq_hooked_unsubscribe_eventid_cb(struct afb_request *closure, struct afb_eventid *eventid);
 656 static int xreq_hooked_unsubscribe_cb(struct afb_request *closure, struct afb_event event)
 657 {
 658         return xreq_hooked_unsubscribe_eventid_cb(closure, event.closure);
 659 }
 660
 661 static int xreq_hooked_unsubscribe_eventid_cb(struct afb_request *closure, struct afb_eventid *eventid)
 662 {
 663         int r = xreq_unsubscribe_eventid_cb(closure, eventid);
 664         struct afb_xreq *xreq = from_request(closure);
 665         return afb_hook_xreq_unsubscribe(xreq, eventid, r);
 666 }
 667
 668 static void xreq_hooked_subcall_cb(struct afb_request *closure, const char *api, const char *verb, struct json_object *args, void (*callback)(void*, int, struct json_object*), void *cb_closure)
 669 {
 670         struct afb_xreq *xreq = from_request(closure);
 671         struct subcall *sc;
 672
 673         afb_hook_xreq_subcall(xreq, api, verb, args);
 674         sc = subcall_alloc(xreq, api, verb, args);
 675         if (sc == NULL) {
 676                 if (callback)
 677                         callback(cb_closure, 1, afb_msg_json_internal_error());
 678                 json_object_put(args);
 679         } else {
 680                 subcall_hooked(sc, callback, cb_closure);
 681         }
 682 }
 683
 684 static void xreq_hooked_subcall_req_cb(struct afb_request *closure, const char *api, const char *verb, struct json_object *args, void (*callback)(void*, int, struct json_object*, struct afb_req), void *cb_closure)
 685 {
 686         struct afb_xreq *xreq = from_request(closure);
 687         struct subcall *sc;
 688
 689         afb_hook_xreq_subcall_req(xreq, api, verb, args);
 690         sc = subcall_alloc(xreq, api, verb, args);
 691         if (sc == NULL) {
 692                 if (callback)
 693                         callback(cb_closure, 1, afb_msg_json_internal_error(), to_req(xreq));
 694                 json_object_put(args);
 695         } else {
 696                 subcall_req_hooked(sc, callback, cb_closure);
 697         }
 698 }
 699
 700 static int xreq_hooked_subcallsync_cb(struct afb_request *closure, const char *api, const char *verb, struct json_object *args, struct json_object **result)
 701 {
 702         int r;
 703         struct afb_xreq *xreq = from_request(closure);
 704         afb_hook_xreq_subcallsync(xreq, api, verb, args);
 705         r = xreq_subcallsync_cb(closure, api, verb, args, result);
 706         return afb_hook_xreq_subcallsync_result(xreq, r, *result);
 707 }
 708
 709 static void xreq_hooked_vverbose_cb(struct afb_request *closure, int level, const char *file, int line, const char *func, const char *fmt, va_list args)
 710 {
 711         struct afb_xreq *xreq = from_request(closure);
 712         va_list ap;
 713         va_copy(ap, args);
 714         xreq_vverbose_cb(closure, level, file, line, func, fmt, args);
 715         afb_hook_xreq_vverbose(xreq, level, file, line, func, fmt, ap);
 716         va_end(ap);
 717 }
 718
 719 static struct afb_stored_req *xreq_hooked_store_cb(struct afb_request *closure)
 720 {
 721         struct afb_xreq *xreq = from_request(closure);
 722         struct afb_stored_req *r = xreq_store_cb(closure);
 723         afb_hook_xreq_store(xreq, r);
 724         return r;
 725 }
 726
 727 static int xreq_hooked_has_permission_cb(struct afb_request *closure, const char *permission)
 728 {
 729         struct afb_xreq *xreq = from_request(closure);
 730         int r = xreq_has_permission_cb(closure, permission);
 731         return afb_hook_xreq_has_permission(xreq, permission, r);
 732 }
 733
 734 static char *xreq_hooked_get_application_id_cb(struct afb_request *closure)
 735 {
 736         struct afb_xreq *xreq = from_request(closure);
 737         char *r = xreq_get_application_id_cb(closure);
 738         return afb_hook_xreq_get_application_id(xreq, r);
 739 }
 740
 741 static void *xreq_hooked_context_make_cb(struct afb_request *closure, int replace, void *(*create_value)(void*), void (*free_value)(void*), void *create_closure)
 742 {
 743         struct afb_xreq *xreq = from_request(closure);
 744         void *result = xreq_context_make_cb(closure, replace, create_value, free_value, create_closure);
 745         return afb_hook_xreq_context_make(xreq, replace, create_value, free_value, create_closure, result);
 746 }
 747
 748 /******************************************************************************/
 749
 750 const struct afb_request_itf xreq_itf = {
 751         .json = xreq_json_cb,
 752         .get = xreq_get_cb,
 753         .success = xreq_success_cb,
 754         .fail = xreq_fail_cb,
 755         .vsuccess = xreq_vsuccess_cb,
 756         .vfail = xreq_vfail_cb,
 757         .context_get = xreq_context_get_cb,
 758         .context_set = xreq_context_set_cb,
 759         .addref = xreq_addref_cb,
 760         .unref = xreq_unref_cb,
 761         .session_close = xreq_session_close_cb,
 762         .session_set_LOA = xreq_session_set_LOA_cb,
 763         .subscribe = xreq_subscribe_cb,
 764         .unsubscribe = xreq_unsubscribe_cb,
 765         .subcall = xreq_subcall_cb,
 766         .subcallsync = xreq_subcallsync_cb,
 767         .vverbose = xreq_vverbose_cb,
 768         .store = xreq_store_cb,
 769         .subcall_req = xreq_subcall_req_cb,
 770         .has_permission = xreq_has_permission_cb,
 771         .get_application_id = xreq_get_application_id_cb,
 772         .context_make = xreq_context_make_cb,
 773         .subscribe_eventid = xreq_subscribe_eventid_cb,
 774         .unsubscribe_eventid = xreq_unsubscribe_eventid_cb,
 775 };
 776
 777 const struct afb_request_itf xreq_hooked_itf = {
 778         .json = xreq_hooked_json_cb,
 779         .get = xreq_hooked_get_cb,
 780         .success = xreq_hooked_success_cb,
 781         .fail = xreq_hooked_fail_cb,
 782         .vsuccess = xreq_hooked_vsuccess_cb,
 783         .vfail = xreq_hooked_vfail_cb,
 784         .context_get = xreq_hooked_context_get_cb,
 785         .context_set = xreq_hooked_context_set_cb,
 786         .addref = xreq_hooked_addref_cb,
 787         .unref = xreq_hooked_unref_cb,
 788         .session_close = xreq_hooked_session_close_cb,
 789         .session_set_LOA = xreq_hooked_session_set_LOA_cb,
 790         .subscribe = xreq_hooked_subscribe_cb,
 791         .unsubscribe = xreq_hooked_unsubscribe_cb,
 792         .subcall = xreq_hooked_subcall_cb,
 793         .subcallsync = xreq_hooked_subcallsync_cb,
 794         .vverbose = xreq_hooked_vverbose_cb,
 795         .store = xreq_hooked_store_cb,
 796         .subcall_req = xreq_hooked_subcall_req_cb,
 797         .has_permission = xreq_hooked_has_permission_cb,
 798         .get_application_id = xreq_hooked_get_application_id_cb,
 799         .context_make = xreq_hooked_context_make_cb,
 800         .subscribe_eventid = xreq_hooked_subscribe_eventid_cb,
 801         .unsubscribe_eventid = xreq_hooked_unsubscribe_eventid_cb,
 802 };
 803
 804 /******************************************************************************/
 805
 806 struct afb_req afb_xreq_unstore(struct afb_stored_req *sreq)
 807 {
 808         struct afb_xreq *xreq = (struct afb_xreq *)sreq;
 809         if (xreq->hookflags)
 810                 afb_hook_xreq_unstore(xreq);
 811         return to_req(xreq);
 812 }
 813
 814 struct json_object *afb_xreq_json(struct afb_xreq *xreq)
 815 {
 816         return afb_request_json(to_request(xreq));
 817 }
 818
 819 void afb_xreq_success(struct afb_xreq *xreq, struct json_object *obj, const char *info)
 820 {
 821         afb_request_success(to_request(xreq), obj, info);
 822 }
 823
 824 void afb_xreq_success_f(struct afb_xreq *xreq, struct json_object *obj, const char *info, ...)
 825 {
 826         va_list args;
 827
 828         va_start(args, info);
 829         afb_request_success_v(to_request(xreq), obj, info, args);
 830         va_end(args);
 831 }
 832
 833 void afb_xreq_fail(struct afb_xreq *xreq, const char *status, const char *info)
 834 {
 835         afb_request_fail(to_request(xreq), status, info);
 836 }
 837
 838 void afb_xreq_fail_f(struct afb_xreq *xreq, const char *status, const char *info, ...)
 839 {
 840         va_list args;
 841
 842         va_start(args, info);
 843         afb_request_fail_v(to_request(xreq), status, info, args);
 844         va_end(args);
 845
 846 }
 847
 848 const char *afb_xreq_raw(struct afb_xreq *xreq, size_t *size)
 849 {
 850         struct json_object *obj = xreq_json_cb(to_request(xreq));
 851         const char *result = json_object_to_json_string(obj);
 852         if (size != NULL)
 853                 *size = strlen(result);
 854         return result;
 855 }
 856
 857 void afb_xreq_addref(struct afb_xreq *xreq)
 858 {
 859         afb_request_addref(to_request(xreq));
 860 }
 861
 862 void afb_xreq_unref(struct afb_xreq *xreq)
 863 {
 864         afb_request_unref(to_request(xreq));
 865 }
 866
 867 void afb_xreq_unhooked_subcall(struct afb_xreq *xreq, const char *api, const char *verb, struct json_object *args, void (*callback)(void*, int, struct json_object*), void *cb_closure)
 868 {
 869         xreq_subcall_cb(to_request(xreq), api, verb, args, callback, cb_closure);
 870 }
 871
 872 void afb_xreq_subcall(struct afb_xreq *xreq, const char *api, const char *verb, struct json_object *args, void (*callback)(void*, int, struct json_object*), void *cb_closure)
 873 {
 874         afb_request_subcall(to_request(xreq), api, verb, args, callback, cb_closure);
 875 }
 876
 877 int afb_xreq_unhooked_subcall_sync(struct afb_xreq *xreq, const char *api, const char *verb, struct json_object *args, struct json_object **result)
 878 {
 879         return xreq_subcallsync_cb(to_request(xreq), api, verb, args, result);
 880 }
 881
 882 int afb_xreq_subcall_sync(struct afb_xreq *xreq, const char *api, const char *verb, struct json_object *args, struct json_object **result)
 883 {
 884         return afb_request_subcall_sync(to_request(xreq), api, verb, args, result);
 885 }
 886
 887 static int xreq_session_check_apply_v1(struct afb_xreq *xreq, int sessionflags)
 888 {
 889         int loa;
 890
 891         if ((sessionflags & (AFB_SESSION_CLOSE_V1|AFB_SESSION_RENEW_V1|AFB_SESSION_CHECK_V1|AFB_SESSION_LOA_EQ_V1)) != 0) {
 892                 if (!afb_context_check(&xreq->context)) {
 893                         afb_context_close(&xreq->context);
 894                         afb_xreq_fail_f(xreq, "denied", "invalid token's identity");
 895                         errno = EINVAL;
 896                         return -1;
 897                 }
 898         }
 899
 900         if ((sessionflags & AFB_SESSION_LOA_GE_V1) != 0) {
 901                 loa = (sessionflags >> AFB_SESSION_LOA_SHIFT_V1) & AFB_SESSION_LOA_MASK_V1;
 902                 if (!afb_context_check_loa(&xreq->context, loa)) {
 903                         afb_xreq_fail_f(xreq, "denied", "invalid LOA");
 904                         errno = EPERM;
 905                         return -1;
 906                 }
 907         }
 908
 909         if ((sessionflags & AFB_SESSION_LOA_LE_V1) != 0) {
 910                 loa = (sessionflags >> AFB_SESSION_LOA_SHIFT_V1) & AFB_SESSION_LOA_MASK_V1;
 911                 if (afb_context_check_loa(&xreq->context, loa + 1)) {
 912                         afb_xreq_fail_f(xreq, "denied", "invalid LOA");
 913                         errno = EPERM;
 914                         return -1;
 915                 }
 916         }
 917
 918         if ((sessionflags & AFB_SESSION_RENEW_V1) != 0) {
 919                 afb_context_refresh(&xreq->context);
 920         }
 921         if ((sessionflags & AFB_SESSION_CLOSE_V1) != 0) {
 922                 afb_context_change_loa(&xreq->context, 0);
 923                 afb_context_close(&xreq->context);
 924         }
 925
 926         return 0;
 927 }
 928
 929 static int xreq_session_check_apply_v2(struct afb_xreq *xreq, uint32_t sessionflags, const struct afb_auth *auth)
 930 {
 931         int loa;
 932
 933         if (sessionflags != 0) {
 934                 if (!afb_context_check(&xreq->context)) {
 935                         afb_context_close(&xreq->context);
 936                         afb_xreq_fail_f(xreq, "denied", "invalid token's identity");
 937                         errno = EINVAL;
 938                         return -1;
 939                 }
 940         }
 941
 942         loa = (int)(sessionflags & AFB_SESSION_LOA_MASK_V2);
 943         if (loa && !afb_context_check_loa(&xreq->context, loa)) {
 944                 afb_xreq_fail_f(xreq, "denied", "invalid LOA");
 945                 errno = EPERM;
 946                 return -1;
 947         }
 948
 949         if (auth && !afb_auth_check(xreq, auth)) {
 950                 afb_xreq_fail_f(xreq, "denied", "authorisation refused");
 951                 errno = EPERM;
 952                 return -1;
 953         }
 954
 955         if ((sessionflags & AFB_SESSION_REFRESH_V2) != 0) {
 956                 afb_context_refresh(&xreq->context);
 957         }
 958         if ((sessionflags & AFB_SESSION_CLOSE_V2) != 0) {
 959                 afb_context_close(&xreq->context);
 960         }
 961
 962         return 0;
 963 }
 964
 965 void afb_xreq_call_verb_v1(struct afb_xreq *xreq, const struct afb_verb_desc_v1 *verb)
 966 {
 967         if (!verb)
 968                 afb_xreq_fail_unknown_verb(xreq);
 969         else
 970                 if (!xreq_session_check_apply_v1(xreq, verb->session))
 971                         verb->callback(to_req(xreq));
 972 }
 973
 974 void afb_xreq_call_verb_v2(struct afb_xreq *xreq, const struct afb_verb_v2 *verb)
 975 {
 976         if (!verb)
 977                 afb_xreq_fail_unknown_verb(xreq);
 978         else
 979                 if (!xreq_session_check_apply_v2(xreq, verb->session, verb->auth))
 980                         verb->callback(to_req(xreq));
 981 }
 982
 983 void afb_xreq_init(struct afb_xreq *xreq, const struct afb_xreq_query_itf *queryitf)
 984 {
 985         memset(xreq, 0, sizeof *xreq);
 986         xreq->request.itf = &xreq_hooked_itf; /* hook by default */
 987         xreq->refcount = 1;
 988         xreq->queryitf = queryitf;
 989 }
 990
 991 void afb_xreq_fail_unknown_api(struct afb_xreq *xreq)
 992 {
 993         afb_xreq_fail_f(xreq, "unknown-api", "api %s not found (for verb %s)", xreq->api, xreq->verb);
 994 }
 995
 996 void afb_xreq_fail_unknown_verb(struct afb_xreq *xreq)
 997 {
 998         afb_xreq_fail_f(xreq, "unknown-verb", "verb %s unknown within api %s", xreq->verb, xreq->api);
 999 }
1000
1001 static void init_hooking(struct afb_xreq *xreq)
1002 {
1003         afb_hook_init_xreq(xreq);
1004         if (xreq->hookflags)
1005                 afb_hook_xreq_begin(xreq);
1006         else
1007                 xreq->request.itf = &xreq_itf; /* unhook the interface */
1008 }
1009
1010 /**
1011  * job callback for asynchronous and secured processing of the request.
1012  */
1013 static void process_async(int signum, void *arg)
1014 {
1015         struct afb_xreq *xreq = arg;
1016         const struct afb_api *api;
1017
1018         if (signum != 0) {
1019                 /* emit the error (assumes that hooking is initialised) */
1020                 afb_xreq_fail_f(xreq, "aborted", "signal %s(%d) caught", strsignal(signum), signum);
1021         } else {
1022                 /* init hooking */
1023                 init_hooking(xreq);
1024                 /* invoke api call method to process the reqiest */
1025                 api = (const struct afb_api*)xreq->context.api_key;
1026                 api->itf->call(api->closure, xreq);
1027         }
1028         /* release the request */
1029         afb_xreq_unhooked_unref(xreq);
1030 }
1031
1032 /**
1033  * Early request failure of the request 'xreq' with, as usual, 'status' and 'info'
1034  * The early failure occurs only in function 'afb_xreq_process' where normally,
1035  * the hooking is not initialised. So this "early" failure takes care to initialise
1036  * the hooking in first.
1037  */
1038 static void early_failure(struct afb_xreq *xreq, const char *status, const char *info, ...)
1039 {
1040         va_list args;
1041
1042         /* init hooking */
1043         init_hooking(xreq);
1044
1045         /* send error */
1046         va_start(args, info);
1047         afb_request_fail_v(to_request(xreq), status, info, args);
1048         va_end(args);
1049 }
1050
1051 /**
1052  * Enqueue a job for processing the request 'xreq' using the given 'apiset'.
1053  * Errors are reported as request failures.
1054  */
1055 void afb_xreq_process(struct afb_xreq *xreq, struct afb_apiset *apiset)
1056 {
1057         const struct afb_api *api;
1058         struct afb_xreq *caller;
1059
1060         /* lookup at the api */
1061         xreq->apiset = apiset;
1062         api = afb_apiset_lookup_started(apiset, xreq->api, 1);
1063         if (!api) {
1064                 if (errno == ENOENT)
1065                         early_failure(xreq, "unknown-api", "api %s not found (for verb %s)", xreq->api, xreq->verb);
1066                 else
1067                         early_failure(xreq, "bad-api-state", "api %s not started correctly: %m", xreq->api);
1068                 goto end;
1069         }
1070         xreq->context.api_key = api;
1071
1072         /* check self locking */
1073         if (api->group) {
1074                 caller = xreq->caller;
1075                 while (caller) {
1076                         if (((const struct afb_api *)caller->context.api_key)->group == api->group) {
1077                                 /* noconcurrency lock detected */
1078                                 ERROR("self-lock detected in call stack for API %s", xreq->api);
1079                                 early_failure(xreq, "self-locked", "recursive self lock, API %s", xreq->api);
1080                                 goto end;
1081                         }
1082                         caller = caller->caller;
1083                 }
1084         }
1085
1086         /* queue the request job */
1087         afb_xreq_unhooked_addref(xreq);
1088         if (jobs_queue(api->group, afb_apiset_timeout_get(apiset), process_async, xreq) < 0) {
1089                 /* TODO: allows or not to proccess it directly as when no threading? (see above) */
1090                 ERROR("can't process job with threads: %m");
1091                 early_failure(xreq, "cancelled", "not able to create a job for the task");
1092                 afb_xreq_unhooked_unref(xreq);
1093         }
1094 end:
1095         afb_xreq_unhooked_unref(xreq);
1096 }
1097