pdns/dynmessenger.cc

   1 /*
   2  * This file is part of PowerDNS or dnsdist.
   3  * Copyright -- PowerDNS.COM B.V. and its contributors
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of version 2 of the GNU General Public License as
   7  * published by the Free Software Foundation.
   8  *
   9  * In addition, for the avoidance of any doubt, permission is granted to
  10  * link this program with OpenSSL and to (re)distribute the binaries
  11  * produced as the result of such linking.
  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 Free Software
  20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  21  */
  22 #ifdef HAVE_CONFIG_H
  23 #include "config.h"
  24 #endif
  25 #include "dynmessenger.hh"
  26 #include <cstdio>
  27 #include "utility.hh"
  28 #include <cstdlib>
  29 #include <cstring>
  30 #include <iostream>
  31 #include <sys/types.h>
  32 #include <sys/stat.h>
  33
  34 DynMessenger::DynMessenger(const string &fname,
  35     int timeout_sec,
  36     int timeout_usec)
  37 {
  38   d_s=socket(AF_UNIX,SOCK_STREAM,0);
  39   setCloseOnExec(d_s);
  40
  41   if(d_s<0) {
  42     throw PDNSException(string("socket")+stringerror());
  43   }
  44
  45   try {
  46     if(makeUNsockaddr(fname, &d_remote))
  47       throw PDNSException("Unable to connect to remote '"+fname+"': Path is not a valid UNIX socket path.");
  48
  49     struct timeval timeout;
  50     timeout.tv_sec = timeout_sec;
  51     timeout.tv_usec = timeout_usec;
  52
  53     if (setsockopt (d_s, SOL_SOCKET, SO_RCVTIMEO, (char *)&timeout, sizeof(timeout)) < 0)
  54       throw PDNSException("Unable to set SO_RCVTIMEO option on socket: " + stringerror());
  55
  56     if (setsockopt (d_s, SOL_SOCKET, SO_SNDTIMEO, (char *)&timeout, sizeof(timeout)) < 0)
  57       throw PDNSException("Unable to set SO_SNDTIMEO option on socket: " + stringerror());
  58
  59     int ret = Utility::timed_connect(d_s,(sockaddr*)&d_remote,sizeof(d_remote), timeout_sec, timeout_usec);
  60
  61     if (ret == 0)
  62       throw TimeoutException("Unable to connect to remote '"+fname+"': "+stringerror());
  63     else if (ret < 0)
  64       throw PDNSException("Unable to connect to remote '"+fname+"': "+stringerror());
  65
  66   } catch(...) {
  67     close(d_s);
  68     d_s=-1;
  69     throw;
  70   }
  71 }
  72
  73 DynMessenger::DynMessenger(const ComboAddress& remote,
  74     const string &secret,
  75     int timeout_sec,
  76     int timeout_usec)
  77 {
  78   d_s=socket(AF_INET, SOCK_STREAM,0);
  79   setCloseOnExec(d_s);
  80
  81   if(d_s<0) {
  82     throw PDNSException(string("socket")+stringerror());
  83   }
  84
  85   try {
  86     struct timeval timeout;
  87     timeout.tv_sec = timeout_sec;
  88     timeout.tv_usec = timeout_usec;
  89
  90     if (setsockopt (d_s, SOL_SOCKET, SO_RCVTIMEO, (char *)&timeout, sizeof(timeout)) < 0)
  91       throw PDNSException("Unable to set SO_RCVTIMEO option on socket: " + stringerror());
  92
  93     if (setsockopt (d_s, SOL_SOCKET, SO_SNDTIMEO, (char *)&timeout, sizeof(timeout)) < 0)
  94       throw PDNSException("Unable to set SO_SNDTIMEO option on socket: " + stringerror());
  95
  96     int ret = Utility::timed_connect(d_s, (sockaddr*)&remote, remote.getSocklen(), timeout_sec, timeout_usec);
  97
  98     if (ret == 0)
  99       throw TimeoutException("Unable to connect to remote '"+remote.toStringWithPort()+"': "+stringerror());
 100     else if (ret < 0)
 101       throw PDNSException("Unable to connect to remote '"+remote.toStringWithPort()+"': "+stringerror());
 102
 103     string login=secret+"\n";
 104     writen2(d_s, login);
 105   } catch(...) {
 106     close(d_s);
 107     d_s=-1;
 108     throw;
 109   }
 110 }
 111
 112 DynMessenger::~DynMessenger()
 113 {
 114   if (d_s > 0)
 115     close(d_s);
 116 }
 117
 118 int DynMessenger::send(const string &msg) const
 119 {
 120   try {
 121     writen2(d_s, msg+"\n");
 122     return 0;
 123   } catch(std::runtime_error& e) {
 124     if (errno == EAGAIN)
 125       throw TimeoutException("Error from remote in send(): " + string(e.what()));
 126     else
 127       throw PDNSException("Error from remote in send(): " + string(e.what()));
 128   }
 129 }
 130
 131 string DynMessenger::receive() const
 132 {
 133   char buffer[1500];
 134
 135   int retlen;
 136   string answer;
 137   for(;;) {
 138     retlen=recv(d_s,buffer,sizeof(buffer),0);
 139     if(retlen<0) {
 140       if (errno == EAGAIN)
 141         throw TimeoutException("Error from remote in receive(): " + stringerror());
 142       else
 143         throw PDNSException("Error from remote in receive(): " + stringerror());
 144     }
 145
 146     answer.append(buffer,retlen);
 147     if (retlen == 0)
 148       break;
 149   }
 150
 151   return answer;
 152 }
 153