static void stor_cleanup(void) { int i; for (i = 0; i < stor_info_no; i++) if (stor_info[i].opened > 0) ub_dev_close(stor_info[i].handle); disk_cleanup(&uboot_storage); }
static void beri_cfi_disk_cleanup(void) { disk_cleanup(&beri_cfi_disk); }
int main(int oargc, char* oargv[]) { int ret; int argc = oargc - 1; char** argv = oargv + 1; // first parameter must be the image file. if (argc == 0) { fprintf(stderr, "Error: First parameter must be a filename.\n"); PRINT_HELP_AND_QUIT(); } if (is_command(argv[0])) { fprintf(stderr, "Error: First parameter must be a filename, found '%s' instead.\n", argv[0]); PRINT_HELP_AND_QUIT(); } if (disk_openimage(0, argv[0])) { fprintf(stderr, "Error: Could not open image file '%s'.\n", argv[0]); ret = 1; goto exit; } argc--; argv++; while (argc > 0) { char* parg = *argv; int nargs = 0; int i = 0; if (!is_command(parg)) { fprintf(stderr, "Error: Expected a command, found '%s' instead.\n", parg); PRINT_HELP_AND_QUIT(); } parg++; argv++; argc--; // find next command, to calculare number of args while ((argv[i] != NULL) && !is_command(argv[i++])) nargs++; if (strcmp(parg, "format") == 0) { // NOTE: The fs driver detects which FAT format fits best based on size int sectors; NEED_PARAMS(1, 2); // Arg 1: number of sectors sectors = atoi(argv[0]); if (sectors <= 0) { fprintf(stderr, "Error: Sectors must be > 0\n"); ret = 1; goto exit; } if (disk_ioctl(0, SET_SECTOR_COUNT, §ors)) { fprintf(stderr, "Error: Failed to set sector count to %d.\n", sectors); ret = 1; goto exit; } NEED_MOUNT(); ret = f_mkfs("0:", 1, sectors < 4096 ? 1 : 8); if (ret) { fprintf(stderr, "Error: Formatting drive: %d.\n", ret); goto exit; } // Arg 2: custom header label (optional) if (nargs > 1) { #define FAT_VOL_LABEL_LEN 11 char vol_label[2 + FAT_VOL_LABEL_LEN + 1]; // Null-terminated buffer char* label = vol_label + 2; // The first two characters are reserved for the drive number "0:" char ch; int i, invalid = 0; int len = strlen(argv[1]); if (len <= FAT_VOL_LABEL_LEN) { // Verify each character (should be printable ASCII) // and copy it in uppercase. for (i = 0; i < len; i++) { ch = toupper(argv[1][i]); if ((ch < 0x20) || !isprint(ch)) { invalid = 1; break; } label[i] = ch; } if (!invalid) { // Pad the label with spaces while (len < FAT_VOL_LABEL_LEN) { label[len++] = ' '; } } } else { invalid = 1; } if (invalid) { fprintf(stderr, "Error: Header label is limited to 11 printable uppercase ASCII symbols."); ret = 1; goto exit; } if (disk_read(0, buff, 0, 1)) { fprintf(stderr, "Error: Unable to read existing boot sector from image."); ret = 1; goto exit; } if (g_Filesystem.fs_type == FS_FAT32) { memcpy(buff + 71, label, FAT_VOL_LABEL_LEN); } else { memcpy(buff + 43, label, FAT_VOL_LABEL_LEN); } if (disk_write(0, buff, 0, 1)) { fprintf(stderr, "Error: Unable to write new boot sector to image."); ret = 1; goto exit; } // Set also the directory volume label memcpy(vol_label, "0:", 2); vol_label[2 + FAT_VOL_LABEL_LEN] = '\0'; if (f_setlabel(vol_label)) { fprintf(stderr, "Error: Unable to set the volume label."); ret = 1; goto exit; } } } else if (strcmp(parg, "boot") == 0) { FILE* fe; BYTE* temp = buff + 1024; NEED_PARAMS(1, 1); // Arg 1: boot file fe = fopen(argv[0], "rb"); if (!fe) { fprintf(stderr, "Error: Unable to open external file '%s' for reading.", argv[0]); ret = 1; goto exit; } if (!fread(buff, 512, 1, fe)) { fprintf(stderr, "Error: Unable to read boot sector from file '%s'.", argv[0]); fclose(fe); ret = 1; goto exit; } fclose(fe); NEED_MOUNT(); if (disk_read(0, temp, 0, 1)) { fprintf(stderr, "Error: Unable to read existing boot sector from image."); ret = 1; goto exit; } if (g_Filesystem.fs_type == FS_FAT32) { printf("TODO: Writing boot sectors for FAT32 images not yet supported."); ret = 1; goto exit; } else { #define FAT16_HEADER_START 3 #define FAT16_HEADER_END 62 memcpy(buff + FAT16_HEADER_START, temp + FAT16_HEADER_START, FAT16_HEADER_END - FAT16_HEADER_START); } if (disk_write(0, buff, 0, 1)) { fprintf(stderr, "Error: Unable to write new boot sector to image."); ret = 1; goto exit; } } else if (strcmp(parg, "add") == 0) { FILE* fe; FIL fv = { 0 }; UINT rdlen = 0; UINT wrlen = 0; NEED_PARAMS(2, 2); NEED_MOUNT(); // Arg 1: external file to add // Arg 2: virtual filename fe = fopen(argv[0], "rb"); if (!fe) { fprintf(stderr, "Error: Unable to open external file '%s' for reading.", argv[0]); ret = 1; goto exit; } if (f_open(&fv, argv[1], FA_WRITE | FA_CREATE_ALWAYS)) { fprintf(stderr, "Error: Unable to open file '%s' for writing.", argv[1]); fclose(fe); ret = 1; goto exit; } while ((rdlen = fread(buff, 1, sizeof(buff), fe)) > 0) { if (f_write(&fv, buff, rdlen, &wrlen) || wrlen < rdlen) { fprintf(stderr, "Error: Unable to write '%d' bytes to disk.", wrlen); ret = 1; goto exit; } } fclose(fe); f_close(&fv); } else if (strcmp(parg, "extract") == 0) { FIL fe = { 0 }; FILE* fv; UINT rdlen = 0; UINT wrlen = 0; NEED_PARAMS(2, 2); NEED_MOUNT(); // Arg 1: virtual file to extract // Arg 2: external filename if (f_open(&fe, argv[0], FA_READ)) { fprintf(stderr, "Error: Unable to open file '%s' for reading.", argv[0]); ret = 1; goto exit; } fv = fopen(argv[1], "wb"); if (!fv) { fprintf(stderr, "Error: Unable to open external file '%s' for writing.", argv[1]); f_close(&fe); ret = 1; goto exit; } while ((f_read(&fe, buff, sizeof(buff), &rdlen) == 0) && (rdlen > 0)) { if (fwrite(buff, 1, rdlen, fv) < rdlen) { fprintf(stderr, "Error: Unable to write '%d' bytes to file.", rdlen); ret = 1; goto exit; } } f_close(&fe); fclose(fv); } else if (strcmp(parg, "move") == 0) { NEED_PARAMS(2, 2); NEED_MOUNT(); // Arg 1: src path & filename // Arg 2: new path & filename if (f_rename(argv[0], argv[1])) { fprintf(stderr, "Error: Unable to move/rename '%s' to '%s'", argv[0], argv[1]); ret = 1; goto exit; } } else if (strcmp(parg, "copy") == 0) { FIL fe = { 0 }; FIL fv = { 0 }; UINT rdlen = 0; UINT wrlen = 0; NEED_PARAMS(2, 2); NEED_MOUNT(); // Arg 1: src path & filename // Arg 2: new path & filename if (f_open(&fe, argv[0], FA_READ)) { fprintf(stderr, "Error: Unable to open file '%s' for reading.", argv[0]); ret = 1; goto exit; } if (f_open(&fv, argv[1], FA_WRITE | FA_CREATE_ALWAYS)) { fprintf(stderr, "Error: Unable to open file '%s' for writing.", argv[1]); f_close(&fe); ret = 1; goto exit; } while ((f_read(&fe, buff, sizeof(buff), &rdlen) == 0) && (rdlen > 0)) { if (f_write(&fv, buff, rdlen, &wrlen) || wrlen < rdlen) { fprintf(stderr, "Error: Unable to write '%d' bytes to disk.", wrlen); ret = 1; goto exit; } } f_close(&fe); f_close(&fv); } else if (strcmp(parg, "mkdir") == 0) { NEED_PARAMS(1, 1); NEED_MOUNT(); // Arg 1: folder path if (f_mkdir(argv[0])) { fprintf(stderr, "Error: Unable to create directory."); ret = 1; goto exit; } } else if (strcmp(parg, "delete") == 0) { NEED_PARAMS(1, 1); NEED_MOUNT(); // Arg 1: file/folder path (cannot delete non-empty folders) if (f_unlink(argv[0])) { fprintf(stderr, "Error: Unable to delete file or directory."); ret = 1; goto exit; } } else if (strcmp(parg, "list") == 0) { char* root = "/"; DIR dir = { 0 }; FILINFO info = { 0 }; char lfname[257]; NEED_PARAMS(0, 1); // Arg 1: folder path (optional) if (nargs == 1) { root = argv[0]; } if (f_opendir(&dir, root)) { fprintf(stderr, "Error: Unable to opening directory '%s' for listing.\n", root); ret = 1; goto exit; } printf("Listing directory contents of: %s\n", root); info.lfname = lfname; info.lfsize = sizeof(lfname)-1; while ((!f_readdir(&dir, &info)) && (strlen(info.fname) > 0)) { if (strlen(info.lfname) > 0) printf(" - %s (%s)\n", info.lfname, info.fname); else printf(" - %s\n", info.fname); } } else { fprintf(stderr, "Error: Unknown or invalid command: %s\n", argv[-1]); PRINT_HELP_AND_QUIT(); } argv += nargs; argc -= nargs; } ret = 0; exit: disk_cleanup(0); return ret; }
static void beri_sdcard_disk_cleanup(void) { disk_cleanup(&beri_sdcard_disk); }