Viewing: liblnetconfig_netlink.c

// SPDX-License-Identifier: LGPL-2.1+

/*
 * Copyright (c) 2021  UT-Battelle, LLC
 */

/*
 * This file is part of Lustre, http://www.lustre.org/
 *
 * Netlink handling.
 *
 * Author: James Simmons <jsimmons@infradead.org>
 */

#include <ctype.h>
#include <unistd.h>
#include <fcntl.h>
#include <yaml.h>

#include <linux/lnet/lnet-nl.h>
#include "liblnetconfig.h"

/**
 * Set NETLINK_BROADCAST_ERROR flags on socket to report ENOBUFS errors.
 *
 * @sk		Socket to change the flags.
 *
 * Return	0 on success or a Netlink error code.
 */
static int nl_socket_enable_broadcast_error(struct nl_sock *sk)
{
	const int state = 1; /* enable errors */
	int err;

	if (nl_socket_get_fd(sk) < 0)
		return -NLE_BAD_SOCK;

	err = setsockopt(nl_socket_get_fd(sk), SOL_NETLINK,
			 NETLINK_BROADCAST_ERROR, &state, sizeof(state));
	if (err < 0)
		return -nl_syserr2nlerr(errno);

	return 0;
}

/**
 * Enable/disable extending ACK for netlink socket. Used for
 * sending extra debugging information.
 *
 * @arg sk              Netlink socket.
 * @arg state           New state (0 - disabled, 1 - enabled)
 *
 * @return 0 on success or a negative error code
 */
static int nl_socket_set_ext_ack(struct nl_sock *sk, int state)
{
	int err;

	if (nl_socket_get_fd(sk) < 0)
		return -NLE_BAD_SOCK;

	err = setsockopt(nl_socket_get_fd(sk), SOL_NETLINK,
			 NETLINK_EXT_ACK, &state, sizeof(state));
	if (err < 0 && errno != ENOPROTOOPT)
		return -nl_syserr2nlerr(errno);

	return 0;
}

/**
 * Create a Netlink socket
 *
 * @sk			The nl_sock which we used to handle the Netlink
 *			connection.
 * @async_events	tell the Netlink socket this will receive asynchronous
 *			data
 *
 * Return		0 on success or a negative error code.
 */
static int lustre_netlink_register(struct nl_sock *sk, bool async_events)
{
	int rc;

	rc = genl_connect(sk);
	if (rc < 0)
		return rc;

	rc = nl_socket_enable_broadcast_error(sk);
	if (rc < 0)
		return rc;

	rc = nl_socket_set_ext_ack(sk, true);
	if (rc < 0)
		return rc;

	if (async_events) {
		/* Required to receive async netlink event notifications */
		nl_socket_disable_seq_check(sk);
		/* Don't need ACK for events generated by kernel */
		nl_socket_disable_auto_ack(sk);
	}

	return rc;
}

/* A YAML file is used to describe data. In a YAML document the content is
 * all about a collection of scalars used to create new data types such as
 * key-value pairs. This allows complex documents to represent anything from
 * a string to a tree.
 *
 * Scalar:
 * ---------
 * YAML scalars are a simple value which can be a string, number or Boolean.
 * They are the simplest data types. They can exist in a YAML document but
 * are typically used to build more complex data formats.
 *
 * Collections:
 * ------------
 * In YAML collections are scalar elements presented in the form of
 * an array, called a sequence, or mappings (hashes) that are scalar
 * key value pairs. All elements belonging to the same collection are
 * the lines that begin at the same indentation level
 *
 * Sequences use a dash followed by a space.
 * Mappings use a colon followed by a space (: ) to mark each key/value pair:
 *
 * Collections can be represented in two forms, flow and block.
 * Note they are equivalent. Example of block sequence is;
 *
 * - string
 * - integer
 * - boolean
 *
 * and a block mapping example is:
 *
 * string: hello
 * integer: 5
 * boolean: False
 *
 * YAML flow styles for collections uses explicit indicators rather than
 * indentation to denote scope.
 *
 * A sequence can be written as a comma separated list within
 * square brackets ([]):
 *
 * [ PHP, Perl, Python ]
 *
 * A mapping can be written as a comma separated list of key/values within
 * curly braces ({}):
 *
 * { PHP: 5.2, MySQL: 5.1, Apache: 2.2.20 }
 *
 * NOTE!! flow and block are equivalent.
 *
 * List:
 * ------
 * A list is a defined array of data which can be either an flow or block
 * sequence. Lists can be nested. Example
 *
 * numbers: [ 1, 2, 3, 4 ]
 *
 * numbers:
 *  - 1
 *  - 2
 *  - 3
 *  - 4
 *
 * Dictionaries:
 * --------------
 * Are comprised of a key: value format with contents indented. This is
 * built on top of the flow or block mapping. Like lists they can be nested.
 *
 * ports:
 * - port: 8080
 *     targetPort: 8080
 *       nodePort: 30012
 */

/* In YAML you have the concept of parsers and emitters. Parser
 * consume YAML input from a file, character buffer, or in our
 * case Netlink and emitters take data from some source and
 * present it in a YAML format.
 *
 * In this section of the code we are handling the parsing of the
 * Netlink packets coming in and using them to piece together a
 * YAML document. We could in theory just dump a YAML document
 * one line at a time over Netlink but the amount of data could
 * become very large and impact performance. Additionally, having
 * pseudo-YAML code in the kernel would be frowned on. We can
 * optimize the network traffic by taking advantage of the fact
 * that for key/value pairs the keys rarely change. We can
 * break up the data into keys and the values. The first Netlink
 * data packets received will be a nested keys table which we
 * can cache locally. As we receive the value pairs we can then
 * reconstruct the key : value pair by looking up the the key
 * in the stored table. In effect we end up with a one key to
 * many values stream of data.
 *
 * The data structures below are used to create a tree data
 * structure which is the natural flow of both YAML and
 * Netlink.
 */
struct yaml_nl_node {
	struct nl_list_head	list;
	struct nl_list_head	children;
	struct ln_key_list	keys;
};

struct yaml_netlink_input {
	yaml_parser_t		*parser;
	void			*start;
	void			*read;
	void			*buffer;
	void			*end;
	const char		*errmsg;
	int			error;
	struct nl_sock		*nl;
	bool			complete;
	bool			async;
	unsigned int		indent;
	unsigned int		version;
	struct yaml_nl_node	*cur;
	struct yaml_nl_node	*root;
};

/* Sadly this is not exported out of libyaml. We want to
 * give descent error message to help people track down
 * issues. This is internal only to this code. The end
 * user will never need to use this.
 */
static int
yaml_parser_set_reader_error(yaml_parser_t *parser, const char *problem,
			     size_t offset, int value)
{
	parser->error = YAML_READER_ERROR;
	parser->problem = problem;
	parser->problem_offset = offset;
	parser->problem_value = value;

	return 0;
}

/* This is used to handle all the Netlink packets containing the keys
 * for the key/value pairs. Instead of creating unique code to handle
 * every type of Netlink attributes possible we create a generic
 * abstract so the same code be used with everything. To make this
 * work the key table trasmitted must report the tree structure and
 * state of the keys. We use nested attributes as a way to notify libyaml
 * we have a new collection. This is used to create the tree structure
 * of the YAML document. Each collection of attributes define the following:
 *
 * LN_SCALAR_ATTR_INDEX:
 *	enum XXX_ATTR that defines which value we are dealing with. This
 *	varies greatly depending on the subsystem we have developed for.
 *
 * LN_SCALAR_ATTR_NLA_TYPE:
 *	The Netlink attribute type (NLA_STRING, NLA_U32, etc..) the coming
 *	value will be.
 *
 * LN_SCALAR_ATTR_VALUE:
 *	The string represnting key's actually scalar value.
 *
 * LN_SCALAR_ATTR_INT_VALUE:
 *	For this case the key is an integer value. This shouldn't be
 *	sent for the receive case since we are going to just turn it
 *	into a string for YAML. Sending packets will make use of this.
 *
 * LN_SCALAR_ATTR_KEY_TYPE:
 *	What YAML format is it? block or flow. Only useful for
 *	LN_SCALAR_ATTR_NLA_TYPE of type NLA_NESTED or NLA_NUL_STRING
 *
 * LN_SCALAR_ATTR_LIST + LN_SCALAR_LIST_SIZE:
 *	Defined the next collection which is a collection of nested
 *	attributes of the above.
 */
static struct nla_policy scalar_attr_policy[LN_SCALAR_MAX + 1] = {
	[LN_SCALAR_ATTR_LIST]		= { .type = NLA_NESTED },
	[LN_SCALAR_ATTR_LIST_SIZE]	= { .type = NLA_U16 },
	[LN_SCALAR_ATTR_INDEX]		= { .type = NLA_U16 },
	[LN_SCALAR_ATTR_NLA_TYPE]	= { .type = NLA_U16 },
	[LN_SCALAR_ATTR_VALUE]		= { .type = NLA_STRING },
	[LN_SCALAR_ATTR_INT_VALUE]	= { .type = NLA_S64 },
	[LN_SCALAR_ATTR_KEY_FORMAT]	= { .type = NLA_U16 },
};

static int yaml_parse_key_list(struct yaml_netlink_input *data,
			       struct yaml_nl_node *parent,
			       struct nlattr *list)
{
	struct nlattr *tbl_info[LN_SCALAR_MAX + 1];
	struct yaml_nl_node *node = NULL;
	struct nlattr *attr;
	int rem;

	nla_for_each_nested(attr, list, rem) {
		uint16_t index = 0;

		if (nla_parse_nested(tbl_info, LN_SCALAR_MAX, attr,
				     scalar_attr_policy))
			break;

		if (tbl_info[LN_SCALAR_ATTR_LIST_SIZE]) {
			size_t cnt;

			cnt = nla_get_u16(tbl_info[LN_SCALAR_ATTR_LIST_SIZE]) + 1;
			if (!node) {
				size_t len = sizeof(struct nl_list_head) * 2;

				len += sizeof(struct ln_key_props) * cnt;
				node = calloc(1, len);
				if (!node)
					return NL_STOP;

				node->keys.lkl_maxattr = cnt;
				NL_INIT_LIST_HEAD(&node->children);
				nl_init_list_head(&node->list);

				if (!data->root)
					data->root = node;
				if (!data->cur)
					data->cur = node;
				if (parent)
					nl_list_add_tail(&node->list,
							 &parent->children);
			}
		}

		if (tbl_info[LN_SCALAR_ATTR_INDEX])
			index = nla_get_u16(tbl_info[LN_SCALAR_ATTR_INDEX]);

		if (!node || index == 0)
			return NL_STOP;

		if (tbl_info[LN_SCALAR_ATTR_KEY_FORMAT]) {
			uint16_t format;

			format = nla_get_u16(tbl_info[LN_SCALAR_ATTR_KEY_FORMAT]);
			node->keys.lkl_list[index].lkp_key_format = format;
		}

		if (tbl_info[LN_SCALAR_ATTR_NLA_TYPE]) {
			uint16_t type;

			type = nla_get_u16(tbl_info[LN_SCALAR_ATTR_NLA_TYPE]);
			node->keys.lkl_list[index].lkp_data_type = type;
		}

		if (tbl_info[LN_SCALAR_ATTR_VALUE]) {
			char *name;

			name = nla_strdup(tbl_info[LN_SCALAR_ATTR_VALUE]);
			if (!name)
				return NL_STOP;
			node->keys.lkl_list[index].lkp_value = name;
		}

		if (tbl_info[LN_SCALAR_ATTR_LIST]) {
			int rc = yaml_parse_key_list(data, node,
						     tbl_info[LN_SCALAR_ATTR_LIST]);
			if (rc != NL_OK)
				return rc;
		}
	}
	return NL_OK;
}

/* We translate Netlink nested list into either a YAML mappping or sequence.
 * This generates the start of such a YAML block.
 */
static int yaml_nested_header(struct yaml_netlink_input *data,
			      int *size, unsigned int *indent,
			      int mapping, struct ln_key_props *keys)
{
	int len = 0;

	if (keys->lkp_key_format & LNKF_FLOW) {
		char brace = '{';

		if (keys->lkp_key_format & LNKF_SEQUENCE)
			brace = '[';

		len = snprintf(data->buffer, *size, "%*s%s: %c ", data->indent,
			       "", keys->lkp_value, brace);
	} else {
		int count = mapping & LNKF_SEQUENCE ? 0 : data->indent;

		if (keys->lkp_key_format & LNKF_MAPPING)
			*indent += 2;
		if (keys->lkp_key_format & LNKF_SEQUENCE)
			*indent += 2;

		len = snprintf(data->buffer, *size, "%*s%s:\n", count, "",
			       keys->lkp_value);
	}

	return len;
}

static struct yaml_nl_node *get_next_child(struct yaml_nl_node *node,
					   unsigned int idx)
{
	struct yaml_nl_node *child;
	unsigned int i = 0;

	nl_list_for_each_entry(child, &node->children, list)
		if (idx == i++)
			return child;

	return NULL;
}

/**
 * In the YAML C implementation the scanner transforms the input stream
 * (Netlink in this case) into a sequence of keys. First we need to
 * examine the potential keys involved to see the mapping to Netlink.
 * We have chosen to examine the YAML stack with keys since they are
 * more detailed when compared to yaml_document_t / yaml_nodes and
 * yaml_event_t.
 *
 *	STREAM-START(encoding)		# The stream start.
 *	STREAM-END			# The stream end.
 *      VERSION-DIRECTIVE(major,minor)	# The '%YAML' directive.
 *      TAG-DIRECTIVE(handle,prefix)	# The '%TAG' directive.
 *      DOCUMENT-START			# '---'
 *      DOCUMENT-END			# '...'
 *      BLOCK-SEQUENCE-START		# Indentation increase denoting a block
 *      BLOCK-MAPPING-START		# sequence or a block mapping.
 *      BLOCK-END			# Indentation decrease.
 *      FLOW-SEQUENCE-START		# '['
 *      FLOW-SEQUENCE-END		# ']'
 *      FLOW-MAPPING-START		# '{'
 *      FLOW-MAPPING-END		# '}'
 *      BLOCK-ENTRY			# '-'
 *      FLOW-ENTRY			# ','
 *      KEY				# '?' or nothing (simple keys).
 *      VALUE				# ':'
 *      ALIAS(anchor)			# '*anchor'
 *      ANCHOR(anchor)			# '&anchor'
 *      TAG(handle,suffix)		# '!handle!suffix'
 *      SCALAR(value,style)		# A scalar.
 *
 * For our read_handler / write_handler STREAM-START / STREAM-END,
 * VERSION-DIRECTIVE, and TAG-DIRECTIVE are hanndler by the libyaml
 * internal scanner so we don't need to deal with it. Normally for
 * LNet / Lustre DOCUMENT-START / DOCUMENT-END are not needed but it
 * could be easily handled. In the case of multiplex streams we could
 * see these used to differentiate data coming in.
 *
 * It is here we handle any simple scalars or values of the key /value
 * pair. How the YAML document is formated is dependent on the key
 * table's data.
 */
static void yaml_parse_value_list(struct yaml_netlink_input *data, int *size,
				  struct nlattr *attr_array[],
				  struct ln_key_props *parent)
{
	struct yaml_nl_node *node = data->cur;
	struct ln_key_props *keys = node->keys.lkl_list;
	int mapping = parent->lkp_key_format;
	int child_idx = 0, len = 0, i;
	bool first = true;

	for (i = 1; i < node->keys.lkl_maxattr; i++) {
		struct nlattr *attr;

		attr = attr_array[i];
		if (!attr && !keys[i].lkp_value)
			continue;

		/* This function is called for each Netlink nested list.
		 * Each nested list is treated as a YAML block. It is here
		 * we handle data for the YAML block. How that data is seen
		 * for YAML is based on the parents mapping and the type of
		 * data value sent.
		 *
		 * The cases are:
		 *
		 * the value type is NLA_NUL_STRING which is interepted as
		 *      key:\n
		 *
		 * Also NLA_NUL_STRING is used to update a single key value.
		 *
		 * the key has no lkp_value and we do receive a 'value'
		 * that is not a nested list in the Netlink packet. This is
		 * treated as a plain scalar.
		 *
		 * we have a key lkp_value and the parent mapping is
		 * LNKF_MAPPING then we have a key : value pair. During
		 * our loop the key normally doesn't change.
		 *
		 * This data belongs to a YAML block which can be of
		 * different kinds (FLOW, SEQUENCE, MAPPING). We determine
		 * the type and adjust the first line of output for the
		 * YAML results if needed. Most of the time the creation
		 * of the nested header is done in the NLA_NESTED case
		 * switch below which happens before this function is
		 * called. Specific handling is done here.
		 *
		 * The common case handled here is for building of the
		 * mapping key : value pair. Another case is that we
		 * are at the start of a SEQUENCE block. If this is the
		 * case we add '-' to the output and clear the flag
		 * LNKF_SEQUENCE to prevent multiple instanstances of
		 * '-'. Only one '-' per SEQUENCE block. We need to
		 * manually add '-' also in the case of were our nested
		 * block first PROCESSED attr instance is another nested
		 * block. For example:
		 *	local NI(s):
		 *	-     interfaces:
		 *		   0: ib0
		 */
		if ((first && (mapping & LNKF_SEQUENCE) &&
		     keys[i].lkp_data_type == NLA_NESTED) ||
		    (keys[i].lkp_data_type != NLA_NUL_STRING &&
		     keys[i].lkp_data_type != NLA_NESTED)) {
			if (!attr && keys[i].lkp_data_type != NLA_FLAG)
				continue;

			/* Mark this as the start of a SEQUENCE block */
			if (!(mapping & LNKF_FLOW)) {
				unsigned int indent = data->indent ?
						      data->indent : 2;

				memset(data->buffer, ' ', indent);
				if (mapping & LNKF_SEQUENCE) {
					((char *)data->buffer)[indent - 2] = '-';
					if (keys[i].lkp_data_type != NLA_NESTED &&
					    mapping & LNKF_MAPPING)
						mapping &= ~LNKF_SEQUENCE;
				}
				data->buffer += indent;
				*size -= indent;
			}

			/* Start of the build of the key : value pair.
			 * Very common case.
			 */
			if (keys[i].lkp_data_type != NLA_NESTED &&
			    mapping & LNKF_MAPPING) {
				len = snprintf(data->buffer, *size, "%s: ",
					       keys[i].lkp_value);
				if (len < 0)
					goto unwind;
				data->buffer += len;
				*size -= len;
			}
		}

		switch (keys[i].lkp_data_type) {
		case NLA_NESTED: {
			struct yaml_nl_node *next = get_next_child(node,
								   child_idx++);
			int num = next ? next->keys.lkl_maxattr : 0;
			struct nla_policy nest_policy[num];
			struct yaml_nl_node *old;
			struct nlattr *cnt_attr;
			unsigned int indent = 0;
			bool start = true;
			int rem, j;

			if (!attr || !next)
				continue;

			memset(nest_policy, 0, sizeof(struct nla_policy) * num);
			for (j = 1; j < num; j++)
				nest_policy[j].type = next->keys.lkl_list[j].lkp_data_type;

			/* We might have a empty list but by YAML standards
			 * we still need to display the header.
			 */
			if (!nla_len(attr)) {
				len = yaml_nested_header(data, size, &indent,
							 first ? mapping : 0,
							 &keys[i]);
				if (len < 0)
					goto unwind;
				data->buffer += len;
				*size -= len;
				len = 0;
			}

			old = data->cur;
			data->cur = next;
			nla_for_each_nested(cnt_attr, attr, rem) {
				struct nlattr *nest_info[num];

				if (nla_parse_nested(nest_info, num, cnt_attr,
						     nest_policy))
					break;

				/* Create the nested header only once at start */
				if (!start)
					goto skip_nested_header;
				start = false;

				/* Update the header's first key */
				if (next->keys.lkl_list[1].lkp_data_type == NLA_NUL_STRING &&
				    !keys[i].lkp_value)
					keys[i].lkp_value = nla_strdup(nest_info[1]);

				len = yaml_nested_header(data, size, &indent,
							 first ? mapping : 0,
							 &keys[i]);
				if (len < 0)
					goto unwind;
				data->buffer += len;
				*size -= len;
				len = 0;
skip_nested_header:
				data->indent += indent;
				yaml_parse_value_list(data, size, nest_info,
						      &keys[i]);
				data->indent -= indent;
			}

			/* nested bookend header */
			if (keys[i].lkp_key_format & LNKF_FLOW) {
				char *tmp = (char *)data->buffer - 2;
				char *brace = " }\n";

				if (keys[i].lkp_key_format &
				    LNKF_SEQUENCE)
					brace = " ]\n";

				memcpy(tmp, brace, strlen(brace));
				data->buffer++;
				*size -= 1;
			}
			data->cur = old;

			/* This is for the special case of the first attr of
			 * a nested list is another nested list. We had to
			 * insert a '-' but that is only done once so clear
			 * the mapping of LNKF_SEQUENCE.
			 */
			if (first && attr) {
				if (mapping & LNKF_MAPPING)
					mapping &= ~LNKF_SEQUENCE;
				first = false;
			}
			break;
		}

		/* Handle the key:\n YAML case or updating an individual key */
		case NLA_NUL_STRING:
			if (i == 1) {
				if (data->cur != data->root)
					goto not_first;

				/* The top level is special so only print
				 * once
				 */
				if (strlen(keys[i].lkp_value)) {
					len = snprintf(data->buffer,
						       *size, "%s:\n",
						       keys[i].lkp_value);
					if (len < 0)
						goto unwind;
					data->buffer += len;
					*size -= len;
					len = 0;
				}
				data->indent = 0;
				if (!(mapping & LNKF_FLOW)) {
					if (mapping & LNKF_SEQUENCE)
						data->indent += 2;
					else if (mapping & LNKF_MAPPING)
						data->indent += 2;
				}
not_first:
				if (attr && parent->lkp_value) {
					free(parent->lkp_value);
					parent->lkp_value = nla_strdup(attr);
				}
			}
			break;

		/* The below is used for a plain scalar or to complete the
		*  key : value pair.
		*/
		case NLA_STRING:
			len = snprintf(data->buffer, *size, "%s",
				       nla_get_string(attr));
			break;

		case NLA_FLAG:
			len = snprintf(data->buffer, *size, "%s",
				       attr ? "true" : "false");
			break;

		case NLA_U16:
			len = snprintf(data->buffer, *size, "%hu",
				       nla_get_u16(attr));
			break;

		case NLA_U32:
			len = snprintf(data->buffer, *size, "%u",
				       nla_get_u32(attr));
			break;

		case NLA_U64:
			len = snprintf(data->buffer, *size, "%ju",
				       nla_get_u64(attr));
			break;

		case NLA_S16:
			len = snprintf(data->buffer, *size, "%hd",
				       nla_get_u16(attr));
			break;

		case NLA_S32:
			len = snprintf(data->buffer, *size, "%d",
				       nla_get_s32(attr));
			break;

		case NLA_S64:
			len = snprintf(data->buffer, *size, "%jd",
				       nla_get_s64(attr));
			/* fallthrough */
		default:
			break;
		}

		if (len) {
			if (mapping & LNKF_FLOW) {
				strcat((char *)data->buffer, ", ");
				len += 2;
			} else {
				if ((mapping == LNKF_SEQUENCE) &&
				    !keys[i].lkp_value)
					((char *)data->buffer)[len++] = ':';

				((char *)data->buffer)[len++] = '\n';
			}
			data->buffer += len;
			*size += len;
		} else if (len < 0) {
unwind:
			data->buffer -= data->indent + 2;
			*size -= data->indent + 2;
		}
	}
}

static bool cleanup_children(struct yaml_nl_node *parent)
{
	struct yaml_nl_node *child;

	if (nl_list_empty(&parent->children)) {
		struct ln_key_props *keys = parent->keys.lkl_list;
		int i;

		for (i = 1; i < parent->keys.lkl_maxattr; i++)
			if (keys[i].lkp_value)
				free(keys[i].lkp_value);
		nl_list_del(&parent->list);
		return true;
	}

	while ((child = get_next_child(parent, 0)) != NULL) {
		if (cleanup_children(child))
			free(child);
	}

	return false;
}

/* This is the CB_VALID callback for the Netlink library that we
 * have hooked into. Any successful Netlink message is passed to
 * this function which handles both the incoming key tables and
 * the values of the key/value pairs being received. We use
 * the NLM_F_CREATE flag to determine if the incoming Netlink
 * message is a key table or a packet containing value pairs.
 */
static int yaml_netlink_msg_parse(struct nl_msg *msg, void *arg)
{
	yaml_parser_t *parser = arg;
	struct yaml_netlink_input *data = parser->read_handler_data;
	struct nlmsghdr *nlh = nlmsg_hdr(msg);

	if (nlh->nlmsg_flags & NLM_F_CREATE) {
		struct genlmsghdr *ghdr = genlmsg_hdr(nlh);
		struct nlattr *attrs[LN_SCALAR_MAX + 1];

		if (genlmsg_parse(nlh, 0, attrs, LN_SCALAR_MAX,
				  scalar_attr_policy))
			return NL_SKIP;

		/* If root already exists this means we are updating the
		 * key table. Free old key table.
		 */
		if (data->root && (nlh->nlmsg_flags & NLM_F_REPLACE)) {
			cleanup_children(data->root);
			free(data->root);
			data->root = NULL;
		}

		if (attrs[LN_SCALAR_ATTR_LIST]) {
			int rc = yaml_parse_key_list(data, NULL,
						     attrs[LN_SCALAR_ATTR_LIST]);
			if (rc != NL_OK)
				return rc;

			/* reset to root node */
			data->cur = data->root;
		}

		/* For streaming insert '---' to define start of
		 * YAML document. This allows use to extract
		 * documents out of a multiplexed stream.
		 */
		if (data->async) {
			char *start_doc = "---\n";
			size_t len = strlen(start_doc) + 1;

			strncpy(data->buffer, start_doc, len);
			data->buffer += len - 1;
		}
		data->version = ghdr->version;
	} else {
		uint16_t maxtype = data->cur->keys.lkl_maxattr;
		struct nla_policy policy[maxtype];
		struct nlattr *attrs[maxtype];
		int size, i;

		memset(policy, 0, sizeof(struct nla_policy) * maxtype);
		for (i = 1; i < maxtype; i++)
			policy[i].type = data->cur->keys.lkl_list[i].lkp_data_type;

		if (genlmsg_parse(nlh, 0, attrs, maxtype, policy))
			return NL_SKIP;

		size = data->end - data->buffer;
		if (size < 1024) {
			size_t len = (data->end - data->start) * 2;
			size_t off = data->buffer - data->start;

			data->start = realloc(data->start, len);
			if (!data->start)
				return NL_STOP;
			data->end = data->start + len;

			data->buffer = data->start + off;
			data->read = data->start;

			size = data->end - data->buffer;
		}
		yaml_parse_value_list(data, &size, attrs,
				      &data->cur->keys.lkl_list[1]);
	}

	/* Let yaml_netlink_msg_complete end collecting data */
	return NL_OK;
}

/* This is the libnl callback for when an error has happened
 * kernel side. An error message is sent back to the user.
 */
static int yaml_netlink_parse_msg_error(struct nlmsgerr *errmsg,
					yaml_parser_t *parser)
{
	struct nlmsghdr *nlh = (void *)errmsg - NLMSG_HDRLEN;

	if ((nlh->nlmsg_type == NLMSG_ERROR ||
	     nlh->nlmsg_flags & NLM_F_ACK_TLVS) && errmsg->error) {
		/* libyaml stomps on the reader error so we need to
		 * cache the source of the error.
		 */
		const char *errstr = nl_geterror(nl_syserr2nlerr(errmsg->error));
		struct yaml_netlink_input *data = parser->read_handler_data;

		/* Newer kernels support NLM_F_ACK_TLVS in nlmsg_flags
		 * which gives greater detail why we failed.
		 */
		if ((nlh->nlmsg_flags & NLM_F_ACK_TLVS) &&
		    !(nlh->nlmsg_flags & NLM_F_CAPPED)) {
			struct nlattr *head = ((void *)&errmsg->msg);
			struct nlattr *tb[NLMSGERR_ATTR_MAX];

			if (nla_parse(tb, NLMSGERR_ATTR_MAX, head,
				      nlmsg_attrlen(nlh, 0), NULL) == 0) {
				if (tb[NLMSGERR_ATTR_MSG])
					errstr = nla_strdup(tb[NLMSGERR_ATTR_MSG]);
			}
		}

		parser->error = YAML_READER_ERROR;
		data->errmsg = errstr;
		data->error = errmsg->error;
		data->complete = true;
	}

	return parser->error;
}

/* This is the libnl callback for when an error has happened
 * kernel side. An error message is sent back to the user.
 */
static int yaml_netlink_msg_error(struct sockaddr_nl *who,
				  struct nlmsgerr *errmsg, void *arg)
{
	yaml_netlink_parse_msg_error(errmsg, (yaml_parser_t *)arg);

	return NL_STOP;
}

/* This is the libnl callback for when the last Netlink packet
 * is finished being parsed or its called right away in case
 * the Linux kernel reports back an error from the Netlink layer.
 */
static int yaml_netlink_msg_complete(struct nl_msg *msg, void *arg)
{
	struct nlmsghdr *nlh = nlmsg_hdr(msg);
	struct yaml_netlink_input *data;
	yaml_parser_t *parser = arg;

	/* For the case of NLM_F_DUMP the kernel will send error msgs
	 * yet not be labled NLMSG_ERROR which results in this code
	 * path being executed.
	 */
	if (yaml_netlink_parse_msg_error(nlmsg_data(nlh), parser) ==
	    YAML_READER_ERROR)
		return NL_STOP;

	/* Free internal data. */
	data = parser->read_handler_data;
	if (data->root) {
		cleanup_children(data->root);
		free(data->root);
		data->root = NULL;
	}

	/* For streaming insert '...' to define end of
	 * YAML document
	 */
	if (data->async) {
		char *end_doc = "...\n";
		size_t len = strlen(end_doc) + 1;

		strncpy(data->buffer, end_doc, len);
		data->buffer += len - 1;
	} else {
		data->complete = true;
	}

	return data->async ? NL_OK : NL_STOP;
}

/**
 * In order for yaml_parser_set_input_netlink() to work we have to
 * register a yaml_read_handler_t callback. This is that call back
 * which listens for Netlink packets. Internally nl_recvmsg_report()
 * calls the various callbacks discussed above.
 */
static int yaml_netlink_read_handler(void *arg, unsigned char *buffer,
				     size_t size, size_t *size_read)
{
	struct yaml_netlink_input *data = arg;
	int rc = 0;

	/* First collect the Netlink data and then transfer it
	 * into the internal libyaml buffers.
	 */
	if (!data->complete) {
		struct nl_cb *cb = nl_socket_get_cb(data->nl);

		rc = nl_recvmsgs_report(data->nl, cb);
		if (rc == -NLE_INTR) {
			*size_read = 0;
			return 1;
		} else if (!data->errmsg && rc < 0) {
			data->errmsg = nl_geterror(rc);
			return 0;
		} else if (data->parser->error) {
			/* data->errmsg is set in NL_CB_FINISH */
			return 0;
		}
	}
	rc = data->buffer - data->read;
	if ((int)size > rc)
		size = rc;

	if (size) {
		memcpy(buffer, data->read, size);
		data->read += size;
	} else if (data->complete) {
		free(data->start);
		data->start = NULL;
	}
	*size_read = size;
	return 1;
}

/* libyaml by default just reports "input error" for parser read_handler_t
 * issues which is not useful. This provides away to get better debugging
 * info.
 */
YAML_DECLARE(const char *)
yaml_parser_get_reader_error(yaml_parser_t *parser)
{
	struct yaml_netlink_input *buf = parser->read_handler_data;

	if (!buf)
		return NULL;

	errno = buf->error;
	return buf->errmsg;
}

YAML_DECLARE(int)
yaml_parser_get_reader_proto_version(yaml_parser_t *parser)
{
	struct yaml_netlink_input *buf = parser->read_handler_data;

	if (!buf)
		return 0;

	return buf->version;
}

/* yaml_parser_set_input_netlink() mirrors the libyaml function
 * yaml_parser_set_input_file(). Internally it does setup of the
 * libnl socket callbacks to parse the Netlink messages received
 * as well as register the special yaml_read_handler_t for libyaml.
 * This is exposed for public use.
 */
YAML_DECLARE(int)
yaml_parser_set_input_netlink(yaml_parser_t *reply, struct nl_sock *nl,
			      bool stream)
{
	struct yaml_netlink_input *buf;
	int rc;

	buf = calloc(1, sizeof(*buf));
	if (!buf) {
		reply->error = YAML_MEMORY_ERROR;
		return false;
	}

	rc = lustre_netlink_register(nl, stream);
	if (rc < 0) {
		yaml_parser_set_reader_error(reply,
					     "netlink setup failed", 0,
					     -rc);
		goto failed;
	}

	buf->start = malloc(65536);
	buf->end = buf->start + 65536;
	buf->buffer = buf->start;
	buf->read = buf->start;
	buf->nl = nl;
	buf->async = stream;
	buf->parser = reply;
	yaml_parser_set_input(reply, yaml_netlink_read_handler, buf);

	rc = nl_socket_modify_cb(buf->nl, NL_CB_VALID, NL_CB_CUSTOM,
				 yaml_netlink_msg_parse, reply);
	if (rc < 0) {
		yaml_parser_set_reader_error(reply,
					     "netlink msg recv setup failed",
					     0, -rc);
		goto failed;
	}

	rc = nl_socket_modify_cb(buf->nl, NL_CB_FINISH, NL_CB_CUSTOM,
				 yaml_netlink_msg_complete, reply);
	if (rc < 0) {
		yaml_parser_set_reader_error(reply,
					     "netlink msg cleanup setup failed",
					     0, -rc);
		goto failed;
	}

	rc = nl_socket_modify_err_cb(buf->nl, NL_CB_CUSTOM, yaml_netlink_msg_error,
				     reply);
	if (rc < 0) {
		yaml_parser_set_reader_error(reply,
					     "failed to register error handling",
					     0, -rc);
failed:
		free(buf);
	}

	return rc < 0 ? false : true;
}

/* The role of the YAML emitter for us is to take a YAML document and
 * change into a Netlink stream to send to the kernel to be processed.
 * This provides the infrastructure to do this.
 */
struct yaml_netlink_output {
	yaml_emitter_t		*emitter;
	struct nl_sock		*nl;
	struct nl_sock		*ctrl;
	char			*family;
	int			family_id;
	int			version;
	int			cmd;
	int			pid;
	int			flags;
};

/* Internal use for this file only. We fill in details of why creating
 * a Netlink packet to send failed. The end user will be able to debug
 * what went wrong.
 */
static int
yaml_emitter_set_writer_error(yaml_emitter_t *emitter, const char *problem)
{
	emitter->error = YAML_WRITER_ERROR;
	emitter->problem = problem;

	return 0;
}

static unsigned int indent_level(const char *str)
{
	char *tmp = (char *)str;

	while (isspace(*tmp))
		++tmp;
	return tmp - str;
}

#define LNKF_BLOCK 8

static enum lnet_nl_key_format yaml_format_type(yaml_emitter_t *emitter,
						char *line,
						unsigned int *offset)
{
	unsigned int indent = *offset, new_indent = 0;
	enum lnet_nl_key_format fmt = 0;
	char *tmp, *flow;

	new_indent = indent_level(line);
	if (new_indent < indent) {
		*offset = indent - emitter->best_indent;
		return LNKF_BLOCK;
	}

	if (strncmp(line + new_indent, "- ", 2) == 0) {
		memset(line + new_indent, ' ', 2);
		/* Eat white spaces physical YAML config files have */
		new_indent += strspn(line + new_indent, " ");
		fmt |= LNKF_SEQUENCE;
	}

	/* hdr: [ a : 1, b : 2, c : 3 ] */
	tmp = strstr(line + new_indent, ": ");
	if (!tmp)
		tmp = line + new_indent;
	else
		fmt |= LNKF_MAPPING;

	flow = strchr(line + new_indent, '{');
	if (!flow)
		flow = strchr(line + new_indent, '[');
	if (flow) {
		if (flow < tmp)
			fmt &= ~LNKF_MAPPING;
		fmt |= LNKF_FLOW;
	} else if (strchr(tmp, '}') || strchr(tmp, ']')) {
		if (strchr(tmp, ']'))
			fmt &= ~LNKF_MAPPING;
		fmt |= LNKF_FLOW;
	}

	if (indent != new_indent) {
		*offset = new_indent;
		fmt |= LNKF_BLOCK;
	}

	return fmt;
}

static int yaml_fill_scalar_data(struct nl_msg *msg,
				 enum lnet_nl_key_format fmt,
				 char *line)
{
	char *sep = strstr(line, ": "); /* handle mappings */
	int rc = 0;
	long num;

	if (!sep) {
		char *tmp = strchr(line, ':');

		if (tmp && strlen(tmp) == 1) /* handle simple scalar */
			sep = tmp;
	}
	if (sep)
		*sep = '\0';

	if (strspn(line, "-0123456789") == strlen(line)) {
		num = strtoll(line, NULL, 0);

		NLA_PUT_S64(msg, LN_SCALAR_ATTR_INT_VALUE, num);
	} else {
		NLA_PUT_STRING(msg, LN_SCALAR_ATTR_VALUE, line);
	}

	if (fmt & LNKF_FLOW) {
		memset(line, ' ', strlen(line) + 1);
		goto nla_put_failure;
	}

	if (fmt & LNKF_MAPPING && sep) {
		char *end;
		int len;

		/* restore ':' */
		*sep = ':';
		sep++;
		while (isspace(*sep))
			++sep;

		end = strchr(sep, '\n');
		len = end ? end - sep : strlen(sep);
		if (len <= 0)
			goto nla_put_failure;
		sep[len] = '\0';

		if (strcasecmp(sep, "yes") == 0 ||
		    strcasecmp(sep, "true") == 0 ||
		    strcasecmp(sep, "on") == 0 ||
		    strcasecmp(sep, "y") == 0) {
			NLA_PUT_S64(msg, LN_SCALAR_ATTR_INT_VALUE, 1);
		} else if (strcasecmp(sep, "no") == 0 ||
			   strcasecmp(sep, "false") == 0 ||
			   strcasecmp(sep, "off") == 0 ||
			   strcasecmp(sep, "n") == 0) {
			NLA_PUT_S64(msg, LN_SCALAR_ATTR_INT_VALUE, 0);
		} else if (strspn(sep, "-0123456789") == strlen(sep)) {
			num = strtoll(sep, NULL, 0);
			NLA_PUT_S64(msg, LN_SCALAR_ATTR_INT_VALUE, num);
		} else {
			NLA_PUT_STRING(msg, LN_SCALAR_ATTR_VALUE, sep);
		}
		sep[len] = '\n';
	}
nla_put_failure:
	return rc;
}

static int yaml_create_nested_list(struct yaml_netlink_output *out,
				   struct nl_msg *msg, char **hdr,
				   char **entry, unsigned int *indent,
				   enum lnet_nl_key_format fmt)
{
	struct nlattr *mapping = NULL, *seq = NULL;
	char *line, *tmp;
	int rc = 0;

	/* Not needed for FLOW only case */
	if (fmt & LNKF_SEQUENCE) {
		seq = nla_nest_start(msg, LN_SCALAR_ATTR_LIST);
		if (!seq) {
			yaml_emitter_set_writer_error(out->emitter,
						      "Emmitter netlink list creation failed");
			rc = -EINVAL;
			goto nla_put_failure;
		}
	}

	if (fmt & LNKF_FLOW) {
		struct nlattr *list = NULL;
		bool format = false;
		char *split = NULL;

		if (fmt != LNKF_FLOW) {
			rc = yaml_fill_scalar_data(msg, fmt, *hdr + *indent);
			if (rc < 0)
				goto nla_put_failure;
		}

		tmp = strchr(*hdr, '{');
		if (!tmp) {
			tmp = strchr(*hdr, '[');
			if (!tmp) {
				yaml_emitter_set_writer_error(out->emitter,
							      "Emmitter flow format invalid");
				rc = -EINVAL;
				goto nla_put_failure;
			}
			fmt |= LNKF_SEQUENCE;
		} else
			fmt |= LNKF_MAPPING;
		*tmp = ' ';

		list = nla_nest_start(msg, LN_SCALAR_ATTR_LIST);
		if (!list) {
			yaml_emitter_set_writer_error(out->emitter,
						      "Emmitter netlink list creation failed");
			rc = -EINVAL;
			goto nla_put_failure;
		}

		fmt &= ~LNKF_FLOW;
		while ((line = strsep(hdr, ",")) != NULL) {
			while (!isalnum(line[0]))
				line++;

			/* Flow can be splt across lines by libyaml library. */
			if (strchr(line, ',')) {
				split = line;
				*hdr = line;
				continue;
			}

			tmp = strchr(line, '}');
			if (!tmp)
				tmp = strchr(line, ']');
			if (tmp) {
				format = true;
				*tmp = '\0';
			}

			rc = yaml_fill_scalar_data(msg, fmt, line);
			if (rc < 0)
				goto nla_put_failure;

			/* Move to next YAML line */
			if (format) {
				if (!split)
					line = *entry;
				else
					*entry = NULL;
				break;
			}
		}

		if (!format) {
			yaml_emitter_set_writer_error(out->emitter,
						      "Emmitter flow format invalid");
			rc = -EINVAL;
			goto nla_put_failure;
		}

		if (line && line[0] == '-')
			*indent = 0;

		nla_nest_end(msg, list);
	} else {
next_mapping:
		if (fmt & LNKF_BLOCK && strchr(*hdr, ':')) {
			mapping = nla_nest_start(msg, LN_SCALAR_ATTR_LIST);
			if (!mapping) {
				yaml_emitter_set_writer_error(out->emitter,
							      "Emmitter netlink list creation failed");
				rc = -EINVAL;
				goto nla_put_failure;
			}
		}

		rc = yaml_fill_scalar_data(msg, fmt, *hdr + *indent);
		if (rc < 0)
			goto nla_put_failure;

		do {
			line = strsep(entry, "\n");
have_next_line:
			if (!line || !strlen(line) || strcmp(line, "...") == 0)
				break;

			fmt = yaml_format_type(out->emitter, line, indent);
			if (fmt == LNKF_BLOCK)
				break;

			/* sequences of simple scalars, general mappings, and
			 * plain scalars are not nested structures in a
			 * netlink packet.
			 */
			if (fmt == LNKF_SEQUENCE || fmt == LNKF_MAPPING || fmt == 0) {
				rc = yaml_fill_scalar_data(msg, fmt,
							   line + *indent);
				if (rc < 0)
					goto nla_put_failure;
			} else {
				rc = yaml_create_nested_list(out, msg, &line,
							     entry, indent,
							     fmt);
				if (rc < 0)
					goto nla_put_failure;

				/* if the original line that called
				 * yaml_create_nested_list above was an
				 * sequence and the next line is also
				 * then break to treat it as a mapping / scalar
				 * instead to avoid over nesting.
				 */
				if (line && seq) {
					fmt = yaml_format_type(out->emitter, line, indent);
					if ((fmt & LNKF_SEQUENCE) || (fmt & LNKF_BLOCK))
						break;
				}

				if (line)
					goto have_next_line;
			}
		} while (strcmp(*entry, ""));

		if (mapping) {
			nla_nest_end(msg, mapping);
			mapping = NULL;
		}
	}

	/* test if next line is sequence at the same level. */
	if (line && (line[0] != '\0') && (fmt & LNKF_BLOCK)) {
		int old_indent = indent_level(*hdr);

		fmt = yaml_format_type(out->emitter, line, indent);
		if (fmt != LNKF_BLOCK && old_indent == *indent) {
			/* If we have a normal mapping set then treate
			 * it as a collection of scalars i.e don't create
			 * another nested level. For scalar:\n and plain
			 * scalar case we send it to next_mapping to
			 * create another nested level.
			 */
			tmp = strchr(line, ':');
			if (tmp) {
				fmt = LNKF_BLOCK;
				if (strstr(line, ": "))
					fmt |= LNKF_MAPPING;
				if (strstr(line, "- "))
					fmt |= LNKF_SEQUENCE;
				*hdr = line;
				goto next_mapping;
			}

			goto have_next_line;
		}
	}

	if (seq) {
		if (*indent >= 2)
			*indent -= 2;
		nla_nest_end(msg, seq);
		seq = NULL;
		if (*entry && !strlen(*entry) && fmt != LNKF_BLOCK)
			line = NULL;
	}

	/* strsep in the above loop moves entry to a value pass the end of the
	 * nested list. So to avoid losing this value we replace hdr with line.
	 */
	*hdr = line;
nla_put_failure:
	return rc;
}

/* YAML allows ' and " in its documents but those characters really
 * confuse libc string handling. The workaround is to replace
 * ' and " with another reserved character for YAML '%' which is
 * for tags which shouldn't matter if we send in a Netlink packet.
 * The kernel side will need to handle % in a special way.
 */
static void yaml_quotation_handling(char *buf)
{
	char *tmp = buf, *line;

	line = strstr(tmp, "! \'");
	if (line)
		line[0] = ' ';

	while ((line = strchr(tmp, '\"')) != NULL) {
		line[0] = ' ';
		tmp = strchr(line, '\"');
		tmp[0] = ' ';
	}

	while ((line = strchr(tmp, '\'')) != NULL) {
		line[0] = ' ';
		tmp = strchr(line, '\'');
		tmp[0] = ' ';
	}
}

/**
 * Filter Netlink socket by groups
 *
 * @out		Data structure for YAML write handler.
 * @family	The family name of the Netlink socket.
 * @group	Netlink messages will only been sent if they belong to this
 *		group
 *
 * Return	0 on success or a negative error code.
 */
static int lustre_netlink_add_group(struct yaml_netlink_output *out,
				    const char *group)
{
	int group_id;

	/* Get group ID */
	group_id = genl_ctrl_resolve_grp(out->ctrl, out->family, group);
	if (group_id < 0)
		return group_id;

	/* subscribe to generic netlink multicast group */
	return nl_socket_add_membership(out->nl, group_id);
}

/* libyaml takes the YAML documents and places the data into an
 * internal buffer to the library. We take each line and turn it
 * into a Netlink message using the same format as the key table.
 * The reason for this approach is that we can do filters at the
 * key level or the key + value level.
 */
static int yaml_netlink_write_handler(void *data, unsigned char *buffer,
				      size_t size)
{
	struct yaml_netlink_output *out = data;
	char *buf = strndup((char *)buffer, size);
	char *entry = buf, *tmp = buf, *line;
	enum lnet_nl_key_format fmt = 0;
	struct nl_msg *msg = NULL;
	unsigned int indent = 0;
	bool nogroups = true;
	int rc = 0;

	yaml_quotation_handling(entry);

	while (entry && strcmp(line = strsep(&entry, "\n"), "")) {
already_have_line:
		if (strcmp(line, "---") == 0 || strcmp(line, "...") == 0)
			continue;

		/* In theory we could have a sequence of groups but a bug in
		 * libyaml prevents this from happing
		 */
		if (line[0] != ' ' && line[0] != '-') {
			bool extra = false;

			if (strchr(line, '{') || strchr(line, '['))
				extra = true;

			tmp = strchr(line, ':');
			if (!tmp)
				continue;
			*tmp = '\0';

			rc = lustre_netlink_add_group(out, line);
			if (rc < 0) {
				yaml_emitter_set_writer_error(out->emitter,
							      "Netlink group does not exist");
				goto nla_put_failure;
			}
			nogroups = false;
			/* Handle case first line contains more than a
			 * simple key
			 */
			if (extra) {
				*tmp = ' ';
				line = tmp;
				goto already_have_line;
			}
		} else {
			if (!msg) {
				void *usr_hdr;

				msg = nlmsg_alloc();
				if (!msg) {
					out->emitter->error = YAML_MEMORY_ERROR;
					goto nla_put_failure;
				}

				usr_hdr = genlmsg_put(msg, out->pid,
						      NL_AUTO_SEQ,
						      out->family_id, 0,
						      out->flags, out->cmd,
						      out->version);
				if (!usr_hdr) {
					out->emitter->error = YAML_MEMORY_ERROR;
					nlmsg_free(msg);
					goto nla_put_failure;
				}

				if (line[0] != '-')
					indent = 2;
			}

			fmt = yaml_format_type(out->emitter, line, &indent);
			if (fmt) {
				rc = yaml_create_nested_list(out, msg, &line,
							     &entry, &indent,
							     fmt);
				if (rc < 0) {
					yaml_emitter_set_writer_error(out->emitter,
								      nl_geterror(rc));
					nlmsg_free(msg);
					goto nla_put_failure;
				}
				/* yaml_create_nested_list set line to the next
				 * entry. We can just add it to the msg directly.
				 */
				if (line)
					goto already_have_line;
			} else {
				rc = yaml_fill_scalar_data(msg, fmt,
							   line + indent);
				if (rc < 0) {
					yaml_emitter_set_writer_error(out->emitter,
								      nl_geterror(rc));
					nlmsg_free(msg);
					goto nla_put_failure;
				}
			}
		}
	}

	/* Don't success if no valid groups found */
	if (nogroups) {
		yaml_emitter_set_writer_error(out->emitter,
					      "Emitter contains no valid Netlink groups");
		goto nla_put_failure;
	}

	if (msg) {
		rc = nl_send_auto(out->nl, msg);
		nlmsg_free(msg);
	} else {
		rc = genl_send_simple(out->nl, out->family_id, out->cmd,
				      out->version, out->flags);
	}
	if (rc < 0)
		yaml_emitter_set_writer_error(out->emitter,
					      nl_geterror(rc));
nla_put_failure:
	if (out->ctrl != out->nl)
		nl_socket_free(out->ctrl);
	free(buf);
	return out->emitter->error == YAML_NO_ERROR ? 1 : 0;
}

/* This function is used by external utilities to use Netlink with
 * libyaml so we can turn YAML documentations into Netlink message
 * to send. This behavior mirrors yaml_emitter_set_output_file()
 * which is used to write out a YAML document to a file.
 */
YAML_DECLARE(int)
yaml_emitter_set_streaming_output_netlink(yaml_emitter_t *sender,
					  struct nl_sock *nl, char *family,
					  int version, int cmd, int flags,
					  bool stream)
{
	struct yaml_netlink_output *out;

	out = calloc(1, sizeof(*out));
	if (!out) {
		sender->error = YAML_MEMORY_ERROR;
		return false;
	}

	/* All because RHEL7 is really too old. Once we drop RHEL7
	 * this hack can go away.
	 */
	if (stream) {
		out->ctrl = nl_socket_alloc();
		if (!out->ctrl) {
			sender->problem = "socket allocation failed";
			sender->error = YAML_MEMORY_ERROR;
			free(out);
			return false;
		}

		if (genl_connect(out->ctrl) < 0) {
			yaml_emitter_set_writer_error(sender,
						      "socket failed to connect");
			nl_socket_free(out->ctrl);
			free(out);
			return false;
		}
	} else {
		out->ctrl = nl;
	}

	/* Get family ID */
	out->family_id = genl_ctrl_resolve(out->ctrl, family);
	if (out->family_id < 0) {
		yaml_emitter_set_writer_error(sender,
					      "failed to resolve Netlink family id");
		if (stream)
			nl_socket_free(out->ctrl);
		free(out);
		return false;
	}

	out->emitter = sender;
	out->nl = nl;
	out->family = family;
	out->version = version;
	out->cmd = cmd;
	out->flags = flags;
	out->pid = nl_socket_get_local_port(nl);
	yaml_emitter_set_output(sender, yaml_netlink_write_handler, out);
	return true;
}

/**
 * yaml_emitter_set_output_netlink() - Set output to Netlink socket and not any
 * YAML document. This is wrapper to yaml_emitter_set_streaming_output_netlink()
 * @sender: emitter object
 * @nl: netlink socket
 * @family: name of socket (lnet)
 * @version: version number
 * @cmd: command identifier
 * @flags: netlink flags (NLM_F_* under liblnetconfig.h)
 *
 * Note: Emitter object setup with this function should always call
 * yaml_emitter_cleanup() and not yaml_emitter_delete(). Otherwise it will
 * result in memory leak
 *
 * Return TRUE on success and FALSE on failure
 */
YAML_DECLARE(int)
yaml_emitter_set_output_netlink(yaml_emitter_t *sender, struct nl_sock *nl,
				char *family, int version, int cmd, int flags)
{
	return yaml_emitter_set_streaming_output_netlink(sender, nl, family,
							 version, cmd, flags,
							 false);
}

/* Error handling helpers */
void yaml_emitter_log_error(yaml_emitter_t *emitter, FILE *log)
{
	/* YAML_WRITER_ERROR means no Netlink support so use old API */
	switch (emitter->error) {
	case YAML_MEMORY_ERROR:
		fprintf(log, "Memory error: Not enough memory for emitting\n");
			break;
	case YAML_WRITER_ERROR:
		fprintf(log, "Writer error: %s\n", emitter->problem);
		break;
	case YAML_EMITTER_ERROR:
		fprintf(log, "Emitter error: %s\n", emitter->problem);
	default:
		break;
	}
}

/*
 * yaml_emitter_cleanup - Cleanup request & all memory held by request
 */
void yaml_emitter_cleanup(yaml_emitter_t *request)
{
	struct yaml_netlink_output *out = NULL;

	if (!request || !request->write_handler_data)
		return;

	out = request->write_handler_data;

	/* first destroy emitter */
	yaml_emitter_delete(request);

	if (out)
		free(out);
}

/*
 * yaml_parser_cleanup - Cleanup parser & all memory held by parser
 */
void yaml_parser_cleanup(yaml_parser_t *reply)
{
	struct yaml_netlink_input *input = NULL;

	if (!reply || !reply->read_handler_data)
		return;

	input = reply->read_handler_data;

	/* delete parser first */
	yaml_parser_delete(reply);

	if (input) {
		if (input->start) {
			free(input->start);
			input->start = NULL;
		}

		free(input);
	}
}

void yaml_parser_log_error(yaml_parser_t *parser, FILE *log, const char *errmsg)
{
	const char *extra;

	switch (parser->error) {
	case YAML_MEMORY_ERROR:
		fprintf(log, "Memory error: Not enough memory for parser\n");
		break;

	case YAML_SCANNER_ERROR:
	case YAML_PARSER_ERROR:
		if (parser->context) {
			fprintf(log,
				"%s error: %s at line %d, column %d\n%s at line %d, column %d\n",
				parser->error == YAML_SCANNER_ERROR ? "Scanner" : "Parser",
				parser->context,
				(int)parser->context_mark.line + 1,
				(int)parser->context_mark.column + 1,
				parser->problem,
				(int)parser->problem_mark.line + 1,
				(int)parser->problem_mark.column + 1);
		} else {
			fprintf(log, "%s error: %s at line %d, column %d\n",
				parser->error == YAML_SCANNER_ERROR ? "Scanner" : "Parser",
				parser->problem,
				(int)parser->problem_mark.line + 1,
				(int)parser->problem_mark.column + 1);
		}
		break;

	case YAML_READER_ERROR:
		extra = yaml_parser_get_reader_error(parser);
		if (!extra)
			extra = parser->problem;

		if (parser->problem_value != -1) {
			fprintf(log, "Reader error: '%s':#%X at %ld'\n",
				extra, parser->problem_value,
				(long)parser->problem_offset);
		} else {
			fprintf(log, "Reader error: '%s' at %ld\n",
				extra, (long)parser->problem_offset);
		}
		/* fallthrough */
	default:
		break;
	}
}