Improve parsing of json string
[src/app-framework-binder.git] / src / afb-wsj1.c
index 068a332..c8572a5 100644 (file)
@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2016 "IoT.bzh"
+ * Copyright (C) 2016, 2017, 2018 "IoT.bzh"
  * Author: José Bollo <jose.bollo@iot.bzh>
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
 #include <errno.h>
 #include <string.h>
 #include <stdio.h>
+#include <pthread.h>
 
 #include <json-c/json.h>
+#if !defined(JSON_C_TO_STRING_NOSLASHESCAPE)
+#define JSON_C_TO_STRING_NOSLASHESCAPE 0
+#endif
 
 #include "afb-ws.h"
 #include "afb-wsj1.h"
+#include "fdev.h"
 
 #define CALL 2
 #define RETOK 3
@@ -77,13 +82,16 @@ struct afb_wsj1
        struct afb_ws *ws;
        struct afb_wsj1_msg *messages;
        struct wsj1_call *calls;
+       pthread_mutex_t mutex;
 };
 
-struct afb_wsj1 *afb_wsj1_create(int fd, struct afb_wsj1_itf *itf, void *closure)
+struct afb_wsj1 *afb_wsj1_create(struct fdev *fdev, struct afb_wsj1_itf *itf, void *closure)
 {
        struct afb_wsj1 *result;
 
-       assert(fd >= 0);
+       assert(fdev);
+       assert(itf);
+       assert(itf->on_call);
 
        result = calloc(1, sizeof * result);
        if (result == NULL)
@@ -92,12 +100,13 @@ struct afb_wsj1 *afb_wsj1_create(int fd, struct afb_wsj1_itf *itf, void *closure
        result->refcount = 1;
        result->itf = itf;
        result->closure = closure;
+       pthread_mutex_init(&result->mutex, NULL);
 
        result->tokener = json_tokener_new();
        if (result->tokener == NULL)
                goto error2;
 
-       result->ws = afb_ws_create(fd, &wsj1_itf, result);
+       result->ws = afb_ws_create(fdev, &wsj1_itf, result);
        if (result->ws == NULL)
                goto error3;
 
@@ -108,19 +117,19 @@ error3:
 error2:
        free(result);
 error:
-       close(fd);
+       fdev_unref(fdev);
        return NULL;
 }
 
 void afb_wsj1_addref(struct afb_wsj1 *wsj1)
 {
-       if (wsj1 != NULL)
-               wsj1->refcount++;
+       if (wsj1)
+               __atomic_add_fetch(&wsj1->refcount, 1, __ATOMIC_RELAXED);
 }
 
 void afb_wsj1_unref(struct afb_wsj1 *wsj1)
 {
-       if (wsj1 != NULL && !--wsj1->refcount) {
+       if (wsj1 && !__atomic_sub_fetch(&wsj1->refcount, 1, __ATOMIC_RELAXED)) {
                afb_ws_destroy(wsj1->ws);
                json_tokener_free(wsj1->tokener);
                free(wsj1);
@@ -134,9 +143,10 @@ static void wsj1_on_hangup(struct afb_wsj1 *wsj1)
 }
 
 
-static struct wsj1_call *wsj1_call_search(struct afb_wsj1 *wsj1, const char *id, int remove)
+static struct wsj1_call *wsj1_locked_call_search(struct afb_wsj1 *wsj1, const char *id, int remove)
 {
        struct wsj1_call *r, **p;
+
        p = &wsj1->calls;
        while((r = *p) != NULL) {
                if (strcmp(r->id, id) == 0) {
@@ -146,6 +156,18 @@ static struct wsj1_call *wsj1_call_search(struct afb_wsj1 *wsj1, const char *id,
                }
                p = &r->next;
        }
+
+       return r;
+}
+
+static struct wsj1_call *wsj1_call_search(struct afb_wsj1 *wsj1, const char *id, int remove)
+{
+       struct wsj1_call *r;
+
+       pthread_mutex_lock(&wsj1->mutex);
+       r = wsj1_locked_call_search(wsj1, id, remove);
+       pthread_mutex_unlock(&wsj1->mutex);
+
        return r;
 }
 
@@ -155,15 +177,17 @@ static struct wsj1_call *wsj1_call_create(struct afb_wsj1 *wsj1, void (*on_reply
        if (call == NULL)
                errno = ENOMEM;
        else {
+               pthread_mutex_lock(&wsj1->mutex);
                do {
                        if (wsj1->genid == 0)
                                wsj1->genid = 999999;
                        sprintf(call->id, "%d", wsj1->genid--);
-               } while (wsj1_call_search(wsj1, call->id, 0) != NULL);
+               } while (wsj1_locked_call_search(wsj1, call->id, 0) != NULL);
                call->callback = on_reply;
                call->closure = closure;
                call->next = wsj1->calls;
                wsj1->calls = call;
+               pthread_mutex_unlock(&wsj1->mutex);
        }
        return call;
 }
@@ -301,10 +325,12 @@ static void wsj1_on_text(struct afb_wsj1 *wsj1, char *text, size_t size)
        msg->refcount = 1;
        afb_wsj1_addref(wsj1);
        msg->wsj1 = wsj1;
+       pthread_mutex_lock(&wsj1->mutex);
        msg->next = wsj1->messages;
        if (msg->next != NULL)
                msg->next->previous = msg;
        wsj1->messages = msg;
+       pthread_mutex_unlock(&wsj1->mutex);
 
        /* incoke the handler */
        switch (msg->code) {
@@ -317,7 +343,8 @@ static void wsj1_on_text(struct afb_wsj1 *wsj1, char *text, size_t size)
                free(call);
                break;
        case EVENT:
-               wsj1->itf->on_event(wsj1->closure, msg->event, msg);
+               if (wsj1->itf->on_event != NULL)
+                       wsj1->itf->on_event(wsj1->closure, msg->event, msg);
                break;
        }
        afb_wsj1_msg_unref(msg);
@@ -333,19 +360,21 @@ alloc_error:
 void afb_wsj1_msg_addref(struct afb_wsj1_msg *msg)
 {
        if (msg != NULL)
-               msg->refcount++;
+               __atomic_add_fetch(&msg->refcount, 1, __ATOMIC_RELAXED);
 }
 
 void afb_wsj1_msg_unref(struct afb_wsj1_msg *msg)
 {
-       if (msg != NULL && --msg->refcount == 0) {
+       if (msg != NULL && !__atomic_sub_fetch(&msg->refcount, 1, __ATOMIC_RELAXED)) {
                /* unlink the message */
+               pthread_mutex_lock(&msg->wsj1->mutex);
                if (msg->next != NULL)
                        msg->next->previous = msg->previous;
                if (msg->previous == NULL)
                        msg->wsj1->messages = msg->next;
                else
                        msg->previous->next = msg->next;
+               pthread_mutex_unlock(&msg->wsj1->mutex);
                /* free ressources */
                afb_wsj1_unref(msg->wsj1);
                json_object_put(msg->object_j);
@@ -361,11 +390,15 @@ const char *afb_wsj1_msg_object_s(struct afb_wsj1_msg *msg)
 
 struct json_object *afb_wsj1_msg_object_j(struct afb_wsj1_msg *msg)
 {
+       enum json_tokener_error jerr;
        struct json_object *object = msg->object_j;
        if (object == NULL) {
+               pthread_mutex_lock(&msg->wsj1->mutex);
                json_tokener_reset(msg->wsj1->tokener);
-               object = json_tokener_parse_ex(msg->wsj1->tokener, msg->object_s, (int)msg->object_s_length);
-               if (object == NULL) {
+               object = json_tokener_parse_ex(msg->wsj1->tokener, msg->object_s, 1 + (int)msg->object_s_length);
+               jerr = json_tokener_get_error(msg->wsj1->tokener);
+               pthread_mutex_unlock(&msg->wsj1->mutex);
+               if (jerr != json_tokener_success) {
                        /* lazy error detection of json request. Is it to improve? */
                        object = json_object_new_string_len(msg->object_s, (int)msg->object_s_length);
                }
@@ -443,7 +476,7 @@ static int wsj1_send_issot(struct afb_wsj1 *wsj1, int i1, const char *s1, const
 
 int afb_wsj1_send_event_j(struct afb_wsj1 *wsj1, const char *event, struct json_object *object)
 {
-       const char *objstr = json_object_to_json_string_ext(object, JSON_C_TO_STRING_PLAIN);
+       const char *objstr = json_object_to_json_string_ext(object, JSON_C_TO_STRING_PLAIN|JSON_C_TO_STRING_NOSLASHESCAPE);
        int rc = afb_wsj1_send_event_s(wsj1, event, objstr);
        json_object_put(object);
        return rc;
@@ -456,7 +489,7 @@ int afb_wsj1_send_event_s(struct afb_wsj1 *wsj1, const char *event, const char *
 
 int afb_wsj1_call_j(struct afb_wsj1 *wsj1, const char *api, const char *verb, struct json_object *object, void (*on_reply)(void *closure, struct afb_wsj1_msg *msg), void *closure)
 {
-       const char *objstr = json_object_to_json_string_ext(object, JSON_C_TO_STRING_PLAIN);
+       const char *objstr = json_object_to_json_string_ext(object, JSON_C_TO_STRING_PLAIN|JSON_C_TO_STRING_NOSLASHESCAPE);
        int rc = afb_wsj1_call_s(wsj1, api, verb, objstr, on_reply, closure);
        json_object_put(object);
        return rc;
@@ -490,7 +523,7 @@ int afb_wsj1_call_s(struct afb_wsj1 *wsj1, const char *api, const char *verb, co
 
 int afb_wsj1_reply_j(struct afb_wsj1_msg *msg, struct json_object *object, const char *token, int iserror)
 {
-       const char *objstr = json_object_to_json_string_ext(object, JSON_C_TO_STRING_PLAIN);
+       const char *objstr = json_object_to_json_string_ext(object, JSON_C_TO_STRING_PLAIN|JSON_C_TO_STRING_NOSLASHESCAPE);
        int rc = afb_wsj1_reply_s(msg, objstr, token, iserror);
        json_object_put(object);
        return rc;