/*
 * Copyright 2015 Red Hat Inc., Durham, North Carolina.
 * All Rights Reserved.
 *
 * This library counter 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 library counter 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.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 *
 * Authors:
 *      "Jan Černý" <jcerny@redhat.com>
 */

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <string.h>
#include "../common/util.h"
#include "oval_system_characteristics.h"
#include "results/oval_status_counter.h"
#include "common/_error.h"

void oval_status_counter_clear(struct oval_status_counter *counter)
{
	memset(counter, 0, sizeof(*counter));
}

void oval_status_counter_add_status(struct oval_status_counter *counter, oval_syschar_status_t status)
{
	switch(status) {
	case SYSCHAR_STATUS_ERROR:
		counter->error_cnt++;
		break;
	case SYSCHAR_STATUS_EXISTS:
		counter->exists_cnt++;
		break;
	case SYSCHAR_STATUS_DOES_NOT_EXIST:
		counter->does_not_exist_cnt++;
		break;
	case SYSCHAR_STATUS_NOT_COLLECTED:
		counter->not_collected_cnt++;
		break;
	default:
		oscap_seterr(OSCAP_EFAMILY_OVAL, "Invalid oval status type: %s.", oval_syschar_status_get_text(status));
		break;
	}
}

oval_result_t oval_status_counter_get_result(struct oval_status_counter *counter, oval_existence_t check_existence)
{
	oval_result_t result = OVAL_RESULT_UNKNOWN;
	switch (check_existence) {
	case OVAL_ALL_EXIST:
		if (counter->exists_cnt >= 1 && counter->does_not_exist_cnt == 0
			&& counter->error_cnt == 0 && counter->not_collected_cnt == 0) {
			result = OVAL_RESULT_TRUE;
		} else if (counter->exists_cnt == 0 && counter->does_not_exist_cnt == 0
			&& counter->error_cnt == 0 && counter->not_collected_cnt == 0) {
			result = OVAL_RESULT_FALSE;
		} else if (counter->exists_cnt >= 0 && counter->does_not_exist_cnt >= 1
			&& counter->error_cnt >= 0 && counter->not_collected_cnt >= 0) {
			result = OVAL_RESULT_FALSE;
		} else if (counter->exists_cnt >= 0 && counter->does_not_exist_cnt == 0
			&& counter->error_cnt >= 1 && counter->not_collected_cnt >= 0) {
			result = OVAL_RESULT_ERROR;
		} else if (counter->exists_cnt >= 0 && counter->does_not_exist_cnt == 0
			&& counter->error_cnt == 0 && counter->not_collected_cnt >= 1) {
			result = OVAL_RESULT_UNKNOWN;
		}
		break;
	case OVAL_ANY_EXIST:
		if (counter->exists_cnt >= 0 && counter->does_not_exist_cnt >= 0
			&& counter->error_cnt == 0 && counter->not_collected_cnt >= 0) {
			result = OVAL_RESULT_TRUE;
		} else if (counter->exists_cnt >= 1 && counter->does_not_exist_cnt >= 0
			&& counter->error_cnt >=1 && counter->not_collected_cnt >= 0) {
			result = OVAL_RESULT_TRUE;
		} else if (counter->exists_cnt == 0 && counter->does_not_exist_cnt >= 0
			&& counter->error_cnt >=1 && counter->not_collected_cnt >= 0) {
			result = OVAL_RESULT_ERROR;
		}
		break;
	case OVAL_AT_LEAST_ONE_EXISTS:
		if (counter->exists_cnt >= 1 && counter->does_not_exist_cnt >= 0
			&& counter->error_cnt >= 0 && counter->not_collected_cnt >= 0) {
			result = OVAL_RESULT_TRUE;
		} else if (counter->exists_cnt >= 1 && counter->does_not_exist_cnt == 0
			&& counter->error_cnt == 0 && counter->not_collected_cnt == 0) {
			result = OVAL_RESULT_FALSE;
		} else if (counter->exists_cnt == 0 && counter->does_not_exist_cnt >= 0
			&& counter->error_cnt >= 1 && counter->not_collected_cnt >= 0) {
			result = OVAL_RESULT_ERROR;
		} else if (counter->exists_cnt == 0 && counter->does_not_exist_cnt >= 0
			&& counter->error_cnt == 0 && counter->not_collected_cnt >= 1) {
			result = OVAL_RESULT_UNKNOWN;
		}
		break;
	case OVAL_NONE_EXIST:
		if (counter->exists_cnt == 0 && counter->does_not_exist_cnt >= 0
			&& counter->error_cnt == 0 && counter->not_collected_cnt == 0) {
			result = OVAL_RESULT_TRUE;
		} else if (counter->exists_cnt >= 1 && counter->does_not_exist_cnt >= 0
			&& counter->error_cnt >= 0 && counter->not_collected_cnt >= 0) {
			result = OVAL_RESULT_FALSE;
		} else if (counter->exists_cnt == 0 && counter->does_not_exist_cnt >= 0
			&& counter->error_cnt >= 1 && counter->not_collected_cnt >= 0) {
			result = OVAL_RESULT_ERROR;
		} else if (counter->exists_cnt == 0 && counter->does_not_exist_cnt >= 0
			&& counter->error_cnt == 0 && counter->not_collected_cnt >= 1) {
			result = OVAL_RESULT_UNKNOWN;
		}
		break;
	case OVAL_ONLY_ONE_EXISTS:
		if (counter->exists_cnt == 1 && counter->does_not_exist_cnt >= 0
			&& counter->error_cnt == 0 && counter->not_collected_cnt == 0) {
			result = OVAL_RESULT_TRUE;
		} else if (counter->exists_cnt >= 2 && counter->does_not_exist_cnt >= 0
			&& counter->error_cnt >= 0 && counter->not_collected_cnt >= 0) {
			result = OVAL_RESULT_FALSE;
		} else if (counter->exists_cnt == 0 && counter->does_not_exist_cnt >= 0
			&& counter->error_cnt == 0 && counter->not_collected_cnt == 1) {
			result = OVAL_RESULT_FALSE;
		} else if ((counter->exists_cnt == 0 || counter->exists_cnt == 1)
			&& counter->does_not_exist_cnt >= 0 && counter->error_cnt >= 1
			&& counter->not_collected_cnt >= 0) {
			result = OVAL_RESULT_ERROR;
		} else if ((counter->exists_cnt == 0 || counter->exists_cnt == 1)
			&& counter->does_not_exist_cnt >= 0 && counter->error_cnt == 0
			&& counter->not_collected_cnt >= 1) {
			result = OVAL_RESULT_ERROR;
		}
		break;
	default:
		oscap_seterr(OSCAP_EFAMILY_OVAL, "Invalid check_existence value: %s.",
				oval_existence_get_text(check_existence));
		result = OVAL_RESULT_ERROR;
		break;
	}
	return result;
}