X-Git-Url: http://git.indexdata.com/?a=blobdiff_plain;f=src%2Ffilter_frontend_net.cpp;h=1a91c076be66f2aab18b8f3ec86977d7564256c1;hb=4cd9e521c5e96e3fcc5dcec6192571b54a1351c4;hp=77263f9ff7eb5a0189416ddb1e905e5e813b449f;hpb=4753f6f22d15b41bf1efcee6294ae1907ba455e7;p=metaproxy-moved-to-github.git diff --git a/src/filter_frontend_net.cpp b/src/filter_frontend_net.cpp index 77263f9..1a91c07 100644 --- a/src/filter_frontend_net.cpp +++ b/src/filter_frontend_net.cpp @@ -1,8 +1,20 @@ -/* $Id: filter_frontend_net.cpp,v 1.18 2006-06-09 14:11:42 adam Exp $ - Copyright (c) 2005-2006, Index Data. +/* This file is part of Metaproxy. + Copyright (C) 2005-2009 Index Data -%LICENSE% - */ +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" @@ -15,6 +27,7 @@ #include #include #include +#include #include #include @@ -22,12 +35,18 @@ namespace mp = metaproxy_1; namespace metaproxy_1 { + class My_Timer_Thread; + class ZAssocServer; namespace filter { class FrontendNet::Rep { friend class FrontendNet; int m_no_threads; std::vector m_ports; int m_listen_duration; + int m_session_timeout; + int m_connect_max; + yazpp_1::SocketManager mySocketManager; + ZAssocServer **az; }; } class My_Timer_Thread : public yazpp_1::ISocketObserver { @@ -66,22 +85,23 @@ namespace metaproxy_1 { class ThreadPoolPackage : public mp::IThreadPoolMsg { public: ThreadPoolPackage(mp::Package *package, mp::ZAssocChild *ses) : - m_session(ses), m_package(package) { }; + m_assoc_child(ses), m_package(package) { }; ~ThreadPoolPackage(); IThreadPoolMsg *handle(); void result(); private: - mp::ZAssocChild *m_session; + mp::ZAssocChild *m_assoc_child; mp::Package *m_package; }; class ZAssocServer : public yazpp_1::Z_Assoc { public: ~ZAssocServer(); - ZAssocServer(yazpp_1::IPDU_Observable *PDU_Observable, - mp::ThreadPoolSocketObserver *m_thread_pool_observer, - const mp::Package *package); + ZAssocServer(yazpp_1::IPDU_Observable *PDU_Observable, int timeout, + int connect_max); + void set_package(const mp::Package *package); + void set_thread_pool(ThreadPoolSocketObserver *m_thread_pool_observer); private: yazpp_1::IPDU_Observer* sessionNotify( yazpp_1::IPDU_Observable *the_PDU_Observable, @@ -94,6 +114,9 @@ namespace metaproxy_1 { private: mp::ThreadPoolSocketObserver *m_thread_pool_observer; const mp::Package *m_package; + int m_session_timeout; + int m_connect_max; + yazpp_1::LimitConnect limit_connect; }; } @@ -104,14 +127,14 @@ mp::ThreadPoolPackage::~ThreadPoolPackage() void mp::ThreadPoolPackage::result() { - m_session->m_no_requests--; + 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); } else if (!m_package->session().is_closed()) { @@ -128,13 +151,25 @@ void mp::ThreadPoolPackage::result() z_gdu->u.z3950, Z_Close_systemProblem, "unhandled Z39.50 request"); - m_session->send_Z_PDU(apdu_response, &len); - m_package->session().close(); + 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_session->m_no_requests == 0 && m_package->session().is_closed()) - delete m_session; + if (m_assoc_child->m_no_requests == 0 && m_package->session().is_closed()) + { + m_assoc_child->close(); + } delete this; } @@ -154,6 +189,10 @@ mp::ZAssocChild::ZAssocChild(yazpp_1::IPDU_Observable *PDU_Observable, m_no_requests = 0; m_delete_flag = false; m_package = package; + const char *peername = PDU_Observable->getpeername(); + if (!peername) + peername = "unknown"; + m_origin.set_tcpip_address(std::string(peername), m_session.id()); } @@ -183,7 +222,11 @@ void mp::ZAssocChild::failNotify() { // TODO: send Package to signal "close" if (m_session.is_closed()) + { + if (m_no_requests == 0) + delete this; return; + } m_no_requests++; m_session.close(); @@ -206,21 +249,41 @@ void mp::ZAssocChild::connectNotify() } mp::ZAssocServer::ZAssocServer(yazpp_1::IPDU_Observable *PDU_Observable, - mp::ThreadPoolSocketObserver *thread_pool_observer, - const mp::Package *package) - : Z_Assoc(PDU_Observable) + int timeout, int connect_max) + : Z_Assoc(PDU_Observable), m_session_timeout(timeout), + m_connect_max(connect_max) +{ + m_package = 0; +} + + +void mp::ZAssocServer::set_package(const mp::Package *package) { - m_thread_pool_observer = thread_pool_observer; m_package = package; +} +void mp::ZAssocServer::set_thread_pool(ThreadPoolSocketObserver *observer) +{ + m_thread_pool_observer = observer; } yazpp_1::IPDU_Observer *mp::ZAssocServer::sessionNotify(yazpp_1::IPDU_Observable *the_PDU_Observable, int fd) { + + 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_connect_max && con_sz > m_connect_max) + return 0; + } mp::ZAssocChild *my = new mp::ZAssocChild(the_PDU_Observable, m_thread_pool_observer, m_package); + my->timeout(m_session_timeout); return my; } @@ -248,10 +311,19 @@ mp::filter::FrontendNet::FrontendNet() : m_p(new Rep) { m_p->m_no_threads = 5; m_p->m_listen_duration = 0; + m_p->m_session_timeout = 300; // 5 minutes + m_p->az = 0; } mp::filter::FrontendNet::~FrontendNet() { + if (m_p->az) + { + size_t i; + for (i = 0; im_ports.size(); i++) + delete m_p->az[i]; + delete [] m_p->az; + } } bool mp::My_Timer_Thread::timeout() @@ -276,43 +348,31 @@ void mp::My_Timer_Thread::socketNotify(int event) void mp::filter::FrontendNet::process(Package &package) const { - if (m_p->m_ports.size() == 0) + if (m_p->az == 0) return; - - yazpp_1::SocketManager mySocketManager; - + size_t i; My_Timer_Thread *tt = 0; - if (m_p->m_listen_duration) - tt = new My_Timer_Thread(&mySocketManager, m_p->m_listen_duration); - ThreadPoolSocketObserver threadPool(&mySocketManager, m_p->m_no_threads); - - mp::ZAssocServer **az = new mp::ZAssocServer *[m_p->m_ports.size()]; + if (m_p->m_listen_duration) + tt = new My_Timer_Thread(&m_p->mySocketManager, + m_p->m_listen_duration); + + ThreadPoolSocketObserver tp(&m_p->mySocketManager, m_p->m_no_threads); - // Create mp::ZAssocServer for each port - size_t i; for (i = 0; im_ports.size(); i++) { - // create a PDU assoc object (one per mp::ZAssocServer) - yazpp_1::PDU_Assoc *as = new yazpp_1::PDU_Assoc(&mySocketManager); - - // create ZAssoc with PDU Assoc - az[i] = new mp::ZAssocServer(as, &threadPool, &package); - az[i]->server(m_p->m_ports[i].c_str()); + m_p->az[i]->set_package(&package); + m_p->az[i]->set_thread_pool(&tp); } - while (mySocketManager.processEvent() > 0) + while (m_p->mySocketManager.processEvent() > 0) { if (tt && tt->timeout()) break; } - for (i = 0; im_ports.size(); i++) - delete az[i]; - - delete [] az; delete tt; } -void mp::filter::FrontendNet::configure(const xmlNode * ptr) +void mp::filter::FrontendNet::configure(const xmlNode * ptr, bool test_only) { if (!ptr || !ptr->children) { @@ -338,6 +398,19 @@ void mp::filter::FrontendNet::configure(const xmlNode * ptr) + threads_str); m_p->m_no_threads = threads; } + 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 mp::filter::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->children, 0); + } else { throw mp::filter::FilterException("Bad element " @@ -345,17 +418,39 @@ void mp::filter::FrontendNet::configure(const xmlNode * ptr) ptr->name)); } } - m_p->m_ports = ports; + if (test_only) + return; + set_ports(ports); } -std::vector &mp::filter::FrontendNet::ports() +void mp::filter::FrontendNet::set_ports(std::vector &ports) { - return m_p->m_ports; + m_p->m_ports = ports; + + m_p->az = new mp::ZAssocServer *[m_p->m_ports.size()]; + + // Create mp::ZAssocServer for each port + size_t i; + for (i = 0; im_ports.size(); i++) + { + // create a PDU assoc object (one per mp::ZAssocServer) + yazpp_1::PDU_Assoc *as = new yazpp_1::PDU_Assoc(&m_p->mySocketManager); + + // create ZAssoc with PDU Assoc + m_p->az[i] = new mp::ZAssocServer(as, + m_p->m_session_timeout, + m_p->m_connect_max); + if (m_p->az[i]->server(m_p->m_ports[i].c_str())) + { + throw mp::filter::FilterException("Unable to bind to address " + + std::string(m_p->m_ports[i])); + } + } } -int &mp::filter::FrontendNet::listen_duration() +void mp::filter::FrontendNet::set_listen_duration(int d) { - return m_p->m_listen_duration; + m_p->m_listen_duration = d; } static mp::filter::Base* filter_creator() @@ -374,8 +469,9 @@ extern "C" { /* * Local variables: * c-basic-offset: 4 + * c-file-style: "Stroustrup" * indent-tabs-mode: nil - * c-file-style: "stroustrup" * End: * vim: shiftwidth=4 tabstop=8 expandtab */ +