+/*
+ * 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 <stdlib.h>
+#include <pthread.h>
+#include <yaz/log.h>
+#include <yaz/diagbib1.h>
+#include <yaz/options.h>
+#include "msg-thread.h"
+#include <yaz++/z-assoc.h>
+#include <yaz++/pdu-assoc.h>
+#include <yaz++/gdu.h>
+#include <yaz++/socket-manager.h>
+
+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;
+}
projects / yazproxy-moved-to-github.git / commitdiff