src/r_command.cc

   1 ///
   2 /// \file       r_command.cc
   3 ///             Internal implementation of CommandTable parser class
   4 ///
   5
   6 /*
   7     Copyright (C) 2005-2013, 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 "i18n.h"
  23 #include "record.h"
  24 #include "record-internal.h"
  25 #include "data.h"
  26 #include <sstream>
  27 #include <iomanip>
  28
  29 using namespace std;
  30 using namespace Barry::Protocol;
  31
  32 namespace Barry {
  33
  34 ///////////////////////////////////////////////////////////////////////////////
  35 // CommandTable class
  36
  37 CommandTable::CommandTable()
  38 {
  39 }
  40
  41 CommandTable::~CommandTable()
  42 {
  43 }
  44
  45 const unsigned char* CommandTable::ParseField(const unsigned char *begin,
  46                                               const unsigned char *end)
  47 {
  48         // check if there is enough data for a header
  49         const unsigned char *headend = begin + sizeof(CommandTableField);
  50         if( headend > end )
  51                 return headend;
  52
  53         const CommandTableField *field = (const CommandTableField *) begin;
  54
  55         // advance and check size
  56         begin += COMMAND_FIELD_HEADER_SIZE + field->size;       // size is byte
  57         if( begin > end )               // if begin==end, we are ok
  58                 return begin;
  59
  60         if( !field->size )              // if field has no size, something's up
  61                 return begin;
  62
  63         Command command;
  64         command.Code = field->code;
  65         command.Name.assign((const char *)field->name, field->size);
  66         Commands.push_back(command);
  67         return begin;
  68 }
  69
  70 void CommandTable::Parse(const Data &data, size_t offset)
  71 {
  72         if( offset >= data.GetSize() )
  73                 return;
  74
  75         const unsigned char *begin = data.GetData() + offset;
  76         const unsigned char *end = data.GetData() + data.GetSize();
  77
  78         while( begin < end )
  79                 begin = ParseField(begin, end);
  80 }
  81
  82 void CommandTable::Clear()
  83 {
  84         Commands.clear();
  85 }
  86
  87 unsigned int CommandTable::GetCommand(const std::string &name) const
  88 {
  89         CommandArrayType::const_iterator b = Commands.begin();
  90         for( ; b != Commands.end(); b++ )
  91                 if( b->Name == name )
  92                         return b->Code;
  93         return 0;
  94 }
  95
  96 void CommandTable::Dump(std::ostream &os) const
  97 {
  98         CommandArrayType::const_iterator b = Commands.begin();
  99         os << _("Command table:\n");
 100         for( ; b != Commands.end(); b++ ) {
 101                 os << _("    Command: ") << "0x" << setbase(16) << b->Code
 102                    << " '" << b->Name << "'\n";
 103         }
 104 }
 105
 106
 107 } // namespace Barry
 108