Branch data Line data Source code
1 : : /*
2 : : * Copyright (c) 2005 Anil Madhavapeddy. All rights reserved.
3 : : * Copyright (c) 1995,1999 Theo de Raadt. All rights reserved.
4 : : * 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
8 : : * are met:
9 : : * 1. Redistributions of source code must retain the above copyright
10 : : * notice, this list of conditions and the following disclaimer.
11 : : * 2. Redistributions in binary form must reproduce the above copyright
12 : : * notice, this list of conditions and the following disclaimer in the
13 : : * documentation and/or other materials provided with the distribution.
14 : : *
15 : : * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
16 : : * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
17 : : * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
18 : : * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
19 : : * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
20 : : * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
21 : : * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
22 : : * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23 : : * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
24 : : * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25 : : */
26 : :
27 : : #ifdef HAVE_CONFIG_H
28 : : #include "config.h"
29 : : #endif
30 : :
31 : : #include <stdlib.h>
32 : : #include <errno.h>
33 : : #include "atomicio.h"
34 : :
35 : : /*
36 : : * ensure all of data on socket comes through. f==read || f==vwrite
37 : : */
38 : : size_t
39 : 0 : atomicio(f, fd, _s, n)
40 : : ssize_t (*f) (int, void *, size_t);
41 : : int fd;
42 : : void *_s;
43 : : size_t n;
44 : : {
45 : 0 : char *s = _s;
46 : 0 : size_t pos = 0;
47 : : ssize_t res;
48 : :
49 [ # # ]: 0 : while (n > pos) {
50 : 0 : res = (f) (fd, s + pos, n - pos);
51 [ # # # ]: 0 : switch (res) {
52 : : case -1:
53 [ # # ]: 0 : if (errno == EINTR || errno == EAGAIN)
54 : 0 : continue;
55 : : return 0;
56 : : case 0:
57 : 0 : errno = EPIPE;
58 : 0 : return pos;
59 : : default:
60 : 0 : pos += (size_t)res;
61 : : }
62 : : }
63 : : return (pos);
64 : : }
65 : :
|