util/liblinebasedcomms/tcp_server.h

// copyright 2010 t. schneider tes@mit.edu
//
//
// 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 3 of the License, or
// (at your option) any later version.
//
// This software 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 software.  If not, see <http://www.gnu.org/licenses/>.


#ifndef TCPServer20100719H
#define TCPServer20100719H

#include <iostream>
#include <string>
#include <set>
#include <deque>

#include <boost/bind.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/enable_shared_from_this.hpp>
#include <boost/foreach.hpp>
#include <boost/asio.hpp>

#include "goby/util/string.h"

#include "interface.h"
#include "connection.h"

namespace goby
{
    namespace util
    {
        class TCPConnection;
        
        class TCPServer : public LineBasedInterface
        {
          public:
          TCPServer(unsigned port,
                    const std::string& delimiter = "\r\n")
              : LineBasedInterface(delimiter),
                acceptor_(LineBasedInterface::io_service_,
                          boost::asio::ip::tcp::endpoint(boost::asio::ip::tcp::v4(), port))
                {  }

            std::string local_endpoint() { return goby::util::as<std::string>(acceptor_.local_endpoint()); }

            

            // for access to static members of LineBasedInterface
            friend class TCPConnection;
            friend class LineBasedConnection<boost::asio::ip::tcp::socket>;
          private:
            void do_start()
            { start_accept(); }
        
            void do_write(const protobuf::Datagram& line);
            void do_close(const boost::system::error_code& error);
            
          private:
            void start_accept();
            void handle_accept(boost::shared_ptr<TCPConnection> new_connection,
                               const boost::system::error_code& error);
            
          private:
            static std::map<std::string, TCPServer*> inst_;
            std::string server_;
            boost::asio::ip::tcp::acceptor acceptor_;
            boost::shared_ptr<TCPConnection> new_connection_;
            std::set< boost::shared_ptr<TCPConnection> > connections_;    
        };


        class TCPConnection : public boost::enable_shared_from_this<TCPConnection>,
            public LineBasedConnection<boost::asio::ip::tcp::socket>
            {
              public:
                static boost::shared_ptr<TCPConnection> create();
                
                boost::asio::ip::tcp::socket& socket()
                { return socket_; }
    
                void start()
                { socket_.get_io_service().post(boost::bind(&TCPConnection::read_start, this)); }

                void write(const protobuf::Datagram& msg)
                { socket_.get_io_service().post(boost::bind(&TCPConnection::socket_write, this, msg)); }    

                void close(const boost::system::error_code& error)
                { socket_.get_io_service().post(boost::bind(&TCPConnection::socket_close, this, error)); }
                
                std::string local_endpoint() { return goby::util::as<std::string>(socket_.local_endpoint()); }
                std::string remote_endpoint() { return goby::util::as<std::string>(socket_.remote_endpoint()); }
                
              private:
                void socket_write(const protobuf::Datagram& line);
                void socket_close(const boost::system::error_code& error);
    
              TCPConnection()
                  : socket_(LineBasedInterface::io_service_)
                { }
    

              private:
                boost::asio::ip::tcp::socket socket_;
            };
    }

}

#endif