On Mon, Aug 08, 2016 at 01:17:56PM +0200, Carlos Falgueras García wrote:
> This patch implements the function 'bool nftnl_rule_cmp(const struct
> nftnl_rule *r, const struct nftnl_rule *r2)' for rule comparison.
> 
> Expressions within rules need to be compared, so also has been created the
> function 'nftnl_expr_cmp' which calls new field within
> 'nfntl_expr_<expression>': a function pointer to a comparator. The
> expressions that can be compared with memcmp have this new field set to
> NULL, otherwise they have implemented a comparator.
> 
> Signed-off-by: Carlos Falgueras García <carlo...@riseup.net>
> ---
>  include/data_reg.h      |  3 +++
>  include/expr_ops.h      |  1 +
>  include/libnftnl/expr.h |  2 ++
>  include/libnftnl/rule.h |  2 ++
>  src/expr.c              | 14 ++++++++++++++
>  src/expr/data_reg.c     | 16 ++++++++++++++++
>  src/expr/dynset.c       | 18 ++++++++++++++++++
>  src/expr/immediate.c    | 19 +++++++++++++++++++
>  src/expr/log.c          | 17 +++++++++++++++++
>  src/expr/lookup.c       | 16 ++++++++++++++++
>  src/expr/match.c        | 15 +++++++++++++++
>  src/expr/target.c       | 15 +++++++++++++++
>  src/libnftnl.map        |  5 +++++
>  src/rule.c              | 30 ++++++++++++++++++++++++++++++
>  14 files changed, 173 insertions(+)
> 
> diff --git a/include/data_reg.h b/include/data_reg.h
> index e749b5b..3fec7cd 100644
> --- a/include/data_reg.h
> +++ b/include/data_reg.h
> @@ -3,6 +3,7 @@
>  
>  #include <linux/netfilter/nf_tables.h>
>  #include <stdint.h>
> +#include <stdbool.h>
>  #include <unistd.h>
>  
>  enum {
> @@ -27,6 +28,8 @@ int nftnl_data_reg_snprintf(char *buf, size_t size,
>                           const union nftnl_data_reg *reg,
>                           uint32_t output_format, uint32_t flags,
>                           int reg_type);
> +bool nftnl_data_reg_cmp(const union nftnl_data_reg *r1,
> +                     const union nftnl_data_reg *r2, int reg_type);
>  struct nlattr;
>  
>  int nftnl_parse_data(union nftnl_data_reg *data, struct nlattr *attr, int 
> *type);
> diff --git a/include/expr_ops.h b/include/expr_ops.h
> index 3c0cb18..a334732 100644
> --- a/include/expr_ops.h
> +++ b/include/expr_ops.h
> @@ -13,6 +13,7 @@ struct expr_ops {
>       uint32_t alloc_len;
>       int     max_attr;
>       void    (*free)(const struct nftnl_expr *e);
> +     bool    (*cmp)(const struct nftnl_expr *e1, const struct nftnl_expr 
> *e2);
>       int     (*set)(struct nftnl_expr *e, uint16_t type, const void *data, 
> uint32_t data_len);
>       const void *(*get)(const struct nftnl_expr *e, uint16_t type, uint32_t 
> *data_len);
>       int     (*parse)(struct nftnl_expr *e, struct nlattr *attr);
> diff --git a/include/libnftnl/expr.h b/include/libnftnl/expr.h
> index 17f60bd..8ae6f57 100644
> --- a/include/libnftnl/expr.h
> +++ b/include/libnftnl/expr.h
> @@ -36,6 +36,8 @@ uint32_t nftnl_expr_get_u32(const struct nftnl_expr *expr, 
> uint16_t type);
>  uint64_t nftnl_expr_get_u64(const struct nftnl_expr *expr, uint16_t type);
>  const char *nftnl_expr_get_str(const struct nftnl_expr *expr, uint16_t type);
>  
> +bool nftnl_expr_cmp(const struct nftnl_expr *e1, const struct nftnl_expr 
> *e2);
> +
>  int nftnl_expr_snprintf(char *buf, size_t buflen, const struct nftnl_expr 
> *expr, uint32_t type, uint32_t flags);
>  
>  enum {
> diff --git a/include/libnftnl/rule.h b/include/libnftnl/rule.h
> index 2776a77..969cb4e 100644
> --- a/include/libnftnl/rule.h
> +++ b/include/libnftnl/rule.h
> @@ -50,6 +50,8 @@ uint64_t nftnl_rule_get_u64(const struct nftnl_rule *r, 
> uint16_t attr);
>  
>  void nftnl_rule_add_expr(struct nftnl_rule *r, struct nftnl_expr *expr);
>  
> +bool nftnl_rule_cmp(const struct nftnl_rule *r1, const struct nftnl_rule 
> *r2);
> +
>  struct nlmsghdr;
>  
>  void nftnl_rule_nlmsg_build_payload(struct nlmsghdr *nlh, struct nftnl_rule 
> *t);
> diff --git a/src/expr.c b/src/expr.c
> index e5c1dd3..7f32055 100644
> --- a/src/expr.c
> +++ b/src/expr.c
> @@ -203,6 +203,20 @@ const char *nftnl_expr_get_str(const struct nftnl_expr 
> *expr, uint16_t type)
>  }
>  EXPORT_SYMBOL_ALIAS(nftnl_expr_get_str, nft_rule_expr_get_str);
>  
> +bool nftnl_expr_cmp(const struct nftnl_expr *e1, const struct nftnl_expr *e2)
> +{
> +     if (e1->flags != e2->flags)
> +             return false;
> +
> +     if (strcmp(e1->ops->name, e2->ops->name))
> +             return false;
> +     if (e1->ops->cmp)
> +             return e1->ops->cmp(e1, e2);
> +     else
> +             return !memcmp(e1->data, e2->data, e1->ops->alloc_len);
> +}
> +EXPORT_SYMBOL(nftnl_expr_cmp);
> +
>  void
>  nftnl_expr_build_payload(struct nlmsghdr *nlh, struct nftnl_expr *expr)
>  {
> diff --git a/src/expr/data_reg.c b/src/expr/data_reg.c
> index 688823b..a954e95 100644
> --- a/src/expr/data_reg.c
> +++ b/src/expr/data_reg.c
> @@ -379,6 +379,22 @@ int nftnl_data_reg_snprintf(char *buf, size_t size,
>       return -1;
>  }
>  
> +bool nftnl_data_reg_cmp(const union nftnl_data_reg *r1,
> +                     const union nftnl_data_reg *r2, int reg_type)
> +{
> +     switch (reg_type) {
> +     case DATA_VALUE:
> +             return  r1->len == r2->len &&
> +                     !memcmp(r1->val, r2->val, r1->len);
> +     case DATA_VERDICT:
> +     case DATA_CHAIN:
> +             return  r1->verdict == r2->verdict &&
> +                     !strcmp(r1->chain, r2->chain);
> +     default:
> +             return false;
> +     }
> +}
> +
>  static int nftnl_data_parse_cb(const struct nlattr *attr, void *data)
>  {
>       const struct nlattr **tb = data;
> diff --git a/src/expr/dynset.c b/src/expr/dynset.c
> index 0eaa409..fa8b8d5 100644
> --- a/src/expr/dynset.c
> +++ b/src/expr/dynset.c
> @@ -370,6 +370,23 @@ static void nftnl_expr_dynset_free(const struct 
> nftnl_expr *e)
>       xfree(dynset->set_name);
>  }
>  
> +static bool nftnl_expr_dynset_cmp(const struct nftnl_expr *e1,
> +                               const struct nftnl_expr *e2)
> +{
> +     struct nftnl_expr_dynset *d1, *d2;
> +
> +     d1 = nftnl_expr_data(e1);
> +     d2 = nftnl_expr_data(e2);

Probably this?

        struct nftnl_expr_dynset *d1 = nftnl_expr_data(e1);
        struct nftnl_expr_dynset *d2 = nftnl_expr_data(e2);

> +     d1 = nftnl_expr_data(e1);
> +     d2 = nftnl_expr_data(e2);
> +
> +     return  d1->sreg_key  == d2->sreg_key       &&
> +             d1->sreg_data == d2->sreg_data      &&
> +             d1->op        == d2->op             &&
> +             d1->timeout   == d2->timeout        &&
> +             nftnl_expr_cmp(d1->expr, d2->expr)  &&
> +             !strcmp(d1->set_name, d2->set_name) &&
> +             d1->set_id    == d2->set_id;
> +}
> +
>  struct expr_ops expr_ops_dynset = {
>       .name           = "dynset",
>       .alloc_len      = sizeof(struct nftnl_expr_dynset),
> @@ -382,4 +399,5 @@ struct expr_ops expr_ops_dynset = {
>       .snprintf       = nftnl_expr_dynset_snprintf,
>       .xml_parse      = nftnl_expr_dynset_xml_parse,
>       .json_parse     = nftnl_expr_dynset_json_parse,
> +     .cmp            = nftnl_expr_dynset_cmp,
>  };
> diff --git a/src/expr/immediate.c b/src/expr/immediate.c
> index 22ec864..a41e9f7 100644
> --- a/src/expr/immediate.c
> +++ b/src/expr/immediate.c
> @@ -320,6 +320,24 @@ static void nftnl_expr_immediate_free(const struct 
> nftnl_expr *e)
>               nftnl_free_verdict(&imm->data);
>  }
>  
> +static bool nftnl_expr_immediate_cmp(const struct nftnl_expr *e1,
> +                                  const struct nftnl_expr *e2)
> +{
> +     struct nftnl_expr_immediate *i1, *i2;
> +     int reg_type;
> +
> +     i1 = nftnl_expr_data(e1);
> +     i2 = nftnl_expr_data(e2);
> +
> +     if (e1->flags & (1 << NFTNL_EXPR_IMM_VERDICT))
> +             reg_type = DATA_VERDICT;
> +     else
> +             reg_type = DATA_VALUE;
> +
> +     return  i1->dreg == i2->dreg &&
> +             nftnl_data_reg_cmp(&i1->data, &i2->data, reg_type);
> +}
> +
>  struct expr_ops expr_ops_immediate = {
>       .name           = "immediate",
>       .alloc_len      = sizeof(struct nftnl_expr_immediate),
> @@ -332,4 +350,5 @@ struct expr_ops expr_ops_immediate = {
>       .snprintf       = nftnl_expr_immediate_snprintf,
>       .xml_parse      = nftnl_expr_immediate_xml_parse,
>       .json_parse     = nftnl_expr_immediate_json_parse,
> +     .cmp            = nftnl_expr_immediate_cmp,

Please, place this after free() as in the structure definition for
consistency.

>  };
> diff --git a/src/expr/log.c b/src/expr/log.c
> index b9b3951..1c055e1 100644
> --- a/src/expr/log.c
> +++ b/src/expr/log.c
> @@ -336,6 +336,22 @@ static void nftnl_expr_log_free(const struct nftnl_expr 
> *e)
>       xfree(log->prefix);
>  }
>  
> +static bool nftnl_expr_log_cmp(const struct nftnl_expr *e1,
> +                                  const struct nftnl_expr *e2)
> +{
> +     struct nftnl_expr_log *l1, *l2;
> +
> +     l1 = nftnl_expr_data(e1);
> +     l2 = nftnl_expr_data(e2);
> +
> +     return  l1->snaplen     == l2->snaplen    &&
> +             l1->group       == l2->group      &&
> +             l1->qthreshold  == l2->qthreshold &&
> +             l1->level       == l2->level      &&
> +             l1->flags       == l2->flags      &&
> +             !strcmp(l1->prefix, l2->prefix);
> +}
> +
>  struct expr_ops expr_ops_log = {
>       .name           = "log",
>       .alloc_len      = sizeof(struct nftnl_expr_log),
> @@ -348,4 +364,5 @@ struct expr_ops expr_ops_log = {
>       .snprintf       = nftnl_expr_log_snprintf,
>       .xml_parse      = nftnl_expr_log_xml_parse,
>       .json_parse     = nftnl_expr_log_json_parse,
> +     .cmp            = nftnl_expr_log_cmp,
>  };
> diff --git a/src/expr/lookup.c b/src/expr/lookup.c
> index 97478c2..57612d1 100644
> --- a/src/expr/lookup.c
> +++ b/src/expr/lookup.c
> @@ -296,6 +296,21 @@ static void nftnl_expr_lookup_free(const struct 
> nftnl_expr *e)
>       xfree(lookup->set_name);
>  }
>  
> +static bool nftnl_expr_lookup_cmp(const struct nftnl_expr *e1,
> +                               const struct nftnl_expr *e2)
> +{
> +     struct nftnl_expr_lookup *l1, *l2;
> +
> +     l1 = nftnl_expr_data(e1);
> +     l2 = nftnl_expr_data(e2);
> +
> +     return  l1->sreg   == l2->sreg              &&
> +             l1->dreg   == l2->dreg              &&
> +             !strcmp(l1->set_name, l2->set_name) &&
> +             l1->set_id == l2->set_id            &&
> +             l1->flags  == l2->flags;
> +}
> +
>  struct expr_ops expr_ops_lookup = {
>       .name           = "lookup",
>       .alloc_len      = sizeof(struct nftnl_expr_lookup),
> @@ -308,4 +323,5 @@ struct expr_ops expr_ops_lookup = {
>       .snprintf       = nftnl_expr_lookup_snprintf,
>       .xml_parse      = nftnl_expr_lookup_xml_parse,
>       .json_parse     = nftnl_expr_lookup_json_parse,
> +     .cmp            = nftnl_expr_lookup_cmp,
>  };
> diff --git a/src/expr/match.c b/src/expr/match.c
> index 3342e2c..caefba1 100644
> --- a/src/expr/match.c
> +++ b/src/expr/match.c
> @@ -240,6 +240,20 @@ static void nftnl_expr_match_free(const struct 
> nftnl_expr *e)
>       xfree(match->data);
>  }
>  
> +static bool nftnl_expr_match_cmp(const struct nftnl_expr *e1,
> +                              const struct nftnl_expr *e2)
> +{
> +     struct nftnl_expr_match *m1, *m2;
> +
> +     m1 = nftnl_expr_data(e1);
> +     m2 = nftnl_expr_data(e2);
> +
> +     return  !strcmp(m1->name, m2->name)  &&
> +             m1->rev == m2->rev           &&
> +             m1->data_len == m2->data_len &&
> +             !memcmp(m1->data, m2->data, m1->data_len);
> +}
> +
>  struct expr_ops expr_ops_match = {
>       .name           = "match",
>       .alloc_len      = sizeof(struct nftnl_expr_match),
> @@ -252,4 +266,5 @@ struct expr_ops expr_ops_match = {
>       .snprintf       = nftnl_expr_match_snprintf,
>       .xml_parse      = nftnl_expr_match_xml_parse,
>       .json_parse     = nftnl_expr_match_json_parse,
> +     .cmp            = nftnl_expr_match_cmp,
>  };
> diff --git a/src/expr/target.c b/src/expr/target.c
> index d4c0091..0794464 100644
> --- a/src/expr/target.c
> +++ b/src/expr/target.c
> @@ -241,6 +241,20 @@ static void nftnl_expr_target_free(const struct 
> nftnl_expr *e)
>       xfree(target->data);
>  }
>  
> +static bool nftnl_expr_target_cmp(const struct nftnl_expr *e1,
> +                               const struct nftnl_expr *e2)
> +{
> +     struct nftnl_expr_target *t1, *t2;
> +
> +     t1 = nftnl_expr_data(e1);
> +     t2 = nftnl_expr_data(e2);
> +
> +     return  !strcmp(t1->name, t2->name)  &&
> +             t1->rev == t2->rev           &&
> +             t1->data_len == t2->data_len &&
> +             !memcmp(t1->data, t2->data, t1->data_len);
> +}
> +
>  struct expr_ops expr_ops_target = {
>       .name           = "target",
>       .alloc_len      = sizeof(struct nftnl_expr_target),
> @@ -253,4 +267,5 @@ struct expr_ops expr_ops_target = {
>       .snprintf       = nftnl_expr_target_snprintf,
>       .xml_parse      = nftnl_expr_target_xml_parse,
>       .json_parse     = nftnl_expr_target_json_parse,
> +     .cmp            = nftnl_expr_target_cmp,
>  };
> diff --git a/src/libnftnl.map b/src/libnftnl.map
> index c38e081..748a2c6 100644
> --- a/src/libnftnl.map
> +++ b/src/libnftnl.map
> @@ -528,3 +528,8 @@ LIBNFTNL_4.1 {
>       nftnl_udata_next;
>       nftnl_udata_parse;
>  } LIBNFTNL_4;
> +
> +LIBNFTNL_4.2 {
> +     nftnl_rule_cmp;
> +     nftnl_expr_cmp;
> +} LIBNFTNL_4;

This should be LIBNFTNL_4.1 instead.

> diff --git a/src/rule.c b/src/rule.c
> index 0cfddf2..27e753b 100644
> --- a/src/rule.c
> +++ b/src/rule.c
> @@ -331,6 +331,36 @@ void nftnl_rule_add_expr(struct nftnl_rule *r, struct 
> nftnl_expr *expr)
>  }
>  EXPORT_SYMBOL_ALIAS(nftnl_rule_add_expr, nft_rule_add_expr);
>  
> +bool nftnl_rule_cmp(const struct nftnl_rule *r1, const struct nftnl_rule *r2)
> +{
> +     unsigned int eq = 1;
> +     struct nftnl_expr *e1, *e2;
> +     struct nftnl_expr_iter it1, it2;

Just a comestic comment, we usually prefer this:

        struct nftnl_expr_iter it1, it2;
        struct nftnl_expr *e1, *e2;
        unsigned int eq = 1;

So larger lines in variable declarations go first, if there are no
dependencies of course.

> +     if (r1->flags & r1->flags & (1 << NFTNL_RULE_TABLE))
> +             eq &= !strcmp(r1->table, r2->table);
> +     if (r1->flags & r1->flags & (1 << NFTNL_RULE_CHAIN))
> +             eq &= !strcmp(r1->chain, r2->chain);
> +     if (r1->flags & r1->flags & (1 << NFTNL_RULE_COMPAT_FLAGS))
> +             eq &= r1->compat.flags == r2->compat.flags;
> +     if (r1->flags & r1->flags & (1 << NFTNL_RULE_COMPAT_PROTO))
> +             eq &= r1->compat.proto == r2->compat.proto;
> +
> +     nftnl_expr_iter_init(r1, &it1);
> +     nftnl_expr_iter_init(r2, &it2);
> +     e1 = nftnl_expr_iter_next(&it1);
> +     e2 = nftnl_expr_iter_next(&it2);
> +     while (eq && e1 && e2) {
> +             eq = nftnl_expr_cmp(e1, e2);
> +
> +             e1 = nftnl_expr_iter_next(&it1);
> +             e2 = nftnl_expr_iter_next(&it2);
> +     }

Probably you can use:

        do {
                ...
        while (...);

to simplify this?
--
To unsubscribe from this list: send the line "unsubscribe netfilter-devel" in
the body of a message to majord...@vger.kernel.org
More majordomo info at  http://vger.kernel.org/majordomo-info.html

Reply via email to