130 lines
2.7 KiB
C
130 lines
2.7 KiB
C
#include <arpa/inet.h>
|
|
#include <netinet/in.h>
|
|
#include <signal.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/types.h>
|
|
#include <sys/wait.h>
|
|
#include <unistd.h>
|
|
|
|
static void setup(void) {
|
|
setvbuf(stdin, NULL, _IONBF, 0);
|
|
setvbuf(stdout, NULL, _IONBF, 0);
|
|
setvbuf(stderr, NULL, _IONBF, 0);
|
|
}
|
|
|
|
static void reap_children(int sig) {
|
|
(void)sig;
|
|
while (waitpid(-1, NULL, WNOHANG) > 0) {
|
|
}
|
|
}
|
|
|
|
__attribute__((noreturn)) static void win(void) {
|
|
puts("Storm warning! Here's your emergency shell:");
|
|
char *argv[] = {"/bin/sh", NULL};
|
|
execve("/bin/sh", argv, NULL);
|
|
_exit(0);
|
|
}
|
|
|
|
static void read_query(int fd) {
|
|
char query[48];
|
|
|
|
ssize_t n = read(fd, query, 256);
|
|
if (n <= 0) {
|
|
return;
|
|
}
|
|
}
|
|
|
|
static void handle_client(int fd) {
|
|
dup2(fd, STDIN_FILENO);
|
|
dup2(fd, STDOUT_FILENO);
|
|
dup2(fd, STDERR_FILENO);
|
|
|
|
alarm(2);
|
|
|
|
char location[16];
|
|
|
|
const char *banner = "Welcome to the Weather Station!\n";
|
|
(void)write(fd, banner, strlen(banner));
|
|
|
|
const char *prompt1 = "Enter your location: ";
|
|
(void)write(fd, prompt1, strlen(prompt1));
|
|
(void)read(fd, location, sizeof(location));
|
|
|
|
const char *prompt2 = "Submit your forecast query: ";
|
|
(void)write(fd, prompt2, strlen(prompt2));
|
|
|
|
read_query(fd);
|
|
|
|
const char *ok = "Forecast sent!\n";
|
|
(void)write(fd, ok, strlen(ok));
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
setup();
|
|
|
|
int port = 5555;
|
|
if (argc == 2) {
|
|
port = atoi(argv[1]);
|
|
if (port <= 0 || port > 65535) {
|
|
fprintf(stderr, "Invalid port\n");
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
signal(SIGCHLD, reap_children);
|
|
|
|
int s = socket(AF_INET, SOCK_STREAM, 0);
|
|
if (s < 0) {
|
|
perror("socket");
|
|
return 1;
|
|
}
|
|
|
|
int opt = 1;
|
|
setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
|
|
|
|
struct sockaddr_in addr;
|
|
memset(&addr, 0, sizeof(addr));
|
|
addr.sin_family = AF_INET;
|
|
addr.sin_addr.s_addr = htonl(INADDR_ANY);
|
|
addr.sin_port = htons((uint16_t)port);
|
|
|
|
if (bind(s, (struct sockaddr *)&addr, sizeof(addr)) < 0) {
|
|
perror("bind");
|
|
return 1;
|
|
}
|
|
|
|
if (listen(s, 16) < 0) {
|
|
perror("listen");
|
|
return 1;
|
|
}
|
|
|
|
printf("Weather Station listening on 0.0.0.0:%d\n", port);
|
|
|
|
while (1) {
|
|
int fd = accept(s, NULL, NULL);
|
|
if (fd < 0) {
|
|
perror("accept");
|
|
continue;
|
|
}
|
|
|
|
pid_t pid = fork();
|
|
if (pid < 0) {
|
|
perror("fork");
|
|
close(fd);
|
|
continue;
|
|
}
|
|
|
|
if (pid == 0) {
|
|
close(s);
|
|
handle_client(fd);
|
|
close(fd);
|
|
_exit(0);
|
|
}
|
|
|
|
close(fd);
|
|
}
|
|
}
|