copy_os.c

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <stdbool.h>
#include <sys/stat.h>
#include "oscafs.h"
extern str_t dentry[NUM_DENTRY];
extern int fat[NB_BLOCKS];

int file_size(char *filename)
{
    struct stat statbuf;
    stat(filename, &statbuf);
    int size = statbuf.st_size;

    return size;
}

void os_copy_host(FILE *fp)
{
    FILE *file_host;
    int fd;          // dentry row number
    int fb;          //first block
    int nb;          // next block number
    int data_length; // Count the number of characters
    char file_name_copy[NAME_LEN];
    char *pt;
    bool same_name = false;
    int count_alloc;
    int count_free;
    if ((return_first_entry()) >= 0)
    {
        printf(YELLOW "please input a host file name: " NONE);

        while (s_gets(file_name_copy, NAME_LEN) != NULL && file_name_copy[0] != '\0') // check is it have same name file
        {
            if ((file_host = fopen(file_name_copy, "r+")) == NULL)
            {
                fprintf(stderr, "Cannot open or find the \"%s\" file, please try other file\n", file_name_copy);
                //puts("please try other file");
                continue;
            }

            for (int i = 0; i < NUM_DENTRY; i++)
            {
                if (strcmp(file_name_copy, return_entry(i)) == 0)
                {
                    same_name = true;
                    break;
                }
            }

            if (same_name)
            {
                printf("the file \"%s\" have been exist, cannot copy same name file, try again\n", file_name_copy);
                //puts("try again");
                same_name = false;
                continue;
            }

            else
            {
                int num_block;
                int size = file_size(file_name_copy);
                //printf("for test size: %d\n", size); //for test
                if (size % BLOCK_SIZE == 0)
                    num_block = size / BLOCK_SIZE;
                else
                    num_block = (size / BLOCK_SIZE) + 1;

                //printf("num_block:%d\n", num_block); //for test
                //printf("spare block:%d\n", count_free_block()); // for test
                if (num_block <= count_free_block())
                {
                    char spdata[num_block * BLOCK_SIZE];
                    pt = spdata;
                    fread(spdata, 1, size, file_host);
                    spdata[size] = '\0';
                    //printf("spdata:%s\n", spdata); // for test
                    //printf("spdata strlen is %d\n", strlen(spdata));  //for test
                    fd = write_first_free(file_name_copy);
                    if (fd < 0)
                    {
                        printf("OSCAFS cannot contain more than %d file\n", NUM_DENTRY);
                        break;
                    }
                    fb = write_firstblock_num(fd);
                    count_alloc = 1;

                    //printf("%s is No.%d in dentry and ", file_name_copy, fd); //for test
                    //printf("begin in %d block\n", fb);  //for test
                    //printf("spdata: %s\n", spdata); //fot test
                    //printf("spdata strlen is %d\n", strlen(spdata)); //for test
                    //printf("this file use %d block\n", num_block); //for test
                    char databuf[num_block][BLOCK_SIZE];
                    for (int count = 0; count < num_block; count++)
                    {
                        strncpy(databuf[count], pt + (sizeof(spdata) - (num_block - count) * BLOCK_SIZE), BLOCK_SIZE);
                        databuf[count][BLOCK_SIZE] = '\0';
                        //printf("databuf%d: %-8s | in block: %d\n", count + 1, databuf[count], fb); //for check data if correct
                        write_block(fb, databuf[count], fp);
                        //printf("%d ", fb); // for test
                        if (count_alloc < num_block)
                        {
                            nb = add_next_block(fb);
                            fb = nb;
                            count_alloc++;
                        }
                    }
                    fclose(file_host);
                    printf(YELLOW "Enter file name to copy new one to OSCAFS(empty to quit): " NONE);
                }

                else
                {
                    puts("OSCAFS not enough spare sapce to save this file");
                    break;
                    //printf("need %d block\n", num_block);  //for test
                    //printf("spare blocks:%d\n", count_free_block());  //for test
                }
            }
        }
    }

    else
        printf("OSCAFS cannot contain more than %d file\n", NUM_DENTRY);
}

void os_copy_OSCAFS(FILE *fp)
{
    FILE *file_host;
    int i;  //for check empty dentry
    int j;  //for select file
    int fb; // first block in file
    bool have_file = false;
    char file_name[NAME_LEN];
    char data[BLOCK_SIZE];
    int temp; //for change the value of fat table
    for (i = 0; i < NUM_DENTRY; i++)
    {
        if (dentry[i].str[0] != '\0')
        {
            have_file = true;
        }
    }

    if (have_file)
    {
        //printf("\n");
        os_list();
        printf(YELLOW "plese choice a file: " NONE);
        bool true_name = false;
        while (s_gets(file_name, NAME_LEN) != NULL && file_name[0] != '\0')
        {
            for (j = 0; j < NUM_DENTRY; j++)
            {
                if (strcmp(file_name, dentry[j].str) == 0)
                {
                    true_name = true;
                    break;
                }
            }

            if (true_name)
            {
                if ((file_host = fopen(file_name, "w+")) == NULL)
                {
                    fprintf(stderr, "Cannot create the \"%s\" file", file_name);
                    printf("please try other file or make sure the authority(empty to quit): ");
                    continue;
                }
                else
                {
                    rewind(file_host);
                    //printf("\n");
                    fb = dentry[j].num;
                    // printf("%d\n", fb); //for test
                    read_block(fb, data, fp);
                    data[BLOCK_SIZE] = '\0';
                    //printf("%s", data); // for test
                    fwrite(data, 1, BLOCK_SIZE, file_host);
                    int count_seek = 1;
                    while (fat[fb] != EOF_BLK)
                    {

                        char bufdata[BLOCK_SIZE];
                        temp = next_block(fb);
                        fb = temp;
                        read_block(fb, bufdata, fp);
                        bufdata[BLOCK_SIZE] = '\0';
                        //printf("%s", bufdata); for test
                        fseek(file_host, count_seek * BLOCK_SIZE, SEEK_SET);
                        fwrite(bufdata, 1, BLOCK_SIZE, file_host);
                        count_seek++;
                    }
                    //printf("done\n");
                    fclose(file_host);
                    printf(YELLOW "Enter file name to copy other one (empty to quit): " NONE);
                }
            }

            else
            {
                printf("%s can not find in OSCAFS\n", file_name);
                printf(YELLOW "please try again (empty to quit): " NONE);
                continue;
            }

            true_name = false;
        }
    }

    else
        puts("No any file");
}