application/hub/main/handler/message-types/dht/class_NodeMessageDhtBootstrapHandler.php

   1 <?php
   2 /**
   3  * A NodeMessageDhtBootstrap handler
   4  *
   5  * @author              Roland Haeder <webmaster@shipsimu.org>
   6  * @version             0.0.0
   7  * @copyright   Copyright (c) 2007, 2008 Roland Haeder, 2009 - 2015 Hub Developer Team
   8  * @license             GNU GPL 3.0 or any newer version
   9  * @link                http://www.shipsimu.org
  10  *
  11  * This program is free software: you can redistribute it and/or modify
  12  * it under the terms of the GNU General Public License as published by
  13  * the Free Software Foundation, either version 3 of the License, or
  14  * (at your option) any later version.
  15  *
  16  * This program is distributed in the hope that it will be useful,
  17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19  * GNU General Public License for more details.
  20  *
  21  * You should have received a copy of the GNU General Public License
  22  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  23  */
  24 class NodeMessageDhtBootstrapHandler extends BaseMessageHandler implements HandleableMessage, Registerable {
  25         /**
  26          * Protected constructor
  27          *
  28          * @return      void
  29          */
  30         protected function __construct () {
  31                 // Call parent constructor
  32                 parent::__construct(__CLASS__);
  33
  34                 // Set handler name
  35                 $this->setHandlerName('message_dht_bootstrap');
  36
  37                 // Init message data array
  38                 $this->messageDataElements = array(
  39                         XmlDhtBootstrapTemplateEngine::DHT_BOOTSTRAP_DATA_EXTERNAL_ADDRESS,
  40                         XmlDhtBootstrapTemplateEngine::DHT_BOOTSTRAP_DATA_INTERNAL_ADDRESS,
  41                         XmlDhtBootstrapTemplateEngine::DHT_BOOTSTRAP_DATA_NODE_STATUS,
  42                         XmlDhtBootstrapTemplateEngine::DHT_BOOTSTRAP_DATA_NODE_MODE,
  43                         XmlDhtBootstrapTemplateEngine::DHT_BOOTSTRAP_DATA_SESSION_ID,
  44                 );
  45
  46                 // Init message-data->configuration translation array
  47                 $this->messageToConfig = array(
  48                         XmlDhtBootstrapTemplateEngine::DHT_BOOTSTRAP_DATA_EXTERNAL_ADDRESS => 'your_external_address',
  49                         XmlDhtBootstrapTemplateEngine::DHT_BOOTSTRAP_DATA_INTERNAL_ADDRESS => 'your_internal_address',
  50                         XmlDhtBootstrapTemplateEngine::DHT_BOOTSTRAP_DATA_SESSION_ID       => 'your_session_id'
  51                 );
  52
  53                 // Init config-copy array
  54                 $this->configCopy = array(
  55                         XmlDhtBootstrapAnswerTemplateEngine::DHT_BOOTSTRAP_DATA_EXTERNAL_ADDRESS => 'external_address',
  56                         XmlDhtBootstrapAnswerTemplateEngine::DHT_BOOTSTRAP_DATA_INTERNAL_ADDRESS => 'internal_address',
  57                         XmlDhtBootstrapAnswerTemplateEngine::DHT_BOOTSTRAP_DATA_NODE_STATUS      => 'node_status',
  58                         XmlDhtBootstrapAnswerTemplateEngine::DHT_BOOTSTRAP_DATA_SESSION_ID       => 'session_id',
  59                 );
  60
  61                 // Init array
  62                 $this->searchData = array(
  63                         XmlDhtBootstrapTemplateEngine::DHT_BOOTSTRAP_DATA_EXTERNAL_ADDRESS,
  64                 );
  65         }
  66
  67         /**
  68          * Creates an instance of this class
  69          *
  70          * @return      $handlerInstance        An instance of a HandleableMessage class
  71          */
  72         public final static function createNodeMessageDhtBootstrapHandler () {
  73                 // Get new instance
  74                 $handlerInstance = new NodeMessageDhtBootstrapHandler();
  75
  76                 // Get node instance
  77                 $nodeInstance = NodeObjectFactory::createNodeInstance();
  78
  79                 // Debug message
  80                 self::createDebugInstance(__CLASS__)->debugOutput('POOL[' . __METHOD__ . ':' . __LINE__ . ':] Setting node ...');
  81
  82                 // ... and set it here
  83                 $handlerInstance->setNodeInstance($nodeInstance);
  84
  85                 // Get a DHT instance
  86                 $dhtInstance = DhtObjectFactory::createDhtInstance('node');
  87
  88                 // Set the DHT instance here
  89                 $handlerInstance->setDhtInstance($dhtInstance);
  90
  91                 // Return the prepared instance
  92                 return $handlerInstance;
  93         }
  94
  95         /**
  96          * Handles data array of the message
  97          *
  98          * @param       $messageData            An array with message data to handle
  99          * @param       $packageInstance        An instance of a Receivable class
 100          * @return      void
 101          * @throws      DhtBootstrapNotAcceptedException        If this node does not accept DHT bootstrap requests
 102          */
 103         public function handleMessageData (array $messageData, Receivable $packageInstance) {
 104                 // Is this node accepting DHT bootstrap requests?
 105                 if (!$this->getNodeInstance()->isAcceptingDhtBootstrap()) {
 106                         /*
 107                          * This node is not accepting DHT bootstrap requests.
 108                          */
 109                         throw new DhtBootstrapNotAcceptedException(array($this, $messageData), self::EXCEPTION_DHT_BOOTSTRAP_NOT_ACCEPTED);
 110                 } // END - if
 111
 112                 // Register the DHT bootstrap requesting node with this node
 113                 $this->registerNodeByMessageData($messageData);
 114
 115                 // Prepare answer message to be delivered back to the other node
 116                 $this->prepareAnswerMessage($messageData, $packageInstance);
 117         }
 118
 119         /**
 120          * Adds all required elements from given array into data set instance
 121          *
 122          * @param       $dataSetInstance        An instance of a StoreableCriteria class
 123          * @param       $messageData            An array with all message data
 124          * @return      void
 125          */
 126         public function addArrayToDataSet (StoreableCriteria $dataSetInstance, array $messageData) {
 127                 // Add generic first
 128                 parent::addArrayToDataSet($dataSetInstance, $messageData);
 129
 130                 // Add all ements
 131                 foreach ($this->messageDataElements as $key) {
 132                         // Is it there?
 133                         assert(isset($messageData[$key]));
 134
 135                         // Add it
 136                         $dataSetInstance->addCriteria($key, $messageData[$key]);
 137                 } // END - foreach
 138         }
 139
 140         /**
 141          * Initializes configuration data from given message data array
 142          *
 143          * @param       $messageData    An array with all message data
 144          * @return      void
 145          */
 146         protected function initMessageConfigurationData (array $messageData) {
 147                 // Debug message
 148                 //* NOISY-DEBUG: */ self::createDebugInstance(__CLASS__)->debugOutput('DHT-BOOTSTRAP-HANDLER[' . __METHOD__ . ':' . __LINE__ . ']: messageData=' . print_r($messageData, TRUE));
 149
 150                 // "Walk" throught the translation array
 151                 foreach ($this->messageToConfig as $messageKey => $configKey) {
 152                         // Debug message
 153                         //* NOISY-DEBUG: */ self::createDebugInstance(__CLASS__)->debugOutput('DHT-BOOTSTRAP-HANDLER[' . __METHOD__ . ':' . __LINE__ . ']: Setting messageKey=' . $messageKey . ',configKey=' . $configKey . ':' . $messageData[$messageKey]);
 154
 155                         // Set the element in configuration
 156                         $this->getConfigInstance()->setConfigEntry($configKey, $messageData[$messageKey]);
 157                 } // END - foreach
 158
 159                 // "Walk" throught the config-copy array
 160                 foreach ($this->configCopy as $targetKey => $sourceKey) {
 161                         // Debug message
 162                         //* NOISY-DEBUG: */ self::createDebugInstance(__CLASS__)->debugOutput('DHT-BOOTSTRAP-HANDLER[' . __METHOD__ . ':' . __LINE__ . ']: Copying from sourceKey=' . $sourceKey . ' to targetKey=' . $targetKey . '...');
 163
 164                         // Copy from source to target key
 165                         $this->getConfigInstance()->setConfigEntry($targetKey, $this->getConfigInstance()->getConfigEntry($sourceKey));
 166                 } // END - foreach
 167
 168                 // Translate last exception into a status code
 169                 $statusCode = $this->getTranslatedStatusFromLastException();
 170
 171                 // Set it in configuration (temporarily)
 172                 $this->getConfigInstance()->setConfigEntry(NodeDistributedHashTableDatabaseWrapper::DB_COLUMN_ANSWER_STATUS, $statusCode);
 173
 174                 /*
 175                  * Use the DHT instance to get a list of recipients. This means that all
 176                  * DHT nodes that accept bootstrap requests are read from the DHT
 177                  * database.
 178                  */
 179                 $nodeList = $this->getDhtInstance()->findRecipientsByKey(NodeDistributedHashTableDatabaseWrapper::DB_COLUMN_ACCEPT_BOOTSTRAP, 'Y');
 180
 181                 // Make sure it is an array and has at least one entry
 182                 assert(is_array($nodeList));
 183                 assert(count($nodeList) > 0);
 184
 185                 // Set it in configuration
 186                 $this->getConfigInstance()->setConfigEntry('dht_nodes', base64_encode(json_encode($nodeList)));
 187         }
 188
 189         /**
 190          * Removes configuration data with given message data array from global
 191          * configuration
 192          *
 193          * @param       $messageData    An array with all message data
 194          * @return      void
 195          */
 196         protected function removeMessageConfigurationData (array $messageData) {
 197                 // "Walk" throught the translation array again
 198                 foreach ($this->messageToConfig as $dummy => $configKey) {
 199                         // Now unset this configuration entry (to save some memory)
 200                         $this->getConfigInstance()->unsetConfigEntry($configKey);
 201                 } // END - foreach
 202
 203                 // "Walk" throught the config-copy array again
 204                 foreach ($this->configCopy as $configKey => $dummy) {
 205                         // Now unset this configuration entry (to save some memory again)
 206                         $this->getConfigInstance()->unsetConfigEntry($configKey);
 207                 } // END - foreach
 208
 209                 // Remove temporary "special" values as well
 210                 $this->getConfigInstance()->unsetConfigEntry(NodeDistributedHashTableDatabaseWrapper::DB_COLUMN_ANSWER_STATUS);
 211                 $this->getConfigInstance()->unsetConfigEntry('dht_nodes');
 212         }
 213 }
 214
 215 // [EOF]
 216 ?>