.#iolog.cc.1.3
上传用户:rrhhcc
上传日期:2015-12-11
资源大小:54129k
文件大小:5k
源码类别:

通讯编程

开发平台:

Visual C++

  1. //
  2. // iolog.cc      : IO Log Layer
  3. // Authors       : Fabio Silva and Yutaka Mori
  4. //
  5. // Copyright (C) 2000-2002 by the University of Southern California
  6. // $Id: iolog.cc,v 1.3 2005/09/13 04:53:47 tomh Exp $
  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. // version 2, as published by the Free Software Foundation.
  11. //
  12. // This program is distributed in the hope that it will be useful,
  13. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15. // GNU General Public License for more details.
  16. //
  17. // You should have received a copy of the GNU General Public License along
  18. // with this program; if not, write to the Free Software Foundation, Inc.,
  19. // 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
  20. //
  21. // Linking this file statically or dynamically with other modules is making
  22. // a combined work based on this file.  Thus, the terms and conditions of
  23. // the GNU General Public License cover the whole combination.
  24. //
  25. // In addition, as a special exception, the copyright holders of this file
  26. // give you permission to combine this file with free software programs or
  27. // libraries that are released under the GNU LGPL and with code included in
  28. // the standard release of ns-2 under the Apache 2.0 license or under
  29. // otherwise-compatible licenses with advertising requirements (or modified
  30. // versions of such code, with unchanged license).  You may copy and
  31. // distribute such a system following the terms of the GNU GPL for this
  32. // file and the licenses of the other code concerned, provided that you
  33. // include the source code of that other code when and as the GNU GPL
  34. // requires distribution of source code.
  35. //
  36. // Note that people who make modified versions of this file are not
  37. // obligated to grant this special exception for their modified versions;
  38. // it is their choice whether to do so.  The GNU General Public License
  39. // gives permission to release a modified version without this exception;
  40. // this exception also makes it possible to release a modified version
  41. // which carries forward this exception.
  42. #include <stdio.h>
  43. #include <stdlib.h>
  44. #include "iolog.hh"
  45. IOLog::IOLog(int32_t id) : IOHook()
  46. {
  47.   node_id_ = id;
  48.   DiffPrint(DEBUG_IMPORTANT, "Initializing IOLog...n");
  49. }
  50. DiffPacket IOLog::recvPacket(int fd)
  51. {
  52.   DiffPacket incoming_packet;
  53.   struct timeval tv;
  54.   struct hdr_diff *dfh;
  55.   u_int16_t data_len, packet_len;
  56.   int32_t last_hop, next_hop;
  57.   int msg_type;
  58.   char *msg_name;
  59.   // Receive the packet first
  60.   incoming_packet = IOHook::recvPacket(fd);
  61.   if (incoming_packet){
  62.     // Log incoming packet
  63.     dfh = HDR_DIFF(incoming_packet);
  64.     last_hop = ntohl(LAST_HOP(dfh));
  65.     next_hop = ntohl(NEXT_HOP(dfh));
  66.     data_len = ntohs(DATA_LEN(dfh));
  67.     msg_type = MSG_TYPE(dfh);
  68.     packet_len = data_len + sizeof(hdr_diff);
  69.     switch (msg_type){
  70.     case INTEREST:
  71.       msg_name = strdup("Interest");
  72.       break;
  73.     case DATA:
  74.       msg_name = strdup("Data");
  75.       break;
  76.     case EXPLORATORY_DATA:
  77.       msg_name = strdup("Exploratory Data");
  78.       break;
  79.     case PUSH_EXPLORATORY_DATA:
  80.       msg_name = strdup("Push Exploratory Data");
  81.       break;
  82.     case POSITIVE_REINFORCEMENT:
  83.       msg_name = strdup("Positive Reinforcement");
  84.       break;
  85.     case NEGATIVE_REINFORCEMENT:
  86.       msg_name = strdup("Negative Reinforcement");
  87.       break;
  88.     default:
  89.       msg_name = strdup("Unknown");
  90.       break;
  91.     }
  92.     if (last_hop != LOCALHOST_ADDR){
  93.       GetTime(&tv);
  94.       if (next_hop == BROADCAST_ADDR){
  95. fprintf(stdout,
  96. "Diffusion Log: Time %ld.%06ld Node %d received broadcast %d bytes from node %d message %sn",
  97. tv.tv_sec, (long int) tv.tv_usec, node_id_, packet_len, last_hop, msg_name);
  98.       }
  99.       else{
  100. fprintf(stdout,
  101. "Diffusion Log: Time %ld.%06ld Node %d received unicast %d bytes from node %d message %sn",
  102. tv.tv_sec, (long int) tv.tv_usec, node_id_, packet_len, last_hop, msg_name);
  103.       }
  104.       fflush(NULL);
  105.     }
  106.     free(msg_name);
  107.   }
  108.   return incoming_packet;
  109. }
  110. void IOLog::sendPacket(DiffPacket pkt, int len, int dst)
  111. {
  112.   struct timeval tv;
  113.   struct hdr_diff *dfh;
  114.   int msg_type;
  115.   char *msg_name;
  116.   // Log outgoing packet
  117.   dfh = HDR_DIFF(pkt);
  118.   msg_type = MSG_TYPE(dfh);
  119.   switch (msg_type){
  120.   case INTEREST:
  121.     msg_name = strdup("Interest");
  122.     break;
  123.   case DATA:
  124.     msg_name = strdup("Data");
  125.     break;
  126.   case EXPLORATORY_DATA:
  127.     msg_name = strdup("Exploratory Data");
  128.     break;
  129.   case PUSH_EXPLORATORY_DATA:
  130.     msg_name = strdup("Push Exploratory Data");
  131.     break;
  132.   case POSITIVE_REINFORCEMENT:
  133.     msg_name = strdup("Positive Reinforcement");
  134.     break;
  135.   case NEGATIVE_REINFORCEMENT:
  136.     msg_name = strdup("Negative Reinforcement");
  137.     break;
  138.   default:
  139.     msg_name = strdup("Unknown");
  140.     break;
  141.   }
  142.   // Get local time
  143.   GetTime(&tv);
  144.   if (dst == BROADCAST_ADDR){
  145.     fprintf(stdout,
  146.     "Diffusion Log: Time %ld.%06ld Node %d sending broadcast %d bytes to node %d message %sn",
  147.     tv.tv_sec, (long int) tv.tv_usec, node_id_, len, dst, msg_name);
  148.   }
  149.   else{
  150.     fprintf(stdout,
  151.     "Diffusion Log: Time %ld.%06ld Node %d sending unicast %d bytes to node %d message %sn",
  152.     tv.tv_sec, (long int) tv.tv_usec, node_id_, len, dst, msg_name);
  153.   }
  154.   fflush(NULL);
  155.   free(msg_name);
  156.   // Send packet to device
  157.   IOHook::sendPacket(pkt, len, dst);
  158. }