handle_accept.c 2.14 KB
/**
 * \file
 *
 * \author	Georg Hopp
 *
 * \copyright
 * Copyright (C) 2012  Georg Hopp
 *
 * 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 <errno.h>
#include <stdio.h>
#include <stdlib.h>

#include <openssl/ssl.h>

#include "http/worker.h"
#include "server.h"
#include "interface/class.h"
#include "interface/logger.h"
#include "stream.h"

int
serverHandleAccept(Server this, unsigned int i)
{
	char   remoteAddr[16] = "\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0";
	Sock   acc = NULL;
	Stream st;

	if (this->nfds >= this->max_fds) {
		return -1;
	}

	switch(i) {
		case 0:
			// no SSL
			acc = socketAccept(this->sock, &remoteAddr);
			st = new(Stream, STREAM_FD, acc->handle);
			break;

		case 1:
			// SSL
			{
				SSL * ssl = SSL_new(this->ctx);
				acc = socketAccept(this->sockSSL, &remoteAddr);
				SSL_set_fd(ssl, acc->handle);
				SSL_accept(ssl);
				st = new(Stream, STREAM_SSL, ssl);
			}
			break;

		default:
			break;
	}

	if (-1 != acc->handle) {
		// save the socket handle
		(this->conns)[acc->handle].sock   = acc; 

		// clone worker
		(this->conns)[acc->handle].worker = clone(this->worker);
		(this->conns)[acc->handle].stream = st;

		(this->fds)[this->nfds].fd        = acc->handle;
		(this->fds)[this->nfds].events    = POLLIN;
		this->nfds++;
	} else {
		delete(acc);

		switch(errno) {
			case EAGAIN:
				loggerLog(this->logger,
						LOGGER_DEBUG,
						"server accept blocks");
				break;

			default:
				loggerLog(this->logger,
						LOGGER_DEBUG,
						"server accept error");
				break;
		}
	}

	return (acc)? acc->handle : -1;
}

// vim: set ts=4 sw=4: