[WTF] Add environment variable helpers
[WebKit-https.git] / Tools / TestRunnerShared / IOSLayoutTestCommunication.cpp
1 /*
2  * Copyright (C) 2017 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #include "config.h"
27 #include "IOSLayoutTestCommunication.h"
28
29 #include <netdb.h>
30 #include <netinet/in.h>
31 #include <sys/socket.h>
32 #include <sys/types.h>
33 #include <unistd.h>
34 #include <wtf/Assertions.h>
35 #include <wtf/Environment.h>
36
37 static int stdinSocket;
38 static int stdoutSocket;
39 static int stderrSocket;
40 static bool isUsingTCP = false;
41
42 static int connectToServer(sockaddr_in& serverAddress)
43 {
44     int result = socket(AF_INET, SOCK_STREAM, 0);
45     RELEASE_ASSERT(result >= 0);
46     RELEASE_ASSERT(connect(result, (struct sockaddr *) &serverAddress, sizeof(serverAddress)) >= 0);
47     return result;
48 }
49
50 void setUpIOSLayoutTestCommunication()
51 {
52     auto port = Environment::getInt("PORT");
53     if (!port)
54         return;
55     RELEASE_ASSERT(*port > 0);
56     isUsingTCP = true;
57
58     struct hostent* host = gethostbyname("127.0.0.1");
59     struct sockaddr_in serverAddress;
60     memset((char*) &serverAddress, 0, sizeof(serverAddress));
61     serverAddress.sin_family = AF_INET;
62     memcpy(
63         (char*)&serverAddress.sin_addr.s_addr,
64         (char*)host->h_addr,
65         host->h_length);
66     serverAddress.sin_port = htons(*port);
67     
68     // This order matches the server side listener in Tools/Scripts/webkitpy/port/simulator_process.py SimulatorProcess._start()
69     stdinSocket = connectToServer(serverAddress);
70     dup2(stdinSocket, STDIN_FILENO);
71
72     stdoutSocket = connectToServer(serverAddress);
73     dup2(stdoutSocket, STDOUT_FILENO);
74
75     stderrSocket = connectToServer(serverAddress);
76     dup2(stderrSocket, STDERR_FILENO);
77 }
78
79 void tearDownIOSLayoutTestCommunication()
80 {
81     if (!isUsingTCP)
82         return;
83     close(stdinSocket);
84     close(stdoutSocket);
85     close(stderrSocket);
86 }