mirror of https://github.com/ventoy/Ventoy
160 lines
3.8 KiB
C
160 lines
3.8 KiB
C
/******************************************************************************
|
|
* vtoyloader.c ---- ventoy loader (wapper for binary loader)
|
|
*
|
|
* Copyright (c) 2020, longpanda <admin@ventoy.net>
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License as
|
|
* published by the Free Software Foundation; either version 3 of the
|
|
* License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but
|
|
* WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
#include <unistd.h>
|
|
#include <fcntl.h>
|
|
#include <sys/types.h>
|
|
#include <sys/mman.h>
|
|
#include <sys/ioctl.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
|
|
static int verbose = 0;
|
|
#define debug(fmt, ...) if(verbose) printf(fmt, ##__VA_ARGS__)
|
|
|
|
static int vine_patch_loader(unsigned char *buf, int len, int major)
|
|
{
|
|
int i;
|
|
int ptrlen;
|
|
unsigned int *data1;
|
|
unsigned int *data2;
|
|
|
|
/*
|
|
* http://vinelinux.ime.cmc.osaka-u.ac.jp/Vine-6.5/SRPMS/SRPMS.main/anaconda-vine-11.0.2.1-1vl7.src.rpm
|
|
* http://vinelinux.ime.cmc.osaka-u.ac.jp/Vine-6.5/SRPMS/SRPMS.main/kudzu-1.2.86-3vl6.src.rpm
|
|
* anaconda-vine-11.0.2.1
|
|
* isys/devnodes.c
|
|
* static struct devnum devices[] = {
|
|
* { "aztcd", 29, 0, 0 },
|
|
* { "pcd", 46, 0, 0 },
|
|
*
|
|
* Patch 29 ---> 253
|
|
*/
|
|
|
|
ptrlen = (buf[4] == 1) ? 4 : 8;
|
|
debug("ELF %d bit major:%d ptrlen:%d\n", (buf[4] == 1) ? 32 : 64, major, ptrlen);
|
|
|
|
for (i = 0; i < len - 8 - 8 - ptrlen; i++)
|
|
{
|
|
data1 = (unsigned int *)(buf + i);
|
|
data2 = (unsigned int *)(buf + i + 8 + ptrlen);
|
|
|
|
if (data1[0] == 0x1D && data1[1] == 0x00 && data2[0] == 0x2E && data2[1] == 0x00)
|
|
{
|
|
debug("Find aztcd patch point at %d\n", i);
|
|
data1[0] = major;
|
|
break;
|
|
}
|
|
}
|
|
|
|
for (i = 0; i < len; i++)
|
|
{
|
|
if (buf[i] != '/')
|
|
{
|
|
continue;
|
|
}
|
|
|
|
data1 = (unsigned int *)(buf + i + 1);
|
|
|
|
/* /proc/ide */
|
|
if (data1[0] == 0x636F7270 && data1[1] == 0x6564692F)
|
|
{
|
|
debug("patch string %s\n", (char *)(buf + i));
|
|
buf[i + 1] = 'v';
|
|
buf[i + 2] = 't';
|
|
buf[i + 3] = 'o';
|
|
buf[i + 4] = 'y';
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int vtoyvine_main(int argc, char **argv)
|
|
{
|
|
int i;
|
|
int len;
|
|
unsigned char *buf;
|
|
FILE *fp;
|
|
|
|
for (i = 0; i < argc; i++)
|
|
{
|
|
if (argv[i][0] == '-' && argv[i][1] == 'v')
|
|
{
|
|
verbose = 1;
|
|
break;
|
|
}
|
|
}
|
|
|
|
fp = fopen(argv[1], "rb");
|
|
if (!fp)
|
|
{
|
|
fprintf(stderr, "Failed to open file %s err:%d\n", argv[1], errno);
|
|
return 1;
|
|
}
|
|
|
|
fseek(fp, 0, SEEK_END);
|
|
len = (int)ftell(fp);
|
|
debug("file length:%d\n", len);
|
|
|
|
fseek(fp, 0, SEEK_SET);
|
|
|
|
buf = (unsigned char *)malloc(len);
|
|
if (!buf)
|
|
{
|
|
fclose(fp);
|
|
return 1;
|
|
}
|
|
|
|
fread(buf, 1, len, fp);
|
|
fclose(fp);
|
|
|
|
vine_patch_loader(buf, len, (int)strtoul(argv[2], NULL, 10));
|
|
|
|
fp = fopen(argv[1], "wb+");
|
|
if (!fp)
|
|
{
|
|
fprintf(stderr, "Failed to open file %s err:%d\n", argv[1], errno);
|
|
free(buf);
|
|
return 1;
|
|
}
|
|
|
|
debug("write new data length:%d\n", len);
|
|
fwrite(buf, 1, len, fp);
|
|
fclose(fp);
|
|
|
|
free(buf);
|
|
|
|
return 0;
|
|
}
|
|
|
|
// wrapper main
|
|
#ifndef BUILD_VTOY_TOOL
|
|
int main(int argc, char **argv)
|
|
{
|
|
return vtoyvine_main(argc, argv);
|
|
}
|
|
#endif
|
|
|