1*ebfedea0SLionel Sambuc /* $NetBSD: net_write.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 write 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_write(rk_socket_t fd,const void * buf,size_t nbytes)47*ebfedea0SLionel Sambuc net_write (rk_socket_t fd, const void *buf, size_t nbytes)
48*ebfedea0SLionel Sambuc {
49*ebfedea0SLionel Sambuc const char *cbuf = (const 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 = write (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 }
61*ebfedea0SLionel Sambuc cbuf += count;
62*ebfedea0SLionel Sambuc rem -= count;
63*ebfedea0SLionel Sambuc }
64*ebfedea0SLionel Sambuc return nbytes;
65*ebfedea0SLionel Sambuc }
66*ebfedea0SLionel Sambuc
67*ebfedea0SLionel Sambuc #else
68*ebfedea0SLionel Sambuc
69*ebfedea0SLionel Sambuc ROKEN_LIB_FUNCTION ssize_t ROKEN_LIB_CALL
net_write(rk_socket_t sock,const void * buf,size_t nbytes)70*ebfedea0SLionel Sambuc net_write(rk_socket_t sock, const void *buf, size_t nbytes)
71*ebfedea0SLionel Sambuc {
72*ebfedea0SLionel Sambuc const char *cbuf = (const char *)buf;
73*ebfedea0SLionel Sambuc ssize_t count;
74*ebfedea0SLionel Sambuc size_t rem = nbytes;
75*ebfedea0SLionel Sambuc #ifdef SOCKET_IS_NOT_AN_FD
76*ebfedea0SLionel Sambuc int use_write = 0;
77*ebfedea0SLionel Sambuc #endif
78*ebfedea0SLionel Sambuc
79*ebfedea0SLionel Sambuc while (rem > 0) {
80*ebfedea0SLionel Sambuc #ifdef SOCKET_IS_NOT_AN_FD
81*ebfedea0SLionel Sambuc if (use_write)
82*ebfedea0SLionel Sambuc count = _write (sock, cbuf, rem);
83*ebfedea0SLionel Sambuc else
84*ebfedea0SLionel Sambuc count = send (sock, cbuf, rem, 0);
85*ebfedea0SLionel Sambuc
86*ebfedea0SLionel Sambuc if (use_write == 0 &&
87*ebfedea0SLionel Sambuc rk_IS_SOCKET_ERROR(count) &&
88*ebfedea0SLionel Sambuc (rk_SOCK_ERRNO == WSANOTINITIALISED ||
89*ebfedea0SLionel Sambuc rk_SOCK_ERRNO == WSAENOTSOCK)) {
90*ebfedea0SLionel Sambuc use_write = 1;
91*ebfedea0SLionel Sambuc
92*ebfedea0SLionel Sambuc count = _write (sock, cbuf, rem);
93*ebfedea0SLionel Sambuc }
94*ebfedea0SLionel Sambuc #else
95*ebfedea0SLionel Sambuc count = send (sock, cbuf, rem, 0);
96*ebfedea0SLionel Sambuc #endif
97*ebfedea0SLionel Sambuc if (count < 0) {
98*ebfedea0SLionel Sambuc if (errno == EINTR)
99*ebfedea0SLionel Sambuc continue;
100*ebfedea0SLionel Sambuc else
101*ebfedea0SLionel Sambuc return count;
102*ebfedea0SLionel Sambuc }
103*ebfedea0SLionel Sambuc cbuf += count;
104*ebfedea0SLionel Sambuc rem -= count;
105*ebfedea0SLionel Sambuc }
106*ebfedea0SLionel Sambuc return nbytes;
107*ebfedea0SLionel Sambuc }
108*ebfedea0SLionel Sambuc
109*ebfedea0SLionel Sambuc #endif
110