Diff for /libaitio/src/bpf.c between versions 1.1.2.2 and 1.1.2.3

version 1.1.2.2, 2013/06/24 11:09:55 version 1.1.2.3, 2013/06/24 14:43:00
Line 26  io_get1stiface(char *szIface, int iflen) Line 26  io_get1stiface(char *szIface, int iflen)
  * io_etherClose() - Close BPF interface   * io_etherClose() - Close BPF interface
  *   *
  * @eth = bpf handle   * @eth = bpf handle
    * @zcbuf = zero copy buffer, if BPF supports it and isn't NULL
  * return: none   * return: none
  */   */
 void  void
io_etherClose(int eth)io_etherClose(int eth, void **zcbuf)
 {  {
   #ifdef __FreeBSD__
           struct bpf_zbuf *zbuf = NULL;
   #endif
   
         if (eth > STDERR_FILENO)          if (eth > STDERR_FILENO)
                 close(eth);                  close(eth);
   
           if (zcbuf && *zcbuf) {
   #ifdef __FreeBSD__
                   zbuf = *zcbuf;
                   munmap(zbuf->bz_bufb, zbuf->bz_buflen);
                   munmap(zbuf->bz_bufa, zbuf->bz_buflen);
                   e_free(*zcbuf);
                   *zcbuf = NULL;
   #endif
           }
 }  }
   
   #ifdef __FreeBSD__
   static inline struct bpf_zbuf *
   allocZCbuf(u_int len)
   {
           struct bpf_zbuf *zbuf = NULL;
   
           zbuf = e_malloc(sizeof(struct bpf_zbuf));
           if (!zbuf) {
                   io_SetErr(elwix_GetErrno(), "%s", elwix_GetError());
                   return NULL;
           } else
                   memset(zbuf, 0, sizeof(struct bpf_zbuf));
   
           zbuf->bz_buflen = roundup(len, getpagesize());
           zbuf->bz_bufa = mmap(NULL, zbuf->bz_buflen, PROT_READ | PROT_WRITE, MAP_ANON, -1, 0);
           if (zbuf->bz_bufa == MAP_FAILED) {
                   LOGERR;
                   e_free(zbuf);
                   return NULL;
           } else
                   memset(zbuf->bz_bufa, 0, zbuf->bz_buflen);
           zbuf->bz_bufb = mmap(NULL, zbuf->bz_buflen, PROT_READ | PROT_WRITE, MAP_ANON, -1, 0);
           if (zbuf->bz_bufb == MAP_FAILED) {
                   LOGERR;
                   munmap(zbuf->bz_bufa, zbuf->bz_buflen);
                   e_free(zbuf);
                   return NULL;
           } else
                   memset(zbuf->bz_bufb, 0, zbuf->bz_buflen);
   
           return zbuf;
   }
   #endif
   
 /*  /*
  * io_etherOpen() - Open BPF interface to device   * io_etherOpen() - Open BPF interface to device
  *   *
  * @csIface = interface name   * @csIface = interface name
  * @flags = open flags   * @flags = open flags
    * @whdr = with complete headers
    * @buflen = buffer length
    * @zcbuf = zero copy buffer, if BPF supports it and isn't NULL
  * return: -1 error or >-1 bpf handle   * return: -1 error or >-1 bpf handle
  */   */
 int  int
io_etherOpen(const char *csIface, int flags)io_etherOpen(const char *csIface, int flags, int whdr, u_int buflen, void **zcbuf)
 {  {
         int n = 1, eth = -1;          int n = 1, eth = -1;
         register int i;          register int i;
Line 67  io_etherOpen(const char *csIface, int flags) Line 119  io_etherOpen(const char *csIface, int flags)
                 close(eth);                  close(eth);
                 return -1;                  return -1;
         }          }
        n = USHRT_MAX + 1;        if (whdr && ioctl(eth, BIOCSHDRCMPLT, &n) == -1) {
        if (ioctl(eth, BIOCSBLEN, &n) == -1) { 
                 LOGERR;                  LOGERR;
                 close(eth);                  close(eth);
                 return -1;                  return -1;
         }          }
           if (buflen) {
                   if (!zcbuf) {
                           if (ioctl(eth, BIOCSBLEN, &buflen) == -1) {
                                   LOGERR;
                                   close(eth);
                                   return -1;
                           }
                   } else {
   #ifdef __FreeBSD__
                           n = BPF_BUFMODE_ZBUF;
                           if (ioctl(eth, BIOCSETBUFMODE, &n) == -1) {
                                   LOGERR;
                                   close(eth);
                                   return -1;
                           }
                           if (ioctl(eth, BIOCGETZMAX, &n) == -1) {
                                   LOGERR;
                                   close(eth);
                                   return -1;
                           }
                           if (!(*zcbuf = allocZCbuf(MIN(n, buflen)))) {
                                   close(eth);
                                   return -1;
                           }
                           if (ioctl(eth, BIOCSETZBUF, (struct bpf_zbuf*) *zcbuf) == -1) {
                                   LOGERR;
                                   io_etherClose(eth, zcbuf);
                                   return -1;
                           }
   #endif
                   }
           }
         if (csIface)          if (csIface)
                 strlcpy(szStr, csIface, sizeof szStr);                  strlcpy(szStr, csIface, sizeof szStr);
         else if (io_get1stiface(szStr, sizeof szStr) == -1) {          else if (io_get1stiface(szStr, sizeof szStr) == -1) {
                close(eth);                io_etherClose(eth, zcbuf);
                 return -1;                  return -1;
         }          }
         strlcpy(ifr.ifr_name, szStr, sizeof ifr.ifr_name);          strlcpy(ifr.ifr_name, szStr, sizeof ifr.ifr_name);
         if (ioctl(eth, BIOCSETIF, &ifr) == -1) {          if (ioctl(eth, BIOCSETIF, &ifr) == -1) {
                 LOGERR;                  LOGERR;
                close(eth);                io_etherClose(eth, zcbuf);
                 return -1;                  return -1;
         }          }
   

Removed from v.1.1.2.2  
changed lines
  Added in v.1.1.2.3


FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>