From 3e40a89d43db8cf6c5947baa7221e57e2b9cf351 Mon Sep 17 00:00:00 2001 From: Arturo Borrero Date: Thu, 26 Sep 2013 00:13:02 +0200 Subject: set_elem: xml: avoid code duplication This patch removes 3x code duplication in the XML snprintf function. Signed-off-by: Arturo Borrero Gonzalez Signed-off-by: Pablo Neira Ayuso --- src/set_elem.c | 32 +++++++++----------------------- 1 file changed, 9 insertions(+), 23 deletions(-) diff --git a/src/set_elem.c b/src/set_elem.c index a9e0751..885893b 100644 --- a/src/set_elem.c +++ b/src/set_elem.c @@ -519,7 +519,7 @@ static int nft_set_elem_snprintf_default(char *buf, size_t size, static int nft_set_elem_snprintf_xml(char *buf, size_t size, struct nft_set_elem *e, uint32_t flags) { - int ret, len = size, offset = 0; + int ret, len = size, offset = 0, type = DATA_NONE; ret = snprintf(buf, size, "" "%u", @@ -533,33 +533,19 @@ static int nft_set_elem_snprintf_xml(char *buf, size_t size, ret = snprintf(buf+offset, size, ""); SNPRINTF_BUFFER_SIZE(ret, size, len, offset); - if (e->flags & (1 << NFT_SET_ELEM_ATTR_DATA)) { - ret = snprintf(buf+offset, size, ""); - SNPRINTF_BUFFER_SIZE(ret, size, len, offset); - - ret = nft_data_reg_snprintf(buf+offset, len, &e->data, - NFT_RULE_O_XML, flags, DATA_VALUE); - SNPRINTF_BUFFER_SIZE(ret, size, len, offset); - - ret = snprintf(buf+offset, size, ""); - SNPRINTF_BUFFER_SIZE(ret, size, len, offset); - } else if (e->flags & (1 << NFT_SET_ELEM_ATTR_VERDICT)) { - ret = snprintf(buf+offset, size, ""); - SNPRINTF_BUFFER_SIZE(ret, size, len, offset); - - ret = nft_data_reg_snprintf(buf+offset, len, &e->data, - NFT_RULE_O_XML, flags, - DATA_VERDICT); - SNPRINTF_BUFFER_SIZE(ret, size, len, offset); + if (e->flags & (1 << NFT_SET_ELEM_ATTR_DATA)) + type = DATA_VALUE; + else if (e->flags & (1 << NFT_SET_ELEM_ATTR_CHAIN)) + type = DATA_CHAIN; + else if (e->flags & (1 << NFT_SET_ELEM_ATTR_VERDICT)) + type = DATA_VERDICT; - ret = snprintf(buf+offset, size, ""); - SNPRINTF_BUFFER_SIZE(ret, size, len, offset); - } else if (e->flags & (1 << NFT_SET_ELEM_ATTR_CHAIN)) { + if (type != DATA_NONE) { ret = snprintf(buf+offset, size, ""); SNPRINTF_BUFFER_SIZE(ret, size, len, offset); ret = nft_data_reg_snprintf(buf+offset, len, &e->data, - NFT_RULE_O_XML, flags, DATA_CHAIN); + NFT_RULE_O_XML, flags, type); SNPRINTF_BUFFER_SIZE(ret, size, len, offset); ret = snprintf(buf+offset, size, ""); -- cgit v1.2.3