/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

/*
 * This file defines _PR_MapOptionName().  The purpose of putting
 * _PR_MapOptionName() in a separate file is to work around a Winsock
 * header file problem on Windows NT.
 *
 * On Windows NT, if we define _WIN32_WINNT to be 0x0400 (in order
 * to use Service Pack 3 extensions), windows.h includes winsock2.h
 * (instead of winsock.h), which doesn't define many socket options
 * defined in winsock.h.
 *
 * We need the socket options defined in winsock.h.  So this file
 * includes winsock.h, with _WIN32_WINNT undefined.
 */

#if defined(WINNT) || defined(__MINGW32__)
#  include <winsock.h>
#endif

/* MinGW doesn't define these in its winsock.h. */
#ifdef __MINGW32__
#  ifndef IP_TTL
#    define IP_TTL 7
#  endif
#  ifndef IP_TOS
#    define IP_TOS 8
#  endif
#endif

#include "primpl.h"

#if defined(LINUX) || defined(ANDROID)
#  include <netinet/in.h>
#endif

#ifdef DARWIN
#  include <netinet/in.h>
#  include <netinet/ip.h>
#endif

#ifdef HAVE_NETINET_TCP_H
#  include <netinet/tcp.h> /* TCP_NODELAY, TCP_MAXSEG */
#endif

#ifndef _PR_PTHREADS

PRStatus PR_CALLBACK _PR_SocketGetSocketOption(PRFileDesc* fd,
                                               PRSocketOptionData* data) {
  PRStatus rv;
  PRInt32 length;
  PRInt32 level, name;

  /*
   * PR_SockOpt_Nonblocking is a special case that does not
   * translate to a getsockopt() call
   */
  if (PR_SockOpt_Nonblocking == data->option) {
    data->value.non_blocking = fd->secret->nonblocking;
    return PR_SUCCESS;
  }

  rv = _PR_MapOptionName(data->option, &level, &name);
  if (PR_SUCCESS == rv) {
    switch (data->option) {
      case PR_SockOpt_Linger: {
        struct linger linger;
        length = sizeof(linger);
        rv = _PR_MD_GETSOCKOPT(fd, level, name, (char*)&linger, &length);
        if (PR_SUCCESS == rv) {
          PR_ASSERT(sizeof(linger) == length);
          data->value.linger.polarity = (linger.l_onoff) ? PR_TRUE : PR_FALSE;
          data->value.linger.linger = PR_SecondsToInterval(linger.l_linger);
        }
        break;
      }
      case PR_SockOpt_Reuseaddr:
      case PR_SockOpt_Keepalive:
      case PR_SockOpt_NoDelay:
      case PR_SockOpt_Broadcast:
      case PR_SockOpt_Reuseport: {
#  ifdef WIN32 /* Winsock */
        BOOL value;
#  else
        PRIntn value;
#  endif
        length = sizeof(value);
        rv = _PR_MD_GETSOCKOPT(fd, level, name, (char*)&value, &length);
        if (PR_SUCCESS == rv) {
          data->value.reuse_addr = (0 == value) ? PR_FALSE : PR_TRUE;
        }
        break;
      }
      case PR_SockOpt_McastLoopback: {
#  ifdef WIN32 /* Winsock */
        BOOL bool;
#  else
        PRUint8 bool;
#  endif
        length = sizeof(bool);
        rv = _PR_MD_GETSOCKOPT(fd, level, name, (char*)&bool, &length);
        if (PR_SUCCESS == rv) {
          data->value.mcast_loopback = (0 == bool) ? PR_FALSE : PR_TRUE;
        }
        break;
      }
      case PR_SockOpt_RecvBufferSize:
      case PR_SockOpt_SendBufferSize:
      case PR_SockOpt_MaxSegment: {
        PRIntn value;
        length = sizeof(value);
        rv = _PR_MD_GETSOCKOPT(fd, level, name, (char*)&value, &length);
        if (PR_SUCCESS == rv) {
          data->value.recv_buffer_size = value;
        }
        break;
      }
      case PR_SockOpt_IpTimeToLive:
      case PR_SockOpt_IpTypeOfService: {
        /* These options should really be an int (or PRIntn). */
        length = sizeof(PRUintn);
        rv = _PR_MD_GETSOCKOPT(fd, level, name, (char*)&data->value.ip_ttl,
                               &length);
        break;
      }
      case PR_SockOpt_McastTimeToLive: {
#  ifdef WIN32 /* Winsock */
        int ttl;
#  else
        PRUint8 ttl;
#  endif
        length = sizeof(ttl);
        rv = _PR_MD_GETSOCKOPT(fd, level, name, (char*)&ttl, &length);
        if (PR_SUCCESS == rv) {
          data->value.mcast_ttl = ttl;
        }
        break;
      }
#  ifdef IP_ADD_MEMBERSHIP
      case PR_SockOpt_AddMember:
      case PR_SockOpt_DropMember: {
        struct ip_mreq mreq;
        length = sizeof(mreq);
        rv = _PR_MD_GETSOCKOPT(fd, level, name, (char*)&mreq, &length);
        if (PR_SUCCESS == rv) {
          data->value.add_member.mcaddr.inet.ip = mreq.imr_multiaddr.s_addr;
          data->value.add_member.ifaddr.inet.ip = mreq.imr_interface.s_addr;
        }
        break;
      }
#  endif /* IP_ADD_MEMBERSHIP */
      case PR_SockOpt_McastInterface: {
        /* This option is a struct in_addr. */
        length = sizeof(data->value.mcast_if.inet.ip);
        rv = _PR_MD_GETSOCKOPT(fd, level, name,
                               (char*)&data->value.mcast_if.inet.ip, &length);
        break;
      }
      case PR_SockOpt_DontFrag: {
#  if !defined(WIN32) && !defined(DARWIN) && !defined(LINUX) && \
      !defined(ANDROID)
        PR_SetError(PR_OPERATION_NOT_SUPPORTED_ERROR, 0);
        rv = PR_FAILURE;
#  else
#    ifdef WIN32 /* Winsock */
        DWORD value;
#    else
        PRIntn value;
#    endif
        length = sizeof(value);
        rv = _PR_MD_GETSOCKOPT(fd, level, name, (char*)&value, &length);
#    if defined(WIN32) || defined(DARWIN)
        data->value.dont_fragment = value;
#    else
        data->value.dont_fragment = (value == IP_PMTUDISC_DO) ? 1 : 0;
#    endif
#  endif /* !(!defined(WIN32) && !defined(DARWIN) && !defined(LINUX) && \
            !defined(ANDROID)) */
        break;
      }
      default:
        PR_NOT_REACHED("Unknown socket option");
        break;
    }
  }
  return rv;
} /* _PR_SocketGetSocketOption */

PRStatus PR_CALLBACK _PR_SocketSetSocketOption(PRFileDesc* fd,
                                               const PRSocketOptionData* data) {
  PRStatus rv;
  PRInt32 level, name;

  /*
   * PR_SockOpt_Nonblocking is a special case that does not
   * translate to a setsockopt call.
   */
  if (PR_SockOpt_Nonblocking == data->option) {
#  ifdef WINNT
    PR_ASSERT((fd->secret->md.io_model_committed == PR_FALSE) ||
              (fd->secret->nonblocking == data->value.non_blocking));
    if (fd->secret->md.io_model_committed &&
        (fd->secret->nonblocking != data->value.non_blocking)) {
      /*
       * On NT, once we have associated a socket with the io
       * completion port, we can't disassociate it.  So we
       * can't change the nonblocking option of the socket
       * afterwards.
       */
      PR_SetError(PR_INVALID_ARGUMENT_ERROR, 0);
      return PR_FAILURE;
    }
#  endif
    fd->secret->nonblocking = data->value.non_blocking;
    return PR_SUCCESS;
  }

  rv = _PR_MapOptionName(data->option, &level, &name);
  if (PR_SUCCESS == rv) {
    switch (data->option) {
      case PR_SockOpt_Linger: {
        struct linger linger;
        linger.l_onoff = data->value.linger.polarity;
        linger.l_linger = PR_IntervalToSeconds(data->value.linger.linger);
        rv = _PR_MD_SETSOCKOPT(fd, level, name, (char*)&linger, sizeof(linger));
        break;
      }
      case PR_SockOpt_Reuseaddr:
      case PR_SockOpt_Keepalive:
      case PR_SockOpt_NoDelay:
      case PR_SockOpt_Broadcast:
      case PR_SockOpt_Reuseport: {
#  ifdef WIN32 /* Winsock */
        BOOL value;
#  else
        PRIntn value;
#  endif
        value = (data->value.reuse_addr) ? 1 : 0;
        rv = _PR_MD_SETSOCKOPT(fd, level, name, (char*)&value, sizeof(value));
        break;
      }
      case PR_SockOpt_McastLoopback: {
#  ifdef WIN32 /* Winsock */
        BOOL bool;
#  else
        PRUint8 bool;
#  endif
        bool = data->value.mcast_loopback ? 1 : 0;
        rv = _PR_MD_SETSOCKOPT(fd, level, name, (char*)&bool, sizeof(bool));
        break;
      }
      case PR_SockOpt_RecvBufferSize:
      case PR_SockOpt_SendBufferSize:
      case PR_SockOpt_MaxSegment: {
        PRIntn value = data->value.recv_buffer_size;
        rv = _PR_MD_SETSOCKOPT(fd, level, name, (char*)&value, sizeof(value));
        break;
      }
      case PR_SockOpt_IpTimeToLive:
      case PR_SockOpt_IpTypeOfService: {
        /* These options should really be an int (or PRIntn). */
        rv = _PR_MD_SETSOCKOPT(fd, level, name, (char*)&data->value.ip_ttl,
                               sizeof(PRUintn));
        break;
      }
      case PR_SockOpt_McastTimeToLive: {
#  ifdef WIN32 /* Winsock */
        int ttl;
#  else
        PRUint8 ttl;
#  endif
        ttl = data->value.mcast_ttl;
        rv = _PR_MD_SETSOCKOPT(fd, level, name, (char*)&ttl, sizeof(ttl));
        break;
      }
#  ifdef IP_ADD_MEMBERSHIP
      case PR_SockOpt_AddMember:
      case PR_SockOpt_DropMember: {
        struct ip_mreq mreq;
        mreq.imr_multiaddr.s_addr = data->value.add_member.mcaddr.inet.ip;
        mreq.imr_interface.s_addr = data->value.add_member.ifaddr.inet.ip;
        rv = _PR_MD_SETSOCKOPT(fd, level, name, (char*)&mreq, sizeof(mreq));
        break;
      }
#  endif /* IP_ADD_MEMBERSHIP */
      case PR_SockOpt_McastInterface: {
        /* This option is a struct in_addr. */
        rv = _PR_MD_SETSOCKOPT(fd, level, name,
                               (char*)&data->value.mcast_if.inet.ip,
                               sizeof(data->value.mcast_if.inet.ip));
        break;
      }
      case PR_SockOpt_DontFrag: {
#  if !defined(WIN32) && !defined(DARWIN) && !defined(LINUX) && \
      !defined(ANDROID)
        PR_SetError(PR_OPERATION_NOT_SUPPORTED_ERROR, 0);
        rv = PR_FAILURE;
#  else
#    if defined(WIN32) /* Winsock */
        DWORD value;
        value = (data->value.dont_fragment) ? 1 : 0;
#    elif defined(LINUX) || defined(ANDROID)
        PRIntn value;
        value = (data->value.dont_fragment) ? IP_PMTUDISC_DO : IP_PMTUDISC_DONT;
#    elif defined(DARWIN)
        PRIntn value;
        value = data->value.dont_fragment;
#    endif
        rv = _PR_MD_SETSOCKOPT(fd, level, name, (char*)&value, sizeof(value));
#  endif /* !(!defined(WIN32) && !defined(DARWIN) && !defined(LINUX) && \
            !defined(ANDROID)) */
        break;
      }
      default:
        PR_NOT_REACHED("Unknown socket option");
        break;
    }
  }
  return rv;
} /* _PR_SocketSetSocketOption */

#endif /* ! _PR_PTHREADS */

/*
 *********************************************************************
 *********************************************************************
 **
 ** Make sure that the following is at the end of this file,
 ** because we will be playing with macro redefines.
 **
 *********************************************************************
 *********************************************************************
 */

/*
 * Not every platform has all the socket options we want to
 * support.  Some older operating systems such as SunOS 4.1.3
 * don't have the IP multicast socket options.  Win32 doesn't
 * have TCP_MAXSEG.
 *
 * To deal with this problem, we define the missing socket
 * options as _PR_NO_SUCH_SOCKOPT.  _PR_MapOptionName() fails with
 * PR_OPERATION_NOT_SUPPORTED_ERROR if a socket option not
 * available on the platform is requested.
 */

/*
 * Sanity check.  SO_LINGER and TCP_NODELAY should be available
 * on all platforms.  Just to make sure we have included the
 * appropriate header files.  Then any undefined socket options
 * are really missing.
 */

#if !defined(SO_LINGER)
#  error "SO_LINGER is not defined"
#endif

#if !defined(TCP_NODELAY)
#  error "TCP_NODELAY is not defined"
#endif

/*
 * Make sure the value of _PR_NO_SUCH_SOCKOPT is not
 * a valid socket option.
 */
#define _PR_NO_SUCH_SOCKOPT -1

#ifndef SO_KEEPALIVE
#  define SO_KEEPALIVE _PR_NO_SUCH_SOCKOPT
#endif

#ifndef SO_SNDBUF
#  define SO_SNDBUF _PR_NO_SUCH_SOCKOPT
#endif

#ifndef SO_RCVBUF
#  define SO_RCVBUF _PR_NO_SUCH_SOCKOPT
#endif

#ifndef IP_MULTICAST_IF /* set/get IP multicast interface   */
#  define IP_MULTICAST_IF _PR_NO_SUCH_SOCKOPT
#endif

#ifndef IP_MULTICAST_TTL /* set/get IP multicast timetolive  */
#  define IP_MULTICAST_TTL _PR_NO_SUCH_SOCKOPT
#endif

#ifndef IP_MULTICAST_LOOP /* set/get IP multicast loopback    */
#  define IP_MULTICAST_LOOP _PR_NO_SUCH_SOCKOPT
#endif

#ifndef IP_ADD_MEMBERSHIP /* add  an IP group membership      */
#  define IP_ADD_MEMBERSHIP _PR_NO_SUCH_SOCKOPT
#endif

#ifndef IP_DROP_MEMBERSHIP /* drop an IP group membership      */
#  define IP_DROP_MEMBERSHIP _PR_NO_SUCH_SOCKOPT
#endif

#ifndef IP_TTL /* set/get IP Time To Live          */
#  define IP_TTL _PR_NO_SUCH_SOCKOPT
#endif

#ifndef IP_TOS /* set/get IP Type Of Service       */
#  define IP_TOS _PR_NO_SUCH_SOCKOPT
#endif

/* set/get IP do not fragment */
#if defined(WIN32)
#  ifndef IP_DONTFRAGMENT
#    define IP_DONTFRAGMENT _PR_NO_SUCH_SOCKOPT
#  endif

#elif defined(LINUX) || defined(ANDROID)
#  ifndef IP_MTU_DISCOVER
#    define IP_MTU_DISCOVER _PR_NO_SUCH_SOCKOPT
#  endif

#elif defined(DARWIN)
#  ifndef IP_DONTFRAG
#    define IP_DONTFRAG _PR_NO_SUCH_SOCKOPT
#  endif
#endif

#ifndef TCP_NODELAY /* don't delay to coalesce data     */
#  define TCP_NODELAY _PR_NO_SUCH_SOCKOPT
#endif

#ifndef TCP_MAXSEG /* maxumum segment size for tcp     */
#  define TCP_MAXSEG _PR_NO_SUCH_SOCKOPT
#endif

#ifndef SO_BROADCAST /* enable broadcast on UDP sockets  */
#  define SO_BROADCAST _PR_NO_SUCH_SOCKOPT
#endif

#ifndef SO_REUSEPORT /* allow local address & port reuse */
#  define SO_REUSEPORT _PR_NO_SUCH_SOCKOPT
#endif

PRStatus _PR_MapOptionName(PRSockOption optname, PRInt32* level,
                           PRInt32* name) {
  static PRInt32 socketOptions[PR_SockOpt_Last] = {
      0,
      SO_LINGER,
      SO_REUSEADDR,
      SO_KEEPALIVE,
      SO_RCVBUF,
      SO_SNDBUF,
      IP_TTL,
      IP_TOS,
      IP_ADD_MEMBERSHIP,
      IP_DROP_MEMBERSHIP,
      IP_MULTICAST_IF,
      IP_MULTICAST_TTL,
      IP_MULTICAST_LOOP,
      TCP_NODELAY,
      TCP_MAXSEG,
      SO_BROADCAST,
      SO_REUSEPORT,
#if defined(WIN32)
      IP_DONTFRAGMENT,
#elif defined(LINUX) || defined(ANDROID)
      IP_MTU_DISCOVER,
#elif defined(DARWIN)
      IP_DONTFRAG,
#else
      _PR_NO_SUCH_SOCKOPT,
#endif
  };
  static PRInt32 socketLevels[PR_SockOpt_Last] = {
      0,          SOL_SOCKET,  SOL_SOCKET,  SOL_SOCKET, SOL_SOCKET, SOL_SOCKET,
      IPPROTO_IP, IPPROTO_IP,  IPPROTO_IP,  IPPROTO_IP, IPPROTO_IP, IPPROTO_IP,
      IPPROTO_IP, IPPROTO_TCP, IPPROTO_TCP, SOL_SOCKET, SOL_SOCKET, IPPROTO_IP};

  if ((optname < PR_SockOpt_Linger) || (optname >= PR_SockOpt_Last)) {
    PR_SetError(PR_INVALID_ARGUMENT_ERROR, 0);
    return PR_FAILURE;
  }

  if (socketOptions[optname] == _PR_NO_SUCH_SOCKOPT) {
    PR_SetError(PR_OPERATION_NOT_SUPPORTED_ERROR, 0);
    return PR_FAILURE;
  }
  *name = socketOptions[optname];
  *level = socketLevels[optname];
  return PR_SUCCESS;
} /* _PR_MapOptionName */