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.2 1999-01-28 13:08:44 adam
8 * Yaz_PDU_Assoc better encapsulated. Memory leak fix in
9 * yaz-socket-manager.cc.
11 * Revision 1.1.1.1 1999/01/28 09:41:07 adam
12 * First implementation of YAZ++.
18 #include <yaz-pdu-assoc.h>
23 Yaz_PDU_Assoc::Yaz_PDU_Assoc(IYazSocketObservable *socketObservable,
28 m_socketObservable = socketObservable;
39 IYaz_PDU_Observable *Yaz_PDU_Assoc::clone()
41 Yaz_PDU_Assoc *copy = new Yaz_PDU_Assoc(m_socketObservable, 0);
45 Yaz_PDU_Assoc::~Yaz_PDU_Assoc()
50 void Yaz_PDU_Assoc::socketNotify(int event)
52 logf (LOG_LOG, "socketNotify p=%p event = %d", this, event);
53 if (m_state == Connected)
56 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
57 YAZ_SOCKET_OBSERVE_EXCEPT);
58 m_PDU_Observer->connectNotify();
61 else if (m_state == Connecting)
63 if (event & YAZ_SOCKET_OBSERVE_READ)
66 m_PDU_Observer->failNotify();
71 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
72 YAZ_SOCKET_OBSERVE_EXCEPT);
73 m_PDU_Observer->connectNotify();
77 else if (m_state == Listen)
79 if (event & YAZ_SOCKET_OBSERVE_READ)
84 if ((res = cs_listen(m_cs, 0, 0)) == 1)
88 logf(LOG_FATAL, "cs_listen failed");
91 if (!(new_line = cs_accept(m_cs)))
94 Yaz_PDU_Assoc *assoc = new Yaz_PDU_Assoc (m_socketObservable,
96 assoc->m_parent = this;
97 assoc->m_next = m_children;
100 assoc->m_PDU_Observer = m_PDU_Observer->clone(assoc);
101 assoc->m_state = Ready;
102 assoc->m_socketObservable->addObserver(cs_fileno(new_line), assoc);
103 assoc->m_socketObservable->maskObserver(assoc,
104 YAZ_SOCKET_OBSERVE_READ|
105 YAZ_SOCKET_OBSERVE_EXCEPT);
108 else if (m_state == Ready)
110 if (event & YAZ_SOCKET_OBSERVE_WRITE)
114 if (event & YAZ_SOCKET_OBSERVE_READ)
118 int res = cs_get (m_cs, &m_input_buf, &m_input_len);
123 logf (LOG_LOG, "Connection closed by client");
125 m_PDU_Observer->failNotify();
128 // lock it, so we know if recv_PDU deletes it.
130 m_destroyed = &destroyed;
132 m_PDU_Observer->recv_PDU(m_input_buf, res);
133 if (destroyed) // it really was destroyed, return now.
135 } while (m_cs && cs_more (m_cs));
140 void Yaz_PDU_Assoc::close()
142 m_socketObservable->deleteObserver(this);
146 logf (LOG_LOG, "Yaz_PDU_Assoc::close fd=%d", cs_fileno(m_cs));
150 PDU_Queue **q = &m_queue_out;
153 PDU_Queue *q_this = *q;
162 void Yaz_PDU_Assoc::destroy()
169 // delete from parent's child list (if any)
172 c = &m_parent->m_children;
180 // delete all children ...
184 Yaz_PDU_Assoc *here = *c;
191 Yaz_PDU_Assoc::PDU_Queue::PDU_Queue(const char *buf, int len)
193 m_buf = (char *) malloc (len);
194 memcpy (m_buf, buf, len);
199 Yaz_PDU_Assoc::PDU_Queue::~PDU_Queue()
204 int Yaz_PDU_Assoc::flush_PDU()
208 if (m_state != Ready)
210 PDU_Queue *q = m_queue_out;
213 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
214 YAZ_SOCKET_OBSERVE_EXCEPT);
217 r = cs_put (m_cs, q->m_buf, q->m_len);
221 m_PDU_Observer->failNotify();
226 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
227 YAZ_SOCKET_OBSERVE_EXCEPT|
228 YAZ_SOCKET_OBSERVE_WRITE);
229 logf (LOG_LOG, "put %d bytes (incomplete write)", q->m_len);
232 logf (LOG_LOG, "put %d bytes fd=%d", q->m_len, cs_fileno(m_cs));
233 // whole packet sent... delete this and proceed to next ...
234 m_queue_out = q->m_next;
235 logf (LOG_LOG, "m_queue_out = %p", m_queue_out);
237 // don't select on write if queue is empty ...
239 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
240 YAZ_SOCKET_OBSERVE_EXCEPT);
244 int Yaz_PDU_Assoc::send_PDU(const char *buf, int len)
246 PDU_Queue **pq = &m_queue_out;
247 int is_idle = (*pq ? 0 : 1);
251 logf (LOG_LOG, "send_PDU failed, m_cs == 0");
256 *pq = new PDU_Queue(buf, len);
263 logf (LOG_LOG, "cannot send_PDU fd=%d", cs_fileno(m_cs));
268 COMSTACK Yaz_PDU_Assoc::comstack()
272 CS_TYPE cs_type = tcpip_type;
273 int protocol = PROTO_Z3950;
274 m_cs = cs_create (cs_type, 0, protocol);
279 void Yaz_PDU_Assoc::listen(IYaz_PDU_Observer *observer,
284 COMSTACK cs = comstack();
286 m_PDU_Observer = observer;
289 ap = cs_straddr (cs, addr);
292 if (cs_bind(cs, ap, CS_SERVER) < 0)
294 m_socketObservable->addObserver(cs_fileno(cs), this);
295 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
296 YAZ_SOCKET_OBSERVE_EXCEPT);
300 void Yaz_PDU_Assoc::connect(IYaz_PDU_Observer *observer,
303 logf (LOG_LOG, "Yaz_PDU_Assoc::connect");
305 m_PDU_Observer = observer;
306 COMSTACK cs = comstack();
307 void *ap = cs_straddr (cs, addr);
310 int res = cs_connect (cs, ap);
313 logf (LOG_DEBUG, "Yaz_PDU_Assoc::connect failed");
318 logf (LOG_LOG, "Yaz_PDU_Assoc::connect fd=%d", cs_fileno(cs));
319 m_socketObservable->addObserver(cs_fileno(cs), this);
320 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
321 YAZ_SOCKET_OBSERVE_EXCEPT|
322 YAZ_SOCKET_OBSERVE_WRITE);
324 m_state = Connecting;