// $Id$
//
//  PLP - An implementation of the PSION link protocol
//
//  Copyright (C) 1999  Philip Proudman
//  Modifications for plptools:
//    Copyright (C) 1999 Fritz Elfert <felfert@to.com>
//
//  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.
//
//  This program is distributed in the hope that it will be useful,
//  but WITHOUT ANY WARRANTY; without even the implied warranty of
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//  GNU General Public License for more details.
//
//  You should have received a copy of the GNU General Public License
//  along with this program; if not, write to the Free Software
//  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
//
//  e-mail philip.proudman@btinternet.com

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include <stdio.h>
#include <string>
#include <stream.h>
#include <stdlib.h>
#include <signal.h>
#include <errno.h>

#include "ncp.h"
#include "bufferstore.h"
#include "ppsocket.h"
#include "socketchan.h"
#include "iowatch.h"
#include "linkchan.h"
#include "link.h"
#include "packet.h"
#include "log.h"

static bool verbose = false;
static bool active = true;

static RETSIGTYPE
term_handler(int)
{
	cout << "Got SIGTERM" << endl;
	signal(SIGTERM, term_handler);
	active = false;
};

static RETSIGTYPE
int_handler(int)
{
	cout << "Got SIGINT" << endl;
	signal(SIGINT, int_handler);
	active = false;
};

void
checkForNewSocketConnection(ppsocket & skt, int &numScp, socketChan ** scp, ncp * a, IOWatch &iow)
{
	string peer;
	ppsocket *next = skt.accept(&peer);
	if (next != NULL) {
		// New connect
		if (verbose)
			cout << "New socket connection from " << peer << endl;
		if ((numScp >= a->maxLinks()) || (!a->gotLinkChannel())) {
			bufferStore a;

			// Give the client time to send it's version request.
			next->dataToGet(1, 0);
			next->getBufferStore(a, false);

			a.init();
			a.addStringT("No Psion Connected\n");
			next->sendBufferStore(a);
			next->closeSocket();
			if (verbose)
			    cout << "rejected" << endl;
		} else
			scp[numScp++] = new socketChan(next, a);
	}
}

void
pollSocketConnections(int &numScp, socketChan ** scp)
{
	for (int i = 0; i < numScp; i++) {
		scp[i]->socketPoll();
		if (scp[i]->terminate()) {
			// Requested channel termination
			delete scp[i];
			numScp--;
			for (int j = i; j < numScp; j++)
				scp[j] = scp[j + 1];
			i--;
		}
	}
}

void
usage()
{
	cerr << "Usage : ncpd [-V] [-v logclass] [-d] [-e] [-p <port>] [-s <device>] [-b <baudrate>]\n";
	exit(1);
}

int
main(int argc, char **argv)
{
	ppsocket skt;
	IOWatch iow;
	int pid;
	bool dofork = true;
	bool autoexit = false;

	int sockNum = DPORT;
	int baudRate = DSPEED;
	const char *serialDevice = NULL;
	short int nverbose = 0;
	short int pverbose = 0;
	short int lverbose = 0;

	struct servent *se = getservbyname("psion", "tcp");
	endservent();
	if (se != 0L)
		sockNum = ntohs(se->s_port);

	// Command line parameter processing
	for (int i = 1; i < argc; i++) {
		if (!strcmp(argv[i], "-p") && i + 1 < argc) {
			sockNum = atoi(argv[++i]);
		} else if (!strcmp(argv[i], "-s") && i + 1 < argc) {
			serialDevice = argv[++i];
		} else if (!strcmp(argv[i], "-v") && i + 1 < argc) {
			i++;
			if (!strcmp(argv[i], "nl"))
				nverbose |= NCP_DEBUG_LOG;
			if (!strcmp(argv[i], "nd"))
				nverbose |= NCP_DEBUG_DUMP;
			if (!strcmp(argv[i], "ll"))
				lverbose |= LNK_DEBUG_LOG;
			if (!strcmp(argv[i], "ld"))
				lverbose |= LNK_DEBUG_DUMP;
			if (!strcmp(argv[i], "pl"))
				pverbose |= PKT_DEBUG_LOG;
			if (!strcmp(argv[i], "pd"))
				pverbose |= PKT_DEBUG_DUMP;
			if (!strcmp(argv[i], "ph"))
				pverbose |= PKT_DEBUG_HANDSHAKE;
			if (!strcmp(argv[i], "m"))
				verbose = true;
			if (!strcmp(argv[i], "all")) {
				nverbose = NCP_DEBUG_LOG | NCP_DEBUG_DUMP;
				lverbose = LNK_DEBUG_LOG | LNK_DEBUG_DUMP;
				pverbose = PKT_DEBUG_LOG | PKT_DEBUG_DUMP |
					PKT_DEBUG_HANDSHAKE;
				verbose = true;
			}
		} else if (!strcmp(argv[i], "-b") && i + 1 < argc) {
			baudRate = atoi(argv[++i]);
		} else if (!strcmp(argv[i], "-d")) {
			dofork = 0;
		} else if (!strcmp(argv[i], "-e")) {
			autoexit = true;
		} else if (!strcmp(argv[i], "-V")) {
			cout << "ncpd version " << VERSION << endl;
			exit(0);
		} else
			usage();
	}

	if (serialDevice == NULL) {
		// If started with -e, assume being started from mgetty and
		// use the tty opened by mgetty instead of the builtin default.
		if (autoexit)
			serialDevice = ttyname(0);
		else
			serialDevice = DDEV;
	}

	if (dofork)
		pid = fork();
	else
		pid = 0;
	switch (pid) {
		case 0:
			signal(SIGTERM, term_handler);
			signal(SIGINT, int_handler);
			skt.setWatch(&iow);
			if (!skt.listen("127.0.0.1", sockNum))
				cerr << "listen on port " << sockNum << ": " << strerror(errno) << endl;
			else {
				if (dofork || autoexit) {
					logbuf dlog(LOG_DEBUG);
					logbuf elog(LOG_ERR);
					ostream lout(&dlog);
					ostream lerr(&elog);
					cout = lout;
					cerr = lerr;
					openlog("ncpd", LOG_CONS|LOG_PID, LOG_DAEMON);
					syslog(LOG_INFO,
					       "daemon started. Listening at 127.0.0.1:%d, using device %s\n",
					       sockNum, serialDevice);
				}
				ncp *a = new ncp(serialDevice, baudRate, iow);
				int numScp = 0;
				socketChan *scp[257]; // MAX_CHANNELS_PSION + 1

				a->setVerbose(nverbose);
				a->setLinkVerbose(lverbose);
				a->setPktVerbose(pverbose);
				while (active) {
					// sockets
					pollSocketConnections(numScp, scp);
					checkForNewSocketConnection(skt, numScp, scp, a, iow);

					// psion
					a->poll();

					if (a->stuffToSend())
						iow.watch(0, 100000);
					else
						iow.watch(1, 0);

					if (a->hasFailed()) {
						if (autoexit)
							break;

						iow.watch(5, 0);
						if (verbose)
							cout << "ncp: restarting\n";
						a->reset();
					}
				}
				delete a;
			}
			skt.closeSocket();
			break;
		case -1:
			cerr << "fork: " << strerror(errno) << endl;
			break;
		default:
			exit(0);
	}
}