src/ex_filter_frontend_net.cpp

   1
   2 #include <cstdlib>
   3 #include <iostream>
   4 #include <stdexcept>
   5
   6 #include <boost/program_options.hpp>
   7 namespace po = boost::program_options;
   8
   9 #include "config.hpp"
  10
  11 #include "filter_frontend_net.hpp"
  12
  13 #include "router.hpp"
  14 #include "session.hpp"
  15 #include "package.hpp"
  16
  17 class FilterInit: public yp2::Filter {
  18 public:
  19     void process(yp2::Package & package) const {
  20
  21         if (package.session().is_closed())
  22         {
  23             // std::cout << "Got Close.\n";
  24         }
  25
  26         Z_GDU *gdu = package.request().get();
  27         if (gdu)
  28         {
  29             ODR odr = odr_createmem(ODR_ENCODE);
  30             switch(gdu->which)
  31             {
  32             case Z_GDU_Z3950:
  33                 // std::cout << "Got PDU. Sending init response\n";
  34                 Z_APDU *apdu = zget_APDU(odr, Z_APDU_initResponse);
  35
  36                 apdu->u.initResponse->implementationName = "YP2/YAZ";
  37
  38                 package.response() = apdu;
  39                 break;
  40             case Z_GDU_HTTP_Request:
  41                 Z_GDU *gdu = z_get_HTTP_Response(odr, 200);
  42                 Z_HTTP_Response *http_res = gdu->u.HTTP_Response;
  43
  44                 z_HTTP_header_add(odr, &http_res->headers,
  45                                   "Content-Type", "text/plain");
  46
  47                 http_res->content_buf =
  48                     odr_strdup(odr, "Welcome to YP2");
  49                 http_res->content_len = strlen(http_res->content_buf);
  50
  51                 package.response() = gdu;
  52                 break;
  53             default:
  54                 break;
  55             }
  56             odr_destroy(odr);
  57         }
  58         return package.move();
  59     };
  60 };
  61
  62 int main(int argc, char **argv)
  63 {
  64     try
  65     {
  66         po::options_description desc("Allowed options");
  67         desc.add_options()
  68             ("help", "produce help message")
  69             ("duration", po::value<int>(),
  70              "number of seconds for server to exist")
  71             ;
  72
  73         po::variables_map vm;
  74         po::store(po::parse_command_line(argc, argv, desc), vm);
  75         po::notify(vm);
  76
  77         if (vm.count("help")) {
  78             std::cout << desc << "\n";
  79             return 1;
  80         }
  81
  82         {
  83             yp2::RouterChain router;
  84
  85             // put in frontend first
  86             yp2::FilterFrontendNet filter_front;
  87             filter_front.listen_address() = "tcp:@:9999";
  88
  89             // 0=no time, >0 timeout in seconds
  90             if (vm.count("duration")) {
  91                 filter_front.listen_duration() =
  92                     vm["duration"].as<int>();
  93             }
  94             router.rule(filter_front);
  95
  96             // put in a backend
  97             FilterInit filter_init;
  98             router.rule(filter_init);
  99
 100             yp2::Session session;
 101             yp2::Origin origin;
 102             yp2::Package pack(session, origin);
 103
 104             pack.router(router).move();
 105         }
 106     }
 107     catch ( ... ) {
 108         std::cerr << "unknown exception\n";
 109         std::exit(1);
 110     }
 111     std::exit(0);
 112 }
 113
 114 /*
 115  * Local variables:
 116  * c-basic-offset: 4
 117  * indent-tabs-mode: nil
 118  * End:
 119  * vim: shiftwidth=4 tabstop=8 expandtab
 120  */