jcs.org
Serenity Operating System
1/*
2 * Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org>
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions are met:
7 *
8 * 1. Redistributions of source code must retain the above copyright notice, this
9 * list of conditions and the following disclaimer.
10 *
11 * 2. Redistributions in binary form must reproduce the above copyright notice,
12 * this list of conditions and the following disclaimer in the documentation
13 * and/or other materials provided with the distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
16 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
18 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
21 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
22 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
23 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
24 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25 */
26
27#include <AK/JsonObject.h>
28#include <AK/JsonValue.h>
29#include <LibCore/EventLoop.h>
30#include <LibCore/LocalSocket.h>
31#include <stdio.h>
32#include <stdlib.h>
33
34int main(int argc, char** argv)
35{
36 if (pledge("stdio unix cpath fattr", nullptr) < 0) {
37 perror("pledge");
38 return 1;
39 }
40
41 if (unveil("/tmp", "rwc") < 0) {
42 perror("unveil");
43 return 1;
44 }
45
46 unveil(nullptr, nullptr);
47
48 if (argc != 2) {
49 printf("usage: %s <pid>\n", argv[0]);
50 return 0;
51 }
52
53 Core::EventLoop loop;
54
55 int pid = atoi(argv[1]);
56
57 auto socket = Core::LocalSocket::construct();
58
59 if (pledge("stdio unix", nullptr) < 0) {
60 perror("pledge");
61 return 1;
62 }
63
64 socket->on_connected = [&] {
65 dbg() << "Connected to PID " << pid;
66
67 JsonObject request;
68 request.set("type", "GetAllObjects");
69 auto serialized = request.to_string();
70 i32 length = serialized.length();
71 socket->write((const u8*)&length, sizeof(length));
72 socket->write(serialized);
73 };
74
75 socket->on_ready_to_read = [&] {
76 if (socket->eof()) {
77 dbg() << "Disconnected from PID " << pid;
78 loop.quit(0);
79 return;
80 }
81
82 auto data = socket->read_all();
83
84 for (size_t i = 0; i < data.size(); ++i)
85 putchar(data[i]);
86 printf("\n");
87
88 loop.quit(0);
89 };
90
91 auto success = socket->connect(Core::SocketAddress::local(String::format("/tmp/rpc.%d", pid)));
92 if (!success) {
93 fprintf(stderr, "Couldn't connect to PID %d\n", pid);
94 return 1;
95 }
96
97 return loop.exec();
98}