file.c

#include <ctype.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/stat.h>

#include "fat32disk.h"
#include "directory.h"
#include "recover.h"

int main(int argc, char **argv){
    int flag=0, anyOption=0;
    int information=0, listRootDir=0, recoverFiles=0, sha=0;
    char *recOptarg, *shaOptarg;

    // go through all of the flags
    while((flag=getopt(argc, argv, "ilr:s:")) != -1){
        anyOption=1;
        switch(flag){
            case 'i':
                information=1;
                break;
            case 'l':
                listRootDir=1;
                break;
            case 'r':
                recoverFiles=1;
                recOptarg = optarg;
                break;
            case 's':
                sha = 1;
                shaOptarg = optarg;
                break;
            default:
                showUsage();
                break;
        }
    }

    if (anyOption==0 || optind==argc)
        showUsage();

    // get file system information
    int fd = getFileDirectory(argv[optind]);
    BootEntry* disk = readFileSystem(fd);

    if (information)
        showDiskInformation(disk);
    else if (listRootDir)
        getRootDirectoryEntries(fd, disk);
    else if(sha){ // use sha only when -r flag is given
        if(recoverFiles){
            if (recOptarg == NULL)
            showUsage();
        recoverFile(fd, disk, recOptarg, shaOptarg);
        }
        else{
            showUsage();
        }
    }
    else if(recoverFiles && !sha){ // only recover; no sha
        if (recOptarg == NULL)
            showUsage();
        recoverFile(fd, disk, recOptarg, NULL);
    }
}