/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* 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/. */
#include "nspr.h"
#include "prio.h"
#include "prinit.h"
#include "prprf.h"
#include "obsolete/probslet.h"
#include "plerror.h"
#ifdef XP_UNIX
# include <sys/socket.h>
/* SO_REUSEPORT */
#endif
static PRFileDesc* err = NULL;
static PRBool failed = PR_FALSE;
static void Failed(
const char* msg1,
const char* msg2) {
if (NULL != msg1) {
PR_fprintf(err,
"%s ", msg1);
}
PL_FPrintError(err, msg2);
failed = PR_TRUE;
}
/* Failed */
static PRSockOption Incr(PRSockOption* option) {
PRIntn val = ((PRIntn)*option) + 1;
*option = (PRSockOption)val;
return (PRSockOption)val;
}
/* Incr */
int main(
int argc,
char** argv) {
PRStatus rv;
PRFileDesc* udp = PR_NewUDPSocket();
PRFileDesc* tcp = PR_NewTCPSocket();
const char* tag[] = {
"PR_SockOpt_Nonblocking",
/* nonblocking io */
"PR_SockOpt_Linger",
/* linger on close if data present */
"PR_SockOpt_Reuseaddr",
/* allow local address reuse */
"PR_SockOpt_Keepalive",
/* keep connections alive */
"PR_SockOpt_RecvBufferSize",
/* receive buffer size */
"PR_SockOpt_SendBufferSize",
/* send buffer size */
"PR_SockOpt_IpTimeToLive",
/* time to live */
"PR_SockOpt_IpTypeOfService",
/* type of service and precedence */
"PR_SockOpt_AddMember",
/* add an IP group membership */
"PR_SockOpt_DropMember",
/* drop an IP group membership */
"PR_SockOpt_McastInterface",
/* multicast interface address */
"PR_SockOpt_McastTimeToLive",
/* multicast timetolive */
"PR_SockOpt_McastLoopback",
/* multicast loopback */
"PR_SockOpt_NoDelay",
/* don't delay send to coalesce packets */
"PR_SockOpt_MaxSegment",
/* maximum segment size */
"PR_SockOpt_Broadcast",
/* Enable broadcast */
"PR_SockOpt_Reuseport",
/* allow local address & port reuse */
"PR_SockOpt_DontFrag",
/* do not fragment */
"PR_SockOpt_Last"};
err = PR_GetSpecialFD(PR_StandardError);
PR_STDIO_INIT();
if (NULL == udp) {
Failed(
"PR_NewUDPSocket()", NULL);
}
else if (NULL == tcp) {
Failed(
"PR_NewTCPSocket()", NULL);
}
else {
PRSockOption option;
PRUint32 segment = 1024;
PRNetAddr addr;
rv = PR_InitializeNetAddr(PR_IpAddrAny, 0, &addr);
if (PR_FAILURE == rv) {
Failed(
"PR_InitializeNetAddr()", NULL);
}
rv = PR_Bind(udp, &addr);
if (PR_FAILURE == rv) {
Failed(
"PR_Bind()", NULL);
}
for (option = PR_SockOpt_Linger; option < PR_SockOpt_Last; Incr(&option)) {
PRSocketOptionData data;
PRFileDesc* fd = tcp;
data.option = option;
switch (option) {
case PR_SockOpt_Nonblocking:
data.value.non_blocking = PR_TRUE;
break;
case PR_SockOpt_Linger:
data.value.linger.polarity = PR_TRUE;
data.value.linger.linger = PR_SecondsToInterval(2);
break;
case PR_SockOpt_Reuseaddr:
data.value.reuse_addr = PR_TRUE;
break;
case PR_SockOpt_Keepalive:
data.value.keep_alive = PR_TRUE;
break;
case PR_SockOpt_RecvBufferSize:
data.value.recv_buffer_size = segment;
break;
case PR_SockOpt_SendBufferSize:
data.value.send_buffer_size = segment;
break;
case PR_SockOpt_IpTimeToLive:
data.value.ip_ttl = 64;
break;
case PR_SockOpt_IpTypeOfService:
data.value.tos = 0;
break;
case PR_SockOpt_McastTimeToLive:
fd = udp;
data.value.mcast_ttl = 4;
break;
case PR_SockOpt_McastLoopback:
fd = udp;
data.value.mcast_loopback = PR_TRUE;
break;
case PR_SockOpt_NoDelay:
data.value.no_delay = PR_TRUE;
break;
#ifndef WIN32
case PR_SockOpt_MaxSegment:
data.value.max_segment = segment;
break;
#endif
case PR_SockOpt_Broadcast:
fd = udp;
data.value.broadcast = PR_TRUE;
break;
#ifdef SO_REUSEPORT
case PR_SockOpt_Reuseport:
data.value.reuse_port = PR_TRUE;
break;
#endif
case PR_SockOpt_DontFrag:
data.value.dont_fragment = PR_TRUE;
break;
default:
continue;
}
/*
* TCP_MAXSEG can only be read, not set
*/
if (option != PR_SockOpt_MaxSegment) {
#ifdef WIN32
if (option != PR_SockOpt_McastLoopback)
#endif
{
rv = PR_SetSocketOption(fd, &data);
if (PR_FAILURE == rv) {
Failed(
"PR_SetSocketOption()", tag[option]);
}
}
}
rv = PR_GetSocketOption(fd, &data);
if (PR_FAILURE == rv) {
Failed(
"PR_GetSocketOption()", tag[option]);
}
}
PR_Close(udp);
PR_Close(tcp);
}
PR_fprintf(err,
"%s\n", (failed) ?
"FAILED" :
"PASSED");
return (failed) ? 1 : 0;
}
/* main */
/* sockopt.c */