network.c
上传用户:sddyfurun
上传日期:2007-01-04
资源大小:525k
文件大小:5k
源码类别:

代理服务器

开发平台:

Unix_Linux

  1. /*
  2.  * Copyright (c) 1988, 1993
  3.  * The Regents of the University of California.  All rights reserved.
  4.  *
  5.  * Redistribution and use in source and binary forms, with or without
  6.  * modification, are permitted provided that the following conditions
  7.  * are met:
  8.  * 1. Redistributions of source code must retain the above copyright
  9.  *    notice, this list of conditions and the following disclaimer.
  10.  * 2. Redistributions in binary form must reproduce the above copyright
  11.  *    notice, this list of conditions and the following disclaimer in the
  12.  *    documentation and/or other materials provided with the distribution.
  13.  * 3. All advertising materials mentioning features or use of this software
  14.  *    must display the following acknowledgement:
  15.  * This product includes software developed by the University of
  16.  * California, Berkeley and its contributors.
  17.  * 4. Neither the name of the University nor the names of its contributors
  18.  *    may be used to endorse or promote products derived from this software
  19.  *    without specific prior written permission.
  20.  *
  21.  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  22.  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23.  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24.  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  25.  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26.  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27.  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28.  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29.  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30.  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31.  * SUCH DAMAGE.
  32.  */
  33. #ifdef SHOW_SCCSIDS
  34. static char sccsid[] = "@(#)network.c 8.2 (Berkeley) 12/15/93";
  35. #endif
  36. #include <sys/types.h>
  37. #include <sys/socket.h>
  38. #include <sys/time.h>
  39. #ifdef HAVE_SYS_SELECT_H
  40. #include <sys/select.h>
  41. #endif
  42. #include <errno.h>
  43. #include <arpa/telnet.h>
  44. #include "ring.h"
  45. #include "defines.h"
  46. #include "externs.h"
  47. #include "fdset.h"
  48. Ring netoring, netiring;
  49. unsigned char netobuf[2*BUFSIZ], netibuf[BUFSIZ];
  50. /*
  51.  * Initialize internal network data structures.
  52.  */
  53.     void
  54. init_network()
  55. {
  56.     if (ring_init(&netoring, netobuf, sizeof netobuf) != 1) {
  57. exit(1);
  58.     }
  59.     if (ring_init(&netiring, netibuf, sizeof netibuf) != 1) {
  60. exit(1);
  61.     }
  62.     NetTrace = stdout;
  63. }
  64. /*
  65.  * Check to see if any out-of-band data exists on a socket (for
  66.  * Telnet "synch" processing).
  67.  */
  68.     int
  69. stilloob()
  70. {
  71.     static struct timeval timeout = { 0 };
  72.     fd_set excepts;
  73.     int value;
  74.     do {
  75. FD_ZERO(&excepts);
  76. FD_SET(net, &excepts);
  77. value = select(net+1, (fd_set *)0, (fd_set *)0, &excepts, &timeout);
  78.     } while ((value == -1) && (errno == EINTR));
  79.     if (value < 0) {
  80. perror("select");
  81. (void) quit();
  82. /* NOTREACHED */
  83.     }
  84.     if (FD_ISSET(net, &excepts)) {
  85. return 1;
  86.     } else {
  87. return 0;
  88.     }
  89. }
  90. /*
  91.  *  setneturg()
  92.  *
  93.  * Sets "neturg" to the current location.
  94.  */
  95.     void
  96. setneturg()
  97. {
  98.     ring_mark(&netoring);
  99. }
  100. /*
  101.  *  netflush
  102.  * Send as much data as possible to the network,
  103.  * handling requests for urgent data.
  104.  *
  105.  * The return value indicates whether we did any
  106.  * useful work.
  107.  */
  108.     int
  109. netflush()
  110. {
  111.     register int n, n1;
  112.     if ((n1 = n = ring_full_consecutive(&netoring)) > 0) {
  113. if (!ring_at_mark(&netoring)) {
  114.     n = send(net, (char *)netoring.consume, n, 0); /* normal write */
  115. } else {
  116.     /*
  117.      * In 4.2 (and 4.3) systems, there is some question about
  118.      * what byte in a sendOOB operation is the "OOB" data.
  119.      * To make ourselves compatible, we only send ONE byte
  120.      * out of band, the one WE THINK should be OOB (though
  121.      * we really have more the TCP philosophy of urgent data
  122.      * rather than the Unix philosophy of OOB data).
  123.      */
  124.     n = send(net, (char *)netoring.consume, 1, MSG_OOB);/* URGENT data */
  125. }
  126.     }
  127.     if (n < 0) {
  128. if (errno != ENOBUFS && errno != EWOULDBLOCK) {
  129.     setcommandmode();
  130.     perror(hostname);
  131.     (void)NetClose(net);
  132.     ring_clear_mark(&netoring);
  133.     longjmp(peerdied, -1);
  134.     /*NOTREACHED*/
  135. }
  136. n = 0;
  137.     }
  138.     if (netdata && n) {
  139. Dump('>', netoring.consume, n);
  140.     }
  141.     if (n) {
  142. ring_consumed(&netoring, n);
  143. /*
  144.  * If we sent all, and more to send, then recurse to pick
  145.  * up the other half.
  146.  */
  147. if ((n1 == n) && ring_full_consecutive(&netoring)) {
  148.     (void) netflush();
  149. }
  150. return 1;
  151.     } else {
  152. return 0;
  153.     }
  154. }