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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
/***************************************************************************
* Copyright (C) 2005 by Joris Guisson *
* joris.guisson@gmail.com *
* *
* 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 2 of the License, or *
* (at your option) any later version. *
* *
* This program 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 program; if not, write to the *
* Free Software Foundation, Inc., *
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. *
***************************************************************************/
#include <tqstringlist.h>
#include <torrent/globals.h>
#include "httprequest.h"
#include "array.h"
#include "log.h"
namespace bt
{
HTTPRequest::HTTPRequest(const TQString & hdr,const TQString & payload,const TQString & host,Uint16 port,bool verbose) : hdr(hdr),payload(payload),verbose(verbose)
{
sock = new KNetwork::KStreamSocket(host,TQString::number(port),this,0);
sock->enableRead(true);
sock->enableWrite(true);
sock->setTimeout(30000);
sock->setBlocking(false);
connect(sock,TQT_SIGNAL(readyRead()),this,TQT_SLOT(onReadyRead()));
connect(sock,TQT_SIGNAL(gotError(int)),this,TQT_SLOT(onError(int )));
connect(sock,TQT_SIGNAL(timedOut()),this,TQT_SLOT(onTimeout()));
connect(sock,TQT_SIGNAL(connected(const KResolverEntry&)),
this, TQT_SLOT(onConnect( const KResolverEntry& )));
}
HTTPRequest::~HTTPRequest()
{
sock->close();
delete sock;
}
void HTTPRequest::start()
{
sock->connect();
}
void HTTPRequest::onConnect(const KResolverEntry&)
{
payload = payload.replace("$LOCAL_IP",sock->localAddress().nodeName());
hdr = hdr.replace("$CONTENT_LENGTH",TQString::number(payload.length()));
TQString req = hdr + payload;
if (verbose)
{
Out(SYS_PNP|LOG_DEBUG) << "Sending " << endl;
Out(SYS_PNP|LOG_DEBUG) << hdr << payload << endl;
}
sock->writeBlock(req.ascii(),req.length());
}
void HTTPRequest::onReadyRead()
{
Uint32 ba = sock->bytesAvailable();
if (ba == 0)
{
error(this,false);
sock->close();
return;
}
Array<char> data(ba);
ba = sock->readBlock(data,ba);
TQString strdata((const char*)data);
TQStringList sl = TQStringList::split("\r\n",strdata,false);
if (verbose)
{
Out(SYS_PNP|LOG_DEBUG) << "Got reply : " << endl;
Out(SYS_PNP|LOG_DEBUG) << strdata << endl;
}
if (sl.first().contains("HTTP") && sl.first().contains("200"))
{
// emit reply OK
replyOK(this,sl.last());
}
else
{
// emit reply error
replyError(this,sl.last());
}
operationFinished(this);
}
void HTTPRequest::onError(int)
{
Out() << "HTTPRequest error : " << sock->KSocketBase::errorString() << endl;
error(this,false);
sock->close();
operationFinished(this);
}
void HTTPRequest::onTimeout()
{
Out() << "HTTPRequest timeout" << endl;
error(this,true);
sock->close();
operationFinished(this);
}
}
#include "httprequest.moc"
|