/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License, Version 1.0 only
 * (the "License").  You may not use this file except in compliance
 * with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */
/*
 * Copyright 2004 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

/*      Copyright (c) 1983, 1984, 1985, 1986, 1987, 1988, 1989 AT&T     */
/*        All Rights Reserved   */

/*
 * Portions of this source code were derived from Berkeley 4.3 BSD
 * under license from the Regents of the University of California.
 */

#include <sys/types.h>
#include <errno.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <netinet/udp.h>
#include <string.h>
#include <unistd.h>

#ifdef SYSV
#define bzero(s, len)   (void) memset((s), 0, (len))
#endif


/*
 * Bind a socket to a privileged IP port
 */
int
bindresvport(int sd, struct sockaddr_in *sin)
{
        struct sockaddr_in myaddr;
        struct sockaddr_in *bindaddr;
        int level, optname;
        int optval, len;
        int ret;

        bindaddr = sin;
        if (bindaddr == (struct sockaddr_in *)0) {
                bindaddr = &myaddr;
                bzero(bindaddr, sizeof (*bindaddr));
                bindaddr->sin_family = AF_INET;
        } else if (bindaddr->sin_family != AF_INET) {
                errno = EPFNOSUPPORT;
                return (-1);
        }

        len = sizeof (optval);
        if (getsockopt(sd, SOL_SOCKET, SO_TYPE, &optval, &len) < 0) {
                return (-1);
        }
        /*
         * Use *_ANONPRIVBIND to ask the kernel to pick a port in the
         * priviledged range for us.
         */
        if (optval == SOCK_STREAM) {
                level = IPPROTO_TCP;
                optname = TCP_ANONPRIVBIND;
        } else if (optval == SOCK_DGRAM) {
                level = IPPROTO_UDP;
                optname = UDP_ANONPRIVBIND;
        } else {
                errno = EPROTONOSUPPORT;
                return (-1);
        }

        optval = 1;
        if (setsockopt(sd, level, optname, &optval, sizeof (optval)) < 0) {
                return (-1);
        }

        bindaddr->sin_port = 0;
        ret = bind(sd, (struct sockaddr *)bindaddr,
            sizeof (struct sockaddr_in));

        /*
         * Always turn off the option when we are done.  Note that by doing
         * this, if the caller has set this option before calling
         * bindresvport(), it will be unset.  But this should never happen...
         */
        optval = 0;
        (void) setsockopt(sd, level, optname, &optval, sizeof (optval));

        if (ret >= 0 && sin != NULL) {
                /*
                 * Historical note:
                 *
                 * Past versions of this bindresvport() code have
                 * returned with the reserved port number bound
                 * filled in its "sin" parameter (if passed in), perhaps
                 * "accidently" because of the structure of historical code.
                 *
                 * This is not documented but the behavior is
                 * explicitly retained here for compatibility to minimize
                 * risk to applications, even though it is not clear if this
                 * was a design intent.
                 */
                len = sizeof (struct sockaddr_in);
                (void) getsockname(sd, (struct sockaddr *)bindaddr, &len);
        }
        return (ret);
}