blob: 9de125894dd6fa63c9e83c73a556d978ffcc41d1 (
plain)
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
|
/*
* Copyright (c) 2011 Tobias Markmann
* Licensed under the simplified BSD license.
* See Documentation/Licenses/BSD-simplified.txt for more information.
*/
#pragma once
#include <deque>
#include <boost/optional.hpp>
#include <boost/thread/thread.hpp>
#include <boost/thread/mutex.hpp>
#include <boost/thread/condition_variable.hpp>
#include <Swiften/Network/NATTraverser.h>
#include <Swiften/Network/HostAddressPort.h>
#include <Swiften/Network/PlatformNATTraversalRequest.h>
namespace Swift {
class EventLoop;
class NATTraversalGetPublicIPRequest;
class NATTraversalForwardPortRequest;
class NATTraversalRemovePortForwardingRequest;
class PlatformNATTraversalWorker : public NATTraverser {
private:
enum BackendType {
NotYetDecided,
UPnP,
NATPMP,
None,
};
public:
PlatformNATTraversalWorker(EventLoop* eventLoop);
~PlatformNATTraversalWorker();
boost::shared_ptr<NATTraversalGetPublicIPRequest> createGetPublicIPRequest();
boost::shared_ptr<NATTraversalForwardPortRequest> createForwardPortRequest(unsigned int localPort, unsigned int publicPort);
boost::shared_ptr<NATTraversalRemovePortForwardingRequest> createRemovePortForwardingRequest(unsigned int localPort, unsigned int publicPort);
void run();
void addRequestToQueue(PlatformNATTraversalRequest::ref);
private:
void checkAvailableNATTraversalProtocols();
void handleUPnPGetPublicIPResult(boost::optional<HostAddress> address);
void handleNATPMPGetPublicIPResult(boost::optional<HostAddress> address);
private:
BackendType backendType;
EventLoop* eventLoop;
bool stopRequested;
boost::thread* thread;
std::deque<PlatformNATTraversalRequest::ref> queue;
boost::mutex queueMutex;
boost::condition_variable queueNonEmpty;
};
}
|