mozilla/nsprpub/pr/tests/poll_er.c

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* ***** BEGIN LICENSE BLOCK *****
   3  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   4  *
   5  * The contents of this file are subject to the Mozilla Public License Version
   6  * 1.1 (the "License"); you may not use this file except in compliance with
   7  * the License. You may obtain a copy of the License at
   8  * http://www.mozilla.org/MPL/
   9  *
  10  * Software distributed under the License is distributed on an "AS IS" basis,
  11  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  12  * for the specific language governing rights and limitations under the
  13  * License.
  14  *
  15  * The Original Code is the Netscape Portable Runtime (NSPR).
  16  *
  17  * The Initial Developer of the Original Code is
  18  * Netscape Communications Corporation.
  19  * Portions created by the Initial Developer are Copyright (C) 1998-2000
  20  * the Initial Developer. All Rights Reserved.
  21  *
  22  * Contributor(s):
  23  *
  24  * Alternatively, the contents of this file may be used under the terms of
  25  * either the GNU General Public License Version 2 or later (the "GPL"), or
  26  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  27  * in which case the provisions of the GPL or the LGPL are applicable instead
  28  * of those above. If you wish to allow use of your version of this file only
  29  * under the terms of either the GPL or the LGPL, and not to allow others to
  30  * use your version of this file under the terms of the MPL, indicate your
  31  * decision by deleting the provisions above and replace them with the notice
  32  * and other provisions required by the GPL or the LGPL. If you do not delete
  33  * the provisions above, a recipient may use your version of this file under
  34  * the terms of any one of the MPL, the GPL or the LGPL.
  35  *
  36  * ***** END LICENSE BLOCK ***** */
  37
  38 /***********************************************************************
  39 **
  40 ** Name: prpoll_err.c
  41 **
  42 ** Description: This program tests PR_Poll with sockets.
  43 **              error reporting operation is tested
  44 **
  45 ** Modification History:
  46 ** 19-May-97 AGarcia- Converted the test to accomodate the debug_mode flag.
  47 **               The debug mode will print all of the printfs associated with this test.
  48 **                       The regress mode will be the default mode. Since the regress tool limits
  49 **           the output to a one line status:PASS or FAIL,all of the printf statements
  50 **                       have been handled with an if (debug_mode) statement.
  51 ** 04-June-97 AGarcia removed the Test_Result function. Regress tool has been updated to
  52 **                      recognize the return code from tha main program.
  53 ***********************************************************************/
  54
  55 #ifdef XP_BEOS
  56 #include <stdio.h>
  57 int main()
  58 {
  59     printf( "This test is not ported to the BeOS\n" );
  60     return 0;
  61 }
  62 #else
  63
  64 /***********************************************************************
  65 ** Includes
  66 ***********************************************************************/
  67 /* Used to get the command line option */
  68 #include "plgetopt.h"
  69
  70 #include "primpl.h"
  71
  72 #include <stdio.h>
  73 #include <string.h>
  74 #include <stdlib.h>
  75
  76 PRIntn failed_already=0;
  77 PRIntn debug_mode;
  78
  79 static void
  80 ClientThreadFunc(void *arg)
  81 {
  82     PRFileDesc *badFD = (PRFileDesc *) arg;
  83     /*
  84      * Make the fd invalid
  85      */
  86 #if defined(XP_UNIX)
  87     close(PR_FileDesc2NativeHandle(badFD));
  88 #elif defined(XP_OS2)
  89     soclose(PR_FileDesc2NativeHandle(badFD));
  90 #elif defined(WIN32) || defined(WIN16)
  91     closesocket(PR_FileDesc2NativeHandle(badFD));
  92 #else
  93 #error "Unknown architecture"
  94 #endif
  95 }
  96
  97 int main(int argc, char **argv)
  98 {
  99     PRFileDesc *listenSock1, *listenSock2;
 100     PRFileDesc *badFD;
 101     PRUint16 listenPort1, listenPort2;
 102     PRNetAddr addr;
 103     char buf[128];
 104     PRPollDesc pds0[10], pds1[10], *pds, *other_pds;
 105     PRIntn npds;
 106     PRInt32 retVal;
 107
 108         /* The command line argument: -d is used to determine if the test is being run
 109         in debug mode. The regress tool requires only one line output:PASS or FAIL.
 110         All of the printfs associated with this test has been handled with a if (debug_mode)
 111         test.
 112         Usage: test_name -d
 113         */
 114         PLOptStatus os;
 115         PLOptState *opt = PL_CreateOptState(argc, argv, "d:");
 116         while (PL_OPT_EOL != (os = PL_GetNextOpt(opt)))
 117     {
 118                 if (PL_OPT_BAD == os) continue;
 119         switch (opt->option)
 120         {
 121         case 'd':  /* debug mode */
 122                         debug_mode = 1;
 123             break;
 124          default:
 125             break;
 126         }
 127     }
 128         PL_DestroyOptState(opt);
 129
 130  /* main test */
 131
 132     PR_Init(PR_USER_THREAD, PR_PRIORITY_NORMAL, 0);
 133     PR_STDIO_INIT();
 134
 135     if (debug_mode) {
 136                 printf("This program tests PR_Poll with sockets.\n");
 137                 printf("error reporting is  tested.\n\n");
 138         }
 139
 140     /* Create two listening sockets */
 141     if ((listenSock1 = PR_NewTCPSocket()) == NULL) {
 142         fprintf(stderr, "Can't create a new TCP socket\n");
 143         failed_already=1;
 144         goto exit_now;
 145     }
 146     addr.inet.family = AF_INET;
 147     addr.inet.ip = PR_htonl(INADDR_ANY);
 148     addr.inet.port = PR_htons(0);
 149     if (PR_Bind(listenSock1, &addr) == PR_FAILURE) {
 150         fprintf(stderr, "Can't bind socket\n");
 151         failed_already=1;
 152         goto exit_now;
 153     }
 154     if (PR_GetSockName(listenSock1, &addr) == PR_FAILURE) {
 155         fprintf(stderr, "PR_GetSockName failed\n");
 156         failed_already=1;
 157         goto exit_now;
 158     }
 159     listenPort1 = PR_ntohs(addr.inet.port);
 160     if (PR_Listen(listenSock1, 5) == PR_FAILURE) {
 161         fprintf(stderr, "Can't listen on a socket\n");
 162         failed_already=1;
 163         goto exit_now;
 164     }
 165
 166     if ((listenSock2  = PR_NewTCPSocket()) == NULL) {
 167         fprintf(stderr, "Can't create a new TCP socket\n");
 168         failed_already=1;
 169         goto exit_now;
 170     }
 171     addr.inet.family = AF_INET;
 172     addr.inet.ip = PR_htonl(INADDR_ANY);
 173     addr.inet.port = PR_htons(0);
 174     if (PR_Bind(listenSock2, &addr) == PR_FAILURE) {
 175         fprintf(stderr, "Can't bind socket\n");
 176         failed_already=1;
 177         goto exit_now;
 178     }
 179     if (PR_GetSockName(listenSock2, &addr) == PR_FAILURE) {
 180         fprintf(stderr, "PR_GetSockName failed\n");
 181         failed_already=1;
 182         goto exit_now;
 183     }
 184     listenPort2 = PR_ntohs(addr.inet.port);
 185     if (PR_Listen(listenSock2, 5) == PR_FAILURE) {
 186         fprintf(stderr, "Can't listen on a socket\n");
 187         failed_already=1;
 188         goto exit_now;
 189     }
 190     PR_snprintf(buf, sizeof(buf),
 191             "The server thread is listening on ports %hu and %hu\n\n",
 192             listenPort1, listenPort2);
 193     if (debug_mode) printf("%s", buf);
 194
 195     /* Set up the poll descriptor array */
 196     pds = pds0;
 197     other_pds = pds1;
 198     memset(pds, 0, sizeof(pds));
 199     pds[0].fd = listenSock1;
 200     pds[0].in_flags = PR_POLL_READ;
 201     pds[1].fd = listenSock2;
 202     pds[1].in_flags = PR_POLL_READ;
 203     npds = 2;
 204
 205
 206     /* Testing bad fd */
 207     if (debug_mode) printf("PR_Poll should detect a bad file descriptor\n");
 208     if ((badFD = PR_NewTCPSocket()) == NULL) {
 209         fprintf(stderr, "Can't create a TCP socket\n");
 210         goto exit_now;
 211     }
 212
 213     pds[2].fd = badFD;
 214     pds[2].in_flags = PR_POLL_READ;
 215     npds = 3;
 216
 217     if (PR_CreateThread(PR_USER_THREAD, ClientThreadFunc,
 218             badFD, PR_PRIORITY_NORMAL, PR_LOCAL_THREAD,
 219             PR_UNJOINABLE_THREAD, 0) == NULL) {
 220         fprintf(stderr, "cannot create thread\n");
 221         exit(1);
 222     }
 223
 224     retVal = PR_Poll(pds, npds, PR_INTERVAL_NO_TIMEOUT);
 225     if (retVal != 1 || (unsigned short) pds[2].out_flags != PR_POLL_NVAL) {
 226         fprintf(stderr, "Failed to detect the bad fd: "
 227                 "PR_Poll returns %d, out_flags is 0x%hx\n",
 228                 retVal, pds[2].out_flags);
 229         failed_already=1;
 230         goto exit_now;
 231     }
 232     if (debug_mode) printf("PR_Poll detected the bad fd.  Test passed.\n\n");
 233     PR_Cleanup();
 234         goto exit_now;
 235 exit_now:
 236         if(failed_already)
 237                 return 1;
 238         else
 239                 return 0;
 240 }
 241
 242 #endif /* XP_BEOS */