proto_common.c revision 211452 
1/*-
2 * Copyright (c) 2009-2010 The FreeBSD Foundation
3 * All rights reserved.
4 *
5 * This software was developed by Pawel Jakub Dawidek under sponsorship from
6 * the FreeBSD Foundation.
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
10 * are met:
11 * 1. Redistributions of source code must retain the above copyright
12 *    notice, this list of conditions and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 *    notice, this list of conditions and the following disclaimer in the
15 *    documentation and/or other materials provided with the distribution.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS ``AS IS'' AND
18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE
21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 * SUCH DAMAGE.
28 */
29
30#include <sys/cdefs.h>
31__FBSDID("$FreeBSD: head/sbin/hastd/proto_common.c 211452 2010-08-18 12:09:27Z pjd $");
32
33#include <sys/types.h>
34#include <sys/socket.h>
35
36#include <assert.h>
37#include <errno.h>
38#include <stdlib.h>
39#include <strings.h>
40
41#include "proto_impl.h"
42
43/* Maximum size of packet we want to use when sending data. */
44#ifndef MAX_SEND_SIZE
45#define	MAX_SEND_SIZE	32768
46#endif
47
48int
49proto_common_send(int fd, const unsigned char *data, size_t size)
50{
51	ssize_t done;
52	size_t sendsize;
53
54	do {
55		sendsize = size < MAX_SEND_SIZE ? size : MAX_SEND_SIZE;
56		done = send(fd, data, sendsize, MSG_NOSIGNAL);
57		if (done == 0)
58			return (ENOTCONN);
59		else if (done < 0) {
60			if (errno == EINTR)
61				continue;
62			return (errno);
63		}
64		data += done;
65		size -= done;
66	} while (size > 0);
67
68	return (0);
69}
70
71int
72proto_common_recv(int fd, unsigned char *data, size_t size)
73{
74	ssize_t done;
75
76	do {
77		done = recv(fd, data, size, MSG_WAITALL);
78	} while (done == -1 && errno == EINTR);
79	if (done == 0)
80		return (ENOTCONN);
81	else if (done < 0)
82		return (errno);
83	return (0);
84}
85