| /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ |
| /* ***** BEGIN LICENSE BLOCK ***** |
| * Version: MPL 1.1/GPL 2.0/LGPL 2.1 |
| * |
| * The contents of this file are subject to the Mozilla Public License Version |
| * 1.1 (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * http://www.mozilla.org/MPL/ |
| * |
| * Software distributed under the License is distributed on an "AS IS" basis, |
| * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License |
| * for the specific language governing rights and limitations under the |
| * License. |
| * |
| * The Original Code is the Netscape Portable Runtime (NSPR). |
| * |
| * The Initial Developer of the Original Code is |
| * Netscape Communications Corporation. |
| * Portions created by the Initial Developer are Copyright (C) 2000 |
| * the Initial Developer. All Rights Reserved. |
| * |
| * Contributor(s): |
| * |
| * Alternatively, the contents of this file may be used under the terms of |
| * either the GNU General Public License Version 2 or later (the "GPL"), or |
| * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"), |
| * in which case the provisions of the GPL or the LGPL are applicable instead |
| * of those above. If you wish to allow use of your version of this file only |
| * under the terms of either the GPL or the LGPL, and not to allow others to |
| * use your version of this file under the terms of the MPL, indicate your |
| * decision by deleting the provisions above and replace them with the notice |
| * and other provisions required by the GPL or the LGPL. If you do not delete |
| * the provisions above, a recipient may use your version of this file under |
| * the terms of any one of the MPL, the GPL or the LGPL. |
| * |
| * ***** END LICENSE BLOCK ***** */ |
| |
| /* |
| * File: pipeping2.c |
| * |
| * Description: |
| * This test runs in conjunction with the pipepong2 test. |
| * This test creates two pipes and passes two pipe fd's |
| * to the pipepong2 test. Then this test writes "ping" to |
| * to the pipepong2 test and the pipepong2 test writes "pong" |
| * back. To run this pair of tests, just invoke pipeping2. |
| * |
| * Tested areas: process creation, pipes, file descriptor |
| * inheritance. |
| */ |
| |
| #include "prerror.h" |
| #include "prio.h" |
| #include "prproces.h" |
| |
| #include <stdio.h> |
| #include <string.h> |
| #include <stdlib.h> |
| |
| #define NUM_ITERATIONS 10 |
| |
| static char *child_argv[] = { "pipepong2", NULL }; |
| |
| int main(int argc, char **argv) |
| { |
| PRFileDesc *in_pipe[2]; |
| PRFileDesc *out_pipe[2]; |
| PRStatus status; |
| PRProcess *process; |
| PRProcessAttr *attr; |
| char buf[1024]; |
| PRInt32 nBytes; |
| PRInt32 exitCode; |
| int idx; |
| |
| status = PR_CreatePipe(&in_pipe[0], &in_pipe[1]); |
| if (status == PR_FAILURE) { |
| fprintf(stderr, "PR_CreatePipe failed\n"); |
| exit(1); |
| } |
| status = PR_CreatePipe(&out_pipe[0], &out_pipe[1]); |
| if (status == PR_FAILURE) { |
| fprintf(stderr, "PR_CreatePipe failed\n"); |
| exit(1); |
| } |
| |
| status = PR_SetFDInheritable(in_pipe[0], PR_FALSE); |
| if (status == PR_FAILURE) { |
| fprintf(stderr, "PR_SetFDInheritable failed\n"); |
| exit(1); |
| } |
| status = PR_SetFDInheritable(in_pipe[1], PR_TRUE); |
| if (status == PR_FAILURE) { |
| fprintf(stderr, "PR_SetFDInheritable failed\n"); |
| exit(1); |
| } |
| status = PR_SetFDInheritable(out_pipe[0], PR_TRUE); |
| if (status == PR_FAILURE) { |
| fprintf(stderr, "PR_SetFDInheritable failed\n"); |
| exit(1); |
| } |
| status = PR_SetFDInheritable(out_pipe[1], PR_FALSE); |
| if (status == PR_FAILURE) { |
| fprintf(stderr, "PR_SetFDInheritable failed\n"); |
| exit(1); |
| } |
| |
| attr = PR_NewProcessAttr(); |
| if (attr == NULL) { |
| fprintf(stderr, "PR_NewProcessAttr failed\n"); |
| exit(1); |
| } |
| |
| status = PR_ProcessAttrSetInheritableFD(attr, out_pipe[0], "PIPE_READ"); |
| if (status == PR_FAILURE) { |
| fprintf(stderr, "PR_ProcessAttrSetInheritableFD failed\n"); |
| exit(1); |
| } |
| status = PR_ProcessAttrSetInheritableFD(attr, in_pipe[1], "PIPE_WRITE"); |
| if (status == PR_FAILURE) { |
| fprintf(stderr, "PR_ProcessAttrSetInheritableFD failed\n"); |
| exit(1); |
| } |
| |
| process = PR_CreateProcess(child_argv[0], child_argv, NULL, attr); |
| if (process == NULL) { |
| fprintf(stderr, "PR_CreateProcess failed\n"); |
| exit(1); |
| } |
| PR_DestroyProcessAttr(attr); |
| status = PR_Close(out_pipe[0]); |
| if (status == PR_FAILURE) { |
| fprintf(stderr, "PR_Close failed\n"); |
| exit(1); |
| } |
| status = PR_Close(in_pipe[1]); |
| if (status == PR_FAILURE) { |
| fprintf(stderr, "PR_Close failed\n"); |
| exit(1); |
| } |
| |
| for (idx = 0; idx < NUM_ITERATIONS; idx++) { |
| strcpy(buf, "ping"); |
| printf("ping process: sending \"%s\"\n", buf); |
| nBytes = PR_Write(out_pipe[1], buf, 5); |
| if (nBytes == -1) { |
| fprintf(stderr, "PR_Write failed: (%d, %d)\n", |
| PR_GetError(), PR_GetOSError()); |
| exit(1); |
| } |
| memset(buf, 0, sizeof(buf)); |
| nBytes = PR_Read(in_pipe[0], buf, sizeof(buf)); |
| if (nBytes == -1) { |
| fprintf(stderr, "PR_Read failed\n"); |
| exit(1); |
| } |
| printf("ping process: received \"%s\"\n", buf); |
| if (nBytes != 5) { |
| fprintf(stderr, "ping process: expected 5 bytes but got %d bytes\n", |
| nBytes); |
| exit(1); |
| } |
| if (strcmp(buf, "pong") != 0) { |
| fprintf(stderr, "ping process: expected \"pong\" but got \"%s\"\n", |
| buf); |
| exit(1); |
| } |
| } |
| |
| status = PR_Close(in_pipe[0]); |
| if (status == PR_FAILURE) { |
| fprintf(stderr, "PR_Close failed\n"); |
| exit(1); |
| } |
| status = PR_Close(out_pipe[1]); |
| if (status == PR_FAILURE) { |
| fprintf(stderr, "PR_Close failed\n"); |
| exit(1); |
| } |
| status = PR_WaitProcess(process, &exitCode); |
| if (status == PR_FAILURE) { |
| fprintf(stderr, "PR_WaitProcess failed\n"); |
| exit(1); |
| } |
| if (exitCode == 0) { |
| printf("PASS\n"); |
| return 0; |
| } else { |
| printf("FAIL\n"); |
| return 1; |
| } |
| } |