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