release/src-rt-6.x.4708/router/samba-3.0.25b/examples/pcap2nbench/ip.cpp

   1 /*\
   2  *  pcap2nbench - Converts libpcap network traces to nbench input
   3  *  Copyright (C) 2004  Jim McDonough <jmcd@us.ibm.com>
   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
  16  *  along with this program; if not, write to the Free Software
  17  *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  18  *
  19  *  Written by Anthony Liguori <aliguori@us.ibm.com>
  20 \*/
  21
  22 #include <netinet/in.h>
  23 #include <sys/socket.h>
  24 #include <arpa/inet.h>
  25 #include "ip.hpp"
  26
  27 ip::ip(const uint8_t *data, size_t length)
  28 {
  29   if (length < 20) {
  30     std::cerr << "Invalid ip packet" << std::endl;
  31   }
  32
  33   version = (data[0] >> 4) & 0x0F;
  34   header_length = (data[0] & 0x0F) * 4;
  35   dsfield = data[1];
  36   memcpy(&total_length, data + 2, 2);
  37   total_length = ntohs(total_length);
  38   memcpy(&id, data + 4, 2);
  39   id = ntohs(id);
  40   flags = (data[6] >> 4) & 0x0F;
  41   fragment_offset = ((data[6] & 0x0F) << 4) | data[7];
  42   ttl = data[8];
  43   protocol = data[9];
  44   memcpy(&checksum, data + 10, 2);
  45   checksum = ntohs(checksum);
  46   memcpy(&source, data + 12, 4);
  47   memcpy(&destination, data + 16, 4);
  48 }
  49
  50 std::ostream &operator<<(std::ostream &lhs, const ip &rhs)
  51 {
  52   lhs << "Version: " << (uint32_t)rhs.version << std::endl
  53       << "Header length: " << (uint32_t)rhs.header_length << std::endl
  54       << "Differentiated Services Field: " << (uint32_t)rhs.dsfield << std::endl
  55       << "Total Length: " << (uint32_t)rhs.total_length << std::endl
  56       << "Identification: " << (uint32_t)rhs.id << std::endl
  57       << "Flags: " << (uint32_t)rhs.flags << std::endl
  58       << "Fragment offset: " << (uint32_t)rhs.fragment_offset << std::endl
  59       << "TTL: " << (uint32_t)rhs.ttl << std::endl
  60       << "Protocol: " << (uint32_t)rhs.protocol << std::endl
  61       << "Checksum: " << (uint32_t)rhs.checksum << std::endl
  62       << "Source: " << inet_ntoa(*((in_addr *)&rhs.source)) << std::endl
  63       << "Destination: " << inet_ntoa(*((in_addr *)&rhs.destination)) << std::endl;
  64
  65   return lhs;
  66 }