tcpsrv/tcpsrv.cc

Go to the documentation of this file.

//
// tcpsrv.cc - This is the main component of the tcpsrv program, which will
// accept client connections through a UNIX domain socket, and maintains
// a consistent connection to a host/port on the other end.
//
// ***** BEGIN LICENSE BLOCK *****
// Version: MPL 1.1/GPL 2.0/LGPL 2.1
// 
// The contents of this file are subject to the Mozilla Public License
// Version 1.1 (the "License"); you may not use this file except in
// compliance with the License. You may obtain a copy of the License at
// http://www.mozilla.org/MPL/
// 
// Software distributed under the License is distributed on an "AS IS"
// basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
// the License for the specific language governing rights and limitations
// under the License.
// 
// The Original Code is the consmgr network/serial-line monitoring package.
// 
// The Initial Developer of the Original Code is Chris P. Ross.
// Portions created by the Initial Developer are Copyright (C) 2000-2008
// the Initial Developer.  All Rights Reserved.
// 
// Contributor(s):
// 
// Alternatively, the contents of this file may be used under the terms of
// either the GNU General Public License Version 2 or later (the "GPL"), or
// the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
// in which case the provisions of the GPL or the LGPL are applicable instead
// of those above. If you wish to allow use of your version of this file only
// under the terms of either the GPL or the LGPL, and not to allow others to
// use your version of this file under the terms of the MPL, indicate your
// decision by deleting the provisions above and replace them with the notice
// and other provisions required by the GPL or the LGPL. If you do not delete
// the provisions above, a recipient may use your version of this file under
// the terms of any one of the MPL, the GPL or the LGPL.
// 
// ***** END LICENSE BLOCK *****

extern "C" {
#include <unistd.h>
#include <netinet/in.h>
#include <signal.h>
}

#include <fstream>

#include "consmgr.h"
#include "cmconfig.h"
#include "mastercont.h"

/* Internal function prototypes */
static void usage(void);

int
main(int argc, char *argv[])
{
    int                 ch;

    // Shut off clog until we parse options...
    std::ofstream       *null = new std::ofstream("/dev/null");
    clog.rdbuf(null->rdbuf());

    bool                        debug = false;
    bool                        foreground = false; 
    Config                      *cnf = NULL;

    // ConfigFactory::config_object() may throw exceptions
    try {
        while ((ch = getopt(argc, argv, "c:fDtv")) != -1) {
            switch (ch) {
              case 'c':
                if (cnf)
                    err(5,"Only one configuration file [currently] allowed");
                else
                    cnf = ConfigFactory::config_object(optarg);
                break;
              case 'D':
                debug = true;
                break;
              case 'f':
                foreground = true;
              case 't':
                // XXX - serv.add_decoder(new Telnet);
                break;
              case 'v':
                std::cout << "tcpsrv, part of consmgr, built " << __DATE__
                          << " " << __TIME__ << endl;
                break;
              case '?':
                usage();
                exit(1);
            }
        }

        if (cnf == NULL)
            cnf = ConfigFactory::config_object("/etc/consmgr.conf");
        
        argc -= optind;
        argv += optind;

        if (argc != 1) {
            usage();
            exit(1);
        }
    } catch (ConfigException &e) {
        cerr << e.what() << endl;
        exit(4);
    } catch (CMException &e) {
        cerr << e.what() << endl;
        exit(5);
    } catch (const string &error) {
        cerr << error << endl;
        exit(7);
    } catch (...) {
        cerr << "Other random exception thrown while parsing arguments"
             << endl;
        exit(7);
    }

    // Set up the debugging stream to go to stderr, and nuke the
    // temporary null out we made for it...
    if (debug) {
        clog.rdbuf(cerr.rdbuf());
        delete null;
    }

    // Look up the configuration for the specified connection
    try {
        cnf->load_entry(argv[0]);
    } catch (ConfigUnfoundException &e) {
        // XXX - anything else we care to do here?
        cerr << e.what() << endl;
        exit(2);
    } CATCH_STR_OR_DIE(2) {
        cerr << "** Uncaught exception in Config::load_entry(" << argv[0]
             << ")!!" << endl;
        exit(2);
    }

    // Start and run the MasterController (as a server)
    ConfigP     config(cnf);    // This is safe because cnf was dyn. allocated
    MasterController::run(config, true);

    exit(0);
}

static void
usage(void)
{
    cerr << "Usage: " << __progname << " [-D] [-t] [-c <config-file>]"
         << " <connection-name>" << endl;
    return;
}

---