--- a/src/proto2/NetworkSession.hh	Thu Nov 20 19:25:56 2008 +0000
+++ b/src/proto2/NetworkSession.hh	Thu Nov 20 20:27:12 2008 +0000
@@ -1,6 +1,8 @@
 #ifndef NETWORK_SESSION_HH
 #define NETWORK_SESSION_HH
 
+#include <map>
+
 // forward-declare
 class NetworkSession;
 
@@ -13,9 +15,9 @@
 #include "NetworkUDP.hh"
 #include "NetworkNode.hh"
 
-#include <map>
+class NetworkSession {
+    friend class NetworkNode;
 
-class NetworkSession {
     private:
         uint64_t magic;
         NetworkTCPServer *tcp_srv;
@@ -34,10 +36,11 @@
     
     protected:
         virtual NetworkNode *build_node (NetworkTCPTransport *tcp, NetworkUDP *udp, const NetworkAddress &addr, enum NetworkNodeType type);
+        
+        void handle_disconnect (NetworkNode *node);
 
     private:
         void on_tcp_client (NetworkTCPTransport *client);
-        void handle_disconnect (NetworkNode *node);
 
          CL_Signal_v1<NetworkNode*> _sig_node_connected;