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.3 1999-02-02 14:01:20 adam
8 * First WIN32 port of YAZ++.
10 * Revision 1.2 1999/01/28 13:08:44 adam
11 * Yaz_PDU_Assoc better encapsulated. Memory leak fix in
12 * yaz-socket-manager.cc.
14 * Revision 1.1.1.1 1999/01/28 09:41:07 adam
15 * First implementation of YAZ++.
21 #include <yaz-pdu-assoc.h>
26 Yaz_PDU_Assoc::Yaz_PDU_Assoc(IYazSocketObservable *socketObservable,
31 m_socketObservable = socketObservable;
42 IYaz_PDU_Observable *Yaz_PDU_Assoc::clone()
44 Yaz_PDU_Assoc *copy = new Yaz_PDU_Assoc(m_socketObservable, 0);
48 Yaz_PDU_Assoc::~Yaz_PDU_Assoc()
53 void Yaz_PDU_Assoc::socketNotify(int event)
55 logf (LOG_LOG, "socketNotify p=%p event = %d", this, event);
56 if (m_state == Connected)
59 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
60 YAZ_SOCKET_OBSERVE_EXCEPT);
61 m_PDU_Observer->connectNotify();
64 else if (m_state == Connecting)
66 if (event & YAZ_SOCKET_OBSERVE_READ)
69 m_PDU_Observer->failNotify();
74 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
75 YAZ_SOCKET_OBSERVE_EXCEPT);
76 m_PDU_Observer->connectNotify();
80 else if (m_state == Listen)
82 if (event & YAZ_SOCKET_OBSERVE_READ)
87 if ((res = cs_listen(m_cs, 0, 0)) == 1)
91 logf(LOG_FATAL, "cs_listen failed");
94 if (!(new_line = cs_accept(m_cs)))
97 Yaz_PDU_Assoc *assoc = new Yaz_PDU_Assoc (m_socketObservable,
99 assoc->m_parent = this;
100 assoc->m_next = m_children;
103 assoc->m_PDU_Observer = m_PDU_Observer->clone(assoc);
104 assoc->m_state = Ready;
105 assoc->m_socketObservable->addObserver(cs_fileno(new_line), assoc);
106 assoc->m_socketObservable->maskObserver(assoc,
107 YAZ_SOCKET_OBSERVE_READ|
108 YAZ_SOCKET_OBSERVE_EXCEPT);
111 else if (m_state == Ready)
113 if (event & YAZ_SOCKET_OBSERVE_WRITE)
117 if (event & YAZ_SOCKET_OBSERVE_READ)
121 int res = cs_get (m_cs, &m_input_buf, &m_input_len);
126 logf (LOG_LOG, "Connection closed by client");
128 m_PDU_Observer->failNotify();
131 // lock it, so we know if recv_PDU deletes it.
133 m_destroyed = &destroyed;
135 m_PDU_Observer->recv_PDU(m_input_buf, res);
136 if (destroyed) // it really was destroyed, return now.
138 } while (m_cs && cs_more (m_cs));
143 void Yaz_PDU_Assoc::close()
145 m_socketObservable->deleteObserver(this);
149 logf (LOG_LOG, "Yaz_PDU_Assoc::close fd=%d", cs_fileno(m_cs));
155 PDU_Queue *q_this = m_queue_out;
156 m_queue_out = m_queue_out->m_next;
159 // free (m_input_buf);
164 void Yaz_PDU_Assoc::destroy()
171 // delete from parent's child list (if any)
174 c = &m_parent->m_children;
182 // delete all children ...
186 Yaz_PDU_Assoc *here = *c;
193 Yaz_PDU_Assoc::PDU_Queue::PDU_Queue(const char *buf, int len)
195 m_buf = (char *) malloc (len);
196 memcpy (m_buf, buf, len);
201 Yaz_PDU_Assoc::PDU_Queue::~PDU_Queue()
206 int Yaz_PDU_Assoc::flush_PDU()
210 if (m_state != Ready)
212 PDU_Queue *q = m_queue_out;
215 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
216 YAZ_SOCKET_OBSERVE_EXCEPT);
219 r = cs_put (m_cs, q->m_buf, q->m_len);
223 m_PDU_Observer->failNotify();
228 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
229 YAZ_SOCKET_OBSERVE_EXCEPT|
230 YAZ_SOCKET_OBSERVE_WRITE);
231 logf (LOG_LOG, "put %d bytes (incomplete write)", q->m_len);
234 logf (LOG_LOG, "put %d bytes fd=%d", q->m_len, cs_fileno(m_cs));
235 // whole packet sent... delete this and proceed to next ...
236 m_queue_out = q->m_next;
237 logf (LOG_LOG, "m_queue_out = %p", m_queue_out);
239 // don't select on write if queue is empty ...
241 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
242 YAZ_SOCKET_OBSERVE_EXCEPT);
246 int Yaz_PDU_Assoc::send_PDU(const char *buf, int len)
248 logf (LOG_LOG, "send_PDU");
249 PDU_Queue **pq = &m_queue_out;
250 int is_idle = (*pq ? 0 : 1);
254 logf (LOG_LOG, "send_PDU failed, m_cs == 0");
259 *pq = new PDU_Queue(buf, len);
266 logf (LOG_LOG, "cannot send_PDU fd=%d", cs_fileno(m_cs));
271 COMSTACK Yaz_PDU_Assoc::comstack()
275 CS_TYPE cs_type = tcpip_type;
276 int protocol = PROTO_Z3950;
277 m_cs = cs_create (cs_type, 0, protocol);
282 void Yaz_PDU_Assoc::listen(IYaz_PDU_Observer *observer,
287 COMSTACK cs = comstack();
289 logf (LOG_LOG, "Yaz_PDU_Assoc::listen %s", addr);
290 m_PDU_Observer = observer;
293 ap = cs_straddr (cs, addr);
296 if (cs_bind(cs, ap, CS_SERVER) < 0)
298 m_socketObservable->addObserver(cs_fileno(cs), this);
299 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
300 YAZ_SOCKET_OBSERVE_EXCEPT);
304 void Yaz_PDU_Assoc::connect(IYaz_PDU_Observer *observer,
307 logf (LOG_LOG, "Yaz_PDU_Assoc::connect %s", addr);
309 m_PDU_Observer = observer;
310 COMSTACK cs = comstack();
311 void *ap = cs_straddr (cs, addr);
314 logf (LOG_LOG, "cs_straddr failed");
317 int res = cs_connect (cs, ap);
320 logf (LOG_DEBUG, "Yaz_PDU_Assoc::connect failed");
325 logf (LOG_LOG, "Yaz_PDU_Assoc::connect fd=%d", cs_fileno(cs));
326 m_socketObservable->addObserver(cs_fileno(cs), this);
327 m_socketObservable->maskObserver(this, YAZ_SOCKET_OBSERVE_READ|
328 YAZ_SOCKET_OBSERVE_EXCEPT|
329 YAZ_SOCKET_OBSERVE_WRITE);
331 m_state = Connecting;