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
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
|
/***************************************************************************
* 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. *
***************************************************************************/
#ifndef KTUPNPROUTER_H
#define KTUPNPROUTER_H
#include <ntqtimer.h>
#include <kio/job.h>
#include <ntqvaluelist.h>
#include "exitoperation.h"
#include <kurl.h>
#include <ntqstringlist.h>
#include <kstreamsocket.h>
#include "portlist.h"
#include "forwardportlist.h"
using bt::Uint16;
namespace bt
{
/**
* @author Joris Guisson <joris.guisson@gmail.com>
*
* Job to wait for a certain amount of time or until one or more ExitOperation's have
* finished.
*/
class WaitJob : public KIO::Job
{
Q_OBJECT
public:
WaitJob(Uint32 millis);
virtual ~WaitJob();
virtual void kill(bool quietly=true);
/**
* Add an ExitOperation;
* @param op The operation
*/
void addExitOperation(kt::ExitOperation* op);
/**
* Execute a WaitJob
* @param job The Job
*/
static void execute(WaitJob* job);
/// Are there any ExitOperation's we need to wait for
bool needToWait() const {return exit_ops.count() > 0;}
private slots:
void timerDone();
void operationFinished(kt::ExitOperation* op);
private:
TQTimer timer;
TQValueList<kt::ExitOperation*> exit_ops;
};
void SynchronousWait(Uint32 millis);
}
namespace bt
{
class HTTPRequest;
class WaitJob;
}
namespace net
{
class ForwardPortList;
}
namespace KIO
{
class Job;
}
namespace kt
{
/**
* Structure describing a UPnP service found in an xml file.
*/
struct UPnPService
{
TQString serviceid;
TQString servicetype;
TQString controlurl;
TQString eventsuburl;
TQString scpdurl;
UPnPService();
UPnPService(const UPnPService & s);
/**
* Set a property of the service.
* @param name Name of the property (matches to variable names)
* @param value Value of the property
*/
void setProperty(const TQString & name,const TQString & value);
/**
* Set all strings to empty.
*/
void clear();
/// Print the data of this service
void debugPrintData();
/**
* Assignment operator
* @param s The service to copy
* @return *this
*/
UPnPService & operator = (const UPnPService & s);
};
/**
* Struct to hold the description of a device
*/
struct UPnPDeviceDescription
{
TQString friendlyName;
TQString manufacturer;
TQString modelDescription;
TQString modelName;
TQString modelNumber;
/**
* Set a property of the description
* @param name Name of the property (matches to variable names)
* @param value Value of the property
*/
void setProperty(const TQString & name,const TQString & value);
};
/**
* @author Joris Guisson
*
* Class representing a UPnP enabled router. This class is also used to communicate
* with the router.
*/
class UPnPRouter : public TQObject
{
Q_OBJECT
public:
struct Forwarding
{
net::Port extport;
net::Port intport;
bt::HTTPRequest* pending_req;
UPnPService* service;
};
struct ForwardingRequest
{
net::Port extport;
net::Port intport;
bt::HTTPRequest* pending_req;
};
private:
TQString server;
TQString tmp_file;
KURL location;
UPnPDeviceDescription desc;
TQValueList<UPnPService> services;
TQValueList<Forwarding> fwds;
TQValueList<ForwardingRequest> fwdreqs;
TQValueList<bt::HTTPRequest*> active_reqs;
net::ForwardPortList* forwardedPortList;
public:
/**
* Construct a router.
* @param server The name of the router
* @param location The location of it's xml description file
* @param verbose Print lots of debug info
*/
UPnPRouter(const TQString & server,const KURL & location,bool verbose = false);
virtual ~UPnPRouter();
/// Get the name of the server
TQString getServer() const {return server;}
/// Get the location of it's xml description
KURL getLocation() const {return location;}
/// Get the device description
UPnPDeviceDescription & getDescription() {return desc;}
/// Get the device description (const version)
const UPnPDeviceDescription & getDescription() const {return desc;}
/**
* Download the XML File of the router.
*/
void downloadXMLFile();
/**
* Add a service to the router.
* @param s The service
*/
void addService(const UPnPService & s);
#if 0
/**
* See if a port is forwarded
* @param port The Port
*/
void isPortForwarded(const net::Port & port);
/**
* Get the external IP address.
*/
void getExternalIP();
#endif
/**
* Forward a local port
* @param port The local port to forward
*/
void forward(const net::Port & externalport,
const net::Port & internalport = net::Port(),
bool force = false);
/**
* Undo forwarding
* @param port The port
* @param waitjob When this is set the jobs needs to be added to the waitjob,
* so we can wait for their completeion at exit
*/
void undoForward(const net::Port & externalport,const net::Port & internalport,
bt::WaitJob* waitjob = 0);
void debugPrintData();
TQValueList<ForwardingRequest>::iterator beginReqMappings() {return fwdreqs.begin();}
TQValueList<ForwardingRequest>::iterator endReqMappings() {return fwdreqs.end();}
TQValueList<Forwarding>::iterator beginPortMappings() {return fwds.begin();}
TQValueList<Forwarding>::iterator endPortMappings() {return fwds.end();}
net::ForwardPortList* forwardedPorts() {return forwardedPortList;}
private slots:
void onReplyOK(bt::HTTPRequest* r,const TQString &,bool);
void onReplyError(bt::HTTPRequest* r,const TQString &,bool);
void onError(bt::HTTPRequest* r,bool);
void downloadFinished(KIO::Job* j);
signals:
/**
* Tell the GUI that it needs to be updated.
*/
void updateGUI();
/**
* Signal which indicates that the XML was downloaded successfully or not.
* @param r The router which emitted the signal
* @param success Wether or not it succeeded
*/
void xmlFileDownloaded(UPnPRouter* r,bool success);
void replyOK(kt::UPnPRouter*,bt::HTTPRequest* ,const TQString &,bool);
void replyError(kt::UPnPRouter*,bt::HTTPRequest* ,const TQString &,bool);
private:
TQValueList<UPnPService>::iterator findPortForwardingService();
bt::HTTPRequest* sendSoapQuery(const TQString & query,const TQString & soapact,const TQString & controlurl,bool fwd, bool at_exit = false );
bool verbose;
void forward(UPnPService* srv,const net::Port & externalport,const net::Port & internalport = net::Port());
void undoForward(UPnPService* srv,const net::Port & externalport,const net::Port &
internalport,bt::WaitJob* waitjob);
void httpRequestDone(bt::HTTPRequest* r,bool erase_fwd);
};
}
#endif
|