s3:registry: fix seqnum race in regdb_fetch_keys_internal
[Samba.git] / lib / tsocket / tsocket.c
blob674858de0a5aa416c6ee92cdd7d464a117c76c7b
1 /*
2 Unix SMB/CIFS implementation.
4 Copyright (C) Stefan Metzmacher 2009
6 ** NOTE! The following LGPL license applies to the tsocket
7 ** library. This does NOT imply that all of Samba is released
8 ** under the LGPL
10 This library is free software; you can redistribute it and/or
11 modify it under the terms of the GNU Lesser General Public
12 License as published by the Free Software Foundation; either
13 version 3 of the License, or (at your option) any later version.
15 This library is distributed in the hope that it will be useful,
16 but WITHOUT ANY WARRANTY; without even the implied warranty of
17 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18 Lesser General Public License for more details.
20 You should have received a copy of the GNU Lesser General Public
21 License along with this library; if not, see <http://www.gnu.org/licenses/>.
24 #include "replace.h"
25 #include "system/filesys.h"
26 #include "tsocket.h"
27 #include "tsocket_internal.h"
29 int tsocket_simple_int_recv(struct tevent_req *req, int *perrno)
31 enum tevent_req_state state;
32 uint64_t error;
34 if (!tevent_req_is_error(req, &state, &error)) {
35 return 0;
38 switch (state) {
39 case TEVENT_REQ_NO_MEMORY:
40 *perrno = ENOMEM;
41 return -1;
42 case TEVENT_REQ_TIMED_OUT:
43 *perrno = ETIMEDOUT;
44 return -1;
45 case TEVENT_REQ_USER_ERROR:
46 *perrno = (int)error;
47 return -1;
48 default:
49 break;
52 *perrno = EIO;
53 return -1;
56 struct tsocket_address *_tsocket_address_create(TALLOC_CTX *mem_ctx,
57 const struct tsocket_address_ops *ops,
58 void *pstate,
59 size_t psize,
60 const char *type,
61 const char *location)
63 void **ppstate = (void **)pstate;
64 struct tsocket_address *addr;
66 addr = talloc_zero(mem_ctx, struct tsocket_address);
67 if (!addr) {
68 return NULL;
70 addr->ops = ops;
71 addr->location = location;
72 addr->private_data = talloc_size(addr, psize);
73 if (!addr->private_data) {
74 talloc_free(addr);
75 return NULL;
77 talloc_set_name_const(addr->private_data, type);
79 *ppstate = addr->private_data;
80 return addr;
83 char *tsocket_address_string(const struct tsocket_address *addr,
84 TALLOC_CTX *mem_ctx)
86 if (!addr) {
87 return talloc_strdup(mem_ctx, "NULL");
89 return addr->ops->string(addr, mem_ctx);
92 struct tsocket_address *_tsocket_address_copy(const struct tsocket_address *addr,
93 TALLOC_CTX *mem_ctx,
94 const char *location)
96 return addr->ops->copy(addr, mem_ctx, location);
99 struct tdgram_context {
100 const char *location;
101 const struct tdgram_context_ops *ops;
102 void *private_data;
104 struct tevent_req *recvfrom_req;
105 struct tevent_req *sendto_req;
108 static int tdgram_context_destructor(struct tdgram_context *dgram)
110 if (dgram->recvfrom_req) {
111 tevent_req_received(dgram->recvfrom_req);
114 if (dgram->sendto_req) {
115 tevent_req_received(dgram->sendto_req);
118 return 0;
121 struct tdgram_context *_tdgram_context_create(TALLOC_CTX *mem_ctx,
122 const struct tdgram_context_ops *ops,
123 void *pstate,
124 size_t psize,
125 const char *type,
126 const char *location)
128 struct tdgram_context *dgram;
129 void **ppstate = (void **)pstate;
130 void *state;
132 dgram = talloc(mem_ctx, struct tdgram_context);
133 if (dgram == NULL) {
134 return NULL;
136 dgram->location = location;
137 dgram->ops = ops;
138 dgram->recvfrom_req = NULL;
139 dgram->sendto_req = NULL;
141 state = talloc_size(dgram, psize);
142 if (state == NULL) {
143 talloc_free(dgram);
144 return NULL;
146 talloc_set_name_const(state, type);
148 dgram->private_data = state;
150 talloc_set_destructor(dgram, tdgram_context_destructor);
152 *ppstate = state;
153 return dgram;
156 void *_tdgram_context_data(struct tdgram_context *dgram)
158 return dgram->private_data;
161 struct tdgram_recvfrom_state {
162 const struct tdgram_context_ops *ops;
163 struct tdgram_context *dgram;
164 uint8_t *buf;
165 size_t len;
166 struct tsocket_address *src;
169 static int tdgram_recvfrom_destructor(struct tdgram_recvfrom_state *state)
171 if (state->dgram) {
172 state->dgram->recvfrom_req = NULL;
175 return 0;
178 static void tdgram_recvfrom_done(struct tevent_req *subreq);
180 struct tevent_req *tdgram_recvfrom_send(TALLOC_CTX *mem_ctx,
181 struct tevent_context *ev,
182 struct tdgram_context *dgram)
184 struct tevent_req *req;
185 struct tdgram_recvfrom_state *state;
186 struct tevent_req *subreq;
188 req = tevent_req_create(mem_ctx, &state,
189 struct tdgram_recvfrom_state);
190 if (req == NULL) {
191 return NULL;
194 state->ops = dgram->ops;
195 state->dgram = dgram;
196 state->buf = NULL;
197 state->len = 0;
198 state->src = NULL;
200 if (dgram->recvfrom_req) {
201 tevent_req_error(req, EBUSY);
202 goto post;
204 dgram->recvfrom_req = req;
206 talloc_set_destructor(state, tdgram_recvfrom_destructor);
208 subreq = state->ops->recvfrom_send(state, ev, dgram);
209 if (tevent_req_nomem(subreq, req)) {
210 goto post;
212 tevent_req_set_callback(subreq, tdgram_recvfrom_done, req);
214 return req;
216 post:
217 tevent_req_post(req, ev);
218 return req;
221 static void tdgram_recvfrom_done(struct tevent_req *subreq)
223 struct tevent_req *req = tevent_req_callback_data(subreq,
224 struct tevent_req);
225 struct tdgram_recvfrom_state *state = tevent_req_data(req,
226 struct tdgram_recvfrom_state);
227 ssize_t ret;
228 int sys_errno;
230 ret = state->ops->recvfrom_recv(subreq, &sys_errno, state,
231 &state->buf, &state->src);
232 if (ret == -1) {
233 tevent_req_error(req, sys_errno);
234 return;
237 state->len = ret;
239 tevent_req_done(req);
242 ssize_t tdgram_recvfrom_recv(struct tevent_req *req,
243 int *perrno,
244 TALLOC_CTX *mem_ctx,
245 uint8_t **buf,
246 struct tsocket_address **src)
248 struct tdgram_recvfrom_state *state = tevent_req_data(req,
249 struct tdgram_recvfrom_state);
250 ssize_t ret;
252 ret = tsocket_simple_int_recv(req, perrno);
253 if (ret == 0) {
254 *buf = talloc_move(mem_ctx, &state->buf);
255 ret = state->len;
256 if (src) {
257 *src = talloc_move(mem_ctx, &state->src);
261 tevent_req_received(req);
262 return ret;
265 struct tdgram_sendto_state {
266 const struct tdgram_context_ops *ops;
267 struct tdgram_context *dgram;
268 ssize_t ret;
271 static int tdgram_sendto_destructor(struct tdgram_sendto_state *state)
273 if (state->dgram) {
274 state->dgram->sendto_req = NULL;
277 return 0;
280 static void tdgram_sendto_done(struct tevent_req *subreq);
282 struct tevent_req *tdgram_sendto_send(TALLOC_CTX *mem_ctx,
283 struct tevent_context *ev,
284 struct tdgram_context *dgram,
285 const uint8_t *buf, size_t len,
286 const struct tsocket_address *dst)
288 struct tevent_req *req;
289 struct tdgram_sendto_state *state;
290 struct tevent_req *subreq;
292 req = tevent_req_create(mem_ctx, &state,
293 struct tdgram_sendto_state);
294 if (req == NULL) {
295 return NULL;
298 state->ops = dgram->ops;
299 state->dgram = dgram;
300 state->ret = -1;
302 if (len == 0) {
303 tevent_req_error(req, EINVAL);
304 goto post;
307 if (dgram->sendto_req) {
308 tevent_req_error(req, EBUSY);
309 goto post;
311 dgram->sendto_req = req;
313 talloc_set_destructor(state, tdgram_sendto_destructor);
315 subreq = state->ops->sendto_send(state, ev, dgram,
316 buf, len, dst);
317 if (tevent_req_nomem(subreq, req)) {
318 goto post;
320 tevent_req_set_callback(subreq, tdgram_sendto_done, req);
322 return req;
324 post:
325 tevent_req_post(req, ev);
326 return req;
329 static void tdgram_sendto_done(struct tevent_req *subreq)
331 struct tevent_req *req = tevent_req_callback_data(subreq,
332 struct tevent_req);
333 struct tdgram_sendto_state *state = tevent_req_data(req,
334 struct tdgram_sendto_state);
335 ssize_t ret;
336 int sys_errno;
338 ret = state->ops->sendto_recv(subreq, &sys_errno);
339 if (ret == -1) {
340 tevent_req_error(req, sys_errno);
341 return;
344 state->ret = ret;
346 tevent_req_done(req);
349 ssize_t tdgram_sendto_recv(struct tevent_req *req,
350 int *perrno)
352 struct tdgram_sendto_state *state = tevent_req_data(req,
353 struct tdgram_sendto_state);
354 ssize_t ret;
356 ret = tsocket_simple_int_recv(req, perrno);
357 if (ret == 0) {
358 ret = state->ret;
361 tevent_req_received(req);
362 return ret;
365 struct tdgram_disconnect_state {
366 const struct tdgram_context_ops *ops;
369 static void tdgram_disconnect_done(struct tevent_req *subreq);
371 struct tevent_req *tdgram_disconnect_send(TALLOC_CTX *mem_ctx,
372 struct tevent_context *ev,
373 struct tdgram_context *dgram)
375 struct tevent_req *req;
376 struct tdgram_disconnect_state *state;
377 struct tevent_req *subreq;
379 req = tevent_req_create(mem_ctx, &state,
380 struct tdgram_disconnect_state);
381 if (req == NULL) {
382 return NULL;
385 state->ops = dgram->ops;
387 if (dgram->recvfrom_req || dgram->sendto_req) {
388 tevent_req_error(req, EBUSY);
389 goto post;
392 subreq = state->ops->disconnect_send(state, ev, dgram);
393 if (tevent_req_nomem(subreq, req)) {
394 goto post;
396 tevent_req_set_callback(subreq, tdgram_disconnect_done, req);
398 return req;
400 post:
401 tevent_req_post(req, ev);
402 return req;
405 static void tdgram_disconnect_done(struct tevent_req *subreq)
407 struct tevent_req *req = tevent_req_callback_data(subreq,
408 struct tevent_req);
409 struct tdgram_disconnect_state *state = tevent_req_data(req,
410 struct tdgram_disconnect_state);
411 int ret;
412 int sys_errno;
414 ret = state->ops->disconnect_recv(subreq, &sys_errno);
415 if (ret == -1) {
416 tevent_req_error(req, sys_errno);
417 return;
420 tevent_req_done(req);
423 int tdgram_disconnect_recv(struct tevent_req *req,
424 int *perrno)
426 int ret;
428 ret = tsocket_simple_int_recv(req, perrno);
430 tevent_req_received(req);
431 return ret;
434 struct tstream_context {
435 const char *location;
436 const struct tstream_context_ops *ops;
437 void *private_data;
439 struct tevent_req *readv_req;
440 struct tevent_req *writev_req;
443 static int tstream_context_destructor(struct tstream_context *stream)
445 if (stream->readv_req) {
446 tevent_req_received(stream->readv_req);
449 if (stream->writev_req) {
450 tevent_req_received(stream->writev_req);
453 return 0;
456 struct tstream_context *_tstream_context_create(TALLOC_CTX *mem_ctx,
457 const struct tstream_context_ops *ops,
458 void *pstate,
459 size_t psize,
460 const char *type,
461 const char *location)
463 struct tstream_context *stream;
464 void **ppstate = (void **)pstate;
465 void *state;
467 stream = talloc(mem_ctx, struct tstream_context);
468 if (stream == NULL) {
469 return NULL;
471 stream->location = location;
472 stream->ops = ops;
473 stream->readv_req = NULL;
474 stream->writev_req = NULL;
476 state = talloc_size(stream, psize);
477 if (state == NULL) {
478 talloc_free(stream);
479 return NULL;
481 talloc_set_name_const(state, type);
483 stream->private_data = state;
485 talloc_set_destructor(stream, tstream_context_destructor);
487 *ppstate = state;
488 return stream;
491 void *_tstream_context_data(struct tstream_context *stream)
493 return stream->private_data;
496 ssize_t tstream_pending_bytes(struct tstream_context *stream)
498 return stream->ops->pending_bytes(stream);
501 struct tstream_readv_state {
502 const struct tstream_context_ops *ops;
503 struct tstream_context *stream;
504 int ret;
507 static int tstream_readv_destructor(struct tstream_readv_state *state)
509 if (state->stream) {
510 state->stream->readv_req = NULL;
513 return 0;
516 static void tstream_readv_done(struct tevent_req *subreq);
518 struct tevent_req *tstream_readv_send(TALLOC_CTX *mem_ctx,
519 struct tevent_context *ev,
520 struct tstream_context *stream,
521 struct iovec *vector,
522 size_t count)
524 struct tevent_req *req;
525 struct tstream_readv_state *state;
526 struct tevent_req *subreq;
527 int to_read = 0;
528 size_t i;
530 req = tevent_req_create(mem_ctx, &state,
531 struct tstream_readv_state);
532 if (req == NULL) {
533 return NULL;
536 state->ops = stream->ops;
537 state->stream = stream;
538 state->ret = -1;
540 /* first check if the input is ok */
541 #ifdef IOV_MAX
542 if (count > IOV_MAX) {
543 tevent_req_error(req, EMSGSIZE);
544 goto post;
546 #endif
548 for (i=0; i < count; i++) {
549 int tmp = to_read;
550 tmp += vector[i].iov_len;
552 if (tmp < to_read) {
553 tevent_req_error(req, EMSGSIZE);
554 goto post;
557 to_read = tmp;
560 if (to_read == 0) {
561 tevent_req_error(req, EINVAL);
562 goto post;
565 if (stream->readv_req) {
566 tevent_req_error(req, EBUSY);
567 goto post;
569 stream->readv_req = req;
571 talloc_set_destructor(state, tstream_readv_destructor);
573 subreq = state->ops->readv_send(state, ev, stream, vector, count);
574 if (tevent_req_nomem(subreq, req)) {
575 goto post;
577 tevent_req_set_callback(subreq, tstream_readv_done, req);
579 return req;
581 post:
582 tevent_req_post(req, ev);
583 return req;
586 static void tstream_readv_done(struct tevent_req *subreq)
588 struct tevent_req *req = tevent_req_callback_data(subreq,
589 struct tevent_req);
590 struct tstream_readv_state *state = tevent_req_data(req,
591 struct tstream_readv_state);
592 ssize_t ret;
593 int sys_errno;
595 ret = state->ops->readv_recv(subreq, &sys_errno);
596 TALLOC_FREE(subreq);
597 if (ret == -1) {
598 tevent_req_error(req, sys_errno);
599 return;
602 state->ret = ret;
604 tevent_req_done(req);
607 int tstream_readv_recv(struct tevent_req *req,
608 int *perrno)
610 struct tstream_readv_state *state = tevent_req_data(req,
611 struct tstream_readv_state);
612 int ret;
614 ret = tsocket_simple_int_recv(req, perrno);
615 if (ret == 0) {
616 ret = state->ret;
619 tevent_req_received(req);
620 return ret;
623 struct tstream_writev_state {
624 const struct tstream_context_ops *ops;
625 struct tstream_context *stream;
626 int ret;
629 static int tstream_writev_destructor(struct tstream_writev_state *state)
631 if (state->stream) {
632 state->stream->writev_req = NULL;
635 return 0;
638 static void tstream_writev_done(struct tevent_req *subreq);
640 struct tevent_req *tstream_writev_send(TALLOC_CTX *mem_ctx,
641 struct tevent_context *ev,
642 struct tstream_context *stream,
643 const struct iovec *vector,
644 size_t count)
646 struct tevent_req *req;
647 struct tstream_writev_state *state;
648 struct tevent_req *subreq;
649 int to_write = 0;
650 size_t i;
652 req = tevent_req_create(mem_ctx, &state,
653 struct tstream_writev_state);
654 if (req == NULL) {
655 return NULL;
658 state->ops = stream->ops;
659 state->stream = stream;
660 state->ret = -1;
662 /* first check if the input is ok */
663 #ifdef IOV_MAX
664 if (count > IOV_MAX) {
665 tevent_req_error(req, EMSGSIZE);
666 goto post;
668 #endif
670 for (i=0; i < count; i++) {
671 int tmp = to_write;
672 tmp += vector[i].iov_len;
674 if (tmp < to_write) {
675 tevent_req_error(req, EMSGSIZE);
676 goto post;
679 to_write = tmp;
682 if (to_write == 0) {
683 tevent_req_error(req, EINVAL);
684 goto post;
687 if (stream->writev_req) {
688 tevent_req_error(req, EBUSY);
689 goto post;
691 stream->writev_req = req;
693 talloc_set_destructor(state, tstream_writev_destructor);
695 subreq = state->ops->writev_send(state, ev, stream, vector, count);
696 if (tevent_req_nomem(subreq, req)) {
697 goto post;
699 tevent_req_set_callback(subreq, tstream_writev_done, req);
701 return req;
703 post:
704 tevent_req_post(req, ev);
705 return req;
708 static void tstream_writev_done(struct tevent_req *subreq)
710 struct tevent_req *req = tevent_req_callback_data(subreq,
711 struct tevent_req);
712 struct tstream_writev_state *state = tevent_req_data(req,
713 struct tstream_writev_state);
714 ssize_t ret;
715 int sys_errno;
717 ret = state->ops->writev_recv(subreq, &sys_errno);
718 if (ret == -1) {
719 tevent_req_error(req, sys_errno);
720 return;
723 state->ret = ret;
725 tevent_req_done(req);
728 int tstream_writev_recv(struct tevent_req *req,
729 int *perrno)
731 struct tstream_writev_state *state = tevent_req_data(req,
732 struct tstream_writev_state);
733 int ret;
735 ret = tsocket_simple_int_recv(req, perrno);
736 if (ret == 0) {
737 ret = state->ret;
740 tevent_req_received(req);
741 return ret;
744 struct tstream_disconnect_state {
745 const struct tstream_context_ops *ops;
748 static void tstream_disconnect_done(struct tevent_req *subreq);
750 struct tevent_req *tstream_disconnect_send(TALLOC_CTX *mem_ctx,
751 struct tevent_context *ev,
752 struct tstream_context *stream)
754 struct tevent_req *req;
755 struct tstream_disconnect_state *state;
756 struct tevent_req *subreq;
758 req = tevent_req_create(mem_ctx, &state,
759 struct tstream_disconnect_state);
760 if (req == NULL) {
761 return NULL;
764 state->ops = stream->ops;
766 if (stream->readv_req || stream->writev_req) {
767 tevent_req_error(req, EBUSY);
768 goto post;
771 subreq = state->ops->disconnect_send(state, ev, stream);
772 if (tevent_req_nomem(subreq, req)) {
773 goto post;
775 tevent_req_set_callback(subreq, tstream_disconnect_done, req);
777 return req;
779 post:
780 tevent_req_post(req, ev);
781 return req;
784 static void tstream_disconnect_done(struct tevent_req *subreq)
786 struct tevent_req *req = tevent_req_callback_data(subreq,
787 struct tevent_req);
788 struct tstream_disconnect_state *state = tevent_req_data(req,
789 struct tstream_disconnect_state);
790 int ret;
791 int sys_errno;
793 ret = state->ops->disconnect_recv(subreq, &sys_errno);
794 if (ret == -1) {
795 tevent_req_error(req, sys_errno);
796 return;
799 tevent_req_done(req);
802 int tstream_disconnect_recv(struct tevent_req *req,
803 int *perrno)
805 int ret;
807 ret = tsocket_simple_int_recv(req, perrno);
809 tevent_req_received(req);
810 return ret;