| 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 | #elif defined(XP_MAC)
|
|---|
| 93 | _PR_MD_CLOSE_SOCKET(PR_FileDesc2NativeHandle(badFD));
|
|---|
| 94 | #else
|
|---|
| 95 | #error "Unknown architecture"
|
|---|
| 96 | #endif
|
|---|
| 97 | }
|
|---|
| 98 |
|
|---|
| 99 | int main(int argc, char **argv)
|
|---|
| 100 | {
|
|---|
| 101 | PRFileDesc *listenSock1, *listenSock2;
|
|---|
| 102 | PRFileDesc *badFD;
|
|---|
| 103 | PRUint16 listenPort1, listenPort2;
|
|---|
| 104 | PRNetAddr addr;
|
|---|
| 105 | char buf[128];
|
|---|
| 106 | PRPollDesc pds0[10], pds1[10], *pds, *other_pds;
|
|---|
| 107 | PRIntn npds;
|
|---|
| 108 | PRInt32 retVal;
|
|---|
| 109 |
|
|---|
| 110 | /* The command line argument: -d is used to determine if the test is being run
|
|---|
| 111 | in debug mode. The regress tool requires only one line output:PASS or FAIL.
|
|---|
| 112 | All of the printfs associated with this test has been handled with a if (debug_mode)
|
|---|
| 113 | test.
|
|---|
| 114 | Usage: test_name -d
|
|---|
| 115 | */
|
|---|
| 116 | PLOptStatus os;
|
|---|
| 117 | PLOptState *opt = PL_CreateOptState(argc, argv, "d:");
|
|---|
| 118 | while (PL_OPT_EOL != (os = PL_GetNextOpt(opt)))
|
|---|
| 119 | {
|
|---|
| 120 | if (PL_OPT_BAD == os) continue;
|
|---|
| 121 | switch (opt->option)
|
|---|
| 122 | {
|
|---|
| 123 | case 'd': /* debug mode */
|
|---|
| 124 | debug_mode = 1;
|
|---|
| 125 | break;
|
|---|
| 126 | default:
|
|---|
| 127 | break;
|
|---|
| 128 | }
|
|---|
| 129 | }
|
|---|
| 130 | PL_DestroyOptState(opt);
|
|---|
| 131 |
|
|---|
| 132 | /* main test */
|
|---|
| 133 |
|
|---|
| 134 | PR_Init(PR_USER_THREAD, PR_PRIORITY_NORMAL, 0);
|
|---|
| 135 | PR_STDIO_INIT();
|
|---|
| 136 |
|
|---|
| 137 | if (debug_mode) {
|
|---|
| 138 | printf("This program tests PR_Poll with sockets.\n");
|
|---|
| 139 | printf("error reporting is tested.\n\n");
|
|---|
| 140 | }
|
|---|
| 141 |
|
|---|
| 142 | /* Create two listening sockets */
|
|---|
| 143 | if ((listenSock1 = PR_NewTCPSocket()) == NULL) {
|
|---|
| 144 | fprintf(stderr, "Can't create a new TCP socket\n");
|
|---|
| 145 | failed_already=1;
|
|---|
| 146 | goto exit_now;
|
|---|
| 147 | }
|
|---|
| 148 | addr.inet.family = AF_INET;
|
|---|
| 149 | addr.inet.ip = PR_htonl(INADDR_ANY);
|
|---|
| 150 | addr.inet.port = PR_htons(0);
|
|---|
| 151 | if (PR_Bind(listenSock1, &addr) == PR_FAILURE) {
|
|---|
| 152 | fprintf(stderr, "Can't bind socket\n");
|
|---|
| 153 | failed_already=1;
|
|---|
| 154 | goto exit_now;
|
|---|
| 155 | }
|
|---|
| 156 | if (PR_GetSockName(listenSock1, &addr) == PR_FAILURE) {
|
|---|
| 157 | fprintf(stderr, "PR_GetSockName failed\n");
|
|---|
| 158 | failed_already=1;
|
|---|
| 159 | goto exit_now;
|
|---|
| 160 | }
|
|---|
| 161 | listenPort1 = PR_ntohs(addr.inet.port);
|
|---|
| 162 | if (PR_Listen(listenSock1, 5) == PR_FAILURE) {
|
|---|
| 163 | fprintf(stderr, "Can't listen on a socket\n");
|
|---|
| 164 | failed_already=1;
|
|---|
| 165 | goto exit_now;
|
|---|
| 166 | }
|
|---|
| 167 |
|
|---|
| 168 | if ((listenSock2 = PR_NewTCPSocket()) == NULL) {
|
|---|
| 169 | fprintf(stderr, "Can't create a new TCP socket\n");
|
|---|
| 170 | failed_already=1;
|
|---|
| 171 | goto exit_now;
|
|---|
| 172 | }
|
|---|
| 173 | addr.inet.family = AF_INET;
|
|---|
| 174 | addr.inet.ip = PR_htonl(INADDR_ANY);
|
|---|
| 175 | addr.inet.port = PR_htons(0);
|
|---|
| 176 | if (PR_Bind(listenSock2, &addr) == PR_FAILURE) {
|
|---|
| 177 | fprintf(stderr, "Can't bind socket\n");
|
|---|
| 178 | failed_already=1;
|
|---|
| 179 | goto exit_now;
|
|---|
| 180 | }
|
|---|
| 181 | if (PR_GetSockName(listenSock2, &addr) == PR_FAILURE) {
|
|---|
| 182 | fprintf(stderr, "PR_GetSockName failed\n");
|
|---|
| 183 | failed_already=1;
|
|---|
| 184 | goto exit_now;
|
|---|
| 185 | }
|
|---|
| 186 | listenPort2 = PR_ntohs(addr.inet.port);
|
|---|
| 187 | if (PR_Listen(listenSock2, 5) == PR_FAILURE) {
|
|---|
| 188 | fprintf(stderr, "Can't listen on a socket\n");
|
|---|
| 189 | failed_already=1;
|
|---|
| 190 | goto exit_now;
|
|---|
| 191 | }
|
|---|
| 192 | PR_snprintf(buf, sizeof(buf),
|
|---|
| 193 | "The server thread is listening on ports %hu and %hu\n\n",
|
|---|
| 194 | listenPort1, listenPort2);
|
|---|
| 195 | if (debug_mode) printf("%s", buf);
|
|---|
| 196 |
|
|---|
| 197 | /* Set up the poll descriptor array */
|
|---|
| 198 | pds = pds0;
|
|---|
| 199 | other_pds = pds1;
|
|---|
| 200 | memset(pds, 0, sizeof(pds));
|
|---|
| 201 | pds[0].fd = listenSock1;
|
|---|
| 202 | pds[0].in_flags = PR_POLL_READ;
|
|---|
| 203 | pds[1].fd = listenSock2;
|
|---|
| 204 | pds[1].in_flags = PR_POLL_READ;
|
|---|
| 205 | npds = 2;
|
|---|
| 206 |
|
|---|
| 207 |
|
|---|
| 208 | /* Testing bad fd */
|
|---|
| 209 | if (debug_mode) printf("PR_Poll should detect a bad file descriptor\n");
|
|---|
| 210 | if ((badFD = PR_NewTCPSocket()) == NULL) {
|
|---|
| 211 | fprintf(stderr, "Can't create a TCP socket\n");
|
|---|
| 212 | goto exit_now;
|
|---|
| 213 | }
|
|---|
| 214 |
|
|---|
| 215 | pds[2].fd = badFD;
|
|---|
| 216 | pds[2].in_flags = PR_POLL_READ;
|
|---|
| 217 | npds = 3;
|
|---|
| 218 |
|
|---|
| 219 | if (PR_CreateThread(PR_USER_THREAD, ClientThreadFunc,
|
|---|
| 220 | badFD, PR_PRIORITY_NORMAL, PR_LOCAL_THREAD,
|
|---|
| 221 | PR_UNJOINABLE_THREAD, 0) == NULL) {
|
|---|
| 222 | fprintf(stderr, "cannot create thread\n");
|
|---|
| 223 | exit(1);
|
|---|
| 224 | }
|
|---|
| 225 |
|
|---|
| 226 | retVal = PR_Poll(pds, npds, PR_INTERVAL_NO_TIMEOUT);
|
|---|
| 227 | if (retVal != 1 || (unsigned short) pds[2].out_flags != PR_POLL_NVAL) {
|
|---|
| 228 | fprintf(stderr, "Failed to detect the bad fd: "
|
|---|
| 229 | "PR_Poll returns %d, out_flags is 0x%hx\n",
|
|---|
| 230 | retVal, pds[2].out_flags);
|
|---|
| 231 | failed_already=1;
|
|---|
| 232 | goto exit_now;
|
|---|
| 233 | }
|
|---|
| 234 | if (debug_mode) printf("PR_Poll detected the bad fd. Test passed.\n\n");
|
|---|
| 235 | PR_Cleanup();
|
|---|
| 236 | goto exit_now;
|
|---|
| 237 | exit_now:
|
|---|
| 238 | if(failed_already)
|
|---|
| 239 | return 1;
|
|---|
| 240 | else
|
|---|
| 241 | return 0;
|
|---|
| 242 | }
|
|---|
| 243 |
|
|---|
| 244 | #endif /* XP_BEOS */
|
|---|
