2 * Copyright (C) 2015-2018 "IoT.bzh"
3 * Author José Bollo <jose.bollo@iot.bzh>
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
9 * http://www.apache.org/licenses/LICENSE-2.0
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.
28 #include <sys/types.h>
29 #include <sys/socket.h>
33 #include <json-c/json.h>
36 #include "afb-msg-json.h"
37 #include "afb-proto-ws.h"
44 /******** implementation of internal binder protocol per api **************/
47 This protocol is asymetric: there is a client and a server
49 The client can require the following actions:
55 The server must reply to the previous actions by
57 - answering success or failure of the call
59 - answering the required description
61 The server can also within the context of a call
63 - subscribe or unsubscribe an event
65 For the purpose of handling events the server can:
67 - create/destroy an event
69 - push or brodcast data as an event
72 /************** constants for protocol definition *************************/
74 #define CHAR_FOR_CALL 'C'
75 #define CHAR_FOR_REPLY 'Y'
76 #define CHAR_FOR_EVT_BROADCAST '*'
77 #define CHAR_FOR_EVT_ADD '+'
78 #define CHAR_FOR_EVT_DEL '-'
79 #define CHAR_FOR_EVT_PUSH '!'
80 #define CHAR_FOR_EVT_SUBSCRIBE 'S'
81 #define CHAR_FOR_EVT_UNSUBSCRIBE 'U'
82 #define CHAR_FOR_DESCRIBE 'D'
83 #define CHAR_FOR_DESCRIPTION 'd'
85 /******************* handling calls *****************************/
88 * structure for recording calls on client side
91 struct client_call *next; /* the next call */
92 struct afb_proto_ws *protows; /* the proto_ws */
93 void *request; /* the request closure */
94 uint32_t callid; /* the message identifier */
98 * structure for a ws request
100 struct afb_proto_ws_call {
101 struct client_call *next; /* the next call */
102 struct afb_proto_ws *protows; /* the client of the request */
103 uint32_t refcount; /* reference count */
104 uint32_t callid; /* the incoming request callid */
105 char *buffer; /* the incoming buffer */
109 * structure for recording describe requests
111 struct client_describe
113 struct client_describe *next;
114 struct afb_proto_ws *protows;
115 void (*callback)(void*, struct json_object*);
121 * structure for jobs of describing
123 struct afb_proto_ws_describe
125 struct afb_proto_ws *protows;
129 /******************* proto description for client or servers ******************/
133 /* count of references */
136 /* file descriptor */
139 /* resource control */
140 pthread_mutex_t mutex;
145 /* the client closure */
148 /* the client side interface */
149 const struct afb_proto_ws_client_itf *client_itf;
151 /* the server side interface */
152 const struct afb_proto_ws_server_itf *server_itf;
154 /* emitted calls (client side) */
155 struct client_call *calls;
157 /* pending description (client side) */
158 struct client_describe *describes;
160 /* on hangup callback */
161 void (*on_hangup)(void *closure);
163 /* queuing facility for processing messages */
164 int (*queuing)(void (*process)(int s, void *c), void *closure);
167 /******************* streaming objects **********************************/
169 #define WRITEBUF_COUNT_MAX 32
172 struct iovec iovec[WRITEBUF_COUNT_MAX];
173 uint32_t uints[WRITEBUF_COUNT_MAX];
179 char *base, *head, *end;
184 struct afb_proto_ws *protows;
188 /******************* common useful tools **********************************/
191 * translate a pointer to some integer
192 * @param ptr the pointer to translate
195 static inline uint32_t ptr2id(void *ptr)
197 return (uint32_t)(((intptr_t)ptr) >> 6);
200 /******************* serialisation part **********************************/
202 static char *readbuf_get(struct readbuf *rb, uint32_t length)
204 char *before = rb->head;
205 char *after = before + length;
212 __attribute__((unused))
213 static int readbuf_char(struct readbuf *rb, char *value)
215 if (rb->head >= rb->end)
217 *value = *rb->head++;
221 static int readbuf_uint32(struct readbuf *rb, uint32_t *value)
223 char *after = rb->head + sizeof *value;
226 memcpy(value, rb->head, sizeof *value);
228 *value = le32toh(*value);
232 static int _readbuf_string_(struct readbuf *rb, const char **value, size_t *length, int nulok)
235 if (!readbuf_uint32(rb, &len))
246 *length = (size_t)(len - 1);
247 return (*value = readbuf_get(rb, len)) != NULL && rb->head[-1] == 0;
251 static int readbuf_string(struct readbuf *rb, const char **value, size_t *length)
253 return _readbuf_string_(rb, value, length, 0);
256 static int readbuf_nullstring(struct readbuf *rb, const char **value, size_t *length)
258 return _readbuf_string_(rb, value, length, 1);
261 static int readbuf_object(struct readbuf *rb, struct json_object **object)
264 struct json_object *o;
265 enum json_tokener_error jerr;
266 int rc = readbuf_string(rb, &string, NULL);
268 o = json_tokener_parse_verbose(string, &jerr);
269 if (jerr != json_tokener_success)
270 o = json_object_new_string(string);
276 static int writebuf_put(struct writebuf *wb, const void *value, size_t length)
279 if (i == WRITEBUF_COUNT_MAX)
281 wb->iovec[i].iov_base = (void*)value;
282 wb->iovec[i].iov_len = length;
287 static int writebuf_char(struct writebuf *wb, char value)
290 if (i == WRITEBUF_COUNT_MAX)
292 *(char*)&wb->uints[i] = value;
293 wb->iovec[i].iov_base = &wb->uints[i];
294 wb->iovec[i].iov_len = 1;
299 static int writebuf_uint32(struct writebuf *wb, uint32_t value)
302 if (i == WRITEBUF_COUNT_MAX)
304 wb->uints[i] = htole32(value);
305 wb->iovec[i].iov_base = &wb->uints[i];
306 wb->iovec[i].iov_len = sizeof wb->uints[i];
311 static int writebuf_string_length(struct writebuf *wb, const char *value, size_t length)
313 uint32_t len = (uint32_t)++length;
314 return (size_t)len == length && len && writebuf_uint32(wb, len) && writebuf_put(wb, value, length);
317 static int writebuf_string(struct writebuf *wb, const char *value)
319 return writebuf_string_length(wb, value, strlen(value));
322 static int writebuf_nullstring(struct writebuf *wb, const char *value)
324 return value ? writebuf_string_length(wb, value, strlen(value)) : writebuf_uint32(wb, 0);
327 static int writebuf_object(struct writebuf *wb, struct json_object *object)
329 const char *string = json_object_to_json_string_ext(object, JSON_C_TO_STRING_PLAIN);
330 return string != NULL && writebuf_string(wb, string);
333 /******************* queuing of messages *****************/
335 /* queue the processing of the received message (except if size=0 cause it's not a valid message) */
336 static void queue_message_processing(struct afb_proto_ws *protows, char *data, size_t size, void (*processing)(int,void*))
338 struct binary *binary;
341 binary = malloc(sizeof *binary);
343 /* TODO process the problem */
346 binary->protows = protows;
347 binary->rb.base = data;
348 binary->rb.head = data;
349 binary->rb.end = data + size;
350 if (!protows->queuing
351 || protows->queuing(processing, binary) < 0)
352 processing(0, binary);
359 /******************* ws request part for server *****************/
361 void afb_proto_ws_call_addref(struct afb_proto_ws_call *call)
363 __atomic_add_fetch(&call->refcount, 1, __ATOMIC_RELAXED);
366 void afb_proto_ws_call_unref(struct afb_proto_ws_call *call)
368 if (__atomic_sub_fetch(&call->refcount, 1, __ATOMIC_RELAXED))
371 afb_proto_ws_unref(call->protows);
376 int afb_proto_ws_call_reply(struct afb_proto_ws_call *call, struct json_object *obj, const char *error, const char *info)
379 struct writebuf wb = { .count = 0 };
380 struct afb_proto_ws *protows = call->protows;
382 if (writebuf_char(&wb, CHAR_FOR_REPLY)
383 && writebuf_uint32(&wb, call->callid)
384 && writebuf_nullstring(&wb, error)
385 && writebuf_nullstring(&wb, info)
386 && writebuf_object(&wb, obj)) {
387 pthread_mutex_lock(&protows->mutex);
388 rc = afb_ws_binary_v(protows->ws, wb.iovec, wb.count);
389 pthread_mutex_unlock(&protows->mutex);
399 int afb_proto_ws_call_subscribe(struct afb_proto_ws_call *call, const char *event_name, int event_id)
402 struct writebuf wb = { .count = 0 };
403 struct afb_proto_ws *protows = call->protows;
405 if (writebuf_char(&wb, CHAR_FOR_EVT_SUBSCRIBE)
406 && writebuf_uint32(&wb, call->callid)
407 && writebuf_uint32(&wb, (uint32_t)event_id)
408 && writebuf_string(&wb, event_name)) {
409 pthread_mutex_lock(&protows->mutex);
410 rc = afb_ws_binary_v(protows->ws, wb.iovec, wb.count);
411 pthread_mutex_unlock(&protows->mutex);
421 int afb_proto_ws_call_unsubscribe(struct afb_proto_ws_call *call, const char *event_name, int event_id)
424 struct writebuf wb = { .count = 0 };
425 struct afb_proto_ws *protows = call->protows;
427 if (writebuf_char(&wb, CHAR_FOR_EVT_UNSUBSCRIBE)
428 && writebuf_uint32(&wb, call->callid)
429 && writebuf_uint32(&wb, (uint32_t)event_id)
430 && writebuf_string(&wb, event_name)) {
431 pthread_mutex_lock(&protows->mutex);
432 rc = afb_ws_binary_v(protows->ws, wb.iovec, wb.count);
433 pthread_mutex_unlock(&protows->mutex);
443 /******************* client part **********************************/
445 /* search a memorized call */
446 static struct client_call *client_call_search_locked(struct afb_proto_ws *protows, uint32_t callid)
448 struct client_call *call;
450 call = protows->calls;
451 while (call != NULL && call->callid != callid)
457 static struct client_call *client_call_search_unlocked(struct afb_proto_ws *protows, uint32_t callid)
459 struct client_call *result;
461 pthread_mutex_lock(&protows->mutex);
462 result = client_call_search_locked(protows, callid);
463 pthread_mutex_unlock(&protows->mutex);
467 /* free and release the memorizing call */
468 static void client_call_destroy(struct client_call *call)
470 struct client_call **prv;
471 struct afb_proto_ws *protows = call->protows;
473 pthread_mutex_lock(&protows->mutex);
474 prv = &call->protows->calls;
475 while (*prv != NULL) {
482 pthread_mutex_unlock(&protows->mutex);
486 /* get event data from the message */
487 static int client_msg_event_read(struct readbuf *rb, uint32_t *eventid, const char **name)
489 return readbuf_uint32(rb, eventid) && readbuf_string(rb, name, NULL);
492 /* get event from the message */
493 static int client_msg_call_get(struct afb_proto_ws *protows, struct readbuf *rb, struct client_call **call)
497 /* get event data from the message */
498 if (!readbuf_uint32(rb, &callid)) {
503 *call = client_call_search_unlocked(protows, callid);
512 static void client_on_event_create(struct afb_proto_ws *protows, struct readbuf *rb)
514 const char *event_name;
517 if (protows->client_itf->on_event_create && client_msg_event_read(rb, &event_id, &event_name))
518 protows->client_itf->on_event_create(protows->closure, event_name, (int)event_id);
520 ERROR("Ignoring creation of event");
523 /* removes an event */
524 static void client_on_event_remove(struct afb_proto_ws *protows, struct readbuf *rb)
526 const char *event_name;
529 if (protows->client_itf->on_event_remove && client_msg_event_read(rb, &event_id, &event_name))
530 protows->client_itf->on_event_remove(protows->closure, event_name, (int)event_id);
532 ERROR("Ignoring deletion of event");
535 /* subscribes an event */
536 static void client_on_event_subscribe(struct afb_proto_ws *protows, struct readbuf *rb)
538 const char *event_name;
540 struct client_call *call;
542 if (protows->client_itf->on_event_subscribe && client_msg_call_get(protows, rb, &call) && client_msg_event_read(rb, &event_id, &event_name))
543 protows->client_itf->on_event_subscribe(protows->closure, call->request, event_name, (int)event_id);
545 ERROR("Ignoring subscription to event");
548 /* unsubscribes an event */
549 static void client_on_event_unsubscribe(struct afb_proto_ws *protows, struct readbuf *rb)
551 const char *event_name;
553 struct client_call *call;
555 if (protows->client_itf->on_event_unsubscribe && client_msg_call_get(protows, rb, &call) && client_msg_event_read(rb, &event_id, &event_name))
556 protows->client_itf->on_event_unsubscribe(protows->closure, call->request, event_name, (int)event_id);
558 ERROR("Ignoring unsubscription to event");
561 /* receives broadcasted events */
562 static void client_on_event_broadcast(struct afb_proto_ws *protows, struct readbuf *rb)
564 const char *event_name;
565 struct json_object *object;
567 if (protows->client_itf->on_event_broadcast && readbuf_string(rb, &event_name, NULL) && readbuf_object(rb, &object))
568 protows->client_itf->on_event_broadcast(protows->closure, event_name, object);
570 ERROR("Ignoring broadcast of event");
574 static void client_on_event_push(struct afb_proto_ws *protows, struct readbuf *rb)
576 const char *event_name;
578 struct json_object *object;
580 if (protows->client_itf->on_event_push && client_msg_event_read(rb, &event_id, &event_name) && readbuf_object(rb, &object))
581 protows->client_itf->on_event_push(protows->closure, event_name, (int)event_id, object);
583 ERROR("Ignoring push of event");
586 static void client_on_reply(struct afb_proto_ws *protows, struct readbuf *rb)
588 struct client_call *call;
589 struct json_object *object;
590 const char *error, *info;
592 if (!client_msg_call_get(protows, rb, &call))
595 if (readbuf_nullstring(rb, &error, NULL) && readbuf_nullstring(rb, &info, NULL) && readbuf_object(rb, &object)) {
596 protows->client_itf->on_reply(protows->closure, call->request, object, error, info);
598 protows->client_itf->on_reply(protows->closure, call->request, NULL, "proto-error", "can't process success");
600 client_call_destroy(call);
603 static void client_on_description(struct afb_proto_ws *protows, struct readbuf *rb)
606 struct client_describe *desc, **prv;
607 struct json_object *object;
609 if (readbuf_uint32(rb, &descid)) {
610 pthread_mutex_lock(&protows->mutex);
611 prv = &protows->describes;
612 while ((desc = *prv) && desc->descid != descid)
615 pthread_mutex_unlock(&protows->mutex);
618 pthread_mutex_unlock(&protows->mutex);
619 if (!readbuf_object(rb, &object))
621 desc->callback(desc->closure, object);
627 /* callback when receiving binary data */
628 static void client_on_binary_job(int sig, void *closure)
630 struct binary *binary = closure;
633 switch (*binary->rb.head++) {
634 case CHAR_FOR_REPLY: /* reply */
635 client_on_reply(binary->protows, &binary->rb);
637 case CHAR_FOR_EVT_BROADCAST: /* broadcast */
638 client_on_event_broadcast(binary->protows, &binary->rb);
640 case CHAR_FOR_EVT_ADD: /* creates the event */
641 client_on_event_create(binary->protows, &binary->rb);
643 case CHAR_FOR_EVT_DEL: /* removes the event */
644 client_on_event_remove(binary->protows, &binary->rb);
646 case CHAR_FOR_EVT_PUSH: /* pushs the event */
647 client_on_event_push(binary->protows, &binary->rb);
649 case CHAR_FOR_EVT_SUBSCRIBE: /* subscribe event for a request */
650 client_on_event_subscribe(binary->protows, &binary->rb);
652 case CHAR_FOR_EVT_UNSUBSCRIBE: /* unsubscribe event for a request */
653 client_on_event_unsubscribe(binary->protows, &binary->rb);
655 case CHAR_FOR_DESCRIPTION: /* description */
656 client_on_description(binary->protows, &binary->rb);
658 default: /* unexpected message */
659 /* TODO: close the connection */
663 free(binary->rb.base);
667 /* callback when receiving binary data */
668 static void client_on_binary(void *closure, char *data, size_t size)
670 struct afb_proto_ws *protows = closure;
672 queue_message_processing(protows, data, size, client_on_binary_job);
675 int afb_proto_ws_client_call(
676 struct afb_proto_ws *protows,
678 struct json_object *args,
679 const char *sessionid,
681 const char *user_creds
685 struct client_call *call;
686 struct writebuf wb = { .count = 0 };
688 /* allocate call data */
689 call = malloc(sizeof *call);
694 call->request = request;
697 pthread_mutex_lock(&protows->mutex);
698 call->callid = ptr2id(call);
699 while(client_call_search_locked(protows, call->callid) != NULL)
701 call->protows = protows;
702 call->next = protows->calls;
703 protows->calls = call;
704 pthread_mutex_unlock(&protows->mutex);
706 /* creates the call message */
707 if (!writebuf_char(&wb, CHAR_FOR_CALL)
708 || !writebuf_uint32(&wb, call->callid)
709 || !writebuf_string(&wb, verb)
710 || !writebuf_string(&wb, sessionid)
711 || !writebuf_object(&wb, args)
712 || !writebuf_nullstring(&wb, user_creds)) {
718 pthread_mutex_lock(&protows->mutex);
719 rc = afb_ws_binary_v(protows->ws, wb.iovec, wb.count);
720 pthread_mutex_unlock(&protows->mutex);
727 client_call_destroy(call);
732 /* get the description */
733 int afb_proto_ws_client_describe(struct afb_proto_ws *protows, void (*callback)(void*, struct json_object*), void *closure)
735 struct client_describe *desc, *d;
736 struct writebuf wb = { .count = 0 };
738 desc = malloc(sizeof *desc);
744 /* fill in stack the description of the task */
745 pthread_mutex_lock(&protows->mutex);
746 desc->descid = ptr2id(desc);
747 d = protows->describes;
749 if (d->descid != desc->descid)
753 d = protows->describes;
756 desc->callback = callback;
757 desc->closure = closure;
758 desc->protows = protows;
759 desc->next = protows->describes;
760 protows->describes = desc;
763 if (writebuf_char(&wb, CHAR_FOR_DESCRIBE)
764 && writebuf_uint32(&wb, desc->descid)
765 && afb_ws_binary_v(protows->ws, wb.iovec, wb.count) >= 0) {
766 pthread_mutex_unlock(&protows->mutex);
770 d = protows->describes;
772 protows->describes = desc->next;
774 while(d && d->next != desc)
777 d->next = desc->next;
779 pthread_mutex_unlock(&protows->mutex);
782 /* TODO? callback(closure, NULL); */
786 /******************* client description part for server *****************************/
788 /* on call, propagate it to the ws service */
789 static void server_on_call(struct afb_proto_ws *protows, struct readbuf *rb)
791 struct afb_proto_ws_call *call;
792 const char *uuid, *verb, *user_creds;
795 struct json_object *object;
797 afb_proto_ws_addref(protows);
799 /* reads the call message data */
800 if (!readbuf_uint32(rb, &callid)
801 || !readbuf_string(rb, &verb, &lenverb)
802 || !readbuf_string(rb, &uuid, NULL)
803 || !readbuf_object(rb, &object)
804 || !readbuf_nullstring(rb, &user_creds, NULL))
807 /* create the request */
808 call = malloc(sizeof *call);
812 call->protows = protows;
813 call->callid = callid;
815 call->buffer = rb->base;
816 rb->base = NULL; /* don't free the buffer */
818 protows->server_itf->on_call(protows->closure, call, verb, object, uuid, user_creds);
822 json_object_put(object);
825 afb_proto_ws_unref(protows);
828 static int server_send_description(struct afb_proto_ws *protows, uint32_t descid, struct json_object *descobj)
831 struct writebuf wb = { .count = 0 };
833 if (writebuf_char(&wb, CHAR_FOR_DESCRIPTION)
834 && writebuf_uint32(&wb, descid)
835 && writebuf_object(&wb, descobj)) {
836 pthread_mutex_lock(&protows->mutex);
837 rc = afb_ws_binary_v(protows->ws, wb.iovec, wb.count);
838 pthread_mutex_unlock(&protows->mutex);
845 int afb_proto_ws_describe_put(struct afb_proto_ws_describe *describe, struct json_object *description)
847 int rc = server_send_description(describe->protows, describe->descid, description);
848 afb_proto_ws_unref(describe->protows);
853 /* on describe, propagate it to the ws service */
854 static void server_on_describe(struct afb_proto_ws *protows, struct readbuf *rb)
857 struct afb_proto_ws_describe *desc;
859 /* reads the descid */
860 if (readbuf_uint32(rb, &descid)) {
861 if (protows->server_itf->on_describe) {
862 /* create asynchronous job */
863 desc = malloc(sizeof *desc);
865 desc->descid = descid;
866 desc->protows = protows;
867 afb_proto_ws_addref(protows);
868 protows->server_itf->on_describe(protows->closure, desc);
872 server_send_description(protows, descid, NULL);
876 /* callback when receiving binary data */
877 static void server_on_binary_job(int sig, void *closure)
879 struct binary *binary = closure;
882 switch (*binary->rb.head++) {
884 server_on_call(binary->protows, &binary->rb);
886 case CHAR_FOR_DESCRIBE:
887 server_on_describe(binary->protows, &binary->rb);
889 default: /* unexpected message */
890 /* TODO: close the connection */
894 free(binary->rb.base);
898 static void server_on_binary(void *closure, char *data, size_t size)
900 struct afb_proto_ws *protows = closure;
902 queue_message_processing(protows, data, size, server_on_binary_job);
905 /******************* server part: manage events **********************************/
907 static int server_event_send(struct afb_proto_ws *protows, char order, const char *event_name, int event_id, struct json_object *data)
909 struct writebuf wb = { .count = 0 };
912 if (writebuf_char(&wb, order)
913 && (order == CHAR_FOR_EVT_BROADCAST || writebuf_uint32(&wb, event_id))
914 && writebuf_string(&wb, event_name)
915 && (order == CHAR_FOR_EVT_ADD || order == CHAR_FOR_EVT_DEL || writebuf_object(&wb, data))) {
916 pthread_mutex_lock(&protows->mutex);
917 rc = afb_ws_binary_v(protows->ws, wb.iovec, wb.count);
918 pthread_mutex_unlock(&protows->mutex);
925 int afb_proto_ws_server_event_create(struct afb_proto_ws *protows, const char *event_name, int event_id)
927 return server_event_send(protows, CHAR_FOR_EVT_ADD, event_name, event_id, NULL);
930 int afb_proto_ws_server_event_remove(struct afb_proto_ws *protows, const char *event_name, int event_id)
932 return server_event_send(protows, CHAR_FOR_EVT_DEL, event_name, event_id, NULL);
935 int afb_proto_ws_server_event_push(struct afb_proto_ws *protows, const char *event_name, int event_id, struct json_object *data)
937 return server_event_send(protows, CHAR_FOR_EVT_PUSH, event_name, event_id, data);
940 int afb_proto_ws_server_event_broadcast(struct afb_proto_ws *protows, const char *event_name, struct json_object *data)
942 return server_event_send(protows, CHAR_FOR_EVT_BROADCAST, event_name, 0, data);
945 /*****************************************************/
947 /* callback when receiving a hangup */
948 static void on_hangup(void *closure)
950 struct afb_proto_ws *protows = closure;
951 struct client_describe *cd, *ncd;
952 struct client_call *call, *ncall;
954 ncd = __atomic_exchange_n(&protows->describes, NULL, __ATOMIC_RELAXED);
955 ncall = __atomic_exchange_n(&protows->calls, NULL, __ATOMIC_RELAXED);
960 protows->client_itf->on_reply(protows->closure, call->request, NULL, "disconnected", "server hung up");
967 cd->callback(cd->closure, NULL);
972 fdev_unref(protows->fdev);
974 if (protows->on_hangup)
975 protows->on_hangup(protows->closure);
979 /*****************************************************/
981 static const struct afb_ws_itf proto_ws_client_ws_itf =
985 .on_binary = client_on_binary,
987 .on_hangup = on_hangup
990 static const struct afb_ws_itf server_ws_itf =
994 .on_binary = server_on_binary,
996 .on_hangup = on_hangup
999 /*****************************************************/
1001 static struct afb_proto_ws *afb_proto_ws_create(struct fdev *fdev, const struct afb_proto_ws_server_itf *itfs, const struct afb_proto_ws_client_itf *itfc, void *closure, const struct afb_ws_itf *itf)
1003 struct afb_proto_ws *protows;
1005 protows = calloc(1, sizeof *protows);
1006 if (protows == NULL)
1009 fcntl(fdev_fd(fdev), F_SETFD, FD_CLOEXEC);
1010 fcntl(fdev_fd(fdev), F_SETFL, O_NONBLOCK);
1011 protows->ws = afb_ws_create(fdev, itf, protows);
1012 if (protows->ws != NULL) {
1013 protows->fdev = fdev;
1014 protows->refcount = 1;
1015 protows->closure = closure;
1016 protows->server_itf = itfs;
1017 protows->client_itf = itfc;
1018 pthread_mutex_init(&protows->mutex, NULL);
1026 struct afb_proto_ws *afb_proto_ws_create_client(struct fdev *fdev, const struct afb_proto_ws_client_itf *itf, void *closure)
1028 return afb_proto_ws_create(fdev, NULL, itf, closure, &proto_ws_client_ws_itf);
1031 struct afb_proto_ws *afb_proto_ws_create_server(struct fdev *fdev, const struct afb_proto_ws_server_itf *itf, void *closure)
1033 return afb_proto_ws_create(fdev, itf, NULL, closure, &server_ws_itf);
1036 void afb_proto_ws_unref(struct afb_proto_ws *protows)
1038 if (protows && !__atomic_sub_fetch(&protows->refcount, 1, __ATOMIC_RELAXED)) {
1039 afb_proto_ws_hangup(protows);
1040 afb_ws_destroy(protows->ws);
1041 pthread_mutex_destroy(&protows->mutex);
1046 void afb_proto_ws_addref(struct afb_proto_ws *protows)
1048 __atomic_add_fetch(&protows->refcount, 1, __ATOMIC_RELAXED);
1051 int afb_proto_ws_is_client(struct afb_proto_ws *protows)
1053 return !!protows->client_itf;
1056 int afb_proto_ws_is_server(struct afb_proto_ws *protows)
1058 return !!protows->server_itf;
1061 void afb_proto_ws_hangup(struct afb_proto_ws *protows)
1063 afb_ws_hangup(protows->ws);
1066 void afb_proto_ws_on_hangup(struct afb_proto_ws *protows, void (*on_hangup)(void *closure))
1068 protows->on_hangup = on_hangup;
1071 void afb_proto_ws_set_queuing(struct afb_proto_ws *protows, int (*queuing)(void (*)(int,void*), void*))
1073 protows->queuing = queuing;