2 * Copyright (c) 1998-2004, Index Data.
3 * See the file LICENSE for details.
5 * $Id: yaz-pdu-assoc.cpp,v 1.40 2005-06-02 06:40:21 adam Exp $
11 #include <yaz/tcpip.h>
13 #include <yaz++/pdu-assoc.h>
15 using namespace yazpp_1;
17 void Yaz_PDU_Assoc::init(IYazSocketObservable *socketObservable)
21 m_socketObservable = socketObservable;
35 Yaz_PDU_Assoc::Yaz_PDU_Assoc(IYazSocketObservable *socketObservable)
37 init (socketObservable);
40 Yaz_PDU_Assoc::Yaz_PDU_Assoc(IYazSocketObservable *socketObservable,
43 init(socketObservable);
46 if (cs->io_pending & CS_WANT_WRITE)
47 mask |= YAZ_SOCKET_OBSERVE_WRITE;
48 if (cs->io_pending & CS_WANT_READ)
49 mask |= YAZ_SOCKET_OBSERVE_READ;
50 m_socketObservable->addObserver(cs_fileno(cs), this);
53 yaz_log (m_log, "new PDU_Assoc. Ready");
59 yaz_log (m_log, "new PDU_Assoc. Accepting");
60 // assume comstack is accepting...
62 m_socketObservable->addObserver(cs_fileno(cs), this);
63 yaz_log(m_log, "maskObserver 1");
64 m_socketObservable->maskObserver(this,
65 mask |YAZ_SOCKET_OBSERVE_EXCEPT);
70 IYaz_PDU_Observable *Yaz_PDU_Assoc::clone()
72 Yaz_PDU_Assoc *copy = new Yaz_PDU_Assoc(m_socketObservable);
76 void Yaz_PDU_Assoc::socketNotify(int event)
78 yaz_log (m_log, "Yaz_PDU_Assoc::socketNotify p=%p state=%d event = %d",
79 this, m_state, event);
80 if (event & YAZ_SOCKET_OBSERVE_EXCEPT)
83 m_PDU_Observer->failNotify();
86 else if (event & YAZ_SOCKET_OBSERVE_TIMEOUT)
88 m_PDU_Observer->timeoutNotify();
94 if (!cs_accept (m_cs))
96 yaz_log (m_log, "Yaz_PDU_Assoc::cs_accept failed");
99 m_PDU_Observer->failNotify();
104 if (m_cs->io_pending & CS_WANT_WRITE)
105 mask |= YAZ_SOCKET_OBSERVE_WRITE;
106 if (m_cs->io_pending & CS_WANT_READ)
107 mask |= YAZ_SOCKET_OBSERVE_READ;
109 { // accept is complete. turn to ready state and write if needed
114 { // accept still incomplete.
115 yaz_log(m_log, "maskObserver 2");
116 m_socketObservable->maskObserver(this,
117 mask|YAZ_SOCKET_OBSERVE_EXCEPT);
122 if (event & YAZ_SOCKET_OBSERVE_READ &&
123 event & YAZ_SOCKET_OBSERVE_WRITE)
125 // For Unix: if both read and write is set, then connect failed.
127 m_PDU_Observer->failNotify();
131 yaz_log (m_log, "cs_rcvconnect");
132 int res = cs_rcvconnect (m_cs);
135 unsigned mask = YAZ_SOCKET_OBSERVE_EXCEPT;
136 if (m_cs->io_pending & CS_WANT_WRITE)
137 mask |= YAZ_SOCKET_OBSERVE_WRITE;
138 if (m_cs->io_pending & CS_WANT_READ)
139 mask |= YAZ_SOCKET_OBSERVE_READ;
140 yaz_log(m_log, "maskObserver 3");
141 m_socketObservable->maskObserver(this, mask);
147 m_PDU_Observer->connectNotify();
153 if (event & YAZ_SOCKET_OBSERVE_READ)
158 if ((res = cs_listen(m_cs, 0, 0)) == 1)
162 yaz_log(YLOG_FATAL|YLOG_ERRNO, "cs_listen failed");
165 if (!(new_line = cs_accept(m_cs)))
167 /* 1. create socket-manager
169 3. create top-level object
170 setup observer for child fileid in pdu-assoc
173 yaz_log (m_log, "new session: parent fd=%d child fd=%d",
174 cs_fileno(m_cs), cs_fileno(new_line));
175 childNotify (new_line);
179 if (event & (YAZ_SOCKET_OBSERVE_READ|YAZ_SOCKET_OBSERVE_WRITE))
183 if (event & (YAZ_SOCKET_OBSERVE_READ|YAZ_SOCKET_OBSERVE_WRITE))
187 int res = cs_get (m_cs, &m_input_buf, &m_input_len);
190 unsigned mask = YAZ_SOCKET_OBSERVE_EXCEPT;
191 if (m_cs->io_pending & CS_WANT_WRITE)
192 mask |= YAZ_SOCKET_OBSERVE_WRITE;
193 if (m_cs->io_pending & CS_WANT_READ)
194 mask |= YAZ_SOCKET_OBSERVE_READ;
195 yaz_log(m_log, "maskObserver 4");
196 m_socketObservable->maskObserver(this, mask);
201 yaz_log (m_log, "Yaz_PDU_Assoc::Connection closed by peer");
204 m_PDU_Observer->failNotify(); // problem here..
207 // lock it, so we know if recv_PDU deletes it.
209 m_destroyed = &destroyed;
214 PDU_Queue **pq = &m_queue_in;
218 *pq = new PDU_Queue(m_input_buf, res);
220 m_PDU_Observer->recv_PDU(m_input_buf, res);
222 if (destroyed) // it really was destroyed, return now.
225 } while (m_cs && cs_more (m_cs));
226 if (m_cs && m_state == Ready)
228 yaz_log(m_log, "maskObserver 5");
229 m_socketObservable->maskObserver(this,
230 YAZ_SOCKET_OBSERVE_EXCEPT|
231 YAZ_SOCKET_OBSERVE_READ);
236 yaz_log (m_log, "CLOSING state=%d event was %d", m_state, event);
238 m_PDU_Observer->failNotify();
241 yaz_log (m_log, "Unknown state=%d event was %d", m_state, event);
243 m_PDU_Observer->failNotify();
247 void Yaz_PDU_Assoc::close()
250 for (ch = m_children; ch; ch = ch->m_next)
253 m_socketObservable->deleteObserver(this);
257 yaz_log (m_log, "Yaz_PDU_Assoc::close fd=%d", cs_fileno(m_cs));
263 PDU_Queue *q_this = m_queue_out;
264 m_queue_out = m_queue_out->m_next;
272 void Yaz_PDU_Assoc::destroy()
280 // delete from parent's child list (if any)
283 c = &m_parent->m_children;
291 // delete all children ...
295 Yaz_PDU_Assoc *here = *c;
300 yaz_log (m_log, "Yaz_PDU_Assoc::destroy this=%p", this);
303 Yaz_PDU_Assoc::PDU_Queue::PDU_Queue(const char *buf, int len)
305 m_buf = (char *) xmalloc (len);
306 memcpy (m_buf, buf, len);
311 Yaz_PDU_Assoc::PDU_Queue::~PDU_Queue()
316 int Yaz_PDU_Assoc::flush_PDU()
320 if (m_state != Ready && m_state != Writing)
322 yaz_log (m_log, "YAZ_PDU_Assoc::flush_PDU, not ready");
325 PDU_Queue *q = m_queue_out;
329 yaz_log (m_log, "YAZ_PDU_Assoc::flush_PDU queue empty");
330 yaz_log(m_log, "maskObserver 6");
331 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
332 YAZ_SOCKET_OBSERVE_WRITE|
333 YAZ_SOCKET_OBSERVE_EXCEPT);
336 r = cs_put (m_cs, q->m_buf, q->m_len);
339 yaz_log (m_log, "Yaz_PDU_Assoc::flush_PDU cs_put failed");
341 m_PDU_Observer->failNotify();
346 unsigned mask = YAZ_SOCKET_OBSERVE_EXCEPT;
348 if (m_cs->io_pending & CS_WANT_WRITE)
349 mask |= YAZ_SOCKET_OBSERVE_WRITE;
350 if (m_cs->io_pending & CS_WANT_READ)
351 mask |= YAZ_SOCKET_OBSERVE_READ;
353 mask |= YAZ_SOCKET_OBSERVE_WRITE;
354 yaz_log(m_log, "maskObserver 7");
355 m_socketObservable->maskObserver(this, mask);
356 yaz_log (m_log, "Yaz_PDU_Assoc::flush_PDU cs_put %d bytes fd=%d (inc)",
357 q->m_len, cs_fileno(m_cs));
360 yaz_log (m_log, "Yaz_PDU_Assoc::flush_PDU cs_put %d bytes", q->m_len);
361 // whole packet sent... delete this and proceed to next ...
362 m_queue_out = q->m_next;
364 // don't select on write if queue is empty ...
368 yaz_log(m_log, "maskObserver 8");
369 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
370 YAZ_SOCKET_OBSERVE_EXCEPT);
375 int Yaz_PDU_Assoc::send_PDU(const char *buf, int len)
377 yaz_log (m_log, "Yaz_PDU_Assoc::send_PDU");
378 PDU_Queue **pq = &m_queue_out;
379 int is_idle = (*pq ? 0 : 1);
383 yaz_log (m_log, "Yaz_PDU_Assoc::send_PDU failed, m_cs == 0");
388 *pq = new PDU_Queue(buf, len);
392 yaz_log (m_log, "Yaz_PDU_Assoc::cannot send_PDU fd=%d",
397 COMSTACK Yaz_PDU_Assoc::comstack(const char *type_and_host, void **vp)
399 return cs_create_host(type_and_host, 2, vp);
402 int Yaz_PDU_Assoc::listen(IYaz_PDU_Observer *observer,
407 m_PDU_Observer = observer;
409 m_cs = comstack(addr, &ap);
413 if (cs_bind(m_cs, ap, CS_SERVER) < 0)
415 m_socketObservable->addObserver(cs_fileno(m_cs), this);
416 yaz_log(m_log, "maskObserver 9");
417 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
418 YAZ_SOCKET_OBSERVE_EXCEPT);
419 yaz_log (m_log, "Yaz_PDU_Assoc::listen ok fd=%d", cs_fileno(m_cs));
424 void Yaz_PDU_Assoc::idleTime(int idleTime)
426 m_idleTime = idleTime;
427 yaz_log (m_log, "Yaz_PDU_Assoc::idleTime(%d)", idleTime);
428 m_socketObservable->timeoutObserver(this, m_idleTime);
431 int Yaz_PDU_Assoc::connect(IYaz_PDU_Observer *observer,
434 yaz_log (m_log, "Yaz_PDU_Assoc::connect %s", addr);
436 m_PDU_Observer = observer;
438 m_cs = comstack(addr, &ap);
441 int res = cs_connect (m_cs, ap);
442 yaz_log (m_log, "Yaz_PDU_Assoc::connect fd=%d res=%d", cs_fileno(m_cs),
444 m_socketObservable->addObserver(cs_fileno(m_cs), this);
447 { // Connect complete
448 m_state = Connecting;
449 unsigned mask = YAZ_SOCKET_OBSERVE_EXCEPT;
450 mask |= YAZ_SOCKET_OBSERVE_WRITE;
451 mask |= YAZ_SOCKET_OBSERVE_READ;
452 yaz_log(m_log, "maskObserver 11");
453 m_socketObservable->maskObserver(this, mask);
457 m_state = Connecting;
458 unsigned mask = YAZ_SOCKET_OBSERVE_EXCEPT;
459 if (m_cs->io_pending & CS_WANT_WRITE)
460 mask |= YAZ_SOCKET_OBSERVE_WRITE;
461 if (m_cs->io_pending & CS_WANT_READ)
462 mask |= YAZ_SOCKET_OBSERVE_READ;
463 yaz_log(m_log, "maskObserver 11");
464 m_socketObservable->maskObserver(this, mask);
467 { // Connect failed immediately
468 // Since m_state is Closed we can distinguish this case from
469 // normal connect in socketNotify handler
470 yaz_log(m_log, "maskObserver 12");
471 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_WRITE|
472 YAZ_SOCKET_OBSERVE_EXCEPT);
477 // Single-threaded... Only useful for non-blocking handlers
478 void Yaz_PDU_Assoc::childNotify(COMSTACK cs)
480 Yaz_PDU_Assoc *new_observable =
481 new Yaz_PDU_Assoc (m_socketObservable, cs);
483 new_observable->m_next = m_children;
484 m_children = new_observable;
485 new_observable->m_parent = this;
487 // Clone PDU Observer
488 new_observable->m_PDU_Observer = m_PDU_Observer->sessionNotify
489 (new_observable, cs_fileno(cs));
492 const char*Yaz_PDU_Assoc::getpeername()
494 return cs_addrstr(m_cs);