1 /* This file is part of the yazpp toolkit.
2 * Copyright (C) 1998-2010 Index Data and Mike Taylor
3 * See the file LICENSE for details.
11 #include <yazpp/pdu-assoc.h>
13 using namespace yazpp_1;
15 void PDU_Assoc::init(ISocketObservable *socketObservable)
19 m_socketObservable = socketObservable;
31 m_session_is_dead = false;
34 PDU_Assoc::PDU_Assoc(ISocketObservable *socketObservable)
36 init (socketObservable);
39 PDU_Assoc::PDU_Assoc(ISocketObservable *socketObservable,
42 init(socketObservable);
45 if (cs->io_pending & CS_WANT_WRITE)
46 mask |= SOCKET_OBSERVE_WRITE;
47 if (cs->io_pending & CS_WANT_READ)
48 mask |= SOCKET_OBSERVE_READ;
49 m_socketObservable->addObserver(cs_fileno(cs), this);
52 yaz_log (m_log, "new PDU_Assoc. Ready");
58 yaz_log (m_log, "new PDU_Assoc. Accepting");
59 // assume comstack is accepting...
61 m_socketObservable->addObserver(cs_fileno(cs), this);
62 yaz_log(m_log, "maskObserver 1");
63 m_socketObservable->maskObserver(this,
64 mask |SOCKET_OBSERVE_EXCEPT);
69 IPDU_Observable *PDU_Assoc::clone()
71 PDU_Assoc *copy = new PDU_Assoc(m_socketObservable);
75 void PDU_Assoc::socketNotify(int event)
77 yaz_log (m_log, "PDU_Assoc::socketNotify p=%p state=%d event = %d",
78 this, m_state, event);
79 if (event & SOCKET_OBSERVE_EXCEPT)
82 m_PDU_Observer->failNotify();
85 else if (event & SOCKET_OBSERVE_TIMEOUT)
87 m_PDU_Observer->timeoutNotify();
93 if (!cs_accept (m_cs))
95 yaz_log (m_log, "PDU_Assoc::cs_accept failed");
98 m_PDU_Observer->failNotify();
103 if (m_cs->io_pending & CS_WANT_WRITE)
104 mask |= SOCKET_OBSERVE_WRITE;
105 if (m_cs->io_pending & CS_WANT_READ)
106 mask |= SOCKET_OBSERVE_READ;
108 { // accept is complete. turn to ready state and write if needed
113 { // accept still incomplete.
114 yaz_log(m_log, "maskObserver 2");
115 m_socketObservable->maskObserver(this,
116 mask|SOCKET_OBSERVE_EXCEPT);
121 if (event & SOCKET_OBSERVE_READ &&
122 event & SOCKET_OBSERVE_WRITE)
124 // For Unix: if both read and write is set, then connect failed.
126 m_PDU_Observer->failNotify();
130 yaz_log (m_log, "cs_rcvconnect");
131 int res = cs_rcvconnect (m_cs);
134 unsigned mask = SOCKET_OBSERVE_EXCEPT;
135 if (m_cs->io_pending & CS_WANT_WRITE)
136 mask |= SOCKET_OBSERVE_WRITE;
137 if (m_cs->io_pending & CS_WANT_READ)
138 mask |= SOCKET_OBSERVE_READ;
139 yaz_log(m_log, "maskObserver 3");
140 m_socketObservable->maskObserver(this, mask);
146 m_PDU_Observer->connectNotify();
152 if (event & SOCKET_OBSERVE_READ)
157 if ((res = cs_listen(m_cs, 0, 0)) == 1)
161 yaz_log(YLOG_FATAL|YLOG_ERRNO, "cs_listen failed");
164 if (!(new_line = cs_accept(m_cs)))
166 /* 1. create socket-manager
168 3. create top-level object
169 setup observer for child fileid in pdu-assoc
172 yaz_log (m_log, "new session: parent fd=%d child fd=%d",
173 cs_fileno(m_cs), cs_fileno(new_line));
174 childNotify (new_line);
178 if (event & (SOCKET_OBSERVE_READ|SOCKET_OBSERVE_WRITE))
182 if (event & (SOCKET_OBSERVE_READ|SOCKET_OBSERVE_WRITE))
186 int res = cs_get (m_cs, &m_input_buf, &m_input_len);
189 unsigned mask = SOCKET_OBSERVE_EXCEPT;
190 if (m_cs->io_pending & CS_WANT_WRITE)
191 mask |= SOCKET_OBSERVE_WRITE;
192 if (m_cs->io_pending & CS_WANT_READ)
193 mask |= SOCKET_OBSERVE_READ;
194 yaz_log(m_log, "maskObserver 4");
195 m_socketObservable->maskObserver(this, mask);
200 yaz_log (m_log, "PDU_Assoc::Connection closed by peer");
203 m_PDU_Observer->failNotify(); // problem here..
206 // lock it, so we know if recv_PDU deletes it.
208 m_destroyed = &destroyed;
213 PDU_Queue **pq = &m_queue_in;
217 *pq = new PDU_Queue(m_input_buf, res);
219 m_PDU_Observer->recv_PDU(m_input_buf, res);
221 if (destroyed) // it really was destroyed, return now.
224 } while (m_cs && cs_more (m_cs));
225 if (m_cs && m_state == Ready)
227 yaz_log(m_log, "maskObserver 5");
228 m_socketObservable->maskObserver(this,
229 SOCKET_OBSERVE_EXCEPT|
230 SOCKET_OBSERVE_READ);
235 yaz_log (m_log, "CLOSING state=%d event was %d", m_state, event);
237 m_PDU_Observer->failNotify();
240 yaz_log (m_log, "Unknown state=%d event was %d", m_state, event);
242 m_PDU_Observer->failNotify();
246 void PDU_Assoc::close_session()
248 m_session_is_dead = true;
252 m_PDU_Observer->failNotify();
256 void PDU_Assoc::shutdown()
259 for (ch = m_children; ch; ch = ch->m_next)
262 m_socketObservable->deleteObserver(this);
266 yaz_log (m_log, "PDU_Assoc::close fd=%d", cs_fileno(m_cs));
272 PDU_Queue *q_this = m_queue_out;
273 m_queue_out = m_queue_out->m_next;
281 void PDU_Assoc::destroy()
289 // delete from parent's child list (if any)
292 c = &m_parent->m_children;
300 // delete all children ...
304 PDU_Assoc *here = *c;
309 yaz_log (m_log, "PDU_Assoc::destroy this=%p", this);
312 PDU_Assoc::PDU_Queue::PDU_Queue(const char *buf, int len)
314 m_buf = (char *) xmalloc (len);
315 memcpy (m_buf, buf, len);
320 PDU_Assoc::PDU_Queue::~PDU_Queue()
325 int PDU_Assoc::flush_PDU()
329 if (m_state != Ready && m_state != Writing)
331 yaz_log (m_log, "YAZ_PDU_Assoc::flush_PDU, not ready");
334 PDU_Queue *q = m_queue_out;
338 yaz_log (m_log, "YAZ_PDU_Assoc::flush_PDU queue empty");
339 yaz_log(m_log, "maskObserver 6");
340 m_socketObservable->maskObserver(this, SOCKET_OBSERVE_READ|
341 SOCKET_OBSERVE_WRITE|
342 SOCKET_OBSERVE_EXCEPT);
343 if (m_session_is_dead)
346 m_PDU_Observer->failNotify();
350 r = cs_put (m_cs, q->m_buf, q->m_len);
353 yaz_log (m_log, "PDU_Assoc::flush_PDU cs_put failed");
355 m_PDU_Observer->failNotify();
360 unsigned mask = SOCKET_OBSERVE_EXCEPT;
362 if (m_cs->io_pending & CS_WANT_WRITE)
363 mask |= SOCKET_OBSERVE_WRITE;
364 if (m_cs->io_pending & CS_WANT_READ)
365 mask |= SOCKET_OBSERVE_READ;
367 mask |= SOCKET_OBSERVE_WRITE;
368 yaz_log(m_log, "maskObserver 7");
369 m_socketObservable->maskObserver(this, mask);
370 yaz_log (m_log, "PDU_Assoc::flush_PDU cs_put %d bytes fd=%d (inc)",
371 q->m_len, cs_fileno(m_cs));
374 yaz_log (m_log, "PDU_Assoc::flush_PDU cs_put %d bytes", q->m_len);
375 // whole packet sent... delete this and proceed to next ...
376 m_queue_out = q->m_next;
378 // don't select on write if queue is empty ...
382 yaz_log(m_log, "maskObserver 8");
383 m_socketObservable->maskObserver(this, SOCKET_OBSERVE_READ|
384 SOCKET_OBSERVE_EXCEPT);
389 int PDU_Assoc::send_PDU(const char *buf, int len)
391 yaz_log (m_log, "PDU_Assoc::send_PDU");
392 PDU_Queue **pq = &m_queue_out;
393 int is_idle = (*pq ? 0 : 1);
397 yaz_log (m_log, "PDU_Assoc::send_PDU failed, m_cs == 0");
402 *pq = new PDU_Queue(buf, len);
406 yaz_log (m_log, "PDU_Assoc::cannot send_PDU fd=%d",
411 COMSTACK PDU_Assoc::comstack(const char *type_and_host, void **vp)
413 return cs_create_host(type_and_host, 2, vp);
416 int PDU_Assoc::listen(IPDU_Observer *observer, const char *addr)
420 m_PDU_Observer = observer;
422 m_cs = comstack(addr, &ap);
426 if (cs_bind(m_cs, ap, CS_SERVER) < 0)
428 m_socketObservable->addObserver(cs_fileno(m_cs), this);
429 yaz_log(m_log, "maskObserver 9");
430 m_socketObservable->maskObserver(this, SOCKET_OBSERVE_READ|
431 SOCKET_OBSERVE_EXCEPT);
432 yaz_log (m_log, "PDU_Assoc::listen ok fd=%d", cs_fileno(m_cs));
437 void PDU_Assoc::idleTime(int idleTime)
439 m_idleTime = idleTime;
440 yaz_log (m_log, "PDU_Assoc::idleTime(%d)", idleTime);
441 m_socketObservable->timeoutObserver(this, m_idleTime);
444 int PDU_Assoc::connect(IPDU_Observer *observer, const char *addr)
446 yaz_log (m_log, "PDU_Assoc::connect %s", addr);
448 m_PDU_Observer = observer;
450 m_cs = comstack(addr, &ap);
453 int res = cs_connect (m_cs, ap);
454 yaz_log (m_log, "PDU_Assoc::connect fd=%d res=%d", cs_fileno(m_cs),
456 m_socketObservable->addObserver(cs_fileno(m_cs), this);
459 { // Connect complete
460 m_state = Connecting;
461 unsigned mask = SOCKET_OBSERVE_EXCEPT;
462 mask |= SOCKET_OBSERVE_WRITE;
463 mask |= SOCKET_OBSERVE_READ;
464 yaz_log(m_log, "maskObserver 11");
465 m_socketObservable->maskObserver(this, mask);
469 m_state = Connecting;
470 unsigned mask = SOCKET_OBSERVE_EXCEPT;
471 if (m_cs->io_pending & CS_WANT_WRITE)
472 mask |= SOCKET_OBSERVE_WRITE;
473 if (m_cs->io_pending & CS_WANT_READ)
474 mask |= SOCKET_OBSERVE_READ;
475 yaz_log(m_log, "maskObserver 11");
476 m_socketObservable->maskObserver(this, mask);
479 { // Connect failed immediately
480 // Since m_state is Closed we can distinguish this case from
481 // normal connect in socketNotify handler
482 yaz_log(m_log, "maskObserver 12");
483 m_socketObservable->maskObserver(this, SOCKET_OBSERVE_WRITE|
484 SOCKET_OBSERVE_EXCEPT);
489 // Single-threaded... Only useful for non-blocking handlers
490 void PDU_Assoc::childNotify(COMSTACK cs)
492 PDU_Assoc *new_observable =
493 new PDU_Assoc (m_socketObservable, cs);
495 // Clone PDU Observer
496 new_observable->m_PDU_Observer = m_PDU_Observer->sessionNotify
497 (new_observable, cs_fileno(cs));
499 if (!new_observable->m_PDU_Observer)
501 new_observable->shutdown();
502 delete new_observable;
505 new_observable->m_next = m_children;
506 m_children = new_observable;
507 new_observable->m_parent = this;
510 const char*PDU_Assoc::getpeername()
514 return cs_addrstr(m_cs);
519 * c-file-style: "Stroustrup"
520 * indent-tabs-mode: nil
522 * vim: shiftwidth=4 tabstop=8 expandtab