1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
#include <arpa/inet.h>
#include <json-c/json.h>
#include <signal.h>
#include <threads.h>
#include "logger.c"
#define BACKLOG 10
struct Client {
int fd;
struct sockaddr_in addr;
};
void sig_handler(int signo) {
log_warn("Signal %d received, shutting down data server", signo);
exit(0);
}
void respond_name(int client_fd) {
log_info("Client requested simulation name");
struct json_object *sim_name = json_object_new_object();
json_object_object_add(sim_name, "name", json_object_new_string(NAME));
json_object_to_fd(client_fd, sim_name, JSON_C_TO_STRING_PRETTY);
json_object_put(sim_name);
}
void respond_opts(int client_fd) {
log_info("Client requested simulation options");
struct json_object *sim_opts = json_object_from_file(SIM_OPTS);
json_object_to_fd(client_fd, sim_opts, JSON_C_TO_STRING_PRETTY);
json_object_put(sim_opts);
}
void respond_hash(int client_fd) {
log_info("Client requested git hash");
char buff[41] = { 0 };
FILE *pipe = popen("git rev-parse HEAD", "r");
fread(buff, sizeof(char), 40, pipe);
pclose(pipe);
struct json_object *git_hash = json_object_new_object();
json_object_object_add(git_hash, "hash", json_object_new_string(buff));
json_object_to_fd(client_fd, git_hash, JSON_C_TO_STRING_PRETTY);
json_object_put(git_hash);
}
int handle_client(struct Client *client) {
assert(client);
char client_ip[INET_ADDRSTRLEN];
inet_ntop(AF_INET, &client->addr.sin_addr, client_ip, INET_ADDRSTRLEN);
log_info("Client connected: %s:%d", client_ip, ntohs(client->addr.sin_port));
char request = '\0';
recv(client->fd, &request, 1, 0);
switch (request) {
case REQUEST_NAME:
respond_name(client->fd);
break;
case REQUEST_OPTS:
respond_opts(client->fd);
break;
case REQUEST_HASH:
respond_hash(client->fd);
break;
default:
log_warn("Client made invalid request");
}
log_info("Client disconnected: %s:%d", client_ip, ntohs(client->addr.sin_port));
close(client->fd);
free(client);
return 0;
}
int main(void) {
signal(SIGINT, sig_handler);
signal(SIGTERM, sig_handler);
log_info("Initializing salis data server");
int opt = 1;
int server_fd = socket(AF_INET, SOCK_STREAM, 0);
setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
log_info("Binding to port: %d", PORT);
struct sockaddr_in server_addr = { 0 };
server_addr.sin_family = AF_INET;
server_addr.sin_addr.s_addr = INADDR_ANY;
server_addr.sin_port = htons(PORT);
bind(server_fd, (struct sockaddr *)&server_addr, sizeof(struct sockaddr_in));
log_info("Listening...");
listen(server_fd, BACKLOG);
while (true) {
struct Client *client = calloc(1, sizeof(struct Client));
socklen_t client_len = sizeof(struct sockaddr_in);
client->fd = accept(server_fd, (struct sockaddr *)&client->addr, &client_len);
thrd_t thrd;
thrd_create(&thrd, (thrd_start_t)handle_client, client);
thrd_detach(thrd);
}
close(server_fd);
return 0;
}
|
