X-Git-Url: http://git.indexdata.com/?a=blobdiff_plain;f=src%2Ffilter_frontend_net.cpp;h=fe5b3ace6b49ecdc0393a694b90182ef5c1652b2;hb=b70b9ec78f0ab1c3ed3b432de986159129a0e4ed;hp=5b7f59d0dcd05ef90f2a2f67da1cdd40455b8730;hpb=b84218609efb976d5b5270bd9e95a3c421af3a13;p=metaproxy-moved-to-github.git diff --git a/src/filter_frontend_net.cpp b/src/filter_frontend_net.cpp index 5b7f59d..fe5b3ac 100644 --- a/src/filter_frontend_net.cpp +++ b/src/filter_frontend_net.cpp @@ -1,120 +1,192 @@ +/* $Id: filter_frontend_net.cpp,v 1.23 2007-01-25 14:05:54 adam Exp $ + Copyright (c) 2005-2007, Index Data. + See the LICENSE file for details + */ #include "config.hpp" +#include "util.hpp" +#include "pipe.hpp" #include "filter.hpp" -#include "router.hpp" #include "package.hpp" #include "thread_pool_observer.hpp" #include "filter_frontend_net.hpp" -#include -#include -#include +#include +#include +#include #include #include -class ZAssocServerChild : public yazpp_1::Z_Assoc { -public: - ~ZAssocServerChild(); - ZAssocServerChild(yazpp_1::IPDU_Observable *the_PDU_Observable, - yp2::ThreadPoolSocketObserver *m_thread_pool_observer, - const yp2::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(); +namespace mp = metaproxy_1; + +namespace metaproxy_1 { + namespace filter { + class FrontendNet::Rep { + friend class FrontendNet; + int m_no_threads; + std::vector m_ports; + int m_listen_duration; + int m_session_timeout; + }; + } + class 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 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); + 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: + mp::ThreadPoolSocketObserver *m_thread_pool_observer; + mp::Session m_session; + mp::Origin m_origin; + bool m_delete_flag; + const mp::Package *m_package; + }; + class ThreadPoolPackage : public mp::IThreadPoolMsg { + public: + ThreadPoolPackage(mp::Package *package, mp::ZAssocChild *ses) : + m_session(ses), m_package(package) { }; + ~ThreadPoolPackage(); + IThreadPoolMsg *handle(); + void result(); + + private: + mp::ZAssocChild *m_session; + 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, + int timeout); + 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: - yp2::ThreadPoolSocketObserver *m_thread_pool_observer; - yp2::Session m_session; - yp2::Origin m_origin; - bool m_delete_flag; - const yp2::Package *m_package; -}; - - -class ThreadPoolPackage : public yp2::IThreadPoolMsg { -public: - ThreadPoolPackage(yp2::Package *package, ZAssocServerChild *ses) : - m_session(ses), m_package(package) { }; - ~ThreadPoolPackage(); - IThreadPoolMsg *handle(); - void result(); - -private: - ZAssocServerChild *m_session; - yp2::Package *m_package; - -}; - -ThreadPoolPackage::~ThreadPoolPackage() + private: + mp::ThreadPoolSocketObserver *m_thread_pool_observer; + const mp::Package *m_package; + int m_session_timeout; + }; +} + +mp::ThreadPoolPackage::~ThreadPoolPackage() { delete m_package; } -void ThreadPoolPackage::result() +void mp::ThreadPoolPackage::result() { m_session->m_no_requests--; yazpp_1::GDU *gdu = &m_package->response(); + if (gdu->get()) { int len; m_session->send_GDU(gdu->get(), &len); } + 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_session->send_Z_PDU(apdu_response, &len); + m_package->session().close(); + } + } + if (m_session->m_no_requests == 0 && m_package->session().is_closed()) delete m_session; delete this; } -yp2::IThreadPoolMsg *ThreadPoolPackage::handle() +mp::IThreadPoolMsg *mp::ThreadPoolPackage::handle() { m_package->move(); return this; } -ZAssocServerChild::ZAssocServerChild(yazpp_1::IPDU_Observable *PDU_Observable, - yp2::ThreadPoolSocketObserver *my_thread_pool, - const yp2::Package *package) +mp::ZAssocChild::ZAssocChild(yazpp_1::IPDU_Observable *PDU_Observable, + mp::ThreadPoolSocketObserver *my_thread_pool, + const mp::Package *package) : Z_Assoc(PDU_Observable) { m_thread_pool_observer = my_thread_pool; 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()); } -yazpp_1::IPDU_Observer *ZAssocServerChild::sessionNotify(yazpp_1::IPDU_Observable +yazpp_1::IPDU_Observer *mp::ZAssocChild::sessionNotify(yazpp_1::IPDU_Observable *the_PDU_Observable, int fd) { return 0; } -ZAssocServerChild::~ZAssocServerChild() +mp::ZAssocChild::~ZAssocChild() { } -void ZAssocServerChild::recv_GDU(Z_GDU *z_pdu, int len) +void mp::ZAssocChild::recv_GDU(Z_GDU *z_pdu, int len) { m_no_requests++; - yp2::Package *p = new yp2::Package(m_session, m_origin); + mp::Package *p = new mp::Package(m_session, m_origin); - ThreadPoolPackage *tp = new ThreadPoolPackage(p, this); + mp::ThreadPoolPackage *tp = new mp::ThreadPoolPackage(p, this); p->copy_filter(*m_package); p->request() = yazpp_1::GDU(z_pdu); m_thread_pool_observer->put(tp); } -void ZAssocServerChild::failNotify() +void mp::ZAssocChild::failNotify() { // TODO: send Package to signal "close" if (m_session.is_closed()) @@ -123,164 +195,207 @@ void ZAssocServerChild::failNotify() m_session.close(); - yp2::Package *p = new yp2::Package(m_session, m_origin); + mp::Package *p = new mp::Package(m_session, m_origin); - ThreadPoolPackage *tp = new ThreadPoolPackage(p, this); + mp::ThreadPoolPackage *tp = new mp::ThreadPoolPackage(p, this); p->copy_filter(*m_package); m_thread_pool_observer->put(tp); } -void ZAssocServerChild::timeoutNotify() +void mp::ZAssocChild::timeoutNotify() { failNotify(); } -void ZAssocServerChild::connectNotify() +void mp::ZAssocChild::connectNotify() { } -class ZAssocServer : public yazpp_1::Z_Assoc { -public: - ~ZAssocServer(); - ZAssocServer(yazpp_1::IPDU_Observable *PDU_Observable, - yp2::ThreadPoolSocketObserver *m_thread_pool_observer, - const yp2::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: - yp2::ThreadPoolSocketObserver *m_thread_pool_observer; - const yp2::Package *m_package; -}; - - -ZAssocServer::ZAssocServer(yazpp_1::IPDU_Observable *PDU_Observable, - yp2::ThreadPoolSocketObserver *thread_pool_observer, - const yp2::Package *package) - : Z_Assoc(PDU_Observable) +mp::ZAssocServer::ZAssocServer(yazpp_1::IPDU_Observable *PDU_Observable, + mp::ThreadPoolSocketObserver *thread_pool_observer, + const mp::Package *package, + int timeout) + : Z_Assoc(PDU_Observable), m_session_timeout(timeout) { m_thread_pool_observer = thread_pool_observer; m_package = package; } -yazpp_1::IPDU_Observer *ZAssocServer::sessionNotify(yazpp_1::IPDU_Observable +yazpp_1::IPDU_Observer *mp::ZAssocServer::sessionNotify(yazpp_1::IPDU_Observable *the_PDU_Observable, int fd) { - ZAssocServerChild *my = - new ZAssocServerChild(the_PDU_Observable, m_thread_pool_observer, - m_package); + mp::ZAssocChild *my = + new mp::ZAssocChild(the_PDU_Observable, m_thread_pool_observer, + m_package); + my->timeout(m_session_timeout); return my; } -ZAssocServer::~ZAssocServer() +mp::ZAssocServer::~ZAssocServer() +{ +} + +void mp::ZAssocServer::recv_GDU(Z_GDU *apdu, int len) { } -void ZAssocServer::recv_GDU(Z_GDU *apdu, int len) +void mp::ZAssocServer::failNotify() { } -void ZAssocServer::failNotify() +void mp::ZAssocServer::timeoutNotify() { } -void ZAssocServer::timeoutNotify() +void mp::ZAssocServer::connectNotify() { } -void ZAssocServer::connectNotify() +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 } -yp2::filter::FrontendNet::FrontendNet() +mp::filter::FrontendNet::~FrontendNet() { - m_no_threads = 5; - m_listen_duration = 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(); -}; - -bool My_Timer_Thread::timeout() +bool mp::My_Timer_Thread::timeout() { return m_timeout; } -My_Timer_Thread::My_Timer_Thread(yazpp_1::ISocketObservable *obs, +mp::My_Timer_Thread::My_Timer_Thread(yazpp_1::ISocketObservable *obs, int duration) : - m_obs(obs), m_timeout(false) + 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 mp::My_Timer_Thread::socketNotify(int event) { m_timeout = true; m_obs->deleteObserver(this); - close(m_fd[0]); - close(m_fd[1]); } -void yp2::filter::FrontendNet::process(Package &package) const { +void mp::filter::FrontendNet::process(Package &package) const +{ + if (m_p->m_ports.size() == 0) + return; + yazpp_1::SocketManager mySocketManager; My_Timer_Thread *tt = 0; - if (m_listen_duration) - tt = new My_Timer_Thread(&mySocketManager, m_listen_duration); + if (m_p->m_listen_duration) + tt = new My_Timer_Thread(&mySocketManager, m_p->m_listen_duration); - ThreadPoolSocketObserver threadPool(&mySocketManager, m_no_threads); + ThreadPoolSocketObserver threadPool(&mySocketManager, m_p->m_no_threads); - ZAssocServer **az = new ZAssocServer *[m_ports.size()]; + mp::ZAssocServer **az = new mp::ZAssocServer *[m_p->m_ports.size()]; - // Create ZAssocServer for each port + // Create mp::ZAssocServer for each port size_t i; - for (i = 0; im_ports.size(); i++) { - // create a PDU assoc object (one per ZAssocServer) + // 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 ZAssocServer(as, &threadPool, &package); - az[i]->server(m_ports[i].c_str()); + az[i] = new mp::ZAssocServer(as, &threadPool, &package, + m_p->m_session_timeout); + az[i]->server(m_p->m_ports[i].c_str()); } while (mySocketManager.processEvent() > 0) { if (tt && tt->timeout()) break; } - for (i = 0; im_ports.size(); i++) delete az[i]; delete [] az; delete tt; } -std::vector &yp2::filter::FrontendNet::ports() +void mp::filter::FrontendNet::configure(const xmlNode * ptr) +{ + if (!ptr || !ptr->children) + { + throw mp::filter::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")) + { + std::string port = mp::xml::get_text(ptr); + 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 mp::filter::FilterException("Bad value for threads: " + + 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 + { + throw mp::filter::FilterException("Bad element " + + std::string((const char *) + ptr->name)); + } + } + m_p->m_ports = ports; +} + +std::vector &mp::filter::FrontendNet::ports() { - return m_ports; + return m_p->m_ports; } -int &yp2::filter::FrontendNet::listen_duration() +int &mp::filter::FrontendNet::listen_duration() { - return m_listen_duration; + return m_p->m_listen_duration; +} + +static mp::filter::Base* filter_creator() +{ + return new mp::filter::FrontendNet; +} + +extern "C" { + struct metaproxy_1_filter_struct metaproxy_1_filter_frontend_net = { + 0, + "frontend_net", + filter_creator + }; } +/* + * Local variables: + * c-basic-offset: 4 + * indent-tabs-mode: nil + * c-file-style: "stroustrup" + * End: + * vim: shiftwidth=4 tabstop=8 expandtab + */