common/bsd/writev.c

	writev.c revision 4bc0a2ef2b7ba50a7a717e7ddbf31472ad28e358
/*
 * Copyright 2005 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

#ifndef LINT
static const char rcsid[] = "$Id: writev.c,v 8.6 2003/04/30 05:21:18 marka Exp $";
#endif


#pragma ident   "%Z%%M% %I% %E% SMI"

#include "port_before.h"

#include <sys/types.h>
#include <sys/uio.h>
#include <sys/stat.h>
#include <sys/socket.h>

#include "port_after.h"

#ifndef NEED_WRITEV
int __bindcompat_writev;
#else

#ifdef _CRAY
#define OWN_WRITEV
int
__writev(int fd, struct iovec *iov, int iovlen)
{
    struct stat statbuf;

    if (fstat(fd, &statbuf) < 0)
        return (-1);

    /*
     * Allow for atomic writes to network.
     */
    if (S_ISSOCK(statbuf.st_mode)) {
        struct msghdr   mesg;

        memset(&mesg, 0, sizeof(mesg));
        mesg.msg_name = 0;
        mesg.msg_namelen = 0;
        mesg.msg_iov = iov;
        mesg.msg_iovlen = iovlen;
        mesg.msg_accrights = 0;
        mesg.msg_accrightslen = 0;
        return (sendmsg(fd, &mesg, 0));
    } else {
        struct iovec *tv;
        int i, rcode = 0, count = 0;

        for (i = 0, tv = iov; i <= iovlen; tv++) {
            rcode = write(fd, tv->iov_base, tv->iov_len);

            if (rcode < 0)
                break;

            count += rcode;
        }

        if (count == 0)
            return (rcode);
        else
            return (count);
    }
}

#else /*_CRAY*/

int
__writev(fd, vp, vpcount)
    int fd;
    const struct iovec *vp;
    int vpcount;
{
    int count = 0;

    while (vpcount-- > 0) {
        int written = write(fd, vp->iov_base, vp->iov_len);

        if (written < 0)
            return (-1);
        count += written;
        if (written != vp->iov_len)
            break;
        vp++;
    }
    return (count);
}

#endif /*_CRAY*/

#endif /*NEED_WRITEV*/