PPI/SocketSink.cc

   1 // $Id: SocketSink.cc 661 2008-02-05 09:53:54Z dw6 $
   2 //
   3 // Copyright (C) 2007
   4 // Fraunhofer Institute for Open Communication Systems (FOKUS)
   5 // Competence Center NETwork research (NET), St. Augustin, GERMANY
   6 //    David Wagner <dw6@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 SocketSink non-inline non-template implementation */
  25
  26 // Custom includes
  27 #include "SocketSink.hh"
  28 #include "../Socket/ClientSocketHandle.hh"
  29
  30 #define prefix_
  31
  32 prefix_ senf::ppi::IPv4SourceForcingDgramWriter::IPv4SourceForcingDgramWriter(){
  33     source_ = senf::INet4Address::from_string("0.0.0.0");
  34     destination_ = senf::INet4Address::from_string("0.0.0.0");
  35     protocolId_ = 0;
  36 }
  37
  38 prefix_ senf::ppi::IPv4SourceForcingDgramWriter::IPv4SourceForcingDgramWriter(senf::INet4Address sourceAddr, senf::INet4SocketAddress destAddr){
  39     source(sourceAddr);
  40     destination(destAddr);
  41 }
  42
  43 prefix_ void senf::ppi::IPv4SourceForcingDgramWriter::source(senf::INet4Address & source){
  44     source_ = source;
  45 }
  46 prefix_ void senf::ppi::IPv4SourceForcingDgramWriter::destination(senf::INet4SocketAddress & dest){
  47     destination_ = dest.address();
  48     protocolId_ = dest.port();
  49 }
  50
  51 prefix_ void senf::ppi::IPv4SourceForcingDgramWriter::operator()(Handle handle,
  52                                                                  Packet const & packet){
  53     sendtoandfrom(
  54             handle.fd(),
  55             reinterpret_cast<void*> (&*packet.data().begin()),
  56             packet.size(),
  57             reinterpret_cast<in_addr*> (&destination_),
  58             protocolId_,
  59             reinterpret_cast<in_addr*> (&source_));
  60 }
  61
  62 prefix_ int senf::ppi::IPv4SourceForcingDgramWriter::sendtoandfrom(
  63         int sock,
  64         const void *data,
  65         size_t dataLen,
  66         const in_addr *dst,
  67         int port,
  68         const in_addr *src)
  69 {
  70     uint8_t cbuf[CMSG_SPACE(sizeof(struct in_pktinfo))];
  71     struct cmsghdr *c = (struct cmsghdr *)cbuf;
  72     struct in_pktinfo *pi;
  73     struct iovec iov;
  74     struct msghdr h;
  75
  76     c->cmsg_len = CMSG_LEN(sizeof(struct in_pktinfo));
  77     c->cmsg_level = IPPROTO_IP;
  78     c->cmsg_type = IP_PKTINFO;
  79
  80     pi = (struct in_pktinfo *)CMSG_DATA(c);
  81     pi->ipi_ifindex = 0;
  82     memcpy(&pi->ipi_addr, src, 16);
  83
  84     iov.iov_base = (void *)data;
  85     iov.iov_len = dataLen;
  86
  87     sockaddr_in dstpeer;
  88     memset(&dstpeer, 0, sizeof(dstpeer));
  89     dstpeer.sin_family = AF_INET;
  90     dstpeer.sin_addr = *dst;
  91     dstpeer.sin_port = htons(port);
  92
  93     memset(&h, 0, sizeof(h));
  94     h.msg_name = (struct sockaddr *)&dstpeer;
  95     h.msg_namelen = sizeof(dstpeer);
  96     h.msg_iov = &iov;
  97     h.msg_iovlen = 1;
  98     h.msg_control = c;
  99     h.msg_controllen = sizeof(cbuf);
 100
 101     return sendmsg(sock, &h, 0);
 102 }
 103
 104 prefix_ senf::ppi::IPv6SourceForcingDgramWriter::IPv6SourceForcingDgramWriter(){
 105     source_ = senf::INet6Address::from_string("0::0");
 106     destination_ = senf::INet6Address::from_string("0::0");
 107     protocolId_ = 0;
 108 }
 109
 110 prefix_ senf::ppi::IPv6SourceForcingDgramWriter::IPv6SourceForcingDgramWriter(senf::INet6Address sourceAddr, senf::INet6SocketAddress destAddr){
 111     source(sourceAddr);
 112     destination(destAddr);
 113 }
 114
 115 prefix_ void senf::ppi::IPv6SourceForcingDgramWriter::source(senf::INet6Address & source){
 116     source_ = source;
 117 }
 118 prefix_ void senf::ppi::IPv6SourceForcingDgramWriter::destination(senf::INet6SocketAddress & dest){
 119     destination_ = dest.address();
 120     protocolId_ = dest.port();
 121 }
 122
 123 prefix_ void senf::ppi::IPv6SourceForcingDgramWriter::operator()(Handle handle,
 124                                                                  Packet const & packet){
 125     sendtoandfrom(
 126             handle.fd(),
 127             reinterpret_cast<void*> (&*packet.data().begin()),
 128             packet.size(),
 129             reinterpret_cast<in6_addr*> (&destination_),
 130             protocolId_,
 131             reinterpret_cast<in6_addr*> (&source_));
 132 }
 133
 134 prefix_ int senf::ppi::IPv6SourceForcingDgramWriter::sendtoandfrom(
 135         int sock,
 136         const void *data,
 137         size_t dataLen,
 138         const in6_addr *dst,
 139         int port,
 140         const in6_addr *src)
 141 {
 142     uint8_t cbuf[CMSG_SPACE(sizeof(struct in6_pktinfo))];
 143     struct cmsghdr *c = (struct cmsghdr *)cbuf;
 144     struct in6_pktinfo *pi;
 145     struct iovec iov;
 146     struct msghdr h;
 147
 148     c->cmsg_len = CMSG_LEN(sizeof(struct in6_pktinfo));
 149     c->cmsg_level = IPPROTO_IPV6;
 150     c->cmsg_type = IPV6_PKTINFO;
 151
 152     pi = (struct in6_pktinfo *)CMSG_DATA(c);
 153     pi->ipi6_ifindex = 0;
 154     memcpy(&pi->ipi6_addr, src, 16);
 155
 156     iov.iov_base = (void *)data;
 157     iov.iov_len = dataLen;
 158
 159     sockaddr_in6 dstpeer;
 160     memset(&dstpeer, 0, sizeof(dstpeer));
 161     dstpeer.sin6_family = AF_INET6;
 162     dstpeer.sin6_addr = *dst;
 163     dstpeer.sin6_port = htons(port);
 164
 165     memset(&h, 0, sizeof(h));
 166     h.msg_name = (struct sockaddr *)&dstpeer;
 167     h.msg_namelen = sizeof(dstpeer);
 168     h.msg_iov = &iov;
 169     h.msg_iovlen = 1;
 170     h.msg_control = c;
 171     h.msg_controllen = sizeof(cbuf);
 172
 173     return sendmsg(sock, &h, 0);
 174 }
 175
 176 #undef prefix_