2 * Copyright (c) 1998-2000, Index Data.
3 * See the file LICENSE for details.
5 * $Log: yaz-pdu-assoc.cpp,v $
6 * Revision 1.21 2000-11-20 14:17:36 adam
7 * Yet another WIN32 fix for connect notify.
9 * Revision 1.20 2000/11/20 11:27:33 adam
10 * Fixes for connect operation (timeout and notify fix).
12 * Revision 1.19 2000/11/01 14:22:59 adam
13 * Added fd parameter for method IYaz_PDU_Observer::clone.
15 * Revision 1.18 2000/10/24 12:29:57 adam
16 * Fixed bug in proxy where a Yaz_ProxyClient could be owned by
17 * two Yaz_Proxy's (fatal).
19 * Revision 1.17 2000/10/11 11:58:16 adam
20 * Moved header files to include/yaz++. Switched to libtool and automake.
21 * Configure script creates yaz++-config script.
23 * Revision 1.16 2000/09/22 09:54:11 heikki
26 * Revision 1.15 2000/09/21 21:43:20 adam
27 * Better high-level server API.
29 * Revision 1.14 2000/09/12 12:09:53 adam
30 * More work on high-level server.
32 * Revision 1.13 2000/09/08 10:23:42 adam
33 * Added skeleton of yaz-z-server.
35 * Revision 1.12 2000/09/06 14:23:45 adam
38 * Revision 1.11 2000/09/04 08:29:22 adam
39 * Fixed memory leak(s). Added re-use of associations, rather than
40 * re-init, when maximum number of targets are in use.
42 * Revision 1.10 2000/08/10 08:42:42 adam
43 * Fixes for {set,get}_APDU_log.
45 * Revision 1.9 1999/12/06 13:52:45 adam
46 * Modified for new location of YAZ header files. Experimental threaded
49 * Revision 1.8 1999/04/28 13:04:03 adam
50 * Fixed setting of proxy otherInfo so that database(s) are removed.
52 * Revision 1.7 1999/04/21 12:09:01 adam
53 * Many improvements. Modified to proxy server to work with "sessions"
56 * Revision 1.6 1999/04/20 10:30:05 adam
57 * Implemented various stuff for client and proxy. Updated calls
58 * to ODR to reflect new name parameter.
60 * Revision 1.5 1999/04/09 11:46:57 adam
61 * Added object Yaz_Z_Assoc. Much more functional client.
63 * Revision 1.4 1999/03/23 14:17:57 adam
64 * More work on timeout handling. Work on yaz-client.
66 * Revision 1.3 1999/02/02 14:01:20 adam
67 * First WIN32 port of YAZ++.
69 * Revision 1.2 1999/01/28 13:08:44 adam
70 * Yaz_PDU_Assoc better encapsulated. Memory leak fix in
71 * yaz-socket-manager.cc.
73 * Revision 1.1.1.1 1999/01/28 09:41:07 adam
74 * First implementation of YAZ++.
80 #include <yaz++/yaz-pdu-assoc.h>
83 #include <yaz/tcpip.h>
85 Yaz_PDU_Assoc::Yaz_PDU_Assoc(IYazSocketObservable *socketObservable)
89 m_socketObservable = socketObservable;
102 IYaz_PDU_Observable *Yaz_PDU_Assoc::clone()
104 Yaz_PDU_Assoc *copy = new Yaz_PDU_Assoc(m_socketObservable);
108 void Yaz_PDU_Assoc::socketNotify(int event)
110 logf (m_log, "Yaz_PDU_Assoc::socketNotify p=%p event = %d", this, event);
114 if (event & (YAZ_SOCKET_OBSERVE_READ|YAZ_SOCKET_OBSERVE_EXCEPT))
117 m_PDU_Observer->failNotify();
119 else if (event & YAZ_SOCKET_OBSERVE_TIMEOUT)
121 m_PDU_Observer->timeoutNotify();
126 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
127 YAZ_SOCKET_OBSERVE_EXCEPT);
128 m_PDU_Observer->connectNotify();
133 if (event & YAZ_SOCKET_OBSERVE_READ)
138 if ((res = cs_listen(m_cs, 0, 0)) == 1)
142 logf(LOG_FATAL|LOG_ERRNO, "cs_listen failed");
145 if (!(new_line = cs_accept(m_cs)))
147 /* 1. create socket-manager
149 3. create top-level object
150 setup observer for child fileid in pdu-assoc
153 int fd = cs_fileno(new_line);
154 logf (m_log, "accept ok fd = %d", fd);
155 cs_fileno(new_line) = -1;
161 if (event & YAZ_SOCKET_OBSERVE_WRITE)
165 if (event & YAZ_SOCKET_OBSERVE_READ)
169 int res = cs_get (m_cs, &m_input_buf, &m_input_len);
174 logf (m_log, "Connection closed by peer");
176 m_PDU_Observer->failNotify(); // problem here..
179 // lock it, so we know if recv_PDU deletes it.
181 m_destroyed = &destroyed;
183 m_PDU_Observer->recv_PDU(m_input_buf, res);
185 if (destroyed) // it really was destroyed, return now.
187 } while (m_cs && cs_more (m_cs));
189 if (event & YAZ_SOCKET_OBSERVE_TIMEOUT)
191 m_PDU_Observer->timeoutNotify();
195 logf (m_log, "CLOSING state=%d event was %d", m_state, event);
197 m_PDU_Observer->failNotify();
200 logf (m_log, "Unknown state=%d event was %d", m_state, event);
202 m_PDU_Observer->failNotify();
206 void Yaz_PDU_Assoc::close()
208 m_socketObservable->deleteObserver(this);
212 logf (m_log, "Yaz_PDU_Assoc::close fd=%d", cs_fileno(m_cs));
218 PDU_Queue *q_this = m_queue_out;
219 m_queue_out = m_queue_out->m_next;
227 void Yaz_PDU_Assoc::destroy()
234 // delete from parent's child list (if any)
237 c = &m_parent->m_children;
245 // delete all children ...
249 Yaz_PDU_Assoc *here = *c;
256 Yaz_PDU_Assoc::PDU_Queue::PDU_Queue(const char *buf, int len)
258 m_buf = (char *) malloc (len);
259 memcpy (m_buf, buf, len);
264 Yaz_PDU_Assoc::PDU_Queue::~PDU_Queue()
269 int Yaz_PDU_Assoc::flush_PDU()
273 logf (m_log, "Yaz_PDU_Assoc::flush_PDU");
274 if (m_state != Ready)
276 logf (m_log, "YAZ_PDU_Assoc::flush_PDU, not ready");
279 PDU_Queue *q = m_queue_out;
282 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
283 YAZ_SOCKET_OBSERVE_EXCEPT);
286 r = cs_put (m_cs, q->m_buf, q->m_len);
290 m_PDU_Observer->failNotify();
295 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
296 YAZ_SOCKET_OBSERVE_EXCEPT|
297 YAZ_SOCKET_OBSERVE_WRITE);
298 logf (m_log, "Yaz_PDU_Assoc::flush_PDU put %d bytes (incomplete)",
302 logf (m_log, "Yaz_PDU_Assoc::flush_PDU put %d bytes", q->m_len);
303 // whole packet sent... delete this and proceed to next ...
304 m_queue_out = q->m_next;
306 // don't select on write if queue is empty ...
308 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
309 YAZ_SOCKET_OBSERVE_EXCEPT);
313 int Yaz_PDU_Assoc::send_PDU(const char *buf, int len)
315 logf (m_log, "Yaz_PDU_Assoc::send_PDU");
316 PDU_Queue **pq = &m_queue_out;
317 int is_idle = (*pq ? 0 : 1);
321 logf (m_log, "Yaz_PDU_Assoc::send_PDU failed, m_cs == 0");
326 *pq = new PDU_Queue(buf, len);
330 logf (m_log, "Yaz_PDU_Assoc::cannot send_PDU fd=%d",
335 COMSTACK Yaz_PDU_Assoc::comstack()
339 CS_TYPE cs_type = tcpip_type;
340 m_cs = cs_create (cs_type, 0, PROTO_Z3950);
345 void Yaz_PDU_Assoc::listen(IYaz_PDU_Observer *observer,
350 COMSTACK cs = comstack();
352 logf (m_log, "Yaz_PDU_Assoc::listen %s", addr);
353 m_PDU_Observer = observer;
356 ap = cs_straddr (cs, addr);
359 if (cs_bind(cs, ap, CS_SERVER) < 0)
361 m_socketObservable->addObserver(cs_fileno(cs), this);
362 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
363 YAZ_SOCKET_OBSERVE_EXCEPT);
364 logf (m_log, "Yaz_PDU_Assoc::listen ok fd=%d", cs_fileno(cs));
368 void Yaz_PDU_Assoc::idleTime(int idleTime)
370 m_idleTime = idleTime;
371 logf (m_log, "Yaz_PDU_Assoc::idleTime(%d)", idleTime);
372 m_socketObservable->timeoutObserver(this, m_idleTime);
375 void Yaz_PDU_Assoc::connect(IYaz_PDU_Observer *observer,
378 logf (m_log, "Yaz_PDU_Assoc::connect %s", addr);
380 m_PDU_Observer = observer;
381 COMSTACK cs = comstack();
382 void *ap = cs_straddr (cs, addr);
385 logf (m_log, "cs_straddr failed");
388 int res = cs_connect (cs, ap);
389 logf (m_log, "Yaz_PDU_Assoc::connect fd=%d res=%d", cs_fileno(cs), res);
390 m_socketObservable->addObserver(cs_fileno(cs), this);
391 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
392 YAZ_SOCKET_OBSERVE_EXCEPT|
393 YAZ_SOCKET_OBSERVE_WRITE);
395 m_state = Connecting;
396 // if res < 0, then cs_connect failed immediately -> m_state is Closed..
399 void Yaz_PDU_Assoc::socket(IYaz_PDU_Observer *observer, int fd)
402 m_PDU_Observer = observer;
405 CS_TYPE cs_type = tcpip_type;
406 m_cs = cs_createbysocket(fd, cs_type, 0, PROTO_Z3950);
408 m_socketObservable->addObserver(fd, this);
409 m_socketObservable->maskObserver(this,
410 YAZ_SOCKET_OBSERVE_READ|
411 YAZ_SOCKET_OBSERVE_EXCEPT);
412 m_socketObservable->timeoutObserver(this, m_idleTime);
417 // 1 = single-threaded
418 // 0 = multi-threaded
420 // Single-threaded... Only useful for non-blocking handlers
421 void Yaz_PDU_Assoc::childNotify(int fd)
423 // Clone PDU Observable (keep socket manager)
424 IYaz_PDU_Observable *new_observable = clone();
426 // Clone PDU Observer
427 IYaz_PDU_Observer *observer = m_PDU_Observer->clone(new_observable, fd);
429 // Attach new socket to it
430 new_observable->socket(observer, fd);
434 #include <yaz++/yaz-socket-manager.h>
449 Yaz_SocketManager *s = (Yaz_SocketManager *) p;
451 logf (LOG_LOG, "thread started");
452 while (s->processEvent() > 0)
454 logf (LOG_LOG, "thread finished");
461 void Yaz_PDU_Assoc::childNotify(int fd)
463 Yaz_SocketManager *socket_observable = new Yaz_SocketManager;
464 Yaz_PDU_Assoc *new_observable = new Yaz_PDU_Assoc (socket_observable);
466 /// Clone PDU Observer
467 IYaz_PDU_Observer *observer = m_PDU_Observer->clone(new_observable, fd);
469 /// Attach new socket to it
470 new_observable->socket(observer, fd);
474 t_id = _beginthread (events, 0, socket_observable);
477 logf (LOG_FATAL|LOG_ERRNO, "_beginthread failed");
483 int id = pthread_create (&type, 0, events, socket_observable);
484 logf (LOG_LOG, "pthread_create returned id=%d", id);