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 #ifdef NS_DIFFUSION
00050
00051 #include "difftimer.h"
00052 #include "diffagent.h"
00053 #include "diffrtg.h"
00054
00055 DiffEvent::DiffEvent(d_handle hdl, void *payload, int time) {
00056 handle_ = hdl;
00057 payload_ = payload;
00058 GetTime(&tv_);
00059 TimevalAddusecs(&tv_, time*1000);
00060 }
00061
00062 void DiffEventQueue::eqAddAfter(d_handle hdl, void *payload, int delay_msec) {
00063 DiffEvent* de;
00064
00065 de = new DiffEvent(hdl, payload, delay_msec);
00066 DiffEventHandler *dh = timerHandler_;
00067 double delay = ((double)delay_msec)/1000;
00068
00069 (void)Scheduler::instance().schedule(dh, de, delay);
00070 scheduler_uid_t uid = ((Event *)de)->uid_;
00071
00072 setUID(hdl, uid);
00073 }
00074
00075
00076 DiffEvent *DiffEventQueue::eqFindEvent(d_handle hdl) {
00077 scheduler_uid_t uid = getUID(hdl);
00078 if (uid != -1) {
00079 Event *p = Scheduler::instance().lookup(uid);
00080 if (p != 0) {
00081 return ((DiffEvent *)p);
00082 } else {
00083 fprintf(stderr, "Error: Can't find event in scheduler queue\n");
00084 return NULL;
00085 }
00086 } else {
00087 fprintf(stderr, "Error: Can't find event in uidmap!\n");
00088 return NULL;
00089 }
00090 }
00091
00092 bool DiffEventQueue::eqRemove(d_handle hdl) {
00093
00094 scheduler_uid_t uid = getUID(hdl);
00095 if (uid != -1) {
00096
00097 Event *p = Scheduler::instance().lookup(uid);
00098 if (p != 0) {
00099
00100 Scheduler::instance().cancel(p);
00101 TimerEntry* te = (TimerEntry *)((DiffEvent *)p)->payload();
00102 delete te;
00103 delete p;
00104 return 1;
00105 } else {
00106 fprintf(stderr, "Error: Can't find event in scheduler queue\n");
00107 return 0;
00108 }
00109 } else {
00110 fprintf(stderr, "Error: Can't find event in uidmap!\n");
00111 return 0;
00112 }
00113 }
00114
00115
00116 void DiffEventQueue::setUID(d_handle hdl, scheduler_uid_t uid) {
00117 MapEntry *me = new MapEntry;
00118 me->hdl_ = hdl;
00119 me->uid_ = uid;
00120 uidmap_.push_back(me);
00121 }
00122
00123
00124 scheduler_uid_t DiffEventQueue::getUID(d_handle hdl) {
00125 UIDMap_t::iterator itr;
00126 MapEntry* entry;
00127 for (itr = uidmap_.begin(); itr != uidmap_.end(); ++itr) {
00128 entry = *itr;
00129 if (entry->hdl_ == hdl) {
00130
00131 scheduler_uid_t uid = entry->uid_;
00132 itr = uidmap_.erase(itr);
00133 delete entry;
00134 return uid;
00135 }
00136 }
00137 return (-1);
00138 }
00139
00140
00141 void DiffEventHandler::handle(Event *e) {
00142 DiffEvent *de = (DiffEvent *)e;
00143
00144 d_handle hdl = de->getHandle();
00145 queue_->getUID(hdl);
00146 a_->diffTimeout(de);
00147 }
00148
00149
00150 #endif // NS