Ventoy/VtoyTool/vtoyvine.c
2020-08-29 13:12:46 +08:00

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