From 850cba55769a7ed94d4da0e7c05b823e54ce3cc3 Mon Sep 17 00:00:00 2001 From: Adam Dickmeiss Date: Tue, 21 Jun 2005 21:54:05 +0000 Subject: [PATCH] Added small server to test MSG threads. --- src/.cvsignore | 2 + src/Makefile.am | 5 +- src/t-server.cpp | 272 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 3 files changed, 277 insertions(+), 2 deletions(-) create mode 100644 src/t-server.cpp diff --git a/src/.cvsignore b/src/.cvsignore index 57261d1..0fad00d 100644 --- a/src/.cvsignore +++ b/src/.cvsignore @@ -6,3 +6,5 @@ yazproxy Makefile Makefile.in cdetails +t-server +tstthreads diff --git a/src/Makefile.am b/src/Makefile.am index 95c84ca..1747d6d 100644 --- a/src/Makefile.am +++ b/src/Makefile.am @@ -1,4 +1,4 @@ -## $Id: Makefile.am,v 1.8 2005-05-30 20:09:21 adam Exp $ +## $Id: Makefile.am,v 1.9 2005-06-21 21:54:05 adam Exp $ AM_CXXFLAGS = $(YAZPPINC) -I$(srcdir)/../include $(XSLT_CFLAGS) $(USEMARCONINC) @@ -11,13 +11,14 @@ libyazproxy_la_SOURCES= yaz-proxy.cpp yaz-proxy-config.cpp yaz-bw.cpp \ bin_PROGRAMS = yazproxy check_PROGRAMS = cdetails -noinst_PROGRAMS = tstthreads +noinst_PROGRAMS = tstthreads t-server TESTS=$(check_PROGRAMS) yazproxy_SOURCES=yaz-proxy-main.cpp cdetails_SOURCES=cdetails.cpp tstthreads_SOURCES=tstthreads.cpp +t_server_SOURCES=t-server.cpp LDADD=libyazproxy.la $(YAZPPLALIB) $(XSLT_LIBS) $(USEMARCONLALIB) libyazproxy_la_LIBADD = $(XSLT_LIBS) diff --git a/src/t-server.cpp b/src/t-server.cpp new file mode 100644 index 0000000..5489f59 --- /dev/null +++ b/src/t-server.cpp @@ -0,0 +1,272 @@ +/* + * Copyright (c) 1998-2005, Index Data. + * See the file LICENSE for details. + * + * $Id: t-server.cpp,v 1.1 2005-06-21 21:54:05 adam Exp $ + */ + +#include +#include +#include +#include +#include +#include "msg-thread.h" +#include +#include +#include +#include + +using namespace yazpp_1; + +class Mutex { +public: + Mutex(); + ~Mutex(); + void lock(); + void unlock(); +private: + pthread_mutex_t m_mutex; + +}; + +Mutex::Mutex() +{ + pthread_mutex_init(&m_mutex, 0); +} + +Mutex::~Mutex() +{ + pthread_mutex_destroy(&m_mutex); +} + +void Mutex::lock() +{ + pthread_mutex_lock(&m_mutex); +} + +void Mutex::unlock() +{ + pthread_mutex_unlock(&m_mutex); +} + + +class MyServer; + +class Auth_Msg : public IMsg_Thread { +public: + int m_close_flag; + GDU *m_gdu; + GDU *m_output; + MyServer *m_front; + IMsg_Thread *handle(); + void result(); + Auth_Msg(GDU *gdu, MyServer *front); + virtual ~Auth_Msg(); +}; + +Auth_Msg::Auth_Msg(GDU *gdu, MyServer *front) +{ + m_front = front; + m_output = 0; + m_gdu = gdu; + m_close_flag = 0; +} + +Auth_Msg::~Auth_Msg() +{ + delete m_output; + delete m_gdu; +} + +IMsg_Thread *Auth_Msg::handle() +{ + ODR odr = odr_createmem(ODR_ENCODE); + yaz_log(YLOG_LOG, "Auth_Msg:handle begin"); + Z_GDU *z_gdu = m_gdu->get(); + if (z_gdu->which == Z_GDU_Z3950) + { + Z_APDU *apdu = 0; + switch(z_gdu->u.z3950->which) + { + case Z_APDU_initRequest: + apdu = zget_APDU(odr, Z_APDU_initResponse); + break; + case Z_APDU_searchRequest: + sleep(5); + apdu = zget_APDU(odr, Z_APDU_searchResponse); + break; + default: + apdu = zget_APDU(odr, Z_APDU_close); + m_close_flag = 1; + break; + } + if (apdu) + m_output = new GDU(apdu); + } + yaz_log(YLOG_LOG, "Auth_Msg:handle end"); + odr_destroy(odr); + return this; +} + +class MyServer : public Z_Assoc { +public: + ~MyServer(); + MyServer(IPDU_Observable *the_PDU_Observable, + Msg_Thread *m_my_thread + ); + IPDU_Observer* sessionNotify(IPDU_Observable *the_PDU_Observable, + int fd); + + void recv_GDU(Z_GDU *apdu, int len); + + void failNotify(); + void timeoutNotify(); + void connectNotify(); + + int m_no_requests; + int m_delete_flag; +private: + yazpp_1::GDUQueue m_in_queue; + Msg_Thread *m_my_thread; +}; + +void Auth_Msg::result() +{ + m_front->m_no_requests--; + if (!m_front->m_delete_flag) + { + if (m_output) + { + int len; + m_front->send_GDU(m_output->get(), &len); + } + if (m_close_flag) + { + m_front->close(); + m_front->m_delete_flag = 1; + } + } + if (m_front->m_delete_flag && m_front->m_no_requests == 0) + delete m_front; +} + +MyServer::MyServer(IPDU_Observable *the_PDU_Observable, + Msg_Thread *my_thread +) + : Z_Assoc(the_PDU_Observable) +{ + m_my_thread = my_thread; + m_no_requests = 0; + m_delete_flag = 0; + yaz_log(YLOG_LOG, "Construct Myserver=%p", this); +} + +IPDU_Observer *MyServer::sessionNotify(IPDU_Observable + *the_PDU_Observable, int fd) +{ + MyServer *my = new MyServer(the_PDU_Observable, m_my_thread); + yaz_log(YLOG_LOG, "New session %s", the_PDU_Observable->getpeername()); + return my; +} + +MyServer::~MyServer() +{ + yaz_log(YLOG_LOG, "Destroy Myserver=%p", this); +} + +void MyServer::recv_GDU(Z_GDU *apdu, int len) +{ + GDU *gdu = new GDU(apdu); + Auth_Msg *m = new Auth_Msg(gdu, this); + m_no_requests++; + m_my_thread->put(m); +} + +void MyServer::failNotify() +{ + m_delete_flag = 1; + if (m_no_requests == 0) + delete this; + +} + +void MyServer::timeoutNotify() +{ + m_delete_flag = 1; + if (m_no_requests == 0) + delete this; +} + +void MyServer::connectNotify() +{ + +} + +void usage(const char *prog) +{ + fprintf (stderr, "%s: [-a log] [-v level] [-T] @:port\n", prog); + exit (1); +} + +int main(int argc, char **argv) +{ + char *arg; + char *prog = *argv; + int thread_flag = 0; + int ret; + const char *addr = "tcp:@:9999"; + char *apdu_log = 0; + + while ((ret = options("a:v:T", argv, argc, &arg)) != -2) + { + switch (ret) + { + case 0: + addr = xstrdup(arg); + break; + case 'a': + apdu_log = xstrdup(arg); + break; + case 'v': + yaz_log_init_level (yaz_log_mask_str(arg)); + break; + case 'T': + thread_flag = 1; + break; + default: + usage(prog); + return 1; + } + } + + SocketManager mySocketManager; + + PDU_Assoc *my_PDU_Assoc = 0; + + MyServer *z = 0; + + Msg_Thread *my_thread = new Msg_Thread(&mySocketManager); + +#if YAZ_POSIX_THREADS + if (thread_flag) + my_PDU_Assoc = new PDU_AssocThread(&mySocketManager); + else + my_PDU_Assoc = new PDU_Assoc(&mySocketManager); +#else + my_PDU_Assoc = new PDU_Assoc(&mySocketManager); +#endif + + z = new MyServer(my_PDU_Assoc, my_thread); + z->server(addr); + if (apdu_log) + { + yaz_log (YLOG_LOG, "set_APDU_log %s", apdu_log); + z->set_APDU_log(apdu_log); + } + + while (mySocketManager.processEvent() > 0) + ; + delete z; + delete my_thread; + return 0; +} -- 1.7.10.4