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

通讯编程

开发平台:

Visual C++

  1. //
  2. // iohook.cc     : IO Hook for Diffusion
  3. // Authors       : Fabio Silva and Yutaka Mori
  4. //
  5. // Copyright (C) 2000-2002 by the University of Southern California
  6. // $Id: iohook.cc,v 1.2 2005/09/13 04:53:49 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. //
  43. #include <stdio.h>
  44. #include <stdlib.h>
  45. #include "iohook.hh"
  46. IOHook::IOHook() : DiffusionIO()
  47. {
  48. }
  49. DiffPacket IOHook::recvPacket(int fd)
  50. {
  51.   DeviceList::iterator device_itr;
  52.   DiffPacket incoming_packet = NULL;
  53.   for (device_itr = in_devices_.begin();
  54.        device_itr != in_devices_.end(); ++device_itr){
  55.     if ((*device_itr)->hasFD(fd)){
  56.       // We now have the correct device
  57.       incoming_packet = (*device_itr)->recvPacket(fd);
  58.       break;
  59.     }
  60.   }
  61.   // Make sure packet was received
  62.   if (incoming_packet == NULL){
  63.     DiffPrint(DEBUG_ALWAYS, "Cannot read packet from driver !n");
  64.     return NULL;
  65.   }
  66.   return incoming_packet;
  67. }
  68. void IOHook::sendPacket(DiffPacket pkt, int len, int dst)
  69. {
  70.   DeviceList::iterator device_itr;
  71.   for (device_itr = out_devices_.begin();
  72.        device_itr != out_devices_.end(); ++device_itr){
  73.     (*device_itr)->sendPacket(pkt, len, dst);
  74.   }
  75. }
  76. void IOHook::addInFDS(fd_set *fds, int *max)
  77. {
  78.   DeviceList::iterator device_itr;
  79.   for (device_itr = in_devices_.begin();
  80.        device_itr != in_devices_.end(); ++device_itr){
  81.     (*device_itr)->addInFDS(fds, max);
  82.   }
  83. }
  84. int IOHook::checkInFDS(fd_set *fds)
  85. {
  86.   DeviceList::iterator device_itr;
  87.   int fd;
  88.   for (device_itr = in_devices_.begin();
  89.        device_itr != in_devices_.end(); ++device_itr){
  90.     fd = (*device_itr)->checkInFDS(fds);
  91.     if (fd != 0)
  92.       return fd;
  93.   }
  94.   return 0;
  95. }