brmlife/connection.cc

#include <cassert>
#include <cstdio>
#include <cstring>
#include <iostream>

#include <sys/select.h>
#include <pthread.h>
#include <unistd.h>

#include "agent.h"
#include "connection.h"

#define buf_incomplete(buf) \
	(buf.find("\r\n") == std::string::npos || (buf.find("\r\n") > 0 && buf.find("\r\n\r\n") == std::string::npos))

void
connection::senses(int tick_id, class agent &a)
{
	assert(!negotiation);

	char buf[1024];
	snprintf(buf, sizeof(buf),
		"tick %d\r\n"
		"%s"
		"energy %d\r\n"
		"visual %s %s %s %s %s %s %s %s\r\n"
		"\r\n",
		tick_id,
		a.dead ? "DEAD\r\n" : "",
		a.energy,
		a.tile->tile_in_dir(0, -1).str(),
		a.tile->tile_in_dir(1, -1).str(),
		a.tile->tile_in_dir(1, 0).str(),
		a.tile->tile_in_dir(1, 1).str(),
		a.tile->tile_in_dir(0, 1).str(),
		a.tile->tile_in_dir(-1, 1).str(),
		a.tile->tile_in_dir(-1, 0).str(),
		a.tile->tile_in_dir(-1, -1).str()
	);

	pthread_mutex_lock(&buf_lock);
	out_buf.append(buf);
	pthread_mutex_unlock(&buf_lock);
}

void
connection::bump(void)
{
	/* Must be called with buf_lock held! */
	out_buf.append("BUMP\r\n");
}

void
connection::actions(class agent &agent)
{
	pthread_mutex_lock(&buf_lock);
	if (buf_incomplete(in_buf)) {
		/* Not enough data, needs to wait until next turn, sorry. */
		pthread_mutex_unlock(&buf_lock);
		return;
	}

	int mask = 0;
	while (in_buf.c_str()[0] != '\r') {
		int nlofs = in_buf.find("\r\n");
		std::string line = in_buf.substr(0, nlofs);
		in_buf.erase(0, nlofs + 2);

		int spofs = line.find(' ');
		std::string cmd = line.substr(0, spofs);
		line.erase(0, spofs + 1);

		if (negotiation && !cmd.compare("move")) {
			double rate = agent.attr.move;
			sscanf(line.c_str(), "%lf", &rate);
			if (rate >= 0 && rate <= 1)
				agent.attr.move = rate;
		} else if (negotiation && !cmd.compare("attack")) {
			double rate = agent.attr.attack;
			sscanf(line.c_str(), "%lf", &rate);
			if (rate >= 0 && rate <= 1)
				agent.attr.attack = rate;
		} else if (negotiation && !cmd.compare("defense")) {
			double rate = agent.attr.defense;
			sscanf(line.c_str(), "%lf", &rate);
			if (rate >= 0 && rate <= 1)
				agent.attr.defense = rate;

		} else if (!negotiation && !cmd.compare("move_dir") && !(mask & 1)) {
			int x = 0, y = 0;
			sscanf(line.c_str(), "%d %d", &x, &y);
			if (x < -1) x = -1; if (x > 1) x = 1;
			if (y < -1) y = -1; if (y > 1) y = 1;
			if (!agent.move_dir(x, y))
				bump();
			mask |= 1;
		} else if (!negotiation && !cmd.compare("attack_dir") && !(mask & 2)) {
			int x = 0, y = 0;
			sscanf(line.c_str(), "%d %d", &x, &y);
			if (x < -1) x = -1; if (x > 1) x = 1;
			if (y < -1) y = -1; if (y > 1) y = 1;
			if (!agent.attack_dir(x, y))
				bump();
			mask |= 2;

		} else {
			std::cout << "unknown line " << cmd << " " << line << " ...\n";
		}
	}
	in_buf.erase(0, 2);

	if (negotiation) {
		negotiation = false;
		agent.spawn();
	}

	pthread_mutex_unlock(&buf_lock);
}

void
connection::cancel(void)
{
	pthread_cond_signal(&cancel_cond);
}


void *
conn_thread_worker(void *ctx)
{
	class connection *conn = (class connection *) ctx;
	conn->thread_loop();
	return NULL;
}

void
connection::spawn_thread(void)
{
	pthread_mutex_init(&buf_lock, NULL);
	pthread_cond_init(&cancel_cond, NULL);
	pthread_mutex_init(&cancel_lock, NULL);

	pthread_t tid;
	pthread_create(&tid, NULL, conn_thread_worker, (void *) this);
	pthread_detach(tid);
}

void
connection::thread_loop(void)
{
	while (!error) {
		/* Repeatedly try to write and read stuff. */
		std::string buf;
		ssize_t len;

		pthread_mutex_lock(&buf_lock);
		buf = out_buf;
		pthread_mutex_unlock(&buf_lock);

		if (buf.size() > 0) {
			len = write(fd, buf.c_str(), buf.size());
			if (len < 0) {
				pthread_mutex_lock(&cancel_lock);
				error = true;
			} else {
				pthread_mutex_lock(&buf_lock);
				out_buf.erase(0, len);
				pthread_mutex_unlock(&buf_lock);
			}
		}

		struct timeval tv;
		tv.tv_sec = 0; tv.tv_usec = 0;
		fd_set rfds;
		FD_ZERO(&rfds);
		FD_SET(fd, &rfds);
		while (select(fd + 1, &rfds, NULL, NULL, &tv)) {
			char cbuf[1024];
			len = read(fd, cbuf, sizeof(cbuf));
			if (len < 0) {
				error = true;
			} else if (len == 0) {
				break;
			} else {
				bool want_moar = false;
				pthread_mutex_lock(&buf_lock);
				in_buf.append(cbuf, len);
				want_moar = buf_incomplete(in_buf);
				pthread_mutex_unlock(&buf_lock);
				if (!want_moar)
					break;
			}
		}

		usleep(10000); // XXX: Signal-oriented instead.
	}

	pthread_cond_wait(&cancel_cond, &cancel_lock);
	delete this;
}