/*
 * Copyright 2010--2014 Red Hat Inc., Durham, North Carolina.
 * 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 library 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.
 *
 * 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:
 *      Peter Vrabec  <pvrabec@redhat.com>
 */

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

/* Standard header files */
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#ifdef HAVE_GETOPT_H
#include <getopt.h>
#endif
#include <assert.h>
#include <math.h>
#include <time.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <limits.h>

#include <oscap.h>
#include "oscap_source.h"
#include <xccdf_policy.h>
#include <oval_results.h>
#include <oval_variables.h>
#include <oval_system_characteristics.h>
#include <oval_directives.h>
#include <scap_ds.h>
#include <ds_rds_session.h>
#include <ds_sds_session.h>
#include <xccdf_session.h>

#include "oscap-tool.h"


static bool getopt_info(int argc, char **argv, struct oscap_action *action);
static int app_info(const struct oscap_action *action);

struct oscap_module OSCAP_INFO_MODULE = {
    .name = "info",
    .parent = &OSCAP_ROOT_MODULE,
	.summary = "Print information about a SCAP file.",
    .usage = "some-file.xml",
	.help = "Options:\n"
		"   --fetch-remote-resources      - Download remote content referenced by data stream.\n"
		"   --profile <id>                - Show info of the profile with the given ID.\n"
		"   --profiles                    - Show profiles from the input file in the <id>:<title> format, one line per profile.\n",
    .opt_parser = getopt_info,
    .func = app_info
};

static void print_time(const char *file) {
	struct stat buffer;
	char timeStr[ 100 ] = "";

	if(!stat(file, &buffer)) {
		strftime(timeStr, 100, "%Y-%m-%dT%H:%M:%S", localtime(&buffer.st_mtime));
		printf("Imported: %s\n", timeStr);
	}
}

static inline void _print_xccdf_status(struct xccdf_status *status, const char *prefix)
{
	if (status) {
		printf("%sStatus: %s\n", prefix, xccdf_status_type_to_text(xccdf_status_get_status(status)));
		const time_t date_time = xccdf_status_get_date(status);
		if (date_time != 0) {
			struct tm *date = localtime(&date_time);
			char date_str[] = "YYYY-DD-MM";
			int ret = snprintf(date_str, sizeof(date_str), "%04d-%02d-%02d", date->tm_year + 1900, date->tm_mon + 1, date->tm_mday);
			if (ret < 0) {
				return;
			}
			printf("%sGenerated: %s\n", prefix, date_str);
		}
	}
}

static void _remove_occurence_of_character_from_string(char *string, char c)
{
	char *reading_ptr = string, *writing_ptr = string;
	while (*reading_ptr) {
		*writing_ptr = *reading_ptr;
		if (*reading_ptr != c)
			writing_ptr++;
		reading_ptr++;
	}
	*writing_ptr = '\0';
}

static void _translate_character_in_string(char *string, char to_replace, char replace_with)
{
	char *string_iterator = string;
	while (*string_iterator) {
		if (*string_iterator == to_replace)
			*string_iterator = replace_with;
		string_iterator++;
	}
}

static void _print_xccdf_profile_default(const struct xccdf_profile *prof, const char *prefix)
{
	struct oscap_text_iterator *title_it = xccdf_profile_get_title(prof);
	char *profile_title = oscap_textlist_get_preferred_plaintext(title_it, NULL);
	oscap_text_iterator_free(title_it);
	_remove_occurence_of_character_from_string(profile_title, '\n');
	printf("%s\tTitle: %s\n", prefix, profile_title);
	free(profile_title);
	printf("%s\t\tId: %s%s\n", prefix,
		xccdf_profile_get_id(prof),
		xccdf_profile_get_abstract(prof) ? " (abstract)" : "");
}

static void _print_xccdf_profile_verbose(const struct xccdf_profile *prof, const char *prefix)
{
	struct oscap_text_iterator *text_it;

	printf("%sProfile\n", prefix);

	text_it = xccdf_profile_get_title(prof);
	char *profile_title = oscap_textlist_get_preferred_plaintext(text_it, NULL);
	oscap_text_iterator_free(text_it);
	_remove_occurence_of_character_from_string(profile_title, '\n');
	printf("%s\tTitle: %s\n", prefix, profile_title);
	free(profile_title);

	printf("%s\tId: %s%s\n\n", prefix,
		xccdf_profile_get_id(prof),
		xccdf_profile_get_abstract(prof) ? " (abstract)" : "");

	text_it = xccdf_profile_get_description(prof);
	char *profile_description = oscap_textlist_get_preferred_plaintext(text_it, NULL);
	oscap_text_iterator_free(text_it);
	_translate_character_in_string(profile_description, '\n', ' ');
	printf("%s\tDescription: %s\n", prefix, profile_description);
	free(profile_description);
}

static void _print_xccdf_profile_terse(const struct xccdf_profile *prof, const char *prefix)
{
	struct oscap_text_iterator *title_it = xccdf_profile_get_title(prof);
	char *profile_title = oscap_textlist_get_preferred_plaintext(title_it, NULL);
	_remove_occurence_of_character_from_string(profile_title, '\n');
	oscap_text_iterator_free(title_it);
	const char *profile_id = xccdf_profile_get_id(prof);
	char cleaned_profile_id[1024];
	strncpy(cleaned_profile_id, profile_id, sizeof(cleaned_profile_id) - 1);
	_remove_occurence_of_character_from_string(cleaned_profile_id, '\n');
	printf("%s:%s\n",
		cleaned_profile_id,
		profile_title);
	free(profile_title);
}

static void _print_xccdf_profile_with_id(struct xccdf_profile *profile, const char *prefix)
{
	if (profile == NULL)
		return;

	_print_xccdf_profile_verbose(profile, prefix);
}

static inline void _print_xccdf_profiles(struct xccdf_profile_iterator *prof_it, const char *prefix, void (*print_one_profile)(const struct xccdf_profile *, const char *))
{
	if (prefix)
		printf("%sProfiles:\n", prefix);

	if (print_one_profile == NULL)
		print_one_profile = &_print_xccdf_profile_default;

	while (xccdf_profile_iterator_has_more(prof_it)) {
		struct xccdf_profile *prof = xccdf_profile_iterator_next(prof_it);
		print_one_profile(prof, prefix);
	}
}

static inline void _print_xccdf_referenced_files(struct xccdf_policy_model *policy_model, const char *prefix)
{
	struct oscap_file_entry_list *referenced_files = xccdf_policy_model_get_systems_and_files(policy_model);
	struct oscap_file_entry_iterator *files_it = oscap_file_entry_list_get_files(referenced_files);
	printf("%sReferenced check files:\n", prefix);
	while (oscap_file_entry_iterator_has_more(files_it)) {
		struct oscap_file_entry *file_entry = (struct oscap_file_entry *) oscap_file_entry_iterator_next(files_it);
		printf("%s\t%s\n", prefix, oscap_file_entry_get_file(file_entry));
		printf("%s\t\tsystem: %s\n", prefix, oscap_file_entry_get_system(file_entry));
	}
	oscap_file_entry_iterator_free(files_it);
	oscap_file_entry_list_free(referenced_files);
}

static inline void _print_xccdf_result(struct xccdf_result *xccdf_result, const char *prefix)
{
	const char *result_id = xccdf_result_get_id(xccdf_result);
	const char *benchmark_uri = xccdf_result_get_benchmark_uri(xccdf_result);
	const char *xccdf_profile = xccdf_result_get_profile(xccdf_result);
	if (xccdf_profile == NULL) {
		xccdf_profile = "(default)";
	}
	const char *start_time = xccdf_result_get_start_time(xccdf_result);
	const char *end_time = xccdf_result_get_end_time(xccdf_result);
	printf("%s\tResult ID: %s\n", prefix, result_id);
	printf("%s\tSource benchmark: %s\n", prefix, benchmark_uri);
	printf("%s\tSource profile: %s\n", prefix, xccdf_profile);
	printf("%s\tEvaluation started: %s\n", prefix, start_time);
	printf("%s\tEvaluation finished: %s\n", prefix, end_time);
	printf("%s\tPlatform CPEs:\n", prefix);
	struct oscap_string_iterator *platforms_it = xccdf_result_get_platforms(xccdf_result);
	if (!oscap_string_iterator_has_more(platforms_it)) {
		printf("%s\t\t(none)\n", prefix);
	}
	while (oscap_string_iterator_has_more(platforms_it)) {
		const char *platform = oscap_string_iterator_next(platforms_it);
		printf("%s\t\t%s\n", prefix, platform);
	}
	oscap_string_iterator_free(platforms_it);
}


static inline void _print_xccdf_testresults(struct xccdf_benchmark *bench, const char *prefix)
{
	struct xccdf_result_iterator *res_it = xccdf_benchmark_get_results(bench);
	if (xccdf_result_iterator_has_more(res_it))
		printf("%sTest Results:\n", prefix);
	while (xccdf_result_iterator_has_more(res_it)) {
		struct xccdf_result *test_result = xccdf_result_iterator_next(res_it);
		_print_xccdf_result(test_result, prefix);
	}
	xccdf_result_iterator_free(res_it);
}

static inline void _print_xccdf_benchmark(struct xccdf_benchmark *bench, const char *prefix, void (*print_one_profile)(const struct xccdf_profile *, const char *))
{
	_print_xccdf_status(xccdf_benchmark_get_status_current(bench), prefix);
	printf("%sResolved: %s\n", prefix, xccdf_benchmark_get_resolved(bench) ? "true" : "false");

	struct xccdf_profile_iterator *prof_it = xccdf_benchmark_get_profiles(bench);
	_print_xccdf_profiles(prof_it, prefix, print_one_profile);
	xccdf_profile_iterator_free(prof_it);

	struct xccdf_policy_model *policy_model = xccdf_policy_model_new(bench);
	_print_xccdf_referenced_files(policy_model, prefix);
	_print_xccdf_testresults(bench, prefix);

	xccdf_policy_model_free(policy_model);
	// xccdf_benchmark_free not needed, it si already freed by the policy!
}

static inline void _print_xccdf_tailoring_header(struct xccdf_tailoring *tailoring, const char *prefix)
{
	if (tailoring == NULL) {
		return;
	}
	printf("%sBenchmark Hint: %s\n", prefix ? prefix : "", xccdf_tailoring_get_benchmark_ref(tailoring));
}

static inline void _print_xccdf_tailoring(struct oscap_source *source, const char *prefix, void (*print_one_profile)(const struct xccdf_profile *, const char *))
{
	struct xccdf_tailoring *tailoring = xccdf_tailoring_import_source(source, NULL);
	_print_xccdf_tailoring_header(tailoring, prefix);

	struct xccdf_profile_iterator *prof_it = xccdf_tailoring_get_profiles(tailoring);
	_print_xccdf_profiles(prof_it, prefix, print_one_profile);
	xccdf_profile_iterator_free(prof_it);

	xccdf_tailoring_free(tailoring);
}

static void _print_single_benchmark_profiles_only(struct xccdf_benchmark *bench)
{
	struct xccdf_profile_iterator *prof_it = xccdf_benchmark_get_profiles(bench);
	_print_xccdf_profiles(prof_it, 0, _print_xccdf_profile_terse);
	xccdf_profile_iterator_free(prof_it);
}

static void _print_single_benchmark_one_profile(struct xccdf_benchmark *bench, const char *profile_id)
{
	struct xccdf_profile *profile = xccdf_benchmark_get_profile_by_id(bench, profile_id);
	if (profile != NULL)
		_print_xccdf_profile_with_id(profile, "");
}

static void _print_single_benchmark_all(struct xccdf_benchmark *bench, const char *prefix)
{
	_print_xccdf_benchmark(bench, prefix, 0);
	// bench is freed as a side-effect of the function above
}

static int app_info_single_ds_profiles_only(struct ds_stream_index_iterator* sds_it, struct ds_sds_session *session, const struct oscap_action *action)
{
	struct ds_stream_index * stream = ds_stream_index_iterator_next(sds_it);
	struct oscap_string_iterator* checklist_it = ds_stream_index_get_checklists(stream);

	while (oscap_string_iterator_has_more(checklist_it)) {
		const char * id = oscap_string_iterator_next(checklist_it);
		/* decompose */
		struct oscap_source *xccdf_source = ds_sds_session_select_checklist(session, ds_stream_index_get_id(stream), id, NULL);
		if (xccdf_source == NULL) {
			oscap_string_iterator_free(checklist_it);
			ds_stream_index_iterator_free(sds_it);
			ds_sds_session_free(session);
			return OSCAP_ERROR;
		}

		if (oscap_source_get_scap_type(xccdf_source) == OSCAP_DOCUMENT_XCCDF) {
			struct xccdf_benchmark* bench = xccdf_benchmark_import_source(xccdf_source);
			if(!bench) {
				oscap_string_iterator_free(checklist_it);
				ds_stream_index_iterator_free(sds_it);
				ds_sds_session_free(session);
				return OSCAP_ERROR;
			}
			_print_single_benchmark_profiles_only(bench);
			xccdf_benchmark_free(bench);

		} else if (oscap_source_get_scap_type(xccdf_source) == OSCAP_DOCUMENT_XCCDF_TAILORING) {
			_print_xccdf_tailoring(xccdf_source, 0, _print_xccdf_profile_terse);
		}
		ds_sds_session_reset(session);
	}
	oscap_string_iterator_free(checklist_it);
	return OSCAP_OK;
}

static const char *tailoring_get_profile_or_report_multiple_ids(struct xccdf_tailoring *tailoring, const char *profile_suffix, const char *source_file)
{
	int match_status;
	const char *result = xccdf_tailoring_match_profile_id(tailoring, profile_suffix, &match_status);
	evaluate_suffix_match_result_with_custom_reports(match_status, profile_suffix, source_file, NULL, &report_multiple_profile_matches);
	return result;
}

static const char *benchmark_get_profile_or_report_multiple_ids(struct xccdf_benchmark *bench, const char *profile_suffix, const char *source_file)
{
	int match_status;
	const char *result = xccdf_benchmark_match_profile_id(bench, profile_suffix, &match_status);
	evaluate_suffix_match_result_with_custom_reports(match_status, profile_suffix, source_file, NULL, &report_multiple_profile_matches);
	return result;
}

static const char *benchmark_get_profile_or_report_id_issues(struct xccdf_benchmark *bench, const char *profile_suffix, const char *source_file)
{
	int match_status;
	const char *result = xccdf_benchmark_match_profile_id(bench, profile_suffix, &match_status);
	evaluate_suffix_match_result(match_status, profile_suffix, source_file);
	return result;
}

static int app_info_single_ds_one_profile(struct ds_stream_index_iterator* sds_it, struct ds_sds_session *session, const char *profile_suffix, const char *filename)
{
	const char *prefix = "";
	struct ds_stream_index * stream = ds_stream_index_iterator_next(sds_it);
	struct oscap_string_iterator* checklist_it = ds_stream_index_get_checklists(stream);

	printf("\nStream: %s\n", ds_stream_index_get_id(stream));
	printf("Generated: %s\n", ds_stream_index_get_timestamp(stream));
	printf("Version: %s\n", ds_stream_index_get_version(stream));
	bool profile_not_found = true;

	while (oscap_string_iterator_has_more(checklist_it) && profile_not_found) {
		const char * id = oscap_string_iterator_next(checklist_it);

		/* decompose */
		struct oscap_source *xccdf_source = ds_sds_session_select_checklist(session, ds_stream_index_get_id(stream), id, NULL);
		if (xccdf_source == NULL) {
			oscap_string_iterator_free(checklist_it);
			ds_stream_index_iterator_free(sds_it);
			ds_sds_session_free(session);
			return OSCAP_ERROR;
		}

		if (oscap_source_get_scap_type(xccdf_source) == OSCAP_DOCUMENT_XCCDF) {
			struct xccdf_benchmark *bench = xccdf_benchmark_import_source(xccdf_source);
			if(!bench) {
				oscap_string_iterator_free(checklist_it);
				ds_stream_index_iterator_free(sds_it);
				ds_sds_session_free(session);
				return OSCAP_ERROR;
			}
			const char *profile_id = benchmark_get_profile_or_report_multiple_ids(bench, profile_suffix, filename);
			if (profile_id != NULL) {
				_print_single_benchmark_one_profile(bench, profile_id);
				profile_not_found = false;
			}
			xccdf_benchmark_free(bench);
		} else if (oscap_source_get_scap_type(xccdf_source) == OSCAP_DOCUMENT_XCCDF_TAILORING) {
			struct xccdf_tailoring *tailoring = xccdf_tailoring_import_source(xccdf_source, NULL);

			const char *profile_id = tailoring_get_profile_or_report_multiple_ids(tailoring, profile_suffix, filename);
			if (profile_id != NULL) {
				struct xccdf_profile *profile = xccdf_tailoring_get_profile_by_id(tailoring, profile_id);
				_print_xccdf_profile_with_id(profile, prefix);
				profile_not_found = false;
			}

			xccdf_tailoring_free(tailoring);
		}
		ds_sds_session_reset(session);
	}
	oscap_string_iterator_free(checklist_it);
	if (profile_not_found) {
		report_missing_profile(profile_suffix, filename);
	}
	return OSCAP_OK;
}

static int app_info_single_ds_all(struct ds_stream_index_iterator* sds_it, struct ds_sds_session *session, const struct oscap_action *action)
{
	const char *prefix = "\t\t";
	struct ds_stream_index * stream = ds_stream_index_iterator_next(sds_it);
	struct oscap_string_iterator* checklist_it = ds_stream_index_get_checklists(stream);

	printf("\nStream: %s\n", ds_stream_index_get_id(stream));
	printf("Generated: %s\n", ds_stream_index_get_timestamp(stream));
	printf("Version: %s\n", ds_stream_index_get_version(stream));

	printf("Checklists:\n");
	while (oscap_string_iterator_has_more(checklist_it)) {
		const char * id = oscap_string_iterator_next(checklist_it);
		printf("\tRef-Id: %s\n", id);

		/* decompose */
		struct oscap_source *xccdf_source = ds_sds_session_select_checklist(session, ds_stream_index_get_id(stream), id, NULL);
		if (xccdf_source == NULL) {
			oscap_string_iterator_free(checklist_it);
			ds_stream_index_iterator_free(sds_it);
			ds_sds_session_free(session);
			return OSCAP_ERROR;
		}

		if (oscap_source_get_scap_type(xccdf_source) == OSCAP_DOCUMENT_XCCDF) {
			struct xccdf_benchmark* bench = xccdf_benchmark_import_source(xccdf_source);
			if(!bench) {
				oscap_string_iterator_free(checklist_it);
				ds_stream_index_iterator_free(sds_it);
				ds_sds_session_free(session);
				return OSCAP_ERROR;
			}
			_print_xccdf_benchmark(bench, prefix, 0);
			// bench is freed as a side-effect of the function above
		} else if (oscap_source_get_scap_type(xccdf_source) == OSCAP_DOCUMENT_XCCDF_TAILORING) {
			_print_xccdf_tailoring(xccdf_source, prefix, 0);
		}
		ds_sds_session_reset(session);
	}
	oscap_string_iterator_free(checklist_it);

	printf("Checks:\n");
	struct oscap_string_iterator* checks_it = ds_stream_index_get_checks(stream);
	while (oscap_string_iterator_has_more(checks_it)) {
		const char * id = oscap_string_iterator_next(checks_it);
		printf("\tRef-Id: %s\n", id);
	}
	oscap_string_iterator_free(checks_it);

	struct oscap_string_iterator* dict_it = ds_stream_index_get_dictionaries(stream);
	if (oscap_string_iterator_has_more(dict_it)) {
		printf("Dictionaries:\n");
	} else {
		printf("No dictionaries.\n");
	}
	while (oscap_string_iterator_has_more(dict_it)) {
		const char * id = oscap_string_iterator_next(dict_it);
		printf("\tRef-Id: %s\n", id);
	}
	oscap_string_iterator_free(dict_it);
	return OSCAP_OK;
}

static void app_info_single_benchmark(struct xccdf_benchmark *bench, const struct oscap_action *action, struct oscap_source *source)
{
	if (action->show_profiles_only) {
		_print_single_benchmark_profiles_only(bench);
		xccdf_benchmark_free(bench);
	} else if (action->profile) {
		const char *profile_id = benchmark_get_profile_or_report_id_issues(bench, action->profile, action->file);
		if (profile_id != NULL) {
			_print_single_benchmark_one_profile(bench, profile_id);
		}
		xccdf_benchmark_free(bench);
	} else {
		printf("Checklist version: %s\n", oscap_source_get_schema_version(source));
		print_time(action->file);

		_print_single_benchmark_all(bench, "");
		// bench is freed as a side-effect of the function above
	}
}

static int app_info_single_ds(struct ds_stream_index_iterator* sds_it, struct ds_sds_session *session, const struct oscap_action *action)
{
	int return_value;
	if (action->show_profiles_only) {
		return_value = app_info_single_ds_profiles_only(sds_it, session, action);
	} else if (action->profile) {
		return_value = app_info_single_ds_one_profile(sds_it, session, action->profile, action->file);
	} else {
		return_value = app_info_single_ds_all(sds_it, session, action);
	}
	return return_value;
}

static int app_info_sds(struct oscap_source *source, const struct oscap_action *action)
{
	if (! action->provide_machine_readable_output) {
		printf("Document type: Source Data Stream\n");
		print_time(action->file);
	}

	struct ds_sds_session *session = ds_sds_session_new_from_source(source);
	if (session == NULL) {
		return OSCAP_ERROR;
	}

	ds_sds_session_set_remote_resources(session, action->remote_resources, download_reporting_callback);

	/* get collection */
	struct ds_sds_index *sds = ds_sds_session_get_sds_idx(session);
	if (!sds) {
		ds_sds_session_free(session);
		return OSCAP_ERROR;
	}
	/* iterate over streams */
	struct ds_stream_index_iterator* sds_it = ds_sds_index_get_streams(sds);
	while (ds_stream_index_iterator_has_more(sds_it)) {
		if (app_info_single_ds(sds_it, session, action) == OSCAP_ERROR)
			return OSCAP_ERROR;
	}
	ds_stream_index_iterator_free(sds_it);
	ds_sds_session_free(session);
	return OSCAP_OK;
}

static int app_info(const struct oscap_action *action)
{
        int result = OSCAP_ERROR;

	struct oscap_source *source = oscap_source_new_from_file(action->file);

	switch (oscap_source_get_scap_type(source)) {
	case OSCAP_DOCUMENT_OVAL_DEFINITIONS: {
		printf("Document type: OVAL Definitions\n");
		struct oval_definition_model *def_model = oval_definition_model_import_source(source);
		if(!def_model)
			goto cleanup;
		struct oval_generator *gen = oval_definition_model_get_generator(def_model);
		printf("OVAL version: %s\n", oval_generator_get_core_schema_version(gen));
		printf("Generated: %s\n", oval_generator_get_timestamp(gen));
		print_time(action->file);
		oval_definition_model_free(def_model);
	}
	break;
	case OSCAP_DOCUMENT_OVAL_VARIABLES: {
		printf("Document type: OVAL Variables\n");
		struct oval_variable_model *var_model = oval_variable_model_import_source(source);
		if(!var_model)
			goto cleanup;
		struct oval_generator *gen = oval_variable_model_get_generator(var_model);
		printf("OVAL version: %s\n", oval_generator_get_core_schema_version(gen));
		printf("Generated: %s\n", oval_generator_get_timestamp(gen));
		print_time(action->file);
		oval_variable_model_free(var_model);
	}
	break;
	case OSCAP_DOCUMENT_OVAL_DIRECTIVES: {
		printf("Document type: OVAL Directives\n");
		struct oval_directives_model *dir_model = oval_directives_model_new();
		int ret = oval_directives_model_import_source(dir_model, source);
		if(ret)
			goto cleanup;
		struct oval_generator *gen = oval_directives_model_get_generator(dir_model);
		printf("OVAL version: %s\n", oval_generator_get_core_schema_version(gen));
		printf("Generated: %s\n", oval_generator_get_timestamp(gen));
		print_time(action->file);
		oval_directives_model_free(dir_model);
	}
	break;
	case OSCAP_DOCUMENT_OVAL_SYSCHAR: {
		printf("Document type: OVAL System Characteristics\n");
		struct oval_definition_model * def_model = oval_definition_model_new();
		struct oval_syschar_model * sys_model = oval_syschar_model_new(def_model);
		int ret = oval_syschar_model_import_source(sys_model, source);
		if(ret)
			goto cleanup;
		struct oval_generator *gen = oval_syschar_model_get_generator(sys_model);
		printf("OVAL version: %s\n", oval_generator_get_core_schema_version(gen));
		printf("Generated: %s\n", oval_generator_get_timestamp(gen));
		print_time(action->file);
		oval_syschar_model_free(sys_model);
		oval_definition_model_free(def_model);
	}
	break;
	case OSCAP_DOCUMENT_OVAL_RESULTS: {
		printf("Document type: OVAL Results\n");
		struct oval_definition_model * def_model=oval_definition_model_new();
		struct oval_results_model * res_model = oval_results_model_new(def_model,NULL);
		int ret = oval_results_model_import_source(res_model, source);
		if(ret)
			goto cleanup;
		struct oval_generator *gen = oval_results_model_get_generator(res_model);
		printf("OVAL version: %s\n", oval_generator_get_core_schema_version(gen));
		printf("Generated: %s\n", oval_generator_get_timestamp(gen));
		print_time(action->file);
		oval_results_model_free(res_model);
		oval_definition_model_free(def_model);
	}
	break;
	case OSCAP_DOCUMENT_XCCDF: {
                if (!action->provide_machine_readable_output)
			printf("Document type: XCCDF Checklist\n");
		struct xccdf_benchmark* bench = xccdf_benchmark_import_source(source);
		if(!bench)
			goto cleanup;
		app_info_single_benchmark(bench, action, source);
		// bench is freed as a side-effect of the function above
	}
	break;
	case OSCAP_DOCUMENT_CPE_LANGUAGE: {
		printf("Document type: CPE Language\n");
		print_time(action->file);
	}
	break;
	case OSCAP_DOCUMENT_CPE_DICTIONARY: {
		printf("Document type: CPE Dictionary\n");
		struct cpe_dict_model *dict_model = cpe_dict_model_import_source(source);
		if (!dict_model)
			goto cleanup;
		struct cpe_generator *gen = cpe_dict_model_get_generator(dict_model);
		if (gen != NULL) {
			printf("CPE version: %s\n", cpe_generator_get_schema_version(gen));
			printf("Generated: %s\n", cpe_generator_get_timestamp(gen));
		}
		print_time(action->file);
		cpe_dict_model_free(dict_model);
	}
	break;
	case OSCAP_DOCUMENT_SDS: {
		if (app_info_sds(source, action) == OSCAP_ERROR)
			goto cleanup;
	}
	break;
	case OSCAP_DOCUMENT_ARF: {
		printf("Document type: Result Data Stream\n");
		print_time(action->file);
		struct ds_rds_session *session = ds_rds_session_new_from_source(source);
		if (session == NULL) {
			goto cleanup;
		}
		struct rds_index *rds = ds_rds_session_get_rds_idx(session);
		if (!rds) {
			ds_rds_session_free(session);
			goto cleanup;
		}

		struct rds_asset_index_iterator* asset_it = rds_index_get_assets(rds);
		while (rds_asset_index_iterator_has_more(asset_it)) {
			struct rds_asset_index* asset = rds_asset_index_iterator_next(asset_it);
			printf("\nAsset: %s\n", rds_asset_index_get_id(asset));

			struct rds_report_index_iterator* report_it = rds_asset_index_get_reports(asset);
			while (rds_report_index_iterator_has_more(report_it)) {
				struct rds_report_index* report = rds_report_index_iterator_next(report_it);
				struct rds_report_request_index* request = rds_report_index_get_request(report);
				const char *report_request_id = rds_report_request_index_get_id(request);
				const char *report_id = rds_report_index_get_id(report);

				struct oscap_source *report_source = ds_rds_session_select_report(session, report_id);
				if (report_source == NULL) {
					rds_report_index_iterator_free(report_it);
					rds_asset_index_iterator_free(asset_it);
					ds_rds_session_free(session);
					goto cleanup;
				}
				oscap_document_type_t report_source_type = oscap_source_get_scap_type(report_source);
				if (report_source_type != OSCAP_DOCUMENT_XCCDF) {
					rds_report_index_iterator_free(report_it);
					rds_asset_index_iterator_free(asset_it);
					ds_rds_session_free(session);
					oscap_source_free(report_source);
					goto cleanup;
				}
				struct xccdf_result *xccdf_result = xccdf_result_import_source(report_source);
				if (xccdf_result == NULL) {
					rds_report_index_iterator_free(report_it);
					rds_asset_index_iterator_free(asset_it);
					ds_rds_session_free(session);
					oscap_source_free(report_source);
					goto cleanup;
				}
				printf("\tARF report: %s\n", report_id);
				printf("\t\tReport request: %s\n", report_request_id);
				_print_xccdf_result(xccdf_result, "\t");
				xccdf_result_free(xccdf_result);
				// oscap_source_free is not needed, it is already freed by xccdf_result_free

			}
			rds_report_index_iterator_free(report_it);
		}
		rds_asset_index_iterator_free(asset_it);
		ds_rds_session_free(session);
	}
	break;
	case OSCAP_DOCUMENT_XCCDF_TAILORING:
		printf("Document type: XCCDF Tailoring\n");
		print_time(action->file);
		_print_xccdf_tailoring(source, "", 0);
	break;
	case OSCAP_DOCUMENT_CVE_FEED:
		printf("Document type: CVE Feed\n");
		// TODO: Provide more info about CVE feeds
	break;
	case OSCAP_DOCUMENT_CVRF_FEED:
		printf("Document type: CVRF Feed\n");
		// TODO: Provide more info about CVRF feeds
	break;
	case OSCAP_DOCUMENT_SCE_RESULT:
		printf("Document type: SCE Result File\n");
		// Currently, we do not have any SCE result file parsing capabilities.
	break;
	case OSCAP_DOCUMENT_OCIL:
		printf("Document type: OCIL Definitions file\n");
		// we don't support OCIL yet
	break;
	default:
		printf("Could not determine document type\n");
		goto cleanup;
		break;
	}

	result = OSCAP_OK;

cleanup:
	oscap_source_free(source);
	oscap_print_error();

	return result;
}

bool getopt_info(int argc, char **argv, struct oscap_action *action)
{
	assert(action != NULL);

	/* Command-options */
	const struct option long_options[] = {
		{"fetch-remote-resources", no_argument, &action->remote_resources, 1},
		{"profile", required_argument, 0, 'p'},
		{"profiles", no_argument, 0, 'n'},
		// end
		{0, 0, 0, 0}
	};

	int c;
	while ((c = getopt_long(argc, argv, "o:i:p:", long_options, NULL)) != -1) {
		switch(c) {
			case 0: break;
			case 'p':
				action->profile = optarg;
				break;
			case 'n':
				action->show_profiles_only = 1;
				action->provide_machine_readable_output = 1;
				break;
			default: return oscap_module_usage(action->module, stderr, NULL);
		}
	}

	if (optind >= argc) {
		oscap_module_usage(action->module, stderr, "SCAP file needs to be specified!\n");
		return false;
	}

	action->file = argv[optind];

	return true;
}
