Back to index

lightning-sunbird  0.9+nobinonly
sockping.c
Go to the documentation of this file.
00001 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
00002 /* ***** BEGIN LICENSE BLOCK *****
00003  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
00004  *
00005  * The contents of this file are subject to the Mozilla Public License Version
00006  * 1.1 (the "License"); you may not use this file except in compliance with
00007  * the License. You may obtain a copy of the License at
00008  * http://www.mozilla.org/MPL/
00009  *
00010  * Software distributed under the License is distributed on an "AS IS" basis,
00011  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
00012  * for the specific language governing rights and limitations under the
00013  * License.
00014  *
00015  * The Original Code is the Netscape Portable Runtime (NSPR).
00016  *
00017  * The Initial Developer of the Original Code is
00018  * Netscape Communications Corporation.
00019  * Portions created by the Initial Developer are Copyright (C) 1999-2000
00020  * the Initial Developer. All Rights Reserved.
00021  *
00022  * Contributor(s):
00023  *
00024  * Alternatively, the contents of this file may be used under the terms of
00025  * either the GNU General Public License Version 2 or later (the "GPL"), or
00026  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
00027  * in which case the provisions of the GPL or the LGPL are applicable instead
00028  * of those above. If you wish to allow use of your version of this file only
00029  * under the terms of either the GPL or the LGPL, and not to allow others to
00030  * use your version of this file under the terms of the MPL, indicate your
00031  * decision by deleting the provisions above and replace them with the notice
00032  * and other provisions required by the GPL or the LGPL. If you do not delete
00033  * the provisions above, a recipient may use your version of this file under
00034  * the terms of any one of the MPL, the GPL or the LGPL.
00035  *
00036  * ***** END LICENSE BLOCK ***** */
00037 
00038 /*
00039  * File: sockping.c
00040  *
00041  * Description:
00042  * This test runs in conjunction with the sockpong test.
00043  * This test creates a socket pair and passes one socket
00044  * to the sockpong test.  Then this test writes "ping" to
00045  * to the sockpong test and the sockpong test writes "pong"
00046  * back.  To run this pair of tests, just invoke sockping.
00047  *
00048  * Tested areas: process creation, socket pairs, file
00049  * descriptor inheritance.
00050  */
00051 
00052 #include "prerror.h"
00053 #include "prio.h"
00054 #include "prproces.h"
00055 
00056 #include <stdio.h>
00057 #include <string.h>
00058 #include <stdlib.h>
00059 
00060 #define NUM_ITERATIONS 10
00061 
00062 static char *child_argv[] = { "sockpong", NULL };
00063 
00064 int main()
00065 {
00066     PRFileDesc *sock[2];
00067     PRStatus status;
00068     PRProcess *process;
00069     PRProcessAttr *attr;
00070     char buf[1024];
00071     PRInt32 nBytes;
00072     PRInt32 exitCode;
00073     int idx;
00074 
00075     status = PR_NewTCPSocketPair(sock);
00076     if (status == PR_FAILURE) {
00077         fprintf(stderr, "PR_NewTCPSocketPair failed\n");
00078         exit(1);
00079     }
00080 
00081     status = PR_SetFDInheritable(sock[0], PR_FALSE);
00082     if (status == PR_FAILURE) {
00083         fprintf(stderr, "PR_SetFDInheritable failed: (%d, %d)\n",
00084                 PR_GetError(), PR_GetOSError());
00085         exit(1);
00086     }
00087     status = PR_SetFDInheritable(sock[1], PR_TRUE);
00088     if (status == PR_FAILURE) {
00089         fprintf(stderr, "PR_SetFDInheritable failed: (%d, %d)\n",
00090                 PR_GetError(), PR_GetOSError());
00091         exit(1);
00092     }
00093 
00094     attr = PR_NewProcessAttr();
00095     if (attr == NULL) {
00096         fprintf(stderr, "PR_NewProcessAttr failed\n");
00097         exit(1);
00098     }
00099 
00100     status = PR_ProcessAttrSetInheritableFD(attr, sock[1], "SOCKET");
00101     if (status == PR_FAILURE) {
00102         fprintf(stderr, "PR_ProcessAttrSetInheritableFD failed\n");
00103         exit(1);
00104     }
00105 
00106     process = PR_CreateProcess(child_argv[0], child_argv, NULL, attr);
00107     if (process == NULL) {
00108         fprintf(stderr, "PR_CreateProcess failed\n");
00109         exit(1);
00110     }
00111     PR_DestroyProcessAttr(attr);
00112     status = PR_Close(sock[1]);
00113     if (status == PR_FAILURE) {
00114         fprintf(stderr, "PR_Close failed\n");
00115         exit(1);
00116     }
00117 
00118     for (idx = 0; idx < NUM_ITERATIONS; idx++) {
00119         strcpy(buf, "ping");
00120         printf("ping process: sending \"%s\"\n", buf);
00121         nBytes = PR_Write(sock[0], buf, 5);
00122         if (nBytes == -1) {
00123             fprintf(stderr, "PR_Write failed: (%d, %d)\n",
00124                     PR_GetError(), PR_GetOSError());
00125             exit(1);
00126         }
00127         memset(buf, 0, sizeof(buf));
00128         nBytes = PR_Read(sock[0], buf, sizeof(buf));
00129         if (nBytes == -1) {
00130             fprintf(stderr, "PR_Read failed: (%d, %d)\n",
00131                     PR_GetError(), PR_GetOSError());
00132             exit(1);
00133         }
00134         printf("ping process: received \"%s\"\n", buf);
00135         if (nBytes != 5) {
00136             fprintf(stderr, "ping process: expected 5 bytes but got %d bytes\n",
00137                     nBytes);
00138             exit(1);
00139         }
00140         if (strcmp(buf, "pong") != 0) {
00141             fprintf(stderr, "ping process: expected \"pong\" but got \"%s\"\n",
00142                     buf);
00143             exit(1);
00144         }
00145     }
00146 
00147     status = PR_Close(sock[0]);
00148     if (status == PR_FAILURE) {
00149         fprintf(stderr, "PR_Close failed\n");
00150         exit(1);
00151     }
00152     status = PR_WaitProcess(process, &exitCode);
00153     if (status == PR_FAILURE) {
00154         fprintf(stderr, "PR_WaitProcess failed\n");
00155         exit(1);
00156     }
00157     if (exitCode == 0) {
00158         printf("PASS\n");
00159         return 0;
00160     } else {
00161         printf("FAIL\n");
00162         return 1;
00163     }
00164 }