Socket/Protocols/Raw/PacketSocketHandle.cc

   1 // $Id$
   2 //
   3 // Copyright (C) 2006
   4 // Fraunhofer Institut fuer offene Kommunikationssysteme (FOKUS)
   5 // Kompetenzzentrum fuer Satelitenkommunikation (SatCom)
   6 //     Stefan Bund <stefan.bund@fokus.fraunhofer.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 PacketProtocol 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::PacketProtocol::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         throwErrno();
  54     body().fd(sock);
  55 }
  56
  57 prefix_ std::auto_ptr<senf::SocketProtocol> senf::PacketProtocol::clone()
  58     const
  59 {
  60     return std::auto_ptr<SocketProtocol>(new PacketProtocol());
  61 }
  62
  63 prefix_ unsigned senf::PacketProtocol::available()
  64     const
  65 {
  66     if (! body().readable())
  67         return 0;
  68     ssize_t l = ::recv(body().fd(),0,0,MSG_PEEK | MSG_TRUNC);
  69     if (l < 0)
  70         throwErrno();
  71     return l;
  72 }
  73
  74 prefix_ bool senf::PacketProtocol::eof()
  75     const
  76 {
  77     return false;
  78 }
  79
  80 namespace {
  81
  82     void do_mc(int fd, std::string const & interface, senf::MACAddress address, bool add)
  83     {
  84         struct packet_mreq mreq;
  85         mreq.mr_ifindex = ::if_nametoindex(interface.c_str());
  86         if (mreq.mr_ifindex == 0)
  87             senf::throwErrno(EINVAL);
  88         mreq.mr_type = PACKET_MR_MULTICAST;
  89         mreq.mr_alen = 6;
  90         std::copy(address.begin(), address.end(), &mreq.mr_address[0]);
  91         if (::setsockopt(fd, SOL_PACKET,
  92                          add ? PACKET_ADD_MEMBERSHIP : PACKET_DROP_MEMBERSHIP,
  93                          &mreq, sizeof(mreq)) < 0)
  94             senf::throwErrno();
  95     }
  96
  97 }
  98
  99 prefix_ void senf::PacketProtocol::mcAdd(std::string const & interface,
 100                                          MACAddress const & address)
 101     const
 102 {
 103     do_mc(body().fd(),interface,address,true);
 104 }
 105
 106 prefix_ void senf::PacketProtocol::mcDrop(std::string const & interface,
 107                                           MACAddress const & address)
 108     const
 109 {
 110     do_mc(body().fd(),interface,address,false);
 111 }
 112
 113 ///////////////////////////////cc.e////////////////////////////////////////
 114 #undef prefix_
 115 //#include "PacketSocketHandle.mpp"
 116
 117 \f
 118 // Local Variables:
 119 // mode: c++
 120 // fill-column: 100
 121 // c-file-style: "senf"
 122 // indent-tabs-mode: nil
 123 // ispell-local-dictionary: "american"
 124 // compile-command: "scons -u test"
 125 // comment-column: 40
 126 // End: