modules/pipebackend/coprocess.hh

   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 #ifndef PDNS_COPROCESS_HH
  23 #define PDNS_COPROCESS_HH
  24
  25 #include <iostream>
  26 #include <stdio.h>
  27 #include <string>
  28
  29 #include "pdns/namespaces.hh"
  30
  31 class CoRemote
  32 {
  33 public:
  34   virtual ~CoRemote() {}
  35   virtual void sendReceive(const string &send, string &receive) = 0;
  36   virtual void receive(string &rcv) = 0;
  37   virtual void send(const string &send) = 0;
  38
  39 };
  40
  41 class CoProcess : public CoRemote
  42 {
  43 public:
  44   CoProcess(const string &command,int timeout=0, int infd=0, int outfd=1);
  45   ~CoProcess();
  46   void sendReceive(const string &send, string &receive);
  47   void receive(string &rcv);
  48   void send(const string &send);
  49 private:
  50   void launch(const char **argv, int timeout=0, int infd=0, int outfd=1);
  51   void checkStatus();
  52   int d_fd1[2], d_fd2[2];
  53   int d_pid;
  54   int d_infd;
  55   int d_outfd;
  56   int d_timeout;
  57   FILE *d_fp;
  58 };
  59
  60 class UnixRemote : public CoRemote
  61 {
  62 public:
  63   UnixRemote(const string &path, int timeout=0);
  64   ~UnixRemote();
  65   void sendReceive(const string &send, string &receive);
  66   void receive(string &rcv);
  67   void send(const string &send);
  68 private:
  69   int d_fd;
  70   FILE *d_fp;
  71 };
  72 bool isUnixSocket(const string& fname);
  73 #endif