1*ebfedea0SLionel Sambuc /* $NetBSD: net_read.c,v 1.1.1.1 2011/04/13 18:15:42 elric Exp $ */
2*ebfedea0SLionel Sambuc
3*ebfedea0SLionel Sambuc /*
4*ebfedea0SLionel Sambuc * Copyright (c) 1995, 1996, 1997, 1998 Kungliga Tekniska Högskolan
5*ebfedea0SLionel Sambuc * (Royal Institute of Technology, Stockholm, Sweden).
6*ebfedea0SLionel Sambuc * All rights reserved.
7*ebfedea0SLionel Sambuc *
8*ebfedea0SLionel Sambuc * Redistribution and use in source and binary forms, with or without
9*ebfedea0SLionel Sambuc * modification, are permitted provided that the following conditions
10*ebfedea0SLionel Sambuc * are met:
11*ebfedea0SLionel Sambuc *
12*ebfedea0SLionel Sambuc * 1. Redistributions of source code must retain the above copyright
13*ebfedea0SLionel Sambuc * notice, this list of conditions and the following disclaimer.
14*ebfedea0SLionel Sambuc *
15*ebfedea0SLionel Sambuc * 2. Redistributions in binary form must reproduce the above copyright
16*ebfedea0SLionel Sambuc * notice, this list of conditions and the following disclaimer in the
17*ebfedea0SLionel Sambuc * documentation and/or other materials provided with the distribution.
18*ebfedea0SLionel Sambuc *
19*ebfedea0SLionel Sambuc * 3. Neither the name of the Institute nor the names of its contributors
20*ebfedea0SLionel Sambuc * may be used to endorse or promote products derived from this software
21*ebfedea0SLionel Sambuc * without specific prior written permission.
22*ebfedea0SLionel Sambuc *
23*ebfedea0SLionel Sambuc * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
24*ebfedea0SLionel Sambuc * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25*ebfedea0SLionel Sambuc * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26*ebfedea0SLionel Sambuc * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
27*ebfedea0SLionel Sambuc * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28*ebfedea0SLionel Sambuc * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29*ebfedea0SLionel Sambuc * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30*ebfedea0SLionel Sambuc * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31*ebfedea0SLionel Sambuc * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32*ebfedea0SLionel Sambuc * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33*ebfedea0SLionel Sambuc * SUCH DAMAGE.
34*ebfedea0SLionel Sambuc */
35*ebfedea0SLionel Sambuc
36*ebfedea0SLionel Sambuc #include <config.h>
37*ebfedea0SLionel Sambuc
38*ebfedea0SLionel Sambuc #include <krb5/roken.h>
39*ebfedea0SLionel Sambuc
40*ebfedea0SLionel Sambuc /*
41*ebfedea0SLionel Sambuc * Like read but never return partial data.
42*ebfedea0SLionel Sambuc */
43*ebfedea0SLionel Sambuc
44*ebfedea0SLionel Sambuc #ifndef _WIN32
45*ebfedea0SLionel Sambuc
46*ebfedea0SLionel Sambuc ROKEN_LIB_FUNCTION ssize_t ROKEN_LIB_CALL
net_read(rk_socket_t fd,void * buf,size_t nbytes)47*ebfedea0SLionel Sambuc net_read (rk_socket_t fd, void *buf, size_t nbytes)
48*ebfedea0SLionel Sambuc {
49*ebfedea0SLionel Sambuc char *cbuf = (char *)buf;
50*ebfedea0SLionel Sambuc ssize_t count;
51*ebfedea0SLionel Sambuc size_t rem = nbytes;
52*ebfedea0SLionel Sambuc
53*ebfedea0SLionel Sambuc while (rem > 0) {
54*ebfedea0SLionel Sambuc count = read (fd, cbuf, rem);
55*ebfedea0SLionel Sambuc if (count < 0) {
56*ebfedea0SLionel Sambuc if (errno == EINTR)
57*ebfedea0SLionel Sambuc continue;
58*ebfedea0SLionel Sambuc else
59*ebfedea0SLionel Sambuc return count;
60*ebfedea0SLionel Sambuc } else if (count == 0) {
61*ebfedea0SLionel Sambuc return count;
62*ebfedea0SLionel Sambuc }
63*ebfedea0SLionel Sambuc cbuf += count;
64*ebfedea0SLionel Sambuc rem -= count;
65*ebfedea0SLionel Sambuc }
66*ebfedea0SLionel Sambuc return nbytes;
67*ebfedea0SLionel Sambuc }
68*ebfedea0SLionel Sambuc
69*ebfedea0SLionel Sambuc #else
70*ebfedea0SLionel Sambuc
71*ebfedea0SLionel Sambuc ROKEN_LIB_FUNCTION ssize_t ROKEN_LIB_CALL
net_read(rk_socket_t sock,void * buf,size_t nbytes)72*ebfedea0SLionel Sambuc net_read(rk_socket_t sock, void *buf, size_t nbytes)
73*ebfedea0SLionel Sambuc {
74*ebfedea0SLionel Sambuc char *cbuf = (char *)buf;
75*ebfedea0SLionel Sambuc ssize_t count;
76*ebfedea0SLionel Sambuc size_t rem = nbytes;
77*ebfedea0SLionel Sambuc
78*ebfedea0SLionel Sambuc #ifdef SOCKET_IS_NOT_AN_FD
79*ebfedea0SLionel Sambuc int use_read = 0;
80*ebfedea0SLionel Sambuc #endif
81*ebfedea0SLionel Sambuc
82*ebfedea0SLionel Sambuc while (rem > 0) {
83*ebfedea0SLionel Sambuc #ifdef SOCKET_IS_NOT_AN_FD
84*ebfedea0SLionel Sambuc if (use_read)
85*ebfedea0SLionel Sambuc count = _read (sock, cbuf, rem);
86*ebfedea0SLionel Sambuc else
87*ebfedea0SLionel Sambuc count = recv (sock, cbuf, rem, 0);
88*ebfedea0SLionel Sambuc
89*ebfedea0SLionel Sambuc if (use_read == 0 &&
90*ebfedea0SLionel Sambuc rk_IS_SOCKET_ERROR(count) &&
91*ebfedea0SLionel Sambuc (rk_SOCK_ERRNO == WSANOTINITIALISED ||
92*ebfedea0SLionel Sambuc rk_SOCK_ERRNO == WSAENOTSOCK)) {
93*ebfedea0SLionel Sambuc use_read = 1;
94*ebfedea0SLionel Sambuc
95*ebfedea0SLionel Sambuc count = _read (sock, cbuf, rem);
96*ebfedea0SLionel Sambuc }
97*ebfedea0SLionel Sambuc #else
98*ebfedea0SLionel Sambuc count = recv (sock, cbuf, rem, 0);
99*ebfedea0SLionel Sambuc #endif
100*ebfedea0SLionel Sambuc if (count < 0) {
101*ebfedea0SLionel Sambuc
102*ebfedea0SLionel Sambuc /* With WinSock, the error EINTR (WSAEINTR), is used to
103*ebfedea0SLionel Sambuc indicate that a blocking call was cancelled using
104*ebfedea0SLionel Sambuc WSACancelBlockingCall(). */
105*ebfedea0SLionel Sambuc
106*ebfedea0SLionel Sambuc #ifndef HAVE_WINSOCK
107*ebfedea0SLionel Sambuc if (rk_SOCK_ERRNO == EINTR)
108*ebfedea0SLionel Sambuc continue;
109*ebfedea0SLionel Sambuc #endif
110*ebfedea0SLionel Sambuc return count;
111*ebfedea0SLionel Sambuc } else if (count == 0) {
112*ebfedea0SLionel Sambuc return count;
113*ebfedea0SLionel Sambuc }
114*ebfedea0SLionel Sambuc cbuf += count;
115*ebfedea0SLionel Sambuc rem -= count;
116*ebfedea0SLionel Sambuc }
117*ebfedea0SLionel Sambuc return nbytes;
118*ebfedea0SLionel Sambuc }
119*ebfedea0SLionel Sambuc
120*ebfedea0SLionel Sambuc #endif
121