Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with HTTPS or Subversion.

Download ZIP
Fetching contributors…

Cannot retrieve contributors at this time

112 lines (84 sloc) 3.026 kb
/* net.c - stupid boilerplate shit that I shouldn't have to write */
/* Copyright (C) 2007 Keith Rarick and Philotic Inc.
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include <stdio.h>
#include <errno.h>
#include "net.h"
#include "util.h"
static int listen_socket = -1;
static struct event listen_evq;
static evh accept_handler;
static time_t main_deadline = 0;
static int brakes_are_on = 1;
int
make_server_socket(struct in_addr host_addr, int port)
{
int fd, flags, r;
struct linger linger = {0, 0};
struct sockaddr_in addr;
fd = socket(AF_INET, SOCK_STREAM, 0);
if (fd == -1) return twarn("socket()"), -1;
flags = fcntl(fd, F_GETFL, 0);
if (flags < 0) return twarn("getting flags"), close(fd), -1;
r = fcntl(fd, F_SETFL, flags | O_NONBLOCK);
if (flags < 0) return twarn("setting O_NONBLOCK"), close(fd), -1;
flags = 1;
setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &flags, sizeof flags);
setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, &flags, sizeof flags);
setsockopt(fd, SOL_SOCKET, SO_LINGER, &linger, sizeof linger);
setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &flags, sizeof flags);
/*memset(&addr, 0, sizeof addr);*/
addr.sin_family = AF_INET;
addr.sin_port = htons(port);
addr.sin_addr = host_addr;
r = bind(fd, (struct sockaddr *) &addr, sizeof addr);
if (r == -1) return twarn("bind()"), close(fd), -1;
r = listen(fd, 1024);
if (r == -1) return twarn("listen()"), close(fd), -1;
return listen_socket = fd;
}
void
brake()
{
int r;
if (brakes_are_on) return;
brakes_are_on = 1;
twarnx("too many connections; putting on the brakes");
r = event_del(&listen_evq);
if (r == -1) twarn("event_del()");
r = listen(listen_socket, 0);
if (r == -1) twarn("listen()");
}
void
unbrake(evh h)
{
int r;
if (!brakes_are_on) return;
brakes_are_on = 0;
twarnx("releasing the brakes");
accept_handler = h ? : accept_handler;
event_set(&listen_evq, listen_socket, EV_READ | EV_PERSIST,
accept_handler, &listen_evq);
set_main_timeout(main_deadline);
r = listen(listen_socket, 1024);
if (r == -1) twarn("listen()");
}
void
set_main_timeout(time_t deadline)
{
int r;
struct timeval tv = {deadline - time(NULL), 0};
main_deadline = deadline;
r = event_add(&listen_evq, deadline ? &tv : NULL);
if (r == -1) twarn("event_add()");
}
Jump to Line
Something went wrong with that request. Please try again.