Qt_25 - 8BitsCoding/RobotMentor GitHub Wiki

#include "asioclient.h"
#include <qdebug.h>

AsioClient::AsioClient() :
    work(new boost::asio::io_service::work(ioservice)),
    resolver(ioservice),
    socket(ioservice)
{
    worker = std::thread([&](){
        ioservice.run();
    });

}

AsioClient::~AsioClient(){
    ioservice.stop();
    worker.join();
    work.reset();
}

void AsioClient::Get(const QString& url, const QString& p)
{
    // 1. url -> ep resolve
    server = url.toStdString();
    path = p.toStdString();

    boost::asio::ip::tcp::resolver::query query(server, "http");
    resolver.async_resolve(query, boost::bind(&AsioClient::handle_resolve,
                                              this,
                                              boost::asio::placeholders::error,
                                              boost::asio::placeholders::iterator));

    // 2. connect

    // 3. send

    // 4. recv
}


void AsioClient::handle_resolve(const boost::system::error_code& err, boost::asio::ip::tcp::resolver::iterator endpoint_iterator)
{
    if(!err) {
        boost::asio::ip::tcp::endpoint ep = *endpoint_iterator;
        qDebug() << ep.address().to_string().c_str();

        socket.async_connect(*endpoint_iterator,
                             boost::bind(&AsioClient::handle_connect,
                                         this,
                                         boost::asio::placeholders::error,
                                         endpoint_iterator));
    }
}

void AsioClient::handle_connect(const boost::system::error_code& err,
                                boost::asio::ip::tcp::resolver::iterator endpoint_iterator)
{
    if(!err) {
        // success
        // write
        std::ostream os(&requestbuf);

        os << "GET" << path << " HTTP/1.0\r\n";
        os << "Host: " << server << "\r\n";
        os << "Accept: */*\r\n";
        os << "Connection: close\r\n\r\n";

        boost::asio::async_write(socket, requestbuf,
                                 boost::bind(&AsioClient::handle_write,
                                             this,
                                             boost::asio::placeholders::error));
    } else {
        qDebug() << err.message().c_str() << endl;
    }
}

void AsioClient::handle_write(const boost::system::error_code& err)
{
    // recv
    if(!err) {
        boost::asio::async_read_until(socket, responsebuf, "\r\n",
                                      boost::bind(&AsioClient::handle_read_line,
                                                  this,
                                                  boost::asio::placeholders::error));
    } else {
        qDebug() << "handle_Write error : " << err.message().c_str() << endl;
    }
}
void AsioClient::handle_read_line(const boost::system::error_code& err)
{
    if(!err) {
        std::istream is(&responsebuf);
        std::string http_version;
        unsigned int status_code;
        std::string message;

        is >> http_version;
        is >> status_code;
        std::getline(is, message);

        // Error Check
        // 1. http_version HTTP/
        // 2. status_code != 200

        if(status_code != 200) {
            return ;
        }

        // 개행문자(\r\n\r\n) 두번 나오는 부분부터 메시지의 본문이다.
        // 그 이전 메시지는 모두 헤더이다.
        boost::asio::async_read_until(socket, responsebuf, "\r\n\r\n",
                                      boost::bind(&AsioClient::handle_read_header,
                                                  this,
                                                  boost::asio::placeholders::error));

    } else {

    }
}

void AsioClient::handle_read_header(const boost::system::error_code& err)
{
    std::istream istrm(&responsebuf);

    std::string header;
    while(std::getline(istrm, header) && header != "\r") {
        // header
        oss << header << std::endl;
    }

    boost::asio::streambuf::const_buffers_type bufs = responsebuf.data();
    std::string str(boost::asio::buffers_begin(bufs),
                    boost::asio::buffers_begin(bufs) + responsebuf.size());

    oss << str ;

    boost::asio::async_read(socket, responsebuf,
                            boost::asio::transfer_at_least(1),
                            boost::bind(&AsioClient::handle_read_content,
                                        this,
                                        boost::asio::placeholders::error));
}

void AsioClient::handle_read_content(const boost::system::error_code& err)
{
    if(!err) {
        oss << &responsebuf;

        boost::asio::async_read(socket, responsebuf,
                                boost::asio::transfer_at_least(1),
                                boost::bind(&AsioClient::handle_read_content,
                                            this,
                                            boost::asio::placeholders::error));

    } else if(err == boost::asio::error::eof) {
        // file end
        qDebug() << oss.str().c_str() << endl;
    } else {
        // error
    }
}

#ifndef ASIOCLIENT_H
#define ASIOCLIENT_H

#include <boost\asio.hpp>
#include <boost/bind.hpp>

#include <QObject>
#include <QString>
#include <memory>
#include <thread>

class AsioClient
{
public:
    AsioClient();
    virtual ~AsioClient();

    void Get(const QString& url, const QString& path);
    void handle_resolve(const boost::system::error_code& err,
                        boost::asio::ip::tcp::resolver::iterator endpoint_iterator);
    void handle_connect(const boost::system::error_code& err,
                        boost::asio::ip::tcp::resolver::iterator endpoint_iterator);
    void handle_write(const boost::system::error_code& err);
    void handle_read_line(const boost::system::error_code& err);
    void handle_read_header(const boost::system::error_code& err);
    void handle_read_content(const boost::system::error_code& err);

private:
    boost::asio::io_service ioservice;
    std::shared_ptr<boost::asio::io_service::work> work;
    std::thread worker;

    std::string server, path;
    boost::asio::ip::tcp::resolver resolver;

    boost::asio::ip::tcp::socket socket;

    boost::asio::streambuf requestbuf;
    boost::asio::streambuf responsebuf;

    std::ostringstream oss;
};

#endif // ASIOCLIENT_H