posix_main.cpp

Go to the documentation of this file.

//
// posix_main.cpp
// ~~~~~~~~~~~~~~
//
// Copyright (c) 2003-2008 Christopher M. Kohlhoff (chris at kohlhoff dot com)
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//

#include <iostream>
#include <string>
#include <boost/asio.hpp>
#include <boost/thread.hpp>
#include <boost/bind.hpp>
#include <boost/lexical_cast.hpp>
#include <airinv/server/AirInvServer.hpp>

#if !defined(_WIN32)

#include <pthread.h>
#include <signal.h>

// //////////////// M A I N ///////////////////
int main(int argc, char* argv[]) {

  try {

    // Check command line arguments.
    if (argc != 5) {
      std::cerr << "Usage: airinvServer <address> <port> <threads> <doc_root>"
                << std::endl;
      std::cerr << "  For IPv4, try:" << std::endl;
      std::cerr << "    receiver 0.0.0.0 80 1 ." << std::endl;
      std::cerr << "  For IPv6, try:" << std::endl;
      std::cerr << "    receiver 0::0 80 1 ." << std::endl;
      return 1;
    }

    // Block all signals for background thread.
    sigset_t new_mask;
    sigfillset (&new_mask);
    sigset_t old_mask;
    pthread_sigmask (SIG_BLOCK, &new_mask, &old_mask);

    // Run server in background thread.
    std::size_t num_threads = boost::lexical_cast<std::size_t>(argv[3]);
    AIRINV::AirInvServer s (argv[1], argv[2], argv[4], num_threads);
    boost::thread t (boost::bind (&AIRINV::AirInvServer::run, &s));

    // Restore previous signals.
    pthread_sigmask (SIG_SETMASK, &old_mask, 0);

    // Wait for signal indicating time to shut down.
    sigset_t wait_mask;
    sigemptyset (&wait_mask);
    sigaddset (&wait_mask, SIGINT);
    sigaddset (&wait_mask, SIGQUIT);
    sigaddset (&wait_mask, SIGTERM);
    pthread_sigmask (SIG_BLOCK, &wait_mask, 0);
    int sig = 0;
    sigwait (&wait_mask, &sig);

    // Stop the server.
    s.stop();
    t.join();

  } catch (std::exception& e) {
    std::cerr << "exception: " << e.what() << "\n";
  }

  return 0;
}

#endif // !defined(_WIN32)