You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
123 lines
2.2 KiB
C
123 lines
2.2 KiB
C
// Program to replace the partition table in the usbloader
|
|
//
|
|
#include <stdio.h>
|
|
#include <stdint.h>
|
|
|
|
#ifndef WIN32
|
|
//%%%%
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#else
|
|
//%%%%
|
|
#include <windows.h>
|
|
#include "getopt.h"
|
|
#include "printf.h"
|
|
#endif
|
|
|
|
#include "parts.h"
|
|
|
|
//############################################################################################################3
|
|
|
|
void main(int argc, char* argv[]) {
|
|
|
|
|
|
int opt;
|
|
int mflag=0;
|
|
char ptfile[100];
|
|
int rflag=0,xflag=0;
|
|
|
|
uint32_t ptaddr;
|
|
struct ptable_t ptable;
|
|
|
|
FILE* ldr;
|
|
FILE* out;
|
|
FILE* in;
|
|
|
|
while ((opt = getopt(argc, argv, "mr:hx")) != -1) {
|
|
switch (opt) {
|
|
case 'h':
|
|
|
|
printf("\n Utility for partition table replacement in usbloaders\
|
|
\n\n\
|
|
%s [options] <file name usbloader>\n\n\
|
|
Following options are allowed:\n\n\
|
|
-m - Show current partition map in usbloader\n\
|
|
-x - Extract current partition map to ptable.bin file\n\
|
|
-r <file> - Replace current partition map with map from specified file\n\
|
|
\n",argv[0]);
|
|
return;
|
|
|
|
case 'm':
|
|
mflag=1;
|
|
break;
|
|
|
|
case 'x':
|
|
xflag=1;
|
|
break;
|
|
|
|
case 'r':
|
|
rflag=1;
|
|
strcpy (ptfile,optarg);
|
|
break;
|
|
|
|
case '?':
|
|
case ':':
|
|
return;
|
|
|
|
}
|
|
}
|
|
if (optind>=argc) {
|
|
printf("\n - Loader file name is not specified\n");
|
|
return;
|
|
}
|
|
|
|
ldr=fopen(argv[optind],"r+b");
|
|
if (ldr == 0) {
|
|
printf("\n Can't open file %s\n",argv[optind]);
|
|
return;
|
|
}
|
|
|
|
// searching for the partition table in the loader file
|
|
|
|
ptaddr=find_ptable(ldr);
|
|
if (ptaddr == 0) {
|
|
printf("\n Таблица разделов в загрузчике не найдена\n");
|
|
return ;
|
|
}
|
|
// read the current table
|
|
fread(&ptable,sizeof(ptable),1,ldr);
|
|
|
|
if (xflag) {
|
|
out=fopen("ptable.bin","wb");
|
|
fwrite(&ptable,sizeof(ptable),1,out);
|
|
fclose(out);
|
|
}
|
|
|
|
if (mflag) {
|
|
show_map(ptable);
|
|
}
|
|
|
|
if (mflag | xflag) return;
|
|
|
|
if (rflag) {
|
|
in=fopen(ptfile,"rb");
|
|
if (in == 0) {
|
|
printf("\n Can't open file %s",ptfile);
|
|
return;
|
|
}
|
|
fread(&ptable,sizeof(ptable),1,in);
|
|
fclose(in);
|
|
|
|
// check the file
|
|
if (memcmp(ptable.head,headmagic,16) != 0) {
|
|
printf("\n Input file is not a partition table\n");
|
|
return;
|
|
}
|
|
fseek(ldr,ptaddr,SEEK_SET);
|
|
fwrite(&ptable,sizeof(ptable),1,ldr);
|
|
fclose(ldr);
|
|
|
|
}
|
|
}
|