/* SPDX-License-Identifier: GPL-2.0-or-later */
/*
 *  Copyright (C) 2020 Hyunchul Lee <hyc.lee@gmail.com>
 */
#include <stdio.h>
#include <string.h>
#include <stdarg.h>

#include "exfat_ondisk.h"
#include "libexfat.h"
#include "fsck.h"
#include "repair.h"

struct exfat_repair_problem {
	er_problem_code_t	prcode;
	unsigned int		flags;
	unsigned int		prompt_type;
};

/* Problem flags */
#define ERF_PREEN_YES		0x00000001
#define ERF_DEFAULT_YES		0x00000002
#define ERF_DEFAULT_NO		0x00000004

/* Prompt types */
#define ERP_FIX			0x00000001
#define ERP_TRUNCATE		0x00000002

static const char *prompts[] = {
	"Repair",
	"Fix",
	"Truncate",
};

static struct exfat_repair_problem problems[] = {
	{ER_BS_CHECKSUM, ERF_PREEN_YES, ERP_FIX},
	{ER_BS_BOOT_REGION, 0, ERP_FIX},
	{ER_DE_CHECKSUM, ERF_PREEN_YES, ERP_FIX},
	{ER_FILE_VALID_SIZE, ERF_PREEN_YES, ERP_FIX},
	{ER_FILE_INVALID_CLUS, ERF_DEFAULT_NO, ERP_TRUNCATE},
	{ER_FILE_FIRST_CLUS, ERF_DEFAULT_NO, ERP_TRUNCATE},
	{ER_FILE_SMALLER_SIZE, ERF_DEFAULT_NO, ERP_TRUNCATE},
	{ER_FILE_LARGER_SIZE, ERF_DEFAULT_NO, ERP_TRUNCATE},
	{ER_FILE_DUPLICATED_CLUS, ERF_DEFAULT_NO, ERP_TRUNCATE},
	{ER_FILE_ZERO_NOFAT, ERF_PREEN_YES, ERP_FIX},
};

static struct exfat_repair_problem *find_problem(er_problem_code_t prcode)
{
	unsigned int i;

	for (i = 0; i < sizeof(problems)/sizeof(problems[0]); i++) {
		if (problems[i].prcode == prcode) {
			return &problems[i];
		}
	}
	return NULL;
}

static bool ask_repair(struct exfat *exfat, struct exfat_repair_problem *pr)
{
	bool repair = false;
	char answer[8];

	if (exfat->options & FSCK_OPTS_REPAIR_NO ||
			pr->flags & ERF_DEFAULT_NO)
		repair = false;
	else if (exfat->options & FSCK_OPTS_REPAIR_YES ||
			pr->flags & ERF_DEFAULT_YES)
		repair = true;
	else {
		if (exfat->options & FSCK_OPTS_REPAIR_ASK) {
			do {
				printf(". %s (y/N)? ",
					prompts[pr->prompt_type]);
				fflush(stdout);

				if (fgets(answer, sizeof(answer), stdin)) {
					if (strcasecmp(answer, "Y\n") == 0)
						return true;
					else if (strcasecmp(answer, "\n") == 0
						|| strcasecmp(answer, "N\n") == 0)
						return false;
				}
			} while (1);
		} else if (exfat->options & FSCK_OPTS_REPAIR_AUTO &&
				pr->flags & ERF_PREEN_YES)
			repair = true;
	}

	printf(". %s (y/N)? %c\n", prompts[pr->prompt_type],
		repair ? 'y' : 'n');
	return repair;
}

bool exfat_repair_ask(struct exfat *exfat, er_problem_code_t prcode,
			const char *desc, ...)
{
	struct exfat_repair_problem *pr = NULL;
	va_list ap;

	pr = find_problem(prcode);
	if (!pr) {
		exfat_err("unknown problem code. %#x\n", prcode);
		return false;
	}

	va_start(ap, desc);
	vprintf(desc, ap);
	va_end(ap);

	if (ask_repair(exfat, pr)) {
		if (pr->prompt_type & ERP_TRUNCATE)
			exfat->dirty_fat = true;
		exfat->dirty = true;
		return true;
	} else
		return false;
}