merged test_privacy updates from RC_0_16

2013-06-13 06:32:13 +00:00
parent 971b979dbc
commit 3823b74586
5 changed files with 215 additions and 5 deletions
--- a/test/peer_server.cpp
+++ b/test/peer_server.cpp
@@ -0,0 +1,147 @@
+/*
+
+Copyright (c) 2013, Arvid Norberg
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions
+are met:
+
+    * Redistributions of source code must retain the above copyright
+      notice, this list of conditions and the following disclaimer.
+    * Redistributions in binary form must reproduce the above copyright
+      notice, this list of conditions and the following disclaimer in
+      the documentation and/or other materials provided with the distribution.
+    * Neither the name of the author nor the names of its
+      contributors may be used to endorse or promote products derived
+      from this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
+LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+POSSIBILITY OF SUCH DAMAGE.
+
+*/
+
+#include "libtorrent/thread.hpp"
+#include "libtorrent/bencode.hpp"
+#include "libtorrent/entry.hpp"
+#include "libtorrent/address.hpp"
+#include "libtorrent/io_service.hpp"
+#include "libtorrent/error_code.hpp"
+#include "libtorrent/socket.hpp"
+
+#include <boost/detail/atomic_count.hpp>
+#include <boost/shared_ptr.hpp>
+#include <boost/bind.hpp>
+
+using namespace libtorrent;
+
+struct peer_server
+{
+
+	boost::asio::io_service m_ios;
+	boost::detail::atomic_count m_peer_requests;
+	tcp::acceptor m_acceptor;
+	int m_port;
+
+	boost::shared_ptr<libtorrent::thread> m_thread;
+
+	peer_server()
+		: m_peer_requests(0)
+		, m_acceptor(m_ios)
+	{
+		error_code ec;
+		m_acceptor.open(tcp::v4(), ec);
+		if (ec)
+		{
+			fprintf(stderr, "Error opening peer listen socket: %s\n", ec.message().c_str());
+			return;
+		}
+
+		m_acceptor.bind(tcp::endpoint(address_v4::any(), 0), ec);
+		if (ec)
+		{
+			fprintf(stderr, "Error binding peer socket to port 0: %s\n", ec.message().c_str());
+			return;
+		}
+		m_port = m_acceptor.local_endpoint(ec).port();
+		if (ec)
+		{
+			fprintf(stderr, "Error getting local endpoint of peer socket: %s\n", ec.message().c_str());
+			return;
+		}
+		m_acceptor.listen(10, ec);
+		if (ec)
+		{
+			fprintf(stderr, "Error listening on peer socket: %s\n", ec.message().c_str());
+			return;
+		}
+
+		fprintf(stderr, "peer initialized on port %d\n", m_port);
+
+		m_thread.reset(new thread(boost::bind(&peer_server::thread_fun, this)));
+	}
+
+	~peer_server()
+	{
+		m_acceptor.close();
+		if (m_thread) m_thread->join();
+	}
+
+	int port() const { return m_port; }
+
+	int num_hits() const { return m_peer_requests; }
+
+	void thread_fun()
+	{
+		for (;;)
+		{
+			error_code ec;
+			tcp::endpoint from;
+			tcp::socket socket(m_ios);
+			m_acceptor.accept(socket, from, ec);
+
+			if (ec == boost::asio::error::operation_aborted
+				|| ec == boost::asio::error::bad_descriptor) return;
+
+			if (ec)
+			{
+				fprintf(stderr, "Error accepting connection on peer socket: %s\n", ec.message().c_str());
+				return;
+			}
+
+			fprintf(stderr, "incoming peer connection\n");
+			++m_peer_requests;
+			socket.close(ec);
+		}
+	}
+};
+
+boost::shared_ptr<peer_server> g_peer;
+
+int start_peer()
+{
+	g_peer.reset(new peer_server);
+	return g_peer->port();
+}
+
+// the number of DHT messages received
+int num_peer_hits()
+{
+	if (g_peer) return g_peer->num_hits();
+	return 0;
+}
+
+void stop_peer()
+{
+	g_peer.reset();
+}
+