2 * Copyright (c) 1998-2004, Index Data.
3 * See the file LICENSE for details.
5 * $Id: yaz-pdu-assoc.cpp,v 1.47 2008-01-21 15:57:27 adam Exp $
11 #include <yaz/tcpip.h>
13 #include <yazpp/pdu-assoc.h>
15 using namespace yazpp_1;
17 void PDU_Assoc::init(ISocketObservable *socketObservable)
21 m_socketObservable = socketObservable;
33 m_session_is_dead = false;
36 PDU_Assoc::PDU_Assoc(ISocketObservable *socketObservable)
38 init (socketObservable);
41 PDU_Assoc::PDU_Assoc(ISocketObservable *socketObservable,
44 init(socketObservable);
47 if (cs->io_pending & CS_WANT_WRITE)
48 mask |= SOCKET_OBSERVE_WRITE;
49 if (cs->io_pending & CS_WANT_READ)
50 mask |= SOCKET_OBSERVE_READ;
51 m_socketObservable->addObserver(cs_fileno(cs), this);
54 yaz_log (m_log, "new PDU_Assoc. Ready");
60 yaz_log (m_log, "new PDU_Assoc. Accepting");
61 // assume comstack is accepting...
63 m_socketObservable->addObserver(cs_fileno(cs), this);
64 yaz_log(m_log, "maskObserver 1");
65 m_socketObservable->maskObserver(this,
66 mask |SOCKET_OBSERVE_EXCEPT);
71 IPDU_Observable *PDU_Assoc::clone()
73 PDU_Assoc *copy = new PDU_Assoc(m_socketObservable);
77 void PDU_Assoc::socketNotify(int event)
79 yaz_log (m_log, "PDU_Assoc::socketNotify p=%p state=%d event = %d",
80 this, m_state, event);
81 if (event & SOCKET_OBSERVE_EXCEPT)
84 m_PDU_Observer->failNotify();
87 else if (event & SOCKET_OBSERVE_TIMEOUT)
89 m_PDU_Observer->timeoutNotify();
95 if (!cs_accept (m_cs))
97 yaz_log (m_log, "PDU_Assoc::cs_accept failed");
100 m_PDU_Observer->failNotify();
105 if (m_cs->io_pending & CS_WANT_WRITE)
106 mask |= SOCKET_OBSERVE_WRITE;
107 if (m_cs->io_pending & CS_WANT_READ)
108 mask |= SOCKET_OBSERVE_READ;
110 { // accept is complete. turn to ready state and write if needed
115 { // accept still incomplete.
116 yaz_log(m_log, "maskObserver 2");
117 m_socketObservable->maskObserver(this,
118 mask|SOCKET_OBSERVE_EXCEPT);
123 if (event & SOCKET_OBSERVE_READ &&
124 event & SOCKET_OBSERVE_WRITE)
126 // For Unix: if both read and write is set, then connect failed.
128 m_PDU_Observer->failNotify();
132 yaz_log (m_log, "cs_rcvconnect");
133 int res = cs_rcvconnect (m_cs);
136 unsigned mask = SOCKET_OBSERVE_EXCEPT;
137 if (m_cs->io_pending & CS_WANT_WRITE)
138 mask |= SOCKET_OBSERVE_WRITE;
139 if (m_cs->io_pending & CS_WANT_READ)
140 mask |= SOCKET_OBSERVE_READ;
141 yaz_log(m_log, "maskObserver 3");
142 m_socketObservable->maskObserver(this, mask);
148 m_PDU_Observer->connectNotify();
154 if (event & SOCKET_OBSERVE_READ)
159 if ((res = cs_listen(m_cs, 0, 0)) == 1)
163 yaz_log(YLOG_FATAL|YLOG_ERRNO, "cs_listen failed");
166 if (!(new_line = cs_accept(m_cs)))
168 /* 1. create socket-manager
170 3. create top-level object
171 setup observer for child fileid in pdu-assoc
174 yaz_log (m_log, "new session: parent fd=%d child fd=%d",
175 cs_fileno(m_cs), cs_fileno(new_line));
176 childNotify (new_line);
180 if (event & (SOCKET_OBSERVE_READ|SOCKET_OBSERVE_WRITE))
184 if (event & (SOCKET_OBSERVE_READ|SOCKET_OBSERVE_WRITE))
188 int res = cs_get (m_cs, &m_input_buf, &m_input_len);
191 unsigned mask = SOCKET_OBSERVE_EXCEPT;
192 if (m_cs->io_pending & CS_WANT_WRITE)
193 mask |= SOCKET_OBSERVE_WRITE;
194 if (m_cs->io_pending & CS_WANT_READ)
195 mask |= SOCKET_OBSERVE_READ;
196 yaz_log(m_log, "maskObserver 4");
197 m_socketObservable->maskObserver(this, mask);
202 yaz_log (m_log, "PDU_Assoc::Connection closed by peer");
205 m_PDU_Observer->failNotify(); // problem here..
208 // lock it, so we know if recv_PDU deletes it.
210 m_destroyed = &destroyed;
215 PDU_Queue **pq = &m_queue_in;
219 *pq = new PDU_Queue(m_input_buf, res);
221 m_PDU_Observer->recv_PDU(m_input_buf, res);
223 if (destroyed) // it really was destroyed, return now.
226 } while (m_cs && cs_more (m_cs));
227 if (m_cs && m_state == Ready)
229 yaz_log(m_log, "maskObserver 5");
230 m_socketObservable->maskObserver(this,
231 SOCKET_OBSERVE_EXCEPT|
232 SOCKET_OBSERVE_READ);
237 yaz_log (m_log, "CLOSING state=%d event was %d", m_state, event);
239 m_PDU_Observer->failNotify();
242 yaz_log (m_log, "Unknown state=%d event was %d", m_state, event);
244 m_PDU_Observer->failNotify();
248 void PDU_Assoc::close_session()
250 m_session_is_dead = true;
254 m_PDU_Observer->failNotify();
258 void PDU_Assoc::shutdown()
261 for (ch = m_children; ch; ch = ch->m_next)
264 m_socketObservable->deleteObserver(this);
268 yaz_log (m_log, "PDU_Assoc::close fd=%d", cs_fileno(m_cs));
274 PDU_Queue *q_this = m_queue_out;
275 m_queue_out = m_queue_out->m_next;
283 void PDU_Assoc::destroy()
291 // delete from parent's child list (if any)
294 c = &m_parent->m_children;
302 // delete all children ...
306 PDU_Assoc *here = *c;
311 yaz_log (m_log, "PDU_Assoc::destroy this=%p", this);
314 PDU_Assoc::PDU_Queue::PDU_Queue(const char *buf, int len)
316 m_buf = (char *) xmalloc (len);
317 memcpy (m_buf, buf, len);
322 PDU_Assoc::PDU_Queue::~PDU_Queue()
327 int PDU_Assoc::flush_PDU()
331 if (m_state != Ready && m_state != Writing)
333 yaz_log (m_log, "YAZ_PDU_Assoc::flush_PDU, not ready");
336 PDU_Queue *q = m_queue_out;
340 yaz_log (m_log, "YAZ_PDU_Assoc::flush_PDU queue empty");
341 yaz_log(m_log, "maskObserver 6");
342 m_socketObservable->maskObserver(this, SOCKET_OBSERVE_READ|
343 SOCKET_OBSERVE_WRITE|
344 SOCKET_OBSERVE_EXCEPT);
345 if (m_session_is_dead)
348 m_PDU_Observer->failNotify();
352 r = cs_put (m_cs, q->m_buf, q->m_len);
355 yaz_log (m_log, "PDU_Assoc::flush_PDU cs_put failed");
357 m_PDU_Observer->failNotify();
362 unsigned mask = SOCKET_OBSERVE_EXCEPT;
364 if (m_cs->io_pending & CS_WANT_WRITE)
365 mask |= SOCKET_OBSERVE_WRITE;
366 if (m_cs->io_pending & CS_WANT_READ)
367 mask |= SOCKET_OBSERVE_READ;
369 mask |= SOCKET_OBSERVE_WRITE;
370 yaz_log(m_log, "maskObserver 7");
371 m_socketObservable->maskObserver(this, mask);
372 yaz_log (m_log, "PDU_Assoc::flush_PDU cs_put %d bytes fd=%d (inc)",
373 q->m_len, cs_fileno(m_cs));
376 yaz_log (m_log, "PDU_Assoc::flush_PDU cs_put %d bytes", q->m_len);
377 // whole packet sent... delete this and proceed to next ...
378 m_queue_out = q->m_next;
380 // don't select on write if queue is empty ...
384 yaz_log(m_log, "maskObserver 8");
385 m_socketObservable->maskObserver(this, SOCKET_OBSERVE_READ|
386 SOCKET_OBSERVE_EXCEPT);
391 int PDU_Assoc::send_PDU(const char *buf, int len)
393 yaz_log (m_log, "PDU_Assoc::send_PDU");
394 PDU_Queue **pq = &m_queue_out;
395 int is_idle = (*pq ? 0 : 1);
399 yaz_log (m_log, "PDU_Assoc::send_PDU failed, m_cs == 0");
404 *pq = new PDU_Queue(buf, len);
408 yaz_log (m_log, "PDU_Assoc::cannot send_PDU fd=%d",
413 COMSTACK PDU_Assoc::comstack(const char *type_and_host, void **vp)
415 return cs_create_host(type_and_host, 2, vp);
418 int PDU_Assoc::listen(IPDU_Observer *observer, const char *addr)
422 m_PDU_Observer = observer;
424 m_cs = comstack(addr, &ap);
428 if (cs_bind(m_cs, ap, CS_SERVER) < 0)
430 m_socketObservable->addObserver(cs_fileno(m_cs), this);
431 yaz_log(m_log, "maskObserver 9");
432 m_socketObservable->maskObserver(this, SOCKET_OBSERVE_READ|
433 SOCKET_OBSERVE_EXCEPT);
434 yaz_log (m_log, "PDU_Assoc::listen ok fd=%d", cs_fileno(m_cs));
439 void PDU_Assoc::idleTime(int idleTime)
441 m_idleTime = idleTime;
442 yaz_log (m_log, "PDU_Assoc::idleTime(%d)", idleTime);
443 m_socketObservable->timeoutObserver(this, m_idleTime);
446 int PDU_Assoc::connect(IPDU_Observer *observer, const char *addr)
448 yaz_log (m_log, "PDU_Assoc::connect %s", addr);
450 m_PDU_Observer = observer;
452 m_cs = comstack(addr, &ap);
455 int res = cs_connect (m_cs, ap);
456 yaz_log (m_log, "PDU_Assoc::connect fd=%d res=%d", cs_fileno(m_cs),
458 m_socketObservable->addObserver(cs_fileno(m_cs), this);
461 { // Connect complete
462 m_state = Connecting;
463 unsigned mask = SOCKET_OBSERVE_EXCEPT;
464 mask |= SOCKET_OBSERVE_WRITE;
465 mask |= SOCKET_OBSERVE_READ;
466 yaz_log(m_log, "maskObserver 11");
467 m_socketObservable->maskObserver(this, mask);
471 m_state = Connecting;
472 unsigned mask = SOCKET_OBSERVE_EXCEPT;
473 if (m_cs->io_pending & CS_WANT_WRITE)
474 mask |= SOCKET_OBSERVE_WRITE;
475 if (m_cs->io_pending & CS_WANT_READ)
476 mask |= SOCKET_OBSERVE_READ;
477 yaz_log(m_log, "maskObserver 11");
478 m_socketObservable->maskObserver(this, mask);
481 { // Connect failed immediately
482 // Since m_state is Closed we can distinguish this case from
483 // normal connect in socketNotify handler
484 yaz_log(m_log, "maskObserver 12");
485 m_socketObservable->maskObserver(this, SOCKET_OBSERVE_WRITE|
486 SOCKET_OBSERVE_EXCEPT);
491 // Single-threaded... Only useful for non-blocking handlers
492 void PDU_Assoc::childNotify(COMSTACK cs)
494 PDU_Assoc *new_observable =
495 new PDU_Assoc (m_socketObservable, cs);
497 // Clone PDU Observer
498 new_observable->m_PDU_Observer = m_PDU_Observer->sessionNotify
499 (new_observable, cs_fileno(cs));
501 if (!new_observable->m_PDU_Observer)
503 new_observable->shutdown();
504 delete new_observable;
507 new_observable->m_next = m_children;
508 m_children = new_observable;
509 new_observable->m_parent = this;
512 const char*PDU_Assoc::getpeername()
516 return cs_addrstr(m_cs);
521 * indent-tabs-mode: nil
523 * vim: shiftwidth=4 tabstop=8 expandtab