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 a DHT instance
  77                 $dhtInstance = DhtObjectFactory::createDhtInstance('node');
  78
  79                 // Set the DHT instance here
  80                 $handlerInstance->setDhtInstance($dhtInstance);
  81
  82                 // Return the prepared instance
  83                 return $handlerInstance;
  84         }
  85
  86         /**
  87          * Handles data array of the message
  88          *
  89          * @param       $messageData            An array with message data to handle
  90          * @param       $packageInstance        An instance of a Receivable class
  91          * @return      void
  92          * @throws      DhtBootstrapNotAcceptedException        If this node does not accept DHT bootstrap requests
  93          */
  94         public function handleMessageData (array $messageData, Receivable $packageInstance) {
  95                 // Is this node accepting DHT bootstrap requests?
  96                 if (!NodeObjectFactory::createNodeInstance()->isAcceptingDhtBootstrap()) {
  97                         /*
  98                          * This node is not accepting DHT bootstrap requests.
  99                          */
 100                         throw new DhtBootstrapNotAcceptedException(array($this, $messageData), self::EXCEPTION_DHT_BOOTSTRAP_NOT_ACCEPTED);
 101                 } // END - if
 102
 103                 // Register the DHT bootstrap requesting node with this node
 104                 $this->registerNodeByMessageData($messageData);
 105
 106                 // Prepare answer message to be delivered back to the other node
 107                 $this->prepareAnswerMessage($messageData, $packageInstance);
 108         }
 109
 110         /**
 111          * Adds all required elements from given array into data set instance
 112          *
 113          * @param       $dataSetInstance        An instance of a StoreableCriteria class
 114          * @param       $messageData            An array with all message data
 115          * @return      void
 116          */
 117         public function addArrayToDataSet (StoreableCriteria $dataSetInstance, array $messageData) {
 118                 // Add generic first
 119                 parent::addArrayToDataSet($dataSetInstance, $messageData);
 120
 121                 // Add all ements
 122                 foreach ($this->messageDataElements as $key) {
 123                         // Is it there?
 124                         assert(isset($messageData[$key]));
 125
 126                         // Add it
 127                         $dataSetInstance->addCriteria($key, $messageData[$key]);
 128                 } // END - foreach
 129         }
 130
 131         /**
 132          * Initializes configuration data from given message data array
 133          *
 134          * @param       $messageData    An array with all message data
 135          * @return      void
 136          */
 137         protected function initMessageConfigurationData (array $messageData) {
 138                 // Debug message
 139                 //* NOISY-DEBUG: */ self::createDebugInstance(__CLASS__)->debugOutput('DHT-BOOTSTRAP-HANDLER[' . __METHOD__ . ':' . __LINE__ . ']: messageData=' . print_r($messageData, TRUE));
 140
 141                 // "Walk" throught the translation array
 142                 foreach ($this->messageToConfig as $messageKey => $configKey) {
 143                         // Debug message
 144                         //* NOISY-DEBUG: */ self::createDebugInstance(__CLASS__)->debugOutput('DHT-BOOTSTRAP-HANDLER[' . __METHOD__ . ':' . __LINE__ . ']: Setting messageKey=' . $messageKey . ',configKey=' . $configKey . ':' . $messageData[$messageKey]);
 145
 146                         // Set the element in configuration
 147                         $this->getConfigInstance()->setConfigEntry($configKey, $messageData[$messageKey]);
 148                 } // END - foreach
 149
 150                 // "Walk" throught the config-copy array
 151                 foreach ($this->configCopy as $targetKey => $sourceKey) {
 152                         // Debug message
 153                         //* NOISY-DEBUG: */ self::createDebugInstance(__CLASS__)->debugOutput('DHT-BOOTSTRAP-HANDLER[' . __METHOD__ . ':' . __LINE__ . ']: Copying from sourceKey=' . $sourceKey . ' to targetKey=' . $targetKey . '...');
 154
 155                         // Copy from source to target key
 156                         $this->getConfigInstance()->setConfigEntry($targetKey, $this->getConfigInstance()->getConfigEntry($sourceKey));
 157                 } // END - foreach
 158
 159                 // Translate last exception into a status code
 160                 $statusCode = $this->getTranslatedStatusFromLastException();
 161
 162                 // Set it in configuration (temporarily)
 163                 $this->getConfigInstance()->setConfigEntry(NodeDistributedHashTableDatabaseWrapper::DB_COLUMN_ANSWER_STATUS, $statusCode);
 164
 165                 /*
 166                  * Use the DHT instance to get a list of recipients. This means that all
 167                  * DHT nodes that accept bootstrap requests are read from the DHT
 168                  * database.
 169                  */
 170                 $nodeList = $this->getDhtInstance()->findRecipientsByKey(NodeDistributedHashTableDatabaseWrapper::DB_COLUMN_ACCEPT_BOOTSTRAP, 'Y');
 171
 172                 // Make sure it is an array and has at least one entry
 173                 assert(is_array($nodeList));
 174                 assert(count($nodeList) > 0);
 175
 176                 // Set it in configuration
 177                 $this->getConfigInstance()->setConfigEntry('dht_nodes', base64_encode(json_encode($nodeList)));
 178         }
 179
 180         /**
 181          * Removes configuration data with given message data array from global
 182          * configuration
 183          *
 184          * @param       $messageData    An array with all message data
 185          * @return      void
 186          */
 187         protected function removeMessageConfigurationData (array $messageData) {
 188                 // "Walk" throught the translation array again
 189                 foreach ($this->messageToConfig as $dummy => $configKey) {
 190                         // Now unset this configuration entry (to save some memory)
 191                         $this->getConfigInstance()->unsetConfigEntry($configKey);
 192                 } // END - foreach
 193
 194                 // "Walk" throught the config-copy array again
 195                 foreach ($this->configCopy as $configKey => $dummy) {
 196                         // Now unset this configuration entry (to save some memory again)
 197                         $this->getConfigInstance()->unsetConfigEntry($configKey);
 198                 } // END - foreach
 199
 200                 // Remove temporary "special" values as well
 201                 $this->getConfigInstance()->unsetConfigEntry(NodeDistributedHashTableDatabaseWrapper::DB_COLUMN_ANSWER_STATUS);
 202                 $this->getConfigInstance()->unsetConfigEntry('dht_nodes');
 203         }
 204 }
 205
 206 // [EOF]
 207 ?>