balong-usbload/ptable-injector.c

127 lines
2.6 KiB
C
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

// Программа для замены таблицы разделов в загрузчике 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 Утилита для замены таблицы разделов в загрузчиках usbloader\
\n\n\
%s [ключи] <имя файла usbloader>\n\n\
Допустимы следующие ключи:\n\n\
-m - показать текущую карту разделов в usbloader\n\
-x - извлечь текущую карту в файл ptable.bin\n\
-r <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 - Не указано имя файла загрузчика\n");
return;
}
ldr=fopen(argv[optind],"r+b");
if (ldr == 0) {
printf("\n Ошибка открытия файла %s\n",argv[optind]);
return;
}
// Ищем таблицу разделов в файле загрузчика
ptaddr=find_ptable(ldr);
if (ptaddr == 0) {
printf("\n Таблица разделов в загрузчике не найдена\n");
return ;
}
// читаем текущую таблицу
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 Ошибка открытия файла %s",ptfile);
return;
}
fread(&ptable,sizeof(ptable),1,in);
fclose(in);
// проверяем файл
if (memcmp(ptable.head,headmagic,16) != 0) {
printf("\n Входной файл не является таблицей разделов\n");
return;
}
fseek(ldr,ptaddr,SEEK_SET);
fwrite(&ptable,sizeof(ptable),1,ldr);
fclose(ldr);
}
}