72 lines
2.1 KiB
C
72 lines
2.1 KiB
C
#include <stdio.h>
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <unistd.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
|
|
int main(int argc, char **argv){
|
|
unsigned char *buffer = malloc(64 * 1024);
|
|
struct stat s;
|
|
unsigned int size_vmlinux = 0, real_size_vmlinux = 0;
|
|
const unsigned char *magic_str = "ACME_PART_MAGIC";
|
|
unsigned int loop;
|
|
unsigned char *magic;
|
|
|
|
if(argc != 3){
|
|
printf("%s in out\n", argv[0]);
|
|
return 1;
|
|
}
|
|
|
|
printf("Generating image\n");
|
|
|
|
FILE *vmlinux = fopen(argv[1], "r");
|
|
FILE *vmlinux_out = fopen(argv[2], "w");
|
|
if((!vmlinux) || (!vmlinux_out)){
|
|
printf("Error opening a file\n");
|
|
return 1;
|
|
}
|
|
|
|
stat(argv[1], &s);
|
|
size_vmlinux = s.st_size;
|
|
real_size_vmlinux = (size_vmlinux & 0xffff0000) + 0x10000;
|
|
|
|
printf("vmlinux = 0x%.08X / 0x%.08X\n", size_vmlinux, real_size_vmlinux);
|
|
|
|
unsigned int t = fread(buffer, 1, 64 * 1024, vmlinux);
|
|
for(loop = 0; loop < (64 * 1024) - sizeof(magic_str); loop++){
|
|
if(buffer[loop] == magic_str[0]){
|
|
if((magic = strstr(&buffer[loop], magic_str))){
|
|
printf("Magic at 0x%.08X %p %p\n", magic - buffer, magic, buffer);
|
|
printf("Found Magic %X%X%X%X\n",
|
|
buffer[loop + strlen(magic_str)],
|
|
buffer[loop + strlen(magic_str) + 2],
|
|
buffer[loop + strlen(magic_str) + 1],
|
|
buffer[loop + strlen(magic_str) + 3]);
|
|
|
|
buffer[loop + strlen(magic_str)] = real_size_vmlinux >> 24;
|
|
buffer[loop + strlen(magic_str) + 2] = (real_size_vmlinux >> 16) & 0xff;
|
|
buffer[loop + strlen(magic_str) + 1] = (real_size_vmlinux >> 8) & 0xff;
|
|
buffer[loop + strlen(magic_str) + 3] = (real_size_vmlinux) & 0xff;
|
|
|
|
printf("Replaced with %.02X%.02X%.02X%.02X\n",
|
|
buffer[loop + strlen(magic_str)],
|
|
buffer[loop + strlen(magic_str) + 2],
|
|
buffer[loop + strlen(magic_str) + 1],
|
|
buffer[loop + strlen(magic_str) + 3]);
|
|
|
|
}
|
|
}
|
|
}
|
|
|
|
fwrite(buffer, 1, 64 * 1024, vmlinux_out);
|
|
real_size_vmlinux -= 64 * 1024;
|
|
do {
|
|
real_size_vmlinux -= 64 * 1024;
|
|
memset(buffer, 0, 64 * 1024);
|
|
fread(buffer, 1, 64 * 1024, vmlinux);
|
|
fwrite(buffer, 1, 64 * 1024, vmlinux_out);
|
|
} while (real_size_vmlinux);
|
|
|
|
return 0;
|
|
}
|