2 * Copyright (c) 1998-2005, Index Data.
3 * See the file LICENSE for details.
5 * $Id: t-server.cpp,v 1.5 2005-09-26 09:24:14 adam Exp $
11 #include <yaz/diagbib1.h>
12 #include <yaz/options.h>
13 #include "msg-thread.h"
14 #include <yaz++/z-assoc.h>
15 #include <yaz++/pdu-assoc.h>
16 #include <yaz++/gdu.h>
17 #include <yaz++/socket-manager.h>
19 using namespace yazpp_1;
23 class Auth_Msg : public IMsg_Thread {
29 IMsg_Thread *handle();
31 Auth_Msg(GDU *gdu, MyServer *front);
35 Auth_Msg::Auth_Msg(GDU *gdu, MyServer *front)
49 IMsg_Thread *Auth_Msg::handle()
51 ODR odr = odr_createmem(ODR_ENCODE);
52 yaz_log(YLOG_LOG, "Auth_Msg:handle begin");
53 Z_GDU *z_gdu = m_gdu->get();
54 if (z_gdu->which == Z_GDU_Z3950)
57 switch(z_gdu->u.z3950->which)
59 case Z_APDU_initRequest:
60 apdu = zget_APDU(odr, Z_APDU_initResponse);
61 ODR_MASK_SET(apdu->u.initResponse->options, Z_Options_triggerResourceCtrl);
62 ODR_MASK_SET(apdu->u.initResponse->options, Z_Options_search);
63 ODR_MASK_SET(apdu->u.initResponse->options, Z_Options_present);
65 case Z_APDU_searchRequest:
67 apdu = zget_APDU(odr, Z_APDU_searchResponse);
69 case Z_APDU_triggerResourceControlRequest:
72 apdu = zget_APDU(odr, Z_APDU_close);
77 m_output = new GDU(apdu);
79 yaz_log(YLOG_LOG, "Auth_Msg:handle end");
84 class MyServer : public Z_Assoc {
87 MyServer(IPDU_Observable *the_PDU_Observable,
88 Msg_Thread *m_my_thread
90 IPDU_Observer* sessionNotify(IPDU_Observable *the_PDU_Observable,
93 void recv_GDU(Z_GDU *apdu, int len);
102 yazpp_1::GDUQueue m_in_queue;
103 Msg_Thread *m_my_thread;
106 void Auth_Msg::result()
108 m_front->m_no_requests--;
109 if (!m_front->m_delete_flag)
114 m_front->send_GDU(m_output->get(), &len);
119 m_front->m_delete_flag = 1;
122 if (m_front->m_delete_flag && m_front->m_no_requests == 0)
127 MyServer::MyServer(IPDU_Observable *the_PDU_Observable,
128 Msg_Thread *my_thread
130 : Z_Assoc(the_PDU_Observable)
132 m_my_thread = my_thread;
135 yaz_log(YLOG_LOG, "Construct Myserver=%p", this);
138 IPDU_Observer *MyServer::sessionNotify(IPDU_Observable
139 *the_PDU_Observable, int fd)
141 MyServer *my = new MyServer(the_PDU_Observable, m_my_thread);
142 yaz_log(YLOG_LOG, "New session %s", the_PDU_Observable->getpeername());
146 MyServer::~MyServer()
148 yaz_log(YLOG_LOG, "Destroy Myserver=%p", this);
151 void MyServer::recv_GDU(Z_GDU *apdu, int len)
153 GDU *gdu = new GDU(apdu);
154 Auth_Msg *m = new Auth_Msg(gdu, this);
159 void MyServer::failNotify()
162 if (m_no_requests == 0)
167 void MyServer::timeoutNotify()
170 if (m_no_requests == 0)
174 void MyServer::connectNotify()
179 void usage(const char *prog)
181 fprintf (stderr, "%s: [-a log] [-v level] [-T] @:port\n", prog);
185 int main(int argc, char **argv)
191 const char *addr = "tcp:@:9999";
195 while ((ret = options("n:a:v:T", argv, argc, &arg)) != -2)
203 no_threads = atoi(arg);
206 apdu_log = xstrdup(arg);
209 yaz_log_init_level (yaz_log_mask_str(arg));
220 SocketManager mySocketManager;
222 PDU_Assoc *my_PDU_Assoc = 0;
226 Msg_Thread *my_thread = new Msg_Thread(&mySocketManager, no_threads);
228 #if YAZ_POSIX_THREADS
230 my_PDU_Assoc = new PDU_AssocThread(&mySocketManager);
232 my_PDU_Assoc = new PDU_Assoc(&mySocketManager);
234 my_PDU_Assoc = new PDU_Assoc(&mySocketManager);
237 z = new MyServer(my_PDU_Assoc, my_thread);
241 yaz_log (YLOG_LOG, "set_APDU_log %s", apdu_log);
242 z->set_APDU_log(apdu_log);
245 while (mySocketManager.processEvent() > 0)
254 * indent-tabs-mode: nil
256 * vim: shiftwidth=4 tabstop=8 expandtab