1 |
// sockstream.C -*- C++ -*- socket library |
2 |
// Copyright (C) 1992,1993,1994 Gnanasekaran Swaminathan <gs4t@virginia.edu> |
3 |
// |
4 |
// Permission is granted to use at your own risk and distribute this software |
5 |
// in source and binary forms provided the above copyright |
6 |
// notice and this paragraph are preserved on all copies. |
7 |
// This software is provided "as is" with no express or implied warranty. |
8 |
// |
9 |
// Version: 17Oct95 1.10 |
10 |
// |
11 |
// You can simultaneously read and write into |
12 |
// a sockbuf just like you can listen and talk |
13 |
// through a telephone. Hence, the read and the |
14 |
// write buffers are different. That is, they do not |
15 |
// share the same memory. |
16 |
// |
17 |
// Read: |
18 |
// gptr() points to the start of the get area. |
19 |
// The unread chars are gptr() - egptr(). |
20 |
// base() points to the read buffer |
21 |
// |
22 |
// eback() is set to base() so that pbackfail() |
23 |
// is called only when there is no place to |
24 |
// putback a char. And pbackfail() always returns EOF. |
25 |
// |
26 |
// Write: |
27 |
// pptr() points to the start of the put area |
28 |
// The unflushed chars are pbase() - pptr() |
29 |
// pbase() points to the write buffer. |
30 |
// epptr() points to the end of the write buffer. |
31 |
// |
32 |
// Output is flushed whenever one of the following conditions |
33 |
// holds: |
34 |
// (1) pptr() == epptr() |
35 |
// (2) EOF is written |
36 |
// (3) linebuffered and '\n' is written |
37 |
// |
38 |
// Unbuffered: |
39 |
// Input buffer size is assumed to be of size 1 and output |
40 |
// buffer is of size 0. That is, egptr() <= base()+1 and |
41 |
// epptr() == pbase(). |
42 |
|
43 |
|
44 |
#include <config.h> |
45 |
#include <sockstream.h> |
46 |
|
47 |
EXTERN_C_BEGIN |
48 |
#include <sys/time.h> |
49 |
#include <sys/socket.h> |
50 |
#include <unistd.h> |
51 |
#include <errno.h> |
52 |
EXTERN_C_END |
53 |
|
54 |
#ifndef BUFSIZ |
55 |
# define BUFSIZ 1024 |
56 |
#endif |
57 |
|
58 |
#ifdef FD_ZERO |
59 |
# undef FD_ZERO // bzero causes so much trouble to us |
60 |
#endif |
61 |
#define FD_ZERO(p) (memset ((p), 0, sizeof *(p))) |
62 |
|
63 |
void sock_error (const char* classname, const char* msg) |
64 |
{ |
65 |
if (errno) |
66 |
perror (msg); |
67 |
cerr << classname << ' ' << msg << endl; |
68 |
errno = 0; |
69 |
} |
70 |
|
71 |
void sockAddr::error (const char* msg) const |
72 |
{ |
73 |
sock_error ("class sockAddr ", msg); |
74 |
} |
75 |
|
76 |
sockbuf::sockbuf (int soc) |
77 |
: rep (new sockcnt (soc, 1)), |
78 |
stmo (-1), rtmo (-1) |
79 |
{ |
80 |
#ifdef _S_NOLIBGXX |
81 |
xflags (0); |
82 |
#endif |
83 |
xsetflags (_S_LINE_BUF); |
84 |
} |
85 |
|
86 |
sockbuf::sockbuf (int domain, sockbuf::type st, int proto) |
87 |
: rep (0), stmo (-1), rtmo (-1) |
88 |
{ |
89 |
int soc = ::socket (domain, st, proto); |
90 |
rep = new sockcnt (soc, 1); |
91 |
#ifdef _S_NOLIBGXX |
92 |
xflags (0); |
93 |
#endif |
94 |
if (rep->sock == -1) perror ("sockbuf::sockbuf"); |
95 |
xsetflags (_S_LINE_BUF); |
96 |
} |
97 |
|
98 |
sockbuf::sockbuf (const sockbuf& sb) |
99 |
: rep (sb.rep), stmo (sb.stmo), rtmo (sb.rtmo) |
100 |
{ |
101 |
#ifdef _S_NOLIBGXX |
102 |
xflags (0); |
103 |
#endif |
104 |
rep->cnt++; |
105 |
xsetflags (_S_LINE_BUF); |
106 |
} |
107 |
|
108 |
sockbuf& sockbuf::operator = (const sockbuf& sb) |
109 |
{ |
110 |
if (this != &sb && rep != sb.rep && rep->sock != sb.rep->sock) { |
111 |
this->sockbuf::~sockbuf(); |
112 |
rep = sb.rep; stmo = sb.stmo; rtmo = sb.rtmo; |
113 |
rep->cnt++; |
114 |
#ifdef _S_NOLIBGXX |
115 |
xflags (sb.xflags ()); |
116 |
#else |
117 |
// xflags () is a non-const member function in libg++. |
118 |
xflags (((sockbuf&)sb).xflags ()); |
119 |
#endif |
120 |
} |
121 |
return *this; |
122 |
} |
123 |
|
124 |
sockbuf::~sockbuf () |
125 |
{ |
126 |
overflow (EOF); |
127 |
if (rep->cnt == 1 && !(xflags () & _S_DELETE_DONT_CLOSE)) close (); |
128 |
delete [] base (); |
129 |
if (--rep->cnt == 0) delete rep; |
130 |
} |
131 |
|
132 |
sockbuf* sockbuf::open (type, int) |
133 |
{ |
134 |
return 0; |
135 |
} |
136 |
|
137 |
sockbuf* sockbuf::close() |
138 |
{ |
139 |
if (rep->sock >= 0) { |
140 |
if (::close (rep->sock) == -1) return this; |
141 |
rep->sock = -1; |
142 |
} |
143 |
return 0; |
144 |
} |
145 |
|
146 |
_G_ssize_t sockbuf::sys_read (char* buf, _G_ssize_t len) |
147 |
// return EOF on eof, 0 on timeout, and # of chars read on success |
148 |
{ |
149 |
return read (buf, len); |
150 |
} |
151 |
|
152 |
_G_ssize_t sockbuf::sys_write (const void* buf, long len) |
153 |
// return written_length; < len indicates error |
154 |
{ |
155 |
return write (buf, (int) len); |
156 |
} |
157 |
|
158 |
int sockbuf::flush_output() |
159 |
// return 0 when there is nothing to flush or when the flush is a success |
160 |
// return EOF when it could not flush |
161 |
{ |
162 |
if (pptr () <= pbase ()) return 0; |
163 |
if (!(xflags () & _S_NO_WRITES)) { |
164 |
int wlen = pptr () - pbase (); |
165 |
int wval = sys_write (pbase (), wlen); |
166 |
int status = (wval == wlen)? 0: EOF; |
167 |
if (unbuffered()) setp (pbase (), pbase ()); |
168 |
else setp (pbase (), pbase () + BUFSIZ); |
169 |
return status; |
170 |
} |
171 |
return EOF; |
172 |
} |
173 |
|
174 |
int sockbuf::sync () |
175 |
{ |
176 |
return flush_output (); |
177 |
} |
178 |
|
179 |
int sockbuf::doallocate () |
180 |
// return 1 on allocation and 0 if there is no need |
181 |
{ |
182 |
if (!base ()) { |
183 |
char* buf = new char[2*BUFSIZ]; |
184 |
setb (buf, buf+BUFSIZ, 0); |
185 |
setg (buf, buf, buf); |
186 |
|
187 |
buf += BUFSIZ; |
188 |
setp (buf, buf+BUFSIZ); |
189 |
return 1; |
190 |
} |
191 |
return 0; |
192 |
} |
193 |
|
194 |
int sockbuf::underflow () |
195 |
{ |
196 |
if (xflags () & _S_NO_READS) return EOF; |
197 |
|
198 |
if (gptr () < egptr ()) return *(unsigned char*)gptr (); |
199 |
|
200 |
if (base () == 0 && doallocate () == 0) return EOF; |
201 |
|
202 |
int bufsz = unbuffered () ? 1: BUFSIZ; |
203 |
int rval = sys_read (base (), bufsz); |
204 |
if (rval == EOF) { |
205 |
xsetflags (_S_EOF_SEEN); |
206 |
return EOF; |
207 |
}else if (rval == 0) |
208 |
return EOF; |
209 |
setg (eback (), base (), base () + rval); |
210 |
return *(unsigned char*)gptr (); |
211 |
} |
212 |
|
213 |
int sockbuf::overflow (int c) |
214 |
// if c == EOF, return flush_output(); |
215 |
// if c == '\n' and linebuffered, insert c and |
216 |
// return (flush_output()==EOF)? EOF: c; |
217 |
// otherwise insert c into the buffer and return c |
218 |
{ |
219 |
if (c == EOF) return flush_output (); |
220 |
|
221 |
if (xflags () & _S_NO_WRITES) return EOF; |
222 |
|
223 |
if (pbase () == 0 && doallocate () == 0) return EOF; |
224 |
|
225 |
if (pptr () >= epptr() && flush_output () == EOF) |
226 |
return EOF; |
227 |
|
228 |
xput_char (c); |
229 |
if ((unbuffered () || linebuffered () && c == '\n' || pptr () >= epptr ()) |
230 |
&& flush_output () == EOF) |
231 |
return EOF; |
232 |
|
233 |
return c; |
234 |
} |
235 |
|
236 |
int sockbuf::xsputn (const char* s, int n) |
237 |
{ |
238 |
if (n <= 0) return 0; |
239 |
const unsigned char* p = (const unsigned char*)s; |
240 |
|
241 |
for (int i=0; i<n; i++, p++) { |
242 |
if (*p == '\n') { |
243 |
if (overflow (*p) == EOF) return i; |
244 |
} else |
245 |
if (sputc (*p) == EOF) return i; |
246 |
} |
247 |
return n; |
248 |
} |
249 |
|
250 |
int sockbuf::bind (sockAddr& sa) |
251 |
{ |
252 |
if (::bind (rep->sock, sa.addr (), sa.size ()) == -1) { |
253 |
// error ("sockbuf::bind"); |
254 |
return errno; |
255 |
} |
256 |
return 0; |
257 |
} |
258 |
|
259 |
int sockbuf::connect (sockAddr& sa) |
260 |
{ |
261 |
if (::connect(rep->sock, sa.addr (), sa.size()) == -1) { |
262 |
// error ("sockbuf::connect"); |
263 |
return errno; |
264 |
} |
265 |
return 0; |
266 |
} |
267 |
|
268 |
void sockbuf::listen (int num) |
269 |
{ |
270 |
if (::listen (rep->sock, num) == -1) |
271 |
error ("sockbuf::listen"); |
272 |
} |
273 |
|
274 |
sockbuf sockbuf::accept (sockAddr& sa) |
275 |
{ |
276 |
int len = sa.size (); |
277 |
int soc = -1; |
278 |
while ((soc = ::accept (rep->sock, sa.addr (), &len)) == -1 |
279 |
&& errno == EINTR) |
280 |
errno = 0; |
281 |
if (soc == -1) |
282 |
error ("sockbuf::accept"); |
283 |
return soc; |
284 |
} |
285 |
|
286 |
sockbuf sockbuf::accept () |
287 |
{ |
288 |
int soc = -1; |
289 |
while ((soc = ::accept (rep->sock, 0, 0)) == -1 |
290 |
&& errno == EINTR) |
291 |
errno = 0; |
292 |
if (soc == -1) |
293 |
error ("sockbuf::accept"); |
294 |
return soc; |
295 |
} |
296 |
|
297 |
int sockbuf::read (void* buf, int len) |
298 |
{ |
299 |
if (rtmo != -1 && is_readready (rtmo)==0) return 0; |
300 |
|
301 |
int rval; |
302 |
if ((rval = ::read (rep->sock, (char*) buf, len)) == -1) |
303 |
error("sockbuf::read"); |
304 |
return (rval==0) ? EOF: rval; |
305 |
} |
306 |
|
307 |
int sockbuf::recv (void* buf, int len, int msgf) |
308 |
{ |
309 |
if (rtmo != -1 && is_readready (rtmo)==0) return 0; |
310 |
|
311 |
int rval; |
312 |
if ((rval = ::recv (rep->sock, (char*) buf, len, msgf)) == -1) |
313 |
error ("sockbuf::recv"); |
314 |
return (rval==0) ? EOF: rval; |
315 |
} |
316 |
|
317 |
int sockbuf::recvfrom (sockAddr& sa, void* buf, int len, int msgf) |
318 |
{ |
319 |
if (rtmo != -1 && is_readready (rtmo)==0) return 0; |
320 |
|
321 |
int rval; |
322 |
int sa_len = sa.size (); |
323 |
|
324 |
if ((rval = ::recvfrom (rep->sock, (char*) buf, len, |
325 |
msgf, sa.addr (), &sa_len)) == -1) |
326 |
error ("sockbuf::recvfrom"); |
327 |
return (rval==0) ? EOF: rval; |
328 |
} |
329 |
|
330 |
int sockbuf::write(const void* buf, int len) |
331 |
{ |
332 |
if (stmo != -1 && is_writeready (stmo)==0) return 0; |
333 |
|
334 |
int wlen=0; |
335 |
while(len>0) { |
336 |
int wval; |
337 |
if ((wval = ::write (rep->sock, (char*) buf, len)) == -1) { |
338 |
error ("sockbuf::write"); |
339 |
return wval; |
340 |
} |
341 |
len -= wval; |
342 |
wlen += wval; |
343 |
} |
344 |
return wlen; // == len if every thing is all right |
345 |
} |
346 |
|
347 |
int sockbuf::send (const void* buf, int len, int msgf) |
348 |
{ |
349 |
if (stmo != -1 && is_writeready (stmo)==0) return 0; |
350 |
|
351 |
int wlen=0; |
352 |
while(len>0) { |
353 |
int wval; |
354 |
if ((wval = ::send (rep->sock, (char*) buf, len, msgf)) == -1) { |
355 |
error ("sockbuf::send"); |
356 |
return wval; |
357 |
} |
358 |
len -= wval; |
359 |
wlen += wval; |
360 |
} |
361 |
return wlen; |
362 |
} |
363 |
|
364 |
int sockbuf::sendto (sockAddr& sa, const void* buf, int len, int msgf) |
365 |
{ |
366 |
if (stmo != -1 && is_writeready (stmo)==0) return 0; |
367 |
|
368 |
int wlen=0; |
369 |
while(len>0) { |
370 |
int wval; |
371 |
if ((wval = ::sendto (rep->sock, (char*) buf, len, |
372 |
msgf, sa.addr (), sa.size())) == -1) { |
373 |
error ("sockbuf::sendto"); |
374 |
return wval; |
375 |
} |
376 |
len -= wval; |
377 |
wlen += wval; |
378 |
} |
379 |
return wlen; |
380 |
} |
381 |
|
382 |
#ifndef __linux__ |
383 |
// linux does not have sendmsg or recvmsg |
384 |
|
385 |
int sockbuf::recvmsg (msghdr* msg, int msgf) |
386 |
{ |
387 |
if (rtmo != -1 && is_readready (rtmo)==0) return 0; |
388 |
|
389 |
int rval; |
390 |
if ((rval = ::recvmsg(rep->sock, msg, msgf)) == -1) |
391 |
error ("sockbuf::recvmsg"); |
392 |
return (rval==0)? EOF: rval; |
393 |
} |
394 |
|
395 |
int sockbuf::sendmsg (msghdr* msg, int msgf) |
396 |
{ |
397 |
if (stmo != -1 && is_writeready (stmo)==0) return 0; |
398 |
|
399 |
int wval; |
400 |
if ((wval = ::sendmsg (rep->sock, msg, msgf)) == -1) |
401 |
error("sockbuf::sendmsg"); |
402 |
return wval; |
403 |
} |
404 |
#endif //!__linux__ |
405 |
|
406 |
int sockbuf::sendtimeout (int wp) |
407 |
{ |
408 |
int oldstmo = stmo; |
409 |
stmo = (wp < 0) ? -1: wp; |
410 |
return oldstmo; |
411 |
} |
412 |
|
413 |
int sockbuf::recvtimeout (int wp) |
414 |
{ |
415 |
int oldrtmo = rtmo; |
416 |
rtmo = (wp < 0) ? -1: wp; |
417 |
return oldrtmo; |
418 |
} |
419 |
|
420 |
int sockbuf::is_readready (int wp_sec, int wp_usec) const |
421 |
{ |
422 |
fd_set fds; |
423 |
FD_ZERO (&fds); |
424 |
FD_SET (rep->sock, &fds); |
425 |
|
426 |
timeval tv; |
427 |
tv.tv_sec = wp_sec; |
428 |
tv.tv_usec = wp_usec; |
429 |
|
430 |
int ret = select (rep->sock+1, &fds, 0, 0, (wp_sec == -1) ? 0: &tv); |
431 |
if (ret == -1) { |
432 |
error ("sockbuf::readready"); |
433 |
return 0; |
434 |
} |
435 |
return ret; |
436 |
} |
437 |
|
438 |
int sockbuf::is_writeready (int wp_sec, int wp_usec) const |
439 |
{ |
440 |
fd_set fds; |
441 |
FD_ZERO (&fds); |
442 |
FD_SET (rep->sock, &fds); |
443 |
|
444 |
timeval tv; |
445 |
tv.tv_sec = wp_sec; |
446 |
tv.tv_usec = wp_usec; |
447 |
|
448 |
int ret = select (rep->sock+1, 0, &fds, 0, (wp_sec == -1) ? 0: &tv); |
449 |
if (ret == -1) { |
450 |
error ("Select::operator ()"); |
451 |
return 0; |
452 |
} |
453 |
return ret; |
454 |
} |
455 |
|
456 |
int sockbuf::is_exceptionpending (int wp_sec, int wp_usec) const |
457 |
{ |
458 |
fd_set fds; |
459 |
FD_ZERO (&fds); |
460 |
FD_SET (rep->sock, &fds); |
461 |
|
462 |
timeval tv; |
463 |
tv.tv_sec = wp_sec; |
464 |
tv.tv_usec = wp_usec; |
465 |
|
466 |
int ret = select (rep->sock+1, 0, 0, &fds, (wp_sec == -1) ? 0: &tv); |
467 |
if (ret == -1) { |
468 |
error ("Select::operator ()"); |
469 |
return 0; |
470 |
} |
471 |
return ret; |
472 |
} |
473 |
|
474 |
void sockbuf::shutdown (shuthow sh) |
475 |
{ |
476 |
switch (sh) { |
477 |
case shut_read: |
478 |
xsetflags(_S_NO_READS); |
479 |
break; |
480 |
case shut_write: |
481 |
xsetflags(_S_NO_WRITES); |
482 |
break; |
483 |
case shut_readwrite: |
484 |
xsetflags(_S_NO_READS|_S_NO_WRITES); |
485 |
break; |
486 |
} |
487 |
if (::shutdown(rep->sock, sh) == -1) |
488 |
error("sockbuf::shutdown"); |
489 |
} |
490 |
|
491 |
int sockbuf::getopt (option op, void* buf, int len, level l) const |
492 |
{ |
493 |
int rlen = len; |
494 |
if (::getsockopt (rep->sock, l, op, (char*) buf, &rlen) == -1) |
495 |
perror ("sockbuf::getopt"); |
496 |
return rlen; |
497 |
} |
498 |
|
499 |
void sockbuf::setopt (option op, void* buf, int len, level l) const |
500 |
{ |
501 |
if (::setsockopt (rep->sock, l, op, (char*) buf, len) == -1) |
502 |
perror ("sockbuf::setopt"); |
503 |
} |
504 |
|
505 |
sockbuf::type sockbuf::gettype () const |
506 |
{ |
507 |
int ty=0; |
508 |
getopt (so_type, &ty, sizeof (ty)); |
509 |
return sockbuf::type(ty); |
510 |
} |
511 |
|
512 |
int sockbuf::clearerror () const |
513 |
{ |
514 |
int err=0; |
515 |
getopt (so_error, &err, sizeof (err)); |
516 |
return err; |
517 |
} |
518 |
|
519 |
int sockbuf::debug (int opt) const |
520 |
{ |
521 |
int old=0; |
522 |
getopt (so_debug, &old, sizeof (old)); |
523 |
if (opt != -1) |
524 |
setopt (so_debug, &opt, sizeof (opt)); |
525 |
return old; |
526 |
} |
527 |
|
528 |
int sockbuf::reuseaddr (int opt) const |
529 |
{ |
530 |
int old=0; |
531 |
getopt (so_reuseaddr, &old, sizeof (old)); |
532 |
if (opt != -1) |
533 |
setopt (so_reuseaddr, &opt, sizeof (opt)); |
534 |
return old; |
535 |
} |
536 |
|
537 |
int sockbuf::keepalive (int opt) const |
538 |
{ |
539 |
int old=0; |
540 |
getopt (so_keepalive, &old, sizeof (old)); |
541 |
if (opt != -1) |
542 |
setopt (so_keepalive, &opt, sizeof (opt)); |
543 |
return old; |
544 |
} |
545 |
|
546 |
int sockbuf::dontroute (int opt) const |
547 |
{ |
548 |
int old=0; |
549 |
getopt (so_dontroute, &old, sizeof (old)); |
550 |
if (opt != -1) |
551 |
setopt (so_dontroute, &opt, sizeof (opt)); |
552 |
return old; |
553 |
} |
554 |
|
555 |
int sockbuf::broadcast (int opt) const |
556 |
{ |
557 |
int old=0; |
558 |
getopt (so_broadcast, &old, sizeof (old)); |
559 |
if (opt != -1) |
560 |
setopt (so_broadcast, &opt, sizeof (opt)); |
561 |
return old; |
562 |
} |
563 |
|
564 |
int sockbuf::oobinline (int opt) const |
565 |
{ |
566 |
int old=0; |
567 |
getopt (so_oobinline, &old, sizeof (old)); |
568 |
if (opt != -1) |
569 |
setopt (so_oobinline, &opt, sizeof (opt)); |
570 |
return old; |
571 |
} |
572 |
|
573 |
int sockbuf::linger (int opt) const |
574 |
{ |
575 |
socklinger old (0, 0); |
576 |
getopt (so_linger, &old, sizeof (old)); |
577 |
if (opt > 0) { |
578 |
socklinger nw (1, opt); |
579 |
setopt (so_linger, &nw, sizeof (nw)); |
580 |
}else if (opt == 0) { |
581 |
socklinger nw (0, old.l_linger); |
582 |
setopt (so_linger, &nw, sizeof (nw)); |
583 |
} |
584 |
return old.l_onoff ? old.l_linger: -1; |
585 |
} |
586 |
|
587 |
int sockbuf::sendbufsz (int sz) const |
588 |
{ |
589 |
int old=0; |
590 |
getopt (so_sndbuf, &old, sizeof (old)); |
591 |
if (sz >= 0) |
592 |
setopt (so_sndbuf, &sz, sizeof (sz)); |
593 |
return old; |
594 |
} |
595 |
|
596 |
int sockbuf::recvbufsz (int sz) const |
597 |
{ |
598 |
int old=0; |
599 |
getopt (so_rcvbuf, &old, sizeof (old)); |
600 |
if (sz >= 0) |
601 |
setopt (so_rcvbuf, &sz, sizeof (sz)); |
602 |
return old; |
603 |
} |
604 |
|
605 |
void sockbuf::error (const char* msg) const |
606 |
{ |
607 |
sock_error ("class sockbuf: ", msg); |
608 |
} |
609 |
|
610 |
isockstream::~isockstream () |
611 |
{ |
612 |
delete ios::rdbuf (); |
613 |
init (0); |
614 |
} |
615 |
|
616 |
void isockstream::error (const char* msg) const |
617 |
{ |
618 |
sock_error ("class isockstream: ", msg); |
619 |
} |
620 |
|
621 |
osockstream::~osockstream () |
622 |
{ |
623 |
delete ios::rdbuf (); |
624 |
init (0); |
625 |
} |
626 |
|
627 |
void osockstream::error (const char* msg) const |
628 |
{ |
629 |
sock_error ("class osockstream: ", msg); |
630 |
} |
631 |
|
632 |
iosockstream::~iosockstream () |
633 |
{ |
634 |
delete ios::rdbuf (); |
635 |
init (0); |
636 |
} |
637 |
|
638 |
void iosockstream::error (const char* msg) const |
639 |
{ |
640 |
sock_error ("class iosockstream: ", msg); |
641 |
} |
642 |
|