155682Smarkm/* 2233294Sstas * Copyright (c) 1995, 1996, 1997, 1998 Kungliga Tekniska H��gskolan 355682Smarkm * (Royal Institute of Technology, Stockholm, Sweden). 455682Smarkm * All rights reserved. 5233294Sstas * 655682Smarkm * Redistribution and use in source and binary forms, with or without 755682Smarkm * modification, are permitted provided that the following conditions 855682Smarkm * are met: 9233294Sstas * 1055682Smarkm * 1. Redistributions of source code must retain the above copyright 1155682Smarkm * notice, this list of conditions and the following disclaimer. 12233294Sstas * 1355682Smarkm * 2. Redistributions in binary form must reproduce the above copyright 1455682Smarkm * notice, this list of conditions and the following disclaimer in the 1555682Smarkm * documentation and/or other materials provided with the distribution. 16233294Sstas * 1755682Smarkm * 3. Neither the name of the Institute nor the names of its contributors 1855682Smarkm * may be used to endorse or promote products derived from this software 1955682Smarkm * without specific prior written permission. 20233294Sstas * 2155682Smarkm * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND 2255682Smarkm * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 2355682Smarkm * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 2455682Smarkm * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE 2555682Smarkm * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 2655682Smarkm * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 2755682Smarkm * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 2855682Smarkm * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 2955682Smarkm * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 3055682Smarkm * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 3155682Smarkm * SUCH DAMAGE. 3255682Smarkm */ 3355682Smarkm 3455682Smarkm#include <config.h> 3555682Smarkm 36178825Sdfr#include "roken.h" 3755682Smarkm 3855682Smarkm/* 3955682Smarkm * Like write but never return partial data. 4055682Smarkm */ 4155682Smarkm 42233294Sstas#ifndef _WIN32 43233294Sstas 44233294SstasROKEN_LIB_FUNCTION ssize_t ROKEN_LIB_CALL 45233294Sstasnet_write (rk_socket_t fd, const void *buf, size_t nbytes) 4655682Smarkm{ 4755682Smarkm const char *cbuf = (const char *)buf; 4855682Smarkm ssize_t count; 4955682Smarkm size_t rem = nbytes; 5055682Smarkm 5155682Smarkm while (rem > 0) { 52233294Sstas count = write (fd, cbuf, rem); 53233294Sstas if (count < 0) { 54233294Sstas if (errno == EINTR) 55233294Sstas continue; 56233294Sstas else 57233294Sstas return count; 58233294Sstas } 59233294Sstas cbuf += count; 60233294Sstas rem -= count; 61233294Sstas } 62233294Sstas return nbytes; 63233294Sstas} 64233294Sstas 6555682Smarkm#else 66233294Sstas 67233294SstasROKEN_LIB_FUNCTION ssize_t ROKEN_LIB_CALL 68233294Sstasnet_write(rk_socket_t sock, const void *buf, size_t nbytes) 69233294Sstas{ 70233294Sstas const char *cbuf = (const char *)buf; 71233294Sstas ssize_t count; 72233294Sstas size_t rem = nbytes; 73233294Sstas#ifdef SOCKET_IS_NOT_AN_FD 74233294Sstas int use_write = 0; 7555682Smarkm#endif 76233294Sstas 77233294Sstas while (rem > 0) { 78233294Sstas#ifdef SOCKET_IS_NOT_AN_FD 79233294Sstas if (use_write) 80233294Sstas count = _write (sock, cbuf, rem); 81233294Sstas else 82233294Sstas count = send (sock, cbuf, rem, 0); 83233294Sstas 84233294Sstas if (use_write == 0 && 85233294Sstas rk_IS_SOCKET_ERROR(count) && 86233294Sstas (rk_SOCK_ERRNO == WSANOTINITIALISED || 87233294Sstas rk_SOCK_ERRNO == WSAENOTSOCK)) { 88233294Sstas use_write = 1; 89233294Sstas 90233294Sstas count = _write (sock, cbuf, rem); 91233294Sstas } 92233294Sstas#else 93233294Sstas count = send (sock, cbuf, rem, 0); 94233294Sstas#endif 9555682Smarkm if (count < 0) { 9655682Smarkm if (errno == EINTR) 9755682Smarkm continue; 9855682Smarkm else 9955682Smarkm return count; 10055682Smarkm } 10155682Smarkm cbuf += count; 10255682Smarkm rem -= count; 10355682Smarkm } 10455682Smarkm return nbytes; 10555682Smarkm} 106233294Sstas 107233294Sstas#endif 108