36d61fb4f187d9ed5c9cd56ed161e0c597859acd
[WebKit-https.git] / Tools / LayoutTestRelay / LayoutTestRelay / main.m
1 /*
2  * Copyright (C) 2014 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 #import "CoreSimulatorSPI.h"
27 #import "LTRelayController.h"
28 #import <Foundation/Foundation.h>
29
30 static LTRelayController *relayController;
31
32 void usage()
33 {
34     NSString *helpText = @"LayoutTestRelay: run a dump tool in the simulator. Not for direct consumption.\n"
35     "Usage: LayoutTestRelay [-h] [options] -- [dump tool arguments]\n"
36     "Required options:\n"
37     "    -app <app_path>          Path to the built iOS .app bundle directory.\n"
38     "    -runtime <identifier>    iOS Simulator Runtime identifier (see `xcrun -sdk iphonesimulator simctl`)\n"
39     "    -deviceType <identifier> iOS Simulator device identifier (see simctl).\n"
40     "    -suffix <text>           Used to create multiple unique instances when installing to the simulator.\n"
41     "    -productDir <dir>        /path/to/WebKitBuild/{configuration}-{short platform}.\n";
42
43     fprintf(stderr, "%s\n", [helpText UTF8String]);
44 }
45
46 SimDevice *getTestingSimDevice(SimDeviceType *deviceType, SimRuntime *runtime, NSString *suffix)
47 {
48     NSString *deviceName = [[[[deviceType identifier] componentsSeparatedByString:@"."] lastObject] stringByReplacingOccurrencesOfString:@"-" withString:@" "];
49     deviceName = [NSString stringWithFormat:@"%@%@%@", deviceName, @" WebKit Tester", suffix];
50
51     for (SimDevice *device in [[SimDeviceSet defaultSet] devices]) {
52         if ([[device name] isEqualToString:deviceName] && [[device deviceType] isEqualTo:deviceType] && [[device runtime] isEqualTo:runtime])
53             return device;
54     }
55
56     NSError *error;
57     SimDevice *device = [[SimDeviceSet defaultSet] createDeviceWithType:deviceType runtime:runtime name:deviceName error:&error];
58
59     if (error) {
60         NSLog(@"Couldn't create device: %@", [error description]);
61         return nil;
62     }
63
64     while ([device state] == SimDeviceStateCreating)
65         sleep(1);
66
67     return device;
68 }
69
70
71 NSString *getRequiredStringArgument(NSString *parameter)
72 {
73     NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
74     NSString *argument = [defaults stringForKey:parameter];
75
76     if (!argument) {
77         NSLog(@"-%@ is required.", parameter);
78         usage();
79         exit(EXIT_FAILURE);
80     }
81     return argument;
82 }
83
84 NSArray *getDumpToolArguments()
85 {
86     NSMutableArray *dumpToolArguments = [[NSMutableArray alloc] init];
87     NSArray *processArgs = [[NSProcessInfo processInfo] arguments];
88     BOOL appending = NO;
89     for (NSString *arg in processArgs) {
90         if ([arg isEqualToString:@"--"]) {
91             appending = YES;
92             continue;
93         }
94         if (appending)
95             [dumpToolArguments addObject:arg];
96     }
97     return dumpToolArguments;
98 }
99
100 void finish()
101 {
102     [relayController finish];
103 }
104
105 void receivedSignal(int signal)
106 {
107     exit(EXIT_SUCCESS);
108 }
109
110 int main(int argc, const char * argv[])
111 {
112     @autoreleasepool {
113         NSArray *helpArguments = @[@"-h", @"--help"];
114         for (NSString *helpArgument in helpArguments) {
115             if ([[[NSProcessInfo processInfo] arguments] indexOfObject:helpArgument] != NSNotFound) {
116                 usage();
117                 exit(EXIT_FAILURE);
118             }
119         }
120         NSString *appPath = getRequiredStringArgument(@"app");
121         NSString *runtimeIdentifier = getRequiredStringArgument(@"runtime");
122         SimRuntime *runtime = [SimRuntime supportedRuntimesByIdentifier][runtimeIdentifier];
123         if (!runtime) {
124             NSLog(@"There is no supported runtime \"%@\"", runtimeIdentifier);
125             exit(EXIT_FAILURE);
126         }
127
128         NSString *deviceTypeIdentifier = getRequiredStringArgument(@"deviceType");
129         SimDeviceType *deviceType = [SimDeviceType supportedDeviceTypesByIdentifier][deviceTypeIdentifier];
130         if (!deviceType) {
131             NSLog(@"There is no supported device type \"%@\"", deviceTypeIdentifier);
132             exit(EXIT_FAILURE);
133         }
134
135         NSString *suffix = getRequiredStringArgument(@"suffix");
136         NSString *productDir = getRequiredStringArgument(@"productDir");
137         NSArray *dumpToolArguments = getDumpToolArguments();
138
139         SimDevice *device = getTestingSimDevice(deviceType, runtime, suffix);
140
141         relayController = [[LTRelayController alloc] initWithDevice:device productDir:productDir appPath:appPath identifierSuffix:suffix dumpToolArguments:dumpToolArguments];
142         [relayController start];
143
144         atexit(finish);
145         signal(SIGINT, receivedSignal);
146         signal(SIGTERM, receivedSignal);
147
148         [[NSRunLoop mainRunLoop] run];
149     }
150     return 0;
151 }