cregit-Linux how code gets into the kernel

Release 4.11 net/netfilter/nft_counter.c

Directory: net/netfilter
/*
 * Copyright (c) 2008-2009 Patrick McHardy <kaber@trash.net>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 *
 * Development of this code funded by Astaro AG (http://www.astaro.com/)
 */

#include <linux/kernel.h>
#include <linux/init.h>
#include <linux/module.h>
#include <linux/seqlock.h>
#include <linux/netlink.h>
#include <linux/netfilter.h>
#include <linux/netfilter/nf_tables.h>
#include <net/netfilter/nf_tables.h>


struct nft_counter {
	
s64		bytes;
	
s64		packets;
};


struct nft_counter_percpu_priv {
	
struct nft_counter __percpu *counter;
};

static DEFINE_PER_CPU(seqcount_t, nft_counter_seq);


static inline void nft_counter_do_eval(struct nft_counter_percpu_priv *priv, struct nft_regs *regs, const struct nft_pktinfo *pkt) { struct nft_counter *this_cpu; seqcount_t *myseq; local_bh_disable(); this_cpu = this_cpu_ptr(priv->counter); myseq = this_cpu_ptr(&nft_counter_seq); write_seqcount_begin(myseq); this_cpu->bytes += pkt->skb->len; this_cpu->packets++; write_seqcount_end(myseq); local_bh_enable(); }

Contributors

PersonTokensPropCommitsCommitProp
Patrick McHardy4151.25%240.00%
Pablo Neira Ayuso3948.75%360.00%
Total80100.00%5100.00%


static inline void nft_counter_obj_eval(struct nft_object *obj, struct nft_regs *regs, const struct nft_pktinfo *pkt) { struct nft_counter_percpu_priv *priv = nft_obj_data(obj); nft_counter_do_eval(priv, regs, pkt); }

Contributors

PersonTokensPropCommitsCommitProp
Pablo Neira Ayuso42100.00%1100.00%
Total42100.00%1100.00%


static int nft_counter_do_init(const struct nlattr * const tb[], struct nft_counter_percpu_priv *priv) { struct nft_counter __percpu *cpu_stats; struct nft_counter *this_cpu; cpu_stats = alloc_percpu(struct nft_counter); if (cpu_stats == NULL) return -ENOMEM; preempt_disable(); this_cpu = this_cpu_ptr(cpu_stats); if (tb[NFTA_COUNTER_PACKETS]) { this_cpu->packets = be64_to_cpu(nla_get_be64(tb[NFTA_COUNTER_PACKETS])); } if (tb[NFTA_COUNTER_BYTES]) { this_cpu->bytes = be64_to_cpu(nla_get_be64(tb[NFTA_COUNTER_BYTES])); } preempt_enable(); priv->counter = cpu_stats; return 0; }

Contributors

PersonTokensPropCommitsCommitProp
Pablo Neira Ayuso118100.00%2100.00%
Total118100.00%2100.00%


static int nft_counter_obj_init(const struct nlattr * const tb[], struct nft_object *obj) { struct nft_counter_percpu_priv *priv = nft_obj_data(obj); return nft_counter_do_init(tb, priv); }

Contributors

PersonTokensPropCommitsCommitProp
Pablo Neira Ayuso37100.00%1100.00%
Total37100.00%1100.00%


static void nft_counter_do_destroy(struct nft_counter_percpu_priv *priv) { free_percpu(priv->counter); }

Contributors

PersonTokensPropCommitsCommitProp
Pablo Neira Ayuso18100.00%1100.00%
Total18100.00%1100.00%


static void nft_counter_obj_destroy(struct nft_object *obj) { struct nft_counter_percpu_priv *priv = nft_obj_data(obj); nft_counter_do_destroy(priv); }

Contributors

PersonTokensPropCommitsCommitProp
Pablo Neira Ayuso26100.00%1100.00%
Total26100.00%1100.00%


static void nft_counter_reset(struct nft_counter_percpu_priv __percpu *priv, struct nft_counter *total) { struct nft_counter *this_cpu; local_bh_disable(); this_cpu = this_cpu_ptr(priv->counter); this_cpu->packets -= total->packets; this_cpu->bytes -= total->bytes; local_bh_enable(); }

Contributors

PersonTokensPropCommitsCommitProp
Pablo Neira Ayuso3871.70%583.33%
Patrick McHardy1528.30%116.67%
Total53100.00%6100.00%


static void nft_counter_fetch(struct nft_counter_percpu_priv *priv, struct nft_counter *total) { struct nft_counter *this_cpu; const seqcount_t *myseq; u64 bytes, packets; unsigned int seq; int cpu; memset(total, 0, sizeof(*total)); for_each_possible_cpu(cpu) { myseq = per_cpu_ptr(&nft_counter_seq, cpu); this_cpu = per_cpu_ptr(priv->counter, cpu); do { seq = read_seqcount_begin(myseq); bytes = this_cpu->bytes; packets = this_cpu->packets; } while (read_seqcount_retry(myseq, seq)); total->bytes += bytes; total->packets += packets; } }

Contributors

PersonTokensPropCommitsCommitProp
Pablo Neira Ayuso122100.00%2100.00%
Total122100.00%2100.00%


static int nft_counter_do_dump(struct sk_buff *skb, struct nft_counter_percpu_priv *priv, bool reset) { struct nft_counter total; nft_counter_fetch(priv, &total); if (nla_put_be64(skb, NFTA_COUNTER_BYTES, cpu_to_be64(total.bytes), NFTA_COUNTER_PAD) || nla_put_be64(skb, NFTA_COUNTER_PACKETS, cpu_to_be64(total.packets), NFTA_COUNTER_PAD)) goto nla_put_failure; if (reset) nft_counter_reset(priv, &total); return 0; nla_put_failure: return -1; }

Contributors

PersonTokensPropCommitsCommitProp
Pablo Neira Ayuso89100.00%3100.00%
Total89100.00%3100.00%


static int nft_counter_obj_dump(struct sk_buff *skb, struct nft_object *obj, bool reset) { struct nft_counter_percpu_priv *priv = nft_obj_data(obj); return nft_counter_do_dump(skb, priv, reset); }

Contributors

PersonTokensPropCommitsCommitProp
Pablo Neira Ayuso39100.00%2100.00%
Total39100.00%2100.00%

static const struct nla_policy nft_counter_policy[NFTA_COUNTER_MAX + 1] = { [NFTA_COUNTER_PACKETS] = { .type = NLA_U64 }, [NFTA_COUNTER_BYTES] = { .type = NLA_U64 }, }; static struct nft_object_type nft_counter_obj __read_mostly = { .type = NFT_OBJECT_COUNTER, .size = sizeof(struct nft_counter_percpu_priv), .maxattr = NFTA_COUNTER_MAX, .policy = nft_counter_policy, .eval = nft_counter_obj_eval, .init = nft_counter_obj_init, .destroy = nft_counter_obj_destroy, .dump = nft_counter_obj_dump, .owner = THIS_MODULE, };
static void nft_counter_eval(const struct nft_expr *expr, struct nft_regs *regs, const struct nft_pktinfo *pkt) { struct nft_counter_percpu_priv *priv = nft_expr_priv(expr); nft_counter_do_eval(priv, regs, pkt); }

Contributors

PersonTokensPropCommitsCommitProp
Pablo Neira Ayuso4197.62%266.67%
Patrick McHardy12.38%133.33%
Total42100.00%3100.00%


static int nft_counter_dump(struct sk_buff *skb, const struct nft_expr *expr) { struct nft_counter_percpu_priv *priv = nft_expr_priv(expr); return nft_counter_do_dump(skb, priv, false); }

Contributors

PersonTokensPropCommitsCommitProp
Pablo Neira Ayuso3183.78%375.00%
Patrick McHardy616.22%125.00%
Total37100.00%4100.00%


static int nft_counter_init(const struct nft_ctx *ctx, const struct nft_expr *expr, const struct nlattr * const tb[]) { struct nft_counter_percpu_priv *priv = nft_expr_priv(expr); return nft_counter_do_init(tb, priv); }

Contributors

PersonTokensPropCommitsCommitProp
Patrick McHardy3988.64%133.33%
Pablo Neira Ayuso511.36%266.67%
Total44100.00%3100.00%


static void nft_counter_destroy(const struct nft_ctx *ctx, const struct nft_expr *expr) { struct nft_counter_percpu_priv *priv = nft_expr_priv(expr); nft_counter_do_destroy(priv); }

Contributors

PersonTokensPropCommitsCommitProp
Pablo Neira Ayuso33100.00%2100.00%
Total33100.00%2100.00%


static int nft_counter_clone(struct nft_expr *dst, const struct nft_expr *src) { struct nft_counter_percpu_priv *priv = nft_expr_priv(src); struct nft_counter_percpu_priv *priv_clone = nft_expr_priv(dst); struct nft_counter __percpu *cpu_stats; struct nft_counter *this_cpu; struct nft_counter total; nft_counter_fetch(priv, &total); cpu_stats = alloc_percpu_gfp(struct nft_counter, GFP_ATOMIC); if (cpu_stats == NULL) return -ENOMEM; preempt_disable(); this_cpu = this_cpu_ptr(cpu_stats); this_cpu->packets = total.packets; this_cpu->bytes = total.bytes; preempt_enable(); priv_clone->counter = cpu_stats; return 0; }

Contributors

PersonTokensPropCommitsCommitProp
Pablo Neira Ayuso11799.15%266.67%
Anton Protopopov10.85%133.33%
Total118100.00%3100.00%

static struct nft_expr_type nft_counter_type; static const struct nft_expr_ops nft_counter_ops = { .type = &nft_counter_type, .size = NFT_EXPR_SIZE(sizeof(struct nft_counter_percpu_priv)), .eval = nft_counter_eval, .init = nft_counter_init, .destroy = nft_counter_destroy, .dump = nft_counter_dump, .clone = nft_counter_clone, }; static struct nft_expr_type nft_counter_type __read_mostly = { .name = "counter", .ops = &nft_counter_ops, .policy = nft_counter_policy, .maxattr = NFTA_COUNTER_MAX, .flags = NFT_EXPR_STATEFUL, .owner = THIS_MODULE, };
static int __init nft_counter_module_init(void) { int cpu, err; for_each_possible_cpu(cpu) seqcount_init(per_cpu_ptr(&nft_counter_seq, cpu)); err = nft_register_obj(&nft_counter_obj); if (err < 0) return err; err = nft_register_expr(&nft_counter_type); if (err < 0) goto err1; return 0; err1: nft_unregister_obj(&nft_counter_obj); return err; }

Contributors

PersonTokensPropCommitsCommitProp
Pablo Neira Ayuso6179.22%250.00%
Patrick McHardy1620.78%250.00%
Total77100.00%4100.00%


static void __exit nft_counter_module_exit(void) { nft_unregister_expr(&nft_counter_type); nft_unregister_obj(&nft_counter_obj); }

Contributors

PersonTokensPropCommitsCommitProp
Patrick McHardy1571.43%266.67%
Pablo Neira Ayuso628.57%133.33%
Total21100.00%3100.00%

module_init(nft_counter_module_init); module_exit(nft_counter_module_exit); MODULE_LICENSE("GPL"); MODULE_AUTHOR("Patrick McHardy <kaber@trash.net>"); MODULE_ALIAS_NFT_EXPR("counter"); MODULE_ALIAS_NFT_OBJ(NFT_OBJECT_COUNTER);

Overall Contributors

PersonTokensPropCommitsCommitProp
Pablo Neira Ayuso99178.15%550.00%
Patrick McHardy27621.77%440.00%
Anton Protopopov10.08%110.00%
Total1268100.00%10100.00%
Directory: net/netfilter
Information contained on this website is for historical information purposes only and does not indicate or represent copyright ownership.
Created with cregit.