xref: /illumos-gate/usr/src/cmd/bhyve/sockstream.c (revision 32640292)
1bf21cd93STycho Nightingale /*-
2*32640292SAndy Fiddaman  * SPDX-License-Identifier: BSD-2-Clause
34c87aefeSPatrick Mooney  *
44c87aefeSPatrick Mooney  * Copyright (c) 2015 Nahanni Systems, Inc.
5bf21cd93STycho Nightingale  * All rights reserved.
6bf21cd93STycho Nightingale  *
7bf21cd93STycho Nightingale  * Redistribution and use in source and binary forms, with or without
8bf21cd93STycho Nightingale  * modification, are permitted provided that the following conditions
9bf21cd93STycho Nightingale  * are met:
10bf21cd93STycho Nightingale  * 1. Redistributions of source code must retain the above copyright
11bf21cd93STycho Nightingale  *    notice, this list of conditions and the following disclaimer.
12bf21cd93STycho Nightingale  * 2. Redistributions in binary form must reproduce the above copyright
13bf21cd93STycho Nightingale  *    notice, this list of conditions and the following disclaimer in the
14bf21cd93STycho Nightingale  *    documentation and/or other materials provided with the distribution.
15bf21cd93STycho Nightingale  *
164c87aefeSPatrick Mooney  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND
17bf21cd93STycho Nightingale  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18bf21cd93STycho Nightingale  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
194c87aefeSPatrick Mooney  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20bf21cd93STycho Nightingale  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21bf21cd93STycho Nightingale  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22bf21cd93STycho Nightingale  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23bf21cd93STycho Nightingale  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24bf21cd93STycho Nightingale  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25bf21cd93STycho Nightingale  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26bf21cd93STycho Nightingale  * SUCH DAMAGE.
27bf21cd93STycho Nightingale  */
28bf21cd93STycho Nightingale 
294c87aefeSPatrick Mooney #include <sys/cdefs.h>
304c87aefeSPatrick Mooney 
314c87aefeSPatrick Mooney #include <sys/types.h>
324c87aefeSPatrick Mooney #include <unistd.h>
334c87aefeSPatrick Mooney 
3438864087SAndy Fiddaman #ifndef	__FreeBSD__
3538864087SAndy Fiddaman #include <sys/socket.h>
3638864087SAndy Fiddaman #endif
3738864087SAndy Fiddaman 
384c87aefeSPatrick Mooney #include <errno.h>
394c87aefeSPatrick Mooney 
404c87aefeSPatrick Mooney #include "sockstream.h"
414c87aefeSPatrick Mooney 
424c87aefeSPatrick Mooney ssize_t
stream_read(int fd,void * buf,ssize_t nbytes)434c87aefeSPatrick Mooney stream_read(int fd, void *buf, ssize_t nbytes)
444c87aefeSPatrick Mooney {
454c87aefeSPatrick Mooney 	uint8_t *p;
464c87aefeSPatrick Mooney 	ssize_t len = 0;
474c87aefeSPatrick Mooney 	ssize_t n;
484c87aefeSPatrick Mooney 
494c87aefeSPatrick Mooney 	p = buf;
504c87aefeSPatrick Mooney 
514c87aefeSPatrick Mooney 	while (len < nbytes) {
524c87aefeSPatrick Mooney 		n = read(fd, p + len, nbytes - len);
534c87aefeSPatrick Mooney 		if (n == 0)
544c87aefeSPatrick Mooney 			break;
554c87aefeSPatrick Mooney 
564c87aefeSPatrick Mooney 		if (n < 0) {
574c87aefeSPatrick Mooney 			if (errno == EINTR || errno == EAGAIN)
584c87aefeSPatrick Mooney 				continue;
594c87aefeSPatrick Mooney 			return (n);
604c87aefeSPatrick Mooney 		}
614c87aefeSPatrick Mooney 		len += n;
624c87aefeSPatrick Mooney 	}
634c87aefeSPatrick Mooney 	return (len);
644c87aefeSPatrick Mooney }
654c87aefeSPatrick Mooney 
664c87aefeSPatrick Mooney ssize_t
stream_write(int fd,const void * buf,ssize_t nbytes)674c87aefeSPatrick Mooney stream_write(int fd, const void *buf, ssize_t nbytes)
684c87aefeSPatrick Mooney {
694c87aefeSPatrick Mooney 	const uint8_t *p;
704c87aefeSPatrick Mooney 	ssize_t len = 0;
714c87aefeSPatrick Mooney 	ssize_t n;
72bf21cd93STycho Nightingale 
734c87aefeSPatrick Mooney 	p = buf;
74bf21cd93STycho Nightingale 
754c87aefeSPatrick Mooney 	while (len < nbytes) {
7638864087SAndy Fiddaman #ifdef	__FreeBSD__
774c87aefeSPatrick Mooney 		n = write(fd, p + len, nbytes - len);
7838864087SAndy Fiddaman #else
7938864087SAndy Fiddaman 		n = send(fd, p + len, nbytes - len, MSG_NOSIGNAL);
8038864087SAndy Fiddaman #endif
814c87aefeSPatrick Mooney 		if (n == 0)
824c87aefeSPatrick Mooney 			break;
834c87aefeSPatrick Mooney 		if (n < 0) {
844c87aefeSPatrick Mooney 			if (errno == EINTR || errno == EAGAIN)
854c87aefeSPatrick Mooney 				continue;
864c87aefeSPatrick Mooney 			return (n);
874c87aefeSPatrick Mooney 		}
884c87aefeSPatrick Mooney 		len += n;
894c87aefeSPatrick Mooney 	}
904c87aefeSPatrick Mooney 	return (len);
914c87aefeSPatrick Mooney }
92