| 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 |  | 
|---|