src/utilities/utils.cpp

   1 /*
   2  * <one line to give the program's name and a brief idea of what it does.>
   3  * Copyright (C) 2013  Ta Thanh Dinh <email>
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation; either version 2 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License along
  16  * with this program; if not, write to the Free Software Foundation, Inc.,
  17  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  */
  20
  21 #include "utils.h"
  22 #include "../main.h"
  23 #include <sstream>
  24 #include <ctime>
  25 #include <boost/random.hpp>
  26
  27 namespace utilities
  28 {
  29
  30 // initialize a random generator and a uniform distribution
  31 boost::random::taus88 random_generator(static_cast<unsigned int>(std::time(0)));
  32 boost::random::uniform_int_distribution<UINT8> uint8_uniform;
  33
  34
  35 /**
  36  * @brief generate a random number of type UINT8
  37  *
  38  * @return UINT8
  39  */
  40 UINT8 utils::random_uint8()
  41 {
  42   return uint8_uniform(random_generator);
  43 }
  44
  45
  46 /**
  47  * @brief reformat the string showing some memory address.
  48  *
  49  * @param input input string
  50  * @return std::string
  51  */
  52 std::string utils::remove_leading_zeros(std::string input)
  53 {
  54   std::string::iterator str_iter = input.begin();
  55   std::string output("0x");
  56
  57   while ((str_iter != input.end()) && ((*str_iter == '0') || (*str_iter == 'x')))
  58   {
  59     ++str_iter;
  60   }
  61
  62   while (str_iter != input.end())
  63   {
  64     output.push_back(*str_iter);
  65     ++str_iter;
  66   }
  67
  68   return output;
  69 }
  70
  71
  72 /**
  73  * @brief convert an ADDRINT to a hexadecimal string
  74  *
  75  * @param input ...
  76  * @return std::string
  77  */
  78 std::string utils::addrint2hexstring(ADDRINT input)
  79 {
  80   std::stringstream num_stream;
  81   num_stream << "0x" << std::hex << input;
  82   return num_stream.str();
  83 }
  84
  85
  86
  87 /**
  88  * @brief determine if a memory address is located in the input message buffer.
  89  *
  90  * @param memory_address examined memory address
  91  * @return bool
  92  */
  93 bool utils::is_in_input_buffer(ADDRINT memory_address)
  94 {
  95   return ((received_message_address <= memory_address) &&
  96           (memory_address < received_message_address + received_message_length));
  97 }
  98
  99 } // end of utilities namespace