RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-05-25 19:05:07 +00:00
Code Issues Activity Actions

TelnetServer: Accept arbitrary command with -c

Browse source 
For example, this allows you to do something like
`TelnetServer -c /usr/bin/nyancat` to have the TelnetServer run
`/usr/bin/nyancat` instead of `/bin/Shell`.
This commit is contained in:
Larkin Nickle 2019-09-28 16:31:43 -04:00 • committed by Andreas Kling
parent 801fe7beac
commit ab67f74588
1 changed files with 8 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

12
Servers/TelnetServer/main.cpp
View file

@ -85,13 +85,17 @@ int main(int argc, char** argv)
int opt;
u16 port = 23;
while ((opt = getopt(argc, argv, "p:")) != -1) {
const char* command = "";
while ((opt = getopt(argc, argv, "p:c:")) != -1) {
switch (opt) {
case 'p':
port = atoi(optarg);
break;
case 'c':
command = optarg;
break;
default:
fprintf(stderr, "Usage: %s [-p port]", argv[0]);
fprintf(stderr, "Usage: %s [-p port] [-c command]", argv[0]);
exit(1);
}
}
@ -104,7 +108,7 @@ int main(int argc, char** argv)
HashMap<int, NonnullRefPtr<Client>> clients;
int next_id = 0;
server->on_ready_to_accept = [&next_id, &clients, &server] {
server->on_ready_to_accept = [&next_id, &clients, &server, command] {
int id = next_id++;
auto client_socket = server->accept();
@ -120,7 +124,7 @@ int main(int argc, char** argv)
return;
}
run_command(ptm_fd, "");
run_command(ptm_fd, command);
auto client = Client::create(id, move(client_socket), ptm_fd);
client->on_exit = [&clients, id] { clients.remove(id); };