summaryrefslogtreecommitdiffstats
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;
};

}