ccmsg.c revision 1.5
1 /* $NetBSD: ccmsg.c,v 1.5 2021/02/19 16:42:21 christos Exp $ */ 2 3 /* 4 * Portions Copyright (C) Internet Systems Consortium, Inc. ("ISC") 5 * 6 * This Source Code Form is subject to the terms of the Mozilla Public 7 * License, v. 2.0. If a copy of the MPL was not distributed with this 8 * file, you can obtain one at https://mozilla.org/MPL/2.0/. 9 * 10 * See the COPYRIGHT file distributed with this work for additional 11 * information regarding copyright ownership. 12 * 13 * Portions Copyright (C) 2001 Nominum, Inc. 14 * 15 * Permission to use, copy, modify, and/or distribute this software for any 16 * purpose with or without fee is hereby granted, provided that the above 17 * copyright notice and this permission notice appear in all copies. 18 * 19 * THE SOFTWARE IS PROVIDED "AS IS" AND ISC AND NOMINUM DISCLAIMS ALL 20 * WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES 21 * OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR ANY 22 * SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES 23 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN 24 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF 25 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. 26 */ 27 28 /*! \file */ 29 30 #include <inttypes.h> 31 32 #include <isc/mem.h> 33 #include <isc/result.h> 34 #include <isc/task.h> 35 #include <isc/util.h> 36 37 #include <isccc/ccmsg.h> 38 #include <isccc/events.h> 39 40 #define CCMSG_MAGIC ISC_MAGIC('C', 'C', 'm', 's') 41 #define VALID_CCMSG(foo) ISC_MAGIC_VALID(foo, CCMSG_MAGIC) 42 43 static void 44 recv_length(isc_task_t *, isc_event_t *); 45 static void 46 recv_message(isc_task_t *, isc_event_t *); 47 48 static void 49 recv_length(isc_task_t *task, isc_event_t *ev_in) { 50 isc_socketevent_t *ev = (isc_socketevent_t *)ev_in; 51 isc_event_t *dev; 52 isccc_ccmsg_t *ccmsg = ev_in->ev_arg; 53 isc_region_t region; 54 isc_result_t result; 55 56 INSIST(VALID_CCMSG(ccmsg)); 57 58 dev = &ccmsg->event; 59 60 if (ev->result != ISC_R_SUCCESS) { 61 ccmsg->result = ev->result; 62 goto send_and_free; 63 } 64 65 /* 66 * Success. 67 */ 68 ccmsg->size = ntohl(ccmsg->size); 69 if (ccmsg->size == 0) { 70 ccmsg->result = ISC_R_UNEXPECTEDEND; 71 goto send_and_free; 72 } 73 if (ccmsg->size > ccmsg->maxsize) { 74 ccmsg->result = ISC_R_RANGE; 75 goto send_and_free; 76 } 77 78 region.base = isc_mem_get(ccmsg->mctx, ccmsg->size); 79 region.length = ccmsg->size; 80 if (region.base == NULL) { 81 ccmsg->result = ISC_R_NOMEMORY; 82 goto send_and_free; 83 } 84 85 isc_buffer_init(&ccmsg->buffer, region.base, region.length); 86 result = isc_socket_recv(ccmsg->sock, ®ion, 0, task, recv_message, 87 ccmsg); 88 if (result != ISC_R_SUCCESS) { 89 ccmsg->result = result; 90 goto send_and_free; 91 } 92 93 isc_event_free(&ev_in); 94 return; 95 96 send_and_free: 97 isc_task_send(ccmsg->task, &dev); 98 ccmsg->task = NULL; 99 isc_event_free(&ev_in); 100 return; 101 } 102 103 static void 104 recv_message(isc_task_t *task, isc_event_t *ev_in) { 105 isc_socketevent_t *ev = (isc_socketevent_t *)ev_in; 106 isc_event_t *dev; 107 isccc_ccmsg_t *ccmsg = ev_in->ev_arg; 108 109 (void)task; 110 111 INSIST(VALID_CCMSG(ccmsg)); 112 113 dev = &ccmsg->event; 114 115 if (ev->result != ISC_R_SUCCESS) { 116 ccmsg->result = ev->result; 117 goto send_and_free; 118 } 119 120 ccmsg->result = ISC_R_SUCCESS; 121 isc_buffer_add(&ccmsg->buffer, ev->n); 122 ccmsg->address = ev->address; 123 124 send_and_free: 125 isc_task_send(ccmsg->task, &dev); 126 ccmsg->task = NULL; 127 isc_event_free(&ev_in); 128 } 129 130 void 131 isccc_ccmsg_init(isc_mem_t *mctx, isc_socket_t *sock, isccc_ccmsg_t *ccmsg) { 132 REQUIRE(mctx != NULL); 133 REQUIRE(sock != NULL); 134 REQUIRE(ccmsg != NULL); 135 136 ccmsg->magic = CCMSG_MAGIC; 137 ccmsg->size = 0; 138 ccmsg->buffer.base = NULL; 139 ccmsg->buffer.length = 0; 140 ccmsg->maxsize = 4294967295U; /* Largest message possible. */ 141 ccmsg->mctx = mctx; 142 ccmsg->sock = sock; 143 ccmsg->task = NULL; /* None yet. */ 144 ccmsg->result = ISC_R_UNEXPECTED; /* None yet. */ 145 /* 146 * Should probably initialize the 147 *event here, but it can wait. 148 */ 149 } 150 151 void 152 isccc_ccmsg_setmaxsize(isccc_ccmsg_t *ccmsg, unsigned int maxsize) { 153 REQUIRE(VALID_CCMSG(ccmsg)); 154 155 ccmsg->maxsize = maxsize; 156 } 157 158 isc_result_t 159 isccc_ccmsg_readmessage(isccc_ccmsg_t *ccmsg, isc_task_t *task, 160 isc_taskaction_t action, void *arg) { 161 isc_result_t result; 162 isc_region_t region; 163 164 REQUIRE(VALID_CCMSG(ccmsg)); 165 REQUIRE(task != NULL); 166 REQUIRE(ccmsg->task == NULL); /* not currently in use */ 167 168 if (ccmsg->buffer.base != NULL) { 169 isc_mem_put(ccmsg->mctx, ccmsg->buffer.base, 170 ccmsg->buffer.length); 171 ccmsg->buffer.base = NULL; 172 ccmsg->buffer.length = 0; 173 } 174 175 ccmsg->task = task; 176 ccmsg->action = action; 177 ccmsg->arg = arg; 178 ccmsg->result = ISC_R_UNEXPECTED; /* unknown right now */ 179 180 ISC_EVENT_INIT(&ccmsg->event, sizeof(isc_event_t), 0, 0, 181 ISCCC_EVENT_CCMSG, action, arg, ccmsg, NULL, NULL); 182 183 region.base = (unsigned char *)&ccmsg->size; 184 region.length = 4; /* uint32_t */ 185 result = isc_socket_recv(ccmsg->sock, ®ion, 0, ccmsg->task, 186 recv_length, ccmsg); 187 188 if (result != ISC_R_SUCCESS) { 189 ccmsg->task = NULL; 190 } 191 192 return (result); 193 } 194 195 void 196 isccc_ccmsg_cancelread(isccc_ccmsg_t *ccmsg) { 197 REQUIRE(VALID_CCMSG(ccmsg)); 198 199 isc_socket_cancel(ccmsg->sock, NULL, ISC_SOCKCANCEL_RECV); 200 } 201 202 #if 0 203 void 204 isccc_ccmsg_freebuffer(isccc_ccmsg_t*ccmsg) { 205 REQUIRE(VALID_CCMSG(ccmsg)); 206 207 if (ccmsg->buffer.base == NULL) { 208 return; 209 } 210 211 isc_mem_put(ccmsg->mctx,ccmsg->buffer.base,ccmsg->buffer.length); 212 ccmsg->buffer.base = NULL; 213 ccmsg->buffer.length = 0; 214 } 215 #endif /* if 0 */ 216 217 void 218 isccc_ccmsg_invalidate(isccc_ccmsg_t *ccmsg) { 219 REQUIRE(VALID_CCMSG(ccmsg)); 220 221 ccmsg->magic = 0; 222 223 if (ccmsg->buffer.base != NULL) { 224 isc_mem_put(ccmsg->mctx, ccmsg->buffer.base, 225 ccmsg->buffer.length); 226 ccmsg->buffer.base = NULL; 227 ccmsg->buffer.length = 0; 228 } 229 } 230
Indexes created Mon Mar 02 05:31:46 UTC 2026