proto_common.c revision 207371 
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 207371 2010-04-29 15:36:32Z 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#define	MAX_SEND_SIZE	131072
47#endif
48
49int
50proto_common_send(int fd, const unsigned char *data, size_t size)
51{
52	ssize_t done;
53	size_t sendsize;
54
55	do {
56		sendsize = size < MAX_SEND_SIZE ? size : MAX_SEND_SIZE;
57		done = send(fd, data, sendsize, MSG_NOSIGNAL);
58		if (done == 0)
59			return (ENOTCONN);
60		else if (done < 0) {
61			if (errno == EINTR)
62				continue;
63			return (errno);
64		}
65		data += done;
66		size -= done;
67	} while (size > 0);
68
69	return (0);
70}
71
72int
73proto_common_recv(int fd, unsigned char *data, size_t size)
74{
75	ssize_t done;
76
77	do {
78		done = recv(fd, data, size, MSG_WAITALL);
79	} while (done == -1 && errno == EINTR);
80	if (done == 0)
81		return (ENOTCONN);
82	else if (done < 0)
83		return (errno);
84	return (0);
85}
86