comparison: ui/common/message.c
ui/common/message.c
- changeset 115
- e57ca2747782
equal
deleted
inserted
replaced
| 114:3da24640513a | 115:e57ca2747782 |
|---|---|
| 1 /* | |
| 2 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER. | |
| 3 * | |
| 4 * Copyright 2025 Olaf Wintermann. All rights reserved. | |
| 5 * | |
| 6 * Redistribution and use in source and binary forms, with or without | |
| 7 * modification, are permitted provided that the following conditions are met: | |
| 8 * | |
| 9 * 1. Redistributions of source code must retain the above copyright | |
| 10 * notice, this list of conditions and the following disclaimer. | |
| 11 * | |
| 12 * 2. Redistributions in binary form must reproduce the above copyright | |
| 13 * notice, this list of conditions and the following disclaimer in the | |
| 14 * documentation and/or other materials provided with the distribution. | |
| 15 * | |
| 16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" | |
| 17 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | |
| 18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE | |
| 19 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE | |
| 20 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR | |
| 21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF | |
| 22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS | |
| 23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN | |
| 24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) | |
| 25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE | |
| 26 * POSSIBILITY OF SUCH DAMAGE. | |
| 27 */ | |
| 28 | |
| 29 #ifndef _WIN32 | |
| 30 | |
| 31 #include <stdio.h> | |
| 32 #include <stdlib.h> | |
| 33 #include <unistd.h> | |
| 34 | |
| 35 #include "message.h" | |
| 36 | |
| 37 int uic_message_send_(UiMessageHandler *handler, cxstring msg) { | |
| 38 return handler->send(handler, msg); | |
| 39 } | |
| 40 | |
| 41 UiMessageHandler* uic_simple_msg_handler(int in, int out, msg_received_callback callback) { | |
| 42 UiSimpleMessageHandler *handler = malloc(sizeof(UiSimpleMessageHandler)); | |
| 43 handler->handler.start = uic_simple_msg_handler_start; | |
| 44 handler->handler.stop = uic_simple_msg_handler_stop; | |
| 45 handler->handler.send = uic_simple_msg_handler_send; | |
| 46 handler->handler.callback = callback; | |
| 47 handler->in = in; | |
| 48 handler->out = out; | |
| 49 handler->outbuf = cxBufferCreate(NULL, 4096, NULL, CX_BUFFER_FREE_CONTENTS | CX_BUFFER_AUTO_EXTEND); | |
| 50 handler->stop = 0; | |
| 51 pthread_mutex_init(&handler->queue_lock, NULL); | |
| 52 pthread_mutex_init(&handler->avlbl_lock, NULL); | |
| 53 pthread_cond_init(&handler->available, NULL); | |
| 54 return (UiMessageHandler*)handler; | |
| 55 } | |
| 56 | |
| 57 int uic_simple_msg_handler_start(UiMessageHandler *handler) { | |
| 58 UiSimpleMessageHandler *sh = (UiSimpleMessageHandler*)handler; | |
| 59 if(pthread_create(&sh->in_thread, NULL, uic_simple_msg_handler_in_thread, sh)) { | |
| 60 return 1; | |
| 61 } | |
| 62 if(pthread_create(&sh->out_thread, NULL, uic_simple_msg_handler_out_thread, sh)) { | |
| 63 return 1; | |
| 64 } | |
| 65 return 0; | |
| 66 } | |
| 67 | |
| 68 int uic_simple_msg_handler_stop(UiMessageHandler *handler) { | |
| 69 UiSimpleMessageHandler *sh = (UiSimpleMessageHandler*)handler; | |
| 70 pthread_mutex_lock(&sh->queue_lock); | |
| 71 sh->stop = 0; | |
| 72 pthread_cond_signal(&sh->available); | |
| 73 pthread_mutex_unlock(&sh->queue_lock); | |
| 74 close(sh->in); | |
| 75 sh->in = -1; | |
| 76 | |
| 77 pthread_join(sh->in_thread, NULL); | |
| 78 pthread_join(sh->out_thread, NULL); | |
| 79 | |
| 80 return 0; | |
| 81 } | |
| 82 | |
| 83 int uic_simple_msg_handler_send(UiMessageHandler *handler, cxstring msg) { | |
| 84 UiSimpleMessageHandler *sh = (UiSimpleMessageHandler*)handler; | |
| 85 pthread_mutex_lock(&sh->queue_lock); | |
| 86 char header[32]; | |
| 87 snprintf(header, 32, "%zu\n", msg.length); | |
| 88 cxBufferPutString(sh->outbuf, header); | |
| 89 cxBufferWrite(msg.ptr, 1, msg.length, sh->outbuf); | |
| 90 pthread_cond_signal(&sh->available); | |
| 91 pthread_mutex_unlock(&sh->queue_lock); | |
| 92 return 0; | |
| 93 } | |
| 94 | |
| 95 #define HEADERBUF_SIZE 64 | |
| 96 | |
| 97 void* uic_simple_msg_handler_in_thread(void *data) { | |
| 98 UiSimpleMessageHandler *handler = data; | |
| 99 | |
| 100 char *msg = NULL; | |
| 101 size_t msg_size = 0; | |
| 102 size_t msg_pos = 0; // currently received message length | |
| 103 | |
| 104 char headerbuf[HEADERBUF_SIZE]; | |
| 105 size_t headerpos = 0; | |
| 106 | |
| 107 char buf[2048]; | |
| 108 ssize_t r; | |
| 109 while((r = read(handler->in, buf, 2024)) > 0) { | |
| 110 char *buffer = buf; | |
| 111 size_t available = r; | |
| 112 | |
| 113 while(available > 0) { | |
| 114 if(msg) { | |
| 115 // read message | |
| 116 size_t need = msg_size - msg_pos; | |
| 117 size_t cplen = r > need ? need : available; | |
| 118 memcpy(msg+msg_pos, buffer, cplen); | |
| 119 buffer += cplen; | |
| 120 available -= cplen; | |
| 121 msg_pos += cplen; | |
| 122 if(msg_pos == msg_size) { | |
| 123 // message complete | |
| 124 //fprintf(stderr, "send: %.*s\n", (int)msg_size, msg); | |
| 125 if(handler->handler.callback) { | |
| 126 handler->handler.callback(cx_strn(msg, msg_size)); | |
| 127 } | |
| 128 free(msg); | |
| 129 msg = NULL; | |
| 130 msg_size = 0; | |
| 131 msg_pos = 0; | |
| 132 } | |
| 133 } else { | |
| 134 size_t header_max = HEADERBUF_SIZE - headerpos - 1; | |
| 135 if(header_max > available) { | |
| 136 header_max = available; | |
| 137 } | |
| 138 // search for line break | |
| 139 int i; | |
| 140 int header_complete = 0; | |
| 141 for(i=0;i<header_max;i++) { | |
| 142 if(buffer[i] == '\n') { | |
| 143 header_complete = 1; | |
| 144 break; | |
| 145 } | |
| 146 } | |
| 147 i++; | |
| 148 memcpy(headerbuf+headerpos, buffer, i); | |
| 149 headerpos += i; | |
| 150 buffer += i; | |
| 151 available -= i; | |
| 152 | |
| 153 if(header_complete) { | |
| 154 headerbuf[headerpos-1] = 0; // terminate buffer | |
| 155 char *end; | |
| 156 long length = strtol(headerbuf, &end, 10); | |
| 157 if(*end == '\0') { | |
| 158 //fprintf(stderr, "header: %d\n", (int)length); | |
| 159 msg = malloc(length); | |
| 160 msg_size = length; | |
| 161 headerpos = 0; | |
| 162 } else { | |
| 163 fprintf(stderr, "Error: invalid message {%s}\n", headerbuf); | |
| 164 } | |
| 165 } else if(headerpos+1 >= HEADERBUF_SIZE) { | |
| 166 fprintf(stderr, "Error: message header too big\n"); | |
| 167 exit(-1); | |
| 168 } | |
| 169 } | |
| 170 } | |
| 171 | |
| 172 | |
| 173 } | |
| 174 perror("error"); | |
| 175 fprintf(stderr, "stop simple_msg_handler_in_thread\n"); | |
| 176 | |
| 177 return NULL; | |
| 178 } | |
| 179 | |
| 180 void* uic_simple_msg_handler_out_thread(void *data) { | |
| 181 UiSimpleMessageHandler *handler = data; | |
| 182 CxBuffer *buffer = handler->outbuf; | |
| 183 | |
| 184 pthread_mutex_lock(&handler->queue_lock); | |
| 185 | |
| 186 for(;;) { | |
| 187 if(buffer->pos == 0) { | |
| 188 pthread_cond_wait(&handler->available, &handler->queue_lock); | |
| 189 continue; | |
| 190 } else { | |
| 191 size_t n = buffer->pos; | |
| 192 size_t pos = 0; | |
| 193 while(n > 0) { | |
| 194 ssize_t w = write(handler->out, buffer->space + pos, n); | |
| 195 if(w <= 0) { | |
| 196 fprintf(stderr, "Error: output error\n"); | |
| 197 break; | |
| 198 } | |
| 199 n -= w; | |
| 200 pos += w; | |
| 201 } | |
| 202 if(n > 0) { | |
| 203 break; // error | |
| 204 } | |
| 205 buffer->pos = 0; | |
| 206 } | |
| 207 } | |
| 208 | |
| 209 pthread_mutex_unlock(&handler->queue_lock); | |
| 210 | |
| 211 return NULL; | |
| 212 } | |
| 213 | |
| 214 #endif |
