#include "afb-ws.h"
#include "afb-ws-json.h"
+#include "afb-msg-json.h"
#include "session.h"
#include "afb-req-itf.h"
#include "afb-apis.h"
+#include "afb-context.h"
-static void aws_on_close(struct afb_ws_json *ws, uint16_t code, char *text, size_t size);
+static void aws_on_hangup(struct afb_ws_json *ws);
static void aws_on_text(struct afb_ws_json *ws, char *text, size_t size);
static struct afb_ws_itf aws_itf = {
- .on_close = (void*)aws_on_close,
- .on_text = (void*)aws_on_text,
- .on_binary = NULL,
+ .on_hangup = (void*)aws_on_hangup,
+ .on_text = (void*)aws_on_text
};
struct afb_wsreq;
void (*cleanup)(void*);
void *cleanup_closure;
struct afb_wsreq *requests;
- struct AFB_clientCtx *context;
+ struct AFB_clientCtx *session;
struct json_tokener *tokener;
struct afb_ws *ws;
};
-struct afb_ws_json *afb_ws_json_create(int fd, struct AFB_clientCtx *context, void (*cleanup)(void*), void *closure)
+static void aws_send_event(struct afb_ws_json *ws, const char *event, struct json_object *object);
+
+static const struct afb_event_listener_itf event_listener_itf = {
+ .send = (void*)aws_send_event
+};
+
+struct afb_ws_json *afb_ws_json_create(int fd, struct AFB_clientCtx *session, void (*cleanup)(void*), void *cleanup_closure)
{
struct afb_ws_json *result;
assert(fd >= 0);
- assert(context != NULL);
+ assert(session != NULL);
result = malloc(sizeof * result);
if (result == NULL)
goto error;
result->cleanup = cleanup;
- result->cleanup_closure = closure;
+ result->cleanup_closure = cleanup_closure;
result->requests = NULL;
- result->context = ctxClientGet(context);
- if (result->context == NULL)
+ result->session = ctxClientAddRef(session);
+ if (result->session == NULL)
goto error2;
result->tokener = json_tokener_new();
if (result->ws == NULL)
goto error4;
+ if (0 > ctxClientEventListenerAdd(result->session, (struct afb_event_listener){ .itf = &event_listener_itf, .closure = result }))
+ goto error5;
+
return result;
+error5:
+ afb_ws_destroy(result->ws);
error4:
json_tokener_free(result->tokener);
error3:
- ctxClientPut(result->context);
+ ctxClientUnref(result->session);
error2:
free(result);
error:
return NULL;
}
-static void aws_on_close(struct afb_ws_json *ws, uint16_t code, char *text, size_t size)
+static void aws_on_hangup(struct afb_ws_json *ws)
{
- /* do nothing */
- free(text);
+ ctxClientEventListenerRemove(ws->session, (struct afb_event_listener){ .itf = &event_listener_itf, .closure = ws });
+ afb_ws_destroy(ws->ws);
+ json_tokener_free(ws->tokener);
+ if (ws->cleanup != NULL)
+ ws->cleanup(ws->cleanup_closure);
+ ctxClientUnref(ws->session);
+ free(ws);
}
+#define CALL 2
+#define RETOK 3
+#define RETERR 4
+#define EVENT 5
struct afb_wsreq
{
+ struct afb_context context;
+ int refcount;
struct afb_ws_json *aws;
struct afb_wsreq *next;
- struct json_object *id;
- struct json_object *name;
- struct json_object *token;
- struct json_object *request;
+ char *text;
+ size_t size;
+ int code;
+ char *id;
+ size_t idlen;
+ char *api;
+ size_t apilen;
+ char *verb;
+ size_t verblen;
+ char *obj;
+ size_t objlen;
+ char *tok;
+ size_t toklen;
+ struct json_object *root;
};
+
+static void wsreq_addref(struct afb_wsreq *wsreq);
+static void wsreq_unref(struct afb_wsreq *wsreq);
+static struct json_object *wsreq_json(struct afb_wsreq *wsreq);
static struct afb_arg wsreq_get(struct afb_wsreq *wsreq, const char *name);
-static void wsreq_iterate(struct afb_wsreq *wsreq, int (*iterator)(void *closure, struct afb_arg arg), void *closure);
static void wsreq_fail(struct afb_wsreq *wsreq, const char *status, const char *info);
static void wsreq_success(struct afb_wsreq *wsreq, struct json_object *obj, const char *info);
-static int wsreq_session_create(struct afb_wsreq *wsreq);
-static int wsreq_session_check(struct afb_wsreq *wsreq, int refresh);
-static void wsreq_session_close(struct afb_wsreq *wsreq);
+static const char *wsreq_raw(struct afb_wsreq *wsreq, size_t *size);
+static void wsreq_send(struct afb_wsreq *wsreq, const char *buffer, size_t size);
+
static const struct afb_req_itf wsreq_itf = {
+ .json = (void*)wsreq_json,
.get = (void*)wsreq_get,
- .iterate = (void*)wsreq_iterate,
- .fail = (void*)wsreq_fail,
.success = (void*)wsreq_success,
- .session_create = (void*)wsreq_session_create,
- .session_check = (void*)wsreq_session_check,
- .session_close = (void*)wsreq_session_close
+ .fail = (void*)wsreq_fail,
+ .raw = (void*)wsreq_raw,
+ .send = (void*)wsreq_send,
+ .context_get = (void*)afb_context_get,
+ .context_set = (void*)afb_context_set,
+ .addref = (void*)wsreq_addref,
+ .unref = (void*)wsreq_unref
};
-static int aws_handle_json(struct afb_ws_json *aws, struct json_object *obj)
+static int aws_wsreq_parse(struct afb_wsreq *r, char *text, size_t size)
{
- struct afb_req r;
- int count, num;
- struct json_object *type, *id, *name, *req, *token;
- struct afb_wsreq *wsreq;
- const char *api, *verb;
- size_t lenapi, lenverb;
+ char *pos, *end, c;
+ int aux;
+
+ /* scan */
+ pos = text;
+ end = text + size;
+
+ /* scans: [ */
+ while(pos < end && *pos == ' ') pos++;
+ if (pos == end) goto bad_header;
+ if (*pos++ != '[') goto bad_header;
+
+ /* scans code: 2|3|4 */
+ while(pos < end && *pos == ' ') pos++;
+ if (pos == end) goto bad_header;
+ switch (*pos++) {
+ case '2': r->code = CALL; break;
+ case '3': r->code = RETOK; break;
+ case '4': r->code = RETERR; break;
+ default: goto bad_header;
+ }
- /* protocol inspired by http://www.gir.fr/ocppjs/ocpp_srpc_spec.shtml */
+ /* scans: , */
+ while(pos < end && *pos == ' ') pos++;
+ if (pos == end) goto bad_header;
+ if (*pos++ != ',') goto bad_header;
+
+ /* scans id: "id" */
+ while(pos < end && *pos == ' ') pos++;
+ if (pos == end) goto bad_header;
+ if (*pos++ != '"') goto bad_header;
+ r->id = pos;
+ while(pos < end && *pos != '"') pos++;
+ if (pos == end) goto bad_header;
+ r->idlen = (size_t)(pos++ - r->id);
+
+ /* scans: , */
+ while(pos < end && *pos == ' ') pos++;
+ if (pos == end) goto bad_header;
+ if (*pos++ != ',') goto bad_header;
+
+ /* scans the method if needed */
+ if (r->code == CALL) {
+ /* scans: " */
+ while(pos < end && *pos == ' ') pos++;
+ if (pos == end) goto bad_header;
+ if (*pos++ != '"') goto bad_header;
+
+ /* scans: api/ */
+ r->api = pos;
+ while(pos < end && *pos != '"' && *pos != '/') pos++;
+ if (pos == end) goto bad_header;
+ if (*pos != '/') goto bad_header;
+ r->apilen = (size_t)(pos++ - r->api);
+ if (r->apilen && r->api[r->apilen - 1] == '\\')
+ r->apilen--;
+
+ /* scans: verb" */
+ r->verb = pos;
+ while(pos < end && *pos != '"') pos++;
+ if (pos == end) goto bad_header;
+ r->verblen = (size_t)(pos++ - r->verb);
+
+ /* scans: , */
+ while(pos < end && *pos == ' ') pos++;
+ if (pos == end) goto bad_header;
+ if (*pos++ != ',') goto bad_header;
+ }
- /* the object must be an array of 4 or 5 elements */
- if (!json_object_is_type(obj, json_type_array))
- goto error;
- count = json_object_array_length(obj);
- if (count < 4 || count > 5)
- goto error;
+ /* scan obj */
+ while(pos < end && *pos == ' ') pos++;
+ if (pos == end) goto bad_header;
+ aux = 0;
+ r->obj = pos;
+ while (pos < end && (aux != 0 || (*pos != ',' && *pos != ']'))) {
+ if (pos == end) goto bad_header;
+ switch(*pos) {
+ case '{': case '[': aux++; break;
+ case '}': case ']': if (!aux--) goto bad_header; break;
+ case '"':
+ do {
+ pos += 1 + (*pos == '\\');
+ } while(pos < end && *pos != '"');
+ default:
+ break;
+ }
+ pos++;
+ }
+ if (pos > end) goto bad_header;
+ if (pos == end && aux != 0) goto bad_header;
+ c = *pos;
+ r->objlen = (size_t)(pos++ - r->obj);
+ while (r->objlen && r->obj[r->objlen - 1] == ' ')
+ r->objlen--;
+
+ /* scan the token (if any) */
+ if (c == ',') {
+ /* scans token: "token" */
+ while(pos < end && *pos == ' ') pos++;
+ if (pos == end) goto bad_header;
+ if (*pos++ != '"') goto bad_header;
+ r->tok = pos;
+ while(pos < end && *pos != '"') pos++;
+ if (pos == end) goto bad_header;
+ r->toklen = (size_t)(pos++ - r->tok);
+ while(pos < end && *pos == ' ') pos++;
+ if (pos == end) goto bad_header;
+ c = *pos++;
+ }
- /* get the 5 elements: type id name request token */
- type = json_object_array_get_idx(obj, 0);
- id = json_object_array_get_idx(obj, 1);
- name = json_object_array_get_idx(obj, 2);
- req = json_object_array_get_idx(obj, 3);
- token = json_object_array_get_idx(obj, 4);
+ /* scan: ] */
+ if (c != ']') goto bad_header;
+ while(pos < end && *pos == ' ') pos++;
+ if (pos != end) goto bad_header;
- /* check the types: int string string object string */
- if (!json_object_is_type(type, json_type_int))
- goto error;
- if (!json_object_is_type(id, json_type_string))
- goto error;
- if (!json_object_is_type(name, json_type_string))
- goto error;
- if (!json_object_is_type(req, json_type_object))
- goto error;
- if (token != NULL && !json_object_is_type(token, json_type_string))
- goto error;
+ /* done */
+ r->text = text;
+ r->size = size;
+ return 1;
- /* the type is only 2 */
- num = json_object_get_int(type);
- if (num != 2)
- goto error;
+bad_header:
+ return 0;
+}
- /* checks the api/verb structure of name */
- api = json_object_get_string(name);
- for (lenapi = 0 ; api[lenapi] && api[lenapi] != '/' ; lenapi++);
- if (!lenapi || !api[lenapi])
- goto error;
- verb = &api[lenapi+1];
- for (lenverb = 0 ; verb[lenverb] && verb[lenverb] != '/' ; lenverb++);
- if (!lenverb || verb[lenverb])
- goto error;
+static void aws_on_text(struct afb_ws_json *ws, char *text, size_t size)
+{
+ struct afb_req r;
+ struct afb_wsreq *wsreq;
- /* allocates the request data */
- wsreq = malloc(sizeof *wsreq);
+ /* allocate */
+ wsreq = calloc(1, sizeof *wsreq);
if (wsreq == NULL)
- goto error;
+ goto alloc_error;
+
+ /* init */
+ if (!aws_wsreq_parse(wsreq, text, size))
+ goto bad_header;
/* fill and record the request */
- wsreq->aws = aws;
- wsreq->id = json_object_get(id);
- wsreq->name = json_object_get(name);
- wsreq->token = json_object_get(token);
- wsreq->request = json_object_get(req);
- wsreq->next = aws->requests;
- aws->requests = wsreq;
- json_object_put(obj);
-
- r.data = wsreq;
+ if (wsreq->tok != NULL)
+ wsreq->tok[wsreq->toklen] = 0;
+ afb_context_init(&wsreq->context, ws->session, wsreq->tok);
+ if (!wsreq->context.invalidated)
+ wsreq->context.validated = 1;
+ wsreq->refcount = 1;
+ wsreq->aws = ws;
+ wsreq->next = ws->requests;
+ ws->requests = wsreq;
+
+ r.closure = wsreq;
r.itf = &wsreq_itf;
- afb_apis_call(r, aws->context, api, lenapi, verb, lenverb);
- return 1;
+ afb_apis_call(r, &wsreq->context, wsreq->api, wsreq->apilen, wsreq->verb, wsreq->verblen);
+ wsreq_unref(wsreq);
+ return;
-error:
- json_object_put(obj);
- return 0;
+bad_header:
+ free(wsreq);
+alloc_error:
+ free(text);
+ afb_ws_close(ws->ws, 1008, NULL);
+ return;
}
-static void aws_on_text(struct afb_ws_json *ws, char *text, size_t size)
+static void wsreq_addref(struct afb_wsreq *wsreq)
+{
+ wsreq->refcount++;
+}
+
+static void wsreq_unref(struct afb_wsreq *wsreq)
+{
+ if (--wsreq->refcount == 0) {
+ afb_context_disconnect(&wsreq->context);
+ free(wsreq->text);
+ free(wsreq);
+ }
+}
+
+static struct json_object *wsreq_json(struct afb_wsreq *wsreq)
{
- struct json_object *obj;
- json_tokener_reset(ws->tokener);
- obj = json_tokener_parse_ex(ws->tokener, text, (int)size);
- if (obj == NULL) {
- afb_ws_close(ws->ws, 1008);
- } else if (!aws_handle_json(ws, obj)) {
- afb_ws_close(ws->ws, 1008);
+ struct json_object *root = wsreq->root;
+ if (root == NULL) {
+ json_tokener_reset(wsreq->aws->tokener);
+ root = json_tokener_parse_ex(wsreq->aws->tokener, wsreq->obj, (int)wsreq->objlen);
+ if (root == NULL) {
+ /* lazy error detection of json request. Is it to improve? */
+ root = json_object_new_string_len(wsreq->obj, (int)wsreq->objlen);
+ }
+ wsreq->root = root;
}
+ return root;
}
static struct afb_arg wsreq_get(struct afb_wsreq *wsreq, const char *name)
{
struct afb_arg arg;
- struct json_object *value;
+ struct json_object *value, *root;
- if (json_object_object_get_ex(wsreq->request, name, &value)) {
+ root = wsreq_json(wsreq);
+ if (json_object_object_get_ex(root, name, &value)) {
arg.name = name;
arg.value = json_object_get_string(value);
- arg.size = strlen(arg.value);
} else {
arg.name = NULL;
arg.value = NULL;
- arg.size = 0;
}
arg.path = NULL;
return arg;
}
-static void wsreq_iterate(struct afb_wsreq *wsreq, int (*iterator)(void *closure, struct afb_arg arg), void *closure)
+static void aws_emit(struct afb_ws_json *aws, int code, const char *id, size_t idlen, struct json_object *data, const char *token)
{
- struct afb_arg arg;
- struct json_object_iterator it = json_object_iter_begin(wsreq->request);
- struct json_object_iterator end = json_object_iter_end(wsreq->request);
+ json_object *msg;
+ const char *txt;
- arg.size = 0;
- arg.path = NULL;
- while(!json_object_iter_equal(&it, &end)) {
- arg.name = json_object_iter_peek_name(&it);
- arg.value = json_object_get_string(json_object_iter_peek_value(&it));
- if (!iterator(closure, arg))
- break;
- json_object_iter_next(&it);
- }
+ /* pack the message */
+ msg = json_object_new_array();
+ json_object_array_add(msg, json_object_new_int(code));
+ json_object_array_add(msg, json_object_new_string_len(id, (int)idlen));
+ json_object_array_add(msg, data);
+ if (token)
+ json_object_array_add(msg, json_object_new_string(token));
+
+ /* emits the reply */
+ txt = json_object_to_json_string(msg);
+ afb_ws_text(aws->ws, txt, strlen(txt));
+ json_object_put(msg);
}
-static int wsreq_session_create(struct afb_wsreq *wsreq)
+static void wsreq_reply(struct afb_wsreq *wsreq, int retcode, const char *status, const char *info, json_object *resp)
{
- struct AFB_clientCtx *context = wsreq->aws->context;
- if (context->created)
- return 0;
- return wsreq_session_check(wsreq, 1);
+ const char *token = afb_context_sent_token(&wsreq->context);
+ aws_emit(wsreq->aws, retcode, wsreq->id, wsreq->idlen, afb_msg_json_reply(status, info, resp, token, NULL), token);
}
-static int wsreq_session_check(struct afb_wsreq *wsreq, int refresh)
+static void wsreq_fail(struct afb_wsreq *wsreq, const char *status, const char *info)
{
- const char *token;
- struct AFB_clientCtx *context = wsreq->aws->context;
-
- if (wsreq->token == NULL)
- return 0;
-
- token = json_object_get_string(wsreq->token);
- if (token == NULL)
- return 0;
-
- if (!ctxTokenCheck (context, token))
- return 0;
-
- if (refresh) {
- ctxTokenNew (context);
- }
-
- return 1;
+ wsreq_reply(wsreq, RETERR, status, info, NULL);
}
-static void wsreq_session_close(struct afb_wsreq *wsreq)
+static void wsreq_success(struct afb_wsreq *wsreq, json_object *obj, const char *info)
{
- struct AFB_clientCtx *context = wsreq->aws->context;
- ctxClientClose(context);
+ wsreq_reply(wsreq, RETOK, "success", info, obj);
}
-
-static void wsreq_reply(struct afb_wsreq *wsreq, int retcode, const char *status, const char *info, json_object *resp)
+static const char *wsreq_raw(struct afb_wsreq *wsreq, size_t *size)
{
- json_object *root, *request, *reply;
- const char *message;
-
- /* builds the answering structure */
- root = json_object_new_object();
- json_object_object_add(root, "jtype", json_object_new_string("afb-reply"));
- request = json_object_new_object();
- json_object_object_add(root, "request", request);
- json_object_object_add(request, "status", json_object_new_string(status));
- if (info)
- json_object_object_add(request, "info", json_object_new_string(info));
- if (resp)
- json_object_object_add(root, "response", resp);
-
- /* make the reply */
- reply = json_object_new_array();
- json_object_array_add(reply, json_object_new_int(retcode));
- json_object_array_add(reply, wsreq->id);
- json_object_array_add(reply, root);
- json_object_array_add(reply, json_object_new_string(wsreq->aws->context->token));
-
- /* emits the reply */
- message = json_object_to_json_string(reply);
- afb_ws_text(wsreq->aws->ws, message, strlen(message));
- json_object_put(reply);
-
- /* TODO eliminates the wsreq */
+ *size = wsreq->objlen;
+ return wsreq->obj;
}
-static void wsreq_fail(struct afb_wsreq *wsreq, const char *status, const char *info)
+static void wsreq_send(struct afb_wsreq *wsreq, const char *buffer, size_t size)
{
- wsreq_reply(wsreq, 4, status, info, NULL);
+ afb_ws_text(wsreq->aws->ws, buffer, size);
}
-static void wsreq_success(struct afb_wsreq *wsreq, json_object *obj, const char *info)
+static void aws_send_event(struct afb_ws_json *aws, const char *event, struct json_object *object)
{
- wsreq_reply(wsreq, 3, "success", info, obj);
+ aws_emit(aws, EVENT, event, strlen(event), afb_msg_json_event(event, object), NULL);
}