2 * Copyright (c) 1998-2001, Index Data.
3 * See the file LICENSE for details.
5 * $Id: yaz-pdu-assoc.cpp,v 1.28 2002-10-09 12:50:26 adam Exp $
11 #include <yaz/tcpip.h>
13 #include <yaz++/pdu-assoc.h>
16 void Yaz_PDU_Assoc::init(IYazSocketObservable *socketObservable)
20 m_socketObservable = socketObservable;
33 Yaz_PDU_Assoc::Yaz_PDU_Assoc(IYazSocketObservable *socketObservable)
35 init (socketObservable);
38 Yaz_PDU_Assoc::Yaz_PDU_Assoc(IYazSocketObservable *socketObservable,
41 init(socketObservable);
44 if (cs->io_pending & CS_WANT_WRITE)
45 mask |= YAZ_SOCKET_OBSERVE_WRITE;
46 if (cs->io_pending & CS_WANT_READ)
47 mask |= YAZ_SOCKET_OBSERVE_READ;
48 m_socketObservable->addObserver(cs_fileno(cs), this);
51 yaz_log (m_log, "new PDU_Assoc. Ready");
57 yaz_log (m_log, "new PDU_Assoc. Accepting");
58 // assume comstack is accepting...
60 m_socketObservable->addObserver(cs_fileno(cs), this);
61 m_socketObservable->maskObserver(this,
62 mask |YAZ_SOCKET_OBSERVE_EXCEPT);
67 IYaz_PDU_Observable *Yaz_PDU_Assoc::clone()
69 Yaz_PDU_Assoc *copy = new Yaz_PDU_Assoc(m_socketObservable);
73 void Yaz_PDU_Assoc::socketNotify(int event)
75 yaz_log (m_log, "Yaz_PDU_Assoc::socketNotify p=%p state=%d event = %d",
76 this, m_state, event);
77 if (event & YAZ_SOCKET_OBSERVE_EXCEPT)
80 m_PDU_Observer->failNotify();
83 else if (event & YAZ_SOCKET_OBSERVE_TIMEOUT)
85 m_PDU_Observer->timeoutNotify();
91 if (!cs_accept (m_cs))
93 yaz_log (m_log, "Yaz_PDU_Assoc::cs_accept failed");
96 m_PDU_Observer->failNotify();
101 if (m_cs->io_pending & CS_WANT_WRITE)
102 mask |= YAZ_SOCKET_OBSERVE_WRITE;
103 if (m_cs->io_pending & CS_WANT_READ)
104 mask |= YAZ_SOCKET_OBSERVE_READ;
106 { // accept is complete. turn to ready state and write if needed
111 { // accept still incomplete.
112 m_socketObservable->maskObserver(this,
113 mask|YAZ_SOCKET_OBSERVE_EXCEPT);
118 if (event & YAZ_SOCKET_OBSERVE_READ &&
119 event & YAZ_SOCKET_OBSERVE_WRITE)
121 // For Unix: if both read and write is set, then connect failed.
123 m_PDU_Observer->failNotify();
127 yaz_log (m_log, "cs_rcvconnect");
128 int res = cs_rcvconnect (m_cs);
131 unsigned mask = YAZ_SOCKET_OBSERVE_EXCEPT;
132 if (m_cs->io_pending & CS_WANT_WRITE)
133 mask |= YAZ_SOCKET_OBSERVE_WRITE;
134 if (m_cs->io_pending & CS_WANT_READ)
135 mask |= YAZ_SOCKET_OBSERVE_READ;
136 m_socketObservable->maskObserver(this, mask);
142 m_PDU_Observer->connectNotify();
148 if (event & YAZ_SOCKET_OBSERVE_READ)
153 if ((res = cs_listen(m_cs, 0, 0)) == 1)
157 yaz_log(LOG_FATAL|LOG_ERRNO, "cs_listen failed");
160 if (!(new_line = cs_accept(m_cs)))
162 /* 1. create socket-manager
164 3. create top-level object
165 setup observer for child fileid in pdu-assoc
168 yaz_log (m_log, "new session: parent fd=%d child fd=%d",
169 cs_fileno(m_cs), cs_fileno(new_line));
170 childNotify (new_line);
174 if (event & (YAZ_SOCKET_OBSERVE_READ|YAZ_SOCKET_OBSERVE_WRITE))
178 if (event & (YAZ_SOCKET_OBSERVE_READ|YAZ_SOCKET_OBSERVE_WRITE))
182 int res = cs_get (m_cs, &m_input_buf, &m_input_len);
185 unsigned mask = YAZ_SOCKET_OBSERVE_EXCEPT;
186 if (m_cs->io_pending & CS_WANT_WRITE)
187 mask |= YAZ_SOCKET_OBSERVE_WRITE;
188 if (m_cs->io_pending & CS_WANT_READ)
189 mask |= YAZ_SOCKET_OBSERVE_READ;
190 m_socketObservable->maskObserver(this, mask);
195 yaz_log (m_log, "Yaz_PDU_Assoc::Connection closed by peer");
198 m_PDU_Observer->failNotify(); // problem here..
201 // lock it, so we know if recv_PDU deletes it.
203 m_destroyed = &destroyed;
208 m_PDU_Observer->recv_PDU(m_input_buf, res);
210 if (destroyed) // it really was destroyed, return now.
212 } while (m_cs && cs_more (m_cs));
214 m_socketObservable->maskObserver(this,
215 YAZ_SOCKET_OBSERVE_EXCEPT|
216 YAZ_SOCKET_OBSERVE_READ);
220 yaz_log (m_log, "CLOSING state=%d event was %d", m_state, event);
222 m_PDU_Observer->failNotify();
225 yaz_log (m_log, "Unknown state=%d event was %d", m_state, event);
227 m_PDU_Observer->failNotify();
231 void Yaz_PDU_Assoc::close()
234 for (ch = m_children; ch; ch = ch->m_next)
237 m_socketObservable->deleteObserver(this);
241 yaz_log (m_log, "Yaz_PDU_Assoc::close fd=%d", cs_fileno(m_cs));
247 PDU_Queue *q_this = m_queue_out;
248 m_queue_out = m_queue_out->m_next;
256 void Yaz_PDU_Assoc::destroy()
264 // delete from parent's child list (if any)
267 c = &m_parent->m_children;
275 // delete all children ...
279 Yaz_PDU_Assoc *here = *c;
284 yaz_log (m_log, "Yaz_PDU_Assoc::destroy this=%p", this);
287 Yaz_PDU_Assoc::PDU_Queue::PDU_Queue(const char *buf, int len)
289 m_buf = (char *) xmalloc (len);
290 memcpy (m_buf, buf, len);
295 Yaz_PDU_Assoc::PDU_Queue::~PDU_Queue()
300 int Yaz_PDU_Assoc::flush_PDU()
304 if (m_state != Ready && m_state != Writing)
306 yaz_log (m_log, "YAZ_PDU_Assoc::flush_PDU, not ready");
309 PDU_Queue *q = m_queue_out;
313 yaz_log (m_log, "YAZ_PDU_Assoc::flush_PDU queue empty");
314 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
315 YAZ_SOCKET_OBSERVE_WRITE|
316 YAZ_SOCKET_OBSERVE_EXCEPT);
319 r = cs_put (m_cs, q->m_buf, q->m_len);
322 yaz_log (m_log, "Yaz_PDU_Assoc::flush_PDU cs_put failed");
324 m_PDU_Observer->failNotify();
329 unsigned mask = YAZ_SOCKET_OBSERVE_EXCEPT;
331 if (m_cs->io_pending & CS_WANT_WRITE)
332 mask |= YAZ_SOCKET_OBSERVE_WRITE;
333 if (m_cs->io_pending & CS_WANT_READ)
334 mask |= YAZ_SOCKET_OBSERVE_READ;
336 m_socketObservable->maskObserver(this, mask);
337 yaz_log (m_log, "Yaz_PDU_Assoc::flush_PDU cs_put %d bytes (incomp)",
342 yaz_log (m_log, "Yaz_PDU_Assoc::flush_PDU cs_put %d bytes", q->m_len);
343 // whole packet sent... delete this and proceed to next ...
344 m_queue_out = q->m_next;
346 // don't select on write if queue is empty ...
348 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
349 YAZ_SOCKET_OBSERVE_EXCEPT);
353 int Yaz_PDU_Assoc::send_PDU(const char *buf, int len)
355 yaz_log (m_log, "Yaz_PDU_Assoc::send_PDU");
356 PDU_Queue **pq = &m_queue_out;
357 int is_idle = (*pq ? 0 : 1);
361 yaz_log (m_log, "Yaz_PDU_Assoc::send_PDU failed, m_cs == 0");
366 *pq = new PDU_Queue(buf, len);
370 yaz_log (m_log, "Yaz_PDU_Assoc::cannot send_PDU fd=%d",
375 COMSTACK Yaz_PDU_Assoc::comstack(const char *type_and_host, void **vp)
377 return cs_create_host(type_and_host, 0, vp);
380 void Yaz_PDU_Assoc::listen(IYaz_PDU_Observer *observer,
385 yaz_log (LOG_LOG, "Adding listener %s", addr);
387 m_PDU_Observer = observer;
389 m_cs = comstack(addr, &ap);
393 if (cs_bind(m_cs, ap, CS_SERVER) < 0)
395 m_socketObservable->addObserver(cs_fileno(m_cs), this);
396 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
397 YAZ_SOCKET_OBSERVE_EXCEPT);
398 yaz_log (m_log, "Yaz_PDU_Assoc::listen ok fd=%d", cs_fileno(m_cs));
402 void Yaz_PDU_Assoc::idleTime(int idleTime)
404 m_idleTime = idleTime;
405 yaz_log (m_log, "Yaz_PDU_Assoc::idleTime(%d)", idleTime);
406 m_socketObservable->timeoutObserver(this, m_idleTime);
409 void Yaz_PDU_Assoc::connect(IYaz_PDU_Observer *observer,
412 yaz_log (m_log, "Yaz_PDU_Assoc::connect %s", addr);
414 m_PDU_Observer = observer;
416 m_cs = comstack(addr, &ap);
417 int res = cs_connect (m_cs, ap);
418 yaz_log (m_log, "Yaz_PDU_Assoc::connect fd=%d res=%d", cs_fileno(m_cs),
420 m_socketObservable->addObserver(cs_fileno(m_cs), this);
423 { // Connect pending or complete
424 m_state = Connecting;
425 unsigned mask = YAZ_SOCKET_OBSERVE_EXCEPT;
426 if (m_cs->io_pending & CS_WANT_WRITE)
427 mask |= YAZ_SOCKET_OBSERVE_WRITE;
428 if (m_cs->io_pending & CS_WANT_READ)
429 mask |= YAZ_SOCKET_OBSERVE_READ;
430 m_socketObservable->maskObserver(this, mask);
433 { // Connect failed immediately
434 // Since m_state is Closed we can distinguish this case from
435 // normal connect in socketNotify handler
436 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_WRITE|
437 YAZ_SOCKET_OBSERVE_EXCEPT);
441 // Single-threaded... Only useful for non-blocking handlers
442 void Yaz_PDU_Assoc::childNotify(COMSTACK cs)
444 Yaz_PDU_Assoc *new_observable =
445 new Yaz_PDU_Assoc (m_socketObservable, cs);
447 new_observable->m_next = m_children;
448 m_children = new_observable;
449 new_observable->m_parent = this;
451 // Clone PDU Observer
452 new_observable->m_PDU_Observer = m_PDU_Observer->sessionNotify
453 (new_observable, cs_fileno(cs));