sockstream.c revision 302408 
1/*-
2 * Copyright (c) 2015 Nahanni Systems, Inc.
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24 * SUCH DAMAGE.
25 *
26 * $FreeBSD: stable/11/usr.sbin/bhyve/sockstream.c 302408 2016-07-08 00:04:57Z gjb $
27 */
28
29#include <sys/cdefs.h>
30__FBSDID("$FreeBSD: stable/11/usr.sbin/bhyve/sockstream.c 302408 2016-07-08 00:04:57Z gjb $");
31
32#include <sys/types.h>
33#include <unistd.h>
34
35#include <errno.h>
36
37#include "sockstream.h"
38
39ssize_t
40stream_read(int fd, void *buf, ssize_t nbytes)
41{
42	uint8_t *p;
43	ssize_t len = 0;
44	ssize_t n;
45
46	p = buf;
47
48	while (len < nbytes) {
49		n = read(fd, p + len, nbytes - len);
50		if (n == 0)
51			break;
52
53		if (n < 0) {
54			if (errno == EINTR || errno == EAGAIN)
55				continue;
56			return (n);
57		}
58		len += n;
59	}
60	return (len);
61}
62
63ssize_t
64stream_write(int fd, const void *buf, ssize_t nbytes)
65{
66	const uint8_t *p;
67	ssize_t len = 0;
68	ssize_t n;
69
70	p = buf;
71
72	while (len < nbytes) {
73		n = write(fd, p + len, nbytes - len);
74		if (n == 0)
75			break;
76		if (n < 0) {
77			if (errno == EINTR || errno == EAGAIN)
78				continue;
79			return (n);
80		}
81		len += n;
82	}
83	return (len);
84}
85
86
87