From 28158192742ead144454e071720d10bf5218a20b Mon Sep 17 00:00:00 2001 From: =?utf8?q?Jos=C3=A9=20Bollo?= Date: Wed, 18 May 2016 11:35:12 +0200 Subject: [PATCH] propagate context creation to websocket MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Change-Id: I9ff53f5920d6134030f3e1d63c46f4a30a5e1968 Signed-off-by: José Bollo --- src/afb-websock.c | 2 +- src/afb-ws-json1.c | 16 ++++++++++++---- src/afb-ws-json1.h | 4 ++-- 3 files changed, 15 insertions(+), 7 deletions(-) diff --git a/src/afb-websock.c b/src/afb-websock.c index a4237cfb..3ad9e20d 100644 --- a/src/afb-websock.c +++ b/src/afb-websock.c @@ -208,7 +208,7 @@ int afb_websock_check_upgrade(struct afb_hreq *hreq) return 0; ws = NULL; - rc = check_websocket_upgrade(hreq->connection, protodefs, hreq->context.session, &ws); + rc = check_websocket_upgrade(hreq->connection, protodefs, &hreq->context, &ws); if (rc == 1) { hreq->replied = 1; if (ws != NULL) diff --git a/src/afb-ws-json1.c b/src/afb-ws-json1.c index 85dc0259..78b8c210 100644 --- a/src/afb-ws-json1.c +++ b/src/afb-ws-json1.c @@ -51,6 +51,7 @@ struct afb_ws_json1 struct AFB_clientCtx *session; struct json_tokener *tokener; struct afb_ws *ws; + int new_session; }; static void aws_send_event(struct afb_ws_json1 *ws, const char *event, struct json_object *object); @@ -65,12 +66,12 @@ static inline struct afb_event_listener listener_for(struct afb_ws_json1 *aws) return (struct afb_event_listener){ .itf = &event_listener_itf, .closure = aws }; } -struct afb_ws_json1 *afb_ws_json1_create(int fd, struct AFB_clientCtx *session, void (*cleanup)(void*), void *cleanup_closure) +struct afb_ws_json1 *afb_ws_json1_create(int fd, struct afb_context *context, void (*cleanup)(void*), void *cleanup_closure) { struct afb_ws_json1 *result; assert(fd >= 0); - assert(session != NULL); + assert(context != NULL); result = malloc(sizeof * result); if (result == NULL) @@ -79,7 +80,8 @@ struct afb_ws_json1 *afb_ws_json1_create(int fd, struct AFB_clientCtx *session, result->cleanup = cleanup; result->cleanup_closure = cleanup_closure; result->requests = NULL; - result->session = ctxClientAddRef(session); + result->session = ctxClientAddRef(context->session); + result->new_session = context->created != 0; if (result->session == NULL) goto error2; @@ -316,6 +318,10 @@ static void aws_on_text(struct afb_ws_json1 *ws, char *text, size_t size) afb_context_init(&wsreq->context, ws->session, wsreq->tok); if (!wsreq->context.invalidated) wsreq->context.validated = 1; + if (ws->new_session != 0) { + wsreq->context.created = 1; + ws->new_session = 0; + } wsreq->refcount = 1; wsreq->aws = ws; wsreq->next = ws->requests; @@ -411,8 +417,10 @@ static void aws_emit(struct afb_ws_json1 *aws, int code, const char *id, size_t static void wsreq_reply(struct afb_wsreq *wsreq, int retcode, const char *status, const char *info, json_object *resp) { + const char *uuid = afb_context_sent_uuid(&wsreq->context); 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); + struct json_object *reply = afb_msg_json_reply(status, info, resp, token, uuid); + aws_emit(wsreq->aws, retcode, wsreq->id, wsreq->idlen, reply, token); } static void wsreq_fail(struct afb_wsreq *wsreq, const char *status, const char *info) diff --git a/src/afb-ws-json1.h b/src/afb-ws-json1.h index 964835bd..fedbcf6c 100644 --- a/src/afb-ws-json1.h +++ b/src/afb-ws-json1.h @@ -18,7 +18,7 @@ #pragma once struct afb_ws_json1; -struct AFB_clientCtx; +struct afb_context; -extern struct afb_ws_json1 *afb_ws_json1_create(int fd, struct AFB_clientCtx *context, void (*cleanup)(void*), void *closure); +extern struct afb_ws_json1 *afb_ws_json1_create(int fd, struct afb_context *context, void (*cleanup)(void*), void *closure); -- 2.16.6