The Tcpdump Group git mirrors - libpcap/blob - pcap-netfilter-linux.c

   1 /*
   2  * Copyright (c) 2011 Jakub Zawadzki
   3  * 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  *
   9  * 1. Redistributions of source code must retain the above copyright
  10  * notice, this list of conditions and the following disclaimer.
  11  * 2. Redistributions in binary form must reproduce the above copyright
  12  * notice, this list of conditions and the following disclaimer in the
  13  * documentation and/or other materials provided with the distribution.
  14  * 3. The name of the author may not be used to endorse or promote
  15  * products derived from this software without specific prior written
  16  * permission.
  17  *
  18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  29  */
  30
  31 #ifdef HAVE_CONFIG_H
  32 #include <config.h>
  33 #endif
  34
  35 #include "pcap-int.h"
  36
  37 #ifdef NEED_STRERROR_H
  38 #include "strerror.h"
  39 #endif
  40
  41 #include <errno.h>
  42 #include <stdlib.h>
  43 #include <unistd.h>
  44 #include <string.h>
  45 #include <sys/socket.h>
  46 #include <arpa/inet.h>
  47
  48 #include <time.h>
  49 #include <sys/time.h>
  50 #include <netinet/in.h>
  51 #include <linux/types.h>
  52
  53 #include <linux/netlink.h>
  54 #include <linux/netfilter.h>
  55 #include <linux/netfilter/nfnetlink.h>
  56 #include <linux/netfilter/nfnetlink_log.h>
  57 #include <linux/netfilter/nfnetlink_queue.h>
  58
  59 /* NOTE: if your program drops privilages after pcap_activate() it WON'T work with nfqueue.
  60  *       It took me quite some time to debug ;/
  61  *
  62  *       Sending any data to nfnetlink socket requires CAP_NET_ADMIN privilages,
  63  *       and in nfqueue we need to send verdict reply after recving packet.
  64  *
  65  *       In tcpdump you can disable dropping privilages with -Z root
  66  */
  67
  68 #include "pcap-netfilter-linux.h"
  69
  70 #define HDR_LENGTH (NLMSG_LENGTH(NLMSG_ALIGN(sizeof(struct nfgenmsg))))
  71
  72 #define NFLOG_IFACE "nflog"
  73 #define NFQUEUE_IFACE "nfqueue"
  74
  75 typedef enum { OTHER = -1, NFLOG, NFQUEUE } nftype_t;
  76
  77 /*
  78  * Private data for capturing on Linux netfilter sockets.
  79  */
  80 struct pcap_netfilter {
  81         u_int   packets_read;   /* count of packets read with recvfrom() */
  82         u_int   packets_nobufs; /* ENOBUFS counter */
  83 };
  84
  85 static int nfqueue_send_verdict(const pcap_t *handle, uint16_t group_id, u_int32_t id, u_int32_t verdict);
  86
  87
  88 static int
  89 netfilter_read_linux(pcap_t *handle, int max_packets, pcap_handler callback, u_char *user)
  90 {
  91         struct pcap_netfilter *handlep = handle->priv;
  92         register u_char *bp, *ep;
  93         int count = 0;
  94         ssize_t len;
  95
  96         /*
  97          * Has "pcap_breakloop()" been called?
  98          */
  99         if (handle->break_loop) {
 100                 /*
 101                  * Yes - clear the flag that indicates that it
 102                  * has, and return PCAP_ERROR_BREAK to indicate
 103                  * that we were told to break out of the loop.
 104                  */
 105                 handle->break_loop = 0;
 106                 return PCAP_ERROR_BREAK;
 107         }
 108         len = handle->cc;
 109         if (len == 0) {
 110                 /*
 111                  * The buffer is empty; refill it.
 112                  *
 113                  * We ignore EINTR, as that might just be due to a signal
 114                  * being delivered - if the signal should interrupt the
 115                  * loop, the signal handler should call pcap_breakloop()
 116                  * to set handle->break_loop (we ignore it on other
 117                  * platforms as well).
 118                  */
 119                 do {
 120                         len = recv(handle->fd, handle->buffer, handle->bufsize, 0);
 121                         if (handle->break_loop) {
 122                                 handle->break_loop = 0;
 123                                 return PCAP_ERROR_BREAK;
 124                         }
 125                         if (errno == ENOBUFS)
 126                                 handlep->packets_nobufs++;
 127                 } while ((len == -1) && (errno == EINTR || errno == ENOBUFS));
 128
 129                 if (len < 0) {
 130                         pcap_fmt_errmsg_for_errno(handle->errbuf,
 131                             PCAP_ERRBUF_SIZE, errno, "Can't receive packet");
 132                         return PCAP_ERROR;
 133                 }
 134
 135                 bp = (unsigned char *)handle->buffer;
 136         } else
 137                 bp = handle->bp;
 138         ep = bp + len;
 139         while (bp < ep) {
 140                 const struct nlmsghdr *nlh = (const struct nlmsghdr *) bp;
 141                 uint32_t msg_len;
 142                 nftype_t type = OTHER;
 143                 /*
 144                  * Has "pcap_breakloop()" been called?
 145                  * If so, return immediately - if we haven't read any
 146                  * packets, clear the flag and return PCAP_ERROR_BREAK
 147                  * to indicate that we were told to break out of the loop,
 148                  * otherwise leave the flag set, so that the *next* call
 149                  * will break out of the loop without having read any
 150                  * packets, and return the number of packets we've
 151                  * processed so far.
 152                  */
 153                 if (handle->break_loop) {
 154                         handle->bp = bp;
 155                         handle->cc = (int)(ep - bp);
 156                         if (count == 0) {
 157                                 handle->break_loop = 0;
 158                                 return PCAP_ERROR_BREAK;
 159                         } else
 160                                 return count;
 161                 }
 162                 if (ep - bp < NLMSG_SPACE(0)) {
 163                         /*
 164                          * There's less than one netlink message left
 165                          * in the buffer.  Give up.
 166                          */
 167                         break;
 168                 }
 169
 170                 if (nlh->nlmsg_len < sizeof(struct nlmsghdr) || (u_int)len < nlh->nlmsg_len) {
 171                         snprintf(handle->errbuf, PCAP_ERRBUF_SIZE, "Message truncated: (got: %zd) (nlmsg_len: %u)", len, nlh->nlmsg_len);
 172                         return -1;
 173                 }
 174
 175                 if (NFNL_SUBSYS_ID(nlh->nlmsg_type) == NFNL_SUBSYS_ULOG &&
 176                     NFNL_MSG_TYPE(nlh->nlmsg_type) == NFULNL_MSG_PACKET)
 177                         type = NFLOG;
 178                 else if (NFNL_SUBSYS_ID(nlh->nlmsg_type) == NFNL_SUBSYS_QUEUE &&
 179                          NFNL_MSG_TYPE(nlh->nlmsg_type) == NFQNL_MSG_PACKET)
 180                         type = NFQUEUE;
 181
 182                 if (type != OTHER) {
 183                         const unsigned char *payload = NULL;
 184                         struct pcap_pkthdr pkth;
 185
 186                         const struct nfgenmsg *nfg = NULL;
 187                         int id = 0;
 188
 189                         if (handle->linktype != DLT_NFLOG) {
 190                                 const struct nfattr *payload_attr = NULL;
 191
 192                                 if (nlh->nlmsg_len < HDR_LENGTH) {
 193                                         snprintf(handle->errbuf, PCAP_ERRBUF_SIZE, "Malformed message: (nlmsg_len: %u)", nlh->nlmsg_len);
 194                                         return -1;
 195                                 }
 196
 197                                 nfg = NLMSG_DATA(nlh);
 198                                 if (nlh->nlmsg_len > HDR_LENGTH) {
 199                                         struct nfattr *attr = NFM_NFA(nfg);
 200                                         int attr_len = nlh->nlmsg_len - NLMSG_ALIGN(HDR_LENGTH);
 201
 202                                         while (NFA_OK(attr, attr_len)) {
 203                                                 if (type == NFQUEUE) {
 204                                                         switch (NFA_TYPE(attr)) {
 205                                                                 case NFQA_PACKET_HDR:
 206                                                                         {
 207                                                                                 const struct nfqnl_msg_packet_hdr *pkt_hdr = (const struct nfqnl_msg_packet_hdr *) NFA_DATA(attr);
 208
 209                                                                                 id = ntohl(pkt_hdr->packet_id);
 210                                                                                 break;
 211                                                                         }
 212                                                                 case NFQA_PAYLOAD:
 213                                                                         payload_attr = attr;
 214                                                                         break;
 215                                                         }
 216
 217                                                 } else if (type == NFLOG) {
 218                                                         switch (NFA_TYPE(attr)) {
 219                                                                 case NFULA_PAYLOAD:
 220                                                                         payload_attr = attr;
 221                                                                         break;
 222                                                         }
 223                                                 }
 224                                                 attr = NFA_NEXT(attr, attr_len);
 225                                         }
 226                                 }
 227
 228                                 if (payload_attr) {
 229                                         payload = NFA_DATA(payload_attr);
 230                                         pkth.len = pkth.caplen = NFA_PAYLOAD(payload_attr);
 231                                 }
 232
 233                         } else {
 234                                 payload = NLMSG_DATA(nlh);
 235                                 pkth.caplen = pkth.len = nlh->nlmsg_len-NLMSG_ALIGN(sizeof(struct nlmsghdr));
 236                         }
 237
 238                         if (payload) {
 239                                 /* pkth.caplen = min (payload_len, handle->snapshot); */
 240
 241                                 gettimeofday(&pkth.ts, NULL);
 242                                 if (handle->fcode.bf_insns == NULL ||
 243                                                 pcap_filter(handle->fcode.bf_insns, payload, pkth.len, pkth.caplen))
 244                                 {
 245                                         handlep->packets_read++;
 246                                         callback(user, &pkth, payload);
 247                                         count++;
 248                                 }
 249                         }
 250
 251                         if (type == NFQUEUE) {
 252                                 /* XXX, possible responses: NF_DROP, NF_ACCEPT, NF_STOLEN, NF_QUEUE, NF_REPEAT, NF_STOP */
 253                                 /* if type == NFQUEUE, handle->linktype is always != DLT_NFLOG,
 254                                    so nfg is always initialized to NLMSG_DATA(nlh). */
 255                                 if (nfg != NULL)
 256                                         nfqueue_send_verdict(handle, ntohs(nfg->res_id), id, NF_ACCEPT);
 257                         }
 258                 }
 259
 260                 msg_len = NLMSG_ALIGN(nlh->nlmsg_len);
 261                 /*
 262                  * If the message length would run past the end of the
 263                  * buffer, truncate it to the remaining space in the
 264                  * buffer.
 265                  */
 266                 if (msg_len > ep - bp)
 267                         msg_len = (uint32_t)(ep - bp);
 268
 269                 bp += msg_len;
 270                 if (count >= max_packets && !PACKET_COUNT_IS_UNLIMITED(max_packets)) {
 271                         handle->bp = bp;
 272                         handle->cc = (int)(ep - bp);
 273                         if (handle->cc < 0)
 274                                 handle->cc = 0;
 275                         return count;
 276                 }
 277         }
 278
 279         handle->cc = 0;
 280         return count;
 281 }
 282
 283 static int
 284 netfilter_set_datalink(pcap_t *handle, int dlt)
 285 {
 286         handle->linktype = dlt;
 287         return 0;
 288 }
 289
 290 static int
 291 netfilter_stats_linux(pcap_t *handle, struct pcap_stat *stats)
 292 {
 293         struct pcap_netfilter *handlep = handle->priv;
 294
 295         stats->ps_recv = handlep->packets_read;
 296         stats->ps_drop = handlep->packets_nobufs;
 297         stats->ps_ifdrop = 0;
 298         return 0;
 299 }
 300
 301 static int
 302 netfilter_inject_linux(pcap_t *handle, const void *buf _U_, int size _U_)
 303 {
 304         snprintf(handle->errbuf, PCAP_ERRBUF_SIZE,
 305             "Packet injection is not supported on netfilter devices");
 306         return (-1);
 307 }
 308
 309 struct my_nfattr {
 310         uint16_t nfa_len;
 311         uint16_t nfa_type;
 312         void *data;
 313 };
 314
 315 static int
 316 netfilter_send_config_msg(const pcap_t *handle, uint16_t msg_type, int ack, u_int8_t family, u_int16_t res_id, const struct my_nfattr *mynfa)
 317 {
 318         char buf[1024] __attribute__ ((aligned));
 319         memset(buf, 0, sizeof(buf));
 320
 321         struct nlmsghdr *nlh = (struct nlmsghdr *) buf;
 322         struct nfgenmsg *nfg = (struct nfgenmsg *) (buf + sizeof(struct nlmsghdr));
 323
 324         struct sockaddr_nl snl;
 325         static unsigned int seq_id;
 326
 327         if (!seq_id)
 328                 seq_id = time(NULL);
 329         ++seq_id;
 330
 331         nlh->nlmsg_len = NLMSG_LENGTH(sizeof(struct nfgenmsg));
 332         nlh->nlmsg_type = msg_type;
 333         nlh->nlmsg_flags = NLM_F_REQUEST | (ack ? NLM_F_ACK : 0);
 334         nlh->nlmsg_pid = 0;     /* to kernel */
 335         nlh->nlmsg_seq = seq_id;
 336
 337         nfg->nfgen_family = family;
 338         nfg->version = NFNETLINK_V0;
 339         nfg->res_id = htons(res_id);
 340
 341         if (mynfa) {
 342                 struct nfattr *nfa = (struct nfattr *) (buf + NLMSG_ALIGN(nlh->nlmsg_len));
 343
 344                 nfa->nfa_type = mynfa->nfa_type;
 345                 nfa->nfa_len = NFA_LENGTH(mynfa->nfa_len);
 346                 memcpy(NFA_DATA(nfa), mynfa->data, mynfa->nfa_len);
 347                 nlh->nlmsg_len = NLMSG_ALIGN(nlh->nlmsg_len) + NFA_ALIGN(nfa->nfa_len);
 348         }
 349
 350         memset(&snl, 0, sizeof(snl));
 351         snl.nl_family = AF_NETLINK;
 352
 353         if (sendto(handle->fd, nlh, nlh->nlmsg_len, 0, (struct sockaddr *) &snl, sizeof(snl)) == -1)
 354                 return -1;
 355
 356         if (!ack)
 357                 return 0;
 358
 359         /* waiting for reply loop */
 360         do {
 361                 socklen_t addrlen = sizeof(snl);
 362                 int len;
 363
 364                 /* ignore interrupt system call error */
 365                 do {
 366                         /*
 367                          * The buffer is not so big that its size won't
 368                          * fit into an int.
 369                          */
 370                         len = (int)recvfrom(handle->fd, buf, sizeof(buf), 0, (struct sockaddr *) &snl, &addrlen);
 371                 } while ((len == -1) && (errno == EINTR));
 372
 373                 if (len <= 0)
 374                         return len;
 375
 376                 if (addrlen != sizeof(snl) || snl.nl_family != AF_NETLINK) {
 377                         errno = EINVAL;
 378                         return -1;
 379                 }
 380
 381                 nlh = (struct nlmsghdr *) buf;
 382                 if (snl.nl_pid != 0 || seq_id != nlh->nlmsg_seq)        /* if not from kernel or wrong sequence skip */
 383                         continue;
 384
 385                 while ((u_int)len >= NLMSG_SPACE(0) && NLMSG_OK(nlh, (u_int)len)) {
 386                         if (nlh->nlmsg_type == NLMSG_ERROR || (nlh->nlmsg_type == NLMSG_DONE && nlh->nlmsg_flags & NLM_F_MULTI)) {
 387                                 if (nlh->nlmsg_len < NLMSG_ALIGN(sizeof(struct nlmsgerr))) {
 388                                         errno = EBADMSG;
 389                                         return -1;
 390                                 }
 391                                 errno = -(*((int *)NLMSG_DATA(nlh)));
 392                                 return (errno == 0) ? 0 : -1;
 393                         }
 394                         nlh = NLMSG_NEXT(nlh, len);
 395                 }
 396         } while (1);
 397
 398         return -1; /* never here */
 399 }
 400
 401 static int
 402 nflog_send_config_msg(const pcap_t *handle, uint8_t family, u_int16_t group_id, const struct my_nfattr *mynfa)
 403 {
 404         return netfilter_send_config_msg(handle, (NFNL_SUBSYS_ULOG << 8) | NFULNL_MSG_CONFIG, 1, family, group_id, mynfa);
 405 }
 406
 407 static int
 408 nflog_send_config_cmd(const pcap_t *handle, uint16_t group_id, u_int8_t cmd, u_int8_t family)
 409 {
 410         struct nfulnl_msg_config_cmd msg;
 411         struct my_nfattr nfa;
 412
 413         msg.command = cmd;
 414
 415         nfa.data = &msg;
 416         nfa.nfa_type = NFULA_CFG_CMD;
 417         nfa.nfa_len = sizeof(msg);
 418
 419         return nflog_send_config_msg(handle, family, group_id, &nfa);
 420 }
 421
 422 static int
 423 nflog_send_config_mode(const pcap_t *handle, uint16_t group_id, u_int8_t copy_mode, u_int32_t copy_range)
 424 {
 425         struct nfulnl_msg_config_mode msg;
 426         struct my_nfattr nfa;
 427
 428         msg.copy_range = htonl(copy_range);
 429         msg.copy_mode = copy_mode;
 430
 431         nfa.data = &msg;
 432         nfa.nfa_type = NFULA_CFG_MODE;
 433         nfa.nfa_len = sizeof(msg);
 434
 435         return nflog_send_config_msg(handle, AF_UNSPEC, group_id, &nfa);
 436 }
 437
 438 static int
 439 nfqueue_send_verdict(const pcap_t *handle, uint16_t group_id, u_int32_t id, u_int32_t verdict)
 440 {
 441         struct nfqnl_msg_verdict_hdr msg;
 442         struct my_nfattr nfa;
 443
 444         msg.id = htonl(id);
 445         msg.verdict = htonl(verdict);
 446
 447         nfa.data = &msg;
 448         nfa.nfa_type = NFQA_VERDICT_HDR;
 449         nfa.nfa_len = sizeof(msg);
 450
 451         return netfilter_send_config_msg(handle, (NFNL_SUBSYS_QUEUE << 8) | NFQNL_MSG_VERDICT, 0, AF_UNSPEC, group_id, &nfa);
 452 }
 453
 454 static int
 455 nfqueue_send_config_msg(const pcap_t *handle, uint8_t family, u_int16_t group_id, const struct my_nfattr *mynfa)
 456 {
 457         return netfilter_send_config_msg(handle, (NFNL_SUBSYS_QUEUE << 8) | NFQNL_MSG_CONFIG, 1, family, group_id, mynfa);
 458 }
 459
 460 static int
 461 nfqueue_send_config_cmd(const pcap_t *handle, uint16_t group_id, u_int8_t cmd, u_int16_t pf)
 462 {
 463         struct nfqnl_msg_config_cmd msg;
 464         struct my_nfattr nfa;
 465
 466         msg.command = cmd;
 467         msg.pf = htons(pf);
 468
 469         nfa.data = &msg;
 470         nfa.nfa_type = NFQA_CFG_CMD;
 471         nfa.nfa_len = sizeof(msg);
 472
 473         return nfqueue_send_config_msg(handle, AF_UNSPEC, group_id, &nfa);
 474 }
 475
 476 static int
 477 nfqueue_send_config_mode(const pcap_t *handle, uint16_t group_id, u_int8_t copy_mode, u_int32_t copy_range)
 478 {
 479         struct nfqnl_msg_config_params msg;
 480         struct my_nfattr nfa;
 481
 482         msg.copy_range = htonl(copy_range);
 483         msg.copy_mode = copy_mode;
 484
 485         nfa.data = &msg;
 486         nfa.nfa_type = NFQA_CFG_PARAMS;
 487         nfa.nfa_len = sizeof(msg);
 488
 489         return nfqueue_send_config_msg(handle, AF_UNSPEC, group_id, &nfa);
 490 }
 491
 492 static int
 493 netfilter_activate(pcap_t* handle)
 494 {
 495         const char *dev = handle->opt.device;
 496         unsigned short groups[32];
 497         int group_count = 0;
 498         nftype_t type = OTHER;
 499         int i;
 500
 501         if (strncmp(dev, NFLOG_IFACE, strlen(NFLOG_IFACE)) == 0) {
 502                 dev += strlen(NFLOG_IFACE);
 503                 type = NFLOG;
 504
 505         } else if (strncmp(dev, NFQUEUE_IFACE, strlen(NFQUEUE_IFACE)) == 0) {
 506                 dev += strlen(NFQUEUE_IFACE);
 507                 type = NFQUEUE;
 508         }
 509
 510         if (type != OTHER && *dev == ':') {
 511                 dev++;
 512                 while (*dev) {
 513                         long int group_id;
 514                         char *end_dev;
 515
 516                         if (group_count == 32) {
 517                                 snprintf(handle->errbuf, PCAP_ERRBUF_SIZE,
 518                                                 "Maximum 32 netfilter groups! dev: %s",
 519                                                 handle->opt.device);
 520                                 return PCAP_ERROR;
 521                         }
 522
 523                         group_id = strtol(dev, &end_dev, 0);
 524                         if (end_dev != dev) {
 525                                 if (group_id < 0 || group_id > 65535) {
 526                                         snprintf(handle->errbuf, PCAP_ERRBUF_SIZE,
 527                                                         "Netfilter group range from 0 to 65535 (got %ld)",
 528                                                         group_id);
 529                                         return PCAP_ERROR;
 530                                 }
 531
 532                                 groups[group_count++] = (unsigned short) group_id;
 533                                 dev = end_dev;
 534                         }
 535                         if (*dev != ',')
 536                                 break;
 537                         dev++;
 538                 }
 539         }
 540
 541         if (type == OTHER || *dev) {
 542                 snprintf(handle->errbuf, PCAP_ERRBUF_SIZE,
 543                                 "Can't get netfilter group(s) index from %s",
 544                                 handle->opt.device);
 545                 return PCAP_ERROR;
 546         }
 547
 548         /* if no groups, add default: 0 */
 549         if (!group_count) {
 550                 groups[0] = 0;
 551                 group_count = 1;
 552         }
 553
 554         /*
 555          * Turn a negative snapshot value (invalid), a snapshot value of
 556          * 0 (unspecified), or a value bigger than the normal maximum
 557          * value, into the maximum allowed value.
 558          *
 559          * If some application really *needs* a bigger snapshot
 560          * length, we should just increase MAXIMUM_SNAPLEN.
 561          */
 562         if (handle->snapshot <= 0 || handle->snapshot > MAXIMUM_SNAPLEN)
 563                 handle->snapshot = MAXIMUM_SNAPLEN;
 564
 565         /* Initialize some components of the pcap structure. */
 566         handle->bufsize = 128 + handle->snapshot;
 567         handle->offset = 0;
 568         handle->read_op = netfilter_read_linux;
 569         handle->inject_op = netfilter_inject_linux;
 570         handle->setfilter_op = install_bpf_program; /* no kernel filtering */
 571         handle->setdirection_op = NULL;
 572         handle->set_datalink_op = netfilter_set_datalink;
 573         handle->getnonblock_op = pcap_getnonblock_fd;
 574         handle->setnonblock_op = pcap_setnonblock_fd;
 575         handle->stats_op = netfilter_stats_linux;
 576
 577         /* Create netlink socket */
 578         handle->fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_NETFILTER);
 579         if (handle->fd < 0) {
 580                 pcap_fmt_errmsg_for_errno(handle->errbuf, PCAP_ERRBUF_SIZE,
 581                     errno, "Can't create raw socket");
 582                 return PCAP_ERROR;
 583         }
 584
 585         if (type == NFLOG) {
 586                 handle->linktype = DLT_NFLOG;
 587                 handle->dlt_list = (u_int *) malloc(sizeof(u_int) * 2);
 588                 if (handle->dlt_list != NULL) {
 589                         handle->dlt_list[0] = DLT_NFLOG;
 590                         handle->dlt_list[1] = DLT_IPV4;
 591                         handle->dlt_count = 2;
 592                 }
 593
 594         } else
 595                 handle->linktype = DLT_IPV4;
 596
 597         handle->buffer = malloc(handle->bufsize);
 598         if (!handle->buffer) {
 599                 pcap_fmt_errmsg_for_errno(handle->errbuf, PCAP_ERRBUF_SIZE,
 600                     errno, "Can't allocate dump buffer");
 601                 goto close_fail;
 602         }
 603
 604         if (type == NFLOG) {
 605                 if (nflog_send_config_cmd(handle, 0, NFULNL_CFG_CMD_PF_UNBIND, AF_INET) < 0) {
 606                         pcap_fmt_errmsg_for_errno(handle->errbuf,
 607                             PCAP_ERRBUF_SIZE, errno,
 608                             "NFULNL_CFG_CMD_PF_UNBIND");
 609                         goto close_fail;
 610                 }
 611
 612                 if (nflog_send_config_cmd(handle, 0, NFULNL_CFG_CMD_PF_BIND, AF_INET) < 0) {
 613                         pcap_fmt_errmsg_for_errno(handle->errbuf,
 614                             PCAP_ERRBUF_SIZE, errno, "NFULNL_CFG_CMD_PF_BIND");
 615                         goto close_fail;
 616                 }
 617
 618                 /* Bind socket to the nflog groups */
 619                 for (i = 0; i < group_count; i++) {
 620                         if (nflog_send_config_cmd(handle, groups[i], NFULNL_CFG_CMD_BIND, AF_UNSPEC) < 0) {
 621                                 pcap_fmt_errmsg_for_errno(handle->errbuf,
 622                                     PCAP_ERRBUF_SIZE, errno,
 623                                     "Can't listen on group group index");
 624                                 goto close_fail;
 625                         }
 626
 627                         if (nflog_send_config_mode(handle, groups[i], NFULNL_COPY_PACKET, handle->snapshot) < 0) {
 628                                 pcap_fmt_errmsg_for_errno(handle->errbuf,
 629                                     PCAP_ERRBUF_SIZE, errno,
 630                                     "NFULNL_COPY_PACKET");
 631                                 goto close_fail;
 632                         }
 633                 }
 634
 635         } else {
 636                 if (nfqueue_send_config_cmd(handle, 0, NFQNL_CFG_CMD_PF_UNBIND, AF_INET) < 0) {
 637                         pcap_fmt_errmsg_for_errno(handle->errbuf,
 638                             PCAP_ERRBUF_SIZE, errno, "NFQNL_CFG_CMD_PF_UNBIND");
 639                         goto close_fail;
 640                 }
 641
 642                 if (nfqueue_send_config_cmd(handle, 0, NFQNL_CFG_CMD_PF_BIND, AF_INET) < 0) {
 643                         pcap_fmt_errmsg_for_errno(handle->errbuf,
 644                             PCAP_ERRBUF_SIZE, errno, "NFQNL_CFG_CMD_PF_BIND");
 645                         goto close_fail;
 646                 }
 647
 648                 /* Bind socket to the nfqueue groups */
 649                 for (i = 0; i < group_count; i++) {
 650                         if (nfqueue_send_config_cmd(handle, groups[i], NFQNL_CFG_CMD_BIND, AF_UNSPEC) < 0) {
 651                                 pcap_fmt_errmsg_for_errno(handle->errbuf,
 652                                     PCAP_ERRBUF_SIZE, errno,
 653                                     "Can't listen on group group index");
 654                                 goto close_fail;
 655                         }
 656
 657                         if (nfqueue_send_config_mode(handle, groups[i], NFQNL_COPY_PACKET, handle->snapshot) < 0) {
 658                                 pcap_fmt_errmsg_for_errno(handle->errbuf,
 659                                     PCAP_ERRBUF_SIZE, errno,
 660                                     "NFQNL_COPY_PACKET");
 661                                 goto close_fail;
 662                         }
 663                 }
 664         }
 665
 666         if (handle->opt.rfmon) {
 667                 /*
 668                  * Monitor mode doesn't apply to netfilter devices.
 669                  */
 670                 pcap_cleanup_live_common(handle);
 671                 return PCAP_ERROR_RFMON_NOTSUP;
 672         }
 673
 674         if (handle->opt.buffer_size != 0) {
 675                 /*
 676                  * Set the socket buffer size to the specified value.
 677                  */
 678                 if (setsockopt(handle->fd, SOL_SOCKET, SO_RCVBUF, &handle->opt.buffer_size, sizeof(handle->opt.buffer_size)) == -1) {
 679                         pcap_fmt_errmsg_for_errno(handle->errbuf,
 680                             PCAP_ERRBUF_SIZE, errno, "SO_RCVBUF");
 681                         goto close_fail;
 682                 }
 683         }
 684
 685         handle->selectable_fd = handle->fd;
 686         return 0;
 687
 688 close_fail:
 689         pcap_cleanup_live_common(handle);
 690         return PCAP_ERROR;
 691 }
 692
 693 pcap_t *
 694 netfilter_create(const char *device, char *ebuf, int *is_ours)
 695 {
 696         const char *cp;
 697         pcap_t *p;
 698
 699         /* Does this look like an netfilter device? */
 700         cp = strrchr(device, '/');
 701         if (cp == NULL)
 702                 cp = device;
 703
 704         /* Does it begin with NFLOG_IFACE or NFQUEUE_IFACE? */
 705         if (strncmp(cp, NFLOG_IFACE, sizeof NFLOG_IFACE - 1) == 0)
 706                 cp += sizeof NFLOG_IFACE - 1;
 707         else if (strncmp(cp, NFQUEUE_IFACE, sizeof NFQUEUE_IFACE - 1) == 0)
 708                 cp += sizeof NFQUEUE_IFACE - 1;
 709         else {
 710                 /* Nope, doesn't begin with NFLOG_IFACE nor NFQUEUE_IFACE */
 711                 *is_ours = 0;
 712                 return NULL;
 713         }
 714
 715         /*
 716          * Yes - is that either the end of the name, or is it followed
 717          * by a colon?
 718          */
 719         if (*cp != ':' && *cp != '\0') {
 720                 /* Nope */
 721                 *is_ours = 0;
 722                 return NULL;
 723         }
 724
 725         /* OK, it's probably ours. */
 726         *is_ours = 1;
 727
 728         p = pcap_create_common(ebuf, sizeof (struct pcap_netfilter));
 729         if (p == NULL)
 730                 return (NULL);
 731
 732         p->activate_op = netfilter_activate;
 733         return (p);
 734 }
 735
 736 int
 737 netfilter_findalldevs(pcap_if_list_t *devlistp, char *err_str)
 738 {
 739         int sock;
 740
 741         sock = socket(AF_NETLINK, SOCK_RAW, NETLINK_NETFILTER);
 742         if (sock < 0) {
 743                 /* if netlink is not supported this is not fatal */
 744                 if (errno == EAFNOSUPPORT || errno == EPROTONOSUPPORT)
 745                         return 0;
 746                 pcap_fmt_errmsg_for_errno(err_str, PCAP_ERRBUF_SIZE,
 747                     errno, "Can't open netlink socket");
 748                 return -1;
 749         }
 750         close(sock);
 751
 752         /*
 753          * The notion of "connected" vs. "disconnected" doesn't apply.
 754          * XXX - what about "up" and "running"?
 755          */
 756         if (add_dev(devlistp, NFLOG_IFACE,
 757             PCAP_IF_CONNECTION_STATUS_NOT_APPLICABLE,
 758             "Linux netfilter log (NFLOG) interface", err_str) == NULL)
 759                 return -1;
 760         if (add_dev(devlistp, NFQUEUE_IFACE,
 761             PCAP_IF_CONNECTION_STATUS_NOT_APPLICABLE,
 762             "Linux netfilter queue (NFQUEUE) interface", err_str) == NULL)
 763                 return -1;
 764         return 0;
 765 }