senf/Socket/Protocols/Raw/PacketSocketHandle.cc

   1 // $Id$
   2 //
   3 // Copyright (C) 2006
   4 // Fraunhofer Institute for Open Communication Systems (FOKUS)
   5 // Competence Center NETwork research (NET), St. Augustin, GERMANY
   6 //     Stefan Bund <g0dil@berlios.de>
   7 //
   8 // This program is free software; you can redistribute it and/or modify
   9 // it under the terms of the GNU General Public License as published by
  10 // the Free Software Foundation; either version 2 of the License, or
  11 // (at your option) any later version.
  12 //
  13 // This program is distributed in the hope that it will be useful,
  14 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 // GNU General Public License for more details.
  17 //
  18 // You should have received a copy of the GNU General Public License
  19 // along with this program; if not, write to the
  20 // Free Software Foundation, Inc.,
  21 // 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  22
  23 /** \file
  24     \brief PacketSocketProtocol and PacketSocketHandle non-inline non-template implementation
  25  */
  26
  27 #include "PacketSocketHandle.hh"
  28 //#include "PacketSocketHandle.ih"
  29
  30 // Custom includes
  31 #include <sys/types.h>
  32 #include <sys/socket.h>
  33 #include <netpacket/packet.h>
  34 #include <net/ethernet.h>
  35 #include <netinet/in.h>
  36 #include <net/if.h>
  37 #include <errno.h>
  38
  39 //#include "PacketSocketHandle.mpp"
  40 #define prefix_
  41 ///////////////////////////////cc.p////////////////////////////////////////
  42
  43 prefix_ void senf::PacketSocketProtocol::init_client(SocketType type, int protocol)
  44     const
  45 {
  46     int socktype = SOCK_RAW;
  47     if (type == DatagramSocket)
  48         socktype = SOCK_DGRAM;
  49     if (protocol == -1)
  50         protocol = ETH_P_ALL;
  51     int sock = ::socket(PF_PACKET, socktype, htons(protocol));
  52     if (sock < 0)
  53         SENF_THROW_SYSTEM_EXCEPTION("::socket(...) failed.");
  54     fd(sock);
  55 }
  56
  57 prefix_ unsigned senf::PacketSocketProtocol::available()
  58     const
  59 {
  60     if (! fh().readable())
  61         return 0;
  62     ssize_t l = ::recv(fd(),0,0,MSG_PEEK | MSG_TRUNC);
  63     if (l < 0)
  64         SENF_THROW_SYSTEM_EXCEPTION("::recv(socket_fd) failed.");
  65     return l;
  66 }
  67
  68 prefix_ bool senf::PacketSocketProtocol::eof()
  69     const
  70 {
  71     return false;
  72 }
  73
  74 namespace {
  75
  76     void do_mc(int fd, std::string const & interface, senf::MACAddress address, bool add)
  77     {
  78         struct packet_mreq mreq;
  79         ::memset(&mreq, 0, sizeof(mreq));
  80         mreq.mr_ifindex = ::if_nametoindex(interface.c_str());
  81         if (mreq.mr_ifindex == 0)
  82             throw senf::SystemException(EINVAL);
  83         mreq.mr_type = PACKET_MR_MULTICAST;
  84         mreq.mr_alen = 6;
  85         std::copy(address.begin(), address.end(), &mreq.mr_address[0]);
  86         if (::setsockopt(fd, SOL_PACKET,
  87                         add ? PACKET_ADD_MEMBERSHIP : PACKET_DROP_MEMBERSHIP,
  88                         &mreq, sizeof(mreq)) < 0)
  89             throw senf::SystemException();
  90     }
  91 }
  92
  93 prefix_ void senf::PacketSocketProtocol::mcAdd(std::string const & interface,
  94                                          MACAddress const & address)
  95     const
  96 {
  97     do_mc(fd(), interface, address, true);
  98 }
  99
 100 prefix_ void senf::PacketSocketProtocol::mcDrop(std::string const & interface,
 101                                           MACAddress const & address)
 102     const
 103 {
 104     do_mc(fd(), interface, address, false);
 105 }
 106
 107 prefix_ void senf::PacketSocketProtocol::promisc(std::string const & interface, bool mode)
 108     const
 109 {
 110     struct packet_mreq mreq;
 111     ::memset(&mreq, 0, sizeof(mreq));
 112     mreq.mr_ifindex = ::if_nametoindex(interface.c_str());
 113     if (mreq.mr_ifindex == 0)
 114         throw senf::SystemException(EINVAL);
 115     mreq.mr_type = PACKET_MR_PROMISC;
 116     if (::setsockopt(fd(), SOL_PACKET,
 117                      mode ? PACKET_ADD_MEMBERSHIP : PACKET_DROP_MEMBERSHIP,
 118                      &mreq, sizeof(mreq)) < 0)
 119         throw senf::SystemException();
 120 }
 121
 122 ///////////////////////////////cc.e////////////////////////////////////////
 123 #undef prefix_
 124 //#include "PacketSocketHandle.mpp"
 125
 126 \f
 127 // Local Variables:
 128 // mode: c++
 129 // fill-column: 100
 130 // c-file-style: "senf"
 131 // indent-tabs-mode: nil
 132 // ispell-local-dictionary: "american"
 133 // compile-command: "scons -u test"
 134 // comment-column: 40
 135 // End: