1 files changed, 5 insertions, 0 deletions
diff --git a/server/tcpserver.h b/server/tcpserver.h
index 5d0cbba..47c5056 100644
--- a/server/tcpserver.h
+++ b/server/tcpserver.h
@@ -2,6 +2,7 @@
 #define TCPSERVER_H
 
 #include <boost/asio.hpp>
+#include <boost/function.hpp>
 
 #include "connection.h"
 
@@ -9,6 +10,8 @@ class TCPServer {
 	private:
 		boost::asio::ip::tcp::acceptor acceptor_;
 		
+		boost::function<void (Connection::p)> connect_callback;
+		
 		//! Listen for incoming connection.
 		void listen();
 		
@@ -17,6 +20,8 @@ class TCPServer {
 		
 	public:
 		TCPServer(boost::asio::io_service& io_service);
+		
+		void get_connection(boost::function<void (Connection::p)> f);
 };
 
 #endif