src/r_dbdb.cc

   1 ///
   2 /// \file       r_dbdb.cc
   3 ///             DatabaseDatabase record parser class
   4 ///
   5
   6 /*
   7     Copyright (C) 2005-2012, Net Direct Inc. (http://www.netdirect.ca/)
   8
   9     This program is free software; you can redistribute it and/or modify
  10     it under the terms of the GNU General Public License as published by
  11     the Free Software Foundation; either version 2 of the License, or
  12     (at your option) any later version.
  13
  14     This program is distributed in the hope that it will be useful,
  15     but WITHOUT ANY WARRANTY; without even the implied warranty of
  16     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  17
  18     See the GNU General Public License in the COPYING file at the
  19     root directory of this project for more details.
  20 */
  21
  22 #include "record.h"
  23 #include "record-internal.h"
  24 #include "data.h"
  25 #include "protocol.h"
  26 #include "debug.h"
  27 #include <algorithm>
  28
  29 using namespace std;
  30 using namespace Barry::Protocol;
  31
  32 namespace Barry {
  33
  34 ///////////////////////////////////////////////////////////////////////////////
  35 // DatabaseDatabase class
  36
  37 DatabaseDatabase::DatabaseDatabase()
  38 {
  39 }
  40
  41 DatabaseDatabase::~DatabaseDatabase()
  42 {
  43 }
  44
  45 template <class RecordType, class FieldType>
  46 void DatabaseDatabase::ParseRec(const RecordType &rec, const unsigned char *end)
  47 {
  48 }
  49
  50 template <class FieldType>
  51 const unsigned char* DatabaseDatabase::ParseField(const unsigned char *begin,
  52                                                   const unsigned char *end)
  53 {
  54         // check if there is enough data for a header
  55         const unsigned char *headend = begin + sizeof(FieldType);
  56         if( headend > end )
  57                 return headend;
  58
  59         // get our header
  60         const FieldType *field = (const FieldType *) begin;
  61
  62         // advance and check size
  63         begin += sizeof(FieldType) - sizeof(field->name) + ConvertHtoB(field->nameSize);
  64         if( begin > end )               // if begin==end, we are ok
  65                 return begin;
  66
  67         if( !ConvertHtoB(field->nameSize) ) // if field has no size, something's up
  68                 return begin;
  69
  70         Database db;
  71         db.Number = ConvertHtoB(field->dbNumber);
  72         db.RecordCount = ConvertHtoB(field->dbRecordCount);
  73         db.Name.assign((const char *)field->name, ConvertHtoB(field->nameSize) - 1);
  74         Databases.push_back(db);
  75         return begin;
  76 }
  77
  78 void DatabaseDatabase::Parse(const Data &data)
  79 {
  80         // check size to make sure we have up to the DBAccess operation byte
  81         if( data.GetSize() < (SB_PACKET_DBACCESS_HEADER_SIZE + 1) )
  82                 return;
  83
  84         MAKE_PACKET(pack, data);
  85         const unsigned char *begin = 0;
  86         const unsigned char *end = data.GetData() + data.GetSize();
  87
  88         switch( pack->u.db.u.response.operation )
  89         {
  90         case SB_DBOP_GET_DBDB:
  91                 // using the new protocol
  92                 if( data.GetSize() > SB_PACKET_DBDB_HEADER_SIZE ) {
  93                         begin = (const unsigned char *)
  94                                 &pack->u.db.u.response.u.dbdb.field[0];
  95
  96                         // this while check is ok, since ParseField checks
  97                         // for header size
  98                         while( begin < end )
  99                                 begin = ParseField<DBDBField>(begin, end);
 100                 }
 101                 else
 102                         dout("DatabaseDatabase: not enough data for parsing");
 103                 break;
 104
 105         case SB_DBOP_OLD_GET_DBDB:
 106                 // using the old protocol
 107                 if( data.GetSize() > SB_PACKET_OLD_DBDB_HEADER_SIZE ) {
 108                         begin = (const unsigned char *)
 109                                 &pack->u.db.u.response.u.old_dbdb.field[0];
 110
 111                         // this while check is ok, since ParseField checks
 112                         // for header size
 113                         while( begin < end )
 114                                 begin = ParseField<OldDBDBField>(begin, end);
 115                 }
 116                 else
 117                         dout("DatabaseDatabase: not enough data for parsing");
 118                 break;
 119
 120         default:
 121                 // unknown protocol
 122                 dout("Unknown protocol");
 123                 break;
 124         }
 125
 126
 127 }
 128
 129 void DatabaseDatabase::Clear()
 130 {
 131         Databases.clear();
 132 }
 133
 134 namespace {
 135         bool NameSort(const DatabaseDatabase::Database &one,
 136                 const DatabaseDatabase::Database &two)
 137         {
 138                 return one.Name < two.Name;
 139         }
 140
 141         bool CountSort(const DatabaseDatabase::Database &one,
 142                 const DatabaseDatabase::Database &two)
 143         {
 144                 return one.RecordCount < two.RecordCount;
 145         }
 146 }
 147
 148 void DatabaseDatabase::SortByName()
 149 {
 150         std::sort(Databases.begin(), Databases.end(), NameSort);
 151 }
 152
 153 void DatabaseDatabase::SortByRecordCount()
 154 {
 155         std::sort(Databases.begin(), Databases.end(), CountSort);
 156 }
 157
 158 unsigned int DatabaseDatabase::GetTotalRecordCount() const
 159 {
 160         unsigned int sum = 0;
 161
 162         DatabaseArrayType::const_iterator b = Databases.begin();
 163         for( ; b != Databases.end(); ++b ) {
 164                 sum += b->RecordCount;
 165         }
 166         return sum;
 167 }
 168
 169 bool DatabaseDatabase::GetDBNumber(const std::string &name,
 170                                    unsigned int &number) const
 171 {
 172         DatabaseArrayType::const_iterator b = Databases.begin();
 173         for( ; b != Databases.end(); ++b )
 174                 if( b->Name == name ) {
 175                         number = b->Number;
 176                         return true;
 177                 }
 178         return false;
 179 }
 180
 181 bool DatabaseDatabase::GetDBName(unsigned int number,
 182                                  std::string &name) const
 183 {
 184         DatabaseArrayType::const_iterator b = Databases.begin();
 185         for( ; b != Databases.end(); ++b )
 186                 if( b->Number == number ) {
 187                         name = b->Name;
 188                         return true;
 189                 }
 190         return false;
 191 }
 192
 193 void DatabaseDatabase::Dump(std::ostream &os) const
 194 {
 195         DatabaseArrayType::const_iterator b = Databases.begin();
 196         os << "Database database:\n";
 197         for( ; b != Databases.end(); b++ ) {
 198                 os << "    Database: 0x" << setbase(16) << b->Number
 199                    << " '" << b->Name << "' (records: "
 200                    << setbase(10) << b->RecordCount << ")\n";
 201         }
 202 }
 203
 204 } // namespace Barry
 205