1 /* This file is part of the yazpp toolkit.
2 * Copyright (C) 1998-2011 Index Data and Mike Taylor
3 * See the file LICENSE for details.
11 #include <yazpp/pdu-assoc.h>
17 using namespace yazpp_1;
19 void PDU_Assoc::init(ISocketObservable *socketObservable)
23 m_socketObservable = socketObservable;
35 m_session_is_dead = false;
38 PDU_Assoc::PDU_Assoc(ISocketObservable *socketObservable)
40 init (socketObservable);
43 PDU_Assoc::PDU_Assoc(ISocketObservable *socketObservable,
46 init(socketObservable);
49 if (cs->io_pending & CS_WANT_WRITE)
50 mask |= SOCKET_OBSERVE_WRITE;
51 if (cs->io_pending & CS_WANT_READ)
52 mask |= SOCKET_OBSERVE_READ;
53 m_socketObservable->addObserver(cs_fileno(cs), this);
56 yaz_log (m_log, "new PDU_Assoc. Ready");
62 yaz_log (m_log, "new PDU_Assoc. Accepting");
63 // assume comstack is accepting...
65 m_socketObservable->addObserver(cs_fileno(cs), this);
66 yaz_log(m_log, "maskObserver 1");
67 m_socketObservable->maskObserver(this,
68 mask |SOCKET_OBSERVE_EXCEPT);
73 IPDU_Observable *PDU_Assoc::clone()
75 PDU_Assoc *copy = new PDU_Assoc(m_socketObservable);
79 void PDU_Assoc::socketNotify(int event)
81 yaz_log (m_log, "PDU_Assoc::socketNotify p=%p state=%d event = %d",
82 this, m_state, event);
83 if (event & SOCKET_OBSERVE_EXCEPT)
86 m_PDU_Observer->failNotify();
89 else if (event & SOCKET_OBSERVE_TIMEOUT)
91 m_PDU_Observer->timeoutNotify();
97 if (!cs_accept (m_cs))
99 yaz_log (m_log, "PDU_Assoc::cs_accept failed");
102 m_PDU_Observer->failNotify();
107 if (m_cs->io_pending & CS_WANT_WRITE)
108 mask |= SOCKET_OBSERVE_WRITE;
109 if (m_cs->io_pending & CS_WANT_READ)
110 mask |= SOCKET_OBSERVE_READ;
112 { // accept is complete. turn to ready state and write if needed
117 { // accept still incomplete.
118 yaz_log(m_log, "maskObserver 2");
119 m_socketObservable->maskObserver(this,
120 mask|SOCKET_OBSERVE_EXCEPT);
125 if (event & SOCKET_OBSERVE_READ &&
126 event & SOCKET_OBSERVE_WRITE)
128 // For Unix: if both read and write is set, then connect failed.
130 m_PDU_Observer->failNotify();
134 yaz_log (m_log, "cs_rcvconnect");
135 int res = cs_rcvconnect (m_cs);
138 unsigned mask = SOCKET_OBSERVE_EXCEPT;
139 if (m_cs->io_pending & CS_WANT_WRITE)
140 mask |= SOCKET_OBSERVE_WRITE;
141 if (m_cs->io_pending & CS_WANT_READ)
142 mask |= SOCKET_OBSERVE_READ;
143 yaz_log(m_log, "maskObserver 3");
144 m_socketObservable->maskObserver(this, mask);
150 m_PDU_Observer->connectNotify();
156 if (event & SOCKET_OBSERVE_READ)
161 if ((res = cs_listen(m_cs, 0, 0)) == 1)
165 yaz_log(YLOG_FATAL|YLOG_ERRNO, "cs_listen failed");
168 if (!(new_line = cs_accept(m_cs)))
170 /* 1. create socket-manager
172 3. create top-level object
173 setup observer for child fileid in pdu-assoc
176 yaz_log (m_log, "new session: parent fd=%d child fd=%d",
177 cs_fileno(m_cs), cs_fileno(new_line));
178 childNotify (new_line);
182 if (event & (SOCKET_OBSERVE_READ|SOCKET_OBSERVE_WRITE))
186 if (event & (SOCKET_OBSERVE_READ|SOCKET_OBSERVE_WRITE))
190 int res = cs_get (m_cs, &m_input_buf, &m_input_len);
193 unsigned mask = SOCKET_OBSERVE_EXCEPT;
194 if (m_cs->io_pending & CS_WANT_WRITE)
195 mask |= SOCKET_OBSERVE_WRITE;
196 if (m_cs->io_pending & CS_WANT_READ)
197 mask |= SOCKET_OBSERVE_READ;
198 yaz_log(m_log, "maskObserver 4");
199 m_socketObservable->maskObserver(this, mask);
204 yaz_log (m_log, "PDU_Assoc::Connection closed by peer");
207 m_PDU_Observer->failNotify(); // problem here..
210 // lock it, so we know if recv_PDU deletes it.
212 m_destroyed = &destroyed;
217 PDU_Queue **pq = &m_queue_in;
221 *pq = new PDU_Queue(m_input_buf, res);
223 m_PDU_Observer->recv_PDU(m_input_buf, res);
225 if (destroyed) // it really was destroyed, return now.
228 } while (m_cs && cs_more (m_cs));
229 if (m_cs && m_state == Ready)
231 yaz_log(m_log, "maskObserver 5");
232 m_socketObservable->maskObserver(this,
233 SOCKET_OBSERVE_EXCEPT|
234 SOCKET_OBSERVE_READ);
239 yaz_log (m_log, "CLOSING state=%d event was %d", m_state, event);
241 m_PDU_Observer->failNotify();
244 yaz_log (m_log, "Unknown state=%d event was %d", m_state, event);
246 m_PDU_Observer->failNotify();
250 void PDU_Assoc::close_session()
252 m_session_is_dead = true;
256 m_PDU_Observer->failNotify();
260 void PDU_Assoc::shutdown()
263 for (ch = m_children; ch; ch = ch->m_next)
266 m_socketObservable->deleteObserver(this);
270 yaz_log (m_log, "PDU_Assoc::close fd=%d", cs_fileno(m_cs));
276 PDU_Queue *q_this = m_queue_out;
277 m_queue_out = m_queue_out->m_next;
285 void PDU_Assoc::destroy()
293 // delete from parent's child list (if any)
296 c = &m_parent->m_children;
304 // delete all children ...
308 PDU_Assoc *here = *c;
313 yaz_log (m_log, "PDU_Assoc::destroy this=%p", this);
316 PDU_Assoc::PDU_Queue::PDU_Queue(const char *buf, int len)
318 m_buf = (char *) xmalloc (len);
319 memcpy (m_buf, buf, len);
324 PDU_Assoc::PDU_Queue::~PDU_Queue()
329 int PDU_Assoc::flush_PDU()
333 if (m_state != Ready && m_state != Writing)
335 yaz_log (m_log, "YAZ_PDU_Assoc::flush_PDU, not ready");
338 PDU_Queue *q = m_queue_out;
342 yaz_log (m_log, "YAZ_PDU_Assoc::flush_PDU queue empty");
343 yaz_log(m_log, "maskObserver 6");
344 m_socketObservable->maskObserver(this, SOCKET_OBSERVE_READ|
345 SOCKET_OBSERVE_WRITE|
346 SOCKET_OBSERVE_EXCEPT);
347 if (m_session_is_dead)
350 m_PDU_Observer->failNotify();
354 r = cs_put (m_cs, q->m_buf, q->m_len);
357 yaz_log (m_log, "PDU_Assoc::flush_PDU cs_put failed");
359 m_PDU_Observer->failNotify();
364 unsigned mask = SOCKET_OBSERVE_EXCEPT;
366 if (m_cs->io_pending & CS_WANT_WRITE)
367 mask |= SOCKET_OBSERVE_WRITE;
368 if (m_cs->io_pending & CS_WANT_READ)
369 mask |= SOCKET_OBSERVE_READ;
371 mask |= SOCKET_OBSERVE_WRITE;
372 yaz_log(m_log, "maskObserver 7");
373 m_socketObservable->maskObserver(this, mask);
374 yaz_log (m_log, "PDU_Assoc::flush_PDU cs_put %d bytes fd=%d (inc)",
375 q->m_len, cs_fileno(m_cs));
378 yaz_log (m_log, "PDU_Assoc::flush_PDU cs_put %d bytes", q->m_len);
379 // whole packet sent... delete this and proceed to next ...
380 m_queue_out = q->m_next;
382 // don't select on write if queue is empty ...
386 yaz_log(m_log, "maskObserver 8");
387 m_socketObservable->maskObserver(this, SOCKET_OBSERVE_READ|
388 SOCKET_OBSERVE_EXCEPT);
393 int PDU_Assoc::send_PDU(const char *buf, int len)
395 yaz_log (m_log, "PDU_Assoc::send_PDU");
396 PDU_Queue **pq = &m_queue_out;
397 int is_idle = (*pq ? 0 : 1);
401 yaz_log (m_log, "PDU_Assoc::send_PDU failed, m_cs == 0");
406 *pq = new PDU_Queue(buf, len);
410 yaz_log (m_log, "PDU_Assoc::cannot send_PDU fd=%d",
415 COMSTACK PDU_Assoc::comstack(const char *type_and_host, void **vp)
417 return cs_create_host(type_and_host, 2, vp);
420 int PDU_Assoc::listen(IPDU_Observer *observer, const char *addr)
424 m_PDU_Observer = observer;
426 m_cs = comstack(addr, &ap);
430 if (cs_bind(m_cs, ap, CS_SERVER) < 0)
433 int fd = cs_fileno(m_cs);
435 int oldflags = fcntl(fd, F_GETFD, 0);
438 oldflags |= FD_CLOEXEC;
439 fcntl(fd, F_SETFD, oldflags);
442 m_socketObservable->addObserver(fd, this);
443 yaz_log(m_log, "maskObserver 9");
444 m_socketObservable->maskObserver(this, SOCKET_OBSERVE_READ|
445 SOCKET_OBSERVE_EXCEPT);
446 yaz_log (m_log, "PDU_Assoc::listen ok fd=%d", fd);
451 void PDU_Assoc::idleTime(int idleTime)
453 m_idleTime = idleTime;
454 yaz_log (m_log, "PDU_Assoc::idleTime(%d)", idleTime);
455 m_socketObservable->timeoutObserver(this, m_idleTime);
458 int PDU_Assoc::connect(IPDU_Observer *observer, const char *addr)
460 yaz_log (m_log, "PDU_Assoc::connect %s", addr);
462 m_PDU_Observer = observer;
464 m_cs = comstack(addr, &ap);
467 int res = cs_connect (m_cs, ap);
468 yaz_log (m_log, "PDU_Assoc::connect fd=%d res=%d", cs_fileno(m_cs),
470 m_socketObservable->addObserver(cs_fileno(m_cs), this);
473 { // Connect complete
474 m_state = Connecting;
475 unsigned mask = SOCKET_OBSERVE_EXCEPT;
476 mask |= SOCKET_OBSERVE_WRITE;
477 mask |= SOCKET_OBSERVE_READ;
478 yaz_log(m_log, "maskObserver 11");
479 m_socketObservable->maskObserver(this, mask);
483 m_state = Connecting;
484 unsigned mask = SOCKET_OBSERVE_EXCEPT;
485 if (m_cs->io_pending & CS_WANT_WRITE)
486 mask |= SOCKET_OBSERVE_WRITE;
487 if (m_cs->io_pending & CS_WANT_READ)
488 mask |= SOCKET_OBSERVE_READ;
489 yaz_log(m_log, "maskObserver 11");
490 m_socketObservable->maskObserver(this, mask);
493 { // Connect failed immediately
494 // Since m_state is Closed we can distinguish this case from
495 // normal connect in socketNotify handler
496 yaz_log(m_log, "maskObserver 12");
497 m_socketObservable->maskObserver(this, SOCKET_OBSERVE_WRITE|
498 SOCKET_OBSERVE_EXCEPT);
503 // Single-threaded... Only useful for non-blocking handlers
504 void PDU_Assoc::childNotify(COMSTACK cs)
506 PDU_Assoc *new_observable =
507 new PDU_Assoc (m_socketObservable, cs);
509 // Clone PDU Observer
510 new_observable->m_PDU_Observer = m_PDU_Observer->sessionNotify
511 (new_observable, cs_fileno(cs));
513 if (!new_observable->m_PDU_Observer)
515 new_observable->shutdown();
516 delete new_observable;
519 new_observable->m_next = m_children;
520 m_children = new_observable;
521 new_observable->m_parent = this;
524 const char*PDU_Assoc::getpeername()
528 return cs_addrstr(m_cs);
533 * c-file-style: "Stroustrup"
534 * indent-tabs-mode: nil
536 * vim: shiftwidth=4 tabstop=8 expandtab