IEserver/session.cpp at master · qdore/IEserver · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
//
//  session.cpp
//  3
//
//  Created by adore on 14-4-7.
//  Copyright (c) 2014年 adore. All rights reserved.
//

#include "session.h"

Session::Session(io_service& io):
    sock(io),
    io(io)
{}
void Session::start()
{
    char *a = new char[1024];
    bzero(a, 1024);
    sock.async_read_some(buffer(a, 1024), [this, a](boost::system::error_code  er, int len)
    {
        if(!er)
        {
            do_read(a, len);
            start();
        }
        else
        {
            delete this;
        }
    });
}
void Session::do_read(char* a, int len)
{
    buf = buf + a;
    delete[] a;
    while(buf.find("\r\n\r\n") != std::string::npos)
    {
        std::string tmp = buf.substr(0, buf.find("\r\n\r\n") + 4);
        std::string tmp2 = tmp.substr(tmp.find("Host:") + 6, 30);
        auto i = tmp2.find("\r\n");
        tmp2 = tmp.substr(tmp.find("Host:") + 6, i);
        buf = buf.substr(buf.find("\r\n\r\n") + 4);
        sendto(tmp2, tmp);
    }
}
void Session::sendto(std::string acce, std::string content)
{
    try
    {
        ip::tcp::socket *socksend = new ip::tcp::socket(this->io);
        ip::tcp::resolver rlv(this->io);
        ip::tcp::resolver::query qry(acce, "80");
        ip::tcp::resolver::iterator end, begin = rlv.resolve(qry);
        boost::system::error_code ec = error::host_not_found;
        for(; ec && begin != end; begin++)
        {
            socksend->close();
            socksend->connect(*begin, ec);
        }
        if(!ec)
        {
            socksend->write_some(buffer(content));
            readback(socksend);
        }
    }
    catch(...) {}
}
void Session::readback(ip::tcp::socket* socksend)
{
    char *a = new char[1024];
    bzero(a, 1024);
    socksend->async_read_some(buffer(a, 1024), [this, socksend, a](boost::system::error_code er, int len)
    {
        if(!er)
        {
            try
            {
                sock.write_some(buffer(a, len));
                delete[] a;
                readback(socksend);
            }
            catch(...) {}
        }
        else
        {
            delete[] a;
            delete socksend;
        }
    });
}