/*

 * 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 DataStream.\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;

}