mirror of https://github.com/dirtbags/moth.git
369 lines
10 KiB
C
369 lines
10 KiB
C
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <sysexits.h>
|
|
#include <errno.h>
|
|
#include <time.h>
|
|
#include <string.h>
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/select.h>
|
|
#include <netinet/in.h>
|
|
#include <netinet/tcp.h>
|
|
#include <arpa/inet.h>
|
|
|
|
#define OUTPUT_MAX 1024
|
|
#define INPUT_MAX 1024
|
|
|
|
#ifndef max
|
|
#define max(a,b) (((a)>(b))?(a):(b))
|
|
#endif
|
|
|
|
const char token[100];
|
|
size_t tokenlen;
|
|
|
|
char const octopus[] =
|
|
(" ___\n"
|
|
" .-' `'.\n"
|
|
" / \\\n"
|
|
" | ;\n"
|
|
" | | ___.--,\n"
|
|
" _.._ |8) ~ (8) | _.---'`__.-( (_.\n"
|
|
" __.--'`_.. '.__.\\ '--. \\_.-' ,.--'` `""`\n"
|
|
" ( ,.--'` ',__ /./; ;, '.__.'` __\n"
|
|
" _`) ) .---.__.' / | |\\ \\__..--\"\" \"\"\"--.,_\n"
|
|
" `---' .'.''-._.-'`_./ /\\ '. \\ _.-~~~````~~~-._`-.__.'\n"
|
|
" | | .' _.-' | | \\ \\ '. `~---`\n"
|
|
" \\ \\/ .' \\ \\ '. '-._)\n"
|
|
" \\/ / \\ \\ `=.__`~-.\n"
|
|
" jgs / /\\ `) ) / / `\"\".`\\\n"
|
|
" , _.-'.'\\ \\ / / ( ( / /\n"
|
|
" `--~` ) ) .-'.' '.'. | (\n"
|
|
" (/` ( (` ) ) '-;\n"
|
|
" ` '-; (-'\n"
|
|
);
|
|
|
|
const char *friends[8] = {
|
|
("Welcome to Olive Octopus's house!\n"
|
|
"Help Olive visit all 8 of her friends to receive a prize!\n"
|
|
"Hurry though, things change quickly in the ocean!\n"
|
|
"Next friend: %08o\n"
|
|
"%s"
|
|
),
|
|
("Thanks for stopping by, Olive! Good luck finding that prize!\n"
|
|
"Next friend: %08o\n"
|
|
" ,__\n"
|
|
" | `'.\n"
|
|
"__ |`-._/_.:---`-.._\n"
|
|
"\\='. _/..--'`__ `'-._\n"
|
|
" \\- '-.--\"` === / o `',\n"
|
|
" )= ( .--_ | _.'\n"
|
|
" /_=.'-._ {=_-_ | .--`-.\n"
|
|
"/_.' `\\`'-._ '-= \\ _.'\n"
|
|
" jgs ) _.-'`'-.. _..-'`\n"
|
|
" /_.' `/\";';`|\n"
|
|
" \\` .'/\n"
|
|
" '--'\n"
|
|
),
|
|
("Snap, snap! Good luck on your quest, Olive!\n"
|
|
"Next friend: %08o\n"
|
|
" .\"\".-._.-.\"\".\n"
|
|
" | \\ | / |\n"
|
|
" \\ \\.T./ /\n"
|
|
" '-./ \\.-'\n"
|
|
" / \\\n"
|
|
" ; ;\n"
|
|
" | |\n"
|
|
" | |\n"
|
|
" / \\\n"
|
|
" | . |\n"
|
|
" __.| : |.__\n"
|
|
" .-'` | : | `'-.\n"
|
|
" /` .\"\\ 0 : 0 /\". `\\\n"
|
|
" | _/ './ : \\.' \\_ |\n"
|
|
" | / /`\"\"\"`\\ \\ |\n"
|
|
" \\ \\ .-' '._ / /\n"
|
|
" jgs '-._\\ /_.-'\n"
|
|
),
|
|
("Nice talking with you, Olive. I'd best get back to my babies now!\n"
|
|
"Next friend: %08o\n"
|
|
" , ,\n"
|
|
" \\:.|`._\n"
|
|
" /\\/;.:':::;;;._\n"
|
|
" < .' ':::;(\n"
|
|
" < ' _ '::;>\n"
|
|
" \\ (9) _ :::;(\n"
|
|
" | / \\ ::;`>\n"
|
|
" | / | :;(\n"
|
|
" | ( <=- .::;>\n"
|
|
" ( a) )=- .::;(\n"
|
|
" '-' <=- .::;>\n"
|
|
" )==- ::::( ,\n"
|
|
" <==- :::(,-'(\n"
|
|
" )=- ':: _.->\n"
|
|
" <==- ':.' _(\n"
|
|
" <==- .:'_ (\n"
|
|
" )==- .::' '->\n"
|
|
" <=- .:;(`'.(\n"
|
|
" `) ':;> `\n"
|
|
" .-. < :;(\n"
|
|
" <`.':\\ ) :;>\n"
|
|
" < :/<_/ < .:;>\n"
|
|
" < '`---'` .::(`\n"
|
|
" jgs < .:;>'\n"
|
|
" `-..:::-'`\n"
|
|
),
|
|
("Spshhh! Good to see you, Olive! You're on the right track!\n"
|
|
"Next friend: %08o\n"
|
|
" ,_\n"
|
|
" \\::,\n"
|
|
" |::::\\\n"
|
|
" |:::::\\\n"
|
|
" __/:::::::\\,____\n"
|
|
" _.-::::::::::::::::::::==..,____\n"
|
|
" .-::::::::::::::::::::::::::::::::::::.,__\n"
|
|
" .:::::::::::::::::::::::::::::::::::::::::::::)\n"
|
|
" .:::::'```'-::::::::::::::::::::::(__,__`)::::-'\n"
|
|
" .;;;;;;::. ':::::::::::::::::::-:::::@::-'\"\"-,\n"
|
|
" .------:::::::::::' '-::::::::::' / `'--'\"\"\"\".-'\n"
|
|
"/:::::::::/:::/` _,..-----.,__ `''''`/ ;__,..--''--'`\n"
|
|
"`'--::::::::::|-'` `'---'| |\n"
|
|
" `\\::::\\ \\ /\n"
|
|
" |:::::| '-'\n"
|
|
" \\::::|\n"
|
|
" jgs `\\::|\n"
|
|
" \\/\n"
|
|
),
|
|
("You're getting close, Olive!\n"
|
|
"Next friend: %08o\n"
|
|
" .-------------'```'----......,,__ _,\n"
|
|
" | `'`'`'`'-.,.__ .'(\n"
|
|
" | `'--._.' )\n"
|
|
" | `'-.<\n"
|
|
" \\ .-'`'-. -. `\\\n"
|
|
" \\ -.o_. _ _,-'`\\ |\n"
|
|
" ``````''--.._.-=-._ .' \\ _,,--'` `-._(\n"
|
|
" (^^^^^^^^`___ '-. | \\ __,,,...--' `\n"
|
|
" ````````` `'--..___\\ |`\n"
|
|
" jgs `-.,'\n"
|
|
),
|
|
("Hi, Olive! Not much further now!\n"
|
|
"Next friend: %08o\n"
|
|
" , ,\n"
|
|
" /(_, ,_)\\\n"
|
|
" \\ _/ \\_ /\n"
|
|
" // \\\\\n"
|
|
" \\\\ (@)(@) //\n"
|
|
" \\'=\"==\"='/\n"
|
|
" ,===/ \\===,\n"
|
|
" \",===\\ /===,\"\n"
|
|
" \" ,==='------'===, \"\n"
|
|
" jgs \" \"\n"
|
|
),
|
|
("Aha! You found me!\n"
|
|
"Prize: %.*s\n"
|
|
" (\\.-./)\n"
|
|
" / \\\n"
|
|
" .' : '.\n"
|
|
" _.-'` ' `'-._\n"
|
|
" .-' : '-.\n"
|
|
" ,'_.._ . _.._',\n"
|
|
" '` `'-. ' .-'` `'\n"
|
|
" '. : .'\n"
|
|
" \\_. ._/\n"
|
|
" \\ |^|\n"
|
|
" | jgs | ;\n"
|
|
" \\'.___.' /\n"
|
|
" '-....-'\n")
|
|
};
|
|
|
|
const char invalid[] = "Who are you? Go away!\n";
|
|
|
|
#ifdef EASY
|
|
# define PORTS 15
|
|
#else
|
|
# define PORTS 8
|
|
#endif
|
|
|
|
struct bound_port {
|
|
int fd;
|
|
char output[OUTPUT_MAX];
|
|
size_t output_len;
|
|
} bound_ports[PORTS];
|
|
|
|
int
|
|
bind_port(struct in6_addr *addr, int fd, uint16_t port) {
|
|
struct sockaddr_in6 saddr = {0};
|
|
|
|
saddr.sin6_family = AF_INET6;
|
|
saddr.sin6_port = htons(port);
|
|
memcpy(&saddr.sin6_addr, addr, sizeof *addr);
|
|
return bind(fd, (struct sockaddr *)&saddr, sizeof saddr);
|
|
}
|
|
|
|
int
|
|
rebind(struct in6_addr *addr)
|
|
{
|
|
static int offset = 0;
|
|
int i;
|
|
|
|
for (i = 1; i < 8; i += 1) {
|
|
int ret;
|
|
int last_guy;
|
|
in_port_t port;
|
|
|
|
if (-1 != bound_ports[i + offset].fd) {
|
|
while (-1 == close(bound_ports[i + offset].fd)) {
|
|
if (errno != EINTR) {
|
|
return -1;
|
|
}
|
|
}
|
|
}
|
|
|
|
/* Bind to a port */
|
|
bound_ports[i + offset].fd = socket(AF_INET6, SOCK_DGRAM, 0);
|
|
do {
|
|
port = (random() % 56635) + 10000;
|
|
ret = bind_port(addr, bound_ports[i + offset].fd, port);
|
|
} while (-1 == ret);
|
|
|
|
/* Set the last guy's port number */
|
|
last_guy = i + offset - 1;
|
|
switch (i) {
|
|
case 1:
|
|
/* Always change the port 8888 one */
|
|
last_guy = 0;
|
|
case 2:
|
|
case 3:
|
|
case 4:
|
|
case 5:
|
|
case 6:
|
|
case 7:
|
|
bound_ports[last_guy].output_len =
|
|
snprintf(bound_ports[last_guy].output, OUTPUT_MAX,
|
|
friends[i - 1], port, octopus);
|
|
break;
|
|
}
|
|
}
|
|
bound_ports[7 + offset].output_len =
|
|
snprintf(bound_ports[7 + offset].output, OUTPUT_MAX,
|
|
friends[7], tokenlen, token);
|
|
|
|
if (offset == 0) {
|
|
offset = PORTS - 8;
|
|
} else {
|
|
offset = 0;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
void
|
|
do_io(int which)
|
|
{
|
|
struct bound_port *bp = &bound_ports[which];
|
|
char input[INPUT_MAX];
|
|
ssize_t inlen;
|
|
struct sockaddr_in6 from;
|
|
socklen_t fromlen = sizeof(from);
|
|
|
|
inlen = recvfrom(bp->fd, input, INPUT_MAX, 0,
|
|
(struct sockaddr *)&from, &fromlen);
|
|
if (-1 == inlen) {
|
|
/* Well don't that just beat all. */
|
|
return;
|
|
}
|
|
|
|
if (which > 0) {
|
|
if ((inlen != sizeof(octopus) - 1) ||
|
|
(0 != memcmp(input, octopus, inlen))) {
|
|
/* Didn't send the octopus */
|
|
sendto(bp->fd, invalid, sizeof(invalid), 0,
|
|
&from, fromlen);
|
|
return;
|
|
}
|
|
}
|
|
|
|
sendto(bp->fd, bp->output, bp->output_len, 0,
|
|
&from, fromlen);
|
|
}
|
|
|
|
int
|
|
loop()
|
|
{
|
|
int i;
|
|
int nfds = 0;
|
|
fd_set rfds;
|
|
|
|
FD_ZERO(&rfds);
|
|
for (i = 0; i < PORTS; i += 1) {
|
|
nfds = max(nfds, bound_ports[i].fd);
|
|
FD_SET(bound_ports[i].fd, &rfds);
|
|
}
|
|
|
|
while (-1 == select(nfds+1, &rfds, NULL, NULL, NULL)) {
|
|
if (EINTR == errno) {
|
|
continue;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
for (i = 0; i < PORTS; i += 1) {
|
|
if (FD_ISSET(bound_ports[i].fd, &rfds)) {
|
|
do_io(i);
|
|
}
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
int
|
|
main(int argc, char *argv[])
|
|
{
|
|
int ret;
|
|
int i;
|
|
time_t last_bind = 0;
|
|
time_t last_token = 0;
|
|
struct in6_addr addr;
|
|
|
|
/* The random seed isn't super important here. */
|
|
srand(time(NULL));
|
|
|
|
if (argc > 1) {
|
|
if (0 >= inet_pton(AF_INET6, argv[1], &addr)) {
|
|
fprintf(stderr, "invalid address: %s\n", argv[1]);
|
|
return EX_IOERR;
|
|
}
|
|
} else {
|
|
memcpy(&addr, &in6addr_any, sizeof addr);
|
|
}
|
|
|
|
if (NULL == fgets(token, sizeof(token), stdin)) {
|
|
perror("Unable to read token");
|
|
return EX_IOERR;
|
|
}
|
|
tokenlen = strlen(token);
|
|
|
|
bound_ports[0].fd = socket(AF_INET6, SOCK_DGRAM, 0);
|
|
ret = bind_port(&addr, bound_ports[0].fd, 8888);
|
|
if (-1 == ret) {
|
|
perror("bind port 8888");
|
|
return EX_IOERR;
|
|
}
|
|
|
|
for (i = 1; i < PORTS; i += 1) {
|
|
bound_ports[i].fd = -1;
|
|
}
|
|
|
|
do {
|
|
time_t now = time(NULL);
|
|
|
|
if (last_bind + 4 < now) {
|
|
last_bind = now;
|
|
if (-1 == rebind(&addr)) break;
|
|
}
|
|
} while (loop());
|
|
|
|
return 0;
|
|
}
|