1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
/*
Authors:
Jan Cholasta <jcholast@redhat.com>
Copyright (C) 2012 Red Hat
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include <stdint.h>
#include "util/atomic_io.h"
/* based on code from libssh <http://www.libssh.org> */
ssize_t sss_atomic_io_s(int fd, void *buf, size_t n, bool do_read)
{
char *b = buf;
size_t pos = 0;
ssize_t res;
struct pollfd pfd;
pfd.fd = fd;
pfd.events = do_read ? POLLIN : POLLOUT;
while (n > pos) {
if (do_read) {
res = read(fd, b + pos, n - pos);
} else {
res = write(fd, b + pos, n - pos);
}
switch (res) {
case -1:
if (errno == EINTR) {
continue;
}
if (errno == EAGAIN || errno == EWOULDBLOCK) {
(void) poll(&pfd, 1, -1);
continue;
}
return -1;
case 0:
/* read returns 0 on end-of-file */
errno = do_read ? 0 : EPIPE;
return pos;
default:
pos += (size_t) res;
}
}
return pos;
}
ssize_t sss_atomic_write_safe_s(int fd, void *buf, size_t len)
{
uint32_t ulen = len;
ssize_t ret;
ret = sss_atomic_write_s(fd, &ulen, sizeof(uint32_t));
if (ret != sizeof(uint32_t)) {
errno = errno == 0 ? EIO : errno;
return -1;
}
return sss_atomic_write_s(fd, buf, len);
}
ssize_t sss_atomic_read_safe_s(int fd, void *buf, size_t buf_len, size_t *_len)
{
uint32_t ulen = (uint32_t)-1;
ssize_t ret;
ret = sss_atomic_read_s(fd, &ulen, sizeof(uint32_t));
if (ret != sizeof(uint32_t)) {
errno = errno == 0 ? EIO : errno;
return -1;
}
if (ulen > buf_len) {
if (_len != NULL) {
*_len = ulen;
}
errno = ERANGE;
return -1;
}
if (_len != NULL) {
*_len = ulen;
}
return sss_atomic_read_s(fd, buf, ulen);
}
|
