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.5 1999-04-09 11:46:57 adam
8 * Added object Yaz_Z_Assoc. Much more functional client.
10 * Revision 1.4 1999/03/23 14:17:57 adam
11 * More work on timeout handling. Work on yaz-client.
13 * Revision 1.3 1999/02/02 14:01:20 adam
14 * First WIN32 port of YAZ++.
16 * Revision 1.2 1999/01/28 13:08:44 adam
17 * Yaz_PDU_Assoc better encapsulated. Memory leak fix in
18 * yaz-socket-manager.cc.
20 * Revision 1.1.1.1 1999/01/28 09:41:07 adam
21 * First implementation of YAZ++.
27 #include <yaz-pdu-assoc.h>
32 Yaz_PDU_Assoc::Yaz_PDU_Assoc(IYazSocketObservable *socketObservable,
37 m_socketObservable = socketObservable;
48 IYaz_PDU_Observable *Yaz_PDU_Assoc::clone()
50 Yaz_PDU_Assoc *copy = new Yaz_PDU_Assoc(m_socketObservable, 0);
54 Yaz_PDU_Assoc::~Yaz_PDU_Assoc()
59 void Yaz_PDU_Assoc::socketNotify(int event)
61 logf (LOG_LOG, "Yaz_PDU_Assoc::socketNotify p=%p event = %d", this, event);
62 if (m_state == Connected)
65 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
66 YAZ_SOCKET_OBSERVE_EXCEPT);
67 m_PDU_Observer->connectNotify();
70 else if (m_state == Connecting)
72 if (event & YAZ_SOCKET_OBSERVE_READ)
75 m_PDU_Observer->failNotify();
80 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
81 YAZ_SOCKET_OBSERVE_EXCEPT);
82 m_PDU_Observer->connectNotify();
86 else if (m_state == Listen)
88 if (event & YAZ_SOCKET_OBSERVE_READ)
93 if ((res = cs_listen(m_cs, 0, 0)) == 1)
97 logf(LOG_FATAL, "cs_listen failed");
100 if (!(new_line = cs_accept(m_cs)))
103 Yaz_PDU_Assoc *assoc = new Yaz_PDU_Assoc (m_socketObservable,
105 assoc->m_parent = this;
106 assoc->m_next = m_children;
109 assoc->m_PDU_Observer = m_PDU_Observer->clone(assoc);
110 assoc->m_state = Ready;
111 assoc->m_socketObservable->addObserver(cs_fileno(new_line), assoc);
112 assoc->m_socketObservable->maskObserver(assoc,
113 YAZ_SOCKET_OBSERVE_READ|
114 YAZ_SOCKET_OBSERVE_EXCEPT);
116 assoc->m_socketObservable->timeoutObserver(assoc, m_idleTime);
119 else if (m_state == Ready)
121 if (event & YAZ_SOCKET_OBSERVE_WRITE)
125 if (event & YAZ_SOCKET_OBSERVE_READ)
129 int res = cs_get (m_cs, &m_input_buf, &m_input_len);
134 logf (LOG_LOG, "Connection closed by client");
136 m_PDU_Observer->failNotify();
139 // lock it, so we know if recv_PDU deletes it.
141 m_destroyed = &destroyed;
143 m_PDU_Observer->recv_PDU(m_input_buf, res);
144 if (destroyed) // it really was destroyed, return now.
146 } while (m_cs && cs_more (m_cs));
148 if (event & YAZ_SOCKET_OBSERVE_TIMEOUT)
150 m_PDU_Observer->timeoutNotify();
155 void Yaz_PDU_Assoc::close()
157 m_socketObservable->deleteObserver(this);
161 logf (LOG_LOG, "Yaz_PDU_Assoc::close fd=%d", cs_fileno(m_cs));
167 PDU_Queue *q_this = m_queue_out;
168 m_queue_out = m_queue_out->m_next;
171 // free (m_input_buf);
176 void Yaz_PDU_Assoc::destroy()
183 // delete from parent's child list (if any)
186 c = &m_parent->m_children;
194 // delete all children ...
198 Yaz_PDU_Assoc *here = *c;
205 Yaz_PDU_Assoc::PDU_Queue::PDU_Queue(const char *buf, int len)
207 m_buf = (char *) malloc (len);
208 memcpy (m_buf, buf, len);
213 Yaz_PDU_Assoc::PDU_Queue::~PDU_Queue()
218 int Yaz_PDU_Assoc::flush_PDU()
222 logf (LOG_LOG, "Yaz_PDU_Assoc::flush_PDU");
223 if (m_state != Ready)
227 PDU_Queue *q = m_queue_out;
230 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
231 YAZ_SOCKET_OBSERVE_EXCEPT);
234 r = cs_put (m_cs, q->m_buf, q->m_len);
238 m_PDU_Observer->failNotify();
243 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
244 YAZ_SOCKET_OBSERVE_EXCEPT|
245 YAZ_SOCKET_OBSERVE_WRITE);
246 logf (LOG_LOG, "flush_PDU put %d bytes (incomplete write)", q->m_len);
249 logf (LOG_LOG, "flush_PDU put %d bytes fd=%d", q->m_len, cs_fileno(m_cs));
250 // whole packet sent... delete this and proceed to next ...
251 m_queue_out = q->m_next;
253 // don't select on write if queue is empty ...
255 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
256 YAZ_SOCKET_OBSERVE_EXCEPT);
260 int Yaz_PDU_Assoc::send_PDU(const char *buf, int len)
262 logf (LOG_LOG, "Yaz_PDU_Assoc::send_PDU");
263 PDU_Queue **pq = &m_queue_out;
264 int is_idle = (*pq ? 0 : 1);
268 logf (LOG_LOG, "send_PDU failed, m_cs == 0");
273 *pq = new PDU_Queue(buf, len);
277 logf (LOG_LOG, "cannot send_PDU fd=%d", cs_fileno(m_cs));
281 COMSTACK Yaz_PDU_Assoc::comstack()
285 CS_TYPE cs_type = tcpip_type;
286 int protocol = PROTO_Z3950;
287 m_cs = cs_create (cs_type, 0, protocol);
292 void Yaz_PDU_Assoc::listen(IYaz_PDU_Observer *observer,
297 COMSTACK cs = comstack();
299 logf (LOG_LOG, "Yaz_PDU_Assoc::listen %s", addr);
300 m_PDU_Observer = observer;
303 ap = cs_straddr (cs, addr);
306 if (cs_bind(cs, ap, CS_SERVER) < 0)
308 m_socketObservable->addObserver(cs_fileno(cs), this);
309 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
310 YAZ_SOCKET_OBSERVE_EXCEPT);
314 void Yaz_PDU_Assoc::idleTime(int idleTime)
316 m_idleTime = idleTime;
319 void Yaz_PDU_Assoc::connect(IYaz_PDU_Observer *observer,
322 logf (LOG_LOG, "Yaz_PDU_Assoc::connect %s", addr);
324 m_PDU_Observer = observer;
325 COMSTACK cs = comstack();
326 void *ap = cs_straddr (cs, addr);
329 logf (LOG_LOG, "cs_straddr failed");
332 int res = cs_connect (cs, ap);
335 logf (LOG_DEBUG, "Yaz_PDU_Assoc::connect failed");
340 logf (LOG_LOG, "Yaz_PDU_Assoc::connect fd=%d", cs_fileno(cs));
341 m_socketObservable->addObserver(cs_fileno(cs), this);
342 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
343 YAZ_SOCKET_OBSERVE_EXCEPT|
344 YAZ_SOCKET_OBSERVE_WRITE);
347 logf (LOG_LOG, "Connect pending");
348 m_state = Connecting;
352 logf (LOG_LOG, "Connect complete");