7 #include <sys/socket.h>
10 #include <sys/ioctl.h>
11 #include <sys/epoll.h>
24 #define STREAM_ID "stream"
25 #define STREAM_URL "http://gruessi.zrh.sesse.net:4013/"
30 Server *servers = NULL;
31 volatile bool hupped = false;
38 int create_server_socket(int port)
40 int server_sock = socket(PF_INET6, SOCK_STREAM, IPPROTO_TCP);
41 if (server_sock == -1) {
47 if (setsockopt(server_sock, SOL_SOCKET, SO_REUSEADDR, &one, sizeof(one)) == -1) {
48 perror("setsockopt(SO_REUSEADDR)");
52 // We want dual-stack sockets. (Sorry, OpenBSD and Windows XP...)
54 if (setsockopt(server_sock, IPPROTO_IPV6, IPV6_V6ONLY, &zero, sizeof(zero)) == -1) {
55 perror("setsockopt(IPV6_V6ONLY)");
60 memset(&addr, 0, sizeof(addr));
61 addr.sin6_family = AF_INET6;
62 addr.sin6_port = htons(port);
64 if (bind(server_sock, reinterpret_cast<sockaddr *>(&addr), sizeof(addr)) == -1) {
69 if (listen(server_sock, 128) == -1) {
77 void *acceptor_thread_run(void *arg)
79 int server_sock = int(intptr_t(arg));
83 socklen_t addrlen = sizeof(addr);
86 int sock = accept(server_sock, reinterpret_cast<sockaddr *>(&addr), &addrlen);
87 if (sock == -1 && errno == EINTR) {
95 // Set the socket as nonblocking.
97 if (ioctl(sock, FIONBIO, &one) == -1) {
102 // Pick a server, round-robin, and hand over the socket to it.
103 servers[num_accepted % NUM_SERVERS].add_client(sock);
108 int main(int argc, char **argv)
110 servers = new Server[NUM_SERVERS];
111 for (int i = 0; i < NUM_SERVERS; ++i) {
112 servers[i].add_stream(STREAM_ID);
116 int server_sock = create_server_socket(PORT);
118 pthread_t acceptor_thread;
119 pthread_create(&acceptor_thread, NULL, acceptor_thread_run, reinterpret_cast<void *>(server_sock));
121 Input input(STREAM_ID, STREAM_URL);
132 CubemapStateProto state;
133 for (int i = 0; i < NUM_SERVERS; ++i) {
136 CubemapStateProto local_state = servers[i].serialize();
138 // The stream state should be identical between the servers, so we only store it once.
140 state.mutable_streams()->MergeFrom(local_state.streams());
142 for (int j = 0; j < local_state.clients_size(); ++j) {
143 state.add_clients()->MergeFrom(local_state.clients(j));
148 printf("SERIALIZED: [%s]\n", state.DebugString().c_str());