| 1 | /*
 | 
|---|
| 2 |  * Project: MoleCuilder
 | 
|---|
| 3 |  * Description: creates and alters molecular systems
 | 
|---|
| 4 |  * Copyright (C)  2011 University of Bonn. All rights reserved.
 | 
|---|
| 5 |  * Please see the LICENSE file or "Copyright notice" in builder.cpp for details.
 | 
|---|
| 6 |  */
 | 
|---|
| 7 | 
 | 
|---|
| 8 | /*
 | 
|---|
| 9 |  * Listener.cpp
 | 
|---|
| 10 |  *
 | 
|---|
| 11 |  *  Created on: 22.02.2012
 | 
|---|
| 12 |  *      Author: heber
 | 
|---|
| 13 |  */
 | 
|---|
| 14 | 
 | 
|---|
| 15 | // include config.h
 | 
|---|
| 16 | #ifdef HAVE_CONFIG_H
 | 
|---|
| 17 | #include <config.h>
 | 
|---|
| 18 | #endif
 | 
|---|
| 19 | 
 | 
|---|
| 20 | // boost asio needs specific operator new
 | 
|---|
| 21 | #include <boost/asio.hpp>
 | 
|---|
| 22 | 
 | 
|---|
| 23 | #include "CodePatterns/MemDebug.hpp"
 | 
|---|
| 24 | 
 | 
|---|
| 25 | #include "Listener.hpp"
 | 
|---|
| 26 | 
 | 
|---|
| 27 | #include "CodePatterns/Log.hpp"
 | 
|---|
| 28 | 
 | 
|---|
| 29 | /** Constructor for class Listener.
 | 
|---|
| 30 |  *
 | 
|---|
| 31 |  * @param io_service io service to obtain connections from
 | 
|---|
| 32 |  * @param port service or port number to listen on
 | 
|---|
| 33 |  */
 | 
|---|
| 34 | Listener::Listener(boost::asio::io_service& io_service, unsigned short port) :
 | 
|---|
| 35 |     Exitflag(OkFlag),
 | 
|---|
| 36 |     acceptor_(io_service,
 | 
|---|
| 37 |         boost::asio::ip::tcp::endpoint(boost::asio::ip::tcp::v4(), port)
 | 
|---|
| 38 |     )
 | 
|---|
| 39 | {
 | 
|---|
| 40 |   LOG(1, "Listening on port " << port << ".");
 | 
|---|
| 41 | }
 | 
|---|
| 42 | 
 | 
|---|
| 43 | /** Destructor for class Listener.
 | 
|---|
| 44 |  *
 | 
|---|
| 45 |  */
 | 
|---|
| 46 | Listener::~Listener()
 | 
|---|
| 47 | {}
 | 
|---|
| 48 | 
 | 
|---|
| 49 | /** Initialises the socket in a \a Connection for listening.
 | 
|---|
| 50 |  *
 | 
|---|
| 51 |  */
 | 
|---|
| 52 | void Listener::initiateSocket()
 | 
|---|
| 53 | {
 | 
|---|
| 54 |   // Start an accept operation for worker connections.
 | 
|---|
| 55 |   connection.reset(new Connection(acceptor_.get_io_service()));
 | 
|---|
| 56 |   acceptor_.async_accept(connection->socket(),
 | 
|---|
| 57 |     boost::bind(&Listener::handle_Accept, this,
 | 
|---|
| 58 |       boost::asio::placeholders::error, connection));
 | 
|---|
| 59 | }
 | 
|---|
| 60 | 
 | 
|---|
| 61 | /** Initialises the socket in a \a Connection for listening.
 | 
|---|
| 62 |  *
 | 
|---|
| 63 |  */
 | 
|---|
| 64 | void Listener::closeSocket()
 | 
|---|
| 65 | {
 | 
|---|
| 66 |   boost::system::error_code errorcode;
 | 
|---|
| 67 | 
 | 
|---|
| 68 |   if (connection->socket().is_open())
 | 
|---|
| 69 |   {
 | 
|---|
| 70 |     LOG(1, "INFO: Shutting socket down ...");
 | 
|---|
| 71 |     // Boost documentation recommends calling shutdown first
 | 
|---|
| 72 |     // for "graceful" closing of socket.
 | 
|---|
| 73 |     connection->socket().shutdown(boost::asio::ip::tcp::socket::shutdown_both, errorcode);
 | 
|---|
| 74 |     if (errorcode)
 | 
|---|
| 75 |     {
 | 
|---|
| 76 |       ELOG(1, "socket.shutdown error: " << errorcode.message());
 | 
|---|
| 77 |     }
 | 
|---|
| 78 | 
 | 
|---|
| 79 |     LOG(1, "INFO: Closing socket ...");
 | 
|---|
| 80 |     connection->socket().close(errorcode);
 | 
|---|
| 81 |     if (errorcode)
 | 
|---|
| 82 |     {
 | 
|---|
| 83 |       ELOG(1, "socket.close error: " << errorcode.message());
 | 
|---|
| 84 |     }
 | 
|---|
| 85 |   } else {
 | 
|---|
| 86 |     LOG(1, "INFO: Socket is not open, no need to shut down.");
 | 
|---|
| 87 |   }
 | 
|---|
| 88 | }
 | 
|---|
| 89 | 
 | 
|---|