通讯编程

开发平台：
Visual C++

omni_mcast.cc：源码内容
							
/*
 * omni_mcast.cc
 * Copyright (C) 2000 by the University of Southern California
 * $Id: omni_mcast.cc,v 1.14 2006/03/10 12:25:28 mahrenho 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.
 *
 *
 * The copyright of this module includes the following
 * linking-with-specific-other-licenses addition:
 *
 * In addition, as a special exception, the copyright holders of
 * this module give you permission to combine (via static or
 * dynamic linking) this module 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 module 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 module
 * 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.
 *
 */
/*****************************************************************/
/* omni_mcast.cc : Chalermek Intanagonwiwat (USC/ISI)  05/18/99  */
/*****************************************************************/
// Share api with diffusion and flooding
// Using diffusion packet header
#include <assert.h>
#include <math.h>
#include <stdio.h>
#include <signal.h>
#include <float.h>
#include <stdlib.h>
#include <tcl.h>
#include "diff_header.h"
#include "agent.h"
#include "tclcl.h"
#include "ip.h"
#include "config.h"
#include "packet.h"
#include "trace.h"
#include "random.h"
#include "classifier.h"
#include "node.h"
#include "omni_mcast.h"
#include "iflist.h"
#include "hash_table.h"
#include "arp.h"
#include "mac.h"
#include "ll.h"
#include "dsr/path.h"
#include "god.h"
/* Callback helper */
void OmniMcastXmitFailedCallback(Packet *pkt, void *data)
{
  OmniMcastAgent *agent = (OmniMcastAgent *)data;  // cast of trust
  agent->xmitFailed(pkt);
}
static class OmniMcastClass : public TclClass {
public:
  OmniMcastClass() : TclClass("Agent/OmniMcast") {}
  TclObject* create(int argc, const char*const* argv) {
    return(new OmniMcastAgent());
  }
} class_omni_mcast;
void OmniMcastArpBufferTimer::expire(Event *e)
{
  a_->ArpBufferCheck();
  resched(ARP_BUFFER_CHECK + ARP_BUFFER_CHECK * 
	  (double) ((long) e>>5 & 0xff) /256.0);
}
void OmniMcastSendBufTimer::expire(Event *e)
{
  a_->SendBufferCheck();
  resched(SEND_BUFFER_CHECK + SEND_BUFFER_CHECK * (double) ((long) e>>5 & 0xff)/256.0);
}
void OmniMcastAgent::DataForSink(Packet *pkt)
{
  hdr_cdiff     *dfh  = HDR_CDIFF(pkt);
  unsigned int dtype = dfh->data_type;
  Agent_List   *cur_agent;
  Packet       *cur_pkt;
  hdr_cdiff     *cur_dfh;
  hdr_ip       *cur_iph;
  for (cur_agent= (routing_table[dtype]).sink; cur_agent != NULL; 
	   cur_agent= AGENT_NEXT(cur_agent) ) {
      cur_pkt       = pkt->copy();
      cur_iph       = HDR_IP(cur_pkt);
      cur_iph->dst_ = AGT_ADDR(cur_agent);
      cur_dfh       = HDR_CDIFF(cur_pkt);
      cur_dfh->forward_agent_id = here_;
      cur_dfh->num_next = 1;
      cur_dfh->next_nodes[0] = NODE_ADDR(cur_agent);
      send_to_dmux(cur_pkt, 0);
  }
}
void OmniMcastAgent::GodForwardData(Packet *pkt)
{
  hdr_cdiff     *dfh  = HDR_CDIFF(pkt);
  unsigned int dtype = dfh->data_type;
  Packet       *cur_pkt;
  hdr_cdiff     *cur_dfh;
  hdr_ip       *cur_iph;
  nsaddr_t     src_node = (dfh->sender_id).addr_;
  int          ret_num_oif;
  int *next_oif= God::instance()->NextOIFs(dtype, src_node, THIS_NODE, 
					   &ret_num_oif);
  if (ret_num_oif == 0) {
    Packet::free(pkt);
    return;
  }
  assert(next_oif != NULL);
  for (int i=0; i<ret_num_oif; i++) {
      cur_pkt       = pkt->copy();
      cur_iph       = HDR_IP(cur_pkt);
      (cur_iph->dst_).addr_ = next_oif[i];
      (cur_iph->dst_).port_ = ROUTING_PORT;
      cur_dfh       = HDR_CDIFF(cur_pkt);
      cur_dfh->forward_agent_id = here_;
      cur_dfh->num_next = 1;
      cur_dfh->next_nodes[0] = next_oif[i];
      MACprepare(cur_pkt, next_oif[i], NS_AF_INET, MAC_RETRY_);
      MACsend(cur_pkt, 0);
  }
   delete []next_oif;
   Packet::free(pkt);
}
Packet *OmniMcastAgent::prepare_message(unsigned int dtype, ns_addr_t to_addr, 
				  int msg_type)
{
  Packet *pkt;
  hdr_cdiff *dfh;
  hdr_ip *iph;
    pkt = create_packet();
    dfh = HDR_CDIFF(pkt);
    iph = HDR_IP(pkt);
    
    dfh->mess_type = msg_type;
    dfh->pk_num = pk_count;
    pk_count++;
    dfh->sender_id = here_;
    dfh->data_type = dtype;
    dfh->forward_agent_id = here_;
    dfh->ts_ = NOW;
    dfh->num_next = 1;
    dfh->next_nodes[0] = to_addr.addr_;
    
    iph->src_ = here_;
    iph->dst_ = to_addr;
    return pkt;
}
OmniMcastAgent::OmniMcastAgent() : Agent(PT_DIFF), arp_buf_timer(this), 
  send_buf_timer(this)
{
  pk_count = 0;
  target_ = 0;
  node = NULL;
  tracetarget = NULL;
}
void OmniMcastAgent::recv(Packet* packet, Handler*)
{
  hdr_cdiff* dfh = HDR_CDIFF(packet);
  // Packet Hash Table is used to keep info about experienced pkts.
  Pkt_Hash_Entry *hashPtr= PktTable.GetHash(dfh->sender_id, dfh->pk_num);
     // Received this packet before ?
     if (hashPtr != NULL) {
       Packet::free(packet);
       return;
     }
     // Never receive it before ? Put in hash table.
     PktTable.put_in_hash(dfh);
     // Take action for a new pkt.
     ConsiderNew(packet);     
}
void OmniMcastAgent::ConsiderNew(Packet *pkt)
{
  hdr_cdiff* dfh = HDR_CDIFF(pkt);
  unsigned char msg_type = dfh->mess_type;
  unsigned int dtype = dfh->data_type;
  Pkt_Hash_Entry *hashPtr;
  Agent_List *agentPtr;
  PrvCurPtr  RetVal;
  nsaddr_t   from_nodeID, forward_nodeID;
  Packet *gen_pkt;
  hdr_cdiff *gen_dfh;
  switch (msg_type) {
    case INTEREST : 
      hashPtr = PktTable.GetHash(dfh->sender_id, dfh->pk_num);
      // Check if it comes from sink agent of this node
      // If so we have to keep it in sink list 
      from_nodeID = (dfh->sender_id).addr_;
      forward_nodeID = (dfh->forward_agent_id).addr_;
      if (THIS_NODE == from_nodeID) {       
	// It's from a sink on this node.
	// Is it already in list ?
	RetVal = INTF_FIND(routing_table[dtype].sink, dfh->sender_id);
	if (RetVal.cur == NULL) {            
	  // No, it's not.
	  agentPtr = new Agent_List;
	  AGT_ADDR(agentPtr) = dfh->sender_id;
	  INTF_INSERT(routing_table[dtype].sink, agentPtr);
	  God::instance()->AddSink(dtype, THIS_NODE);
	}	
      }
      Packet::free(pkt);
      return;
    case DATA_READY :
      // put source_agent in source list of routing table
      agentPtr = new Agent_List;
      AGT_ADDR(agentPtr) = dfh->sender_id;
      agentPtr->next = routing_table[dtype].source;
      routing_table[dtype].source = agentPtr;
      God::instance()->AddSource(dtype, (dfh->sender_id).addr_);
      gen_pkt = prepare_message(dtype, dfh->sender_id, DATA_REQUEST);
      gen_dfh = HDR_CDIFF(gen_pkt);
      gen_dfh->report_rate = ORIGINAL;
      send_to_dmux(gen_pkt, 0);
      Packet::free(pkt);
      return;
    case DATA :
      DataForSink(pkt);
      GodForwardData(pkt);
      return;
    default : 
      
      Packet::free(pkt);        
      break;
  }
}
void OmniMcastAgent::Terminate() 
{
#ifdef DEBUG_OUTPUT
	printf("node %d: remaining energy %f, initial energy %fn", THIS_NODE, 
	       node->energy_model()->energy(), 
	       node->energy_model()->initialenergy() );
#endif
}
void OmniMcastAgent::Start()
{
  arp_buf_timer.sched(ARP_BUFFER_CHECK + ARP_BUFFER_CHECK * 
		      Random::uniform(1.0));	  
  send_buf_timer.sched(SEND_BUFFER_CHECK + SEND_BUFFER_CHECK *
			   Random::uniform(1.0));
}
void OmniMcastAgent::StopSource()
{
  Agent_List *cur;
  for (int i=0; i<MAX_DATA_TYPE; i++) {
    for (cur=routing_table[i].source; cur!=NULL; cur=AGENT_NEXT(cur) ) {
      SEND_MESSAGE(i, AGT_ADDR(cur), DATA_STOP);
    }
  }
}
Packet * OmniMcastAgent:: create_packet()
{
  Packet *pkt = allocpkt();
  if (pkt==NULL) return NULL;
  hdr_cmn*  cmh = HDR_CMN(pkt);
  cmh->size() = 36;
  hdr_cdiff* dfh = HDR_CDIFF(pkt);
  dfh->ts_ = NOW;
  return pkt;
}
void OmniMcastAgent::MACprepare(Packet *pkt, nsaddr_t next_hop, 
				unsigned int type, bool lk_dtct)
{
  hdr_cdiff* dfh = HDR_CDIFF(pkt);
  hdr_cmn* cmh = HDR_CMN(pkt);
  hdr_ip*  iph = HDR_IP(pkt);
  dfh->forward_agent_id = here_; 
  if (type == NS_AF_ILINK && next_hop == (nsaddr_t)MAC_BROADCAST) {
      cmh->xmit_failure_ = 0;
      cmh->next_hop() = MAC_BROADCAST;
      cmh->addr_type() = NS_AF_ILINK;
      cmh->direction() = hdr_cmn::DOWN;
      
      iph->src_ = here_;
      iph->dst_.addr_ = next_hop;
      iph->dst_.port_ = ROUTING_PORT;
      dfh->num_next = 1;
      dfh->next_nodes[0] = next_hop;
      return;     
  }  
  if (lk_dtct != 0) {
    cmh->xmit_failure_ = OmniMcastXmitFailedCallback;
    cmh->xmit_failure_data_ = (void *) this;
  }
  else {
    cmh->xmit_failure_ = 0;
  }
  cmh->direction() = hdr_cmn::DOWN;
  cmh->next_hop() = next_hop;
  cmh->addr_type() = type;  
  iph->src_ = here_;
  iph->dst_.addr_ = next_hop;
  iph->dst_.port_ = ROUTING_PORT;
  
  dfh->num_next = 1;
  dfh->next_nodes[0] = next_hop;
}
void OmniMcastAgent::MACsend(Packet *pkt, Time delay)
{
  hdr_cmn*  cmh = HDR_CMN(pkt);
  hdr_cdiff* dfh = HDR_CDIFF(pkt);
  if (dfh->mess_type == DATA)
    cmh->size() = (God::instance()->data_pkt_size) + 4*(dfh->num_next - 1);
  else
    cmh->size() = 36 + 4*(dfh->num_next -1);
  Scheduler::instance().schedule(ll, pkt, delay);
}
void OmniMcastAgent::xmitFailed(Packet *pkt)
{
  // For future extension if needed.
}
void OmniMcastAgent::StickPacketInArpBuffer(Packet *pkt)
{
  Time min = DBL_MAX;
  int  min_index = 0;
  int  c;
  for (c=0; c < ARP_BUF_SIZE; c++) {
    if (arp_buf[c].p == NULL) {
      arp_buf[c].t = NOW;
      arp_buf[c].attempt = 1;
      arp_buf[c].p = pkt;
      return;
    }
    else if (arp_buf[c].t < min) {
      min = arp_buf[c].t;
      min_index = c;
    }
  }
  // Before killing somebody, let him get a last chance to send.
  ARPEntry *llinfo;
  hdr_cmn*  cmh = HDR_CMN(arp_buf[min_index].p);
  llinfo= arp_table->arplookup(cmh->next_hop());
  if (llinfo == 0) {
    // printf("ARP fails. And must give up slot.n");
      xmitFailed(arp_buf[min_index].p);
  }
  else
      MACsend(arp_buf[min_index].p, 0);
  // The new packet is taking over the slot of the dead guy.
  arp_buf[min_index].t = NOW;
  arp_buf[min_index].attempt = 1;
  arp_buf[min_index].p = pkt;
}
void OmniMcastAgent::ArpBufferCheck()
{
  int c;
  ARPEntry *llinfo;
  hdr_cmn*  cmh;
  for (c = 0; c < ARP_BUF_SIZE; c++) {
    if (arp_buf[c].p == NULL)
      continue;
    
    cmh = HDR_CMN(arp_buf[c].p);
    llinfo= arp_table->arplookup(cmh->next_hop());
    if (llinfo != 0) {
      MACsend(arp_buf[c].p, 0);
      arp_buf[c].p = NULL;
      continue;}
    
    if (arp_buf[c].attempt > ARP_MAX_ATTEMPT) {
      // printf("ARP fails. Too many attempts.n");
      xmitFailed(arp_buf[c].p);
      arp_buf[c].p = NULL;
      continue;
    }
    arp_table->arprequest(THIS_NODE, cmh->next_hop(), (LL *)ll);
    arp_buf[c].attempt ++;
  }
}
void OmniMcastAgent::StickPacketInSendBuffer(Packet *p)
{
  Time min = DBL_MAX;
  int min_index = 0;
  int c;
  for (c = 0 ; c < SEND_BUF_SIZE ; c ++) {
    if (send_buf[c].p  == NULL)
      {
	send_buf[c].t = NOW;
	send_buf[c].p = p;
	return;
      }
    else if (send_buf[c].t < min)
      {
	min = send_buf[c].t;
	min_index = c;
      }
  }
  // Before killing somebody, you'd better give him the last chance.
  if (send_buf[min_index].p != NULL) {
    MACsend(send_buf[min_index].p, 0);
  }
  // A new packet is taking over the slot.
  send_buf[min_index].t = Scheduler::instance().clock();
  send_buf[min_index].p = p;
}
void OmniMcastAgent::SendBufferCheck()
{
  for (int c = 0; c < SEND_BUF_SIZE; c++) {
    if (send_buf[c].p != NULL) {
      MACsend(send_buf[c].p, 0);
      send_buf[c].p = NULL;
    }
  }
}
void OmniMcastAgent::trace (char *fmt,...)
{
  va_list ap;
  if (!tracetarget)
    return;
  va_start (ap, fmt);
  vsprintf (tracetarget->pt_->buffer (), fmt, ap);
  tracetarget->pt_->dump ();
  va_end (ap);
}
int OmniMcastAgent::command(int argc, const char*const* argv)
{  
  Tcl& tcl =  Tcl::instance();
  if (argc == 2) {
    if (strcasecmp(argv[1], "reset-state")==0) {
      reset();
      return TCL_OK;
    }
    if (strcasecmp(argv[1], "reset")==0) {
      return Agent::command(argc, argv);
    }
    if (strcasecmp(argv[1], "start")==0) {
      Start();
      return TCL_OK;
    }
    if (strcasecmp(argv[1], "stop")==0) {
      return TCL_OK;
    }
    if (strcasecmp(argv[1], "terminate")==0) {
      Terminate();
      return TCL_OK;
    }
    if (strcasecmp(argv[1], "stop-source")==0) {
      StopSource();
      return TCL_OK;
    }
  } else if (argc == 3) {
    if (strcasecmp(argv[1], "on-node")==0) {
      node = (Node *)tcl.lookup(argv[2]);
      return TCL_OK;
    }
    if (strcasecmp(argv[1], "add-ll") == 0) {
      TclObject *obj;
      if ( (obj = TclObject::lookup(argv[2])) == 0) {
	fprintf(stderr, "OmniMcast Node: %d lookup of %s failedn", THIS_NODE,
		argv[2]);
	return TCL_ERROR;
      }
      ll = (NsObject *) obj;
      // What a hack !!!
      arp_table = ((LL *)ll)->arp_table();
      if (arp_table == NULL) 
	return TCL_ERROR;
     return TCL_OK;
    }
    if (strcasecmp (argv[1], "tracetarget") == 0) {
      TclObject *obj;
      if ((obj = TclObject::lookup (argv[2])) == 0) {
	  fprintf (stderr, "%s: %s lookup of %s failedn", __FILE__, argv[1],
		   argv[2]);
	  return TCL_ERROR;
      }
      tracetarget = (Trace *) obj;
      return TCL_OK;
    }
    if (strcasecmp(argv[1], "port-dmux") == 0) {
      TclObject *obj;
      if ( (obj = TclObject::lookup(argv[2])) == 0) {
	fprintf(stderr, "OmniMcast Node: %d lookup of %s failedn", THIS_NODE,
		argv[2]);
	return TCL_ERROR;
      }
      port_dmux = (NsObject *) obj;
      return TCL_OK;
    }
  } 
  return Agent::command(argc, argv);
}
void OmniMcastAgent::reset()
{
  PktTable.reset();
  for (int i=0; i<MAX_DATA_TYPE; i++) {
    routing_table[i].reset();
  }
  clear_arp_buf();
  clear_send_buf();
}
void OmniMcastAgent::clear_arp_buf()
{
  for (int i=0; i<ARP_BUF_SIZE; i++) {
    arp_buf[i].t = 0;
    arp_buf[i].attempt = 0;
    if (arp_buf[i].p != NULL) 
      Packet::free(arp_buf[i].p);
    arp_buf[i].p = NULL;
  }
}
void OmniMcastAgent::clear_send_buf()
{
  for (int i=0; i<SEND_BUF_SIZE; i++) {
    send_buf[i].t = 0;
    if (send_buf[i].p != NULL)
      Packet::free(send_buf[i].p);
    send_buf[i].p = NULL;
  }
}
				    
void OmniMcast_Entry::reset()
{
  clear_agentlist(source);
  clear_agentlist(sink);
  source = NULL;
  sink = NULL;
}
void OmniMcast_Entry::clear_agentlist(Agent_List *list)
{
  Agent_List *cur=list;
  Agent_List *temp = NULL;
  while (cur != NULL) {
    temp = AGENT_NEXT(cur);
    delete cur;
    cur = temp;
  }
}

						