通讯编程

开发平台：
Visual C++

iolog.cc：源码内容
							//
// iolog.cc      : IO Log Layer
// Authors       : Fabio Silva and Yutaka Mori
//
// Copyright (C) 2000-2002 by the University of Southern California
// $Id: iolog.cc,v 1.4 2008/03/27 05:24:46 tom_henderson Exp $
//
// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License,
// version 2, as published by the Free Software Foundation.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License along
// with this program; if not, write to the Free Software Foundation, Inc.,
// 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
//
// Linking this file statically or dynamically with other modules is making
// a combined work based on this file.  Thus, the terms and conditions of
// the GNU General Public License cover the whole combination.
//
// In addition, as a special exception, the copyright holders of this file
// give you permission to combine this file with free software programs or
// libraries that are released under the GNU LGPL and with code included in
// the standard release of ns-2 under the Apache 2.0 license or under
// otherwise-compatible licenses with advertising requirements (or modified
// versions of such code, with unchanged license).  You may copy and
// distribute such a system following the terms of the GNU GPL for this
// file and the licenses of the other code concerned, provided that you
// include the source code of that other code when and as the GNU GPL
// requires distribution of source code.
//
// Note that people who make modified versions of this file are not
// obligated to grant this special exception for their modified versions;
// it is their choice whether to do so.  The GNU General Public License
// gives permission to release a modified version without this exception;
// this exception also makes it possible to release a modified version
// which carries forward this exception.
#include <stdio.h>
#include <stdlib.h>
#include "iolog.hh"
IOLog::IOLog(int32_t id) : IOHook()
{
  node_id_ = id;
  DiffPrint(DEBUG_IMPORTANT, "Initializing IOLog...n");
}
DiffPacket IOLog::recvPacket(int fd)
{
  DiffPacket incoming_packet;
  struct timeval tv;
  struct hdr_diff *dfh;
  u_int16_t data_len, packet_len;
  int32_t last_hop, next_hop;
  int msg_type;
  char *msg_name;
  // Receive the packet first
  incoming_packet = IOHook::recvPacket(fd);
  if (incoming_packet){
    // Log incoming packet
    dfh = HDR_DIFF(incoming_packet);
    last_hop = ntohl(LAST_HOP(dfh));
    next_hop = ntohl(NEXT_HOP(dfh));
    data_len = ntohs(DATA_LEN(dfh));
    msg_type = MSG_TYPE(dfh);
    packet_len = data_len + sizeof(hdr_diff);
    switch (msg_type){
    case INTEREST:
      msg_name = strdup("Interest");
      break;
    case DATA:
      msg_name = strdup("Data");
      break;
    case EXPLORATORY_DATA:
      msg_name = strdup("Exploratory Data");
      break;
    case PUSH_EXPLORATORY_DATA:
      msg_name = strdup("Push Exploratory Data");
      break;
    case POSITIVE_REINFORCEMENT:
      msg_name = strdup("Positive Reinforcement");
      break;
    case NEGATIVE_REINFORCEMENT:
      msg_name = strdup("Negative Reinforcement");
      break;
    default:
      msg_name = strdup("Unknown");
      break;
    }
    if (last_hop != LOCALHOST_ADDR){
      GetTime(&tv);
      if (next_hop == BROADCAST_ADDR){
	fprintf(stdout,
		"Diffusion Log: Time %ld.%06ld Node %d received broadcast %d bytes from node %d message %sn",
		tv.tv_sec, (long int) tv.tv_usec, node_id_, packet_len, last_hop, msg_name);
      }
      else{
	fprintf(stdout,
		"Diffusion Log: Time %ld.%06ld Node %d received unicast %d bytes from node %d message %sn",
		tv.tv_sec, (long int) tv.tv_usec, node_id_, packet_len, last_hop, msg_name);
      }
      fflush(NULL);
    }
    free(msg_name);
  }
  return incoming_packet;
}
void IOLog::sendPacket(DiffPacket pkt, int len, int dst)
{
  struct timeval tv;
  struct hdr_diff *dfh;
  int msg_type;
  char *msg_name;
  // Log outgoing packet
  dfh = HDR_DIFF(pkt);
  msg_type = MSG_TYPE(dfh);
  switch (msg_type){
  case INTEREST:
    msg_name = strdup("Interest");
    break;
  case DATA:
    msg_name = strdup("Data");
    break;
  case EXPLORATORY_DATA:
    msg_name = strdup("Exploratory Data");
    break;
  case PUSH_EXPLORATORY_DATA:
    msg_name = strdup("Push Exploratory Data");
    break;
  case POSITIVE_REINFORCEMENT:
    msg_name = strdup("Positive Reinforcement");
    break;
  case NEGATIVE_REINFORCEMENT:
    msg_name = strdup("Negative Reinforcement");
    break;
  default:
    msg_name = strdup("Unknown");
    break;
  }
  // Get local time
  GetTime(&tv);
  if (dst == BROADCAST_ADDR){
    fprintf(stdout,
	    "Diffusion Log: Time %ld.%06ld Node %d sending broadcast %d bytes to node %d message %sn",
	    tv.tv_sec, (long int) tv.tv_usec, node_id_, len, dst, msg_name);
  }
  else{
    fprintf(stdout,
	    "Diffusion Log: Time %ld.%06ld Node %d sending unicast %d bytes to node %d message %sn",
	    tv.tv_sec, (long int) tv.tv_usec, node_id_, len, dst, msg_name);
  }
  fflush(NULL);
  free(msg_name);
  // Send packet to device
  IOHook::sendPacket(pkt, len, dst);
}

						