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.9 1999-12-06 13:52:45 adam
8 * Modified for new location of YAZ header files. Experimental threaded
11 * Revision 1.8 1999/04/28 13:04:03 adam
12 * Fixed setting of proxy otherInfo so that database(s) are removed.
14 * Revision 1.7 1999/04/21 12:09:01 adam
15 * Many improvements. Modified to proxy server to work with "sessions"
18 * Revision 1.6 1999/04/20 10:30:05 adam
19 * Implemented various stuff for client and proxy. Updated calls
20 * to ODR to reflect new name parameter.
22 * Revision 1.5 1999/04/09 11:46:57 adam
23 * Added object Yaz_Z_Assoc. Much more functional client.
25 * Revision 1.4 1999/03/23 14:17:57 adam
26 * More work on timeout handling. Work on yaz-client.
28 * Revision 1.3 1999/02/02 14:01:20 adam
29 * First WIN32 port of YAZ++.
31 * Revision 1.2 1999/01/28 13:08:44 adam
32 * Yaz_PDU_Assoc better encapsulated. Memory leak fix in
33 * yaz-socket-manager.cc.
35 * Revision 1.1.1.1 1999/01/28 09:41:07 adam
36 * First implementation of YAZ++.
42 #include <yaz-pdu-assoc.h>
45 #include <yaz/tcpip.h>
47 Yaz_PDU_Assoc::Yaz_PDU_Assoc(IYazSocketObservable *socketObservable)
51 m_socketObservable = socketObservable;
64 IYaz_PDU_Observable *Yaz_PDU_Assoc::clone()
66 Yaz_PDU_Assoc *copy = new Yaz_PDU_Assoc(m_socketObservable);
70 Yaz_PDU_Assoc::~Yaz_PDU_Assoc()
75 void Yaz_PDU_Assoc::socketNotify(int event)
77 logf (m_log, "Yaz_PDU_Assoc::socketNotify p=%p event = %d", this, event);
78 if (0 /* m_state == Connected */)
81 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
82 YAZ_SOCKET_OBSERVE_EXCEPT);
83 m_PDU_Observer->connectNotify();
86 else if (m_state == Connecting)
88 if (event & YAZ_SOCKET_OBSERVE_READ)
91 m_PDU_Observer->failNotify();
93 else if (event & YAZ_SOCKET_OBSERVE_TIMEOUT)
95 m_PDU_Observer->timeoutNotify();
100 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
101 YAZ_SOCKET_OBSERVE_EXCEPT);
102 m_PDU_Observer->connectNotify();
106 else if (m_state == Listen)
108 if (event & YAZ_SOCKET_OBSERVE_READ)
113 if ((res = cs_listen(m_cs, 0, 0)) == 1)
117 logf(LOG_FATAL, "cs_listen failed");
120 if (!(new_line = cs_accept(m_cs)))
123 /* 1. create socket-manager
125 3. create top-level object
126 setup observer for child fileid in pdu-assoc
129 int fd = cs_fileno(new_line);
130 cs_fileno(new_line) = -1;
135 Yaz_PDU_Assoc *assoc = new Yaz_PDU_Assoc (m_socketObservable);
136 assoc->m_parent = this;
137 assoc->m_next = m_children;
140 assoc->m_PDU_Observer = m_PDU_Observer->clone(assoc);
145 else if (m_state == Ready)
147 if (event & YAZ_SOCKET_OBSERVE_WRITE)
151 if (event & YAZ_SOCKET_OBSERVE_READ)
155 int res = cs_get (m_cs, &m_input_buf, &m_input_len);
160 logf (m_log, "Connection closed by peer");
162 m_PDU_Observer->failNotify();
165 // lock it, so we know if recv_PDU deletes it.
167 m_destroyed = &destroyed;
169 m_PDU_Observer->recv_PDU(m_input_buf, res);
170 if (destroyed) // it really was destroyed, return now.
172 } while (m_cs && cs_more (m_cs));
174 if (event & YAZ_SOCKET_OBSERVE_TIMEOUT)
176 m_PDU_Observer->timeoutNotify();
181 void Yaz_PDU_Assoc::close()
183 m_socketObservable->deleteObserver(this);
187 logf (m_log, "Yaz_PDU_Assoc::close fd=%d", cs_fileno(m_cs));
193 PDU_Queue *q_this = m_queue_out;
194 m_queue_out = m_queue_out->m_next;
197 // free (m_input_buf);
202 void Yaz_PDU_Assoc::destroy()
209 // delete from parent's child list (if any)
212 c = &m_parent->m_children;
220 // delete all children ...
224 Yaz_PDU_Assoc *here = *c;
231 Yaz_PDU_Assoc::PDU_Queue::PDU_Queue(const char *buf, int len)
233 m_buf = (char *) malloc (len);
234 memcpy (m_buf, buf, len);
239 Yaz_PDU_Assoc::PDU_Queue::~PDU_Queue()
244 int Yaz_PDU_Assoc::flush_PDU()
248 logf (m_log, "Yaz_PDU_Assoc::flush_PDU");
249 if (m_state != Ready)
251 logf (m_log, "YAZ_PDU_Assoc::flush_PDU, not ready");
254 PDU_Queue *q = m_queue_out;
257 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
258 YAZ_SOCKET_OBSERVE_EXCEPT);
261 r = cs_put (m_cs, q->m_buf, q->m_len);
265 m_PDU_Observer->failNotify();
270 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
271 YAZ_SOCKET_OBSERVE_EXCEPT|
272 YAZ_SOCKET_OBSERVE_WRITE);
273 logf (m_log, "Yaz_PDU_Assoc::flush_PDU put %d bytes (incomplete)",
277 logf (m_log, "Yaz_PDU_Assoc::flush_PDU put %d bytes", q->m_len);
278 // whole packet sent... delete this and proceed to next ...
279 m_queue_out = q->m_next;
281 // don't select on write if queue is empty ...
283 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
284 YAZ_SOCKET_OBSERVE_EXCEPT);
288 int Yaz_PDU_Assoc::send_PDU(const char *buf, int len)
290 logf (m_log, "Yaz_PDU_Assoc::send_PDU");
291 PDU_Queue **pq = &m_queue_out;
292 int is_idle = (*pq ? 0 : 1);
296 logf (m_log, "Yaz_PDU_Assoc::send_PDU failed, m_cs == 0");
301 *pq = new PDU_Queue(buf, len);
305 logf (m_log, "Yaz_PDU_Assoc::cannot send_PDU fd=%d",
310 COMSTACK Yaz_PDU_Assoc::comstack()
314 CS_TYPE cs_type = tcpip_type;
315 m_cs = cs_create (cs_type, 0, PROTO_Z3950);
320 void Yaz_PDU_Assoc::listen(IYaz_PDU_Observer *observer,
325 COMSTACK cs = comstack();
327 logf (m_log, "Yaz_PDU_Assoc::listen %s", addr);
328 m_PDU_Observer = observer;
331 ap = cs_straddr (cs, addr);
334 if (cs_bind(cs, ap, CS_SERVER) < 0)
336 m_socketObservable->addObserver(cs_fileno(cs), this);
337 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
338 YAZ_SOCKET_OBSERVE_EXCEPT);
342 void Yaz_PDU_Assoc::idleTime(int idleTime)
344 m_idleTime = idleTime;
345 logf (m_log, "Yaz_PDU_Assoc::idleTime(%d)", idleTime);
346 m_socketObservable->timeoutObserver(this, m_idleTime);
349 void Yaz_PDU_Assoc::connect(IYaz_PDU_Observer *observer,
352 logf (m_log, "Yaz_PDU_Assoc::connect %s", addr);
354 m_PDU_Observer = observer;
355 COMSTACK cs = comstack();
356 void *ap = cs_straddr (cs, addr);
359 logf (m_log, "cs_straddr failed");
362 int res = cs_connect (cs, ap);
363 logf (m_log, "Yaz_PDU_Assoc::connect fd=%d res=%d", cs_fileno(cs), res);
364 m_socketObservable->addObserver(cs_fileno(cs), this);
365 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
366 YAZ_SOCKET_OBSERVE_EXCEPT|
367 YAZ_SOCKET_OBSERVE_WRITE);
368 m_state = Connecting;
371 void Yaz_PDU_Assoc::socket(IYaz_PDU_Observer *observer, int fd)
374 m_PDU_Observer = observer;
377 CS_TYPE cs_type = tcpip_type;
378 m_cs = cs_createbysocket(fd, cs_type, 0, PROTO_Z3950);
380 m_socketObservable->addObserver(fd, this);
381 m_socketObservable->maskObserver(this,
382 YAZ_SOCKET_OBSERVE_READ|
383 YAZ_SOCKET_OBSERVE_EXCEPT);
384 m_socketObservable->timeoutObserver(this, m_idleTime);
389 void Yaz_PDU_Assoc::childNotify(int fd)
391 /// Clone PDU Observable (keep socket manager)
392 IYaz_PDU_Observable *new_observable = clone();
394 /// Clone PDU Observer
395 IYaz_PDU_Observer *observer = m_PDU_Observer->clone(new_observable);
397 /// Attach new socket to it
398 new_observable->socket(observer, fd);
402 #include <yaz-socket-manager.h>
406 Yaz_SocketManager *socketManager;
407 IYaz_PDU_Observable *
411 static void *events(void *p)
413 Yaz_SocketManager *s = (Yaz_SocketManager *) p;
415 while (s->processEvent() > 0)
420 void Yaz_PDU_Assoc::childNotify(int fd)
422 Yaz_SocketManager *socket_observable = new Yaz_SocketManager;
423 IYaz_PDU_Observable *new_observable = clone();
425 m_socketObservable = socket_observable;
427 /// Clone PDU Observer
428 IYaz_PDU_Observer *observer = m_PDU_Observer->clone(new_observable);
430 /// Attach new socket to it
431 new_observable->socket(observer, fd);
435 int id = pthread_create (&type, 0, events, socket_observable);
436 logf (LOG_LOG, "pthread_create returned id=%d", id);