120 lines
2.7 KiB
C
120 lines
2.7 KiB
C
/* 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;
|
|
}
|