Logo Search packages:      
Sourcecode: libnl2 version File versions  Download package

queue_msg.c

/*
 * lib/netfilter/queue_msg.c  Netfilter Queue Messages
 *
 *    This library is free software; you can redistribute it and/or
 *    modify it under the terms of the GNU Lesser General Public
 *    License as published by the Free Software Foundation version 2.1
 *    of the License.
 *
 * Copyright (c) 2007, 2008 Patrick McHardy <kaber@trash.net>
 */

/**
 * @ingroup nfnl
 * @defgroup queue Queue
 * @brief
 * @{
 */

#include <sys/types.h>
#include <linux/netfilter/nfnetlink_queue.h>

#include <netlink-local.h>
#include <netlink/attr.h>
#include <netlink/netfilter/nfnl.h>
#include <netlink/netfilter/queue_msg.h>

static struct nl_cache_ops nfnl_queue_msg_ops;

#if __BYTE_ORDER == __BIG_ENDIAN
static uint64_t ntohll(uint64_t x)
{
      return x;
}
#elif __BYTE_ORDER == __LITTLE_ENDIAN
static uint64_t ntohll(uint64_t x)
{
      return __bswap_64(x);
}
#endif

static struct nla_policy queue_policy[NFQA_MAX+1] = {
      [NFQA_PACKET_HDR]       = {
            .minlen     = sizeof(struct nfqnl_msg_packet_hdr),
      },
      [NFQA_VERDICT_HDR]            = {
            .minlen     = sizeof(struct nfqnl_msg_verdict_hdr),
      },
      [NFQA_MARK]             = { .type = NLA_U32 },
      [NFQA_TIMESTAMP]        = {
            .minlen = sizeof(struct nfqnl_msg_packet_timestamp),
      },
      [NFQA_IFINDEX_INDEV]          = { .type = NLA_U32 },
      [NFQA_IFINDEX_OUTDEV]         = { .type = NLA_U32 },
      [NFQA_IFINDEX_PHYSINDEV]      = { .type = NLA_U32 },
      [NFQA_IFINDEX_PHYSOUTDEV]     = { .type = NLA_U32 },
      [NFQA_HWADDR]                 = {
            .minlen     = sizeof(struct nfqnl_msg_packet_hw),
      },
};

int nfnlmsg_queue_msg_parse(struct nlmsghdr *nlh,
                      struct nfnl_queue_msg **result)
{
      struct nfnl_queue_msg *msg;
      struct nlattr *tb[NFQA_MAX+1];
      struct nlattr *attr;
      int err;

      msg = nfnl_queue_msg_alloc();
      if (!msg)
            return -NLE_NOMEM;

      msg->ce_msgtype = nlh->nlmsg_type;

      err = nlmsg_parse(nlh, sizeof(struct nfgenmsg), tb, NFQA_MAX,
                    queue_policy);
      if (err < 0)
            goto errout;

      nfnl_queue_msg_set_group(msg, nfnlmsg_res_id(nlh));
      nfnl_queue_msg_set_family(msg, nfnlmsg_family(nlh));

      attr = tb[NFQA_PACKET_HDR];
      if (attr) {
            struct nfqnl_msg_packet_hdr *hdr = nla_data(attr);

            nfnl_queue_msg_set_packetid(msg, ntohl(hdr->packet_id));
            if (hdr->hw_protocol)
                  nfnl_queue_msg_set_hwproto(msg, hdr->hw_protocol);
            nfnl_queue_msg_set_hook(msg, hdr->hook);
      }

      attr = tb[NFQA_MARK];
      if (attr)
            nfnl_queue_msg_set_mark(msg, ntohl(nla_get_u32(attr)));

      attr = tb[NFQA_TIMESTAMP];
      if (attr) {
            struct nfqnl_msg_packet_timestamp *timestamp = nla_data(attr);
            struct timeval tv;

            tv.tv_sec = ntohll(timestamp->sec);
            tv.tv_usec = ntohll(timestamp->usec);
            nfnl_queue_msg_set_timestamp(msg, &tv);
      }

      attr = tb[NFQA_IFINDEX_INDEV];
      if (attr)
            nfnl_queue_msg_set_indev(msg, ntohl(nla_get_u32(attr)));

      attr = tb[NFQA_IFINDEX_OUTDEV];
      if (attr)
            nfnl_queue_msg_set_outdev(msg, ntohl(nla_get_u32(attr)));

      attr = tb[NFQA_IFINDEX_PHYSINDEV];
      if (attr)
            nfnl_queue_msg_set_physindev(msg, ntohl(nla_get_u32(attr)));

      attr = tb[NFQA_IFINDEX_PHYSOUTDEV];
      if (attr)
            nfnl_queue_msg_set_physoutdev(msg, ntohl(nla_get_u32(attr)));

      attr = tb[NFQA_HWADDR];
      if (attr) {
            struct nfqnl_msg_packet_hw *hw = nla_data(attr);

            nfnl_queue_msg_set_hwaddr(msg, hw->hw_addr,
                                ntohs(hw->hw_addrlen));
      }

      attr = tb[NFQA_PAYLOAD];
      if (attr) {
            err = nfnl_queue_msg_set_payload(msg, nla_data(attr),
                                     nla_len(attr));
            if (err < 0)
                  goto errout;
      }

      *result = msg;
      return 0;

errout:
      nfnl_queue_msg_put(msg);
      return err;
}

static int queue_msg_parser(struct nl_cache_ops *ops, struct sockaddr_nl *who,
                      struct nlmsghdr *nlh, struct nl_parser_param *pp)
{
      struct nfnl_queue_msg *msg;
      int err;

      if ((err = nfnlmsg_queue_msg_parse(nlh, &msg)) < 0)
            goto errout;

      err = pp->pp_cb((struct nl_object *) msg, pp);
errout:
      nfnl_queue_msg_put(msg);
      return err;
}

/** @} */

struct nl_msg *nfnl_queue_msg_build_verdict(const struct nfnl_queue_msg *msg)
{
      struct nl_msg *nlmsg;
      struct nfqnl_msg_verdict_hdr verdict;

      nlmsg = nfnlmsg_alloc_simple(NFNL_SUBSYS_QUEUE, NFQNL_MSG_VERDICT, 0,
                             nfnl_queue_msg_get_family(msg),
                             nfnl_queue_msg_get_group(msg));
      if (nlmsg == NULL)
            return NULL;

      verdict.id = htonl(nfnl_queue_msg_get_packetid(msg));
      verdict.verdict = htonl(nfnl_queue_msg_get_verdict(msg));
      if (nla_put(nlmsg, NFQA_VERDICT_HDR, sizeof(verdict), &verdict) < 0)
            goto nla_put_failure;

      if (nfnl_queue_msg_test_mark(msg) &&
          nla_put_u32(nlmsg, NFQA_MARK,
                  ntohl(nfnl_queue_msg_get_mark(msg))) < 0)
            goto nla_put_failure;

      return nlmsg;

nla_put_failure:
      nlmsg_free(nlmsg);
      return NULL;
}

int nfnl_queue_msg_send_verdict(struct nl_sock *nlh,
                        const struct nfnl_queue_msg *msg)
{
      struct nl_msg *nlmsg;
      int err;

      nlmsg = nfnl_queue_msg_build_verdict(msg);
      if (nlmsg == NULL)
            return -NLE_NOMEM;

      err = nl_send_auto_complete(nlh, nlmsg);
      nlmsg_free(nlmsg);
      if (err < 0)
            return err;
      return wait_for_ack(nlh);
}

#define NFNLMSG_QUEUE_TYPE(type) NFNLMSG_TYPE(NFNL_SUBSYS_QUEUE, (type))
static struct nl_cache_ops nfnl_queue_msg_ops = {
      .co_name          = "netfilter/queue_msg",
      .co_hdrsize       = NFNL_HDRLEN,
      .co_msgtypes            = {
            { NFNLMSG_QUEUE_TYPE(NFQNL_MSG_PACKET), NL_ACT_NEW, "new" },
            END_OF_MSGTYPES_LIST,
      },
      .co_protocol            = NETLINK_NETFILTER,
      .co_msg_parser          = queue_msg_parser,
      .co_obj_ops       = &queue_msg_obj_ops,
};

static void __init nfnl_msg_queue_init(void)
{
      nl_cache_mngt_register(&nfnl_queue_msg_ops);
}

static void __exit nfnl_queue_msg_exit(void)
{
      nl_cache_mngt_unregister(&nfnl_queue_msg_ops);
}

/** @} */

Generated by  Doxygen 1.6.0   Back to index