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
123
124
125
|
/***************************************************************************
* 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 <kdebug.h>
#include <ksocks.h>
#include "httprequest.h"
#include "../functions.h"
namespace bt
{
HTTPRequest::HTTPRequest(const TQString & hdr,const TQString & payload,const TQString & host,Uint16 port,bool verbose, bool fwd) : hdr(hdr),payload(payload),verbose(verbose),fwd(fwd)
{
KSocks::self()->disableSocks();
sock = new KNetwork::KStreamSocket(host,TQString::number(port),this,0);
sock->enableRead(true);
sock->enableWrite(true);
sock->setTimeout(30000);
sock->setBlocking(false);
connect(sock,TQ_SIGNAL(readyRead()),this,TQ_SLOT(onReadyRead()));
connect(sock,TQ_SIGNAL(gotError(int)),this,TQ_SLOT(onError(int )));
connect(sock,TQ_SIGNAL(timedOut()),this,TQ_SLOT(onTimeout()));
connect(sock,TQ_SIGNAL(connected(const KResolverEntry&)),
this, TQ_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)
{
KdDebug() << "Sending " << endl;
KdDebug() << 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)
{
KdDebug() << "Got reply : " << endl;
KdDebug() << strdata << endl;
}*/
if (sl.first().contains("HTTP") && sl.first().contains("200"))
{
// emit reply OK
replyOK(this,sl.last(),fwd);
}
else
{
// emit reply error
replyError(this,sl.last(),fwd);
}
operationFinished(this);
}
void HTTPRequest::onError(int)
{
kdDebug() << "HTTPRequest error : " << sock->errorString() << endl;
error(this,false);
sock->close();
operationFinished(this);
}
void HTTPRequest::onTimeout()
{
kdDebug() << "HTTPRequest timeout" << endl;
error(this,true);
sock->close();
operationFinished(this);
}
}
#include "httprequest.moc"
|
