The Tor Browser: nsprpub/pr/tests/pipeping2.c@6474c204b198

Cloned upstream origin tor-browser at tor-browser-31.3.0esr-4.5-1-build1
revision ID fc1c9ff7c1b2defdbc039f12214767608f46423f for hacking purpose.

     1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */

     2 /* This Source Code Form is subject to the terms of the Mozilla Public

     3  * License, v. 2.0. If a copy of the MPL was not distributed with this

     4  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

     6 /*

     7  * File: pipeping2.c

     8  *

     9  * Description:

    10  * This test runs in conjunction with the pipepong2 test.

    11  * This test creates two pipes and passes two pipe fd's

    12  * to the pipepong2 test.  Then this test writes "ping" to

    13  * to the pipepong2 test and the pipepong2 test writes "pong"

    14  * back.  To run this pair of tests, just invoke pipeping2.

    15  *

    16  * Tested areas: process creation, pipes, file descriptor

    17  * inheritance.

    18  */

    20 #include "prerror.h"

    21 #include "prio.h"

    22 #include "prproces.h"

    24 #include <stdio.h>

    25 #include <string.h>

    26 #include <stdlib.h>

    28 #define NUM_ITERATIONS 10

    30 static char *child_argv[] = { "pipepong2", NULL };

    32 int main(int argc, char **argv)

    33 {

    34     PRFileDesc *in_pipe[2];

    35     PRFileDesc *out_pipe[2];

    36     PRStatus status;

    37     PRProcess *process;

    38     PRProcessAttr *attr;

    39     char buf[1024];

    40     PRInt32 nBytes;

    41     PRInt32 exitCode;

    42     int idx;

    44     status = PR_CreatePipe(&in_pipe[0], &in_pipe[1]);

    45     if (status == PR_FAILURE) {

    46         fprintf(stderr, "PR_CreatePipe failed\n");

    47         exit(1);

    48     }

    49     status = PR_CreatePipe(&out_pipe[0], &out_pipe[1]);

    50     if (status == PR_FAILURE) {

    51         fprintf(stderr, "PR_CreatePipe failed\n");

    52         exit(1);

    53     }

    55     status = PR_SetFDInheritable(in_pipe[0], PR_FALSE);

    56     if (status == PR_FAILURE) {

    57         fprintf(stderr, "PR_SetFDInheritable failed\n");

    58         exit(1);

    59     }

    60     status = PR_SetFDInheritable(in_pipe[1], PR_TRUE);

    61     if (status == PR_FAILURE) {

    62         fprintf(stderr, "PR_SetFDInheritable failed\n");

    63         exit(1);

    64     }

    65     status = PR_SetFDInheritable(out_pipe[0], PR_TRUE);

    66     if (status == PR_FAILURE) {

    67         fprintf(stderr, "PR_SetFDInheritable failed\n");

    68         exit(1);

    69     }

    70     status = PR_SetFDInheritable(out_pipe[1], PR_FALSE);

    71     if (status == PR_FAILURE) {

    72         fprintf(stderr, "PR_SetFDInheritable failed\n");

    73         exit(1);

    74     }

    76     attr = PR_NewProcessAttr();

    77     if (attr == NULL) {

    78         fprintf(stderr, "PR_NewProcessAttr failed\n");

    79         exit(1);

    80     }

    82     status = PR_ProcessAttrSetInheritableFD(attr, out_pipe[0], "PIPE_READ");

    83     if (status == PR_FAILURE) {

    84         fprintf(stderr, "PR_ProcessAttrSetInheritableFD failed\n");

    85         exit(1);

    86     }

    87     status = PR_ProcessAttrSetInheritableFD(attr, in_pipe[1], "PIPE_WRITE");

    88     if (status == PR_FAILURE) {

    89         fprintf(stderr, "PR_ProcessAttrSetInheritableFD failed\n");

    90         exit(1);

    91     }

    93     process = PR_CreateProcess(child_argv[0], child_argv, NULL, attr);

    94     if (process == NULL) {

    95         fprintf(stderr, "PR_CreateProcess failed\n");

    96         exit(1);

    97     }

    98     PR_DestroyProcessAttr(attr);

    99     status = PR_Close(out_pipe[0]);

   100     if (status == PR_FAILURE) {

   101         fprintf(stderr, "PR_Close failed\n");

   102         exit(1);

   103     }

   104     status = PR_Close(in_pipe[1]);

   105     if (status == PR_FAILURE) {

   106         fprintf(stderr, "PR_Close failed\n");

   107         exit(1);

   108     }

   110     for (idx = 0; idx < NUM_ITERATIONS; idx++) {

   111         strcpy(buf, "ping");

   112         printf("ping process: sending \"%s\"\n", buf);

   113         nBytes = PR_Write(out_pipe[1], buf, 5);

   114         if (nBytes == -1) {

   115             fprintf(stderr, "PR_Write failed: (%d, %d)\n",

   116                     PR_GetError(), PR_GetOSError());

   117             exit(1);

   118         }

   119         memset(buf, 0, sizeof(buf));

   120         nBytes = PR_Read(in_pipe[0], buf, sizeof(buf));

   121         if (nBytes == -1) {

   122             fprintf(stderr, "PR_Read failed\n");

   123             exit(1);

   124         }

   125         printf("ping process: received \"%s\"\n", buf);

   126         if (nBytes != 5) {

   127             fprintf(stderr, "ping process: expected 5 bytes but got %d bytes\n",

   128                     nBytes);

   129             exit(1);

   130         }

   131         if (strcmp(buf, "pong") != 0) {

   132             fprintf(stderr, "ping process: expected \"pong\" but got \"%s\"\n",

   133                     buf);

   134             exit(1);

   135         }

   136     }

   138     status = PR_Close(in_pipe[0]);

   139     if (status == PR_FAILURE) {

   140         fprintf(stderr, "PR_Close failed\n");

   141         exit(1);

   142     }

   143     status = PR_Close(out_pipe[1]);

   144     if (status == PR_FAILURE) {

   145         fprintf(stderr, "PR_Close failed\n");

   146         exit(1);

   147     }

   148     status = PR_WaitProcess(process, &exitCode);

   149     if (status == PR_FAILURE) {

   150         fprintf(stderr, "PR_WaitProcess failed\n");

   151         exit(1);

   152     }

   153     if (exitCode == 0) {

   154         printf("PASS\n");

   155         return 0;

   156     } else {

   157         printf("FAIL\n");

   158         return 1;

   159     }

   160 }

The Tor Browser / file revision