From mboxrd@z Thu Jan 1 00:00:00 1970 From: Ken-ichirou MATSUZAWA Subject: [ulogd PATCH 6/13] ipfix: add function for ipfix message creation Date: Mon, 28 Apr 2014 20:49:22 +0900 Message-ID: <20140428114922.GG12523@gmail.com> References: <20140308010344.GA4415@gmail.com> <20140428113936.GA12523@gmail.com> Mime-Version: 1.0 Content-Type: text/plain; charset=us-ascii To: The netfilter developer mailinglist Return-path: Received: from mail-pb0-f47.google.com ([209.85.160.47]:50810 "EHLO mail-pb0-f47.google.com" rhost-flags-OK-OK-OK-OK) by vger.kernel.org with ESMTP id S1752830AbaD1Lt2 (ORCPT ); Mon, 28 Apr 2014 07:49:28 -0400 Received: by mail-pb0-f47.google.com with SMTP id up15so5761448pbc.20 for ; Mon, 28 Apr 2014 04:49:28 -0700 (PDT) Received: from gmail.com (softbank220009032006.bbtec.net. [220.9.32.6]) by mx.google.com with ESMTPSA id ky8sm34502899pbc.64.2014.04.28.04.49.25 for (version=TLSv1.2 cipher=RC4-SHA bits=128/128); Mon, 28 Apr 2014 04:49:26 -0700 (PDT) Content-Disposition: inline In-Reply-To: <20140428113936.GA12523@gmail.com> Sender: netfilter-devel-owner@vger.kernel.org List-ID: This function creates ipfix message, template and data part but not scope. Header sequence is kept by struct ipfix_instance, domain id is specified by config file. The returned value has no export time so caller set this and free the value after using it. Signed-off-by Ken-ichirou MATSUZAWA --- include/ulogd/ipfix_protocol.h | 8 ++++- output/ulogd_output_IPFIX.c | 71 +++++++++++++++++++++++++++++++++++++++++- 2 files changed, 77 insertions(+), 2 deletions(-) diff --git a/include/ulogd/ipfix_protocol.h b/include/ulogd/ipfix_protocol.h index 5d7e46a..266897e 100644 --- a/include/ulogd/ipfix_protocol.h +++ b/include/ulogd/ipfix_protocol.h @@ -15,7 +15,13 @@ struct ipfix_msg_hdr { u_int16_t length; u_int32_t export_time; u_int32_t seq; - u_int32_t source_id; + u_int32_t domain_id; +}; + +/* Section 3.3.2 */ +struct ipfix_set_hdr { + u_int16_t set_id; + u_int16_t length; }; /* Section 3.4.1 */ diff --git a/output/ulogd_output_IPFIX.c b/output/ulogd_output_IPFIX.c index f032b50..8246ed3 100644 --- a/output/ulogd_output_IPFIX.c +++ b/output/ulogd_output_IPFIX.c @@ -28,6 +28,7 @@ #include #include +#include #include #include #include @@ -84,7 +85,7 @@ struct sctp_sndrcvinfo { #define IPFIX_DEFAULT_TCPUDP_PORT 4739 static struct config_keyset ipfix_kset = { - .num_ces = 3, + .num_ces = 4, .ces = { { .key = "host", @@ -103,12 +104,19 @@ static struct config_keyset ipfix_kset = { .options = CONFIG_OPT_NONE, .u = { .string = "udp" }, }, + { + .key = "domain_id", + .type = CONFIG_TYPE_INT, + .options = CONFIG_OPT_NONE, + .u.value = 0, + }, }, }; #define host_ce(x) (x->ces[0]) #define port_ce(x) (x->ces[1]) #define proto_ce(x) (x->ces[2]) +#define domain_ce(x) (x->ces[3]) struct ipfix_template { struct ipfix_templ_rec_hdr hdr; @@ -130,6 +138,7 @@ struct ipfix_instance { struct llist_head template_list; struct nfct_bitmask *valid_bitmask; /* bitmask of valid keys */ + u_int32_t seq; }; #define ULOGD_IPFIX_TEMPL_BASE 1024 @@ -295,6 +304,66 @@ static int put_data_records(struct ulogd_pluginstance *upi, return len; } +static struct ipfix_msg_hdr *build_ipfix_msg(struct ulogd_pluginstance *upi, + struct ulogd_ipfix_template *template, + bool need_template) +{ + struct ipfix_instance *ii = (struct ipfix_instance *) &upi->private; + u_int16_t tmpl_len; + struct ipfix_msg_hdr *msg_hdr; + struct ipfix_templ_rec_hdr *tmpl_hdr; + struct ipfix_set_hdr *data_hdr, *tmpl_set_hdr; + void *buf; + int msglen, ret; + + msglen = sizeof(struct ipfix_msg_hdr) + sizeof(struct ipfix_set_hdr) + + template->data_length; + if (need_template) + msglen = msglen + sizeof(struct ipfix_set_hdr) + + (template->tmpl_cur - (void *)&template->tmpl); + buf = malloc(msglen); + if (buf == NULL) + return NULL; + memset(buf, 0, msglen); + + /* ipfix msg header */ + msg_hdr = buf; + msg_hdr->version = htons(10); + msg_hdr->length = htons(msglen); + msg_hdr->seq = htonl(ii->seq++); + msg_hdr->domain_id = htonl(domain_ce(upi->config_kset).u.value); + if (need_template) { + /* put set header and template records */ + tmpl_set_hdr = buf + sizeof(*msg_hdr); + tmpl_set_hdr->set_id = htons(2); + tmpl_len = template->tmpl_cur - (void *)&template->tmpl; + tmpl_set_hdr->length = htons(sizeof(*tmpl_set_hdr) + tmpl_len); + tmpl_hdr = (void *)tmpl_set_hdr + sizeof(*tmpl_set_hdr); + memcpy((void *)tmpl_hdr, (void *)&template->tmpl, tmpl_len); + data_hdr = (void *)tmpl_hdr + tmpl_len; + } else { + data_hdr = buf + sizeof(*msg_hdr); + } + + /* put set header and data records */ + data_hdr->set_id = template->tmpl.hdr.templ_id; /* already ordered */ + data_hdr->length = htons(sizeof(*data_hdr) + template->data_length); + ret = put_data_records(upi, template, (void *)data_hdr + sizeof(*data_hdr)); + if (ret < 0) { + ulogd_log(ULOGD_ERROR, "could not build ipfix dataset"); + goto free_buf; + } else if (ret > msglen) { + ulogd_log(ULOGD_ERROR, "overflowed on building ipfix dataset"); + goto free_buf; + } + + return msg_hdr; + +free_buf: + free(buf); + return NULL; +} + static int output_ipfix(struct ulogd_pluginstance *upi) { struct ipfix_instance *ii = (struct ipfix_instance *) &upi->private; -- 1.9.1