/*
 * Copyright (c) 2014, The Linux Foundation. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met:
 *     * Redistributions of source code must retain the above copyright
 *       notice, this list of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above
 *       copyright notice, this list of conditions and the following
 *       disclaimer in the documentation and/or other materials provided
 *       with the distribution.
 *     * Neither the name of The Linux Foundation nor the names of its
 *       contributors may be used to endorse or promote products derived
 *       from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESS OR IMPLIED
 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
 * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
 * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
 * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <unistd.h>
#include <stdio.h>

#include <linux/msm_ion.h>
#include "memory_prof.h"
#include "memory_prof_util.h"
#include "alloc_profiles.h"

struct ion_cache_op {
	int reps;
	unsigned int heap_id;
	char heap_id_string[MAX_HEAP_ID_STRING_LEN];
	unsigned int flags;
	char flags_string[MAX_FLAGS_STRING_LEN];
	unsigned long size;
	char size_string[MAX_SIZE_STRING_LEN];
	bool cache_clean;
	bool cache_inv;
};

enum ion_cache_op_line_idx {
	LINE_IDX_REPS = 1,
	LINE_IDX_HEAP_ID,
	LINE_IDX_FLAGS,
	LINE_IDX_ALLOC_SIZE,
	LINE_IDX_ALLOC_SIZE_LABEL,
	LINE_IDX_CACHE_INV,
	LINE_IDX_CACHE_CLEAN
};

static double *cache_stats;

static int op_ion_cache_global_setup(struct alloc_profile_entry entries[])
{
	int max_reps = 0;
	struct alloc_profile_entry *entry;
	/*
	 * Rather than malloc'ing and free'ing just enough space on
	 * each iteration, let's get the max needed up front. This
	 * will help keep things a little more equal during the actual
	 * timings.
	 */
	for_each_alloc_profile_entry_op(entry, entries, "ion_cache") {
		struct ion_cache_op *op = (struct ion_cache_op *) entry->priv;
		max_reps = MAX(max_reps, op->reps);
	}

	if (max_reps)
		MALLOC(double *, cache_stats, sizeof(double) * max_reps);
	return 0;
}

static void op_ion_cache_global_teardown(void)
{
	free(cache_stats);
}

static int op_ion_cache_parse(struct alloc_profile_entry *entry,
			struct line_info *li)
{
	unsigned int heap_id;
	struct ion_cache_op *op = (struct ion_cache_op *) entry->priv;
	char **words = li->words;

	STRTOL(op->reps, words[LINE_IDX_REPS], 0);

	if (find_heap_id_value(words[LINE_IDX_HEAP_ID], &heap_id)) {
		warnx("Unknown heap_id: %s", words[LINE_IDX_HEAP_ID]);
		return 1;
	}

	op->heap_id = ION_HEAP(heap_id);
	STRNCPY_SAFE(op->heap_id_string,
		words[LINE_IDX_HEAP_ID], MAX_HEAP_ID_STRING_LEN);

	op->flags = (unsigned int)parse_flags(words[LINE_IDX_FLAGS]);
	STRNCPY_SAFE(op->flags_string,
		words[LINE_IDX_FLAGS], MAX_FLAGS_STRING_LEN);

	if (parse_size_string(words[LINE_IDX_ALLOC_SIZE], &op->size)) {
		warnx("Couldn't parse alloc_size %s into a size",
			words[LINE_IDX_ALLOC_SIZE]);
		return 1;
	}
	STRNCPY_SAFE(op->size_string,
		words[LINE_IDX_ALLOC_SIZE_LABEL], MAX_SIZE_STRING_LEN);

	op->cache_clean = parse_bool(words[LINE_IDX_CACHE_CLEAN]);
	op->cache_inv = parse_bool(words[LINE_IDX_CACHE_INV]);

	return 0;
}

static int op_ion_cache_run(struct alloc_profile_entry *entry)
{
	int i;
	struct ion_cache_op *op = (struct ion_cache_op *) entry->priv;
	char sbuf[70];
	for (i = 0; i < op->reps; ++i) {
		int ret = profile_ion_cache_ops_for_heap(
			op->heap_id,
			op->flags,
			op->size,
			&cache_stats[i],
			op->cache_clean,
			op->cache_inv);
		if (ret)
			warn("Couldn't allocate %s from %s",
				op->size_string,
				op->heap_id_string);
	}
	if (op->cache_clean && op->cache_inv) {
		snprintf(sbuf, 70, "ION_IOC_CLEAN_INV_CACHES %s", op->heap_id_string);
		print_stats_results(sbuf, op->flags_string, op->size_string,
							cache_stats, op->reps);
	}
	else if (op->cache_clean) {
		snprintf(sbuf, 70, "ION_IOC_CLEAN_CACHES %s", op->heap_id_string);
		print_stats_results(sbuf, op->flags_string, op->size_string,
							cache_stats, op->reps);
	}
	else if (op->cache_inv) {
		snprintf(sbuf, 70, "ION_IOC_INV_CACHES %s", op->heap_id_string);
		print_stats_results(sbuf, op->flags_string, op->size_string,
							cache_stats, op->reps);
	}
	return 0;
}

static struct alloc_profile_ops ion_cache_ops = {
	.parse = op_ion_cache_parse,
	.run = op_ion_cache_run,
	.global_setup = op_ion_cache_global_setup,
	.global_teardown = op_ion_cache_global_teardown,
};

ALLOC_PROFILE_OP_SIZED(&ion_cache_ops, ion_cache, sizeof(struct ion_cache_op));