2 * Copyright (c) 1998-2001, Index Data.
3 * See the file LICENSE for details.
5 * $Log: yaz-pdu-assoc.cpp,v $
6 * Revision 1.24 2001-08-13 16:39:12 adam
7 * PDU_Assoc keeps track of children. Using yaz_log instead of logf.
9 * Revision 1.23 2001/03/26 14:43:49 adam
10 * New threaded PDU association.
12 * Revision 1.22 2001/01/29 11:18:24 adam
13 * Server sets OPTIONS search and present.
15 * Revision 1.21 2000/11/20 14:17:36 adam
16 * Yet another WIN32 fix for connect notify.
18 * Revision 1.20 2000/11/20 11:27:33 adam
19 * Fixes for connect operation (timeout and notify fix).
21 * Revision 1.19 2000/11/01 14:22:59 adam
22 * Added fd parameter for method IYaz_PDU_Observer::clone.
24 * Revision 1.18 2000/10/24 12:29:57 adam
25 * Fixed bug in proxy where a Yaz_ProxyClient could be owned by
26 * two Yaz_Proxy's (fatal).
28 * Revision 1.17 2000/10/11 11:58:16 adam
29 * Moved header files to include/yaz++. Switched to libtool and automake.
30 * Configure script creates yaz++-config script.
32 * Revision 1.16 2000/09/22 09:54:11 heikki
35 * Revision 1.15 2000/09/21 21:43:20 adam
36 * Better high-level server API.
38 * Revision 1.14 2000/09/12 12:09:53 adam
39 * More work on high-level server.
41 * Revision 1.13 2000/09/08 10:23:42 adam
42 * Added skeleton of yaz-z-server.
44 * Revision 1.12 2000/09/06 14:23:45 adam
47 * Revision 1.11 2000/09/04 08:29:22 adam
48 * Fixed memory leak(s). Added re-use of associations, rather than
49 * re-init, when maximum number of targets are in use.
51 * Revision 1.10 2000/08/10 08:42:42 adam
52 * Fixes for {set,get}_APDU_log.
54 * Revision 1.9 1999/12/06 13:52:45 adam
55 * Modified for new location of YAZ header files. Experimental threaded
58 * Revision 1.8 1999/04/28 13:04:03 adam
59 * Fixed setting of proxy otherInfo so that database(s) are removed.
61 * Revision 1.7 1999/04/21 12:09:01 adam
62 * Many improvements. Modified to proxy server to work with "sessions"
65 * Revision 1.6 1999/04/20 10:30:05 adam
66 * Implemented various stuff for client and proxy. Updated calls
67 * to ODR to reflect new name parameter.
69 * Revision 1.5 1999/04/09 11:46:57 adam
70 * Added object Yaz_Z_Assoc. Much more functional client.
72 * Revision 1.4 1999/03/23 14:17:57 adam
73 * More work on timeout handling. Work on yaz-client.
75 * Revision 1.3 1999/02/02 14:01:20 adam
76 * First WIN32 port of YAZ++.
78 * Revision 1.2 1999/01/28 13:08:44 adam
79 * Yaz_PDU_Assoc better encapsulated. Memory leak fix in
80 * yaz-socket-manager.cc.
82 * Revision 1.1.1.1 1999/01/28 09:41:07 adam
83 * First implementation of YAZ++.
90 #include <yaz/tcpip.h>
92 #include <yaz++/yaz-pdu-assoc.h>
95 void Yaz_PDU_Assoc::init(IYazSocketObservable *socketObservable)
99 m_socketObservable = socketObservable;
112 Yaz_PDU_Assoc::Yaz_PDU_Assoc(IYazSocketObservable *socketObservable)
114 init (socketObservable);
117 Yaz_PDU_Assoc::Yaz_PDU_Assoc(IYazSocketObservable *socketObservable,
120 init(socketObservable);
123 if (cs->io_pending & CS_WANT_WRITE)
124 mask |= YAZ_SOCKET_OBSERVE_WRITE;
125 if (cs->io_pending & CS_WANT_READ)
126 mask |= YAZ_SOCKET_OBSERVE_READ;
127 m_socketObservable->addObserver(cs_fileno(cs), this);
130 yaz_log (m_log, "new PDU_Assoc. Ready");
136 yaz_log (m_log, "new PDU_Assoc. Accepting");
137 // assume comstack is accepting...
139 m_socketObservable->addObserver(cs_fileno(cs), this);
140 m_socketObservable->maskObserver(this,
141 mask |YAZ_SOCKET_OBSERVE_EXCEPT);
146 IYaz_PDU_Observable *Yaz_PDU_Assoc::clone()
148 Yaz_PDU_Assoc *copy = new Yaz_PDU_Assoc(m_socketObservable);
152 void Yaz_PDU_Assoc::socketNotify(int event)
154 yaz_log (m_log, "Yaz_PDU_Assoc::socketNotify p=%p state=%d event = %d",
155 this, m_state, event);
156 if (event & YAZ_SOCKET_OBSERVE_EXCEPT)
159 m_PDU_Observer->failNotify();
162 else if (event & YAZ_SOCKET_OBSERVE_TIMEOUT)
164 m_PDU_Observer->timeoutNotify();
170 if (!cs_accept (m_cs))
172 yaz_log (m_log, "Yaz_PDU_Assoc::cs_accept failed");
175 m_PDU_Observer->failNotify();
180 if (m_cs->io_pending & CS_WANT_WRITE)
181 mask |= YAZ_SOCKET_OBSERVE_WRITE;
182 if (m_cs->io_pending & CS_WANT_READ)
183 mask |= YAZ_SOCKET_OBSERVE_READ;
185 { // accept is complete. turn to ready state and write if needed
190 { // accept still incomplete.
191 m_socketObservable->maskObserver(this,
192 mask|YAZ_SOCKET_OBSERVE_EXCEPT);
197 if (event & YAZ_SOCKET_OBSERVE_READ &&
198 event & YAZ_SOCKET_OBSERVE_WRITE)
200 // For Unix: if both read and write is set, then connect failed.
202 m_PDU_Observer->failNotify();
206 yaz_log (m_log, "cs_connect again");
207 int res = cs_connect (m_cs, 0);
210 unsigned mask = YAZ_SOCKET_OBSERVE_EXCEPT;
211 if (m_cs->io_pending & CS_WANT_WRITE)
212 mask |= YAZ_SOCKET_OBSERVE_WRITE;
213 if (m_cs->io_pending & CS_WANT_READ)
214 mask |= YAZ_SOCKET_OBSERVE_READ;
215 m_socketObservable->maskObserver(this, mask);
221 m_PDU_Observer->connectNotify();
227 if (event & YAZ_SOCKET_OBSERVE_READ)
232 if ((res = cs_listen(m_cs, 0, 0)) == 1)
236 yaz_log(LOG_FATAL|LOG_ERRNO, "cs_listen failed");
239 if (!(new_line = cs_accept(m_cs)))
241 /* 1. create socket-manager
243 3. create top-level object
244 setup observer for child fileid in pdu-assoc
247 yaz_log (m_log, "new session: parent fd=%d child fd=%d",
248 cs_fileno(m_cs), cs_fileno(new_line));
249 childNotify (new_line);
253 if (event & (YAZ_SOCKET_OBSERVE_READ|YAZ_SOCKET_OBSERVE_WRITE))
257 if (event & (YAZ_SOCKET_OBSERVE_READ|YAZ_SOCKET_OBSERVE_WRITE))
261 int res = cs_get (m_cs, &m_input_buf, &m_input_len);
264 unsigned mask = YAZ_SOCKET_OBSERVE_EXCEPT;
265 if (m_cs->io_pending & CS_WANT_WRITE)
266 mask |= YAZ_SOCKET_OBSERVE_WRITE;
267 if (m_cs->io_pending & CS_WANT_READ)
268 mask |= YAZ_SOCKET_OBSERVE_READ;
269 m_socketObservable->maskObserver(this, mask);
274 yaz_log (m_log, "Yaz_PDU_Assoc::Connection closed by peer");
277 m_PDU_Observer->failNotify(); // problem here..
280 // lock it, so we know if recv_PDU deletes it.
282 m_destroyed = &destroyed;
287 m_PDU_Observer->recv_PDU(m_input_buf, res);
289 if (destroyed) // it really was destroyed, return now.
291 } while (m_cs && cs_more (m_cs));
293 m_socketObservable->maskObserver(this,
294 YAZ_SOCKET_OBSERVE_EXCEPT|
295 YAZ_SOCKET_OBSERVE_READ);
299 yaz_log (m_log, "CLOSING state=%d event was %d", m_state, event);
301 m_PDU_Observer->failNotify();
304 yaz_log (m_log, "Unknown state=%d event was %d", m_state, event);
306 m_PDU_Observer->failNotify();
310 void Yaz_PDU_Assoc::close()
313 for (ch = m_children; ch; ch = ch->m_next)
316 m_socketObservable->deleteObserver(this);
320 yaz_log (m_log, "Yaz_PDU_Assoc::close fd=%d", cs_fileno(m_cs));
326 PDU_Queue *q_this = m_queue_out;
327 m_queue_out = m_queue_out->m_next;
335 void Yaz_PDU_Assoc::destroy()
343 // delete from parent's child list (if any)
346 c = &m_parent->m_children;
354 // delete all children ...
358 Yaz_PDU_Assoc *here = *c;
363 yaz_log (m_log, "Yaz_PDU_Assoc::destroy this=%p", this);
366 Yaz_PDU_Assoc::PDU_Queue::PDU_Queue(const char *buf, int len)
368 m_buf = (char *) malloc (len);
369 memcpy (m_buf, buf, len);
374 Yaz_PDU_Assoc::PDU_Queue::~PDU_Queue()
379 int Yaz_PDU_Assoc::flush_PDU()
383 if (m_state != Ready && m_state != Writing)
385 yaz_log (m_log, "YAZ_PDU_Assoc::flush_PDU, not ready");
388 PDU_Queue *q = m_queue_out;
392 yaz_log (m_log, "YAZ_PDU_Assoc::flush_PDU queue empty");
393 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
394 YAZ_SOCKET_OBSERVE_WRITE|
395 YAZ_SOCKET_OBSERVE_EXCEPT);
398 r = cs_put (m_cs, q->m_buf, q->m_len);
401 yaz_log (m_log, "Yaz_PDU_Assoc::flush_PDU cs_put failed");
403 m_PDU_Observer->failNotify();
408 unsigned mask = YAZ_SOCKET_OBSERVE_EXCEPT;
410 if (m_cs->io_pending & CS_WANT_WRITE)
411 mask |= YAZ_SOCKET_OBSERVE_WRITE;
412 if (m_cs->io_pending & CS_WANT_READ)
413 mask |= YAZ_SOCKET_OBSERVE_READ;
415 m_socketObservable->maskObserver(this, mask);
416 yaz_log (m_log, "Yaz_PDU_Assoc::flush_PDU cs_put %d bytes (incomp)",
421 yaz_log (m_log, "Yaz_PDU_Assoc::flush_PDU cs_put %d bytes", q->m_len);
422 // whole packet sent... delete this and proceed to next ...
423 m_queue_out = q->m_next;
425 // don't select on write if queue is empty ...
427 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
428 YAZ_SOCKET_OBSERVE_EXCEPT);
432 int Yaz_PDU_Assoc::send_PDU(const char *buf, int len)
434 yaz_log (m_log, "Yaz_PDU_Assoc::send_PDU");
435 PDU_Queue **pq = &m_queue_out;
436 int is_idle = (*pq ? 0 : 1);
440 yaz_log (m_log, "Yaz_PDU_Assoc::send_PDU failed, m_cs == 0");
445 *pq = new PDU_Queue(buf, len);
449 yaz_log (m_log, "Yaz_PDU_Assoc::cannot send_PDU fd=%d",
454 COMSTACK Yaz_PDU_Assoc::comstack(const char *type_and_host, void **vp)
456 return cs_create_host(type_and_host, 0, vp);
459 void Yaz_PDU_Assoc::listen(IYaz_PDU_Observer *observer,
464 yaz_log (LOG_LOG, "Adding listener %s", addr);
466 m_PDU_Observer = observer;
468 m_cs = comstack(addr, &ap);
472 if (cs_bind(m_cs, ap, CS_SERVER) < 0)
474 m_socketObservable->addObserver(cs_fileno(m_cs), this);
475 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
476 YAZ_SOCKET_OBSERVE_EXCEPT);
477 yaz_log (m_log, "Yaz_PDU_Assoc::listen ok fd=%d", cs_fileno(m_cs));
481 void Yaz_PDU_Assoc::idleTime(int idleTime)
483 m_idleTime = idleTime;
484 yaz_log (m_log, "Yaz_PDU_Assoc::idleTime(%d)", idleTime);
485 m_socketObservable->timeoutObserver(this, m_idleTime);
488 void Yaz_PDU_Assoc::connect(IYaz_PDU_Observer *observer,
491 yaz_log (m_log, "Yaz_PDU_Assoc::connect %s", addr);
493 m_PDU_Observer = observer;
495 m_cs = comstack(addr, &ap);
496 int res = cs_connect (m_cs, ap);
497 yaz_log (m_log, "Yaz_PDU_Assoc::connect fd=%d res=%d", cs_fileno(m_cs),
499 m_socketObservable->addObserver(cs_fileno(m_cs), this);
502 { // Connect pending or complet
503 m_state = Connecting;
504 unsigned mask = YAZ_SOCKET_OBSERVE_EXCEPT;
505 if (m_cs->io_pending & CS_WANT_WRITE)
506 mask |= YAZ_SOCKET_OBSERVE_WRITE;
507 if (m_cs->io_pending & CS_WANT_READ)
508 mask |= YAZ_SOCKET_OBSERVE_READ;
509 m_socketObservable->maskObserver(this, mask);
512 { // Connect failed immediately
513 // Since m_state is Closed we can distinguish this case from
514 // normal connect in socketNotify handler
515 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_WRITE|
516 YAZ_SOCKET_OBSERVE_EXCEPT);
520 // Single-threaded... Only useful for non-blocking handlers
521 void Yaz_PDU_Assoc::childNotify(COMSTACK cs)
525 Yaz_PDU_Assoc *new_observable =
526 new Yaz_PDU_Assoc (m_socketObservable, cs);
528 new_observable->m_next = m_children;
529 m_children = new_observable;
530 new_observable->m_parent = this;
532 // Clone PDU Observer
533 new_observable->m_PDU_Observer = m_PDU_Observer->sessionNotify
534 (new_observable, cs_fileno(cs));