src/filter_http_client.cpp

   1 /* This file is part of Metaproxy.
   2    Copyright (C) 2005-2013 Index Data
   3
   4 Metaproxy is free software; you can redistribute it and/or modify it under
   5 the terms of the GNU General Public License as published by the Free
   6 Software Foundation; either version 2, or (at your option) any later
   7 version.
   8
   9 Metaproxy is distributed in the hope that it will be useful, but WITHOUT ANY
  10 WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11 FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12 for more details.
  13
  14 You should have received a copy of the GNU General Public License
  15 along with this program; if not, write to the Free Software
  16 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  17 */
  18
  19 #include "config.hpp"
  20 #include <metaproxy/filter.hpp>
  21 #include <metaproxy/package.hpp>
  22 #include <metaproxy/util.hpp>
  23 #include <yaz/url.h>
  24 #include "filter_http_client.hpp"
  25
  26 #include <yaz/zgdu.h>
  27 #include <yaz/log.h>
  28
  29 #include <boost/thread/mutex.hpp>
  30
  31 #include <list>
  32 #include <map>
  33 #include <iostream>
  34
  35 #if HAVE_SYS_TYPES_H
  36 #include <sys/types.h>
  37 #endif
  38
  39 #if HAVE_SYS_STAT_H
  40 #include <sys/stat.h>
  41 #endif
  42
  43 namespace mp = metaproxy_1;
  44 namespace yf = mp::filter;
  45
  46 namespace metaproxy_1 {
  47     namespace filter {
  48         class HTTPClient::Rep {
  49             friend class HTTPClient;
  50             void proxy(mp::Package &package);
  51             std::string proxy_host;
  52         };
  53     }
  54 }
  55
  56 yf::HTTPClient::HTTPClient() : m_p(new Rep)
  57 {
  58 }
  59
  60 yf::HTTPClient::~HTTPClient()
  61 {
  62 }
  63
  64 void yf::HTTPClient::Rep::proxy(mp::Package &package)
  65 {
  66     Z_GDU *req_gdu = package.request().get();
  67     if (req_gdu && req_gdu->which == Z_GDU_HTTP_Request)
  68     {
  69         Z_HTTP_Request *hreq = req_gdu->u.HTTP_Request;
  70         Z_GDU *res_gdu = 0;
  71         mp::odr o;
  72         yaz_url_t yaz_url = yaz_url_create();
  73
  74         if (proxy_host.length())
  75             yaz_url_set_proxy(yaz_url, proxy_host.c_str());
  76         Z_HTTP_Response *http_response =
  77             yaz_url_exec(yaz_url, hreq->path, hreq->method,
  78                          hreq->headers, hreq->content_buf,
  79                          hreq->content_len);
  80         if (http_response)
  81         {
  82             res_gdu = o.create_HTTP_Response(package.session(), hreq, 200);
  83             res_gdu->u.HTTP_Response = http_response;
  84         }
  85         else
  86         {
  87             res_gdu = o.create_HTTP_Response(package.session(), hreq, 404);
  88         }
  89         package.response() = res_gdu;
  90         yaz_url_destroy(yaz_url);
  91     }
  92     else
  93         package.move();
  94 }
  95
  96 void yf::HTTPClient::process(mp::Package &package) const
  97 {
  98     Z_GDU *gdu = package.request().get();
  99     if (gdu && gdu->which == Z_GDU_HTTP_Request)
 100         m_p->proxy(package);
 101     else
 102         package.move();
 103 }
 104
 105 void mp::filter::HTTPClient::configure(const xmlNode * ptr, bool test_only,
 106                                        const char *path)
 107 {
 108     for (ptr = ptr->children; ptr; ptr = ptr->next)
 109     {
 110         if (ptr->type != XML_ELEMENT_NODE)
 111             continue;
 112         else if (!strcmp((const char *) ptr->name, "proxy"))
 113         {
 114             m_p->proxy_host = mp::xml::get_text(ptr);
 115         }
 116         else
 117         {
 118             throw mp::filter::FilterException
 119                 ("Bad element "
 120                  + std::string((const char *) ptr->name)
 121                  + " in virt_db filter");
 122         }
 123     }
 124 }
 125
 126 static mp::filter::Base* filter_creator()
 127 {
 128     return new mp::filter::HTTPClient;
 129 }
 130
 131 extern "C" {
 132     struct metaproxy_1_filter_struct metaproxy_1_filter_http_client = {
 133         0,
 134         "http_client",
 135         filter_creator
 136     };
 137 }
 138
 139
 140 /*
 141  * Local variables:
 142  * c-basic-offset: 4
 143  * c-file-style: "Stroustrup"
 144  * indent-tabs-mode: nil
 145  * End:
 146  * vim: shiftwidth=4 tabstop=8 expandtab
 147  */
 148