src/afb-xreq.c

   1 /*
   2  * Copyright (C) 2017-2019 "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 #include <stdarg.h>
  25
  26 #include <json-c/json.h>
  27 #if !defined(JSON_C_TO_STRING_NOSLASHESCAPE)
  28 #define JSON_C_TO_STRING_NOSLASHESCAPE 0
  29 #endif
  30
  31 #include <afb/afb-binding-v1.h>
  32 #include <afb/afb-binding-v2.h>
  33 #include <afb/afb-binding-v3.h>
  34 #include <afb/afb-req-x2.h>
  35
  36 #include "afb-api.h"
  37 #include "afb-apiset.h"
  38 #include "afb-auth.h"
  39 #include "afb-calls.h"
  40 #include "afb-context.h"
  41 #include "afb-evt.h"
  42 #include "afb-cred.h"
  43 #include "afb-hook.h"
  44 #include "afb-msg-json.h"
  45 #include "afb-xreq.h"
  46 #include "afb-error-text.h"
  47
  48 #include "jobs.h"
  49 #include "verbose.h"
  50
  51 /******************************************************************************/
  52
  53 static void xreq_finalize(struct afb_xreq *xreq)
  54 {
  55         if (!xreq->replied)
  56                 afb_xreq_reply(xreq, NULL, afb_error_text_not_replied, NULL);
  57 #if WITH_AFB_HOOK
  58         if (xreq->hookflags)
  59                 afb_hook_xreq_end(xreq);
  60 #endif
  61         if (xreq->caller)
  62                 afb_xreq_unhooked_unref(xreq->caller);
  63         xreq->queryitf->unref(xreq);
  64 }
  65
  66 inline void afb_xreq_unhooked_addref(struct afb_xreq *xreq)
  67 {
  68         __atomic_add_fetch(&xreq->refcount, 1, __ATOMIC_RELAXED);
  69 }
  70
  71 inline void afb_xreq_unhooked_unref(struct afb_xreq *xreq)
  72 {
  73         if (!__atomic_sub_fetch(&xreq->refcount, 1, __ATOMIC_RELAXED))
  74                 xreq_finalize(xreq);
  75 }
  76
  77 /******************************************************************************/
  78
  79 struct json_object *afb_xreq_unhooked_json(struct afb_xreq *xreq)
  80 {
  81         if (!xreq->json && xreq->queryitf->json)
  82                 xreq->json = xreq->queryitf->json(xreq);
  83         return xreq->json;
  84 }
  85
  86 static struct json_object *xreq_json_cb(struct afb_req_x2 *closure)
  87 {
  88         struct afb_xreq *xreq = xreq_from_req_x2(closure);
  89         return afb_xreq_unhooked_json(xreq);
  90 }
  91
  92 static struct afb_arg xreq_get_cb(struct afb_req_x2 *closure, const char *name)
  93 {
  94         struct afb_xreq *xreq = xreq_from_req_x2(closure);
  95         struct afb_arg arg;
  96         struct json_object *object, *value;
  97
  98         if (xreq->queryitf->get)
  99                 arg = xreq->queryitf->get(xreq, name);
 100         else {
 101                 object = xreq_json_cb(closure);
 102                 if (json_object_object_get_ex(object, name, &value)) {
 103                         arg.name = name;
 104                         arg.value = json_object_get_string(value);
 105                 } else {
 106                         arg.name = NULL;
 107                         arg.value = NULL;
 108                 }
 109                 arg.path = NULL;
 110         }
 111         return arg;
 112 }
 113
 114 static void xreq_reply_cb(struct afb_req_x2 *closure, struct json_object *obj, const char *error, const char *info)
 115 {
 116         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 117
 118         if (xreq->replied) {
 119                 ERROR("reply called more than one time!!");
 120                 json_object_put(obj);
 121         } else {
 122                 xreq->replied = 1;
 123                 xreq->queryitf->reply(xreq, obj, error, info);
 124         }
 125 }
 126
 127 static void xreq_vreply_cb(struct afb_req_x2 *closure, struct json_object *obj, const char *error, const char *fmt, va_list args)
 128 {
 129         char *info;
 130         if (fmt == NULL || vasprintf(&info, fmt, args) < 0)
 131                 info = NULL;
 132         xreq_reply_cb(closure, obj, error, info);
 133         free(info);
 134 }
 135
 136 static void xreq_legacy_success_cb(struct afb_req_x2 *closure, struct json_object *obj, const char *info)
 137 {
 138         xreq_reply_cb(closure, obj, NULL, info);
 139 }
 140
 141 static void xreq_legacy_fail_cb(struct afb_req_x2 *closure, const char *status, const char *info)
 142 {
 143         xreq_reply_cb(closure, NULL, status, info);
 144 }
 145
 146 static void xreq_legacy_vsuccess_cb(struct afb_req_x2 *closure, struct json_object *obj, const char *fmt, va_list args)
 147 {
 148         xreq_vreply_cb(closure, obj, NULL, fmt, args);
 149 }
 150
 151 static void xreq_legacy_vfail_cb(struct afb_req_x2 *closure, const char *status, const char *fmt, va_list args)
 152 {
 153         xreq_vreply_cb(closure, NULL, status, fmt, args);
 154 }
 155
 156 static void *xreq_legacy_context_get_cb(struct afb_req_x2 *closure)
 157 {
 158         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 159         return afb_context_get(&xreq->context);
 160 }
 161
 162 static void xreq_legacy_context_set_cb(struct afb_req_x2 *closure, void *value, void (*free_value)(void*))
 163 {
 164         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 165         afb_context_set(&xreq->context, value, free_value);
 166 }
 167
 168 static struct afb_req_x2 *xreq_addref_cb(struct afb_req_x2 *closure)
 169 {
 170         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 171         afb_xreq_unhooked_addref(xreq);
 172         return closure;
 173 }
 174
 175 static void xreq_unref_cb(struct afb_req_x2 *closure)
 176 {
 177         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 178         afb_xreq_unhooked_unref(xreq);
 179 }
 180
 181 static void xreq_session_close_cb(struct afb_req_x2 *closure)
 182 {
 183         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 184         afb_context_close(&xreq->context);
 185 }
 186
 187 static int xreq_session_set_LOA_cb(struct afb_req_x2 *closure, unsigned level)
 188 {
 189         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 190         return afb_context_change_loa(&xreq->context, level);
 191 }
 192
 193 static int xreq_subscribe_event_x2_cb(struct afb_req_x2 *closure, struct afb_event_x2 *event)
 194 {
 195         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 196         return afb_xreq_subscribe(xreq, event);
 197 }
 198
 199 static int xreq_legacy_subscribe_event_x1_cb(struct afb_req_x2 *closure, struct afb_event_x1 event)
 200 {
 201         return xreq_subscribe_event_x2_cb(closure, afb_event_x1_to_event_x2(event));
 202 }
 203
 204 int afb_xreq_subscribe(struct afb_xreq *xreq, struct afb_event_x2 *event)
 205 {
 206         if (xreq->replied) {
 207                 ERROR("request replied, subscription impossible");
 208                 errno = EINVAL;
 209         } else {
 210                 if (xreq->queryitf->subscribe)
 211                         return xreq->queryitf->subscribe(xreq, event);
 212                 ERROR("no event listener, subscription impossible");
 213                 errno = ENOTSUP;
 214         }
 215         return -1;
 216 }
 217
 218 static int xreq_unsubscribe_event_x2_cb(struct afb_req_x2 *closure, struct afb_event_x2 *event)
 219 {
 220         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 221         return afb_xreq_unsubscribe(xreq, event);
 222 }
 223
 224 static int xreq_legacy_unsubscribe_event_x1_cb(struct afb_req_x2 *closure, struct afb_event_x1 event)
 225 {
 226         return xreq_unsubscribe_event_x2_cb(closure, afb_event_x1_to_event_x2(event));
 227 }
 228
 229 int afb_xreq_unsubscribe(struct afb_xreq *xreq, struct afb_event_x2 *event)
 230 {
 231         if (xreq->replied) {
 232                 ERROR("request replied, unsubscription impossible");
 233                 errno = EINVAL;
 234         } else {
 235                 if (xreq->queryitf->unsubscribe)
 236                         return xreq->queryitf->unsubscribe(xreq, event);
 237                 ERROR("no event listener, unsubscription impossible");
 238                 errno = ENOTSUP;
 239         }
 240         return -1;
 241 }
 242
 243 static void xreq_legacy_subcall_cb(struct afb_req_x2 *req, const char *api, const char *verb, struct json_object *args, void (*callback)(void*, int, struct json_object*), void *closure)
 244 {
 245         struct afb_xreq *xreq = xreq_from_req_x2(req);
 246         return afb_calls_legacy_subcall_v1(xreq, api, verb, args, callback, closure);
 247 }
 248
 249 static void xreq_legacy_subcall_req_cb(struct afb_req_x2 *req, const char *api, const char *verb, struct json_object *args, void (*callback)(void*, int, struct json_object*, struct afb_req_x1), void *closure)
 250 {
 251         struct afb_xreq *xreq = xreq_from_req_x2(req);
 252         return afb_calls_legacy_subcall_v2(xreq, api, verb, args, callback, closure);
 253 }
 254
 255 static void xreq_legacy_subcall_request_cb(struct afb_req_x2 *req, const char *api, const char *verb, struct json_object *args, void (*callback)(void*, int, struct json_object*, struct afb_req_x2*), void *closure)
 256 {
 257         struct afb_xreq *xreq = xreq_from_req_x2(req);
 258         return afb_calls_legacy_subcall_v3(xreq, api, verb, args, callback, closure);
 259 }
 260
 261
 262 static int xreq_legacy_subcallsync_cb(struct afb_req_x2 *req, const char *api, const char *verb, struct json_object *args, struct json_object **result)
 263 {
 264         struct afb_xreq *xreq = xreq_from_req_x2(req);
 265         return afb_calls_legacy_subcall_sync(xreq, api, verb, args, result);
 266 }
 267
 268 static void xreq_vverbose_cb(struct afb_req_x2 *closure, int level, const char *file, int line, const char *func, const char *fmt, va_list args)
 269 {
 270         char *p;
 271         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 272
 273         if (!fmt || vasprintf(&p, fmt, args) < 0)
 274                 vverbose(level, file, line, func, fmt, args);
 275         else {
 276                 verbose(level, file, line, func, "[REQ/API %s] %s", xreq->request.called_api, p);
 277                 free(p);
 278         }
 279 }
 280
 281 static struct afb_stored_req *xreq_legacy_store_cb(struct afb_req_x2 *closure)
 282 {
 283         xreq_addref_cb(closure);
 284         return (struct afb_stored_req*)closure;
 285 }
 286
 287 static int xreq_has_permission_cb(struct afb_req_x2 *closure, const char *permission)
 288 {
 289         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 290         return afb_context_has_permission(&xreq->context, permission);
 291 }
 292
 293 static char *xreq_get_application_id_cb(struct afb_req_x2 *closure)
 294 {
 295         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 296         struct afb_cred *cred = xreq->context.credentials;
 297         return cred && cred->id ? strdup(cred->id) : NULL;
 298 }
 299
 300 static void *xreq_context_make_cb(struct afb_req_x2 *closure, int replace, void *(*create_value)(void*), void (*free_value)(void*), void *create_closure)
 301 {
 302         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 303         return afb_context_make(&xreq->context, replace, create_value, free_value, create_closure);
 304 }
 305
 306 static int xreq_get_uid_cb(struct afb_req_x2 *closure)
 307 {
 308         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 309         struct afb_cred *cred = xreq->context.credentials;
 310         return cred && cred->id ? (int)cred->uid : -1;
 311 }
 312
 313 static struct json_object *xreq_get_client_info_cb(struct afb_req_x2 *closure)
 314 {
 315         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 316         struct afb_cred *cred = xreq->context.credentials;
 317         struct json_object *r = json_object_new_object();
 318         if (cred && cred->id) {
 319                 json_object_object_add(r, "uid", json_object_new_int(cred->uid));
 320                 json_object_object_add(r, "gid", json_object_new_int(cred->gid));
 321                 json_object_object_add(r, "pid", json_object_new_int(cred->pid));
 322                 json_object_object_add(r, "user", json_object_new_string(cred->user));
 323                 json_object_object_add(r, "label", json_object_new_string(cred->label));
 324                 json_object_object_add(r, "id", json_object_new_string(cred->id));
 325         }
 326         if (xreq->context.session) {
 327                 json_object_object_add(r, "uuid", json_object_new_string(afb_context_uuid(&xreq->context)?:""));
 328                 json_object_object_add(r, "LOA", json_object_new_int(afb_context_get_loa(&xreq->context)));
 329         }
 330         return r;
 331 }
 332
 333 static void xreq_subcall_cb(
 334                                 struct afb_req_x2 *req,
 335                                 const char *api,
 336                                 const char *verb,
 337                                 struct json_object *args,
 338                                 int flags,
 339                                 void (*callback)(void *closure, struct json_object *object, const char *error, const char * info, struct afb_req_x2 *req),
 340                                 void *closure)
 341 {
 342         struct afb_xreq *xreq = xreq_from_req_x2(req);
 343         afb_calls_subcall(xreq, api, verb, args, flags, callback, closure);
 344 }
 345
 346 static int xreq_subcallsync_cb(
 347                                 struct afb_req_x2 *req,
 348                                 const char *api,
 349                                 const char *verb,
 350                                 struct json_object *args,
 351                                 int flags,
 352                                 struct json_object **object,
 353                                 char **error,
 354                                 char **info)
 355 {
 356         struct afb_xreq *xreq = xreq_from_req_x2(req);
 357         return afb_calls_subcall_sync(xreq, api, verb, args, flags, object, error, info);
 358 }
 359
 360 /******************************************************************************/
 361
 362 const struct afb_req_x2_itf xreq_itf = {
 363         .json = xreq_json_cb,
 364         .get = xreq_get_cb,
 365         .legacy_success = xreq_legacy_success_cb,
 366         .legacy_fail = xreq_legacy_fail_cb,
 367         .legacy_vsuccess = xreq_legacy_vsuccess_cb,
 368         .legacy_vfail = xreq_legacy_vfail_cb,
 369         .legacy_context_get = xreq_legacy_context_get_cb,
 370         .legacy_context_set = xreq_legacy_context_set_cb,
 371         .addref = xreq_addref_cb,
 372         .unref = xreq_unref_cb,
 373         .session_close = xreq_session_close_cb,
 374         .session_set_LOA = xreq_session_set_LOA_cb,
 375         .legacy_subscribe_event_x1 = xreq_legacy_subscribe_event_x1_cb,
 376         .legacy_unsubscribe_event_x1 = xreq_legacy_unsubscribe_event_x1_cb,
 377         .legacy_subcall = xreq_legacy_subcall_cb,
 378         .legacy_subcallsync = xreq_legacy_subcallsync_cb,
 379         .vverbose = xreq_vverbose_cb,
 380         .legacy_store_req = xreq_legacy_store_cb,
 381         .legacy_subcall_req = xreq_legacy_subcall_req_cb,
 382         .has_permission = xreq_has_permission_cb,
 383         .get_application_id = xreq_get_application_id_cb,
 384         .context_make = xreq_context_make_cb,
 385         .subscribe_event_x2 = xreq_subscribe_event_x2_cb,
 386         .unsubscribe_event_x2 = xreq_unsubscribe_event_x2_cb,
 387         .legacy_subcall_request = xreq_legacy_subcall_request_cb,
 388         .get_uid = xreq_get_uid_cb,
 389         .reply = xreq_reply_cb,
 390         .vreply = xreq_vreply_cb,
 391         .get_client_info = xreq_get_client_info_cb,
 392         .subcall = xreq_subcall_cb,
 393         .subcallsync = xreq_subcallsync_cb,
 394 };
 395 /******************************************************************************/
 396 #if WITH_AFB_HOOK
 397
 398 static struct json_object *xreq_hooked_json_cb(struct afb_req_x2 *closure)
 399 {
 400         struct json_object *r = xreq_json_cb(closure);
 401         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 402         return afb_hook_xreq_json(xreq, r);
 403 }
 404
 405 static struct afb_arg xreq_hooked_get_cb(struct afb_req_x2 *closure, const char *name)
 406 {
 407         struct afb_arg r = xreq_get_cb(closure, name);
 408         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 409         return afb_hook_xreq_get(xreq, name, r);
 410 }
 411
 412 static void xreq_hooked_reply_cb(struct afb_req_x2 *closure, struct json_object *obj, const char *error, const char *info)
 413 {
 414         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 415         afb_hook_xreq_reply(xreq, obj, error, info);
 416         xreq_reply_cb(closure, obj, error, info);
 417 }
 418
 419 static void xreq_hooked_vreply_cb(struct afb_req_x2 *closure, struct json_object *obj, const char *error, const char *fmt, va_list args)
 420 {
 421         char *info;
 422         if (fmt == NULL || vasprintf(&info, fmt, args) < 0)
 423                 info = NULL;
 424         xreq_hooked_reply_cb(closure, obj, error, info);
 425         free(info);
 426 }
 427
 428 static void xreq_hooked_legacy_success_cb(struct afb_req_x2 *closure, struct json_object *obj, const char *info)
 429 {
 430         xreq_hooked_reply_cb(closure, obj, NULL, info);
 431 }
 432
 433 static void xreq_hooked_legacy_fail_cb(struct afb_req_x2 *closure, const char *status, const char *info)
 434 {
 435         xreq_hooked_reply_cb(closure, NULL, status, info);
 436 }
 437
 438 static void xreq_hooked_legacy_vsuccess_cb(struct afb_req_x2 *closure, struct json_object *obj, const char *fmt, va_list args)
 439 {
 440         xreq_hooked_vreply_cb(closure, obj, NULL, fmt, args);
 441 }
 442
 443 static void xreq_hooked_legacy_vfail_cb(struct afb_req_x2 *closure, const char *status, const char *fmt, va_list args)
 444 {
 445         xreq_hooked_vreply_cb(closure, NULL, status, fmt, args);
 446 }
 447
 448 static void *xreq_hooked_legacy_context_get_cb(struct afb_req_x2 *closure)
 449 {
 450         void *r = xreq_legacy_context_get_cb(closure);
 451         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 452         return afb_hook_xreq_legacy_context_get(xreq, r);
 453 }
 454
 455 static void xreq_hooked_legacy_context_set_cb(struct afb_req_x2 *closure, void *value, void (*free_value)(void*))
 456 {
 457         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 458         afb_hook_xreq_legacy_context_set(xreq, value, free_value);
 459         xreq_legacy_context_set_cb(closure, value, free_value);
 460 }
 461
 462 static struct afb_req_x2 *xreq_hooked_addref_cb(struct afb_req_x2 *closure)
 463 {
 464         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 465         afb_hook_xreq_addref(xreq);
 466         return xreq_addref_cb(closure);
 467 }
 468
 469 static void xreq_hooked_unref_cb(struct afb_req_x2 *closure)
 470 {
 471         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 472         afb_hook_xreq_unref(xreq);
 473         xreq_unref_cb(closure);
 474 }
 475
 476 static void xreq_hooked_session_close_cb(struct afb_req_x2 *closure)
 477 {
 478         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 479         afb_hook_xreq_session_close(xreq);
 480         xreq_session_close_cb(closure);
 481 }
 482
 483 static int xreq_hooked_session_set_LOA_cb(struct afb_req_x2 *closure, unsigned level)
 484 {
 485         int r = xreq_session_set_LOA_cb(closure, level);
 486         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 487         return afb_hook_xreq_session_set_LOA(xreq, level, r);
 488 }
 489
 490 static int xreq_hooked_subscribe_event_x2_cb(struct afb_req_x2 *closure, struct afb_event_x2 *event)
 491 {
 492         int r = xreq_subscribe_event_x2_cb(closure, event);
 493         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 494         return afb_hook_xreq_subscribe(xreq, event, r);
 495 }
 496
 497 static int xreq_hooked_legacy_subscribe_event_x1_cb(struct afb_req_x2 *closure, struct afb_event_x1 event)
 498 {
 499         return xreq_hooked_subscribe_event_x2_cb(closure, afb_event_x1_to_event_x2(event));
 500 }
 501
 502 static int xreq_hooked_unsubscribe_event_x2_cb(struct afb_req_x2 *closure, struct afb_event_x2 *event)
 503 {
 504         int r = xreq_unsubscribe_event_x2_cb(closure, event);
 505         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 506         return afb_hook_xreq_unsubscribe(xreq, event, r);
 507 }
 508
 509 static int xreq_hooked_legacy_unsubscribe_event_x1_cb(struct afb_req_x2 *closure, struct afb_event_x1 event)
 510 {
 511         return xreq_hooked_unsubscribe_event_x2_cb(closure, afb_event_x1_to_event_x2(event));
 512 }
 513
 514 static void xreq_hooked_legacy_subcall_cb(struct afb_req_x2 *req, const char *api, const char *verb, struct json_object *args, void (*callback)(void*, int, struct json_object*), void *closure)
 515 {
 516         struct afb_xreq *xreq = xreq_from_req_x2(req);
 517         afb_calls_legacy_hooked_subcall_v1(xreq, api, verb, args, callback, closure);
 518 }
 519
 520 static void xreq_hooked_legacy_subcall_req_cb(struct afb_req_x2 *req, const char *api, const char *verb, struct json_object *args, void (*callback)(void*, int, struct json_object*, struct afb_req_x1), void *closure)
 521 {
 522         struct afb_xreq *xreq = xreq_from_req_x2(req);
 523         afb_calls_legacy_hooked_subcall_v2(xreq, api, verb, args, callback, closure);
 524 }
 525
 526 static void xreq_hooked_legacy_subcall_request_cb(struct afb_req_x2 *req, const char *api, const char *verb, struct json_object *args, void (*callback)(void*, int, struct json_object*, struct afb_req_x2 *), void *closure)
 527 {
 528         struct afb_xreq *xreq = xreq_from_req_x2(req);
 529         afb_calls_legacy_hooked_subcall_v3(xreq, api, verb, args, callback, closure);
 530 }
 531
 532 static int xreq_hooked_legacy_subcallsync_cb(struct afb_req_x2 *req, const char *api, const char *verb, struct json_object *args, struct json_object **result)
 533 {
 534         struct afb_xreq *xreq = xreq_from_req_x2(req);
 535         return afb_calls_legacy_hooked_subcall_sync(xreq, api, verb, args, result);
 536 }
 537
 538 static void xreq_hooked_vverbose_cb(struct afb_req_x2 *closure, int level, const char *file, int line, const char *func, const char *fmt, va_list args)
 539 {
 540         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 541         va_list ap;
 542         va_copy(ap, args);
 543         xreq_vverbose_cb(closure, level, file, line, func, fmt, args);
 544         afb_hook_xreq_vverbose(xreq, level, file, line, func, fmt, ap);
 545         va_end(ap);
 546 }
 547
 548 static struct afb_stored_req *xreq_hooked_legacy_store_cb(struct afb_req_x2 *closure)
 549 {
 550         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 551         struct afb_stored_req *r = xreq_legacy_store_cb(closure);
 552         afb_hook_xreq_legacy_store(xreq, r);
 553         return r;
 554 }
 555
 556 static int xreq_hooked_has_permission_cb(struct afb_req_x2 *closure, const char *permission)
 557 {
 558         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 559         int r = xreq_has_permission_cb(closure, permission);
 560         return afb_hook_xreq_has_permission(xreq, permission, r);
 561 }
 562
 563 static char *xreq_hooked_get_application_id_cb(struct afb_req_x2 *closure)
 564 {
 565         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 566         char *r = xreq_get_application_id_cb(closure);
 567         return afb_hook_xreq_get_application_id(xreq, r);
 568 }
 569
 570 static void *xreq_hooked_context_make_cb(struct afb_req_x2 *closure, int replace, void *(*create_value)(void*), void (*free_value)(void*), void *create_closure)
 571 {
 572         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 573         void *result = xreq_context_make_cb(closure, replace, create_value, free_value, create_closure);
 574         return afb_hook_xreq_context_make(xreq, replace, create_value, free_value, create_closure, result);
 575 }
 576
 577 static int xreq_hooked_get_uid_cb(struct afb_req_x2 *closure)
 578 {
 579         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 580         int r = xreq_get_uid_cb(closure);
 581         return afb_hook_xreq_get_uid(xreq, r);
 582 }
 583
 584 static struct json_object *xreq_hooked_get_client_info_cb(struct afb_req_x2 *closure)
 585 {
 586         struct afb_xreq *xreq = xreq_from_req_x2(closure);
 587         struct json_object *r = xreq_get_client_info_cb(closure);
 588         return afb_hook_xreq_get_client_info(xreq, r);
 589 }
 590
 591 static void xreq_hooked_subcall_cb(
 592                                 struct afb_req_x2 *req,
 593                                 const char *api,
 594                                 const char *verb,
 595                                 struct json_object *args,
 596                                 int flags,
 597                                 void (*callback)(void *closure, struct json_object *object, const char *error, const char * info, struct afb_req_x2 *req),
 598                                 void *closure)
 599 {
 600         struct afb_xreq *xreq = xreq_from_req_x2(req);
 601         afb_calls_hooked_subcall(xreq, api, verb, args, flags, callback, closure);
 602 }
 603
 604 static int xreq_hooked_subcallsync_cb(
 605                                 struct afb_req_x2 *req,
 606                                 const char *api,
 607                                 const char *verb,
 608                                 struct json_object *args,
 609                                 int flags,
 610                                 struct json_object **object,
 611                                 char **error,
 612                                 char **info)
 613 {
 614         struct afb_xreq *xreq = xreq_from_req_x2(req);
 615         return afb_calls_hooked_subcall_sync(xreq, api, verb, args, flags, object, error, info);
 616 }
 617
 618 /******************************************************************************/
 619
 620 const struct afb_req_x2_itf xreq_hooked_itf = {
 621         .json = xreq_hooked_json_cb,
 622         .get = xreq_hooked_get_cb,
 623         .legacy_success = xreq_hooked_legacy_success_cb,
 624         .legacy_fail = xreq_hooked_legacy_fail_cb,
 625         .legacy_vsuccess = xreq_hooked_legacy_vsuccess_cb,
 626         .legacy_vfail = xreq_hooked_legacy_vfail_cb,
 627         .legacy_context_get = xreq_hooked_legacy_context_get_cb,
 628         .legacy_context_set = xreq_hooked_legacy_context_set_cb,
 629         .addref = xreq_hooked_addref_cb,
 630         .unref = xreq_hooked_unref_cb,
 631         .session_close = xreq_hooked_session_close_cb,
 632         .session_set_LOA = xreq_hooked_session_set_LOA_cb,
 633         .legacy_subscribe_event_x1 = xreq_hooked_legacy_subscribe_event_x1_cb,
 634         .legacy_unsubscribe_event_x1 = xreq_hooked_legacy_unsubscribe_event_x1_cb,
 635         .legacy_subcall = xreq_hooked_legacy_subcall_cb,
 636         .legacy_subcallsync = xreq_hooked_legacy_subcallsync_cb,
 637         .vverbose = xreq_hooked_vverbose_cb,
 638         .legacy_store_req = xreq_hooked_legacy_store_cb,
 639         .legacy_subcall_req = xreq_hooked_legacy_subcall_req_cb,
 640         .has_permission = xreq_hooked_has_permission_cb,
 641         .get_application_id = xreq_hooked_get_application_id_cb,
 642         .context_make = xreq_hooked_context_make_cb,
 643         .subscribe_event_x2 = xreq_hooked_subscribe_event_x2_cb,
 644         .unsubscribe_event_x2 = xreq_hooked_unsubscribe_event_x2_cb,
 645         .legacy_subcall_request = xreq_hooked_legacy_subcall_request_cb,
 646         .get_uid = xreq_hooked_get_uid_cb,
 647         .reply = xreq_hooked_reply_cb,
 648         .vreply = xreq_hooked_vreply_cb,
 649         .get_client_info = xreq_hooked_get_client_info_cb,
 650         .subcall = xreq_hooked_subcall_cb,
 651         .subcallsync = xreq_hooked_subcallsync_cb,
 652 };
 653 #endif
 654
 655 /******************************************************************************/
 656
 657 struct afb_req_x1 afb_xreq_unstore(struct afb_stored_req *sreq)
 658 {
 659         struct afb_xreq *xreq = (struct afb_xreq *)sreq;
 660 #if WITH_AFB_HOOK
 661         if (xreq->hookflags)
 662                 afb_hook_xreq_legacy_unstore(xreq);
 663 #endif
 664         return xreq_to_req_x1(xreq);
 665 }
 666
 667 struct json_object *afb_xreq_json(struct afb_xreq *xreq)
 668 {
 669         return afb_req_x2_json(xreq_to_req_x2(xreq));
 670 }
 671
 672 void afb_xreq_reply(struct afb_xreq *xreq, struct json_object *obj, const char *error, const char *info)
 673 {
 674         afb_req_x2_reply(xreq_to_req_x2(xreq), obj, error, info);
 675 }
 676
 677 void afb_xreq_reply_f(struct afb_xreq *xreq, struct json_object *obj, const char *error, const char *info, ...)
 678 {
 679         va_list args;
 680
 681         va_start(args, info);
 682         afb_req_x2_reply_v(xreq_to_req_x2(xreq), obj, error, info, args);
 683         va_end(args);
 684 }
 685
 686 const char *afb_xreq_raw(struct afb_xreq *xreq, size_t *size)
 687 {
 688         struct json_object *obj = xreq_json_cb(xreq_to_req_x2(xreq));
 689         const char *result = json_object_to_json_string_ext(obj, JSON_C_TO_STRING_NOSLASHESCAPE);
 690         if (size != NULL)
 691                 *size = strlen(result);
 692         return result;
 693 }
 694
 695 void afb_xreq_unhooked_legacy_subcall(struct afb_xreq *xreq, const char *api, const char *verb, struct json_object *args, void (*callback)(void*, int, struct json_object*, struct afb_req_x2 *), void *cb_closure)
 696 {
 697         xreq_legacy_subcall_request_cb(xreq_to_req_x2(xreq), api, verb, args, callback, cb_closure);
 698 }
 699
 700 void afb_xreq_unhooked_subcall(struct afb_xreq *xreq, const char *api, const char *verb, struct json_object *args, int flags, void (*callback)(void*, struct json_object*, const char*, const char*, struct afb_req_x2 *), void *closure)
 701 {
 702         xreq_subcall_cb(xreq_to_req_x2(xreq), api, verb, args, flags, callback, closure);
 703 }
 704
 705 int afb_xreq_unhooked_legacy_subcall_sync(struct afb_xreq *xreq, const char *api, const char *verb, struct json_object *args, struct json_object **result)
 706 {
 707         return xreq_legacy_subcallsync_cb(xreq_to_req_x2(xreq), api, verb, args, result);
 708 }
 709
 710 void afb_xreq_addref(struct afb_xreq *xreq)
 711 {
 712         afb_req_x2_addref(xreq_to_req_x2(xreq));
 713 }
 714
 715 void afb_xreq_unref(struct afb_xreq *xreq)
 716 {
 717         afb_req_x2_unref(xreq_to_req_x2(xreq));
 718 }
 719
 720 void afb_xreq_legacy_subcall(struct afb_xreq *xreq, const char *api, const char *verb, struct json_object *args, void (*callback)(void*, int, struct json_object*, struct afb_req_x2 *), void *cb_closure)
 721 {
 722         afb_req_x2_subcall_legacy(xreq_to_req_x2(xreq), api, verb, args, callback, cb_closure);
 723 }
 724
 725 void afb_xreq_subcall(struct afb_xreq *xreq, const char *api, const char *verb, struct json_object *args, int flags, void (*callback)(void*, struct json_object*, const char*, const char*, struct afb_req_x2 *), void *closure)
 726 {
 727         afb_req_x2_subcall(xreq_to_req_x2(xreq), api, verb, args, flags, callback, closure);
 728 }
 729
 730 int afb_xreq_legacy_subcall_sync(struct afb_xreq *xreq, const char *api, const char *verb, struct json_object *args, struct json_object **result)
 731 {
 732         return afb_req_x2_subcall_sync_legacy(xreq_to_req_x2(xreq), api, verb, args, result);
 733 }
 734
 735 int afb_xreq_reply_unknown_api(struct afb_xreq *xreq)
 736 {
 737         afb_xreq_reply_f(xreq, NULL, afb_error_text_unknown_api, "api %s not found (for verb %s)", xreq->request.called_api, xreq->request.called_verb);
 738         errno = EINVAL;
 739         return -1;
 740 }
 741
 742 int afb_xreq_reply_unknown_verb(struct afb_xreq *xreq)
 743 {
 744         afb_xreq_reply_f(xreq, NULL, afb_error_text_unknown_verb, "verb %s unknown within api %s", xreq->request.called_verb, xreq->request.called_api);
 745         errno = EINVAL;
 746         return -1;
 747 }
 748
 749 int afb_xreq_reply_invalid_token(struct afb_xreq *xreq)
 750 {
 751         afb_xreq_reply(xreq, NULL, afb_error_text_invalid_token, "invalid token"); /* TODO: or "no token" */
 752         errno = EINVAL;
 753         return -1;
 754 }
 755
 756 int afb_xreq_reply_insufficient_scope(struct afb_xreq *xreq, const char *scope)
 757 {
 758         afb_xreq_reply(xreq, NULL, afb_error_text_insufficient_scope, scope ?: "insufficient scope");
 759         errno = EPERM;
 760         return -1;
 761 }
 762
 763 #if WITH_LEGACY_BINDING_V1
 764 void afb_xreq_call_verb_v1(struct afb_xreq *xreq, const struct afb_verb_desc_v1 *verb)
 765 {
 766         if (!verb)
 767                 afb_xreq_reply_unknown_verb(xreq);
 768         else
 769                 if (afb_auth_check_and_set_session_x1(xreq, verb->session) >= 0)
 770                         verb->callback(xreq_to_req_x1(xreq));
 771 }
 772 #endif
 773
 774 #if WITH_LEGACY_BINDING_V2
 775 void afb_xreq_call_verb_v2(struct afb_xreq *xreq, const struct afb_verb_v2 *verb)
 776 {
 777         if (!verb)
 778                 afb_xreq_reply_unknown_verb(xreq);
 779         else
 780                 if (afb_auth_check_and_set_session_x2(xreq, verb->auth, verb->session) > 0)
 781                         verb->callback(xreq_to_req_x1(xreq));
 782 }
 783 #endif
 784
 785 void afb_xreq_call_verb_v3(struct afb_xreq *xreq, const struct afb_verb_v3 *verb)
 786 {
 787         if (!verb)
 788                 afb_xreq_reply_unknown_verb(xreq);
 789         else
 790                 if (afb_auth_check_and_set_session_x2(xreq, verb->auth, verb->session) > 0)
 791                         verb->callback(xreq_to_req_x2(xreq));
 792 }
 793
 794 void afb_xreq_init(struct afb_xreq *xreq, const struct afb_xreq_query_itf *queryitf)
 795 {
 796         memset(xreq, 0, sizeof *xreq);
 797         xreq->request.itf = &xreq_itf; /* no hook by default */
 798         xreq->refcount = 1;
 799         xreq->queryitf = queryitf;
 800 }
 801
 802 #if WITH_AFB_HOOK
 803 static void init_hooking(struct afb_xreq *xreq)
 804 {
 805         afb_hook_init_xreq(xreq);
 806         if (xreq->hookflags) {
 807                 xreq->request.itf = &xreq_hooked_itf; /* unhook the interface */
 808                 afb_hook_xreq_begin(xreq);
 809         }
 810 }
 811 #endif
 812
 813 /**
 814  * job callback for asynchronous and secured processing of the request.
 815  */
 816 static void process_async(int signum, void *arg)
 817 {
 818         struct afb_xreq *xreq = arg;
 819         const struct afb_api_item *api;
 820
 821         if (signum != 0) {
 822                 /* emit the error (assumes that hooking is initialised) */
 823                 afb_xreq_reply_f(xreq, NULL, afb_error_text_aborted, "signal %s(%d) caught", strsignal(signum), signum);
 824         } else {
 825 #if WITH_AFB_HOOK
 826                 /* init hooking */
 827                 init_hooking(xreq);
 828 #endif
 829                 /* invoke api call method to process the request */
 830                 api = (const struct afb_api_item*)xreq->context.api_key;
 831                 api->itf->call(api->closure, xreq);
 832         }
 833         /* release the request */
 834         afb_xreq_unhooked_unref(xreq);
 835 }
 836
 837 /**
 838  * Early request failure of the request 'xreq' with, as usual, 'status' and 'info'
 839  * The early failure occurs only in function 'afb_xreq_process' where normally,
 840  * the hooking is not initialised. So this "early" failure takes care to initialise
 841  * the hooking in first.
 842  */
 843 static void early_failure(struct afb_xreq *xreq, const char *status, const char *info, ...)
 844 {
 845         va_list args;
 846
 847 #if WITH_AFB_HOOK
 848         /* init hooking */
 849         init_hooking(xreq);
 850 #endif
 851
 852         /* send error */
 853         va_start(args, info);
 854         afb_req_x2_reply_v(xreq_to_req_x2(xreq), NULL, status, info, args);
 855         va_end(args);
 856 }
 857
 858 /**
 859  * Enqueue a job for processing the request 'xreq' using the given 'apiset'.
 860  * Errors are reported as request failures.
 861  */
 862 void afb_xreq_process(struct afb_xreq *xreq, struct afb_apiset *apiset)
 863 {
 864         const struct afb_api_item *api;
 865         struct afb_xreq *caller;
 866
 867         /* lookup at the api */
 868         xreq->apiset = apiset;
 869         api = afb_apiset_lookup_started(apiset, xreq->request.called_api, 1);
 870         if (!api) {
 871                 if (errno == ENOENT)
 872                         early_failure(xreq, "unknown-api", "api %s not found (for verb %s)", xreq->request.called_api, xreq->request.called_verb);
 873                 else
 874                         early_failure(xreq, "bad-api-state", "api %s not started correctly: %m", xreq->request.called_api);
 875                 goto end;
 876         }
 877         xreq->context.api_key = api;
 878
 879         /* check self locking */
 880         if (api->group) {
 881                 caller = xreq->caller;
 882                 while (caller) {
 883                         if (((const struct afb_api_item*)caller->context.api_key)->group == api->group) {
 884                                 /* noconcurrency lock detected */
 885                                 ERROR("self-lock detected in call stack for API %s", xreq->request.called_api);
 886                                 early_failure(xreq, "self-locked", "recursive self lock, API %s", xreq->request.called_api);
 887                                 goto end;
 888                         }
 889                         caller = caller->caller;
 890                 }
 891         }
 892
 893         /* queue the request job */
 894         afb_xreq_unhooked_addref(xreq);
 895         if (jobs_queue(api->group, afb_apiset_timeout_get(apiset), process_async, xreq) < 0) {
 896                 /* TODO: allows or not to proccess it directly as when no threading? (see above) */
 897                 ERROR("can't process job with threads: %m");
 898                 early_failure(xreq, "cancelled", "not able to create a job for the task");
 899                 afb_xreq_unhooked_unref(xreq);
 900         }
 901 end:
 902         afb_xreq_unhooked_unref(xreq);
 903 }
 904
 905 const char *xreq_on_behalf_cred_export(struct afb_xreq *xreq)
 906 {
 907         return afb_context_on_behalf_export(&xreq->context);
 908 }
 909