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

Telnet服务器

开发平台:

Unix_Linux

  1. /* Nessuslib -- the Nessus Library
  2.  * Copyright (C) 1999 Renaud Deraison
  3.  *
  4.  * This library is free software; you can redistribute it and/or
  5.  * modify it under the terms of the GNU Library General Public
  6.  * License as published by the Free Software Foundation; either
  7.  * version 2 of the License, or (at your option) any later version.
  8.  *
  9.  * This library is distributed in the hope that it will be useful,
  10.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12.  * Library General Public License for more details.
  13.  *
  14.  * You should have received a copy of the GNU Library General Public
  15.  * License along with this library; if not, write to the Free
  16.  * Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17.  */
  18. #include <includes.h>
  19. #include <net/if.h>
  20. #ifdef HAVE_SYS_SOCKIO_H
  21. #include <sys/sockio.h>
  22. #endif
  23. #include "network.h"
  24. #define MAXROUTES 1024
  25. struct interface_info {
  26.     char name[64];
  27.     struct in_addr addr;
  28. };
  29. struct interface_info *getinterfaces(int *howmany);
  30. int is_local_ip(addr)
  31.  struct in_addr addr;
  32. {
  33.  int ifaces;
  34.  struct interface_info * ifs;
  35.  int i;
  36.  
  37.  if ((ifs = getinterfaces(&ifaces)) == NULL) 
  38.   return -1;
  39.  for(i=0;i<ifaces;i++)
  40.  {
  41.   bpf_u_int32 net, mask;
  42.   char errbuf[PCAP_ERRBUF_SIZE];
  43.   pcap_lookupnet(ifs[i].name, &net, &mask, errbuf);
  44.   if((net & mask) == (addr.s_addr & mask))
  45.    return 1;
  46.  }
  47.  return 0;
  48. }
  49. /* 
  50.  * We send an empty UDP packet to the remote host, and read back
  51.  * its mac address. 
  52.  *
  53.  * (we should first interrogate the kernel's arp cache - we may
  54.  * rely on libdnet in the future to do that)
  55.  *
  56.  * As a bonus, this function works well as a local ping
  57.  *
  58.  */
  59. int 
  60. get_mac_addr(addr, mac)
  61.  struct in_addr addr;
  62.  char ** mac;
  63. {
  64.  int soc = socket(AF_INET, SOCK_DGRAM, 0);
  65.  struct sockaddr_in soca;
  66.  int bpf;
  67.  struct in_addr me;
  68.  char * iface = routethrough(&addr, &me);
  69.  char filter[255];
  70.  char * src_host, * dst_host;
  71.  unsigned char * packet;
  72.  int len;
  73.  
  74.  *mac = NULL;
  75.  if(soc < 0)
  76.   return -1;
  77.   
  78.  src_host = estrdup(inet_ntoa(me));
  79.  dst_host = estrdup(inet_ntoa(addr));
  80.  snprintf(filter, sizeof(filter), "ip and (src host %s and dst host %s)",
  81.   src_host, dst_host);
  82.  efree(&src_host);
  83.  efree(&dst_host);
  84.  
  85.   
  86.  bpf = bpf_open_live(iface, filter);
  87.  if(bpf < 0)
  88.   {
  89.   close(soc);
  90.   return -1;
  91.   }
  92.   
  93.  /*
  94.   * We only deal with ethernet
  95.   */
  96.  if(bpf_datalink(bpf) != DLT_EN10MB)
  97.  {
  98.   bpf_close(bpf);
  99.   close(soc);
  100.   return -1;
  101.  }
  102.  
  103.  
  104.  
  105.  soca.sin_addr.s_addr = addr.s_addr;
  106.  soca.sin_port = htons(9); /* or whatever */
  107.  soca.sin_family = AF_INET;
  108.  if(sendto(soc, NULL, 0, 0, (struct sockaddr*)&soca, sizeof(soca)) == 0)
  109.  {
  110.   packet = (unsigned char*)bpf_next(bpf, &len);
  111.   if(packet)
  112.   { 
  113.    if(len >= get_datalink_size(bpf_datalink(bpf)))
  114.    {
  115.     int i;
  116.     for(i=0;i<6;i++)
  117.      if(packet[i]!=0xFF)break;
  118.     
  119.     if(i == 6)
  120.     {
  121.      bpf_close(bpf);
  122.      close(soc);
  123.      return 1;
  124.     }
  125.     
  126.     *mac = emalloc(22);
  127.     snprintf(*mac, 22, "%.2x.%.2x.%.2x.%.2x.%.2x.%.2x",
  128.      (unsigned char)packet[0],
  129. (unsigned char)packet[1], 
  130. (unsigned char)packet[2],
  131. (unsigned char)packet[3],
  132. (unsigned char)packet[4],
  133. (unsigned char)packet[5]);
  134.    bpf_close(bpf);
  135.    close(soc);
  136.    return 0;
  137.    }
  138.   }
  139.   else
  140.   {
  141.    bpf_close(bpf);
  142.    close(soc);
  143.    return 1;
  144.   }
  145.  }
  146.  bpf_close(bpf);
  147.  close(soc);
  148.  return -1;
  149. }
  150. /*
  151.  * Taken straight out of Fyodor's Nmap
  152.  */
  153. int ipaddr2devname( char *dev, int sz, struct in_addr *addr ) {
  154. struct interface_info *mydevs;
  155. int numdevs;
  156. int i;
  157. mydevs = getinterfaces(&numdevs);
  158. if (!mydevs) return -1;
  159. for(i=0; i < numdevs; i++) {
  160.   if (addr->s_addr == mydevs[i].addr.s_addr) {
  161.     dev[sz - 1] = '';
  162.     strncpy(dev, mydevs[i].name, sz);
  163.     return 0;
  164.   }
  165. }
  166. return -1;
  167. }
  168. /* Tests whether a packet sent to  IP is LIKELY to route 
  169.  through the kernel localhost interface */
  170. int islocalhost(struct in_addr *addr) {
  171. char dev[128];
  172.   if(addr == NULL)
  173.    return -1;
  174.   /* If it is 0.0.0.0 or starts with 127.0.0.1 then it is 
  175.      probably localhost */
  176.   if ((addr->s_addr & htonl(0xFF000000)) == htonl(0x7F000000))
  177.     return 1;
  178.   if (!addr->s_addr)
  179.     return 1;
  180.   /* If it is the same addy as a local interface, then it is
  181.      probably localhost */
  182.   if (ipaddr2devname(dev, sizeof(dev), addr) != -1)
  183.     return 1;
  184.   /* OK, so to a first approximation, this addy is probably not
  185.      localhost */
  186.   return 0;
  187. }
  188. int
  189. get_datalink_size(datalink)
  190.  int datalink;
  191. {
  192.  int offset = -1;
  193.  switch(datalink) {
  194.   case DLT_EN10MB: offset = 14; break;
  195.   case DLT_IEEE802: offset = 22; break;
  196.   case DLT_NULL: offset = 4; break;
  197.   case DLT_SLIP:
  198. #if (FREEBSD || OPENBSD || NETBSD || BSDI || DARWIN)
  199.     offset = 16;
  200. #else
  201.     offset = 24; /* Anyone use this??? */
  202. #endif
  203.     break;
  204.   case DLT_PPP: 
  205. #if (FREEBSD || OPENBSD || NETBSD || BSDI || DARWIN)
  206.     offset = 4;
  207. #else
  208. #ifdef SOLARIS
  209.     offset = 8;
  210. #else
  211.     offset = 24; /* Anyone use this? */
  212. #endif /* ifdef solaris */
  213. #endif /* if freebsd || openbsd || netbsd || bsdi */
  214.     break;
  215.   case DLT_RAW: offset = 0; break;
  216.   }
  217.   return(offset);
  218. }
  219. int get_random_bytes(void *buf, int numbytes) {
  220. static char bytebuf[2048];
  221. static char badrandomwarning = 0;
  222. static int bytesleft = 0;
  223. int res;
  224. int tmp;
  225. struct timeval tv;
  226. FILE *fp = NULL;
  227. int i;
  228. short *iptr;
  229. if (numbytes < 0 || numbytes > 0xFFFF) return -1;
  230. if (bytesleft == 0) {
  231.   fp = fopen("/dev/urandom", "r");
  232.   if (!fp) fp = fopen("/dev/random", "r");
  233.   if (fp) {
  234.     res = fread(bytebuf, 1, sizeof(bytebuf), fp);
  235.     if (res != sizeof(bytebuf)) {    
  236.       fclose(fp);
  237.       fp = NULL;
  238.     }      
  239.     bytesleft = sizeof(bytebuf);
  240.   }
  241.   if (!fp) {  
  242.     if (badrandomwarning == 0) {
  243.       badrandomwarning++;
  244.     }
  245.     /* Seed our random generator */
  246.     gettimeofday(&tv, NULL);
  247.     srand((tv.tv_sec ^ tv.tv_usec) ^ getpid());
  248.     for(i=0; i < sizeof(bytebuf) / sizeof(short); i++) {
  249.       iptr = (short *) ((char *)bytebuf + i * sizeof(short));
  250.       *iptr = rand();
  251.     }
  252.     bytesleft = (sizeof(bytebuf) / sizeof(short)) * sizeof(short);
  253.     /*    ^^^^^^^^^^^^^^^not as meaningless as it looks  */
  254.   } else fclose(fp);
  255. }
  256. if (numbytes <= bytesleft) { /* we can cover it */
  257.   memcpy(buf, bytebuf + (sizeof(bytebuf) - bytesleft), numbytes);
  258.   bytesleft -= numbytes;
  259.   return 0;
  260. }
  261. /* We don't have enough */
  262. memcpy(buf, bytebuf + (sizeof(bytebuf) - bytesleft), bytesleft);
  263. tmp = bytesleft;
  264. bytesleft = 0;
  265. return get_random_bytes((char *)buf + tmp, numbytes - tmp);
  266. }
  267. struct interface_info *getinterfaces(int *howmany) {
  268.   static struct interface_info mydevs[1024];
  269.   int numinterfaces = 0;
  270.   int sd;
  271.   int len;
  272.   char *p;
  273.   char buf[10240];
  274.   struct ifconf ifc;
  275.   struct ifreq *ifr;
  276.   struct sockaddr_in *sin;
  277.     /* Dummy socket for ioctl */
  278.     sd = socket(AF_INET, SOCK_DGRAM, 0);
  279.     bzero(buf, sizeof(buf));
  280.     if (sd < 0) printf("socket in getinterfaces");
  281.     ifc.ifc_len = sizeof(buf);
  282.     ifc.ifc_buf = buf;
  283.     if (ioctl(sd, SIOCGIFCONF, &ifc) < 0) {
  284.       printf("Failed to determine your configured interfaces!n");
  285.     }
  286.     close(sd);
  287.     ifr = (struct ifreq *) buf;
  288.     if (ifc.ifc_len == 0) 
  289.       printf("getinterfaces: SIOCGIFCONF claims you have no network interfaces!n");
  290. #ifdef HAVE_SOCKADDR_SA_LEN
  291.     len = ifr->ifr_addr.sa_len;
  292. #else
  293. #ifdef HAVE_STRUCT_IFMAP
  294.     len = sizeof(struct ifmap);
  295. #else
  296.     len = sizeof(struct sockaddr);
  297. #endif
  298. #endif
  299.     for(; ifr && *((char *)ifr) && ((char *)ifr) < buf + ifc.ifc_len; 
  300. ((*(char **)&ifr) +=  sizeof(ifr->ifr_name) + len )) {
  301.       sin = (struct sockaddr_in *) &ifr->ifr_addr;
  302.       memcpy(&(mydevs[numinterfaces].addr), (char *) &(sin->sin_addr), sizeof(struct in_addr));
  303.       /* In case it is a stinkin' alias */
  304.       if ((p = strchr(ifr->ifr_name, ':')))
  305. *p = '';
  306.       strncpy(mydevs[numinterfaces].name, ifr->ifr_name, 63);
  307.       mydevs[numinterfaces].name[63] = '';
  308.       numinterfaces++;
  309.       if (numinterfaces == 1023)  {      
  310. printf("My god!  You seem to have WAY too many interfaces!  Things may not work rightn");
  311. break;
  312.       }
  313. #if HAVE_SOCKADDR_SA_LEN
  314.       /* len = MAX(sizeof(struct sockaddr), ifr->ifr_addr.sa_len);*/
  315.       len = ifr->ifr_addr.sa_len;
  316. #endif 
  317.       mydevs[numinterfaces].name[0] = '';
  318.   }
  319.   if (howmany) *howmany = numinterfaces;
  320.   return mydevs;
  321. }
  322. int getsourceip(struct in_addr *src, struct in_addr *dst) {
  323.   int sd;
  324.   struct sockaddr_in sock;
  325.   unsigned int socklen = sizeof(struct sockaddr_in);
  326.   unsigned short p1;
  327.   
  328.   
  329.  
  330.   
  331.   *src = socket_get_next_source_addr(NULL);
  332.   if ( src->s_addr != INADDR_ANY )
  333.   {
  334.    return 1;
  335.   }
  336.   
  337.   get_random_bytes(&p1, 2);
  338.   if (p1 < 5000) p1 += 5000;
  339.   if ((sd = socket(AF_INET, SOCK_DGRAM, 0)) == -1)
  340.     {perror("Socket troubles"); return 0;}
  341.   sock.sin_family = AF_INET;
  342.   sock.sin_addr = *dst;
  343.   sock.sin_port = htons(p1);
  344.   if (connect(sd, (struct sockaddr *) &sock, sizeof(struct sockaddr_in)) == -1)
  345.     { perror("UDP connect()");
  346.     close(sd);
  347.     return 0;
  348.     }
  349.   bzero(&sock, sizeof(struct sockaddr_in));
  350.   if (getsockname(sd, (struct sockaddr *)&sock, &socklen) == -1) {
  351.     perror("getsockname");
  352.     close(sd);
  353.     return 0;
  354.   }
  355.   src->s_addr = sock.sin_addr.s_addr;
  356.   close(sd);
  357.   return 1; /* Calling function responsible for checking validity */
  358. }
  359. /* An awesome function to determine what interface a packet to a given
  360.    destination should be routed through.  It returns NULL if no appropriate
  361.    interface is found, oterwise it returns the device name and fills in the
  362.    source parameter.   Some of the stuff is
  363.    from Stevens' Unix Network Programming V2.  He had an easier suggestion
  364.    for doing this (in the book), but it isn't portable :( */
  365.    
  366.    
  367. /* An awesome function to determine what interface a packet to a given
  368.    destination should be routed through.  It returns NULL if no appropriate
  369.    interface is found, oterwise it returns the device name and fills in the
  370.    source parameter.   Some of the stuff is
  371.    from Stevens' Unix Network Programming V2.  He had an easier suggestion
  372.    for doing this (in the book), but it isn't portable :( */
  373. char *routethrough(struct in_addr *dest, struct in_addr *source) {
  374.   static int initialized = 0;
  375.   int i;
  376.   struct in_addr addy;
  377.   static enum { procroutetechnique, connectsockettechnique, guesstechnique } technique = procroutetechnique;
  378.   char buf[10240];
  379.   struct interface_info *mydevs;
  380.   static struct myroute {
  381.     struct interface_info *dev;
  382.     unsigned long mask;
  383.     unsigned long dest;
  384.   } myroutes[MAXROUTES];
  385.   int numinterfaces = 0;
  386.   char *p, *endptr;
  387.   char iface[64];
  388.   static int numroutes = 0;
  389.   FILE *routez;
  390.   
  391.   struct in_addr src = socket_get_next_source_addr(NULL);
  392.   
  393.   if (!dest) printf("ipaddr2devname passed a NULL dest address");
  394.   if (!initialized) {  
  395.     /* Dummy socket for ioctl */
  396.     initialized = 1;
  397.     mydevs = getinterfaces(&numinterfaces);
  398.     /* Now we must go through several techniques to determine info */
  399.     routez = fopen("/proc/net/route", "r");
  400.     if (routez) {
  401.       /* OK, linux style /proc/net/route ... we can handle this ... */
  402.       /* Now that we've got the interfaces, we g0 after the r0ut3Z */
  403.       
  404.       fgets(buf, sizeof(buf), routez); /* Kill the first line */
  405.       while(fgets(buf,sizeof(buf), routez)) {
  406. p = strtok(buf, " tn");
  407. if (!p) {
  408.   printf("Could not find interface in /proc/net/route line");
  409.   continue;
  410. }
  411. strncpy(iface, p, sizeof(iface));
  412. if ((p = strchr(iface, ':'))) {
  413.   *p = ''; /* To support IP aliasing */
  414. }
  415. p = strtok(NULL, " tn");
  416. endptr = NULL;
  417. myroutes[numroutes].dest = strtoul(p, &endptr, 16);
  418. if (!endptr || *endptr) {
  419.   printf("Failed to determine Destination from /proc/net/route");
  420.   continue;
  421. }
  422. for(i=0; i < 6; i++) {
  423.   p = strtok(NULL, " tn");
  424.   if (!p) break;
  425. }
  426. if (!p) {
  427.   printf("Failed to find field %d in /proc/net/route", i + 2);
  428.   continue;
  429. }
  430. endptr = NULL;
  431. myroutes[numroutes].mask = strtoul(p, &endptr, 16);
  432. if (!endptr || *endptr) {
  433.   printf("Failed to determine mask from /proc/net/route");
  434.   continue;
  435. }
  436. #if TCPIP_DEBUGGING
  437.   printf("#%d: for dev %s, The dest is %lX and the mask is %lXn", numroutes, iface, myroutes[numroutes].dest, myroutes[numroutes].mask);
  438. #endif
  439.   for(i=0; i < numinterfaces; i++)
  440.     if (!strcmp(iface, mydevs[i].name)) {
  441.       myroutes[numroutes].dev = &mydevs[i];
  442.       break;
  443.     }
  444.   if (i == numinterfaces) 
  445.     printf("Failed to find interface %s mentioned in /proc/net/routen", iface);
  446.   numroutes++;
  447.   if (numroutes >= MAXROUTES)
  448.             {
  449.     printf("My god!  You seem to have WAY to many routes!n");
  450.             break;
  451.             }
  452.       }
  453.       fclose(routez);
  454.     } else {
  455.       technique = connectsockettechnique;
  456.     }
  457.   } else {  
  458.     mydevs = getinterfaces(&numinterfaces);
  459.   }
  460.   /* WHEW, that takes care of initializing, now we have the easy job of 
  461.      finding which route matches */
  462.   if (islocalhost(dest)) {
  463.     if (source)
  464.       source->s_addr = htonl(0x7F000001);
  465.     /* Now we find the localhost interface name, assuming 127.0.0.1 is
  466.        localhost (it damn well better be!)... */
  467.     for(i=0; i < numinterfaces; i++) {    
  468.       if (mydevs[i].addr.s_addr == htonl(0x7F000001)) {
  469. return mydevs[i].name;
  470.       }
  471.     }
  472.     return NULL;
  473.   }
  474.   if (technique == procroutetechnique) {    
  475.     for(i=0; i < numroutes; i++) {  
  476.       if ((dest->s_addr & myroutes[i].mask) == myroutes[i].dest) {
  477. if (source) {
  478.   
  479.   if ( src.s_addr != INADDR_ANY )
  480.    source->s_addr = src.s_addr;
  481.   else
  482.        source->s_addr = myroutes[i].dev->addr.s_addr; 
  483. }
  484. return myroutes[i].dev->name;      
  485.       }
  486.     }
  487.   } else if (technique == connectsockettechnique) {
  488.       if (!getsourceip(&addy, dest))
  489. return NULL;
  490.       if (!addy.s_addr)  {  /* Solaris 2.4 */
  491.         struct hostent *myhostent = NULL;
  492.         char myname[MAXHOSTNAMELEN + 1];
  493. #if defined(USE_PTHREADS) && defined(HAVE_GETHOSTBYNAME_R)
  494.         int Errno = 0;
  495.         char * buf = emalloc(4096);
  496.         struct hostent * res = NULL;
  497.         struct hostent * t = NULL;
  498. myhostent = emalloc(sizeof(struct hostent));
  499. #ifdef HAVE_SOLARIS_GETHOSTBYNAME_R
  500.  gethostbyname_r(myname, myhostent, buf, 4096, &Errno);
  501.  if(Errno){
  502.    free(myhostent);
  503. myhostent = NULL;
  504. }
  505. #else
  506.          gethostbyname_r(myname, myhostent, buf, 4096, &res, &Errno);
  507.          t = myhostent;
  508.          myhostent = res;
  509. #endif /* HAVE_SOLARIS_... */
  510. myhostent = res;
  511. #else
  512. myhostent = gethostbyname(myname);
  513. #endif /* USE_PTHREADS     */
  514.         if (gethostname(myname, MAXHOSTNAMELEN) || 
  515.            !myhostent)
  516.   printf("Cannot get hostname!  Try using -S <my_IP_address> or -e <interface to scan through>n");
  517.         memcpy(&(addy.s_addr), myhostent->h_addr_list[0], sizeof(struct in_addr));
  518. #if defined(USE_PTHREADS) && defined(HAVE_GETHOSTBYNAME_R)
  519. if(myhostent)free(myhostent);
  520. free(buf);
  521. #endif
  522.       }
  523.       /* Now we insure this claimed address is a real interface ... */
  524.       for(i=0; i < numinterfaces; i++)
  525. if (mydevs[i].addr.s_addr == addy.s_addr) {
  526.   if (source) {
  527.     source->s_addr = addy.s_addr;   
  528.   }
  529.   return mydevs[i].name;
  530. }  
  531.       return NULL;
  532.     } else 
  533.       printf("I know sendmail technique ... I know rdist technique ... but I don't know what the hell kindof technique you are attempting!!!");
  534.     return NULL;
  535. }