boost - socket通信

 備忘使用

 

#include <boost/bind.hpp>
#include <boost/asio.hpp>
#include <boost/thread.hpp>
using boost::asio::io_service;
using boost::asio::ip::tcp;
using boost::system::error_code;
using boost::asio::deadline_timer;
using boost::asio::buffer;

class MyNet
{
public:
    void Accept(unsigned short port, boost::function<void ()> fun)
    {
        _pAcceptor = new tcp::acceptor(_ios, tcp::endpoint(tcp::v4(), port));
        _pTimer = new deadline_timer(_ios);
        _pClentSkt = new tcp::socket(_ios);
        _pAcceptor->async_accept(*_pClentSkt,
             boost::bind(&MyNet::AcceptHandle, this,
             boost::asio::placeholders::error, //傳遞錯誤碼
             fun//傳遞一個函數指針
        ));
        _ios.reset(); //充值事件循環狀態
        if (_ios.run() == 0) //啓動事件循環
        {
            throw std::runtime_error("io_service run return 0.");
        }
    }
    void Send(const string& data)
    {
        _pClentSkt->write_some(buffer(data, data.size()));
    }
    string Recv()
    {
        _ret = _pClentSkt->read_some(buffer(_buf, 1024));
        return string(_buf, _ret);
    }
    void Close()
    {
        _mutex.lock();
        _pAcceptor->close();
        _pClentSkt->shutdown(tcp::socket::shutdown_both);
        _pClentSkt->close();
        _ios.stop();
        _mutex.unlock();
    }
    string GetClientIp()
    {//獲得客戶端的ip
        boost::asio::ip::tcp::endpoint ep = _pClentSkt->remote_endpoint();
        return ep.address().to_string();
    }
    unsigned short GetPort()
    {//獲得監控的端口
        return _pAcceptor->local_endpoint().port();
    }
    static unsigned short GetLocalValidPort()
    {//隨機過去一個本機的可用端口
        io_service ios;
        tcp::acceptor acceptor(ios, tcp::endpoint(tcp::v4(), 0));
        tcp::acceptor::endpoint_type et = acceptor.local_endpoint();
        return et.port();
    }
    static string GetLocalIp()
    {//獲取本機ip
        boost::asio::io_service io_service;
        tcp::resolver resolver(io_service);
        tcp::resolver::query query(boost::asio::ip::host_name(), "");
        tcp::resolver::iterator iter = resolver.resolve(query);
        tcp::resolver::iterator end; // End marker.
        if (iter != end)
        {
            tcp::endpoint ep = *iter;
            return ep.address().to_string();
        }
    }
    ~MyNet(void)
    {
        delete _pTimer;
        delete _pAcceptor;
        delete _pClentSkt;
    }
protected:
    void AcceptHandle(const error_code& error, boost::function<void ()> fun)
    {//異步接收連接的回調函數
        if (error)
            return;
        //傳遞進來的函數指針fun未用到，本來是爲了啓動此函數的線程，boost::thread trd(fun);
    }

private:
    io_service _ios;
    tcp::acceptor* _pAcceptor;
    tcp::socket* _pClentSkt;
    tcp::socket* _pClentTmp;
    deadline_timer* _pTimer;
    char _buf[1024];
    size_t _ret;
    static string _ip;
    boost::mutex _mutex;
};