fsck/repair.c

   1 /* SPDX-License-Identifier: GPL-2.0-or-later */
   2 /*
   3  *  Copyright (C) 2020 Hyunchul Lee <hyc.lee@gmail.com>
   4  */
   5 #include <stdio.h>
   6 #include <string.h>
   7 #include <stdarg.h>
   8
   9 #include "exfat_ondisk.h"
  10 #include "libexfat.h"
  11 #include "fsck.h"
  12 #include "repair.h"
  13
  14 struct exfat_repair_problem {
  15         er_problem_code_t       prcode;
  16         unsigned int            flags;
  17         unsigned int            prompt_type;
  18 };
  19
  20 /* Problem flags */
  21 #define ERF_PREEN_YES           0x00000001
  22 #define ERF_DEFAULT_YES         0x00000002
  23 #define ERF_DEFAULT_NO          0x00000004
  24
  25 /* Prompt types */
  26 #define ERP_FIX                 0x00000001
  27 #define ERP_TRUNCATE            0x00000002
  28
  29 static const char *prompts[] = {
  30         "Repair",
  31         "Fix",
  32         "Truncate",
  33 };
  34
  35 static struct exfat_repair_problem problems[] = {
  36         {ER_BS_CHECKSUM, ERF_DEFAULT_YES | ERF_PREEN_YES, ERP_FIX},
  37         {ER_DE_CHECKSUM, ERF_DEFAULT_YES | ERF_PREEN_YES, ERP_FIX},
  38         {ER_FILE_VALID_SIZE, ERF_DEFAULT_YES | ERF_PREEN_YES, ERP_FIX},
  39         {ER_FILE_INVALID_CLUS, ERF_DEFAULT_NO, ERP_TRUNCATE},
  40         {ER_FILE_FIRST_CLUS, ERF_DEFAULT_NO, ERP_TRUNCATE},
  41         {ER_FILE_SMALLER_SIZE, ERF_DEFAULT_NO, ERP_TRUNCATE},
  42         {ER_FILE_LARGER_SIZE, ERF_DEFAULT_NO, ERP_TRUNCATE},
  43         {ER_FILE_DUPLICATED_CLUS, ERF_DEFAULT_NO, ERP_TRUNCATE},
  44         {ER_FILE_ZERO_NOFAT, ERF_DEFAULT_YES | ERF_PREEN_YES, ERP_FIX},
  45 };
  46
  47 static struct exfat_repair_problem *find_problem(er_problem_code_t prcode)
  48 {
  49         unsigned int i;
  50
  51         for (i = 0; i < sizeof(problems)/sizeof(problems[0]); i++) {
  52                 if (problems[i].prcode == prcode) {
  53                         return &problems[i];
  54                 }
  55         }
  56         return NULL;
  57 }
  58
  59 static bool ask_repair(struct exfat *exfat, struct exfat_repair_problem *pr)
  60 {
  61         bool repair = false;
  62         char answer[8];
  63
  64         if (exfat->options & FSCK_OPTS_REPAIR_NO ||
  65                         pr->flags & ERF_DEFAULT_NO)
  66                 repair = false;
  67         else if (exfat->options & FSCK_OPTS_REPAIR_YES ||
  68                         pr->flags & ERF_DEFAULT_YES)
  69                 repair = true;
  70         else {
  71                 if (exfat->options & FSCK_OPTS_REPAIR_ASK) {
  72                         do {
  73                                 printf(". %s (y/N)? ",
  74                                         prompts[pr->prompt_type]);
  75                                 fflush(stdout);
  76
  77                                 if (fgets(answer, sizeof(answer), stdin)) {
  78                                         if (strcasecmp(answer, "Y\n") == 0)
  79                                                 return true;
  80                                         else if (strcasecmp(answer, "\n") == 0
  81                                                 || strcasecmp(answer, "N\n") == 0)
  82                                                 return false;
  83                                 }
  84                         } while (1);
  85                 } else if (exfat->options & FSCK_OPTS_REPAIR_AUTO &&
  86                                 pr->flags & ERF_PREEN_YES)
  87                         repair = true;
  88         }
  89
  90         printf(". %s (y/N)? %c\n", prompts[pr->prompt_type],
  91                 repair ? 'y' : 'n');
  92         return repair;
  93 }
  94
  95 bool exfat_repair_ask(struct exfat *exfat, er_problem_code_t prcode,
  96                         const char *desc, ...)
  97 {
  98         struct exfat_repair_problem *pr = NULL;
  99         va_list ap;
 100
 101         pr = find_problem(prcode);
 102         if (!pr) {
 103                 exfat_err("unknown problem code. %#x\n", prcode);
 104                 return false;
 105         }
 106
 107         va_start(ap, desc);
 108         vprintf(desc, ap);
 109         va_end(ap);
 110
 111         if (ask_repair(exfat, pr)) {
 112                 if (pr->prompt_type & ERP_TRUNCATE)
 113                         exfat->dirty_fat = true;
 114                 exfat->dirty = true;
 115                 return true;
 116         } else
 117                 return false;
 118 }