/****************************************************************************

NAME:
   packet.c -- a packet-sniffing engine for reading from GPS devices

DESCRIPTION:

Initial conditions of the problem:

1. We have an file descriptor open for read. The device on the other end is 
   sending packets at us.  

2. It may require more than one read to gather a packet.  Reads may span packet
   boundaries.
  
3. There may be leading garbage before the first packet.  After the first
   start-of-packet, the input should be well-formed.

The problem: how do we recognize which kind of packet we're getting?

No need to handle Garmin, we know that type by the fact we're connected
to the driver.  But we need to be able to tell the others apart and 
distinguish them from baud barf.

***************************************************************************/
#include <ctype.h>
#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <sys/time.h>
#include "gpsd.h"

static int getch(struct gps_session_t *session)
{
    if (session->inbufptr >= session->inbuffer + session->inbuflen)
    {
	int st, i;
	unsigned char buf2[BUFSIZ];

#ifndef TESTMAIN
	st = read(session->gNMEAdata.gps_fd, 
		  session->inbuffer + session->inbuflen, 
		  MAX_PACKET_LENGTH - session->inbuflen);
	if (st < 0)
#endif /* !TESTMAIN */
	    longjmp(session->packet_error, 1);
	buf2[0] = '\0';
	for (i = 0; i < st; i++)
	    sprintf(buf2+strlen(buf2), "%02x", session->inbuffer[session->inbuflen+i]);
	gpsd_report(6, "Packet buffer fill %d: %s\n", st, buf2);
	session->inbuflen += st;
    }
    return session->outbuffer[session->outbuflen++] = *session->inbufptr++;
}

static int packet_reread(struct gps_session_t *session)
/* packet type grab failed, set us up for reread */
{
    gpsd_report(6, "Packet buffer reread\n");
    session->inbufptr = session->inbuffer;
    session->outbuflen = 0;
    return 0;
}

static void packet_shift(struct gps_session_t *session)
/* packet grab failed, shift the input buffer to discard a character */ 
{
    gpsd_report(6, "Packet buffer shift\n");
    session->inbufptr = memmove(session->inbuffer, 
			       session->inbuffer+1, 
			       --session->inbuflen);
    session->inbuffer[session->inbuflen] = '\0';
    memset(session->outbuffer, '\0', sizeof(session->outbuffer)); 
    session->outbuflen = 0;
}

static int get_nmea(struct gps_session_t *session)
{
    unsigned short crc, n;
    unsigned char c, csum[3], *trailer;

    gpsd_report(6, "NMEA packet get\n");
    if (getch(session) !='$')
	return packet_reread(session);
    c = getch(session);
    if (c == 'G' && getch(session) == 'P')	/* NMEA sentence */
	/* OK */;
    else if (c == 'P')				/* vendor private extension */
	/* OK */;
    else if (c == 'A' && getch(session) == 'c')	/* SiRF $Ack */
	/* OK */;
    else
	return packet_reread(session);
    for (;;) {
	c = getch(session);
	if (c == '\r')
	    break;
	if (!isprint(c))
	    return packet_reread(session);
	if (session->outbuflen > NMEA_MAX)
	    return packet_reread(session);
    }
    if (getch(session) != '\n')
	return packet_reread(session);
    if (*(trailer = &session->outbuffer[session->outbuflen-5]) == '*') {
	crc = 0;
	for (n = 1; n < session->outbuflen-5; n++)
	    crc ^= session->outbuffer[n];
	sprintf(csum, "%02X", crc);
	if (toupper(csum[0])!=toupper(trailer[1])
	    || toupper(csum[1])!=toupper(trailer[2]))
	    return packet_reread(session);
    }
    session->outbuffer[session->outbuflen] = '\0';
    return 1;
}

static int get_sirf(struct gps_session_t *session)
{
    unsigned short length, n, checksum, crc, id;

    gpsd_report(6, "SiRF packet get\n");
    if (getch(session) != 0xa0 || getch(session) != 0xa2)
	return packet_reread(session);
    length = (getch(session) << 8) | getch(session);
    id = (getch(session) << 8) | getch(session);
    if (length > MAX_PACKET_LENGTH-6) {
	gpsd_report(6, "packet rejected, id %d, length %d too large\n", id, length);
	return packet_reread(session);
    }
    for (n = 0; n < length-2; n++)
	getch(session);
    checksum = (getch(session) << 8) | getch(session);
    crc = 0;
    for (n = 0; n < length; n++)
	crc += session->outbuffer[4+n];
    crc &= 0x7fff;
    if (crc != checksum) {
	gpsd_report(6, "SiRF packet rejected, checksum incorrect\n");
	return packet_reread(session);
    }
    if (getch(session) != 0xb0 || getch(session) != 0xb3) {
	gpsd_report(6, "SiRF packet rejected, trailer bytes wrong\n");
	return packet_reread(session);
    }
    return 1;
}

#ifdef __UNUSED__
static int get_zodiac(struct gps_session_t *session)
{
    unsigned short length, n, checksum, crc;

    gpsd_report(6, "Zodiac packet get\n");
    if (getch(session) != 0x81 || getch(session) != 0x00)
	return packet_reread(session);
    getch(session); getch(session);	/* skip the ID */
    length = (getch(session) << 8) | getch(session);
    if (length > MAX_PACKET_LENGTH-10)
	return packet_reread(session);
    checksum = (getch(session) << 8) | getch(session);
    for (n = 0; n < length; n++)
	getch(session);
    crc = 0;
    for (n = 0; n < length; n++)
	crc += session->outbuffer[10+n];
    if (-crc != checksum)
	return packet_reread(session);
    return 1;
}
#endif /* __UNUSED__ */

/* entry points begin here */

void packet_accept(struct gps_session_t *session)
/* discard data in the packet-out buffer */
{
    gpsd_report(6, "Packet buffer accept\n");
    if (session->inbufptr)
	memmove(session->inbuffer, 
	    session->inbuffer + session->outbuflen,
	    session->inbuflen - session->outbuflen);
    session->inbufptr = session->inbuffer;
    session->inbuflen -= session->outbuflen;
    memset(session->outbuffer, '\0', sizeof(session->outbuffer));
    session->outbuflen = 0;
}

int packet_sniff(struct gps_session_t *session)
/* grab a packet, return its type */
{
    if (setjmp(session->packet_error))
	return BAD_PACKET;
    else {
	int n;
	packet_accept(session);
	for (n = MAX_PACKET_LENGTH * 3; n; n--) {
	    if (get_nmea(session)) {
		session->outbuffer[session->outbuflen] = '\0';
		return NMEA_PACKET;
	    } else if (get_sirf(session))
		return SIRF_PACKET;
#ifdef TRIPMATE_ENABLE
	    else if (!memcmp("ASTRAL", session->inbuffer, 6)) {
		strcpy(session->outbuffer, "ASTRAL");
		session->outbuflen = 6;
		return NMEA_PACKET;
	    }
#endif /* TRIPMATE_ENABLE */
#ifdef EARTHMATE_ENABLE
	    else if (!memcmp("EARTHA", session->inbuffer, 6)) {
		strcpy(session->outbuffer, "EARTHA");
		session->outbuflen = 6;
		return NMEA_PACKET;
	    }
#endif /* EARTHMATE_ENABLE */
	    else
		packet_shift(session);
	}
	return BAD_PACKET;
    }
}

#define MAKE_PACKET_GRABBER(outname, inname, maxlength)	int \
	outname(struct gps_session_t *session) \
	{ \
	    int maxgarbage = maxlength; \
	    packet_accept(session); \
	    session->gNMEAdata.d_xmit_time = timestamp(); \
	    while (maxgarbage--) { \
		if (inname(session)) { \
		    return 1; \
		} else \
		    packet_shift(session); \
	    } \
	    session->gNMEAdata.d_recv_time = timestamp(); \
	    return 0; \
	}

MAKE_PACKET_GRABBER(packet_get_nmea, get_nmea, NMEA_MAX * 3)
MAKE_PACKET_GRABBER(packet_get_sirf, get_sirf, MAX_PACKET_LENGTH)
// MAKE_PACKET_GRABBER(packet_get_zodiac, get_zodiac, MAX_PACKET_LENGTH)

#ifdef TESTMAIN


int main(int argc, char *argv[])
{
    struct map {
	char		*legend;
	unsigned char	test[MAX_PACKET_LENGTH];
	int		testlen;
	int		garbage_offset;
	int		type;
    };
    struct map testloads[] = {
	/* NMEA tests */
	{
	    "NMEA packet with checksum (1)",
	    "$GPVTG,308.74,T,,M,0.00,N,0.0,K*68\r\n",
	    36,
	    0,
	    NMEA_PACKET,
	},
	{
	    "NMEA packet with checksum (2)",
	    "$GPGGA,110534.994,4002.1425,N,07531.2585,W,0,00,50.0,172.7,M,-33.8,M,0.0,0000*7A\r\n",
	    82,
	    0,
	    NMEA_PACKET,
	},
	{
	    "NMEA packet with checksum and leading garbage",
	    "\000\277\000\277$GPVTG,308.74,T,,M,0.00,N,0.0,K*68\r\n",
	    40,
	    4,
	    NMEA_PACKET,
	},
	{
	    "NMEA packet without checksum",
	    "$PSRF105,1\r\n",
	    12,
	    0,
	    NMEA_PACKET,
	},
	{
	    "NMEA packet with wrong checksum",
	    "$GPVTG,308.74,T,,M,0.00,N,0.0,K*28\r\n",
	    36,
	    0,
	    BAD_PACKET,
	},
	/* SiRF tests */
	{
	    "SiRF WAAS version ID",
	    {
		0xA0, 0xA2, 0x00, 0x15, 
		0x06, 0x06, 0x31, 0x2E, 0x32, 0x2E, 0x30, 0x44, 
		0x4B, 0x49, 0x54, 0x31, 0x31, 0x39, 0x20, 0x53, 
		0x4D, 0x00, 0x00, 0x00, 0x00,
		0x03, 0x82, 0xB0, 0xB3},
	    29,
	    0,
	    SIRF_PACKET,
	},
	{
	    "SiRF WAAS version ID with leading garbage",
	    {
		0xff, 0x00, 0xff,
		0xA0, 0xA2, 0x00, 0x15, 
		0x06, 0x06, 0x31, 0x2E, 0x32, 0x2E, 0x30, 0x44, 
		0x4B, 0x49, 0x54, 0x31, 0x31, 0x39, 0x20, 0x53, 
		0x4D, 0x00, 0x00, 0x00, 0x00,
		0x03, 0x82, 0xB0, 0xB3},
	    32,
	    3,
	    SIRF_PACKET,
	},
	{
	    "SiRF WAAS version ID with wrong checksum",
	    {
		0xA0, 0xA2, 0x00, 0x15, 
		0x06, 0x06, 0x31, 0x2E, 0x32, 0x2E, 0x30, 0x44, 
		0x4B, 0x49, 0x54, 0x31, 0x31, 0x39, 0x20, 0x53, 
		0x4D, 0x00, 0x00, 0x00, 0x00,
		0x03, 0x00, 0xB0, 0xB3},
	    29,
	    0,
	    BAD_PACKET,
	},
	{
	    "SiRF WAAS version ID with bad length",
	    {
		0xA0, 0xA2, 0xff, 0x15, 
		0x06, 0x06, 0x31, 0x2E, 0x32, 0x2E, 0x30, 0x44, 
		0x4B, 0x49, 0x54, 0x31, 0x31, 0x39, 0x20, 0x53, 
		0x4D, 0x00, 0x00, 0x00, 0x00,
		0x03, 0x82, 0xB0, 0xB3},
	    29,
	    0,
	    BAD_PACKET,
	},
    };
    struct map *mp;
    struct gps_session_t state;
    int st;
    unsigned char *cp;

    for (mp = testloads; mp < testloads + sizeof(testloads)/sizeof(testloads[0]); mp++) {
	if (setjmp(state.packet_error))
	    st = BAD_PACKET;
	else {
	    packet_flush(&state);
	    memcpy(state.inbuffer, mp->test, state.inbuflen = mp->testlen);
	    state.inbufptr = state.inbuffer;
	    st = packet_sniff(&state);
	}
	if (st != mp->type)
	    printf("%s test FAILED (packet type wrong).\n", mp->legend);
	else if (memcmp(mp->test + mp->garbage_offset, state.outbuffer, state.outbuflen))
	    printf("%s test FAILED (data garbled).\n", mp->legend);
	else
	    printf("%s test succeeded.\n", mp->legend);
#ifdef DUMPIT
	for (cp = state.outbuffer; 
	     cp < state.outbuffer + state.outbuflen; 
	     cp++) {
	    if (st != NMEA_PACKET)
		printf(" 0x%02x", *cp);
	    else if (*cp == '\r')
		fputs("\\r", stdout);
	    else if (*cp == '\n')
		fputs("\\n", stdout);
	    else if (isprint(*cp))
		putchar(*cp);
	    else
		printf("\\x%02x", *cp);
	}
	putchar('\n');
#endif /* DUMPIT */
    }
}
#endif /* TESTMAIN */