src/ex_filter_frontend_net.cpp

   1 /* This file is part of Metaproxy.
   2    Copyright (C) 2005-2011 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
  21 #include <cstdlib>
  22 #include <iostream>
  23 #include <stdexcept>
  24
  25 #include <yaz/options.h>
  26 #include <metaproxy/util.hpp>
  27 #include "filter_frontend_net.hpp"
  28 #include "filter_z3950_client.hpp"
  29 #include "filter_virt_db.hpp"
  30 #include "filter_session_shared.hpp"
  31 #include "filter_log.hpp"
  32
  33 #include "router_chain.hpp"
  34 #include <metaproxy/package.hpp>
  35
  36 namespace mp = metaproxy_1;
  37
  38 class HTTPFilter: public mp::filter::Base {
  39 public:
  40     void process(mp::Package & package) const {
  41         if (package.session().is_closed())
  42         {
  43             // std::cout << "Got Close.\n";
  44         }
  45
  46         Z_GDU *gdu = package.request().get();
  47         if (gdu && gdu->which == Z_GDU_HTTP_Request)
  48         {
  49             mp::odr odr;
  50             Z_GDU *gdu = z_get_HTTP_Response(odr, 200);
  51             Z_HTTP_Response *http_res = gdu->u.HTTP_Response;
  52
  53             z_HTTP_header_add(odr, &http_res->headers,
  54                               "Content-Type", "text/plain");
  55
  56             http_res->content_buf =
  57                 odr_strdup(odr, "Welcome to Metaproxy");
  58             http_res->content_len = strlen(http_res->content_buf);
  59
  60             package.response() = gdu;
  61         }
  62         return package.move();
  63     };
  64 };
  65
  66 int main(int argc, char **argv)
  67 {
  68     try
  69     {
  70         std::vector<std::string> ports;
  71         int duration = -1;
  72         int ret;
  73         char *arg;
  74
  75         while ((ret = options("h{help}d{duration}:p{port}:",
  76                               argv, argc, &arg)) != -2)
  77         {
  78             switch(ret)
  79             {
  80             case -1:
  81                 std::cerr << "bad option " << arg << std::endl;
  82             case 'h':
  83                 std::cerr << "ex_filter_frontend_net\n"
  84                     " -h|--help       help\n"
  85                     " -d|--duration n duration\n"
  86                     " -p|--port n     port number\n"
  87                           << std::endl;
  88                 break;
  89             case 'p':
  90                 ports.push_back(arg);
  91                 break;
  92             case 'd':
  93                 duration = atoi(arg);
  94                 break;
  95             }
  96         }
  97         {
  98             for (size_t i = 0; i<ports.size(); i++)
  99                 std::cout << "port " << i << " " << ports[i] << "\n";
 100
 101             mp::RouterChain router;
 102
 103             // put frontend filter in router
 104             mp::filter::FrontendNet filter_front;
 105             filter_front.set_ports(ports);
 106
 107             // 0=no time, >0 timeout in seconds
 108             if (duration != -1)
 109                 filter_front.set_listen_duration(duration);
 110
 111             router.append(filter_front);
 112
 113             // put log filter in router
 114             mp::filter::Log filter_log_front("FRONT");
 115             router.append(filter_log_front);
 116
 117             // put Virt db filter in router
 118             mp::filter::VirtualDB filter_virt_db;
 119             filter_virt_db.add_map_db2target("gils", "indexdata.dk/gils",
 120                                             "");
 121             filter_virt_db.add_map_db2target("Default", "localhost:9999/Default",
 122                                             "");
 123             filter_virt_db.add_map_db2target("2", "localhost:9999/Slow", "");
 124             router.append(filter_virt_db);
 125
 126             mp::filter::SessionShared filter_session_shared;
 127             //router.append(filter_session_shared);
 128
 129             mp::filter::Log filter_log_back("BACK");
 130             router.append(filter_log_back);
 131
 132             // put HTTP backend filter in router
 133             HTTPFilter filter_init;
 134             router.append(filter_init);
 135
 136             // put Z39.50 backend filter in router
 137             mp::filter::Z3950Client z3950_client;
 138             router.append(z3950_client);
 139
 140             mp::Session session;
 141             mp::Origin origin;
 142             mp::Package pack(session, origin);
 143
 144             pack.router(router).move();
 145         }
 146     }
 147     catch ( ... ) {
 148         std::cerr << "unknown exception\n";
 149         std::exit(1);
 150     }
 151     std::exit(0);
 152 }
 153
 154 /*
 155  * Local variables:
 156  * c-basic-offset: 4
 157  * c-file-style: "Stroustrup"
 158  * indent-tabs-mode: nil
 159  * End:
 160  * vim: shiftwidth=4 tabstop=8 expandtab
 161  */
 162