handle_accept.c 2.57 KB
/**
 * \file
 *
 * \author	Georg Hopp
 *
 * \copyright
 * Copyright © 2014 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 "trbase.h"
#include "trio.h"
#include "http/worker.h"
#include "server.h"

int
serverHandleAccept(Server this, unsigned int i)
{
	TR_Socket acc = NULL;
	TR_Stream    st;

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

	acc = (TR_Socket)TR_socketAccept((0 == i)
			? (TR_TcpSocket)this->sock
			: (TR_TcpSocket)this->sockSSL);

	if (NULL != acc && -1 != acc->handle) {
		TR_socketNonblock(acc);

		switch(i) {
			case 0:
				// no SSL
				st = TR_new(TR_Stream, TR_STREAM_FD, acc->handle);
				break;

			case 1:
				// SSL
				{
					SSL * ssl = SSL_new(this->ctx);
					SSL_set_fd(ssl, acc->handle);
					SSL_accept(ssl);
					st = TR_new(TR_Stream, TR_STREAM_SSL, ssl);
				}
				break;

			default:
				st = NULL;
				break;
		}

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

		// clone worker
		(this->conns)[acc->handle].worker        = TR_clone(this->worker);
		/**
		 * TODO
		 * workers need an interface to set the socket within them...
		 * as I currently only have http workers I do a cast here
		 * but this is not future save.
		 */
		((HttpWorker)(this->conns)[acc->handle].worker)->socket = acc;
		(this->conns)[acc->handle].stream        = st;

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

		switch(errno) {
			case EAGAIN|EWOULDBLOCK:
			case EINTR:
				TR_loggerLog(this->logger,
						TR_LOGGER_DEBUG,
						"server accept blocks");
				return -1;
				break;

			default:
				TR_loggerLog(this->logger,
						TR_LOGGER_DEBUG,
						"server accept error");
				return -2;
				break;
		}
	}

	if (0 == this->nfds%200) {
		TR_loggerLog(this->logger,
				TR_LOGGER_DEBUG, "paralel connections: %lu", this->nfds);
	}

	return acc->handle;
}

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