hlswmaster-ng/hlswserver.cpp

#include <unistd.h>
#include <string.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <arpa/inet.h>

#include "config.h"
#include "logging.h"
#include "hlswserver.h"

#define HLSW_MASTER_PORT 7140
static const char hlsw_header[] = "\xFF\xFF\xFF\xFFHLSWLANSEARCH";

#define DEFAULT_REBUILD_INTERVAL 5

struct hlsw_entry {
	uint16_t gameid;
	uint32_t ip;
	uint16_t port1;
	uint16_t port2;
} __attribute__ ((packed));

HlswServer::HlswPacket::HlswPacket(HlswPacket* next)
: next(next), count(0)
{
	memcpy(data, hlsw_header, sizeof(hlsw_header));
	memset(data + sizeof(hlsw_header), 0, sizeof(data) - sizeof(hlsw_header));
}

HlswServer::HlswPacket::~HlswPacket()
{
	if (next)
		delete next;
}

bool HlswServer::HlswPacket::addGame(GameEntry* ge)
{
	if (count >= 140)
		return false;
		
	struct hlsw_entry tmp;
	tmp.gameid = ge->gameid;
	tmp.ip = ge->addr.s_addr;
	tmp.port1 = ge->port1;
	tmp.port2 = ge->port2;

	memcpy(data + sizeof(hlsw_header) + count * 10, &tmp, sizeof(tmp));
	count++;
	return true;
}

void HlswServer::HlswPacket::send(int socket, struct sockaddr_in* dst)
{
	if (next)
		next->send(socket, dst);

	if (sendto(socket, data, sizeof(hlsw_header) + count * 10, 0,
			(struct sockaddr *)dst, sizeof(*dst)) < 0) {
		LogSystem::log(LOG_WARN, "HlswPacket::send()");
	}
}

HlswServer::HlswServer(Config& conf, GameList& glist)
: lastUpdate(0)
{
	if ((sock = socket(PF_INET, SOCK_DGRAM, 0)) < 0) {
		LogSystem::log(LOG_CRIT, "HlswServer(): socket()");
		return;
	}

	const char *ip = conf.getString("global", "master_ip", "0.0.0.0");
	LogSystem::log(LOG_NOTICE, "HlswServer listen on %s:%d", ip, HLSW_MASTER_PORT);

	struct sockaddr_in dst;
	dst.sin_family = AF_INET;
	dst.sin_port = htons(HLSW_MASTER_PORT);
	inet_aton(ip, &dst.sin_addr);

	if (bind(sock, (struct sockaddr *)&dst, sizeof(dst)) < 0) {
		LogSystem::log(LOG_ERROR, "HlswServer(): bind()");
		return;
	}

	int interval = conf.getInteger("global", "rebuild_interval", DEFAULT_REBUILD_INTERVAL);
	TimerService::registerTimer(new Timer(new RebuildEvent(*this, glist), interval));
	
	pktlist = new HlswPacket(NULL);
}

HlswServer::~HlswServer()
{
	close(sock);
	delete pktlist;
}

int HlswServer::execute(void* arg)
{
	struct sockaddr_in src;
	unsigned char buf[32];
	socklen_t len;
	int ret;

	while (1) {
		/* auf clientanfrage warten */
		len = sizeof(src);
		ret = recvfrom(sock, buf, sizeof(buf), 0, (struct sockaddr *)&src, &len);

		if (memcmp(buf, hlsw_header, sizeof(hlsw_header))) {
			LogSystem::log(LOG_NOTICE, "HlswServer: not a hlsw packet");
			continue;
		}

		mutex.lock();
		pktlist->send(sock, &src);
		mutex.unlock();
	}
	return 0;
}


void HlswServer::rebuild(GameList& glist) {
	if (glist.getLastUpdate() > lastUpdate) {
		HlswPacket* newlist = new HlswPacket(NULL);
		
		Iterator<GameEntry>* it = glist.createIterator();
		while (it->hasNext()) {
			GameEntry* d = it->next();
			if (!newlist->addGame(d)) {
				newlist = new HlswPacket(newlist);
				newlist->addGame(d);
			}
		}
		lastUpdate = glist.getLastUpdate();
		delete it;
		
		mutex.lock();
		HlswPacket* oldlist = pktlist;
		pktlist = newlist;
		mutex.unlock();
	
		delete oldlist;
	}
}