App/include/Tourist/App/PeerConnectionWorker.h

   1 #ifndef PEERCONNECTIONWORKER_H
   2 #define PEERCONNECTIONWORKER_H
   3
   4
   5 #include "Tourist/AbstractPipe.h"
   6 #include "Tourist/Message/MessageProcessor.h"
   7 #include "Tourist/Message/RemoteNode.h"
   8 #include "Tourist/Util/CallbackInterface.h"
   9 #include "Poco/Runnable.h"
  10 #include "Poco/Thread.h"
  11 #include "Poco/Logger.h"
  12 #include "Poco/Mutex.h"
  13 #include "Poco/NotificationQueue.h"
  14
  15 using Tourist::Message::MessageProcessor;
  16 using Tourist::Message::RemoteNode;
  17 using Tourist::AbstractPipe;
  18
  19 using Poco::Runnable;
  20 using Poco::Thread;
  21 using Poco::FastMutex;
  22 using Poco::NotificationQueue;
  23
  24 //forward declaration
  25 namespace Tourist {
  26 namespace Message {
  27 }
  28 }
  29
  30 namespace Tourist {
  31 namespace App {
  32
  33 class Application;
  34
  35 /**
  36  * Handle peer handshake in a new thread. Part of thread pool
  37  * in Application class.
  38 */
  39
  40 class PeerConnectionWorker : public Tourist::Util::CallbackInterface,
  41        public Runnable
  42 {
  43
  44 public:
  45   PeerConnectionWorker (Application *app, AbstractPipe *pipe, RemoteNode &response,
  46           bool isActive, int timeout=5);
  47
  48
  49   ~PeerConnectionWorker();
  50
  51   bool isStopped();
  52
  53   bool isStatusOK();
  54
  55   void stopWorker();
  56
  57   //called upon the recept of MsgPeerAnnouncement message.
  58   void handleHandshake();
  59
  60   //un-register pipe connect/disconnect handler
  61   void cleanUp();
  62
  63   Tourist::Message::RemoteNode* getResult();
  64
  65   void run();
  66
  67   void callback(void *);
  68
  69   //reference counting
  70   void duplicate();
  71
  72   void release();
  73
  74
  75 private:
  76
  77   void handleHandshake(AbstractMessage *message);
  78
  79   int sendResponse(bool isAck);
  80
  81   Tourist::Message::MessageProcessor messageProcessor;
  82
  83   Tourist::Message::MessageTypeData msgTypeData;
  84
  85   AbstractPipe *pipe;
  86
  87   //true value indicates that we are initiating peer handshake
  88   bool isActive;
  89
  90   bool shouldBeRunning;
  91
  92   bool stopped;
  93   //Set if we successfuly resolved the peer info
  94   bool statusOK;
  95
  96   //peer connection should be resolved before this timeout value.
  97   int timeout;
  98
  99   //Information of the connected node.
 100   Tourist::Message::RemoteNode *result;
 101
 102   Logger &logger;
 103
 104   static FastMutex mutex;
 105
 106   //The node that should be sent as a response in a handshake.
 107   RemoteNode response;
 108
 109   Application *parent;
 110
 111   NotificationQueue queue;
 112
 113   //for reference counting purpose.
 114   int rc;
 115
 116   static int ref_count;
 117
 118 }; //class PeerConnectionWorker
 119
 120 } //namespace Application
 121 } //namespace Tourist
 122
 123 #endif /*PEERCONNECTIONWORKER_H*/
 124