Planet
navi homePPSaboutscreenshotsdownloaddevelopmentforum

source: code/trunk/src/libraries/network/synchronisable/Synchronisable.cc @ 8425

Last change on this file since 8425 was 8329, checked in by scheusso, 14 years ago

preparing countermeasures against clients sending unwanted data

  • Property svn:eol-style set to native
File size: 15.8 KB
RevLine 
[1505]1/*
2 *   ORXONOX - the hottest 3D action shooter ever to exist
3 *                    > www.orxonox.net <
4 *
5 *
6 *   License notice:
7 *
8 *   This program is free software; you can redistribute it and/or
9 *   modify it under the terms of the GNU General Public License
10 *   as published by the Free Software Foundation; either version 2
11 *   of the License, or (at your option) any later version.
12 *
13 *   This program is distributed in the hope that it will be useful,
14 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
15 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16 *   GNU General Public License for more details.
17 *
18 *   You should have received a copy of the GNU General Public License
19 *   along with this program; if not, write to the Free Software
20 *   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
21 *
22 *   Author:
23 *      Dumeni Manatschal, (C) 2007
24 *      Oliver Scheuss, (C) 2007
25 *   Co-authors:
26 *      ...
27 *
28 */
29
30
31#include "Synchronisable.h"
32
[3214]33#include <cstdlib>
[1505]34#include "core/CoreIncludes.h"
[5742]35#include "core/GameMode.h"
[1735]36#include "core/BaseObject.h"
[3214]37#include "network/Host.h"
[1505]38
[2171]39namespace orxonox
[1505]40{
[1639]41
[2309]42  std::map<uint32_t, Synchronisable *> Synchronisable::objectMap_;
43  std::queue<uint32_t> Synchronisable::deletedObjects_;
[1639]44
[2171]45  uint8_t Synchronisable::state_=0x1; // detemines wheter we are server (default) or client
[1639]46
[1505]47  /**
48  * Constructor:
[5929]49  * Initializes all Variables and sets the right objectID_
[1505]50  */
[7163]51  Synchronisable::Synchronisable(BaseObject* creator )
52  {
[1505]53    RegisterRootObject(Synchronisable);
[1907]54    static uint32_t idCounter=0;
55    objectMode_=0x1; // by default do not send data to server
[7163]56    if ( GameMode::isMaster()/* || ( Host::running() && Host::isServer() )*/ )
[2171]57    {
[5929]58      this->setObjectID( idCounter++ );
[2171]59    }
60    else
[5743]61    {
[5929]62      objectID_=OBJECTID_UNKNOWN;
[5743]63    }
[5929]64    classID_ = static_cast<uint32_t>(-1);
[6417]65
[3084]66    // set dataSize to 0
67    this->dataSize_ = 0;
[2415]68    // set standard priority
[3280]69    this->setPriority( Priority::Normal );
[2171]70
[2415]71    // get creator id
[5929]72    if( creator )
73      this->creatorID_ = creator->getSceneID();
74    else
75      this->creatorID_ = OBJECTID_UNKNOWN;
[1505]76  }
77
[1907]78  /**
[2087]79   * Destructor:
[5929]80   * Delete all callback objects and remove objectID_ from the objectMap_
[1907]81   */
[7163]82  Synchronisable::~Synchronisable()
83  {
[1534]84    // delete callback function objects
[2171]85    if(!Identifier::isCreatingHierarchy()){
[3198]86      // remove object from the static objectMap
87      if (this->objectMode_ != 0x0 && (Host::running() && Host::isServer()))
[5929]88        deletedObjects_.push(objectID_);
[1907]89    }
[7163]90    // delete all Synchronisable Variables from syncList_ ( which are also in stringList_ )
91    for(std::vector<SynchronisableVariableBase*>::iterator it = syncList_.begin(); it!=syncList_.end(); it++)
[3304]92      delete (*it);
[7163]93    syncList_.clear();
94    stringList_.clear();
[7183]95    std::map<uint32_t, Synchronisable*>::iterator it2;
96    it2 = objectMap_.find(objectID_);
97    if (it2 != objectMap_.end())
98      objectMap_.erase(it2);
[2485]99
[1505]100  }
[1639]101
[2087]102
[1907]103  /**
104   * This function sets the internal mode for synchronisation
105   * @param b true if this object is located on a client or on a server
106   */
[7163]107  void Synchronisable::setClient(bool b)
108  {
[1505]109    if(b) // client
110      state_=0x2;
111    else  // server
112      state_=0x1;
113  }
[1856]114
[1907]115  /**
116   * This function fabricated a new synchrnisable (and children of it), sets calls updateData and create
117   * After calling this function the mem pointer will be increased by the size of the needed data
118   * @param mem pointer to where the appropriate data is located
119   * @param mode defines the mode, how the data should be loaded
120   * @return pointer to the newly created synchronisable
121   */
[2171]122  Synchronisable *Synchronisable::fabricate(uint8_t*& mem, uint8_t mode)
[1735]123  {
[2662]124    SynchronisableHeader header(mem);
[7801]125    if( header.isDiffed() )
126    {
127      mem += header.getDataSize() + header.getSize();
128      return 0;
129    }
130//     assert( !header.isDiffed() );
[1856]131
[2662]132    COUT(4) << "fabricating object with id: " << header.getObjectID() << std::endl;
[1856]133
[2662]134    Identifier* id = ClassByID(header.getClassID());
[2485]135    if (!id)
136    {
[3088]137        for(int i = 0; i<160; i++)
[2759]138            COUT(0) << "classid: " << i << " identifier: " << ClassByID(i) << endl;
[2485]139        COUT(0) << "Assertion failed: id" << std::endl;
140        COUT(0) << "Possible reason for this error: Client received a synchronizable object whose class has no factory." << std::endl;
141        abort();
142    }
[1907]143    assert(id);
[2171]144    BaseObject* creator = 0;
[2662]145    if (header.getCreatorID() != OBJECTID_UNKNOWN)
[2087]146    {
[2662]147      Synchronisable* synchronisable_creator = Synchronisable::getSynchronisable(header.getCreatorID());
[2087]148      if (!synchronisable_creator)
149      {
[7163]150        mem += header.getDataSize()+SynchronisableHeader::getSize(); //.TODO: this suckz.... remove size from header
[2662]151        assert(0); // TODO: uncomment this if we have a clean objecthierarchy (with destruction of children of objects) ^^
[2087]152        return 0;
153      }
154      else
[3325]155        creator = orxonox_cast<BaseObject*>(synchronisable_creator);
[2087]156    }
[2662]157    assert(getSynchronisable(header.getObjectID())==0);   //make sure no object with this id exists
[2171]158    BaseObject *bo = id->fabricate(creator);
[2087]159    assert(bo);
[3325]160    Synchronisable *no = orxonox_cast<Synchronisable*>(bo);
[1735]161    assert(no);
[5929]162    assert( Synchronisable::objectMap_.find(header.getObjectID()) == Synchronisable::objectMap_.end() );
163    no->setObjectID(header.getObjectID());
164    //no->creatorID=header.getCreatorID(); //TODO: remove this
165    no->setClassID(header.getClassID());
166    assert(no->creatorID_ == header.getCreatorID());
[7163]167    if( creator )
168      bo->setLevel(creator->getLevel());          // Note: this ensures that the level is known on the client for child objects of the scene (and the scene itself)
[5929]169    //assert(no->classID_ == header.getClassID());
170    COUT(4) << "fabricate objectID_: " << no->objectID_ << " classID_: " << no->classID_ << std::endl;
[1735]171          // update data and create object/entity...
[2087]172    bool b = no->updateData(mem, mode, true);
[1907]173    assert(b);
[2087]174    if (b)
175    {
[2245]176//        b = no->create();
[2087]177        assert(b);
178    }
[1907]179    return no;
180  }
181
[2087]182
[1907]183  /**
[5929]184   * Finds and deletes the Synchronisable with the appropriate objectID_
185   * @param objectID_ objectID_ of the Synchronisable
[1907]186   * @return true/false
187   */
[7163]188  bool Synchronisable::deleteObject(uint32_t objectID_)
189  {
[5929]190    if(!getSynchronisable(objectID_))
[1735]191      return false;
[5929]192    assert(getSynchronisable(objectID_)->objectID_==objectID_);
193    Synchronisable *s = getSynchronisable(objectID_);
[1907]194    if(s)
[5929]195      s->destroy(); // or delete?
[1907]196    else
[1735]197      return false;
198    return true;
199  }
[2087]200
[1907]201  /**
[5929]202   * This function looks up the objectID_ in the objectMap_ and returns a pointer to the right Synchronisable
203   * @param objectID_ objectID_ of the Synchronisable
204   * @return pointer to the Synchronisable with the objectID_
[1907]205   */
[7163]206  Synchronisable* Synchronisable::getSynchronisable(uint32_t objectID_)
207  {
[2309]208    std::map<uint32_t, Synchronisable*>::iterator it1;
[5929]209    it1 = objectMap_.find(objectID_);
[2171]210    if (it1 != objectMap_.end())
211      return it1->second;
[3198]212    // if the objects not in the map it should'nt exist at all anymore
[1907]213    return NULL;
214  }
215
[2087]216
[1505]217  /**
[5929]218   * This function takes all SynchronisableVariables out of the Synchronisable and saves them together with the size, objectID_ and classID_ to the given memory
[1735]219   * takes a pointer to already allocated memory (must have at least getSize bytes length)
[1751]220   * structure of the bitstream:
[5929]221   * |totalsize,objectID_,classID_,var1,var2,string1_length,string1,var3,...|
[1907]222   * length of varx: size saved int syncvarlist
223   * @param mem pointer to allocated memory with enough size
[7801]224   * @param sizes vector containing sizes of all objects in gamestate (to be appended)
[1907]225   * @param id gamestateid of the gamestate to be saved (important for priorities)
226   * @param mode defines the direction in which the data will be send/received
227   *             0x1: server->client
[7801]228   *             0x2: client->server
[1907]229   *             0x3: bidirectional
[2087]230   * @return true: if !doSync or if everything was successfully saved
[1735]231   */
[7163]232  uint32_t Synchronisable::getData(uint8_t*& mem, std::vector<uint32_t>& sizes, int32_t id, uint8_t mode)
233  {
234    unsigned int test = 0;
[2171]235    if(mode==0x0)
236      mode=state_;
[1907]237    //if this tick is we dont synchronise, then abort now
[8329]238    if(!doSync(/*id,*/ mode))
[3084]239      return 0;
[2309]240    uint32_t tempsize = 0;
[3304]241#ifndef NDEBUG
[7163]242    uint8_t* oldmem = mem;
[5929]243    if (this->classID_==0)
[1735]244      COUT(3) << "classid 0 " << this->getIdentifier()->getName() << std::endl;
[3304]245#endif
[2087]246
[5929]247    if (this->classID_ == static_cast<uint32_t>(-1))
248        this->classID_ = this->getIdentifier()->getNetworkID();
[2087]249
[5929]250    assert(ClassByID(this->classID_));
251    assert(this->classID_==this->getIdentifier()->getNetworkID());
252    assert(this->objectID_!=OBJECTID_UNKNOWN);
[3084]253    std::vector<SynchronisableVariableBase*>::iterator i;
[1856]254
[1735]255    // start copy header
[2662]256    SynchronisableHeader header(mem);
257    mem += SynchronisableHeader::getSize();
[1735]258    // end copy header
[1856]259
[7163]260    CCOUT(5) << "getting data from objectID_: " << objectID_ << ", classID_: " << classID_ << std::endl;
261//     COUT(4) << "objectid: " << this->objectID_ << ":";
[1735]262    // copy to location
[7163]263    for(i=syncList_.begin(); i!=syncList_.end(); ++i)
264    {
265      uint32_t varsize = (*i)->getData( mem, mode );
266//       COUT(4) << " " << varsize;
267      tempsize += varsize;
268      sizes.push_back(varsize);
269      ++test;
[3084]270      //tempsize += (*i)->getSize( mode );
[1735]271    }
[7801]272    assert(tempsize!=0);  // if this happens an empty object (with no variables) would be transmitted
[7163]273//     COUT(4) << endl;
[6417]274
[5929]275    header.setObjectID( this->objectID_ );
276    header.setCreatorID( this->creatorID_ );
277    header.setClassID( this->classID_ );
[3084]278    header.setDataSize( tempsize );
[7163]279    assert( tempsize == mem-oldmem-SynchronisableHeader::getSize() );
280    assert( test == this->getNrOfVariables() );
281    header.setDiffed(false);
282    tempsize += SynchronisableHeader::getSize();
[6417]283
[3084]284#ifndef NDEBUG
285    uint32_t size;
286    size=getSize(id, mode);
[1735]287    assert(tempsize==size);
[3084]288#endif
289    return tempsize;
[1735]290  }
[1505]291
[1856]292
[1505]293  /**
[1907]294   * This function takes a bytestream and loads the data into the registered variables
295   * @param mem pointer to the bytestream
296   * @param mode same as in getData
[8327]297   * @param forceCallback this makes updateData call each callback
[1735]298   * @return true/false
299   */
[7163]300  bool Synchronisable::updateData(uint8_t*& mem, uint8_t mode, bool forceCallback)
301  {
[1735]302    if(mode==0x0)
303      mode=state_;
[8329]304   
[7163]305    if(syncList_.empty())
306    {
[2419]307      assert(0);
[7163]308      COUT(2) << "Synchronisable::updateData syncList_ is empty" << std::endl;
[1735]309      return false;
310    }
[1856]311
[2245]312    uint8_t* data=mem;
[1735]313    // start extract header
[7163]314    SynchronisableHeaderLight syncHeaderLight(mem);
315    assert(syncHeaderLight.getObjectID()==this->getObjectID());
[8329]316   
317    if( !this->doReceive(mode) )
318    {
319      uint32_t headerSize;
320      if( syncHeaderLight.isDiffed() )
321        headerSize = SynchronisableHeaderLight::getSize();
322      else
323        headerSize = SynchronisableHeader::getSize();
324      mem += syncHeaderLight.getDataSize() + headerSize;
325      return true;
326    }
[1856]327
[5929]328    //COUT(5) << "Synchronisable: objectID_ " << syncHeader.getObjectID() << ", classID_ " << syncHeader.getClassID() << " size: " << syncHeader.getDataSize() << " synchronising data" << std::endl;
[7163]329    if( !syncHeaderLight.isDiffed() )
[2245]330    {
[7163]331      SynchronisableHeader syncHeader2(mem);
332      assert( this->getClassID() == syncHeader2.getClassID() );
333      assert( this->getCreatorID() == syncHeader2.getCreatorID() );
334      mem += SynchronisableHeader::getSize();
335      std::vector<SynchronisableVariableBase *>::iterator i;
[7801]336      for(i=syncList_.begin(); i!=syncList_.end(); ++i)
[7163]337      {
338        assert( mem <= data+syncHeader2.getDataSize()+SynchronisableHeader::getSize() ); // always make sure we don't exceed the datasize in our stream
339        (*i)->putData( mem, mode, forceCallback );
340      }
341      assert(mem == data+syncHeaderLight.getDataSize()+SynchronisableHeader::getSize() );
[1735]342    }
[7163]343    else
344    {
345      mem += SynchronisableHeaderLight::getSize();
346//       COUT(0) << "objectID: " << this->objectID_ << endl;
347      while( mem < data+syncHeaderLight.getDataSize()+SynchronisableHeaderLight::getSize() )
348      {
349        VariableID varID = *(VariableID*)mem;
350//         COUT(0) << "varID: " << varID << endl;
351        assert( varID < syncList_.size() );
352        mem += sizeof(VariableID);
353        syncList_[varID]->putData( mem, mode, forceCallback );
354      }
355      assert(mem == data+syncHeaderLight.getDataSize()+SynchronisableHeaderLight::getSize() );
356    }
[1735]357    return true;
358  }
[1505]359
360  /**
361  * This function returns the total amount of bytes needed by getData to save the whole content of the variables
[1907]362  * @param id id of the gamestate
363  * @param mode same as getData
[1505]364  * @return amount of bytes
365  */
[7163]366  uint32_t Synchronisable::getSize(int32_t id, uint8_t mode)
367  {
368    uint32_t tsize=SynchronisableHeader::getSize();
[3084]369    if (mode==0x0)
[1505]370      mode=state_;
[8329]371    if (!doSync(/*id, */mode))
[2171]372      return 0;
[3084]373    assert( mode==state_ );
374    tsize += this->dataSize_;
375    std::vector<SynchronisableVariableBase*>::iterator i;
[7163]376    for(i=stringList_.begin(); i!=stringList_.end(); ++i)
377    {
[2245]378      tsize += (*i)->getSize( mode );
[1505]379    }
380    return tsize;
381  }
[1639]382
[1735]383  /**
[1907]384   * This function determines, wheter the object should be saved to the bytestream (according to its syncmode/direction)
385   * @param id gamestate id
[8327]386   * @param mode Synchronisation mode (toclient, toserver or bidirectional)
[1907]387   * @return true/false
[1735]388   */
[8329]389  bool Synchronisable::doSync(/*int32_t id,*/ uint8_t mode)
[7163]390  {
[8327]391//     if(mode==0x0)
392//       mode=state_;
393    assert(mode!=0x0);
[7163]394    return ( (this->objectMode_ & mode)!=0 && (!syncList_.empty() ) );
[1735]395  }
[8329]396 
397  /**
398   * This function determines, wheter the object should accept data from the bytestream (according to its syncmode/direction)
399   * @param id gamestate id
400   * @param mode Synchronisation mode (toclient, toserver or bidirectional)
401   * @return true/false
402   */
403  bool Synchronisable::doReceive( uint8_t mode)
404  {
405    //return mode != this->objectMode_ || this->objectMode_ == ObjectDirection::Bidirectional;
406    return true;
407  }
[1856]408
[1907]409  /**
410   * This function sets the synchronisation mode of the object
[2171]411   * If set to 0x0 variables will not be synchronised at all
[1907]412   * If set to 0x1 variables will only be synchronised to the client
413   * If set to 0x2 variables will only be synchronised to the server
414   * If set to 0x3 variables will be synchronised bidirectionally (only if set so in registerVar)
415   * @param mode same as in registerVar
416   */
[7163]417  void Synchronisable::setSyncMode(uint8_t mode)
418  {
[2087]419    assert(mode==0x0 || mode==0x1 || mode==0x2 || mode==0x3);
[5929]420    this->objectMode_=mode;
[1505]421  }
422
[6417]423  template <> void Synchronisable::registerVariable( std::string& variable, uint8_t mode, NetworkCallbackBase *cb, bool bidirectional)
424  {
425    SynchronisableVariableBase* sv;
426    if (bidirectional)
427      sv = new SynchronisableVariableBidirectional<std::string>(variable, mode, cb);
428    else
429      sv = new SynchronisableVariable<std::string>(variable, mode, cb);
[7163]430    syncList_.push_back(sv);
431    stringList_.push_back(sv);
[6417]432  }
[2485]433
[7163]434template <> void Synchronisable::unregisterVariable( std::string& variable )
435  {
436    bool unregistered_nonexistent_variable = true;
437    std::vector<SynchronisableVariableBase*>::iterator it = syncList_.begin();
438    while(it!=syncList_.end())
439    {
440      if( ((*it)->getReference()) == &variable )
441      {
442        delete (*it);
443        syncList_.erase(it);
444        unregistered_nonexistent_variable = false;
445        break;
446      }
447      else
448        ++it;
449    }
450    assert(unregistered_nonexistent_variable == false);
451   
452    it = stringList_.begin();
453    while(it!=stringList_.end())
454    {
455      if( ((*it)->getReference()) == &variable )
456      {
457        delete (*it);
458        stringList_.erase(it);
459        return;
460      }
461      else
462        ++it;
463    }
464    unregistered_nonexistent_variable = true;
465    assert(unregistered_nonexistent_variable == false); //if we reach this point something went wrong:
466    // the variable has not been registered before
467  }
[6417]468
[7163]469
[1505]470}
Note: See TracBrowser for help on using the repository browser.