3401e7bfc6f6313c3fada9ac00247a6ee07ba2a6
[WebKit-https.git] / Source / WebKit2 / Shared / EntryPointUtilities / mac / XPCService / XPCServiceMain.mm
1 /*
2  * Copyright (C) 2013 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 "config.h"
27
28 #import <CoreFoundation/CoreFoundation.h>
29 #import <wtf/OSObjectPtr.h>
30
31 namespace WebKit {
32
33 static void XPCServiceEventHandler(xpc_connection_t peer)
34 {
35     xpc_connection_set_target_queue(peer, dispatch_get_main_queue());
36     xpc_connection_set_event_handler(peer, ^(xpc_object_t event) {
37         xpc_type_t type = xpc_get_type(event);
38         if (type == XPC_TYPE_ERROR) {
39             if (event == XPC_ERROR_CONNECTION_INVALID || event == XPC_ERROR_TERMINATION_IMMINENT) {
40                 // FIXME: Handle this case more gracefully.
41                 exit(EXIT_FAILURE);
42             }
43         } else {
44             assert(type == XPC_TYPE_DICTIONARY);
45
46             if (!strcmp(xpc_dictionary_get_string(event, "message-name"), "bootstrap")) {
47                 CFBundleRef webKitBundle = CFBundleGetBundleWithIdentifier(CFSTR("com.apple.WebKit"));
48                 CFStringRef entryPointFunctionName = (CFStringRef)CFBundleGetValueForInfoDictionaryKey(CFBundleGetMainBundle(), CFSTR("WebKitEntryPoint"));
49
50                 typedef void (*InitializerFunction)(xpc_connection_t, xpc_object_t);
51                 InitializerFunction initializerFunctionPtr = reinterpret_cast<InitializerFunction>(CFBundleGetFunctionPointerForName(webKitBundle, entryPointFunctionName));
52                 if (!initializerFunctionPtr) {
53                     NSLog(@"Unable to find entry point in WebKit.framework with name: %@", (NSString *)entryPointFunctionName);
54                     exit(EXIT_FAILURE);
55                 }
56
57                 auto reply = adoptOSObject(xpc_dictionary_create_reply(event));
58                 xpc_dictionary_set_string(reply.get(), "message-name", "process-finished-launching");
59                 xpc_connection_send_message(xpc_dictionary_get_remote_connection(event), reply.get());
60
61                 initializerFunctionPtr(peer, event);
62             }
63
64             // Leak a boost onto the NetworkProcess.
65             if (!strcmp(xpc_dictionary_get_string(event, "message-name"), "pre-bootstrap"))
66                 xpc_retain(event);
67         }
68     });
69
70     xpc_connection_resume(peer);
71 }
72
73 } // namespace WebKit
74
75 using namespace WebKit;
76
77 int main(int argc, char** argv)
78 {
79     xpc_main(XPCServiceEventHandler);
80     return 0;
81 }