pcap-enet.c
上传用户:tjescc
上传日期:2021-02-23
资源大小:419k
文件大小:5k
源码类别:

Telnet服务器

开发平台:

Unix_Linux

  1. /*
  2.  * Stanford Enetfilter subroutines for tcpdump
  3.  *
  4.  * Based on the MERIT NNstat etherifrt.c and the Ultrix pcap-pf.c
  5.  * subroutines.
  6.  *
  7.  * Rayan Zachariassen, CA*Net
  8.  */
  9. #include <sys/types.h>
  10. #include <sys/time.h>
  11. #include <sys/file.h>
  12. #include <sys/ioctl.h>
  13. #include <sys/socket.h>
  14. #include <net/if.h>
  15. #include <net/bpf.h>
  16. #include <net/enet.h>
  17. #include <netinet/in.h>
  18. #include <netinet/if_ether.h>
  19. #include <stdio.h>
  20. #include <errno.h>
  21. #include "interface.h"
  22. struct packet_header {
  23. #ifdef IBMRTPC
  24. struct LengthWords length;
  25. struct tap_header tap;
  26. #endif /* IBMRTPC */
  27. u_char packet[8]
  28. };
  29. extern int errno;
  30. #define BUFSPACE (4*1024)
  31. /* Forwards */
  32. static void efReadError(int, char *);
  33. void
  34. readloop(int cnt, int if_fd, struct bpf_program *fp, printfunc printit)
  35. {
  36. #ifdef IBMRTPC
  37. register struct packet_header *ph;
  38. register u_char *bp;
  39. register int inc;
  40. #else /* !IBMRTPC */
  41. static struct timeval tv = { 0 };
  42. #endif /* IBMRTPC */
  43. register int cc, caplen;
  44. register struct bpf_insn *fcode = fp->bf_insns;
  45. union {
  46. struct packet_header hdr;
  47. u_char p[BUFSPACE];
  48. u_short s;
  49. } buf;
  50. while (1) {
  51. if ((cc = read(if_fd, (char *)buf.p, sizeof(buf))) < 0)
  52. efReadError(if_fd, "reader");
  53. #ifdef IBMRTPC
  54. /*
  55.  * Loop through each packet.
  56.  */
  57. bp = buf.p;
  58. while (cc > 0) {
  59. ph = (struct packet_header *)bp;
  60. caplen = ph->tap.th_wirelen > snaplen ? snaplen : ph->tap
  61. .th_wirelen ;
  62. if (bpf_filter(fcode, (char *)ph->packet,
  63. ph->tap.th_wirelen, caplen)) {
  64. if (cnt >= 0 && --cnt < 0)
  65. goto out;
  66. (*printit)((char *)ph->packet,
  67. (struct timeval *)ph->tap.th_timestamp,
  68. ph->tap.th_wirelen, caplen);
  69. }
  70. inc = ph->length.PacketOffset;
  71. cc -= inc;
  72. bp += inc;
  73. }
  74. #else /* !IBMRTPC */
  75. caplen = cc > snaplen ? snaplen : cc ;
  76. if (bpf_filter(fcode, buf.hdr.packet, cc, caplen)) {
  77. if (cnt >= 0 && --cnt < 0)
  78. goto out;
  79. (*printit)(buf.hdr.packet, &tv, cc, caplen);
  80. }
  81. #endif /* IBMRTPC */
  82. }
  83.  out:
  84. wrapup(if_fd);
  85. }
  86. /* Call ONLY if read() has returned an error on packet filter */
  87. static void
  88. efReadError(int fid, char *msg)
  89. {
  90. if (errno == EINVAL) { /* read MAXINT bytes already! */
  91. if (lseek(fid, 0, 0) < 0) {
  92. perror("tcpdump: efReadError/lseek");
  93. exit(-1);
  94. }
  95. else
  96. return;
  97. }
  98. else {
  99. (void) fprintf(stderr, "tcpdump: ");
  100. perror(msg);
  101. exit(-1);
  102. }
  103. }
  104. void
  105. wrapup(int fd)
  106. {
  107. #ifdef IBMRTPC
  108. struct enstats es;
  109. if (ioctl(fd, EIOSTATS, &es) == -1) {
  110. perror("tcpdump: enet ioctl EIOSTATS error");
  111. exit(-1);
  112. }
  113. fprintf(stderr, "%d packets queued", es.enStat_Rcnt);
  114. if (es.enStat_Rdrops > 0)
  115. fprintf(stderr, ", %d dropped", es.enStat_Rdrops);
  116. if (es.enStat_Reads > 0)
  117. fprintf(stderr, ", %d tcpdump %s", es.enStat_Reads,
  118. es.enStat_Reads > 1 ? "reads" : "read");
  119. if (es.enStat_MaxRead > 1)
  120. fprintf(stderr, ", %d packets in largest read", 
  121. es.enStat_MaxRead);
  122. putc('n', stderr);
  123. #endif /* IBMRTPC */
  124. close(fd);
  125. }
  126. int
  127. initdevice(char *device, int pflag, int *linktype)
  128. {
  129. struct eniocb ctl;
  130. struct enfilter filter;
  131. u_int maxwaiting;
  132. int if_fd;
  133. #ifdef IBMRTPC
  134. GETENETDEVICE(0, O_RDONLY, &if_fd);
  135. #else /* !IBMRTPC */
  136. if_fd = open("/dev/enet", O_RDONLY, 0);
  137. #endif /* IBMRTPC */
  138. if (if_fd == -1) {
  139. perror("tcpdump: enet open error");
  140. error(
  141. "your system may not be properly configured; see "man enet(4)"");
  142. exit(-1);
  143. }
  144. /*  Get operating parameters. */
  145. if (ioctl(if_fd, EIOCGETP, (char *)&ctl) == -1) {
  146. perror("tcpdump: enet ioctl EIOCGETP error");
  147. exit(-1);
  148. }
  149. /*  Set operating parameters. */
  150. #ifdef IBMRTPC
  151. ctl.en_rtout = 1 * ctl.en_hz;
  152. ctl.en_tr_etherhead = 1;
  153. ctl.en_tap_network = 1;
  154. ctl.en_multi_packet = 1;
  155. ctl.en_maxlen = BUFSPACE;
  156. #else /* !IBMRTPC */
  157. ctl.en_rtout = 64; /* randomly picked value for HZ */
  158. #endif /* IBMRTPC */
  159. if (ioctl(if_fd, EIOCSETP, &ctl) == -1) {
  160. perror("tcpdump: enet ioctl EIOCSETP error");
  161. exit(-1);
  162. }
  163. /*  Flush the receive queue, since we've changed
  164.     the operating parameters and we otherwise might
  165.     receive data without headers. */
  166. if (ioctl(if_fd, EIOCFLUSH) == -1) {
  167. perror("tcpdump: enet ioctl EIOCFLUSH error");
  168. exit(-1);
  169. }
  170. /*  Set the receive queue depth to its maximum. */
  171. maxwaiting = ctl.en_maxwaiting;
  172. if (ioctl(if_fd, EIOCSETW, &maxwaiting) == -1) {
  173. perror("tcpdump: enet ioctl EIOCSETW error");
  174. exit(-1);
  175. }
  176. #ifdef IBMRTPC
  177. /*  Clear statistics. */
  178. if (ioctl(if_fd, EIOCLRSTAT, 0) == -1) {
  179. perror("tcpdump: enet ioctl EIOCLRSTAT error");
  180. exit(-1);
  181. }
  182. #endif /* IBMRTPC */
  183. /*  Set the filter (accept all packets). */
  184. filter.enf_Priority = 3;
  185. filter.enf_FilterLen = 0;
  186. if (ioctl(if_fd, EIOCSETF, &filter) == -1) {
  187. perror("tcpdump: enet ioctl EIOCSETF error");
  188. exit(-1);
  189. }
  190. /*
  191.  * "enetfilter" supports only ethernets.
  192.  */
  193. *linktype = DLT_EN10MB;
  194. return(if_fd);
  195. }