/*-*-c++-*- * $Id$ * * This file is part of plptools. * * Copyright (C) 1999 Philip Proudman * Copyright (C) 1999-2001 Fritz Elfert * * 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 * */ #ifdef HAVE_CONFIG_H #include "config.h" #endif #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include "ncp.h" #include "socketchan.h" #include "linkchan.h" #include "link.h" #include "packet.h" #ifndef _GNU_SOURCE #define _GNU_SOURCE #endif #include using namespace std; static bool verbose = false; static bool active = true; static bool autoexit = false; static ncp *theNCP = NULL; static IOWatch iow; static IOWatch accept_iow; static ppsocket skt; static int numScp = 0; static socketChan *scp[257]; // MAX_CHANNELS_PSION + 1 logbuf ilog(LOG_INFO, STDOUT_FILENO); logbuf dlog(LOG_DEBUG, STDOUT_FILENO); logbuf elog(LOG_ERR, STDERR_FILENO); ostream linf(&ilog); ostream lout(&dlog); ostream lerr(&elog); static RETSIGTYPE term_handler(int) { linf << _("Got SIGTERM") << endl; signal(SIGTERM, term_handler); active = false; }; static RETSIGTYPE int_handler(int) { linf << _("Got SIGINT") << endl; signal(SIGINT, int_handler); active = false; }; void checkForNewSocketConnection() { string peer; if (accept_iow.watch(5,0) <= 0) { return; } ppsocket *next = skt.accept(&peer, &iow); if (next != NULL) { next->setWatch(&iow); // New connect if (verbose) lout << "New socket connection from " << peer << endl; if ((numScp >= theNCP->maxLinks()) || (!theNCP->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); delete next; if (verbose) lout << "rejected" << endl; } else scp[numScp++] = new socketChan(next, theNCP); } } void * pollSocketConnections(void *) { while (active) { iow.watch(0, 10000); 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--; } } } return NULL; } static void help() { cout << _( "Usage: ncpd [OPTIONS]...\n" "\n" "Supported options:\n" "\n" " -d, --dontfork Run in foreground don't fork\n" " -h, --help Display this text.\n" " -V, --version Print version and exit.\n" " -e, --autoexit Exit after device is disconnected.\n" " -v, --verbose=LOGCLASS Enable logging of LOGCLASS events\n" " Valid log classes are:\n" " m - main program\n" " nl - NCP protocol log\n" " nd - NCP protocol data dump\n" " ll - PLP protocol log\n" " ld - PLP protocol data dump\n" " pl - physical I/O log\n" " ph - physical I/O handshake\n" " pd - physical I/O data dump\n" " all - All of the above\n" " -s, --serial=DEV Use serial device DEV.\n" " -b, --baudrate=RATE Set serial speed to BAUD.\n" ); cout << #if DSPEED > 0 #define SPEEDSTR(x) #x _(" Default: ") << DSPEED << ".\n"; #else _(" Default: Autocycle 115.2k, 57.6k 38.4k, 19.2k\n"); #endif cout << _( " -p, --port=[HOST:]PORT Listen on host HOST, port PORT.\n" " Default for HOST: 127.0.0.1\n" " Default for PORT: " ) << DPORT << "\n\n"; } static void usage() { cerr << _("Try `ncpd --help' for more information") << endl; } static struct option opts[] = { {"dontfork", no_argument, 0, 'd'}, {"autoexit", no_argument, 0, 'e'}, {"help", no_argument, 0, 'h'}, {"version", no_argument, 0, 'V'}, {"verbose", required_argument, 0, 'v'}, {"port", required_argument, 0, 'p'}, {"serial", required_argument, 0, 's'}, {"baudrate", required_argument, 0, 'b'}, {NULL, 0, 0, 0 } }; static void parse_destination(const char *arg, const char **host, int *port) { if (!arg) return; // We don't want to modify argv, therefore copy it first ... char *argcpy = strdup(arg); char *pp = strchr(argcpy, ':'); if (pp) { // host.domain:400 // 10.0.0.1:400 *pp ++= '\0'; *host = argcpy; } else { // 400 // host.domain // host // 10.0.0.1 if (strchr(argcpy, '.') || !isdigit(argcpy[0])) { *host = argcpy; pp = 0L; } else pp = argcpy; } if (pp) *port = atoi(pp); } static void * link_thread(void *arg) { while (active) { // psion iow.watch(1, 0); if (theNCP->hasFailed()) { if (autoexit) { active = false; break; } iow.watch(5, 0); if (verbose) lout << "ncp: restarting\n"; theNCP->reset(); } } return NULL; } int main(int argc, char **argv) { int pid; bool dofork = true; int sockNum = DPORT; int baudRate = DSPEED; const char *host = "127.0.0.1"; const char *serialDevice = NULL; unsigned short nverbose = 0; struct servent *se = getservbyname("psion", "tcp"); dlog.setOn(false); elog.setOn(false); endservent(); if (se != 0L) sockNum = ntohs(se->s_port); while (1) { int c = getopt_long(argc, argv, "hdeVb:s:p:v:", opts, NULL); if (c == -1) break; switch (c) { case '?': usage(); return -1; case 'V': cout << _("ncpd Version ") << VERSION << endl; return 0; case 'h': help(); return 0; case 'v': if (!strcmp(optarg, "nl")) nverbose |= NCP_DEBUG_LOG; if (!strcmp(optarg, "nd")) nverbose |= NCP_DEBUG_DUMP; if (!strcmp(optarg, "ll")) nverbose |= LNK_DEBUG_LOG; if (!strcmp(optarg, "ld")) nverbose |= LNK_DEBUG_DUMP; if (!strcmp(optarg, "pl")) nverbose |= PKT_DEBUG_LOG; if (!strcmp(optarg, "pd")) nverbose |= PKT_DEBUG_DUMP; if (!strcmp(optarg, "ph")) nverbose |= PKT_DEBUG_HANDSHAKE; if (!strcmp(optarg, "m")) verbose = true; if (!strcmp(optarg, "all")) { nverbose = NCP_DEBUG_LOG | NCP_DEBUG_DUMP | LNK_DEBUG_LOG | LNK_DEBUG_DUMP | PKT_DEBUG_LOG | PKT_DEBUG_DUMP | PKT_DEBUG_HANDSHAKE; verbose = true; } break; case 'd': dofork = 0; break; case 'e': autoexit = true; break; case 'b': if (!strcmp(optarg, "auto")) baudRate = -1; else baudRate = atoi(optarg); break; case 's': serialDevice = optarg; break; case 'p': parse_destination(optarg, &host, &sockNum); break; } } if (optind < argc) { usage(); return -1; } 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(&accept_iow); if (!skt.listen(host, sockNum)) cerr << "listen on " << host << ":" << sockNum << ": " << strerror(errno) << endl; else { if (dofork || autoexit) { openlog("ncpd", LOG_CONS|LOG_PID, LOG_DAEMON); dlog.setOn(true); elog.setOn(true); ilog.setOn(true); linf << _("daemon started. Listening at ") << host << ":" << sockNum << _(" using device ") << serialDevice << endl; setsid(); chdir("/"); int devnull = open("/dev/null", O_RDWR, 0); if (devnull != -1) { dup2(devnull, STDIN_FILENO); dup2(devnull, STDOUT_FILENO); dup2(devnull, STDERR_FILENO); if (devnull > 2) close(devnull); } } memset(scp, 0, sizeof(scp)); theNCP = new ncp(serialDevice, baudRate, nverbose); if (!theNCP) { lerr << "Could not create NCP object" << endl; exit(-1); } pthread_t thr_a, thr_b; if (pthread_create(&thr_a, NULL, link_thread, NULL) != 0) { lerr << "Could not create Link thread" << endl; exit(-1); } if (pthread_create(&thr_a, NULL, pollSocketConnections, NULL) != 0) { lerr << "Could not create Socket thread" << endl; exit(-1); } while (active) checkForNewSocketConnection(); linf << _("terminating") << endl; void *ret; pthread_join(thr_a, &ret); pthread_join(thr_b, &ret); delete theNCP; } skt.closeSocket(); break; case -1: lerr << "fork: " << strerror(errno) << endl; break; default: exit(0); } } /* * Local variables: * c-basic-offset: 4 * End: */