/*
 * fw/dispatch.c - Wireless protocol dispatcher
 *
 * Written 2012 by Werner Almesberger
 * Copyright 2012 Werner Almesberger
 *
 * 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 2 of the License, or
 * (at your option) any later version.
 */


#include <stddef.h>
#include <stdbool.h>
#include <stdint.h>

#include "rf.h"
#include "proto.h"
#include "dispatch.h"


static uint8_t seq;	/* last sequence number seen */
static uint8_t type;	/* type currently being processed */
static uint8_t limit;	/* last limit seen */
static const struct handler *curr_proto = NULL;


static void send_ack(const uint8_t *buf)
{
	uint8_t ack[3] = { buf[0]+1, buf[1], 0 };

	rf_send(ack, sizeof(ack));
}


static bool answer_ping(const uint8_t *buf)
{
	uint8_t pong[] = { PONG, 0, 0 };

	if (buf[1])
		return 0;
	rf_send(pong, sizeof(pong));
	return 1;
}


bool dispatch(const uint8_t *buf, uint8_t len, const struct handler **protos)
{
	if (len == 3 && buf[0] == PING)
		return answer_ping(buf);

	if (len != 64+3)
		return 0;

	if (!buf[1]) {
		while (*protos) {
			if ((*protos)->type == buf[0])
				break;
			protos++;
		}
		if (!*protos)
			return 0;
		if (!(*protos)->first(buf+3))
			return 0;
		curr_proto = *protos;
		type = buf[0];
		seq = 0;
		limit = buf[2];
		send_ack(buf);
		return 1;
	} else {
		if (!curr_proto)
			return 0;
		if (buf[0] != type)
			return 0;
		if (buf[2] != limit)
			return 0;
		if (buf[1] > limit)
			return 0;

		if (buf[1]+1 == seq) {
			send_ack(buf);
			return 0;
		}
		if (buf[1] != seq)
			return 0;
	}

	if (!curr_proto->more(buf[1], limit, buf+3))
		return 0;
	seq++;
	send_ack(buf);
	return 1;

}