00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037
00038
00039
00040
00041
00042
00043
00044
00045
00046
00047
00048
00049
00050
00051
00052
00053 #ifndef ns_diffusion_h
00054 #define ns_diffusion_h
00055
00056 #include <assert.h>
00057 #include <math.h>
00058 #include <stdio.h>
00059 #include <signal.h>
00060 #include <float.h>
00061
00062 #include <tcl.h>
00063 #include <stdlib.h>
00064
00065 #include "diff_header.h"
00066 #include "agent.h"
00067 #include "tclcl.h"
00068 #include "ip.h"
00069 #include "config.h"
00070 #include "packet.h"
00071 #include "trace.h"
00072 #include "random.h"
00073 #include "classifier.h"
00074 #include "node.h"
00075 #include "iflist.h"
00076 #include "hash_table.h"
00077 #include "arp.h"
00078 #include "mac.h"
00079 #include "ll.h"
00080 #include "dsr/path.h"
00081 #include "routing_table.h"
00082
00083
00084 #define THIS_NODE here_.addr_
00085 #define MAC_RETRY_ 0 // 0: NO RETRY
00086
00087
00088 #define JITTER 0.11 // (sec) to jitter broadcast
00089 #define ARP_BUFFER_CHECK 0.03 // (sec) between buffer checks
00090 #define ARP_MAX_ATTEMPT 3 // (times)
00091 #define ARP_BUF_SIZE 64 // (slots)
00092
00093 #define SEND_BUFFER_CHECK 0.03 // (sec) between buffer checks
00094 #define SEND_TIMEOUT 30.0 // (sec) a packet can live in sendbuf
00095 #define SEND_BUF_SIZE 64 // (slots)
00096
00097
00098 #define SEND_MESSAGE(x,y,z) send_to_dmux(prepare_message(x,y,z), 0)
00099 #define max(a,b) (((a)<(b))?(b):(a))
00100
00101 class DiffusionAgent;
00102
00103
00104 class ArpBufEntry {
00105 public:
00106 Time t;
00107 int attempt;
00108 Packet *p;
00109
00110 ArpBufEntry() { t=0; attempt=0; p=NULL; }
00111 };
00112
00113
00114 class ArpBufferTimer : public TimerHandler {
00115 public:
00116 ArpBufferTimer(DiffusionAgent *a) : TimerHandler() { a_ = a; }
00117 void expire(Event *e);
00118 protected:
00119 DiffusionAgent *a_;
00120 };
00121
00122
00123 class SendBufferEntry {
00124 public:
00125 Time t;
00126 Packet *p;
00127
00128 SendBufferEntry() { t=0; p=NULL; }
00129 };
00130
00131
00132 class SendBufTimer : public TimerHandler {
00133 public:
00134 SendBufTimer(DiffusionAgent *a) : TimerHandler() { a_ = a; }
00135 void expire(Event *e);
00136 protected:
00137 DiffusionAgent *a_;
00138 };
00139
00140
00141 class DiffusionAgent : public Agent {
00142 public:
00143 DiffusionAgent();
00144 int command(int argc, const char*const* argv);
00145 void recv(Packet*, Handler*);
00146
00147 Diff_Routing_Entry routing_table[MAX_DATA_TYPE];
00148
00149
00150 protected:
00151
00152
00153
00154 bool POS_REINF_;
00155 bool NEG_REINF_;
00156
00157 int pk_count;
00158 int overhead;
00159
00160 Pkt_Hash_Table PktTable;
00161
00162 Node *node;
00163 Trace *tracetarget;
00164 NsObject *ll;
00165 NsObject *port_dmux;
00166 ARPTable *arp_table;
00167
00168 ArpBufferTimer arp_buf_timer;
00169 ArpBufEntry arp_buf[ARP_BUF_SIZE];
00170 SendBufTimer send_buf_timer;
00171 SendBufferEntry send_buf[SEND_BUF_SIZE];
00172
00173
00174
00175 inline void send_to_dmux(Packet *pkt, Handler *h) {
00176 port_dmux->recv(pkt, h);
00177 }
00178
00179 void clear_arp_buf();
00180 void clear_send_buf();
00181 void reset();
00182 void consider_old(Packet *);
00183 void consider_new(Packet *);
00184 void Terminate();
00185 virtual void Start();
00186
00187 Packet *create_packet();
00188 Packet *prepare_message(unsigned int dtype, ns_addr_t to_addr, int msg_type);
00189
00190 virtual void Print_IOlist();
00191 void DataForSink(Packet *);
00192 void StopSource();
00193
00194 void MACprepare(Packet *pkt, nsaddr_t next_hop, int type,
00195 bool lk_dtct);
00196 void MACsend(Packet *pkt, Time delay=0);
00197 void xmitFailed(Packet *pkt);
00198
00199 void StickPacketInArpBuffer(Packet *pkt);
00200 void ArpBufferCheck();
00201 void SendBufferCheck();
00202 void StickPacketInSendBuffer(Packet *p);
00203
00204 void trace(char *fmt,...);
00205
00206 friend void XmitFailedCallback(Packet *pkt, void *data);
00207 friend class ArpBufferTimer;
00208 friend class SendBufTimer;
00209 };
00210
00211 #endif
00212
00213