~ubuntu-branches/ubuntu/precise/kompozer/precise

« back to all changes in this revision

Viewing changes to mozilla/nsprpub/pr/tests/sockping.c

  • Committer: Bazaar Package Importer
  • Author(s): Anthony Yarusso
  • Date: 2007-08-27 01:11:03 UTC
  • Revision ID: james.westby@ubuntu.com-20070827011103-2jgf4s6532gqu2ka
Tags: upstream-0.7.10
ImportĀ upstreamĀ versionĀ 0.7.10

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
 
2
/* 
 
3
 * The contents of this file are subject to the Mozilla Public
 
4
 * License Version 1.1 (the "License"); you may not use this file
 
5
 * except in compliance with the License. You may obtain a copy of
 
6
 * the License at http://www.mozilla.org/MPL/
 
7
 * 
 
8
 * Software distributed under the License is distributed on an "AS
 
9
 * IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
 
10
 * implied. See the License for the specific language governing
 
11
 * rights and limitations under the License.
 
12
 * 
 
13
 * The Original Code is the Netscape Portable Runtime (NSPR).
 
14
 * 
 
15
 * The Initial Developer of the Original Code is Netscape
 
16
 * Communications Corporation.  Portions created by Netscape are 
 
17
 * Copyright (C) 1999-2000 Netscape Communications Corporation.  All
 
18
 * Rights Reserved.
 
19
 * 
 
20
 * Contributor(s):
 
21
 * 
 
22
 * Alternatively, the contents of this file may be used under the
 
23
 * terms of the GNU General Public License Version 2 or later (the
 
24
 * "GPL"), in which case the provisions of the GPL are applicable 
 
25
 * instead of those above.  If you wish to allow use of your 
 
26
 * version of this file only under the terms of the GPL and not to
 
27
 * allow others to use your version of this file under the MPL,
 
28
 * indicate your decision by deleting the provisions above and
 
29
 * replace them with the notice and other provisions required by
 
30
 * the GPL.  If you do not delete the provisions above, a recipient
 
31
 * may use your version of this file under either the MPL or the
 
32
 * GPL.
 
33
 */
 
34
 
 
35
/*
 
36
 * File: sockping.c
 
37
 *
 
38
 * Description:
 
39
 * This test runs in conjunction with the sockpong test.
 
40
 * This test creates a socket pair and passes one socket
 
41
 * to the sockpong test.  Then this test writes "ping" to
 
42
 * to the sockpong test and the sockpong test writes "pong"
 
43
 * back.  To run this pair of tests, just invoke sockping.
 
44
 *
 
45
 * Tested areas: process creation, socket pairs, file
 
46
 * descriptor inheritance.
 
47
 */
 
48
 
 
49
#include "prerror.h"
 
50
#include "prio.h"
 
51
#include "prproces.h"
 
52
 
 
53
#include <stdio.h>
 
54
#include <string.h>
 
55
#include <stdlib.h>
 
56
 
 
57
#define NUM_ITERATIONS 10
 
58
 
 
59
static char *child_argv[] = { "sockpong", NULL };
 
60
 
 
61
int main()
 
62
{
 
63
    PRFileDesc *sock[2];
 
64
    PRStatus status;
 
65
    PRProcess *process;
 
66
    PRProcessAttr *attr;
 
67
    char buf[1024];
 
68
    PRInt32 nBytes;
 
69
    PRInt32 exitCode;
 
70
    int idx;
 
71
 
 
72
    status = PR_NewTCPSocketPair(sock);
 
73
    if (status == PR_FAILURE) {
 
74
        fprintf(stderr, "PR_NewTCPSocketPair failed\n");
 
75
        exit(1);
 
76
    }
 
77
 
 
78
    status = PR_SetFDInheritable(sock[0], PR_FALSE);
 
79
    if (status == PR_FAILURE) {
 
80
        fprintf(stderr, "PR_SetFDInheritable failed: (%d, %d)\n",
 
81
                PR_GetError(), PR_GetOSError());
 
82
        exit(1);
 
83
    }
 
84
    status = PR_SetFDInheritable(sock[1], PR_TRUE);
 
85
    if (status == PR_FAILURE) {
 
86
        fprintf(stderr, "PR_SetFDInheritable failed: (%d, %d)\n",
 
87
                PR_GetError(), PR_GetOSError());
 
88
        exit(1);
 
89
    }
 
90
 
 
91
    attr = PR_NewProcessAttr();
 
92
    if (attr == NULL) {
 
93
        fprintf(stderr, "PR_NewProcessAttr failed\n");
 
94
        exit(1);
 
95
    }
 
96
 
 
97
    status = PR_ProcessAttrSetInheritableFD(attr, sock[1], "SOCKET");
 
98
    if (status == PR_FAILURE) {
 
99
        fprintf(stderr, "PR_ProcessAttrSetInheritableFD failed\n");
 
100
        exit(1);
 
101
    }
 
102
 
 
103
    process = PR_CreateProcess(child_argv[0], child_argv, NULL, attr);
 
104
    if (process == NULL) {
 
105
        fprintf(stderr, "PR_CreateProcess failed\n");
 
106
        exit(1);
 
107
    }
 
108
    PR_DestroyProcessAttr(attr);
 
109
    status = PR_Close(sock[1]);
 
110
    if (status == PR_FAILURE) {
 
111
        fprintf(stderr, "PR_Close failed\n");
 
112
        exit(1);
 
113
    }
 
114
 
 
115
    for (idx = 0; idx < NUM_ITERATIONS; idx++) {
 
116
        strcpy(buf, "ping");
 
117
        printf("ping process: sending \"%s\"\n", buf);
 
118
        nBytes = PR_Write(sock[0], buf, 5);
 
119
        if (nBytes == -1) {
 
120
            fprintf(stderr, "PR_Write failed: (%d, %d)\n",
 
121
                    PR_GetError(), PR_GetOSError());
 
122
            exit(1);
 
123
        }
 
124
        memset(buf, 0, sizeof(buf));
 
125
        nBytes = PR_Read(sock[0], buf, sizeof(buf));
 
126
        if (nBytes == -1) {
 
127
            fprintf(stderr, "PR_Read failed: (%d, %d)\n",
 
128
                    PR_GetError(), PR_GetOSError());
 
129
            exit(1);
 
130
        }
 
131
        printf("ping process: received \"%s\"\n", buf);
 
132
        if (nBytes != 5) {
 
133
            fprintf(stderr, "ping process: expected 5 bytes but got %d bytes\n",
 
134
                    nBytes);
 
135
            exit(1);
 
136
        }
 
137
        if (strcmp(buf, "pong") != 0) {
 
138
            fprintf(stderr, "ping process: expected \"pong\" but got \"%s\"\n",
 
139
                    buf);
 
140
            exit(1);
 
141
        }
 
142
    }
 
143
 
 
144
    status = PR_Close(sock[0]);
 
145
    if (status == PR_FAILURE) {
 
146
        fprintf(stderr, "PR_Close failed\n");
 
147
        exit(1);
 
148
    }
 
149
    status = PR_WaitProcess(process, &exitCode);
 
150
    if (status == PR_FAILURE) {
 
151
        fprintf(stderr, "PR_WaitProcess failed\n");
 
152
        exit(1);
 
153
    }
 
154
    if (exitCode == 0) {
 
155
        printf("PASS\n");
 
156
        return 0;
 
157
    } else {
 
158
        printf("FAIL\n");
 
159
        return 1;
 
160
    }
 
161
}