QnetGateway/IRCReceiver.cpp

#include <sys/types.h>
#include <sys/socket.h>

#include "IRCutils.h"
#include "IRCMessage.h"
#include "IRCReceiver.h"

static int doRead(int sock, char * buf, int buf_size)
{
	struct timeval tv;
	tv.tv_sec = 1;
	tv.tv_usec = 0;
	fd_set rdset;
	fd_set errset;

	FD_ZERO(&rdset);
	FD_ZERO(&errset);
	FD_SET(sock, &rdset);
	FD_SET(sock, &errset);

	int res;

	res = select(sock+1, &rdset, NULL, &errset, &tv);

	if ( res < 0 ) {
		traceit("IRCReceiver::doread: select() error.\n");
		return -1;
	} else if ( res > 0 ) {
		if (FD_ISSET(sock, &errset)) {
			traceit("IRCReceiver::doRead: FD_ISSET error\n");
			return -1;
		}

		if (FD_ISSET(sock, &rdset)) {
			res = recv(sock, buf, buf_size, 0);

			if (res < 0) {
				traceit("IRCReceiver::doRead: recv error\n");
				return -1;
			} else if (res == 0) {
				traceit("IRCReceiver::doRead: EOF read==0\n");
				return -1;
			} else
				return res;
		}

	}

	return 0;
}

void IRCReceiver::Entry()
{
	IRCMessage *m = new IRCMessage();

	int i;
	int state = 0;

	while (!terminateThread) {
		char buf[200];
		int r = doRead(sock, buf, sizeof buf);

		if (r < 0) {
			recvQ->signalEOF();
			delete m;  // delete unfinished IRCMessage
			break;
		}

		for (i=0; i<r; i++) {
			char b = buf[i];

			if (b > 0) {
				if (b == '\n') {
					recvQ->putMessage(m);
					m = new IRCMessage();
					state = 0;
				} else if (b == '\r') {
					// do nothing
				} else switch (state) {
					case 0:
						if (b == ':') {
							state = 1; // prefix
						} else if (b == ' ') {
							// do nothing
						} else {
							m->command.push_back(b);
							state = 2; // command
						}
						break;

					case 1:
						if (b == ' ') {
							state = 2; // command is next
						} else {
							m->prefix.push_back(b);
						}
						break;

					case 2:
						if (b == ' ') {
							state = 3; // params
							m->numParams = 1;
							m->params.push_back("");
						} else {
							m->command.push_back(b);
						}
						break;

					case 3:
						if (b == ' ') {
							m->numParams++;
							if (m->numParams >= 15) {
								state = 5; // ignore the rest
							}

							m->params.push_back("");
						} else if ((b == ':') && (m->params[m->numParams - 1].length() == 0)) {
							state = 4; // rest of line is this param
						} else {
							m->params[m->numParams - 1].push_back(b);
						}
						break;

					case 4:
						m->params[m->numParams - 1].push_back(b);
						break;
					} // switch
			} // if
		} // for
	} // while

	return;
}

IRCReceiver::IRCReceiver(int sock, IRCMessageQueue *q)
{
	this->sock = sock;
	recvQ = q;
}

IRCReceiver::~IRCReceiver()
{
}

bool IRCReceiver::startWork()
{
	terminateThread = false;
	rec_thread = std::async(std::launch::async, &IRCReceiver::Entry, this);
	return true;
}

void IRCReceiver::stopWork()
{
	terminateThread = true;
	rec_thread.get();
}