/*

  Copyright(c) 2014-2015 Intel Corporation

  All rights reserved.

  This library is free software; you can redistribute it and/or modify

  it under the terms of the GNU Lesser General Public License as

  published by the Free Software Foundation; either version 2.1 of

  the License, or (at your option) any later version.

  This program is distributed in the hope that it will be useful,

  but WITHOUT ANY WARRANTY; without even the implied warranty of

  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

  GNU Lesser General Public License for more details.

  Authors: Mengdong Lin <mengdong.lin@intel.com>

           Yao Jin <yao.jin@intel.com>

           Liam Girdwood <liam.r.girdwood@linux.intel.com>

*/

#include "list.h"

#include "tplg_local.h"

#include <ctype.h>

#define UUID_FORMAT "\

%02x:%02x:%02x:%02x:%02x:%02x:%02x:%02x:\

%02x:%02x:%02x:%02x:%02x:%02x:%02x:%02x"

/* Get private data buffer of an element */

struct snd_soc_tplg_private *get_priv_data(struct tplg_elem *elem)

{

	struct snd_soc_tplg_private *priv = NULL;

	switch (elem->type) {

	case SND_TPLG_TYPE_MANIFEST:

		priv = &elem->manifest->priv;

		break;

	case SND_TPLG_TYPE_MIXER:

		priv = &elem->mixer_ctrl->priv;

		break;

	case SND_TPLG_TYPE_ENUM:

		priv = &elem->enum_ctrl->priv;

		break;

	case SND_TPLG_TYPE_BYTES:

		priv = &elem->bytes_ext->priv;

		break;

	case SND_TPLG_TYPE_DAPM_WIDGET:

		priv = &elem->widget->priv;

		break;

	case SND_TPLG_TYPE_DAI:

		priv = &elem->dai->priv;

		break;

	case SND_TPLG_TYPE_BE:

		priv = &elem->link->priv;

		break;

	case SND_TPLG_TYPE_PCM:

		priv = &elem->pcm->priv;

		break;

	default:

		SNDERR("'%s': no support for private data for type %d",

			elem->id, elem->type);

	}

	return priv;

}

/* Parse references for the element, either a single data section

 * or a list of data sections.

 */

int tplg_parse_refs(snd_config_t *cfg, struct tplg_elem *elem,

		    unsigned int type)

{

	snd_config_type_t cfg_type;

	snd_config_iterator_t i, next;

	snd_config_t *n;

	const char *val = NULL;

	int err, count;

	cfg_type = snd_config_get_type(cfg);

	/* refer to a single data section */

	if (cfg_type == SND_CONFIG_TYPE_STRING) {

		if (snd_config_get_string(cfg, &val) < 0)

			return -EINVAL;

		tplg_dbg("\tref data: %s", val);

		err = tplg_ref_add(elem, type, val);

		if (err < 0)

			return err;

		return 1;

	}

	if (cfg_type != SND_CONFIG_TYPE_COMPOUND) {

		SNDERR("compound type expected for %s", elem->id);

		return -EINVAL;

	}

	/* refer to a list of data sections */

	count = 0;

	snd_config_for_each(i, next, cfg) {

		const char *val;

		n = snd_config_iterator_entry(i);

		if (snd_config_get_string(n, &val) < 0)

			continue;

		tplg_dbg("\tref data: %s", val);

		err = tplg_ref_add(elem, type, val);

		if (err < 0)

			return err;

		count++;

	}

	return count;

}

/* save references */

int tplg_save_refs(snd_tplg_t *tplg ATTRIBUTE_UNUSED,

		   struct tplg_elem *elem, unsigned int type,

		   const char *id, char **dst, const char *pfx)

{

	struct tplg_ref *ref, *last;

	struct list_head *pos;

	int err, count;

	count = 0;

	last = NULL;

	list_for_each(pos, &elem->ref_list) {

		ref = list_entry(pos, struct tplg_ref, list);

		if (ref->type == type) {

			last = ref;

			count++;

		}

	}

	if (count == 0)

		return 0;

	if (count == 1)

		return tplg_save_printf(dst, pfx, "%s '%s'\n", id, last->id);

	err = tplg_save_printf(dst, pfx, "%s [\n", id);

	if (err < 0)

		return err;

	list_for_each(pos, &elem->ref_list) {

		ref = list_entry(pos, struct tplg_ref, list);

		if (ref->type == type) {

			err = tplg_save_printf(dst, pfx, "\t'%s'\n", ref->id);

			if (err < 0)

				return err;

		}

	}

	return tplg_save_printf(dst, pfx, "]\n");

}

/* Get Private data from a file. */

static int tplg_parse_data_file(snd_config_t *cfg, struct tplg_elem *elem)

{

	struct snd_soc_tplg_private *priv = NULL;

	const char *value = NULL;

	char filename[PATH_MAX];

	char *env = getenv(ALSA_CONFIG_TPLG_VAR);

	FILE *fp;

	size_t size, bytes_read;

	int ret = 0;

	tplg_dbg("data DataFile: %s", elem->id);

	if (snd_config_get_string(cfg, &value) < 0)

		return -EINVAL;

	/* prepend alsa config directory to path */

	if (env)

		snprintf(filename, sizeof(filename), "%s/%s", env, value);

	else

		snprintf(filename, sizeof(filename), "%s/topology/%s",

			 snd_config_topdir(), value);

	fp = fopen(filename, "r");

	if (fp == NULL) {

		SNDERR("invalid data file path '%s'", filename);

		return -errno;

	}

	fseek(fp, 0L, SEEK_END);

	size = ftell(fp);

	fseek(fp, 0L, SEEK_SET);

	if (size <= 0) {

		SNDERR("invalid data file size %zu", size);

		ret = -EINVAL;

		goto err;

	}

	if (size > TPLG_MAX_PRIV_SIZE) {

		SNDERR("data file too big %zu", size);

		ret = -EINVAL;

		goto err;

	}

	priv = calloc(1, sizeof(*priv) + size);

	if (!priv) {

		ret = -ENOMEM;

		goto err;

	}

	bytes_read = fread(&priv->data, 1, size, fp);

	if (bytes_read != size) {

		ret = -errno;

		goto err;

	}

	elem->data = priv;

	priv->size = size;

	elem->size = sizeof(*priv) + size;

	if (fclose(fp) == EOF) {

		SNDERR("Cannot close data file.");

		return -errno;

	}

	return 0;

err:

	fclose(fp);

	if (priv)

		free(priv);

	return ret;

}

static void dump_priv_data(struct tplg_elem *elem ATTRIBUTE_UNUSED)

{

#ifdef TPLG_DEBUG

	struct snd_soc_tplg_private *priv = elem->data;

	unsigned char *p = (unsigned char *)priv->data;

	char buf[128], buf2[8];

	unsigned int i;

	tplg_dbg(" elem size = %d, priv data size = %d",

		elem->size, priv->size);

	buf[0] = '\0';

	for (i = 0; i < priv->size; i++) {

		if (i > 0 && (i % 16) == 0) {

			tplg_dbg("%s", buf);

			buf[0] = '\0';

		}

		snprintf(buf2, sizeof(buf2), " %02x", *p++);

		strcat(buf, buf2);

	}

	if (buf[0])

		tplg_dbg("%s", buf);

#endif

}

static inline int check_nibble(unsigned char c)

{

	return (c >= '0' && c <= '9') ||

	       (c >= 'a' && c <= 'f') ||

	       (c >= 'A' && c <= 'F');

}

/* get number of hex value elements in CSV list */

static int get_hex_num(const char *str)

{

	int delims, values, len = strlen(str);

	const char *s, *end = str + len;

	/* check "aa:bb:00" syntax */

	s = str;

	delims = values = 0;

	while (s < end) {

		/* skip white space */

		if (isspace(*s)) {

			s++;

			continue;

		}

		/* find delimeters */

		if (*s == ':') {

			delims++;

			s++;

			continue;

		}

		/* check 00 hexadecimal value */

		if (s + 1 <= end) {

			if (check_nibble(s[0]) && check_nibble(s[1])) {

				values++;

			} else {

				goto format2;

			}

			s++;

		}

		s++;

	}

	goto end;

format2:

	/* we expect "0x0, 0x0, 0x0" */

	s = str;

	delims = values = 0;

	while (s < end) {

		/* skip white space */

		if (isspace(*s)) {

			s++;

			continue;

		}

		/* find delimeters */

		if (*s == ',') {

			delims++;

			s++;

			continue;

		}

		/* find 0x[0-9] values */

		if (*s == '0' && s + 2 <= end) {

			if (s[1] == 'x' && check_nibble(s[2])) {

				if (check_nibble(s[3]))

					s++;

				values++;

				s += 2;

			}

		}

		s++;

	}

end:

	/* there should always be one less comma than value */

	if (values - 1 != delims)

		return -EINVAL;

	return values;

}

/* get uuid from a string made by 16 characters separated by commas */

static int get_uuid(const char *str, unsigned char *uuid_le)

{

	unsigned long int  val;

	char *tmp, *s = NULL;

	int values = 0, ret = 0;

	tmp = strdup(str);

	if (tmp == NULL)

		return -ENOMEM;

	if (strchr(tmp, ':') == NULL)

		goto data2;

	s = strtok(tmp, ":");

	while (s != NULL) {

		errno = 0;

		val = strtoul(s, NULL, 16);

		if ((errno == ERANGE && val == ULONG_MAX)

			|| (errno != 0 && val == 0)

			|| (val > UCHAR_MAX)) {

			SNDERR("invalid value for uuid");

			ret = -EINVAL;

			goto out;

		}

		*(uuid_le + values) = (unsigned char)val;

		values++;

		if (values >= 16)

			break;

		s = strtok(NULL, ":");

	}

	goto out;

data2:

	s = strtok(tmp, ",");

	while (s != NULL) {

		errno = 0;

		val = strtoul(s, NULL, 0);

		if ((errno == ERANGE && val == ULONG_MAX)

			|| (errno != 0 && val == 0)

			|| (val > UCHAR_MAX)) {

			SNDERR("invalid value for uuid");

			ret = -EINVAL;

			goto out;

		}

		*(uuid_le + values) = (unsigned char)val;

		values++;

		if (values >= 16)

			break;

		s = strtok(NULL, ",");

	}

	if (values < 16) {

		SNDERR("less than 16 integers for uuid");

		ret = -EINVAL;

	}

out:

	free(tmp);

	return ret;

}

static int write_hex(char *buf, char *str, int width)

{

	long val;

	void *p = &val;

        errno = 0;

	val = strtol(str, NULL, 16);

	if ((errno == ERANGE && (val == LONG_MAX || val == LONG_MIN))

		|| (errno != 0 && val == 0)) {

		return -EINVAL;

        }

	switch (width) {

	case 1:

		*(unsigned char *)buf = *(unsigned char *)p;

		break;

	case 2:

		*(unsigned short *)buf = *(unsigned short *)p;

		break;

	case 4:

		*(unsigned int *)buf = *(unsigned int *)p;

		break;

	default:

		return -EINVAL;

	}

	return 0;

}

static int copy_data_hex(char *data, int off, const char *str, int width)

{

	char *tmp, *s = NULL, *p = data;

	int ret;

	tmp = strdup(str);

	if (tmp == NULL)

		return -ENOMEM;

	p += off;

	s = strtok(tmp, ",:");

	while (s != NULL) {

		ret = write_hex(p, s, width);

		if (ret < 0) {

			free(tmp);

			return ret;

		}

		s = strtok(NULL, ",:");

		p += width;

	}

	free(tmp);

	return 0;

}

static int tplg_parse_data_hex(snd_config_t *cfg, struct tplg_elem *elem,

	int width)

{

	struct snd_soc_tplg_private *priv;

	const char *value = NULL;

	int size, esize, off, num;

	int ret;

	tplg_dbg(" data: %s", elem->id);

	if (snd_config_get_string(cfg, &value) < 0)

		return -EINVAL;

	num = get_hex_num(value);

	if (num <= 0) {

		SNDERR("malformed hex variable list %s", value);

		return -EINVAL;

	}

	size = num * width;

	priv = elem->data;

	if (size > TPLG_MAX_PRIV_SIZE) {

		SNDERR("data too big %d", size);

		return -EINVAL;

	}

	if (priv != NULL) {

		off = priv->size;

		esize = elem->size + size;

		priv = realloc(priv, esize);

	} else {

		off = 0;

		esize = sizeof(*priv) + size;

		priv = calloc(1, esize);

	}

	if (!priv)

		return -ENOMEM;

	elem->data = priv;

	priv->size += size;

	elem->size = esize;

	ret = copy_data_hex(priv->data, off, value, width);

	dump_priv_data(elem);

	return ret;

}

/* get the token integer value from its id */

static int get_token_value(const char *token_id,

			   struct tplg_vendor_tokens *tokens)

{

	unsigned int i;

	for (i = 0; i < tokens->num_tokens; i++) {

		if (strcmp(token_id, tokens->token[i].id) == 0)

			return tokens->token[i].value;

	}

	SNDERR("cannot find token id '%s'", token_id);

	return -1;

}

/* get the vendor tokens referred by the vendor tuples */

static struct tplg_elem *get_tokens(snd_tplg_t *tplg, struct tplg_elem *elem)

{

	struct tplg_ref *ref;

	struct list_head *base, *pos;

	base = &elem->ref_list;

	list_for_each(pos, base) {

		ref = list_entry(pos, struct tplg_ref, list);

		if (ref->type != SND_TPLG_TYPE_TOKEN)

			continue;

		if (!ref->elem) {

			ref->elem = tplg_elem_lookup(&tplg->token_list,

				ref->id, SND_TPLG_TYPE_TOKEN, elem->index);

		}

		return ref->elem;

	}

	return NULL;

}

/* check if a data element has tuples */

static bool has_tuples(struct tplg_elem *elem)

{

	struct tplg_ref *ref;

	struct list_head *base, *pos;

	base = &elem->ref_list;

	list_for_each(pos, base) {

		ref = list_entry(pos, struct tplg_ref, list);

		if (ref->type == SND_TPLG_TYPE_TUPLE)

			return true;

	}

	return false;

}

/* get size of a tuple element from its type */

unsigned int tplg_get_tuple_size(int type)

{

	switch (type) {

	case SND_SOC_TPLG_TUPLE_TYPE_UUID:

		return sizeof(struct snd_soc_tplg_vendor_uuid_elem);

	case SND_SOC_TPLG_TUPLE_TYPE_STRING:

		return sizeof(struct snd_soc_tplg_vendor_string_elem);

	default:

		return sizeof(struct snd_soc_tplg_vendor_value_elem);

	}

}

/* Add a tuples object to the private buffer of its parent data element */

static int copy_tuples(struct tplg_elem *elem,

		       struct tplg_vendor_tuples *tuples,

		       struct tplg_vendor_tokens *tokens)

{

	struct snd_soc_tplg_private *priv = elem->data, *priv2;

	struct tplg_tuple_set *tuple_set;

	struct tplg_tuple *tuple;

	struct snd_soc_tplg_vendor_array *array;

	struct snd_soc_tplg_vendor_uuid_elem *uuid;

	struct snd_soc_tplg_vendor_string_elem *string;

	struct snd_soc_tplg_vendor_value_elem *value;

	int set_size, size, off;

	unsigned int i, j;

	int token_val;

	size = priv ? priv->size : 0; /* original private data size */

	/* scan each tuples set (one set per type) */

	for (i = 0; i < tuples->num_sets ; i++) {

		tuple_set = tuples->set[i];

		set_size = sizeof(struct snd_soc_tplg_vendor_array)

			+ tplg_get_tuple_size(tuple_set->type)

			* tuple_set->num_tuples;

		size += set_size;

		if (size > TPLG_MAX_PRIV_SIZE) {

			SNDERR("data too big %d", size);

			return -EINVAL;

		}

		if (priv != NULL) {

			priv2 = realloc(priv, sizeof(*priv) + size);

			if (priv2 == NULL) {

				free(priv);

				priv = NULL;

			} else {

				priv = priv2;

			}

		} else {

			priv = calloc(1, sizeof(*priv) + size);

		}

		if (!priv)

			return -ENOMEM;

		off = priv->size;

		priv->size = size; /* update private data size */

		elem->data = priv;

		array = (struct snd_soc_tplg_vendor_array *)(priv->data + off);

		memset(array, 0, set_size);

		array->size = set_size;

		array->type = tuple_set->type;

		array->num_elems = tuple_set->num_tuples;

		/* fill the private data buffer */

		for (j = 0; j < tuple_set->num_tuples; j++) {

			tuple = &tuple_set->tuple[j];

			token_val = get_token_value(tuple->token, tokens);

			if (token_val  < 0)

				return -EINVAL;

			switch (tuple_set->type) {

			case SND_SOC_TPLG_TUPLE_TYPE_UUID:

				uuid = &array->uuid[j];

				uuid->token = token_val;

				memcpy(uuid->uuid, tuple->uuid, 16);

				break;

			case SND_SOC_TPLG_TUPLE_TYPE_STRING:

				string = &array->string[j];

				string->token = token_val;

				snd_strlcpy(string->string, tuple->string,

					SNDRV_CTL_ELEM_ID_NAME_MAXLEN);

				break;

			default:

				value = &array->value[j];

				value->token = token_val;

				value->value = tuple->value;

				break;

			}

		}

	}

	return 0;

}

/* build a data element from its tuples */

static int build_tuples(snd_tplg_t *tplg, struct tplg_elem *elem)

{

	struct tplg_ref *ref;

	struct list_head *base, *pos;

	struct tplg_elem *tuples, *tokens;

	int err;

	base = &elem->ref_list;

	list_for_each(pos, base) {

		ref = list_entry(pos, struct tplg_ref, list);

		if (ref->type != SND_TPLG_TYPE_TUPLE)

			continue;

		tplg_dbg("tuples '%s' used by data '%s'", ref->id, elem->id);

		if (!ref->elem)

			ref->elem = tplg_elem_lookup(&tplg->tuple_list,

				ref->id, SND_TPLG_TYPE_TUPLE, elem->index);

		tuples = ref->elem;

		if (!tuples) {

			SNDERR("cannot find tuples %s", ref->id);

			return -EINVAL;

		}

		tokens = get_tokens(tplg, tuples);

		if (!tokens) {

			SNDERR("cannot find token for %s", ref->id);

			return -EINVAL;

		}

		/* a data object can have multiple tuples objects */

		err = copy_tuples(elem, tuples->tuples, tokens->tokens);

		if (err < 0)

			return err;

	}

	return 0;

}

struct tuple_type {

	unsigned int type;

	const char *name;

	unsigned int size;

};

static struct tuple_type tuple_types[] = {

	{

		.type = SND_SOC_TPLG_TUPLE_TYPE_UUID,

		.name = "uuid",

		.size = 4,

	},

	{

		.type = SND_SOC_TPLG_TUPLE_TYPE_STRING,

		.name = "string",

		.size = 6,

	},

	{

		.type = SND_SOC_TPLG_TUPLE_TYPE_BOOL,

		.name = "bool",

		.size = 4,

	},

	{

		.type = SND_SOC_TPLG_TUPLE_TYPE_BYTE,

		.name = "byte",

		.size = 4,

	},

	{

		.type = SND_SOC_TPLG_TUPLE_TYPE_SHORT,

		.name = "short",

		.size = 5,

	},

	{

		.type = SND_SOC_TPLG_TUPLE_TYPE_WORD,

		.name = "word",

		.size = 4

	},

};

static int get_tuple_type(const char *name)

{

	struct tuple_type *t;

	unsigned int i;

	/* skip initial index for sorting */

	while ((*name >= '0' && *name <= '9') || *name == '_')

		name++;

	for (i = 0; i < ARRAY_SIZE(tuple_types); i++) {

		t = &tuple_types[i];

		if (strncasecmp(t->name, name, t->size) == 0)

			return t->type;

	}

	return -EINVAL;

}

static const char *get_tuple_type_name(unsigned int type)

{

	unsigned int i;

	for (i = 0; i < ARRAY_SIZE(tuple_types); i++)

		if (tuple_types[i].type == type)

			return tuple_types[i].name;

	return NULL;

}

static int parse_tuple_set(snd_config_t *cfg,

			   struct tplg_tuple_set **s)

{

	snd_config_iterator_t i, next;

	snd_config_t *n;

	const char *id, *value;

	struct tplg_tuple_set *set;

	unsigned int num_tuples = 0;

	struct tplg_tuple *tuple;

	unsigned int tuple_val;

	int type, ival;

	snd_config_get_id(cfg, &id;;

	type = get_tuple_type(id);

	if (type < 0) {

		SNDERR("invalid tuple type '%s'", id);

		return type;

	}

	snd_config_for_each(i, next, cfg)

		num_tuples++;

	if (!num_tuples)

		return 0;

	tplg_dbg("\t %d %s tuples:", num_tuples, id);

	set = calloc(1, sizeof(*set) + num_tuples * sizeof(struct tplg_tuple));

	if (!set)

		return -ENOMEM;

	set->type = type;

	snd_config_for_each(i, next, cfg) {

		n = snd_config_iterator_entry(i);

		/* get id */

		if (snd_config_get_id(n, &id) < 0)

			continue;

		tuple = &set->tuple[set->num_tuples];

		snd_strlcpy(tuple->token, id,

				SNDRV_CTL_ELEM_ID_NAME_MAXLEN);

		switch (type) {

		case SND_SOC_TPLG_TUPLE_TYPE_UUID:

			if (snd_config_get_string(n, &value) < 0)

				continue;

			if (get_uuid(value, tuple->uuid) < 0)

				goto err;

			break;

		case SND_SOC_TPLG_TUPLE_TYPE_STRING:

			if (snd_config_get_string(n, &value) < 0)

				continue;

			snd_strlcpy(tuple->string, value,

				SNDRV_CTL_ELEM_ID_NAME_MAXLEN);

			tplg_dbg("\t\t%s = %s", tuple->token, tuple->string);

			break;

		case SND_SOC_TPLG_TUPLE_TYPE_BOOL:

			ival = snd_config_get_bool(n);

			if (ival < 0)

				continue;

			tuple->value = ival;

			tplg_dbg("\t\t%s = %d", tuple->token, tuple->value);

			break;

		case SND_SOC_TPLG_TUPLE_TYPE_BYTE:

		case SND_SOC_TPLG_TUPLE_TYPE_SHORT:

		case SND_SOC_TPLG_TUPLE_TYPE_WORD:

			ival = tplg_get_unsigned(n, &tuple_val, 0);

			if (ival < 0) {

				SNDERR("tuple %s: %s", id, snd_strerror(ival));

				goto err;

			}

			if ((type == SND_SOC_TPLG_TUPLE_TYPE_WORD

					&& tuple_val > UINT_MAX)

				|| (type == SND_SOC_TPLG_TUPLE_TYPE_SHORT

					&& tuple_val > USHRT_MAX)

				|| (type == SND_SOC_TPLG_TUPLE_TYPE_BYTE

					&& tuple_val > UCHAR_MAX)) {

				SNDERR("tuple %s: invalid value", id);

				goto err;

			}

			tuple->value = tuple_val;

			tplg_dbg("\t\t%s = 0x%x", tuple->token, tuple->value);

			break;

		default:

			break;

		}

		set->num_tuples++;

	}