X-Git-Url: http://git.indexdata.com/?a=blobdiff_plain;f=src%2Ffilter_frontend_net.cpp;h=2e2ada228645ba4bf1371bf8f92eaa870bc15bfa;hb=1fe3038c15740cf21a2eb7b00edad8d216384c2e;hp=ed9a4759025b9ed14b6482e13a5582bf8384ec63;hpb=57c1c4822f1de4d983c1aee3bc6059ad2600d991;p=metaproxy-moved-to-github.git diff --git a/src/filter_frontend_net.cpp b/src/filter_frontend_net.cpp index ed9a475..2e2ada2 100644 --- a/src/filter_frontend_net.cpp +++ b/src/filter_frontend_net.cpp @@ -1,270 +1,857 @@ +/* This file is part of Metaproxy. + Copyright (C) Index Data +Metaproxy is free software; you can redistribute it and/or modify it under +the terms of the GNU General Public License as published by the Free +Software Foundation; either version 2, or (at your option) any later +version. + +Metaproxy is distributed in the hope that it will be useful, but WITHOUT ANY +WARRANTY; without even the implied warranty of MERCHANTABILITY or +FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License +for more details. + +You should have received a copy of the GNU General Public License +along with this program; if not, write to the Free Software +Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA +*/ #include "config.hpp" -#include "filter.hpp" -#include "router.hpp" -#include "package.hpp" +#if HAVE_GETRLIMIT +#include +#endif +#include +#include +#include +#include "pipe.hpp" +#include +#include #include "thread_pool_observer.hpp" #include "filter_frontend_net.hpp" -#include -#include -#include +#include +#include +#include +#include +#include #include +#include +#include "gduutil.hpp" +#include #include -using namespace yp2; - -class P2_Session : public yazpp_1::Z_Assoc { -public: - ~P2_Session(); - P2_Session(yazpp_1::IPDU_Observable *the_PDU_Observable, - ThreadPoolSocketObserver *m_my_thread, - const Package *package); - int m_no_requests; -private: - yazpp_1::IPDU_Observer* sessionNotify( - yazpp_1::IPDU_Observable *the_PDU_Observable, - int fd); - void recv_GDU(Z_GDU *apdu, int len); - - void failNotify(); - void timeoutNotify(); - void connectNotify(); -private: - ThreadPoolSocketObserver *m_my_thread; - Session m_session; - Origin m_origin; - bool m_delete_flag; - const Package *m_package; -}; - - -class ThreadPoolPackage : public IThreadPoolMsg { -public: - ThreadPoolPackage(Package *package, P2_Session *ses) : - m_session(ses), m_package(package) { }; - ~ThreadPoolPackage(); - IThreadPoolMsg *handle(); - void result(); - -private: - P2_Session *m_session; - Package *m_package; - -}; - -ThreadPoolPackage::~ThreadPoolPackage() +namespace mp = metaproxy_1; +namespace yf = metaproxy_1::filter; + +namespace metaproxy_1 { + namespace filter { + class FrontendNet::Port { + friend class Rep; + friend class FrontendNet; + std::string port; + std::string route; + std::string cert_fname; + int max_recv_bytes; + }; + class FrontendNet::Rep { + friend class FrontendNet; + + int m_no_threads; + int m_max_threads; + int m_stack_size; + std::vector m_ports; + int m_listen_duration; + int m_session_timeout; + int m_connect_max; + int m_http_req_max; + std::string m_msg_config; + std::string m_stat_req; + yazpp_1::SocketManager mySocketManager; + ZAssocServer **az; + yazpp_1::PDU_Assoc **pdu; + int m_duration_freq[22]; + double m_duration_lim[22]; + double m_duration_max; + double m_duration_min; + double m_duration_total; + int m_stop_signo; + public: + Rep(); + ~Rep(); + }; + class FrontendNet::My_Timer_Thread : public yazpp_1::ISocketObserver { + private: + yazpp_1::ISocketObservable *m_obs; + Pipe m_pipe; + bool m_timeout; + public: + My_Timer_Thread(yazpp_1::ISocketObservable *obs, int duration); + void socketNotify(int event); + bool timeout(); + }; + class FrontendNet::ZAssocChild : public yazpp_1::Z_Assoc { + public: + ~ZAssocChild(); + ZAssocChild(yazpp_1::IPDU_Observable *the_PDU_Observable, + mp::ThreadPoolSocketObserver *m_thread_pool_observer, + const mp::Package *package, + Port *port, + Rep *rep, + yazpp_1::LimitConnect &limit_connect); + int m_no_requests; + Port *m_port; + private: + yazpp_1::IPDU_Observer* sessionNotify( + yazpp_1::IPDU_Observable *the_PDU_Observable, + int fd); + void recv_GDU(Z_GDU *apdu, int len); + void report(Z_HTTP_Request *hreq); + void failNotify(); + void timeoutNotify(); + void connectNotify(); + private: + mp::ThreadPoolSocketObserver *m_thread_pool_observer; + mp::Session m_session; + mp::Origin m_origin; + bool m_delete_flag; + const mp::Package *m_package; + Rep *m_p; + yazpp_1::LimitConnect &m_limit_connect; + }; + class FrontendNet::ThreadPoolPackage : public mp::IThreadPoolMsg { + public: + ThreadPoolPackage(mp::Package *package, + yf::FrontendNet::ZAssocChild *ses, + Rep *rep); + ~ThreadPoolPackage(); + IThreadPoolMsg *handle(); + void result(const char *t_info); + bool cleanup(void *info); + private: + yaz_timing_t timer; + ZAssocChild *m_assoc_child; + mp::Package *m_package; + Rep *m_p; + }; + class FrontendNet::ZAssocServer : public yazpp_1::Z_Assoc { + public: + ~ZAssocServer(); + ZAssocServer(yazpp_1::IPDU_Observable *PDU_Observable, + FrontendNet::Port *port, + Rep *rep); + void set_package(const mp::Package *package); + void set_thread_pool(ThreadPoolSocketObserver *observer); + private: + yazpp_1::IPDU_Observer* sessionNotify( + yazpp_1::IPDU_Observable *the_PDU_Observable, + int fd); + void recv_GDU(Z_GDU *apdu, int len); + + void failNotify(); + void timeoutNotify(); + void connectNotify(); + private: + mp::ThreadPoolSocketObserver *m_thread_pool_observer; + const mp::Package *m_package; + yazpp_1::LimitConnect limit_connect; + Port *m_port; + Rep *m_p; + }; + } +} + +yf::FrontendNet::ThreadPoolPackage::ThreadPoolPackage(mp::Package *package, + ZAssocChild *ses, + Rep *rep) : + m_assoc_child(ses), m_package(package), m_p(rep) +{ + timer = yaz_timing_create(); +} + +yf::FrontendNet::ThreadPoolPackage::~ThreadPoolPackage() { + yaz_timing_destroy(&timer); // timer may be NULL delete m_package; } -void ThreadPoolPackage::result() +bool yf::FrontendNet::ThreadPoolPackage::cleanup(void *info) { - m_session->m_no_requests--; + mp::Session *ses = (mp::Session *) info; + + return *ses == m_package->session(); +} + +void yf::FrontendNet::ThreadPoolPackage::result(const char *t_info) +{ + m_assoc_child->m_no_requests--; yazpp_1::GDU *gdu = &m_package->response(); + if (gdu->get()) { int len; - m_session->send_GDU(gdu->get(), &len); + m_assoc_child->send_GDU(gdu->get(), &len); + + yaz_timing_stop(timer); + double duration = yaz_timing_get_real(timer); + + size_t ent = 0; + while (m_p->m_duration_lim[ent] != 0.0 && duration > m_p->m_duration_lim[ent]) + ent++; + m_p->m_duration_freq[ent]++; + + m_p->m_duration_total += duration; + + if (m_p->m_duration_max < duration) + m_p->m_duration_max = duration; + + if (m_p->m_duration_min == 0.0 || m_p->m_duration_min > duration) + m_p->m_duration_min = duration; + + if (m_p->m_msg_config.length()) + { + Z_GDU *z_gdu = gdu->get(); + + std::ostringstream os; + os << m_p->m_msg_config << " " + << *m_package << " " + << std::fixed << std::setprecision (6) << duration << " "; + + if (z_gdu) + os << *z_gdu; + else + os << "-"; + + yaz_log(YLOG_LOG, "%s %s", os.str().c_str(), t_info); + } } + else if (!m_package->session().is_closed()) + { + // no response package and yet the session is still open.. + // means that request is unhandled.. + yazpp_1::GDU *gdu_req = &m_package->request(); + Z_GDU *z_gdu = gdu_req->get(); + if (z_gdu && z_gdu->which == Z_GDU_Z3950) + { + // For Z39.50, response with a Close and shutdown + mp::odr odr; + int len; + Z_APDU *apdu_response = odr.create_close( + z_gdu->u.z3950, Z_Close_systemProblem, + "unhandled Z39.50 request"); + + m_assoc_child->send_Z_PDU(apdu_response, &len); + } + else if (z_gdu && z_gdu->which == Z_GDU_HTTP_Request) + { + // For HTTP, respond with Server Error + int len; + mp::odr odr; + Z_GDU *zgdu_res + = odr.create_HTTP_Response(m_package->session(), + z_gdu->u.HTTP_Request, 500); + m_assoc_child->send_GDU(zgdu_res, &len); + } + m_package->session().close(); + } + + if (m_assoc_child->m_no_requests == 0 && m_package->session().is_closed()) + { + m_assoc_child->close(); + } + + + delete this; } -IThreadPoolMsg *ThreadPoolPackage::handle() +mp::IThreadPoolMsg *yf::FrontendNet::ThreadPoolPackage::handle() { - m_package->move(); + m_package->move(m_assoc_child->m_port->route); return this; } - -P2_Session::P2_Session(yazpp_1::IPDU_Observable *the_PDU_Observable, - ThreadPoolSocketObserver *my_thread_pool, - const Package *package) - : Z_Assoc(the_PDU_Observable) +yf::FrontendNet::ZAssocChild::ZAssocChild( + yazpp_1::IPDU_Observable *PDU_Observable, + mp::ThreadPoolSocketObserver *my_thread_pool, + const mp::Package *package, + Port *port, Rep *rep, + yazpp_1::LimitConnect &limit_connect) + : Z_Assoc(PDU_Observable), m_p(rep), m_limit_connect(limit_connect) { - m_my_thread = my_thread_pool; + m_thread_pool_observer = my_thread_pool; m_no_requests = 0; m_delete_flag = false; m_package = package; + m_port = port; + const char *peername = PDU_Observable->getpeername(); + if (!peername) + peername = "unknown"; + else + { + const char *cp = strchr(peername, ':'); + if (cp) + peername = cp + 1; + } + std::string addr; + addr.append(peername); + addr.append(" "); + addr.append(port->port); + m_origin.set_tcpip_address(addr, m_session.id()); + timeout(m_p->m_session_timeout); } - -yazpp_1::IPDU_Observer *P2_Session::sessionNotify(yazpp_1::IPDU_Observable - *the_PDU_Observable, int fd) +yazpp_1::IPDU_Observer *yf::FrontendNet::ZAssocChild::sessionNotify( + yazpp_1::IPDU_Observable *the_PDU_Observable, int fd) { return 0; } -P2_Session::~P2_Session() +yf::FrontendNet::ZAssocChild::~ZAssocChild() { } -void P2_Session::recv_GDU(Z_GDU *z_pdu, int len) +void yf::FrontendNet::ZAssocChild::report(Z_HTTP_Request *hreq) +{ + mp::odr o; + + Z_GDU *gdu_res = o.create_HTTP_Response(m_session, hreq, 200); + + Z_HTTP_Response *hres = gdu_res->u.HTTP_Response; + + mp::wrbuf w; + size_t i; + int number_total = 0; + + for (i = 0; m_p->m_duration_lim[i] != 0.0; i++) + number_total += m_p->m_duration_freq[i]; + number_total += m_p->m_duration_freq[i]; + + wrbuf_puts(w, "\n"); + wrbuf_puts(w, "\n"); + wrbuf_printf(w, " \n", number_total); + for (i = 0; m_p->m_duration_lim[i] != 0.0; i++) + { + if (m_p->m_duration_freq[i] > 0) + wrbuf_printf( + w, " \n", + i > 0 ? m_p->m_duration_lim[i - 1] : 0.0, + m_p->m_duration_lim[i], m_p->m_duration_freq[i]); + } + + if (m_p->m_duration_freq[i] > 0) + wrbuf_printf( + w, " \n", + m_p->m_duration_lim[i - 1], m_p->m_duration_freq[i]); + + if (m_p->m_duration_max != 0.0) + wrbuf_printf( + w, " \n", + m_p->m_duration_max); + if (m_p->m_duration_min != 0.0) + wrbuf_printf( + w, " \n", + m_p->m_duration_min); + if (m_p->m_duration_total != 0.0) + wrbuf_printf( + w, " \n", + m_p->m_duration_total / number_total); + + wrbuf_puts(w, " \n"); + + int thread_busy; + int thread_total; + m_thread_pool_observer->get_thread_info(thread_busy, thread_total); + + wrbuf_printf(w, " \n", + thread_busy, thread_total); + + wrbuf_puts(w, "\n"); + + hres->content_len = w.len(); + hres->content_buf = (char *) w.buf(); + + int len; + send_GDU(gdu_res, &len); +} + +void yf::FrontendNet::ZAssocChild::recv_GDU(Z_GDU *z_pdu, int len) { m_no_requests++; - Package *p = new Package(m_session, m_origin); + mp::Package *p = new mp::Package(m_session, m_origin); + + if (z_pdu && z_pdu->which == Z_GDU_HTTP_Request) + { + Z_HTTP_Request *hreq = z_pdu->u.HTTP_Request; + + const char *f = z_HTTP_header_lookup(hreq->headers, "X-Forwarded-For"); + if (f) + p->origin().set_tcpip_address(std::string(f), m_session.id()); + + if (m_p->m_stat_req.length() + && !strcmp(hreq->path, m_p->m_stat_req.c_str())) + { + report(hreq); + return; + } + std::string peername = p->origin().get_address(); + + m_limit_connect.add_connect(peername.c_str()); + m_limit_connect.cleanup(false); + int con_sz = m_limit_connect.get_total(peername.c_str()); + + if (m_p->m_http_req_max && con_sz >= m_p->m_http_req_max) + { + mp::odr o; + Z_GDU *gdu_res = o.create_HTTP_Response(m_session, hreq, 500); + int len; + send_GDU(gdu_res, &len); + return; + } + } - ThreadPoolPackage *m = new ThreadPoolPackage(p, this); - p->copy_filter(*m_package); + ThreadPoolPackage *tp = new ThreadPoolPackage(p, this, m_p); + p->copy_route(*m_package); p->request() = yazpp_1::GDU(z_pdu); - m_my_thread->put(m); + + if (m_p->m_msg_config.length()) + { + if (z_pdu) + { + std::ostringstream os; + os << m_p->m_msg_config << " " + << *p << " " + << "0.000000" << " " + << *z_pdu; + yaz_log(YLOG_LOG, "%s", os.str().c_str()); + } + } + m_thread_pool_observer->put(tp); } -void P2_Session::failNotify() +void yf::FrontendNet::ZAssocChild::failNotify() { // TODO: send Package to signal "close" - m_delete_flag = true; - if (m_no_requests == 0) - delete this; - + if (m_session.is_closed()) + { + if (m_no_requests == 0) + delete this; + return; + } + m_no_requests++; + + m_session.close(); + + mp::Package *p = new mp::Package(m_session, m_origin); + + ThreadPoolPackage *tp = new ThreadPoolPackage(p, this, m_p); + p->copy_route(*m_package); + m_thread_pool_observer->cleanup(tp, &m_session); + m_thread_pool_observer->put(tp); } -void P2_Session::timeoutNotify() +void yf::FrontendNet::ZAssocChild::timeoutNotify() { - // TODO: send Package to signal "close" - m_delete_flag = true; - if (m_no_requests == 0) - delete this; + failNotify(); } -void P2_Session::connectNotify() +void yf::FrontendNet::ZAssocChild::connectNotify() { } -class P2_Server : public yazpp_1::Z_Assoc { -public: - ~P2_Server(); - P2_Server(yazpp_1::IPDU_Observable *the_PDU_Observable, - ThreadPoolSocketObserver *m_my_thread, - const Package *package); -private: - yazpp_1::IPDU_Observer* sessionNotify( - yazpp_1::IPDU_Observable *the_PDU_Observable, - int fd); - void recv_GDU(Z_GDU *apdu, int len); - - void failNotify(); - void timeoutNotify(); - void connectNotify(); -private: - ThreadPoolSocketObserver *m_my_thread; - const Package *m_package; -}; +yf::FrontendNet::ZAssocServer::ZAssocServer( + yazpp_1::IPDU_Observable *PDU_Observable, + Port *port, + Rep *rep) + : + Z_Assoc(PDU_Observable), m_port(port), m_p(rep) +{ + m_package = 0; +} -P2_Server::P2_Server(yazpp_1::IPDU_Observable *the_PDU_Observable, - ThreadPoolSocketObserver *my_thread, - const Package *package) - : Z_Assoc(the_PDU_Observable) +void yf::FrontendNet::ZAssocServer::set_package(const mp::Package *package) { - m_my_thread = my_thread; m_package = package; +} +void yf::FrontendNet::ZAssocServer::set_thread_pool( + mp::ThreadPoolSocketObserver *observer) +{ + m_thread_pool_observer = observer; } -yazpp_1::IPDU_Observer *P2_Server::sessionNotify(yazpp_1::IPDU_Observable - *the_PDU_Observable, int fd) +yazpp_1::IPDU_Observer *yf::FrontendNet::ZAssocServer::sessionNotify( + yazpp_1::IPDU_Observable *the_PDU_Observable, int fd) { - P2_Session *my = new P2_Session(the_PDU_Observable, m_my_thread, - m_package); + + const char *peername = the_PDU_Observable->getpeername(); + if (peername) + { + limit_connect.add_connect(peername); + limit_connect.cleanup(false); + int con_sz = limit_connect.get_total(peername); + if (m_p->m_connect_max && con_sz > m_p->m_connect_max) + return 0; + } + ZAssocChild *my = new ZAssocChild(the_PDU_Observable, + m_thread_pool_observer, + m_package, m_port, m_p, limit_connect); return my; } -P2_Server::~P2_Server() +yf::FrontendNet::ZAssocServer::~ZAssocServer() +{ +} + +void yf::FrontendNet::ZAssocServer::recv_GDU(Z_GDU *apdu, int len) { } -void P2_Server::recv_GDU(Z_GDU *apdu, int len) +void yf::FrontendNet::ZAssocServer::failNotify() { } -void P2_Server::failNotify() +void yf::FrontendNet::ZAssocServer::timeoutNotify() { } -void P2_Server::timeoutNotify() +void yf::FrontendNet::ZAssocServer::connectNotify() { } -void P2_Server::connectNotify() +yf::FrontendNet::FrontendNet() : m_p(new Rep) { } -FilterFrontendNet::FilterFrontendNet() +yf::FrontendNet::Rep::Rep() { - m_no_threads = 5; - m_listen_address = "@:9001"; + m_max_threads = m_no_threads = 5; + m_stack_size = 0; m_listen_duration = 0; + m_session_timeout = 300; // 5 minutes + m_connect_max = 0; + az = 0; + size_t i; + for (i = 0; i < 22; i++) + m_duration_freq[i] = 0; + m_duration_lim[0] = 0.000001; + m_duration_lim[1] = 0.00001; + m_duration_lim[2] = 0.0001; + m_duration_lim[3] = 0.001; + m_duration_lim[4] = 0.01; + m_duration_lim[5] = 0.1; + m_duration_lim[6] = 0.2; + m_duration_lim[7] = 0.3; + m_duration_lim[8] = 0.5; + m_duration_lim[9] = 1.0; + m_duration_lim[10] = 1.5; + m_duration_lim[11] = 2.0; + m_duration_lim[12] = 3.0; + m_duration_lim[13] = 4.0; + m_duration_lim[14] = 5.0; + m_duration_lim[15] = 6.0; + m_duration_lim[16] = 8.0; + m_duration_lim[17] = 10.0; + m_duration_lim[18] = 15.0; + m_duration_lim[19] = 20.0; + m_duration_lim[20] = 30.0; + m_duration_lim[21] = 0.0; + m_duration_max = 0.0; + m_duration_min = 0.0; + m_duration_total = 0.0; + m_stop_signo = 0; } -class My_Timer_Thread : public yazpp_1::ISocketObserver { -private: - yazpp_1::ISocketObservable *m_obs; - int m_fd[2]; - bool m_timeout; -public: - My_Timer_Thread(yazpp_1::ISocketObservable *obs, int duration); - void socketNotify(int event); - bool timeout(); -}; +yf::FrontendNet::Rep::~Rep() +{ + if (az) + { + size_t i; + for (i = 0; i < m_ports.size(); i++) + delete az[i]; + delete [] az; + delete [] pdu; + } + az = 0; +} -bool My_Timer_Thread::timeout() +yf::FrontendNet::~FrontendNet() +{ +} + +void yf::FrontendNet::stop(int signo) const +{ + m_p->m_stop_signo = signo; +} + +void yf::FrontendNet::start() const +{ +#if HAVE_GETRLIMIT + struct rlimit limit_data; + getrlimit(RLIMIT_NOFILE, &limit_data); + yaz_log(YLOG_LOG, "getrlimit NOFILE cur=%ld max=%ld", + (long) limit_data.rlim_cur, (long) limit_data.rlim_max); +#endif +} + +bool yf::FrontendNet::My_Timer_Thread::timeout() { return m_timeout; } -My_Timer_Thread::My_Timer_Thread(yazpp_1::ISocketObservable *obs, - int duration) : - m_obs(obs), m_timeout(false) +yf::FrontendNet::My_Timer_Thread::My_Timer_Thread( + yazpp_1::ISocketObservable *obs, + int duration) : + m_obs(obs), m_pipe(9123), m_timeout(false) { - pipe(m_fd); - obs->addObserver(m_fd[0], this); + obs->addObserver(m_pipe.read_fd(), this); obs->maskObserver(this, yazpp_1::SOCKET_OBSERVE_READ); obs->timeoutObserver(this, duration); } -void My_Timer_Thread::socketNotify(int event) +void yf::FrontendNet::My_Timer_Thread::socketNotify(int event) { m_timeout = true; m_obs->deleteObserver(this); - close(m_fd[0]); - close(m_fd[1]); } -Package &FilterFrontendNet::process(Package &package) const { - yazpp_1::SocketManager mySocketManager; - +void yf::FrontendNet::process(mp::Package &package) const +{ + if (m_p->az == 0) + return; + size_t i; My_Timer_Thread *tt = 0; - if (m_listen_duration) - tt = new My_Timer_Thread(&mySocketManager, m_listen_duration); - yazpp_1::PDU_Assoc *my_PDU_Assoc = - new yazpp_1::PDU_Assoc(&mySocketManager); - - ThreadPoolSocketObserver threadPool(&mySocketManager, m_no_threads); + if (m_p->m_listen_duration) + tt = new My_Timer_Thread(&m_p->mySocketManager, + m_p->m_listen_duration); - P2_Server z(my_PDU_Assoc, &threadPool, &package); - z.server(m_listen_address.c_str()); + ThreadPoolSocketObserver tp(&m_p->mySocketManager, m_p->m_no_threads, + m_p->m_max_threads, + m_p->m_stack_size); - while (mySocketManager.processEvent() > 0) + for (i = 0; im_ports.size(); i++) { + m_p->az[i]->set_package(&package); + m_p->az[i]->set_thread_pool(&tp); + } + while (m_p->mySocketManager.processEvent() > 0) + { + if (m_p->m_stop_signo == SIGTERM) + { + yaz_log(YLOG_LOG, "metaproxy received SIGTERM"); + if (m_p->az) + { + size_t i; + for (i = 0; i < m_p->m_ports.size(); i++) + { + m_p->pdu[i]->shutdown(); + m_p->az[i]->server(""); + } + yaz_daemon_stop(); + } + break; /* stop right away */ + } +#ifndef WIN32 + if (m_p->m_stop_signo == SIGUSR1) + { /* just stop listeners and cont till all sessions are done*/ + yaz_log(YLOG_LOG, "metaproxy received SIGUSR1"); + m_p->m_stop_signo = 0; + if (m_p->az) + { + size_t i; + for (i = 0; i < m_p->m_ports.size(); i++) + m_p->az[i]->server(""); + yaz_daemon_stop(); + } + } +#endif + int no = m_p->mySocketManager.getNumberOfObservers(); + if (no <= 1) + break; if (tt && tt->timeout()) break; } - return package; + delete tt; } -std::string &FilterFrontendNet::listen_address() +void yf::FrontendNet::configure(const xmlNode * ptr, bool test_only, + const char *path) { - return m_listen_address; + if (!ptr || !ptr->children) + { + throw yf::FilterException("No ports for Frontend"); + } + std::vector ports; + for (ptr = ptr->children; ptr; ptr = ptr->next) + { + if (ptr->type != XML_ELEMENT_NODE) + continue; + if (!strcmp((const char *) ptr->name, "port")) + { + Port port; + + const char *names[5] = {"route", "max_recv_bytes", "port", + "cert_fname", 0}; + std::string values[4]; + + mp::xml::parse_attr(ptr, names, values); + port.route = values[0]; + if (values[1].length() > 0) + port.max_recv_bytes = atoi(values[1].c_str()); + else + port.max_recv_bytes = 0; + if (values[2].length() > 0) + port.port = values[2]; + else + port.port = mp::xml::get_text(ptr); + port.cert_fname = values[3]; + ports.push_back(port); + } + else if (!strcmp((const char *) ptr->name, "threads")) + { + std::string threads_str = mp::xml::get_text(ptr); + int threads = atoi(threads_str.c_str()); + if (threads < 1) + throw yf::FilterException("Bad value for threads: " + + threads_str); + m_p->m_no_threads = threads; + } + else if (!strcmp((const char *) ptr->name, "max-threads")) + { + std::string threads_str = mp::xml::get_text(ptr); + int threads = atoi(threads_str.c_str()); + if (threads < 1) + throw yf::FilterException("Bad value for max-threads: " + + threads_str); + m_p->m_max_threads = threads; + } + else if (!strcmp((const char *) ptr->name, "stack-size")) + { + std::string sz_str = mp::xml::get_text(ptr); + int sz = atoi(sz_str.c_str()); + if (sz < 0) + throw yf::FilterException("Bad value for stack-size: " + + sz_str); + m_p->m_stack_size = sz * 1024; + } + else if (!strcmp((const char *) ptr->name, "timeout")) + { + std::string timeout_str = mp::xml::get_text(ptr); + int timeout = atoi(timeout_str.c_str()); + if (timeout < 1) + throw yf::FilterException("Bad value for timeout: " + + timeout_str); + m_p->m_session_timeout = timeout; + } + else if (!strcmp((const char *) ptr->name, "connect-max")) + { + m_p->m_connect_max = mp::xml::get_int(ptr, 0); + } + else if (!strcmp((const char *) ptr->name, "http-req-max")) + { + m_p->m_http_req_max = mp::xml::get_int(ptr, 0); + } + else if (!strcmp((const char *) ptr->name, "message")) + { + m_p->m_msg_config = mp::xml::get_text(ptr); + } + else if (!strcmp((const char *) ptr->name, "stat-req")) + { + m_p->m_stat_req = mp::xml::get_text(ptr); + } + else + { + throw yf::FilterException("Bad element " + + std::string((const char *) + ptr->name)); + } + } + if (test_only) + return; + set_ports(ports); } -int &FilterFrontendNet::listen_duration() +void yf::FrontendNet::set_ports(std::vector &ports) { - return m_listen_duration; + std::vector nports; + size_t i; + + for (i = 0; i < ports.size(); i++) + { + Port nport; + + nport.port = ports[i]; + + nports.push_back(nport); + } + set_ports(nports); } + +void yf::FrontendNet::set_ports(std::vector &ports) +{ + m_p->m_ports = ports; + + m_p->az = new yf::FrontendNet::ZAssocServer *[m_p->m_ports.size()]; + m_p->pdu = new yazpp_1::PDU_Assoc *[m_p->m_ports.size()]; + + // Create yf::FrontendNet::ZAssocServer for each port + size_t i; + for (i = 0; i < m_p->m_ports.size(); i++) + m_p->az[i] = 0; + for (i = 0; i < m_p->m_ports.size(); i++) + { + // create a PDU assoc object (one per yf::FrontendNet::ZAssocServer) + yazpp_1::PDU_Assoc *as = new yazpp_1::PDU_Assoc(&m_p->mySocketManager); + + if (m_p->m_ports[i].cert_fname.length()) + as->set_cert_fname(m_p->m_ports[i].cert_fname.c_str()); + // create ZAssoc with PDU Assoc + m_p->pdu[i] = as; + m_p->az[i] = new yf::FrontendNet::ZAssocServer( + as, &m_p->m_ports[i], m_p.get()); + if (m_p->az[i]->server(m_p->m_ports[i].port.c_str())) + { + throw yf::FilterException("Unable to bind to address " + + std::string(m_p->m_ports[i].port)); + } + COMSTACK cs = as->get_comstack(); + + if (cs && m_p->m_ports[i].max_recv_bytes) + cs_set_max_recv_bytes(cs, m_p->m_ports[i].max_recv_bytes); + + } +} + +void yf::FrontendNet::set_listen_duration(int d) +{ + m_p->m_listen_duration = d; +} + +static yf::Base* filter_creator() +{ + return new yf::FrontendNet; +} + +extern "C" { + struct metaproxy_1_filter_struct metaproxy_1_filter_frontend_net = { + 0, + "frontend_net", + filter_creator + }; +} + +/* + * Local variables: + * c-basic-offset: 4 + * c-file-style: "Stroustrup" + * indent-tabs-mode: nil + * End: + * vim: shiftwidth=4 tabstop=8 expandtab + */ +