2 * Copyright (c) 1998-1999, Index Data.
3 * See the file LICENSE for details.
4 * Sebastian Hammer, Adam Dickmeiss
6 * $Log: yaz-pdu-assoc.cpp,v $
7 * Revision 1.12 2000-09-06 14:23:45 adam
10 * Revision 1.11 2000/09/04 08:29:22 adam
11 * Fixed memory leak(s). Added re-use of associations, rather than
12 * re-init, when maximum number of targets are in use.
14 * Revision 1.10 2000/08/10 08:42:42 adam
15 * Fixes for {set,get}_APDU_log.
17 * Revision 1.9 1999/12/06 13:52:45 adam
18 * Modified for new location of YAZ header files. Experimental threaded
21 * Revision 1.8 1999/04/28 13:04:03 adam
22 * Fixed setting of proxy otherInfo so that database(s) are removed.
24 * Revision 1.7 1999/04/21 12:09:01 adam
25 * Many improvements. Modified to proxy server to work with "sessions"
28 * Revision 1.6 1999/04/20 10:30:05 adam
29 * Implemented various stuff for client and proxy. Updated calls
30 * to ODR to reflect new name parameter.
32 * Revision 1.5 1999/04/09 11:46:57 adam
33 * Added object Yaz_Z_Assoc. Much more functional client.
35 * Revision 1.4 1999/03/23 14:17:57 adam
36 * More work on timeout handling. Work on yaz-client.
38 * Revision 1.3 1999/02/02 14:01:20 adam
39 * First WIN32 port of YAZ++.
41 * Revision 1.2 1999/01/28 13:08:44 adam
42 * Yaz_PDU_Assoc better encapsulated. Memory leak fix in
43 * yaz-socket-manager.cc.
45 * Revision 1.1.1.1 1999/01/28 09:41:07 adam
46 * First implementation of YAZ++.
52 #include <yaz-pdu-assoc.h>
55 #include <yaz/tcpip.h>
57 Yaz_PDU_Assoc::Yaz_PDU_Assoc(IYazSocketObservable *socketObservable)
61 m_socketObservable = socketObservable;
74 IYaz_PDU_Observable *Yaz_PDU_Assoc::clone()
76 Yaz_PDU_Assoc *copy = new Yaz_PDU_Assoc(m_socketObservable);
80 Yaz_PDU_Assoc::~Yaz_PDU_Assoc()
85 void Yaz_PDU_Assoc::socketNotify(int event)
87 logf (m_log, "Yaz_PDU_Assoc::socketNotify p=%p event = %d", this, event);
88 if (0 /* m_state == Connected */)
91 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
92 YAZ_SOCKET_OBSERVE_EXCEPT);
93 m_PDU_Observer->connectNotify();
96 else if (m_state == Connecting)
98 if (event & YAZ_SOCKET_OBSERVE_READ)
101 m_PDU_Observer->failNotify();
103 else if (event & YAZ_SOCKET_OBSERVE_TIMEOUT)
105 m_PDU_Observer->timeoutNotify();
110 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
111 YAZ_SOCKET_OBSERVE_EXCEPT);
112 m_PDU_Observer->connectNotify();
116 else if (m_state == Listen)
118 if (event & YAZ_SOCKET_OBSERVE_READ)
123 if ((res = cs_listen(m_cs, 0, 0)) == 1)
127 logf(LOG_FATAL|LOG_ERRNO, "cs_listen failed");
130 if (!(new_line = cs_accept(m_cs)))
133 /* 1. create socket-manager
135 3. create top-level object
136 setup observer for child fileid in pdu-assoc
139 int fd = cs_fileno(new_line);
140 cs_fileno(new_line) = -1;
141 cs_close (new_line); /* potential problem ... */
145 Yaz_PDU_Assoc *assoc = new Yaz_PDU_Assoc (m_socketObservable);
146 assoc->m_parent = this;
147 assoc->m_next = m_children;
150 assoc->m_PDU_Observer = m_PDU_Observer->clone(assoc);
155 else if (m_state == Ready)
157 if (event & YAZ_SOCKET_OBSERVE_WRITE)
161 if (event & YAZ_SOCKET_OBSERVE_READ)
165 int res = cs_get (m_cs, &m_input_buf, &m_input_len);
170 logf (m_log, "Connection closed by peer");
172 m_PDU_Observer->failNotify();
175 // lock it, so we know if recv_PDU deletes it.
177 m_destroyed = &destroyed;
179 m_PDU_Observer->recv_PDU(m_input_buf, res);
181 if (destroyed) // it really was destroyed, return now.
183 } while (m_cs && cs_more (m_cs));
185 if (event & YAZ_SOCKET_OBSERVE_TIMEOUT)
187 m_PDU_Observer->timeoutNotify();
192 void Yaz_PDU_Assoc::close()
194 m_socketObservable->deleteObserver(this);
198 logf (m_log, "Yaz_PDU_Assoc::close fd=%d", cs_fileno(m_cs));
204 PDU_Queue *q_this = m_queue_out;
205 m_queue_out = m_queue_out->m_next;
213 void Yaz_PDU_Assoc::destroy()
220 // delete from parent's child list (if any)
223 c = &m_parent->m_children;
231 // delete all children ...
235 Yaz_PDU_Assoc *here = *c;
242 Yaz_PDU_Assoc::PDU_Queue::PDU_Queue(const char *buf, int len)
244 m_buf = (char *) malloc (len);
245 memcpy (m_buf, buf, len);
250 Yaz_PDU_Assoc::PDU_Queue::~PDU_Queue()
255 int Yaz_PDU_Assoc::flush_PDU()
259 logf (m_log, "Yaz_PDU_Assoc::flush_PDU");
260 if (m_state != Ready)
262 logf (m_log, "YAZ_PDU_Assoc::flush_PDU, not ready");
265 PDU_Queue *q = m_queue_out;
268 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
269 YAZ_SOCKET_OBSERVE_EXCEPT);
272 r = cs_put (m_cs, q->m_buf, q->m_len);
276 m_PDU_Observer->failNotify();
281 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
282 YAZ_SOCKET_OBSERVE_EXCEPT|
283 YAZ_SOCKET_OBSERVE_WRITE);
284 logf (m_log, "Yaz_PDU_Assoc::flush_PDU put %d bytes (incomplete)",
288 logf (m_log, "Yaz_PDU_Assoc::flush_PDU put %d bytes", q->m_len);
289 // whole packet sent... delete this and proceed to next ...
290 m_queue_out = q->m_next;
292 // don't select on write if queue is empty ...
294 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
295 YAZ_SOCKET_OBSERVE_EXCEPT);
299 int Yaz_PDU_Assoc::send_PDU(const char *buf, int len)
301 logf (m_log, "Yaz_PDU_Assoc::send_PDU");
302 PDU_Queue **pq = &m_queue_out;
303 int is_idle = (*pq ? 0 : 1);
307 logf (m_log, "Yaz_PDU_Assoc::send_PDU failed, m_cs == 0");
312 *pq = new PDU_Queue(buf, len);
316 logf (m_log, "Yaz_PDU_Assoc::cannot send_PDU fd=%d",
321 COMSTACK Yaz_PDU_Assoc::comstack()
325 CS_TYPE cs_type = tcpip_type;
326 m_cs = cs_create (cs_type, 0, PROTO_Z3950);
331 void Yaz_PDU_Assoc::listen(IYaz_PDU_Observer *observer,
336 COMSTACK cs = comstack();
338 logf (m_log, "Yaz_PDU_Assoc::listen %s", addr);
339 m_PDU_Observer = observer;
342 ap = cs_straddr (cs, addr);
345 if (cs_bind(cs, ap, CS_SERVER) < 0)
347 m_socketObservable->addObserver(cs_fileno(cs), this);
348 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
349 YAZ_SOCKET_OBSERVE_EXCEPT);
353 void Yaz_PDU_Assoc::idleTime(int idleTime)
355 m_idleTime = idleTime;
356 logf (m_log, "Yaz_PDU_Assoc::idleTime(%d)", idleTime);
357 m_socketObservable->timeoutObserver(this, m_idleTime);
360 void Yaz_PDU_Assoc::connect(IYaz_PDU_Observer *observer,
363 logf (m_log, "Yaz_PDU_Assoc::connect %s", addr);
365 m_PDU_Observer = observer;
366 COMSTACK cs = comstack();
367 void *ap = cs_straddr (cs, addr);
370 logf (m_log, "cs_straddr failed");
373 int res = cs_connect (cs, ap);
374 logf (m_log, "Yaz_PDU_Assoc::connect fd=%d res=%d", cs_fileno(cs), res);
375 m_socketObservable->addObserver(cs_fileno(cs), this);
376 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
377 YAZ_SOCKET_OBSERVE_EXCEPT|
378 YAZ_SOCKET_OBSERVE_WRITE);
379 m_state = Connecting;
382 void Yaz_PDU_Assoc::socket(IYaz_PDU_Observer *observer, int fd)
385 m_PDU_Observer = observer;
388 CS_TYPE cs_type = tcpip_type;
389 m_cs = cs_createbysocket(fd, cs_type, 0, PROTO_Z3950);
391 m_socketObservable->addObserver(fd, this);
392 m_socketObservable->maskObserver(this,
393 YAZ_SOCKET_OBSERVE_READ|
394 YAZ_SOCKET_OBSERVE_EXCEPT);
395 m_socketObservable->timeoutObserver(this, m_idleTime);
401 // Single-threaded... Only useful for non-blocking handlers
402 void Yaz_PDU_Assoc::childNotify(int fd)
404 /// Clone PDU Observable (keep socket manager)
405 IYaz_PDU_Observable *new_observable = clone();
407 /// Clone PDU Observer
408 IYaz_PDU_Observer *observer = m_PDU_Observer->clone(new_observable);
410 /// Attach new socket to it
411 new_observable->socket(observer, fd);
415 #include <yaz-socket-manager.h>
429 Yaz_SocketManager *s = (Yaz_SocketManager *) p;
431 logf (LOG_LOG, "thread started");
432 while (s->processEvent() > 0)
434 logf (LOG_LOG, "thread finished");
441 void Yaz_PDU_Assoc::childNotify(int fd)
443 Yaz_SocketManager *socket_observable = new Yaz_SocketManager;
444 Yaz_PDU_Assoc *new_observable = new Yaz_PDU_Assoc (socket_observable);
446 /// Clone PDU Observer
447 IYaz_PDU_Observer *observer = m_PDU_Observer->clone(new_observable);
449 /// Attach new socket to it
450 new_observable->socket(observer, fd);
454 t_id = _beginthread (events, 0, socket_observable);
457 logf (LOG_FATAL|LOG_ERRNO, "_beginthread failed");
463 int id = pthread_create (&type, 0, events, socket_observable);
464 logf (LOG_LOG, "pthread_create returned id=%d", id);