attributes:
label: Ventoy Version
description: What version of ventoy are you running?
- placeholder: 1.0.65
+ placeholder: 1.0.66
validations:
required: true
- type: dropdown
debug("os_param->vtoy_img_size=<%llu>", chain->os_param.vtoy_img_size);
debug("os_param->vtoy_img_location_addr=<0x%llx>", chain->os_param.vtoy_img_location_addr);
debug("os_param->vtoy_img_location_len=<%u>", chain->os_param.vtoy_img_location_len);
- debug("os_param->vtoy_reserved=<%u %u %u %u %u>",
+ debug("os_param->vtoy_reserved=<%u %u %u %u %u %u %u>",
g_os_param_reserved[0],
g_os_param_reserved[1],
g_os_param_reserved[2],
g_os_param_reserved[3],
- g_os_param_reserved[4]
+ g_os_param_reserved[4],
+ g_os_param_reserved[5],
+ g_os_param_reserved[6]
);
ventoy_debug_pause();
if (CompareMem(g_chain->os_param.vtoy_disk_guid, pBuffer + 0x180, 16) == 0)
{
pMBR = (MBR_HEAD *)pBuffer;
- if (pMBR->PartTbl[0].FsFlag != 0xEE)
+ if (g_os_param_reserved[6] == 0 && pMBR->PartTbl[0].FsFlag != 0xEE)
{
if (pMBR->PartTbl[0].StartSectorId != 2048 ||
pMBR->PartTbl[1].SectorCount != 65536 ||
debug("memdisk mode iso_buf_size:%u", g_iso_buf_size);
g_chain = chain;
+ g_os_param_reserved = (UINT8 *)(g_chain->os_param.vtoy_reserved);
gMemdiskMode = TRUE;
}
else
}
}
+typedef struct grub_vlnk
+{
+ int srclen;
+ char src[512];
+ char dst[512];
+ struct grub_vlnk *next;
+}grub_vlnk;
+
+static grub_vlnk *g_vlnk_list;
+
+int grub_file_is_vlnk_suffix(const char *name, int len)
+{
+ grub_uint32_t suffix;
+
+ if (len > 9)
+ {
+ suffix = *(grub_uint32_t *)(name + len - 4);
+ if (grub_strncmp(name + len - 9, ".vlnk.", 6) == 0)
+ {
+ /* .iso .wim .img .vhd .efi .dat */
+ if (suffix == 0x6F73692E || suffix == 0x6D69772E ||
+ suffix == 0x676D692E || suffix == 0x6468762E ||
+ suffix == 0x6966652E || suffix == 0x7461642E)
+ {
+ return 1;
+ }
+ }
+ else if (len > 10 && grub_strncmp(name + len - 10, ".vlnk.", 6) == 0)
+ {
+ /* vhdx vtoy */
+ if (suffix == 0x78646876 || suffix == 0x796F7476)
+ {
+ return 1;
+ }
+ }
+ }
+
+ return 0;
+}
+
+int grub_file_add_vlnk(const char *src, const char *dst)
+{
+ grub_vlnk *node = NULL;
+
+ if (src && dst)
+ {
+ node = grub_zalloc(sizeof(grub_vlnk));
+ if (node)
+ {
+ node->srclen = (int)grub_strlen(src);
+ grub_strncpy(node->src, src, sizeof(node->src) - 1);
+ grub_strncpy(node->dst, dst, sizeof(node->dst) - 1);
+
+ node->next = g_vlnk_list;
+ g_vlnk_list = node;
+ return 0;
+ }
+ }
+
+ return 1;
+}
+
+const char *grub_file_get_vlnk(const char *name, int *vlnk)
+{
+ int len;
+ grub_vlnk *node = g_vlnk_list;
+
+ len = grub_strlen(name);
+ if (!grub_file_is_vlnk_suffix(name, len))
+ {
+ return name;
+ }
+
+ while (node)
+ {
+ if (node->srclen == len && grub_strcmp(name, node->src) == 0)
+ {
+ *vlnk = 1;
+ return node->dst;
+ }
+ node = node->next;
+ }
+
+ return name;
+}
+
grub_file_t
grub_file_open (const char *name, enum grub_file_type type)
{
+ int vlnk = 0;
grub_device_t device = 0;
grub_file_t file = 0, last_file = 0;
char *device_name;
const char *file_name;
grub_file_filter_id_t filter;
- /* <DESC> : mem:xxx:size:xxx format in chainloader */
- if (grub_strncmp(name, GRUB_MEMFILE_MEM, grub_strlen(GRUB_MEMFILE_MEM)) == 0) {
+ /* <DESC> : mem:xxx:size:xxx format in chainloader grub_strlen(GRUB_MEMFILE_MEM) */
+ if (grub_strncmp(name, GRUB_MEMFILE_MEM, 4) == 0) {
return grub_memfile_open(name);
- }
+ }
+
+ if (g_vlnk_list && (type & GRUB_FILE_TYPE_NO_VLNK) == 0)
+ name = grub_file_get_vlnk(name, &vlnk);
device_name = grub_file_get_device_name (name);
if (grub_errno)
goto fail;
file->device = device;
+ file->vlnk = vlnk;
/* In case of relative pathnames and non-Unix systems (like Windows)
* name of host files may not start with `/'. Blocklists for host files
return 1;
}
+grub_fs_t grub_fs_list_probe(grub_device_t device, const char **list)
+{
+ int i;
+ grub_fs_t p;
+
+ if (!device->disk)
+ return 0;
+
+ for (p = grub_fs_list; p; p = p->next)
+ {
+ for (i = 0; list[i]; i++)
+ {
+ if (grub_strcmp(p->name, list[i]) == 0)
+ break;
+ }
+
+ if (list[i] == NULL)
+ continue;
+
+ grub_dprintf("fs", "Detecting %s...\n", p->name);
+
+ (p->fs_dir) (device, "/", probe_dummy_iter, NULL);
+ if (grub_errno == GRUB_ERR_NONE)
+ return p;
+
+ grub_error_push ();
+ grub_dprintf ("fs", "%s detection failed.\n", p->name);
+ grub_error_pop ();
+
+ if (grub_errno != GRUB_ERR_BAD_FS && grub_errno != GRUB_ERR_OUT_OF_RANGE) {
+ return 0;
+ }
+ grub_errno = GRUB_ERR_NONE;
+ }
+
+ return 0;
+}
+
grub_fs_t
grub_fs_probe (grub_device_t device)
{
int g_wimboot_enable = 0;
ventoy_img_chunk_list g_wimiso_chunk_list;
char *g_wimiso_path = NULL;
+grub_uint32_t g_wimiso_size = 0;
int g_vhdboot_enable = 0;
static char g_json_case_mis_path[32];
+static ventoy_vlnk_part *g_vlnk_part_list = NULL;
+
static int ventoy_get_fs_type(const char *fs)
{
if (NULL == fs)
return file;
}
-int ventoy_is_file_exist(const char *fmt, ...)
-{
- va_list ap;
- int len;
- char *pos = NULL;
- char buf[256] = {0};
-
- grub_snprintf(buf, sizeof(buf), "%s", "[ -f \"");
- pos = buf + 6;
-
- va_start (ap, fmt);
- len = grub_vsnprintf(pos, 255, fmt, ap);
- va_end (ap);
-
- grub_strncpy(pos + len, "\" ]", 3);
-
- debug("script exec %s\n", buf);
-
- if (0 == grub_script_execute_sourcecode(buf))
- {
- return 1;
- }
-
- return 0;
-}
-
int ventoy_is_dir_exist(const char *fmt, ...)
{
va_list ap;
(void)args;
g_wimboot_enable = 0;
+ g_wimiso_size = 0;
grub_check_free(g_wimiso_path);
grub_check_free(g_wimiso_chunk_list.chunk);
g_wimboot_enable = 1;
g_wimiso_path = grub_strdup(args[0]);
-
+ g_wimiso_size = (grub_uint32_t)(file->size);
grub_file_close(file);
return 0;
return 1;
}
+ if (grub_strncmp(args[0], g_iso_path, grub_strlen(g_iso_path)))
+ {
+ file->vlnk = 1;
+ }
+
totlen += ventoy_align_2k(file->size);
dirent = (ventoy_iso9660_override *)(buf + offset);
return 1;
}
+static int ventoy_vlnk_iterate_partition(struct grub_disk *disk, const grub_partition_t partition, void *data)
+{
+ ventoy_vlnk_part *node = NULL;
+ grub_uint32_t *pSig = (grub_uint32_t *)data;
+
+ node = grub_zalloc(sizeof(ventoy_vlnk_part));
+ if (node)
+ {
+ node->disksig = *pSig;
+ node->partoffset = (partition->start << GRUB_DISK_SECTOR_BITS);
+ grub_snprintf(node->disk, sizeof(node->disk) - 1, "%s", disk->name);
+ grub_snprintf(node->device, sizeof(node->device) - 1, "%s,%d", disk->name, partition->number + 1);
+
+ node->next = g_vlnk_part_list;
+ g_vlnk_part_list = node;
+ }
+
+ return 0;
+}
+
+static int ventoy_vlnk_iterate_disk(const char *name, void *data)
+{
+ grub_disk_t disk;
+ grub_uint32_t sig;
+
+ (void)data;
+
+ disk = grub_disk_open(name);
+ if (disk)
+ {
+ grub_disk_read(disk, 0, 0x1b8, 4, &sig);
+
+ /* skip ventoy device self */
+ if (sig != *(grub_uint32_t *)data)
+ {
+ grub_partition_iterate(disk, ventoy_vlnk_iterate_partition, &sig);
+ }
+
+ grub_disk_close(disk);
+ }
+
+ return 0;
+}
+
+static int ventoy_vlnk_probe_fs(ventoy_vlnk_part *cur)
+{
+ const char *fs[ventoy_fs_max + 1] =
+ {
+ "exfat", "ntfs", "ext2", "xfs", "udf", "fat", NULL
+ };
+
+ if (!cur->dev)
+ {
+ cur->dev = grub_device_open(cur->device);
+ }
+
+ if (cur->dev)
+ {
+ cur->fs = grub_fs_list_probe(cur->dev, fs);
+ }
+
+ return 0;
+}
+
+static int ventoy_check_vlnk_data(ventoy_vlnk *vlnk, int print, char *dst, int size)
+{
+ int diskfind = 0;
+ int partfind = 0;
+ int filefind = 0;
+ char *disk, *device;
+ grub_uint32_t readcrc, calccrc;
+ ventoy_vlnk_part *cur;
+ grub_fs_t fs = NULL;
+
+ if (grub_memcmp(&(vlnk->guid), &g_ventoy_guid, sizeof(ventoy_guid)))
+ {
+ if (print)
+ {
+ grub_printf("VLNK invalid guid\n");
+ grub_refresh();
+ }
+ return 1;
+ }
+
+ readcrc = vlnk->crc32;
+ vlnk->crc32 = 0;
+ calccrc = grub_getcrc32c(0, vlnk, sizeof(ventoy_vlnk));
+ if (readcrc != calccrc)
+ {
+ if (print)
+ {
+ grub_printf("VLNK invalid crc 0x%08x 0x%08x\n", calccrc, readcrc);
+ grub_refresh();
+ }
+ return 1;
+ }
+
+ if (!g_vlnk_part_list)
+ {
+ grub_disk_dev_iterate(ventoy_vlnk_iterate_disk, g_ventoy_part_info->MBR.BootCode + 0x1b8);
+ }
+
+ for (cur = g_vlnk_part_list; cur && filefind == 0; cur = cur->next)
+ {
+ if (cur->disksig == vlnk->disk_signature)
+ {
+ diskfind = 1;
+ disk = cur->disk;
+ if (cur->partoffset == vlnk->part_offset)
+ {
+ partfind = 1;
+ device = cur->device;
+
+ if (cur->probe == 0)
+ {
+ cur->probe = 1;
+ ventoy_vlnk_probe_fs(cur);
+ }
+
+ if (!fs)
+ {
+ fs = cur->fs;
+ }
+
+ if (cur->fs)
+ {
+ struct grub_file file;
+
+ grub_memset(&file, 0, sizeof(file));
+ file.device = cur->dev;
+ if (cur->fs->fs_open(&file, vlnk->filepath) == GRUB_ERR_NONE)
+ {
+ filefind = 1;
+ cur->fs->fs_close(&file);
+ grub_snprintf(dst, size - 1, "(%s)%s", cur->device, vlnk->filepath);
+ }
+ }
+ }
+ }
+ }
+
+ if (print)
+ {
+ grub_printf("\n==== VLNK Information ====\n"
+ "Disk Signature: %08x\n"
+ "Partition Offset: %llu\n"
+ "File Path: <%s>\n\n",
+ vlnk->disk_signature, (ulonglong)vlnk->part_offset, vlnk->filepath);
+
+ if (diskfind)
+ {
+ grub_printf("Disk Find: [ YES ] [ %s ]\n", disk);
+ }
+ else
+ {
+ grub_printf("Disk Find: [ NO ]\n");
+ }
+
+ if (partfind)
+ {
+ grub_printf("Part Find: [ YES ] [ %s ] [ %s ]\n", device, fs ? fs->name : "N/A");
+ }
+ else
+ {
+ grub_printf("Part Find: [ NO ]\n");
+ }
+ grub_printf("File Find: [ %s ]\n", filefind ? "YES" : "NO");
+ if (filefind)
+ {
+ grub_printf("VLNK File: <%s>\n", dst);
+ }
+
+ grub_printf("\n");
+ grub_refresh();
+ }
+
+ return (1 - filefind);
+}
+
+int ventoy_add_vlnk_file(char *dir, const char *name)
+{
+ int rc = 1;
+ char src[512];
+ char dst[512];
+ grub_file_t file = NULL;
+ ventoy_vlnk vlnk;
+
+ if (!dir)
+ {
+ grub_snprintf(src, sizeof(src), "%s%s", g_iso_path, name);
+ }
+ else if (dir[0] == '/')
+ {
+ grub_snprintf(src, sizeof(src), "%s%s%s", g_iso_path, dir, name);
+ }
+ else
+ {
+ grub_snprintf(src, sizeof(src), "%s/%s%s", g_iso_path, dir, name);
+ }
+
+ file = grub_file_open(src, VENTOY_FILE_TYPE);
+ if (!file)
+ {
+ return 1;
+ }
+
+ grub_memset(&vlnk, 0, sizeof(vlnk));
+ grub_file_read(file, &vlnk, sizeof(vlnk));
+ grub_file_close(file);
+
+ if (ventoy_check_vlnk_data(&vlnk, 0, dst, sizeof(dst)) == 0)
+ {
+ rc = grub_file_add_vlnk(src, dst);
+ }
+
+ return rc;
+}
+
static int ventoy_collect_img_files(const char *filename, const struct grub_dirhook_info *info, void *data)
{
//int i = 0;
int type = 0;
int ignore = 0;
int index = 0;
+ int vlnk = 0;
grub_size_t len;
img_info *img;
img_info *tail;
return 0;
}
}
+
+ if (info->size == VTOY_FILT_MIN_FILE_SIZE || info->size == 0)
+ {
+ if (grub_file_is_vlnk_suffix(filename, len))
+ {
+ vlnk = 1;
+ if (ventoy_add_vlnk_file(node->dir, filename) != 0)
+ {
+ return 0;
+ }
+ }
+ }
img = grub_zalloc(sizeof(img_info));
if (img)
img->pathlen = grub_snprintf(img->path, sizeof(img->path), "%s%s", node->dir, img->name);
img->size = info->size;
- if (0 == img->size)
+ if (vlnk || 0 == img->size)
{
- img->size = ventoy_grub_get_file_size("%s/%s%s", g_iso_path, node->dir, filename);
+ if (node->dir[0] == '/')
+ {
+ img->size = ventoy_grub_get_file_size("%s%s%s", g_iso_path, node->dir, filename);
+ }
+ else
+ {
+ img->size = ventoy_grub_get_file_size("%s/%s%s", g_iso_path, node->dir, filename);
+ }
}
if (img->size < VTOY_FILT_MIN_FILE_SIZE)
param->vtoy_reserved[5] = 1;
}
+ /* ventoy_disk_signature used for vlnk */
+ param->vtoy_reserved[6] = file->vlnk;
+ grub_memcpy(param->vtoy_reserved + 7, g_ventoy_part_info->MBR.BootCode + 0x1b8, 4);
+
/* calculate checksum */
for (i = 0; i < sizeof(ventoy_os_param); i++)
{
return ret;
}
+static grub_err_t grub_cmd_vlnk_dump_part(grub_extcmd_context_t ctxt, int argc, char **args)
+{
+ int n = 0;
+ ventoy_vlnk_part *node;
+
+ (void)ctxt;
+ (void)argc;
+ (void)args;
+
+ for (node = g_vlnk_part_list; node; node = node->next)
+ {
+ grub_printf("[%d] %s disksig:%08x offset:%llu fs:%s\n",
+ ++n, node->device, node->disksig,
+ (ulonglong)node->partoffset, (node->fs ? node->fs->name : "N/A"));
+ }
+
+ return 0;
+}
+
+static grub_err_t grub_cmd_is_vlnk_name(grub_extcmd_context_t ctxt, int argc, char **args)
+{
+ int len = 0;
+
+ (void)ctxt;
+
+ if (argc == 1)
+ {
+ len = (int)grub_strlen(args[0]);
+ if (grub_file_is_vlnk_suffix(args[0], len))
+ {
+ return 0;
+ }
+ }
+
+ return 1;
+}
+
+static grub_err_t grub_cmd_get_vlnk_dst(grub_extcmd_context_t ctxt, int argc, char **args)
+{
+ int vlnk = 0;
+ const char *name = NULL;
+
+ (void)ctxt;
+
+ if (argc == 2)
+ {
+ grub_env_unset(args[1]);
+ name = grub_file_get_vlnk(args[0], &vlnk);
+ if (vlnk)
+ {
+ debug("VLNK SRC: <%s>\n", args[0]);
+ debug("VLNK DST: <%s>\n", name);
+ grub_env_set(args[1], name);
+ return 0;
+ }
+ }
+
+ return 1;
+}
+
+static grub_err_t grub_cmd_check_vlnk(grub_extcmd_context_t ctxt, int argc, char **args)
+{
+ int ret = 1;
+ int len = 0;
+ grub_file_t file = NULL;
+ ventoy_vlnk vlnk;
+ char dst[512];
+
+ (void)ctxt;
+
+ if (argc != 1)
+ {
+ goto out;
+ }
+
+ len = (int)grub_strlen(args[0]);
+ if (!grub_file_is_vlnk_suffix(args[0], len))
+ {
+ grub_printf("Invalid vlnk suffix\n");
+ goto out;
+ }
+
+ file = grub_file_open(args[0], VENTOY_FILE_TYPE | GRUB_FILE_TYPE_NO_VLNK);
+ if (!file)
+ {
+ grub_printf("Failed to open %s\n", args[0]);
+ goto out;
+ }
+
+ if (file->size != 32768)
+ {
+ grub_printf("Invalid vlnk file (size=%llu).\n", (ulonglong)file->size);
+ goto out;
+ }
+
+ grub_memset(&vlnk, 0, sizeof(vlnk));
+ grub_file_read(file, &vlnk, sizeof(vlnk));
+
+ ret = ventoy_check_vlnk_data(&vlnk, 1, dst, sizeof(dst));
+
+out:
+
+ grub_refresh();
+ check_free(file, grub_file_close);
+ grub_errno = GRUB_ERR_NONE;
+ return ret;
+}
+
int ventoy_env_init(void)
{
char buf[64];
{ "vt_append_extra_sector", ventoy_cmd_append_ext_sector, 0, NULL, "", "", NULL },
{ "gptpriority", grub_cmd_gptpriority, 0, NULL, "", "", NULL },
{ "vt_syslinux_need_nojoliet", grub_cmd_syslinux_nojoliet, 0, NULL, "", "", NULL },
+ { "vt_vlnk_check", grub_cmd_check_vlnk, 0, NULL, "", "", NULL },
+ { "vt_vlnk_dump_part", grub_cmd_vlnk_dump_part, 0, NULL, "", "", NULL },
+ { "vt_is_vlnk_name", grub_cmd_is_vlnk_name, 0, NULL, "", "", NULL },
+ { "vt_get_vlnk_dst", grub_cmd_get_vlnk_dst, 0, NULL, "", "", NULL },
};
int ventoy_register_all_cmd(void)
extern ventoy_img_chunk_list g_img_chunk_list;
extern ventoy_img_chunk_list g_wimiso_chunk_list;
extern char *g_wimiso_path;
+extern grub_uint32_t g_wimiso_size;
extern char g_arch_mode_suffix[64];
extern const char *g_menu_prefix[img_type_max];
int ventoy_cpio_newc_fill_head(void *buf, int filesize, const void *filedata, const char *name);
grub_file_t ventoy_grub_file_open(enum grub_file_type type, const char *fmt, ...);
grub_uint64_t ventoy_grub_get_file_size(const char *fmt, ...);
-int ventoy_is_file_exist(const char *fmt, ...);
int ventoy_is_dir_exist(const char *fmt, ...);
int ventoy_fill_data(grub_uint32_t buflen, char *buffer);
grub_err_t ventoy_cmd_load_plugin(grub_extcmd_context_t ctxt, int argc, char **args);
typedef struct file_fullpath
{
char path[256];
+ int vlnk_add;
}file_fullpath;
typedef struct theme_list
};
#pragma pack()
+typedef struct ventoy_vlnk_part
+{
+ grub_uint32_t disksig;
+ grub_uint64_t partoffset;
+ char disk[64];
+ char device[64];
+ grub_device_t dev;
+ grub_fs_t fs;
+ int probe;
+ struct ventoy_vlnk_part *next;
+}ventoy_vlnk_part;
+
+extern char g_iso_path[256];
+int ventoy_add_vlnk_file(char *dir, const char *name);
+
#endif /* __VENTOY_DEF_H__ */
grub_printf("file: %s\n", value);
if (value[0] == '/')
{
- exist = ventoy_is_file_exist("%s%s", isodisk, value);
+ exist = ventoy_check_file_exist("%s%s", isodisk, value);
}
else
{
- exist = ventoy_is_file_exist("%s/ventoy/%s", isodisk, value);
+ exist = ventoy_check_file_exist("%s/ventoy/%s", isodisk, value);
}
if (exist == 0)
grub_printf("file: %s\n", value);
if (value[0] == '/')
{
- exist = ventoy_is_file_exist("%s%s", isodisk, value);
+ exist = ventoy_check_file_exist("%s%s", isodisk, value);
}
else
{
- exist = ventoy_is_file_exist("%s/ventoy/%s", isodisk, value);
+ exist = ventoy_check_file_exist("%s/ventoy/%s", isodisk, value);
}
if (exist == 0)
return 1;
}
- if (!ventoy_is_file_exist("%s%s", path, file))
+ if (!ventoy_check_file_exist("%s%s", path, file))
{
grub_printf("%s%s does NOT exist\n", path, file);
return 1;
{
debug("%s is string type data\n", node->pcName);
- if ((node->unData.pcStrVal[0] != '/') || (!ventoy_is_file_exist("%s%s", isodisk, node->unData.pcStrVal)))
+ if ((node->unData.pcStrVal[0] != '/') || (!ventoy_check_file_exist("%s%s", isodisk, node->unData.pcStrVal)))
{
debug("%s%s file not found\n", isodisk, node->unData.pcStrVal);
return 1;
for (count = 0, child = node->pstChild; child; child = child->pstNext)
{
- if (ventoy_is_file_exist("%s%s", isodisk, child->unData.pcStrVal))
+ if (ventoy_check_file_exist("%s%s", isodisk, child->unData.pcStrVal))
{
grub_snprintf(path->path, sizeof(path->path), "%s", child->unData.pcStrVal);
path++;
{
grub_printf("image: <%s> [ * ]\n", path);
}
- else if (ventoy_is_file_exist("%s%s", isodisk, path))
+ else if (ventoy_check_file_exist("%s%s", isodisk, path))
{
grub_printf("image: <%s> [ OK ]\n", path);
}
{
grub_printf("image: <%s> [ * ]\n", path);
}
- else if (ventoy_is_file_exist("%s%s", isodisk, path))
+ else if (ventoy_check_file_exist("%s%s", isodisk, path))
{
grub_printf("image: <%s> [ OK ]\n", path);
}
int ventoy_plugin_get_persistent_chunklist(const char *isopath, int index, ventoy_img_chunk_list *chunk_list)
{
int rc = 1;
+ int len = 0;
+ char *path = NULL;
grub_uint64_t start = 0;
grub_file_t file = NULL;
persistence_config *node = NULL;
return 1;
}
- file = ventoy_grub_file_open(VENTOY_FILE_TYPE, "%s%s", g_iso_disk_name, node->backendpath[index].path);
+ path = node->backendpath[index].path;
+
+ if (node->backendpath[index].vlnk_add == 0)
+ {
+ len = grub_strlen(path);
+ if (len > 9 && grub_strncmp(path + len - 9, ".vlnk.dat", 9) == 0)
+ {
+ ventoy_add_vlnk_file(NULL, path);
+ node->backendpath[index].vlnk_add = 1;
+ }
+ }
+
+ file = ventoy_grub_file_open(VENTOY_FILE_TYPE, "%s%s", g_iso_disk_name, path);
if (!file)
{
- debug("Failed to open file %s%s\n", g_iso_disk_name, node->backendpath[index].path);
+ debug("Failed to open file %s%s\n", g_iso_disk_name, path);
goto end;
}
return 1;
}
+ if (grub_strncmp(args[0], g_iso_path, grub_strlen(g_iso_path)))
+ {
+ file->vlnk = 1;
+ }
+
img_chunk_size = g_img_chunk_list.cur_chunk * sizeof(ventoy_img_chunk);
size = sizeof(ventoy_chain_head) + img_chunk_size;
return imgoffset;
}
-static int ventoy_get_wim_chunklist(const char *filename, ventoy_img_chunk_list *wimchunk, grub_uint64_t *wimsize)
+static int ventoy_get_wim_chunklist(grub_file_t wimfile, ventoy_img_chunk_list *wimchunk)
{
- grub_file_t wimfile;
-
- wimfile = ventoy_grub_file_open(VENTOY_FILE_TYPE, "%s", filename);
- if (!wimfile)
- {
- return 1;
- }
-
grub_memset(wimchunk, 0, sizeof(ventoy_img_chunk_list));
wimchunk->chunk = grub_malloc(sizeof(ventoy_img_chunk) * DEFAULT_CHUNK_NUM);
if (NULL == wimchunk->chunk)
{
- grub_file_close(wimfile);
return grub_error(GRUB_ERR_OUT_OF_MEMORY, "Can't allocate image chunk memoty\n");
}
ventoy_get_block_list(wimfile, wimchunk, wimfile->device->disk->partition->start);
- *wimsize = wimfile->size;
- grub_file_close(wimfile);
-
return 0;
}
VENTOY_CMD_RETURN(GRUB_ERR_NONE);
}
-grub_err_t ventoy_cmd_wim_chain_data(grub_extcmd_context_t ctxt, int argc, char **args)
+static grub_err_t ventoy_vlnk_wim_chain_data(grub_file_t wimfile)
{
grub_uint32_t i = 0;
grub_uint32_t imgoffset = 0;
ventoy_img_chunk_list wimchunk;
char envbuf[128];
- (void)ctxt;
- (void)argc;
+ debug("vlnk wim chain data begin <%s> ...\n", wimfile->name);
+
+ if (NULL == g_wimiso_chunk_list.chunk || NULL == g_wimiso_path)
+ {
+ grub_printf("ventoy not ready\n");
+ return 1;
+ }
+
+ imgoffset = ventoy_get_wim_iso_offset(g_wimiso_path);
+ if (imgoffset == 0)
+ {
+ grub_printf("image offset not found\n");
+ return 1;
+ }
+
+ if (0 != ventoy_get_wim_chunklist(wimfile, &wimchunk))
+ {
+ grub_printf("Failed to get wim chunklist\n");
+ return 1;
+ }
+ wimsize = wimfile->size;
+
+ file = ventoy_grub_file_open(VENTOY_FILE_TYPE, "%s", g_wimiso_path);
+ if (!file)
+ {
+ return 1;
+ }
+
+ boot_catlog = ventoy_get_iso_boot_catlog(file);
+
+ img_chunk1_size = g_wimiso_chunk_list.cur_chunk * sizeof(ventoy_img_chunk);
+ img_chunk2_size = wimchunk.cur_chunk * sizeof(ventoy_img_chunk);
+ override_size = sizeof(ventoy_override_chunk) + g_wimiso_size;
+
+ size = sizeof(ventoy_chain_head) + img_chunk1_size + img_chunk2_size + override_size;
+
+ pLastChain = grub_env_get("vtoy_chain_mem_addr");
+ if (pLastChain)
+ {
+ chain = (ventoy_chain_head *)grub_strtoul(pLastChain, NULL, 16);
+ if (chain)
+ {
+ debug("free last chain memory %p\n", chain);
+ grub_free(chain);
+ }
+ }
+
+ chain = grub_malloc(size);
+ if (!chain)
+ {
+ grub_printf("Failed to alloc chain memory size %u\n", size);
+ grub_file_close(file);
+ return 1;
+ }
+
+ grub_snprintf(envbuf, sizeof(envbuf), "0x%lx", (unsigned long)chain);
+ grub_env_set("vtoy_chain_mem_addr", envbuf);
+ grub_snprintf(envbuf, sizeof(envbuf), "%u", size);
+ grub_env_set("vtoy_chain_mem_size", envbuf);
+
+ grub_memset(chain, 0, sizeof(ventoy_chain_head));
+
+ /* part 1: os parameter */
+ g_ventoy_chain_type = ventoy_chain_wim;
+ ventoy_fill_os_param(wimfile, &(chain->os_param));
+
+ /* part 2: chain head */
+ disk = wimfile->device->disk;
+ chain->disk_drive = disk->id;
+ chain->disk_sector_size = (1 << disk->log_sector_size);
+ chain->real_img_size_in_bytes = ventoy_align_2k(file->size) + ventoy_align_2k(wimsize);
+ chain->virt_img_size_in_bytes = chain->real_img_size_in_bytes;
+ chain->boot_catalog = boot_catlog;
+
+ if (!ventoy_is_efi_os())
+ {
+ grub_file_seek(file, boot_catlog * 2048);
+ grub_file_read(file, chain->boot_catalog_sector, sizeof(chain->boot_catalog_sector));
+ }
+
+ /* part 3: image chunk */
+ chain->img_chunk_offset = sizeof(ventoy_chain_head);
+ chain->img_chunk_num = g_wimiso_chunk_list.cur_chunk + wimchunk.cur_chunk;
+ grub_memcpy((char *)chain + chain->img_chunk_offset, g_wimiso_chunk_list.chunk, img_chunk1_size);
+
+ chunknode = (ventoy_img_chunk *)((char *)chain + chain->img_chunk_offset);
+ for (i = 0; i < g_wimiso_chunk_list.cur_chunk; i++)
+ {
+ chunknode->disk_end_sector = chunknode->disk_end_sector - chunknode->disk_start_sector;
+ chunknode->disk_start_sector = 0;
+ chunknode++;
+ }
+
+ /* fs cluster size >= 2048, so don't need to proc align */
+
+ /* align by 2048 */
+ chunknode = wimchunk.chunk + wimchunk.cur_chunk - 1;
+ i = (chunknode->disk_end_sector + 1 - chunknode->disk_start_sector) % 4;
+ if (i)
+ {
+ chunknode->disk_end_sector += 4 - i;
+ }
+
+ isosector = (grub_uint32_t)((file->size + 2047) / 2048);
+ for (i = 0; i < wimchunk.cur_chunk; i++)
+ {
+ chunknode = wimchunk.chunk + i;
+ chunknode->img_start_sector = isosector;
+ chunknode->img_end_sector = chunknode->img_start_sector +
+ ((chunknode->disk_end_sector + 1 - chunknode->disk_start_sector) / 4) - 1;
+ isosector = chunknode->img_end_sector + 1;
+ }
+
+ grub_memcpy((char *)chain + chain->img_chunk_offset + img_chunk1_size, wimchunk.chunk, img_chunk2_size);
+
+ /* part 4: override chunk */
+ chain->override_chunk_offset = chain->img_chunk_offset + img_chunk1_size + img_chunk2_size;
+ chain->override_chunk_num = 1;
- debug("wim chain data begin <%s> ...\n", args[0]);
+ override = (ventoy_override_chunk *)((char *)chain + chain->override_chunk_offset);
+ override->img_offset = 0;
+ override->override_size = g_wimiso_size;
+
+ grub_file_seek(file, 0);
+ grub_file_read(file, override->override_data, file->size);
+
+ dirent = (ventoy_iso9660_override *)(override->override_data + imgoffset);
+ dirent->first_sector = (grub_uint32_t)((file->size + 2047) / 2048);
+ dirent->size = (grub_uint32_t)(wimsize);
+ dirent->first_sector_be = grub_swap_bytes32(dirent->first_sector);
+ dirent->size_be = grub_swap_bytes32(dirent->size);
+
+ debug("imgoffset=%u first_sector=0x%x size=0x%x\n", imgoffset, dirent->first_sector, dirent->size);
+
+ if (ventoy_is_efi_os() == 0)
+ {
+ ventoy_windows_drive_map(chain);
+ }
+
+ grub_file_close(file);
+
+ VENTOY_CMD_RETURN(GRUB_ERR_NONE);
+}
+
+static grub_err_t ventoy_normal_wim_chain_data(grub_file_t wimfile)
+{
+ grub_uint32_t i = 0;
+ grub_uint32_t imgoffset = 0;
+ grub_uint32_t size = 0;
+ grub_uint32_t isosector = 0;
+ grub_uint64_t wimsize = 0;
+ grub_uint32_t boot_catlog = 0;
+ grub_uint32_t img_chunk1_size = 0;
+ grub_uint32_t img_chunk2_size = 0;
+ grub_uint32_t override_size = 0;
+ grub_file_t file;
+ grub_disk_t disk;
+ const char *pLastChain = NULL;
+ ventoy_chain_head *chain;
+ ventoy_iso9660_override *dirent;
+ ventoy_img_chunk *chunknode;
+ ventoy_override_chunk *override;
+ ventoy_img_chunk_list wimchunk;
+ char envbuf[128];
+
+ debug("normal wim chain data begin <%s> ...\n", wimfile->name);
if (NULL == g_wimiso_chunk_list.chunk || NULL == g_wimiso_path)
{
return 1;
}
- if (0 != ventoy_get_wim_chunklist(args[0], &wimchunk, &wimsize))
+ if (0 != ventoy_get_wim_chunklist(wimfile, &wimchunk))
{
grub_printf("Failed to get wim chunklist\n");
return 1;
}
+ wimsize = wimfile->size;
file = ventoy_grub_file_open(VENTOY_FILE_TYPE, "%s", g_wimiso_path);
if (!file)
VENTOY_CMD_RETURN(GRUB_ERR_NONE);
}
+grub_err_t ventoy_cmd_wim_chain_data(grub_extcmd_context_t ctxt, int argc, char **args)
+{
+ grub_err_t ret;
+ grub_file_t wimfile;
+
+ (void)ctxt;
+ (void)argc;
+
+ wimfile = grub_file_open(args[0], VENTOY_FILE_TYPE);
+ if (!wimfile)
+ {
+ return 1;
+ }
+
+ if (wimfile->vlnk)
+ {
+ ret = ventoy_vlnk_wim_chain_data(wimfile);
+ }
+ else
+ {
+ ret = ventoy_normal_wim_chain_data(wimfile);
+ }
+
+ grub_file_close(wimfile);
+ return ret;
+}
+
int ventoy_chain_file_size(const char *path)
{
int size;
/* --skip-sig is specified. */
GRUB_FILE_TYPE_SKIP_SIGNATURE = 0x10000,
- GRUB_FILE_TYPE_NO_DECOMPRESS = 0x20000
+ GRUB_FILE_TYPE_NO_DECOMPRESS = 0x20000,
+ GRUB_FILE_TYPE_NO_VLNK = 0x40000,
};
/* File description. */
/* File name. */
char *name;
+ int vlnk;
+
/* The underlying device. */
grub_device_t device;
grub_off_t EXPORT_FUNC(grub_file_seek) (grub_file_t file, grub_off_t offset);
grub_err_t EXPORT_FUNC(grub_file_close) (grub_file_t file);
+int EXPORT_FUNC(grub_file_is_vlnk_suffix)(const char *name, int len);
+int EXPORT_FUNC(grub_file_add_vlnk)(const char *src, const char *dst);
+const char * EXPORT_FUNC(grub_file_get_vlnk)(const char *name, int *vlnk);
+
/* Return value of grub_file_size() in case file size is unknown. */
#define GRUB_FILE_SIZE_UNKNOWN 0xffffffffffffffffULL
#define FOR_FILESYSTEMS(var) FOR_LIST_ELEMENTS((var), (grub_fs_list))
grub_fs_t EXPORT_FUNC(grub_fs_probe) (grub_device_t device);
+grub_fs_t EXPORT_FUNC(grub_fs_list_probe) (grub_device_t device, const char **list);
#endif /* ! GRUB_FS_HEADER */
* vtoy_reserved[3]: vtoy_iso_format 0:iso9660 1:udf
* vtoy_reserved[4]: vtoy_windows_cd_prompt
* vtoy_reserved[5]: vtoy_linux_remount
+ * vtoy_reserved[6]: vtoy_vlnk
+ * vtoy_reserved[7~10]: vtoy_disk_sig[4] used for vlnk
*
*/
grub_uint8_t vtoy_reserved[32]; // Internal use by ventoy
grub_uint8_t magic2[16]; /* VENTOY_GUID */
}ventoy_secure_data;
+
+typedef struct ventoy_vlnk
+{
+ ventoy_guid guid; // VENTOY_GUID
+ grub_uint32_t crc32; // crc32
+ grub_uint32_t disk_signature;
+ grub_uint64_t part_offset; // in bytes
+ char filepath[384];
+ grub_uint8_t reserved[96];
+}ventoy_vlnk;
+
#pragma pack()
// compile assert check : sizeof(ventoy_os_param) must be 512
COMPILE_ASSERT(1,sizeof(ventoy_os_param) == 512);
COMPILE_ASSERT(2,sizeof(ventoy_secure_data) == 4096);
+COMPILE_ASSERT(3,sizeof(ventoy_vlnk) == 512);
VTOY_DM_PATH=/dev/mapper/ventoy
VTOY_DEBUG_LEVEL=$($BUSYBOX_PATH/hexdump -n 1 -s 450 -e '1/1 "%02x"' $VTOY_PATH/ventoy_os_param)
VTOY_LINUX_REMOUNT=$($BUSYBOX_PATH/hexdump -n 1 -s 454 -e '1/1 "%02x"' $VTOY_PATH/ventoy_os_param)
+VTOY_VLNK_BOOT=$($BUSYBOX_PATH/hexdump -n 1 -s 455 -e '1/1 "%02x"' $VTOY_PATH/ventoy_os_param)
if [ "$VTOY_DEBUG_LEVEL" = "01" ]; then
if [ -e /dev/console ]; then
echo 'Y' > $VTOY_PATH/hook_finish
}
-get_ventoy_disk_name() {
- line=$($VTOY_PATH/tool/vtoydump -f /ventoy/ventoy_os_param)
- if [ $? -eq 0 ]; then
- echo ${line%%#*}
- else
- echo "unknown"
+get_ventoy_disk_name() {
+ if [ "$VTOY_VLNK_BOOT" = "01" ]; then
+ $VTOY_PATH/tool/vtoydump -t /ventoy/ventoy_os_param
+ else
+ line=$($VTOY_PATH/tool/vtoydump -f /ventoy/ventoy_os_param)
+ if [ $? -eq 0 ]; then
+ echo ${line%%#*}
+ else
+ echo "unknown"
+ fi
fi
}
[ -e "${vtpart2}" ]
}
-is_ventoy_disk() {
- if $VTOY_PATH/tool/vtoydump -f $VTOY_PATH/ventoy_os_param -c "$1"; then
- $BUSYBOX_PATH/true
- else
- $BUSYBOX_PATH/false
- fi
-}
-
not_ventoy_disk() {
if echo $1 | $EGREP -q "nvme.*p$|mmc.*p$|nbd.*p$"; then
vtDiskName=${1:0:-1}
vtDiskName=$1
fi
- if $VTOY_PATH/tool/vtoydump -f $VTOY_PATH/ventoy_os_param -c "$vtDiskName"; then
- $BUSYBOX_PATH/false
+ if [ "$VTOY_VLNK_BOOT" = "01" ]; then
+ vtVtoyDisk=$($VTOY_PATH/tool/vtoydump -t $VTOY_PATH/ventoy_os_param)
+ [ "$vtVtoyDisk" != "/dev/$vtDiskName" ]
else
- $BUSYBOX_PATH/true
+ if $VTOY_PATH/tool/vtoydump -f $VTOY_PATH/ventoy_os_param -c "$vtDiskName"; then
+ $BUSYBOX_PATH/false
+ else
+ $BUSYBOX_PATH/true
+ fi
fi
}
fi
if [ -e /dev/$vtShortName ]; then
- $VTOY_PATH/tool/vtoydump -f $VTOY_PATH/ventoy_os_param -c $vtShortName
+ if [ "$VTOY_VLNK_BOOT" = "01" ]; then
+ vtOrgDiskName=$($VTOY_PATH/tool/vtoydump -t $VTOY_PATH/ventoy_os_param)
+ [ "$vtOrgDiskName" = "/dev/$vtShortName" ]
+ else
+ $VTOY_PATH/tool/vtoydump -f $VTOY_PATH/ventoy_os_param -c $vtShortName
+ fi
return
fi
fi
cp -a /dev/mapper/ventoy* /newdev/mapper/
cp -a /dev/ventoy* /newdev/
+if [ "$VTOY_VLNK_BOOT" = "01" ]; then
+ vtLine=$($VTOY_PATH/tool/vtoydump -f /ventoy/ventoy_os_param)
+ vtdiskname=${vtLine%%#*}
+fi
vtshortname="${vtdiskname#/dev/}"
mv /newdev/${vtshortname} /newdev/backup_${vtshortname}
cp -a /dev/mapper/ventoy* /newdev/mapper/
cp -a /dev/ventoy* /newdev/
+if [ "$VTOY_VLNK_BOOT" = "01" ]; then
+ vtLine=$($VTOY_PATH/tool/vtoydump -f /ventoy/ventoy_os_param)
+ vtdiskname=${vtLine%%#*}
+fi
vtshortname="${vtdiskname#/dev/}"
mv /newdev/${vtshortname} /newdev/backup_${vtshortname}
1. sudo sh VentoyPlugson.sh
2. open your browser and visit http://127.0.0.1:24681
+========== VentoyVlnk.sh ===============
+Usage: sudo sh VentoyVlnk.sh CMD FILE
+ CMD:
+ -c FILE create vlnk for FILE
+ -l VLNK parse vlnk file
+ -v print verbose info
+ -h print this help
+
========= VentoyGUI ===================
VentoyGUI is native GUI program for Linux (GTK/QT)
return
fi
- vt_concat_efi_iso "${vtoy_iso_part}${vt_chosen_path}" vtoy_iso_buf
+ unset vt_vlnk_dst
+ if vt_is_vlnk_name "${vt_chosen_path}"; then
+ vt_get_vlnk_dst "${vtoy_iso_part}${vt_chosen_path}" vt_vlnk_dst
+ if [ -z "$vt_vlnk_dst" ]; then
+ echo -e "\n### VLNK FILE NOT FOUND ###\n### VLNK 文件不存在 ###\n"
+ ventoy_pause
+ return
+ fi
+ else
+ vt_vlnk_dst="${vtoy_iso_part}${vt_chosen_path}"
+ fi
+
+ vt_concat_efi_iso "${vt_vlnk_dst}" vtoy_iso_buf
ventoy_debug_pause
if [ -n "$vtoy_dotefi_retry" ]; then
unset vtoy_dotefi_retry
- chainloader "${vtoy_iso_part}${vt_chosen_path}"
+ chainloader "${vt_vlnk_dst}"
boot
fi
}
function vhd_common_menuentry {
-
- if [ "$VTOY_VHD_NO_WARNING" != "1" ]; then
- if [ "$vtoy_iso_fs" != "ntfs" ]; then
- echo -e "!!! WARNING !!!\n"
- echo -e "\nPartition1 ($vtoy_iso_fs) is NOT ntfs, the VHD(x) file may not boot normally \n"
- echo -e "\nVHD(x) 文件所在分区不是 ntfs 格式, 可能无法正常启动 \n\n"
- echo -n "press ENTER to continue boot (请按 回车 键继续) ..."
- read vtInputKey
- fi
- fi
-
vt_chosen_img_path vt_chosen_path vt_chosen_size
if vt_check_password "${vt_chosen_path}"; then
return
fi
- vhdboot_common_func "${vt_chosen_path}"
+ unset vt_vlnk_dst
+ if vt_is_vlnk_name "${vt_chosen_path}"; then
+ vt_get_vlnk_dst "${vtoy_iso_part}${vt_chosen_path}" vt_vlnk_dst
+ if [ -z "$vt_vlnk_dst" ]; then
+ echo -e "\n### VLNK FILE NOT FOUND ###\n### VLNK 文件不存在 ###\n"
+ ventoy_pause
+ return
+ fi
+ else
+ vt_vlnk_dst="${vt_chosen_path}"
+ if [ "$VTOY_VHD_NO_WARNING" != "1" ]; then
+ if [ "$vtoy_iso_fs" != "ntfs" ]; then
+ echo -e "!!! WARNING !!!\n"
+ echo -e "\nPartition1 ($vtoy_iso_fs) is NOT ntfs, the VHD(x) file may not boot normally \n"
+ echo -e "\nVHD(x) 文件所在分区不是 ntfs 格式, 可能无法正常启动 \n\n"
+ echo -n "press ENTER to continue boot (请按 回车 键继续) ..."
+ read vtInputKey
+ fi
+ fi
+ fi
+
+ vhdboot_common_func "${vt_vlnk_dst}"
}
function vhd_unsupport_menuentry {
return
fi
- vtoyboot_common_func "${vtoy_iso_part}${vt_chosen_path}"
+ unset vt_vlnk_dst
+ if vt_is_vlnk_name "${vt_chosen_path}"; then
+ vt_get_vlnk_dst "${vtoy_iso_part}${vt_chosen_path}" vt_vlnk_dst
+ if [ -z "$vt_vlnk_dst" ]; then
+ echo -e "\n### VLNK FILE NOT FOUND ###\n### VLNK 文件不存在 ###\n"
+ ventoy_pause
+ return
+ fi
+ else
+ vt_vlnk_dst="${vtoy_iso_part}${vt_chosen_path}"
+ fi
+
+ vtoyboot_common_func "${vt_vlnk_dst}"
}
function vtoy_unsupport_menuentry {
#############################################################
#############################################################
-set VENTOY_VERSION="1.0.65"
+set VENTOY_VERSION="1.0.66"
#ACPI not compatible with Window7/8, so disable by default
set VTOY_PARAM_NO_ACPI=1
# Default menu display mode, you can change it as you want.
# 0: List mode
# 1: TreeView mode
-set VTOY_DEFAULT_MENU_MODE=0
+set VTOY_DEFAULT_MENU_MODE=1
set VTOY_MEM_DISK_STR="[Memdisk]"
set VTOY_ISO_RAW_STR="Compatible Mode"
set VTOY_HOTKEY_TIP="h:Help F1:Memdisk F2:Power F3:ListView F4:Localboot F5:Tools F6:ExMenu"
fi
+terminal_output console
if [ -n "$vtoy_gfxmode" ]; then
set gfxmode=$vtoy_gfxmode
crypto:
part_bsd: part_msdos
cs5536:
-ventoy: elf fshelp ext2 btrfs font crypto gcry_md5 exfat udf extcmd datetime div normal video gcry_sha1 iso9660
+ventoy: fshelp ext2 elf btrfs font crypto gcry_md5 exfat udf div extcmd datetime normal video gcry_sha1 iso9660
gcry_sha512: crypto
password: crypto normal
fshelp:
part_bsd: part_msdos
cs5536: pci
biosdisk:
-ventoy: elf fshelp ext2 btrfs font crypto gcry_md5 exfat udf extcmd datetime div normal video gcry_sha1 iso9660 acpi
+ventoy: elf fshelp ext2 btrfs font crypto gcry_md5 exfat udf div extcmd datetime normal video gcry_sha1 iso9660 acpi
lsapm:
gcry_sha512: crypto
password: crypto normal
crypto:
part_bsd: part_msdos
cs5536:
-ventoy: elf fshelp ext2 btrfs font crypto gcry_md5 exfat udf extcmd datetime div normal video gcry_sha1 iso9660
+ventoy: fshelp ext2 elf btrfs font crypto gcry_md5 exfat udf div extcmd datetime normal video gcry_sha1 iso9660
gcry_sha512: crypto
password: crypto normal
fshelp:
sh pack.sh
cd -
+cd ../Vlnk
+sh build.sh
+sh pack.sh
+cd -
+
LOOP=$(losetup -f)
cp $OPT Ventoy2Disk.sh $tmpdir/
cp $OPT VentoyWeb.sh $tmpdir/
cp $OPT VentoyPlugson.sh $tmpdir/
+cp $OPT VentoyVlnk.sh $tmpdir/
cp $OPT VentoyGUI* $tmpdir/
dos2unix -q $tmpdir/Ventoy2Disk.sh
dos2unix -q $tmpdir/VentoyWeb.sh
dos2unix -q $tmpdir/VentoyPlugson.sh
+dos2unix -q $tmpdir/VentoyVlnk.sh
dos2unix -q $tmpdir/CreatePersistentImg.sh
chmod +x $tmpdir/Ventoy2Disk.sh
chmod +x $tmpdir/VentoyWeb.sh
chmod +x $tmpdir/VentoyPlugson.sh
+chmod +x $tmpdir/VentoyVlnk.sh
chmod +x $tmpdir/VentoyGUI*
cp $OPT $LANG_DIR/languages.json $tmpdir/tool/
cp $OPT Ventoy2Disk.exe $tmpdir/
cp $OPT VentoyPlugson.exe $tmpdir/
+cp $OPT VentoyVlnk.exe $tmpdir/
cp $OPT FOR_X64_ARM.txt $tmpdir/
mkdir -p $tmpdir/altexe
cp $OPT Ventoy2Disk_*.exe $tmpdir/altexe/
"STR_PART_RESIZE_FAILED":"Hi ha hagut un error durant el procés d'instal·lació no destructiva, comproveu el fitxer log.txt per a més detalls.",
"STR_PART_RESIZE_UNSUPPORTED":"La instal·lació no destructiva de Ventoy no ha estat possible perquè els requeriments no es compleixen. Comproveu el fitxer log.txt per a més detalls.",
+ "STRXXX":""
+ },
+
+ {
+ "name":"Georgian (ქართული)",
+ "FontFamily":"Courier New",
+ "FontSize":16,
+ "Author":"Levan Baindurashvili",
+
+ "STR_ERROR":"შეცდომა",
+ "STR_WARNING":"გაფრთხილება",
+ "STR_INFO":"ინფორმაცია",
+ "STR_INCORRECT_DIR":"გთხოვთ, გაუშვით სხვა კატალოგში!",
+ "STR_INCORRECT_TREE_DIR":"ნუ გამიშვებ აქ, გთხოვთ გადმოწეროთ გამოშვებული ინსტალაციის პაკეტი და გაუშვათ სხვაგან.",
+ "STR_DEVICE":"მოწყობილობა",
+ "STR_LOCAL_VER":"Ventoy პაკეტში",
+ "STR_DISK_VER":"Ventoy მოწყობილობაზე",
+ "STR_STATUS":"სტატუსი - მზადაა",
+ "STR_INSTALL":"დაყენება",
+ "STR_UPDATE":"განახლება",
+ "STR_UPDATE_TIP":"განახლება უსაფრთხოა და ISO ფაილები უცვლელი დარჩება.#@გაგრძელება?",
+ "STR_INSTALL_TIP":"დისკი დაფორმატდება და ყველა მონაცემი დაიკარგება.#@გსურთ გააგრძელოთ?",
+ "STR_INSTALL_TIP2":"დისკი დაფორმატდება და ყველა მონაცემი დაიკარგება.#@ ნამდვილად გსურთ გაგრძელება?",
+ "STR_INSTALL_SUCCESS":"გილოცავთ!#@Ventoy წარმატებით დაინსტალირდა თქვენს მოწყობილობაზე.",
+ "STR_INSTALL_FAILED":"Ventoy-ის ინსტალაციისას მოხდა შეცდომა. ხელახლა შეაერთეთ თქვენი მოწყობილობა და სცადეთ ხელახლა. შეამოწმეთ log.txt შეცდომებისთვის.",
+ "STR_UPDATE_SUCCESS":"გილოცავთ!#@Ventoy წარმატებით განახლდა თქვენს მოწყობილობაზე.",
+ "STR_UPDATE_FAILED":"Ventoy-ის განახლებისას მოხდა შეცდომა. ხელახლა შეაერთეთ თქვენი მოწყობილობა და სცადეთ ხელახლა. შეამოწმეთ log.txt შეცდომებისთვის.",
+ "STR_WAIT_PROCESS":"პროცესი დაიწყო, გთხოვთ დაელოდოთ...",
+ "STR_MENU_OPTION":"Პარამეტრები",
+ "STR_MENU_SECURE_BOOT":"Secure Boot-ის მხარდაჭერა",
+ "STR_MENU_PART_CFG":"დამატებითი განყოფილება",
+ "STR_BTN_OK":"ОК",
+ "STR_BTN_CANCEL":"უარყოფა",
+ "STR_PRESERVE_SPACE":"შექმენით დამატებითი დანაყოფი დისკის ბოლოს",
+ "STR_SPACE_VAL_INVALID":"დანაყოფის ზომა არასწორია",
+ "STR_MENU_CLEAR":"Ventoy-ს წაშლა",
+ "STR_CLEAR_SUCCESS":"Ventoy წარმატებით წაიშალა მოწყობილობიდან.",
+ "STR_CLEAR_FAILED":"Ventoy-ის დეინსტალაციისას მოხდა შეცდომა. ხელახლა შეაერთეთ თქვენი მოწყობილობა და სცადეთ ხელახლა. შეამოწმეთ log.txt შეცდომებზე.",
+ "STR_MENU_PART_STYLE":"განყოფილების განლაგების სტილი",
+ "STR_DISK_2TB_MBR_ERROR":"გთხოვთ, აირჩიოთ GPT 2 ტბაიტზე მეტი დისკებისთვის",
+ "STR_SHOW_ALL_DEV":"ყველა მოწყობილობის ჩვენება",
+ "STR_PART_ALIGN_4KB":"ტიხრების გასწორება 4 KB ზომით",
+ "STR_WEB_COMMUNICATION_ERR":"დაკავშირების შეცდომა:",
+ "STR_WEB_REMOTE_ABNORMAL":"კომუნიკაციის შეცდომა: დისტანციური კავშირი არასწორია",
+ "STR_WEB_REQUEST_TIMEOUT":"კომუნიკაციის შეცდომა: მოთხოვნის დრო ამოიწურა",
+ "STR_WEB_SERVICE_UNAVAILABLE":"კომუნიკაციის შეცდომა: სერვისი მიუწვდომელია",
+ "STR_WEB_TOKEN_MISMATCH":"დემონის სტატუსი განახლებულია. გთხოვთ სცადოთ მოგვიანებით.",
+ "STR_WEB_SERVICE_BUSY":"სერვისი დაკავებულია, გთხოვთ, სცადოთ მოგვიანებით.",
+ "STR_MENU_VTSI_CREATE":"VTSI ფაილის შექმნა",
+ "STR_VTSI_CREATE_TIP":"ამჯერად Ventoy არ დაინსტალირდება მოწყობილობაზე, მაგრამ შეიქმნება მხოლოდ VTSI ფაილი#@გაგრძელება?",
+ "STR_VTSI_CREATE_SUCCESS":"VTSI ფაილი წარმატებით შეიქმნა!#@შეგიძლიათ გამოიყენოთ Rufus(3.15+) ჩაწეროთ მოწყობილობაზე Ventoy ინსტალაციის დასასრულებლად.",
+ "STR_VTSI_CREATE_FAILED":"VTSI ფაილის შექმნის შეცდომა.",
+ "STR_MENU_PART_RESIZE":"Non-destructive Install",
+ "STR_PART_RESIZE_TIP":"Ventoy will try non-destructive installation if possible. #@Continue?",
+ "STR_PART_RESIZE_SUCCESS":"Congratulations!#@Ventoy non-destructive installation successfully finished.",
+ "STR_PART_RESIZE_FAILED":"Non-destructive installation failed, Check log.txt for details.",
+ "STR_PART_RESIZE_UNSUPPORTED":"Ventoy non-destructive installation stopped because some conditions cannot be met. Check log.txt for details.",
+
"STRXXX":""
}
]
{
memset(data, 0, sizeof(data_control));
+ data->default_menu_mode = 1;
data->filter_dot_underscore = 1;
data->max_search_level = -1;
data->menu_timeout = 0;
--- /dev/null
+#!/bin/sh
+
+print_usage() {
+ echo 'Usage: sudo sh VentoyVlnk.sh CMD FILE'
+ echo ' CMD:'
+ echo ' -c FILE create vlnk for FILE'
+ echo ' -l VLNK parse vlnk file'
+ echo ' -v print verbose info'
+ echo ' -h print this help'
+ echo ''
+}
+
+verbose_flag=0
+
+vlog() {
+ if [ $verbose_flag -eq 1 ]; then
+ echo "$@"
+ fi
+}
+
+vlnk_suffix() {
+ echo $1 | egrep -q '.*(.vlnk.iso|.vlnk.img|.vlnk.wim|.vlnk.vhd|.vlnk.vhdx|.vlnk.efi|.vlnk.vtoy|.vlnk.dat)$'
+}
+
+
+uid=$(id -u)
+if [ $uid -ne 0 ]; then
+ echo "Please use sudo or run the script as root."
+ exit 1
+fi
+
+#check system tools used bellow
+for t in 'mountpoint' 'readlink' 'xzcat'; do
+ if ! which "$t" > /dev/null 2>&1; then
+ echo "$t command not found in current system!"
+ exit 1
+ fi
+done
+
+machine=$(uname -m)
+if echo $machine | egrep -q 'aarch64|arm64'; then
+ TOOLDIR=aarch64
+elif echo $machine | egrep -q 'x86_64|amd64'; then
+ TOOLDIR=x86_64
+elif echo $machine | egrep -q 'mips64'; then
+ TOOLDIR=mips64el
+elif echo $machine | egrep -q 'i[3-6]86'; then
+ TOOLDIR=i386
+else
+ echo "Unsupported machine type $machine"
+ exit 1
+fi
+
+fullsh=$(readlink -f "$0")
+vtoydir=${fullsh%/*}
+
+if [ -f "$vtoydir/tool/$TOOLDIR/vlnk.xz" ]; then
+ xzcat "$vtoydir/tool/$TOOLDIR/vlnk.xz" > "$vtoydir/tool/$TOOLDIR/vlnk"
+ rm -f "$vtoydir/tool/$TOOLDIR/vlnk.xz"
+fi
+
+if [ -f "$vtoydir/tool/$TOOLDIR/vlnk" ]; then
+ chmod +x "$vtoydir/tool/$TOOLDIR/vlnk"
+else
+ echo "$vtoydir/tool/$TOOLDIR/vlnk does not exist! "
+ exit 1
+fi
+
+PATH="$vtoydir/tool/$TOOLDIR":$PATH
+
+VLNKCMD=vlnk
+while [ -n "$1" ]; do
+ if [ "$1" = "-h" ] || [ "$1" = "--help" ]; then
+ print_usage
+ exit 0
+ elif [ "$1" = "-c" ]; then
+ shift
+ CMD='c'
+ IMG="$1"
+ elif [ "$1" = "-o" ]; then
+ shift
+ OUT="$1"
+ elif [ "$1" = "-l" ]; then
+ shift
+ CMD='l'
+ VLNK="$1"
+ elif [ "$1" = "-v" ]; then
+ verbose_flag=1
+ VLNKCMD="vlnk -v"
+ else
+ echo "invalid option $1"
+ exit 1
+ fi
+
+ shift
+done
+
+if [ "$CMD" = "c" ]; then
+ vlog "Create vlnk for $IMG"
+
+ if [ ! -f "$IMG" ]; then
+ echo "$IMG does not exist!"
+ exit 1
+ fi
+
+ if echo $IMG | egrep -q -i '.*(.iso|.img|.wim|.vhd|.vhdx|.efi|.vtoy|.dat)$'; then
+ :
+ else
+ echo "This file is not supported for vlnk!"
+ exit 1
+ fi
+
+ if vlnk_suffix "$IMG"; then
+ echo "This is already a vlnk file!"
+ exit 1
+ fi
+
+ if $VLNKCMD -t "$IMG"; then
+ echo "This is already a vlnk file!"
+ exit 1
+ fi
+
+ FULLIMG=$(readlink -f "$IMG")
+ if [ ! -f "$FULLIMG" ]; then
+ echo "$FULLIMG does not exist!"
+ exit 1
+ fi
+ vlog "Full file path is $FULLIMG"
+
+
+ #check img file position is a valid mountpoint
+ FULLDIR=${FULLIMG%/*}
+ while [ -n "${FULLDIR}" ]; do
+ if mountpoint -q "${FULLDIR}"; then
+ break
+ fi
+ FULLDIR="${FULLDIR%/*}"
+ done
+
+ if [ -z "${FULLDIR}" ]; then
+ FULLDIR=/
+ IMGPATH="${FULLIMG}"
+ else
+ IMGPATH="${FULLIMG#$FULLDIR}"
+ fi
+
+ IMGFILE=$(basename "$IMGPATH")
+ vlog "IMGPATH=$IMGPATH IMGFILE=$IMGFILE"
+
+
+ mntdev=$(mountpoint -d "${FULLDIR}")
+ vlog "mountpoint is ${FULLDIR} dev $mntdev"
+
+ #check fs
+ if grep -q " ${FULLDIR} " /proc/mounts; then
+ FS=$(grep " ${FULLDIR} " /proc/mounts | awk '{print $3}')
+ vlog "File system is $FS"
+
+ if echo $FS | egrep -q "ext2|ext3|ext4|exfat|vfat|fat32|fat16|fat12|ntfs|xfs|udf"; then
+ vlog "FS OK"
+ else
+ echo "$FS is not supported!"
+ exit 1
+ fi
+ else
+ echo "${FULLDIR} not found in /proc/mounts"
+ exit 1
+ fi
+
+
+ Major=$(echo $mntdev | awk -F: '{print $1}')
+ Minor=$(echo $mntdev | awk -F: '{print $2}')
+ vlog "Major=$Major Minor=$Minor"
+
+ IMGPARTITION=""
+ while read line; do
+ M1=$(echo $line | awk '{print $1}')
+ M2=$(echo $line | awk '{print $2}')
+ if [ "$Major" = "$M1" -a "$Minor" = "$M2" ]; then
+ IMGPARTITION=$(echo $line | awk '{print $4}')
+ vlog "disk partition is $IMGPARTITION"
+ break
+ fi
+ done < /proc/partitions
+
+ if [ -z "$IMGPARTITION" ]; then
+ echo "Disk partition not found for $FULLDIR"
+ grep " $FULLDIR " /proc/mounts
+ exit 1
+ fi
+
+ if [ -f "/sys/class/block/$IMGPARTITION/start" ]; then
+ PARTSTART=$(cat "/sys/class/block/$IMGPARTITION/start")
+ if echo $IMGPARTITION | egrep -q 'mmc|nbd|nvme'; then
+ DISK=$(echo /dev/$IMGPARTITION | sed "s/^\(.*\)p[0-9][0-9]*$/\1/")
+ else
+ DISK=$(echo /dev/$IMGPARTITION | sed "s/^\(.*[^0-9]\)[0-9][0-9]*$/\1/")
+ fi
+
+ if [ ! -b $DISK ]; then
+ echo "Device $DISK not exist!"
+ exit 1
+ fi
+
+ vlog "PARTSTART=$PARTSTART DISK=$DISK"
+ else
+ if echo $IMGPARTITION | grep -q '^dm-[0-9][0-9]*'; then
+ echo "LVM/DM is not supported!"
+ fi
+ echo "/sys/class/block/$IMGPARTITION/start not exist!"
+ exit 1
+ fi
+
+
+ if [ -n "$OUT" ]; then
+ lowersuffix=$(echo ${IMG##*.} | tr 'A-Z' 'a-z')
+ OUT="${OUT}.vlnk.${lowersuffix}"
+ else
+ name=${IMGFILE%.*}
+ lowersuffix=$(echo ${IMGFILE##*.} | tr 'A-Z' 'a-z')
+ OUT="${name}.vlnk.${lowersuffix}"
+ fi
+
+ echo "Output VLNK file is $OUT"
+ [ -f "${OUT}" ] && rm -f "${OUT}"
+
+ touch "${OUT}"
+ if [ -f "${OUT}" ]; then
+ rm -f "${OUT}"
+ else
+ echo "Failed to create ${OUT}"
+ exit 1
+ fi
+
+ if $VLNKCMD -c "$IMGPATH" -d $DISK -p $PARTSTART -o "${OUT}"; then
+ echo "====== Vlnk file create success ========="
+ echo ""
+ else
+ echo "====== Vlnk file create failed ========="
+ echo ""
+ exit 1
+ fi
+elif [ "$CMD" = "l" ]; then
+ vlog "Parse vlnk for $VLNK"
+
+ if [ ! -f "$VLNK" ]; then
+ echo "$VLNK does not exist!"
+ exit 1
+ fi
+
+ if vlnk_suffix "$VLNK"; then
+ :
+ else
+ echo "Invalid vlnk file suffix!"
+ exit 1
+ fi
+
+ if $VLNKCMD -t "$VLNK"; then
+ vlog "Vlnk data check OK"
+ else
+ echo "This is not a valid vlnk file!"
+ exit 1
+ fi
+
+ $VLNKCMD -l "$VLNK"
+ echo ""
+else
+ echo "invalid cmd"
+ print_usage
+ exit 1
+fi
--- /dev/null
+#!/bin/sh
+
+rm -f vlnk_64
+rm -f vlnk_32
+rm -f vlnk_aa64
+rm -f vlnk_m64e
+
+SRCS="src/crc32.c src/main_linux.c src/vlnk.c"
+
+gcc -specs "/usr/local/musl/lib/musl-gcc.specs" -Os -static -D_FILE_OFFSET_BITS=64 $SRCS -Isrc -o vlnk_64
+
+/opt/diet32/bin/diet -Os gcc -D_FILE_OFFSET_BITS=64 -m32 -static $SRCS -Isrc -o vlnk_32
+
+aarch64-buildroot-linux-uclibc-gcc -static -O2 -D_FILE_OFFSET_BITS=64 $SRCS -Isrc -o vlnk_aa64
+mips64el-linux-musl-gcc -mips64r2 -mabi=64 -static -O2 -D_FILE_OFFSET_BITS=64 $SRCS -Isrc -o vlnk_m64e
+
+if [ -e vlnk_64 ] && [ -e vlnk_32 ] && [ -e vlnk_aa64 ] && [ -e vlnk_m64e ]; then
+ echo -e "\n===== success =======\n"
+
+ strip --strip-all vlnk_32
+ strip --strip-all vlnk_64
+ aarch64-buildroot-linux-uclibc-strip --strip-all vlnk_aa64
+ mips64el-linux-musl-strip --strip-all vlnk_m64e
+
+ [ -d ../INSTALL/tool/i386/ ] && mv vlnk_32 ../INSTALL/tool/i386/vlnk
+ [ -d ../INSTALL/tool/x86_64/ ] && mv vlnk_64 ../INSTALL/tool/x86_64/vlnk
+ [ -d ../INSTALL/tool/aarch64/ ] && mv vlnk_aa64 ../INSTALL/tool/aarch64/vlnk
+ [ -d ../INSTALL/tool/mips64el/ ] && mv vlnk_m64e ../INSTALL/tool/mips64el/vlnk
+else
+ echo -e "\n===== failed =======\n"
+ exit 1
+fi
--- /dev/null
+#!/bin/sh
+
+rm -f ../INSTALL/VentoyVlnk.sh
+cp -a ./VentoyVlnk.sh ../INSTALL/VentoyVlnk.sh
+
+rm -f ../INSTALL/VentoyVlnk.exe
+cp -a ./vs/VentoyVlnk/Release/VentoyVlnk.exe ../INSTALL/VentoyVlnk.exe
--- /dev/null
+/******************************************************************************
+ * crc32.c ----
+ *
+ * Copyright (c) 2022, 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 <stdint.h>
+#include <string.h>
+#include "vlnk.h"
+
+static uint32_t crc32c_table [256];
+
+/* Helper for init_crc32c_table. */
+static uint32_t reflect (uint32_t ref, int len)
+{
+ uint32_t result = 0;
+ int i;
+
+ for (i = 1; i <= len; i++)
+ {
+ if (ref & 1)
+ result |= 1 << (len - i);
+ ref >>= 1;
+ }
+
+ return result;
+}
+
+static void init_crc32c_table (void)
+{
+ uint32_t polynomial = 0x1edc6f41;
+ int i, j;
+
+ for(i = 0; i < 256; i++)
+ {
+ crc32c_table[i] = reflect(i, 8) << 24;
+ for (j = 0; j < 8; j++)
+ crc32c_table[i] = (crc32c_table[i] << 1) ^
+ (crc32c_table[i] & (1 << 31) ? polynomial : 0);
+ crc32c_table[i] = reflect(crc32c_table[i], 32);
+ }
+}
+
+uint32_t ventoy_getcrc32c (uint32_t crc, const void *buf, int size)
+{
+ int i;
+ const uint8_t *data = buf;
+
+ if (! crc32c_table[1])
+ init_crc32c_table ();
+
+ crc^= 0xffffffff;
+
+ for (i = 0; i < size; i++)
+ {
+ crc = (crc >> 8) ^ crc32c_table[(crc & 0xFF) ^ *data];
+ data++;
+ }
+
+ return crc ^ 0xffffffff;
+}
+
--- /dev/null
+#include <stdio.h>
+#include <stdlib.h>
+#include <stdint.h>
+#include <string.h>
+#include <errno.h>
+#include <fcntl.h>
+#include <sys/stat.h>
+#include <sys/mount.h>
+#include <sys/statfs.h>
+#include <dirent.h>
+#include <unistd.h>
+
+#include "vlnk.h"
+
+#ifndef PATH_MAX
+#define PATH_MAX 4096
+#endif
+
+#define IS_DIGIT(x) ((x) >= '0' && (x) <= '9')
+
+static int verbose = 0;
+#define debug(fmt, args...) if(verbose) printf(fmt, ##args)
+
+static uint8_t g_vlnk_buf[VLNK_FILE_LEN];
+
+static int64_t get_file_size(char *file)
+{
+ struct stat stStat;
+
+ if (stat(file, &stStat) < 0)
+ {
+ return -1;
+ }
+
+ return (int64_t)(stStat.st_size);
+}
+
+static int get_disk_sig(char *diskname, uint32_t *sig)
+{
+ int fd;
+ uint8_t buf[512] = {0};
+
+ fd = open(diskname, O_RDONLY);
+ if (fd < 0)
+ {
+ printf("Failed to open %s\n", diskname);
+ return 1;
+ }
+
+ read(fd, buf, 512);
+ close(fd);
+
+ memcpy(sig, buf + 0x1b8, 4);
+ return 0;
+}
+
+static int vtoy_get_disk_guid(const char *diskname, uint8_t *vtguid, uint8_t *vtsig)
+{
+ int i = 0;
+ int fd = 0;
+ char devdisk[128] = {0};
+
+ snprintf(devdisk, sizeof(devdisk) - 1, "/dev/%s", diskname);
+
+ fd = open(devdisk, O_RDONLY);
+ if (fd >= 0)
+ {
+ lseek(fd, 0x180, SEEK_SET);
+ read(fd, vtguid, 16);
+
+ lseek(fd, 0x1b8, SEEK_SET);
+ read(fd, vtsig, 4);
+ close(fd);
+ return 0;
+ }
+ else
+ {
+ debug("failed to open %s %d\n", devdisk, errno);
+ return errno;
+ }
+}
+
+static int vtoy_is_possible_blkdev(const char *name)
+{
+ if (name[0] == '.')
+ {
+ return 0;
+ }
+
+ /* /dev/ramX */
+ if (name[0] == 'r' && name[1] == 'a' && name[2] == 'm')
+ {
+ return 0;
+ }
+
+ /* /dev/loopX */
+ if (name[0] == 'l' && name[1] == 'o' && name[2] == 'o' && name[3] == 'p')
+ {
+ return 0;
+ }
+
+ /* /dev/dm-X */
+ if (name[0] == 'd' && name[1] == 'm' && name[2] == '-' && IS_DIGIT(name[3]))
+ {
+ return 0;
+ }
+
+ /* /dev/srX */
+ if (name[0] == 's' && name[1] == 'r' && IS_DIGIT(name[2]))
+ {
+ return 0;
+ }
+
+ return 1;
+}
+
+
+static int find_disk_by_sig(uint8_t *sig, char *diskname)
+{
+ int rc = 0;
+ int count = 0;
+ DIR* dir = NULL;
+ struct dirent* p = NULL;
+ uint8_t vtguid[16];
+ uint8_t vtsig[16];
+
+ dir = opendir("/sys/block");
+ if (!dir)
+ {
+ return 0;
+ }
+
+ while ((p = readdir(dir)) != NULL)
+ {
+ if (!vtoy_is_possible_blkdev(p->d_name))
+ {
+ continue;
+ }
+
+ rc = vtoy_get_disk_guid(p->d_name, vtguid, vtsig);
+ if (rc == 0 && memcmp(vtsig, sig, 4) == 0)
+ {
+ sprintf(diskname, "%s", p->d_name);
+ count++;
+ }
+ }
+ closedir(dir);
+
+ return count;
+}
+
+static uint64_t get_part_offset(char *partname)
+{
+ int fd;
+ uint64_t offset;
+ char buf[32] = {0};
+ char path[PATH_MAX];
+
+ snprintf(path, PATH_MAX - 1, "/sys/class/block/%s/start", partname);
+
+ fd = open(path, O_RDONLY);
+ if (fd < 0)
+ {
+ return 0;
+ }
+
+ read(fd, buf, sizeof(buf));
+ close(fd);
+
+ offset = (uint64_t)strtoull(buf, NULL, 10);
+ offset *= 512;
+
+ return offset;
+}
+
+static int create_vlnk(char *infile, char *diskname, uint64_t partoff, char *outfile)
+{
+ FILE *fp;
+ int len;
+ uint32_t sig = 0;
+
+ debug("create vlnk\n");
+
+ if (infile[0] == 0 || outfile[0] == 0 || diskname[0] == 0 || partoff == 0)
+ {
+ debug("Invalid parameters: %d %d %d %llu\n", infile[0], outfile[0], diskname[0], (unsigned long long)partoff);
+ return 1;
+ }
+
+ len = (int)strlen(infile);
+ if (len >= VLNK_NAME_MAX)
+ {
+ printf("File name length %d is too long for vlnk!\n", len);
+ return 1;
+ }
+
+ if (get_disk_sig(diskname, &sig))
+ {
+ printf("Failed to read disk sig\n");
+ return 1;
+ }
+
+ fp = fopen(outfile, "wb+");
+ if (!fp)
+ {
+ printf("Failed to create file %s\n", outfile);
+ return 1;
+ }
+
+ memset(g_vlnk_buf, 0, sizeof(g_vlnk_buf));
+ ventoy_create_vlnk(sig, partoff, infile, (ventoy_vlnk *)g_vlnk_buf);
+ fwrite(g_vlnk_buf, 1, VLNK_FILE_LEN, fp);
+ fclose(fp);
+
+ return 0;
+}
+
+static int get_mount_point(char *partname, char *mntpoint)
+{
+ int i;
+ int len;
+ int rc = 1;
+ FILE *fp = NULL;
+ char line[PATH_MAX];
+
+ fp = fopen("/proc/mounts", "r");
+ if (!fp)
+ {
+ return 1;
+ }
+
+ len = (int)strlen(partname);
+ while (fgets(line, sizeof(line), fp))
+ {
+ if (strncmp(line, partname, len) == 0)
+ {
+ for (i = len; i < PATH_MAX && line[i]; i++)
+ {
+ if (line[i] == ' ')
+ {
+ line[i] = 0;
+ rc = 0;
+ strncpy(mntpoint, line + len, PATH_MAX - 1);
+ break;
+ }
+ }
+ break;
+ }
+ }
+
+ fclose(fp);
+ return rc;
+}
+
+static int parse_vlnk(char *infile)
+{
+ int i;
+ int fd;
+ int cnt;
+ int pflag = 0;
+ char diskname[128] = {0};
+ char partname[128] = {0};
+ char partpath[256] = {0};
+ char mntpoint[PATH_MAX];
+ ventoy_vlnk vlnk;
+
+ debug("parse vlnk\n");
+
+ if (infile[0] == 0)
+ {
+ debug("input file null\n");
+ return 1;
+ }
+
+ fd = open(infile, O_RDONLY);
+ if (fd < 0)
+ {
+ printf("Failed to open file %s error %d\n", infile, errno);
+ return 1;
+ }
+
+ memset(&vlnk, 0, sizeof(vlnk));
+ read(fd, &vlnk, sizeof(vlnk));
+ close(fd);
+
+ debug("disk_signature:%08X\n", vlnk.disk_signature);
+ debug("file path:<%s>\n", vlnk.filepath);
+ debug("part offset: %llu\n", (unsigned long long)vlnk.part_offset);
+
+ cnt = find_disk_by_sig((uint8_t *)&(vlnk.disk_signature), diskname);
+ if (cnt != 1)
+ {
+ printf("Disk in vlnk not found!\n");
+ return 1;
+ }
+
+ debug("Disk is <%s>\n", diskname);
+
+ if (strstr(diskname, "nvme") || strstr(diskname, "mmc") || strstr(diskname, "nbd"))
+ {
+ pflag = 1;
+ }
+
+ for (i = 1; i <= 128; i++)
+ {
+ if (pflag)
+ {
+ snprintf(partname, sizeof(partname) - 1, "%sp%d", diskname, i);
+ }
+ else
+ {
+ snprintf(partname, sizeof(partname) - 1, "%s%d", diskname, i);
+ }
+
+ if (get_part_offset(partname) == vlnk.part_offset)
+ {
+ debug("Find correct partition </dev/%s>\n", partname);
+ break;
+ }
+ }
+
+ if (i > 128)
+ {
+ printf("Partition in vlnk not found!");
+ return 1;
+ }
+
+ snprintf(partpath, sizeof(partpath), "/dev/%s ", partname);
+ if (get_mount_point(partpath, mntpoint))
+ {
+ printf("Mountpoint of %s is not found!\n", partpath);
+ return 1;
+ }
+ debug("moutpoint of %s is <%s>\n", partpath, mntpoint);
+
+ strcat(mntpoint, vlnk.filepath);
+ printf("Vlnk Point: %s\n", mntpoint);
+ if (access(mntpoint, F_OK) >= 0)
+ {
+ printf("File Exist: YES\n");
+ }
+ else
+ {
+ printf("File Exist: NO\n");
+ }
+
+ return 0;
+}
+
+static int check_vlnk(char *infile)
+{
+ int fd;
+ int64_t size;
+ ventoy_vlnk vlnk;
+
+ debug("check vlnk\n");
+
+ if (infile[0] == 0)
+ {
+ debug("input file null\n");
+ return 1;
+ }
+
+ size = get_file_size(infile);
+ if (size != VLNK_FILE_LEN)
+ {
+ debug("file size %lld is not a vlnk file size\n", (long long)size);
+ return 1;
+ }
+
+ fd = open(infile, O_RDONLY);
+ if (fd < 0)
+ {
+ debug("Failed to open file %s error %d\n", infile, errno);
+ return 1;
+ }
+
+ memset(&vlnk, 0, sizeof(vlnk));
+ read(fd, &vlnk, sizeof(vlnk));
+ close(fd);
+
+ if (CheckVlnkData(&vlnk))
+ {
+ return 0;
+ }
+
+ return 1;
+}
+
+int main(int argc, char **argv)
+{
+ int ch = 0;
+ int cmd = 0;
+ uint64_t partoff = 0;
+ char infile[PATH_MAX] = {0};
+ char outfile[PATH_MAX] = {0};
+ char diskname[256] = {0};
+
+ while ((ch = getopt(argc, argv, "c:t:l:d:p:o:v::")) != -1)
+ {
+ if (ch == 'c')
+ {
+ cmd = 1;
+ strncpy(infile, optarg, sizeof(infile) - 1);
+ }
+ else if (ch == 'o')
+ {
+ strncpy(outfile, optarg, sizeof(outfile) - 1);
+ }
+ else if (ch == 'l')
+ {
+ cmd = 2;
+ strncpy(infile, optarg, sizeof(infile) - 1);
+ }
+ else if (ch == 't')
+ {
+ cmd = 3;
+ strncpy(infile, optarg, sizeof(infile) - 1);
+ }
+ else if (ch == 'd')
+ {
+ strncpy(diskname, optarg, sizeof(diskname) - 1);
+ }
+ else if (ch == 'p')
+ {
+ partoff = (uint64_t)strtoull(optarg, NULL, 10);
+ partoff *= 512;
+ }
+ else if (ch == 'v')
+ {
+ verbose = 1;
+ }
+ else
+ {
+ return 1;
+ }
+ }
+
+ if (cmd == 1)
+ {
+ return create_vlnk(infile, diskname, partoff, outfile);
+ }
+ else if (cmd == 2)
+ {
+ return parse_vlnk(infile);
+ }
+ else if (cmd == 3)
+ {
+ return check_vlnk(infile);
+ }
+ else
+ {
+ printf("Invalid command %d\n", cmd);
+ return 1;
+ }
+}
+
--- /dev/null
+#include <Windows.h>\r
+#include <stdio.h>\r
+#include <stdlib.h>\r
+#include <string.h>\r
+#include <stdint.h>\r
+#include <time.h>\r
+#include <resource.h>\r
+#include <vlnk.h>\r
+\r
+static WCHAR g_CurDirW[MAX_PATH];\r
+static CHAR g_CurDirA[MAX_PATH];\r
+static CHAR g_LogFile[MAX_PATH];\r
+static HWND g_create_button;\r
+static HWND g_parse_button;\r
+\r
+typedef enum MSGID\r
+{\r
+ MSGID_ERROR = 0,\r
+ MSGID_INFO,\r
+ MSGID_BTN_CREATE,\r
+ MSGID_BTN_PARSE,\r
+ MSGID_SRC_UNSUPPORTED,\r
+ MSGID_FS_UNSUPPORTED,\r
+ MSGID_SUFFIX_UNSUPPORTED,\r
+ MSGID_DISK_INFO_ERR,\r
+ MSGID_VLNK_SUCCESS,\r
+ MSGID_RUNNING_TIP,\r
+ MSGID_CREATE_FILE_ERR,\r
+ MSGID_ALREADY_VLNK,\r
+ MSGID_INVALID_VLNK,\r
+ MSGID_VLNK_POINT_TO,\r
+ MSGID_VLNK_NO_DST,\r
+ MSGID_FILE_NAME_TOO_LONG,\r
+\r
+ MSGID_BUTT\r
+}MSGID;\r
+\r
+\r
+const WCHAR *g_msg_cn[MSGID_BUTT] =\r
+{\r
+ L"´íÎó",\r
+ L"ÌáÐÑ",\r
+ L"´´½¨",\r
+ L"½âÎö", \r
+ L"²»Ö§³ÖΪ´ËÎļþ´´½¨vlnk",\r
+ L"²»Ö§³ÖµÄÎļþϵͳ",\r
+ L"²»Ö§³ÖµÄÎļþºó׺Ãû",\r
+ L"»ñÈ¡´ÅÅÌÐÅϢʱ·¢Éú´íÎó",\r
+ L"Vlnk Îļþ´´½¨³É¹¦¡£",\r
+ L"ÇëÏȹرÕÕýÔÚÔËÐÐµÄ VentoyVlnk ³ÌÐò£¡",\r
+ L"´´½¨Îļþʧ°Ü",\r
+ L"´ËÎļþÒѾÊÇÒ»¸övlnkÎļþÁË£¡",\r
+ L"·Ç·¨µÄvlnkÎļþ!",\r
+ L"´Ë vlnk ÎļþÖ¸Ïò ",\r
+ L"´Ë vlnk Ö¸ÏòµÄÎļþ²»´æÔÚ£¡",\r
+ L"Îļþ·¾¶Ì«³¤£¡",\r
+};\r
+const WCHAR *g_msg_en[MSGID_BUTT] =\r
+{\r
+ L"Error",\r
+ L"Info", \r
+ L"Create",\r
+ L"Parse",\r
+ L"This file is not supported for vlnk",\r
+ L"Unsupported file system!", \r
+ L"Unsupported file suffix!",\r
+ L"Error when getting disk info",\r
+ L"Vlnk file successfully created!",\r
+ L"Please close another running VentoyVlnk instance!",\r
+ L"Failed to create file!",\r
+ L"This file is already a vlnk file!",\r
+ L"Invalid vlnk file!",\r
+ L"The vlnk file point to ",\r
+ L"The file pointed by the vlnk does NOT exist!",\r
+ L"The file full path is too long!",\r
+};\r
+\r
+const WCHAR **g_msg_lang = NULL;\r
+\r
+HINSTANCE g_hInst;\r
+
+static void Log2File(const char *log)
+{
+ time_t stamp;
+ struct tm ttm;
+ FILE *fp;
+
+ time(&stamp);
+ localtime_s(&ttm, &stamp);
+
+ fopen_s(&fp, g_LogFile, "a+");
+ if (fp)
+ {
+ fprintf_s(fp, "[%04u/%02u/%02u %02u:%02u:%02u] %s",
+ ttm.tm_year + 1900, ttm.tm_mon + 1, ttm.tm_mday,
+ ttm.tm_hour, ttm.tm_min, ttm.tm_sec, log);
+ fclose(fp);
+ }
+}
+
+void LogW(const WCHAR *Fmt, ...)
+{
+ WCHAR log[512];
+ CHAR alog[2048];
+ va_list arg;
+
+ if (g_LogFile[0] == 0)
+ {
+ return;
+ }
+
+ va_start(arg, Fmt);
+ vswprintf_s(log, 512, Fmt, arg);
+ va_end(arg);
+
+ WideCharToMultiByte(CP_UTF8, 0, log, -1, alog, 2048, 0, 0);
+
+ Log2File(alog);
+}\r
+\r
+
+void LogA(const CHAR *Fmt, ...)
+{
+ CHAR log[512];
+ va_list arg;
+
+ if (g_LogFile[0] == 0)
+ {
+ return;
+ }
+
+ va_start(arg, Fmt);
+ vsprintf_s(log, 512, Fmt, arg);
+ va_end(arg);
+
+ Log2File(log);
+}\r
+\r
+static int Utf8ToUtf16(const char* src, WCHAR * dst)\r
+{\r
+ int size = MultiByteToWideChar(CP_UTF8, 0, src, -1, dst, 0);\r
+ return MultiByteToWideChar(CP_UTF8, 0, src, -1, dst, size + 1);\r
+}\r
+\r
+static BOOL OnDestroyDialog()\r
+{ \r
+ return TRUE;\r
+}\r
+\r
+\r
+static BOOL InitDialog(HWND hWnd, WPARAM wParam, LPARAM lParam)\r
+{\r
+ HICON hIcon;\r
+\r
+ g_create_button = GetDlgItem(hWnd, IDC_BUTTON1);\r
+ g_parse_button = GetDlgItem(hWnd, IDC_BUTTON2);\r
+ \r
+ hIcon = LoadIcon(g_hInst, MAKEINTRESOURCE(IDI_ICON1));\r
+ SendMessage(hWnd, WM_SETICON, ICON_BIG, (LPARAM)hIcon);\r
+ SendMessage(hWnd, WM_SETICON, ICON_SMALL, (LPARAM)hIcon);\r
+\r
+ SetWindowTextW(g_create_button, g_msg_lang[MSGID_BTN_CREATE]);\r
+ SetWindowTextW(g_parse_button, g_msg_lang[MSGID_BTN_PARSE]);\r
+\r
+ return TRUE;\r
+}\r
+\r
+static int GetPhyDiskInfo(const char LogicalDrive, UINT32 *DiskSig, DISK_EXTENT *DiskExtent)\r
+{\r
+ BOOL Ret;\r
+ DWORD dwSize;\r
+ HANDLE Handle;\r
+ VOLUME_DISK_EXTENTS DiskExtents;\r
+ CHAR PhyPath[128];\r
+ UINT8 SectorBuf[512];\r
+\r
+ LogA("GetPhyDiskInfo %C\n", LogicalDrive);\r
+\r
+ sprintf_s(PhyPath, sizeof(PhyPath), "\\\\.\\%C:", LogicalDrive);\r
+ Handle = CreateFileA(PhyPath, GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE, 0, OPEN_EXISTING, 0, 0);\r
+ if (Handle == INVALID_HANDLE_VALUE)\r
+ {\r
+ LogA("Could not open the disk %C: error:%u\n", LogicalDrive, GetLastError());\r
+ return 1;\r
+ }\r
+\r
+ Ret = DeviceIoControl(Handle,\r
+ IOCTL_VOLUME_GET_VOLUME_DISK_EXTENTS,\r
+ NULL,\r
+ 0,\r
+ &DiskExtents,\r
+ (DWORD)(sizeof(DiskExtents)),\r
+ (LPDWORD)&dwSize,\r
+ NULL);\r
+ if (!Ret || DiskExtents.NumberOfDiskExtents == 0)\r
+ {\r
+ LogA("DeviceIoControl IOCTL_VOLUME_GET_VOLUME_DISK_EXTENTS failed, error:%u\n", GetLastError());\r
+ CloseHandle(Handle);\r
+ return 1;\r
+ }\r
+ CloseHandle(Handle);\r
+\r
+ memcpy(DiskExtent, DiskExtents.Extents, sizeof(DISK_EXTENT));\r
+ LogA("%C: is in PhysicalDrive%d Offset:%llu\n", LogicalDrive, DiskExtents.Extents[0].DiskNumber,\r
+ (ULONGLONG)(DiskExtents.Extents[0].StartingOffset.QuadPart));\r
+\r
+ sprintf_s(PhyPath, sizeof(PhyPath), "\\\\.\\PhysicalDrive%d", DiskExtents.Extents[0].DiskNumber);\r
+ Handle = CreateFileA(PhyPath, GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE, 0, OPEN_EXISTING, 0, 0);\r
+ if (Handle == INVALID_HANDLE_VALUE)\r
+ {\r
+ LogA("Could not open the disk<PhysicalDrive%d>, error:%u\n", DiskExtents.Extents[0].DiskNumber, GetLastError());\r
+ return 1;\r
+ }\r
+\r
+ if (!ReadFile(Handle, SectorBuf, sizeof(SectorBuf), &dwSize, NULL))\r
+ {\r
+ LogA("ReadFile failed, dwSize:%u error:%u\n", dwSize, GetLastError());\r
+ CloseHandle(Handle);\r
+ return 1;\r
+ }\r
+\r
+ memcpy(DiskSig, SectorBuf + 0x1B8, 4);\r
+\r
+ CloseHandle(Handle);\r
+ return 0;\r
+}\r
+\r
+\r
+static int SaveBuffer2File(const WCHAR *Fullpath, void *Buffer, DWORD Length)\r
+{\r
+ int rc = 1;\r
+ DWORD dwSize;\r
+ HANDLE Handle;\r
+\r
+ LogW(L"SaveBuffer2File <%ls> len:%u\n", Fullpath, Length);\r
+\r
+ Handle = CreateFileW(Fullpath, GENERIC_READ | GENERIC_WRITE,\r
+ FILE_SHARE_READ | FILE_SHARE_WRITE, 0, CREATE_ALWAYS, 0, 0);\r
+ if (Handle == INVALID_HANDLE_VALUE)\r
+ {\r
+ LogA("Could not create new file, error:%u\n", GetLastError());\r
+ goto End;\r
+ }\r
+\r
+ WriteFile(Handle, Buffer, Length, &dwSize, NULL);\r
+\r
+ rc = 0;\r
+\r
+End:\r
+\r
+ if (Handle != INVALID_HANDLE_VALUE)\r
+ {\r
+ CloseHandle(Handle);\r
+ }\r
+\r
+\r
+ return rc;\r
+}\r
+\r
+static int DefaultVlnkDstFullPath(WCHAR *Src, WCHAR *Dir, WCHAR *Dst)\r
+{\r
+ int i, j;\r
+ int len;\r
+ int wrlen;\r
+ WCHAR C;\r
+\r
+ len = (int)lstrlen(Src);\r
+ for (i = len - 1; i >= 0; i--)\r
+ {\r
+ if (Src[i] == '.')\r
+ {\r
+ C = Src[i];\r
+ Src[i] = 0;\r
+ wrlen = swprintf_s(Dst, MAX_PATH, L"%ls\\%ls.vlnk.%ls", Dir, Src, Src + i + 1);\r
+ Src[i] = C;\r
+\r
+ for (j = wrlen - (len - i); j < wrlen; j++)\r
+ {\r
+ if (Dst[j] >= 'A' && Dst[j] <= 'Z')\r
+ {\r
+ Dst[j] = 'a' + (Dst[j] - 'A');\r
+ }\r
+ }\r
+\r
+ break;\r
+ }\r
+ }\r
+\r
+ return 0;\r
+}\r
+\r
+static BOOL IsVlnkFile(WCHAR *path, ventoy_vlnk *outvlnk)\r
+{\r
+ BOOL bRet;\r
+ BOOL bVlnk = FALSE;\r
+ DWORD dwSize;\r
+ LARGE_INTEGER FileSize;\r
+ HANDLE Handle;\r
+ ventoy_vlnk vlnk;\r
+\r
+ Handle = CreateFileW(path, GENERIC_READ, FILE_SHARE_READ, 0, OPEN_EXISTING, 0, 0);\r
+ if (Handle == INVALID_HANDLE_VALUE)\r
+ {\r
+ LogA("Could not open this file, error:%u\n", GetLastError());\r
+ return FALSE;\r
+ }\r
+\r
+ if (!GetFileSizeEx(Handle, &FileSize))\r
+ {\r
+ LogA("Failed to get vlnk file size\n");\r
+ goto End;\r
+ }\r
+\r
+ if (FileSize.QuadPart != VLNK_FILE_LEN)\r
+ {\r
+ LogA("Invalid vlnk file length %llu\n", (unsigned long long)FileSize.QuadPart);\r
+ goto End;\r
+ }\r
+\r
+ memset(&vlnk, 0, sizeof(vlnk));\r
+ bRet = ReadFile(Handle, &vlnk, sizeof(vlnk), &dwSize, NULL);\r
+ if (bRet && CheckVlnkData(&vlnk))\r
+ {\r
+ if (outvlnk)\r
+ {\r
+ memcpy(outvlnk, &vlnk, sizeof(vlnk));\r
+ }\r
+\r
+ bVlnk = TRUE;\r
+ }\r
+\r
+End:\r
+\r
+ if (Handle != INVALID_HANDLE_VALUE)\r
+ {\r
+ CloseHandle(Handle);\r
+ }\r
+\r
+ return bVlnk;\r
+}\r
+\r
+\r
+static int CreateVlnk(HWND hWnd, WCHAR *Dir)\r
+{\r
+ int i;\r
+ int end;\r
+ int len;\r
+ UINT32 DiskSig;\r
+ DISK_EXTENT DiskExtend;\r
+ OPENFILENAME ofn = { 0 };\r
+ CHAR UTF8Path[MAX_PATH];\r
+ WCHAR DstFullPath[MAX_PATH];\r
+ WCHAR szFile[MAX_PATH] = { 0 };\r
+ CHAR suffix[8] = { 0 };\r
+ CHAR Drive[8] = { 0 };\r
+ CHAR FsName[64] = { 0 };\r
+ CHAR *Buf = NULL;\r
+ WCHAR *Pos = NULL;\r
+ ventoy_vlnk *vlnk = NULL;\r
+\r
+ ofn.lStructSize = sizeof(ofn);\r
+ ofn.hwndOwner = hWnd;\r
+ ofn.lpstrFile = szFile;\r
+ ofn.nMaxFile = sizeof(szFile);\r
+ ofn.lpstrFilter = L"Vlnk Source File\0*.iso;*.img;*.wim;*.vhd;*.vhdx;*.vtoy;*.efi;*.dat\0";\r
+ ofn.nFilterIndex = 1;\r
+ ofn.lpstrFileTitle = NULL;\r
+ ofn.nMaxFileTitle = 0;\r
+ ofn.lpstrInitialDir = NULL;\r
+ ofn.Flags = OFN_PATHMUSTEXIST | OFN_FILEMUSTEXIST;\r
+\r
+ if (GetOpenFileName(&ofn) != TRUE)\r
+ {\r
+ return 1;\r
+ }\r
+\r
+ LogW(L"Create vlnk for <%ls>\n", szFile);\r
+ \r
+ len = lstrlen(szFile);\r
+\r
+ if (len < 5 || szFile[0] == '.' || szFile[1] != ':')\r
+ {\r
+ MessageBox(hWnd, g_msg_lang[MSGID_SRC_UNSUPPORTED], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+ return 1;\r
+ }\r
+ \r
+ Drive[0] = (CHAR)szFile[0];\r
+ Drive[1] = ':';\r
+ Drive[2] = '\\';\r
+ if (0 == GetVolumeInformationA(Drive, NULL, 0, NULL, NULL, NULL, FsName, sizeof(FsName) - 1))\r
+ {\r
+ LogA("GetVolumeInformationA failed %u\n", GetLastError());\r
+ return 1;\r
+ }\r
+\r
+ LogA("Partition filesystem of <%s> is <%s>\n", Drive, FsName);\r
+ if (_stricmp(FsName, "NTFS") == 0 ||\r
+ _stricmp(FsName, "exFAT") == 0 ||\r
+ _stricmp(FsName, "FAT") == 0 ||\r
+ _stricmp(FsName, "FAT32") == 0 ||\r
+ _stricmp(FsName, "FAT16") == 0 ||\r
+ _stricmp(FsName, "FAT12") == 0 ||\r
+ _stricmp(FsName, "UDF") == 0)\r
+ {\r
+ LogA("FS Check OK\n");\r
+ }\r
+ else\r
+ {\r
+ MessageBox(hWnd, g_msg_lang[MSGID_FS_UNSUPPORTED], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+ return 1;\r
+ }\r
+\r
+\r
+ end = (szFile[len - 5] == '.') ? 5 : 4;\r
+ for (i = 0; i < end; i++)\r
+ {\r
+ suffix[i] = (CHAR)szFile[len - (end - i)];\r
+ }\r
+\r
+ if (!IsSupportedImgSuffix(suffix))\r
+ {\r
+ MessageBox(hWnd, g_msg_lang[MSGID_SUFFIX_UNSUPPORTED], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+ return 1;\r
+ }\r
+\r
+ if (IsVlnkFile(szFile, NULL))\r
+ {\r
+ MessageBox(hWnd, g_msg_lang[MSGID_ALREADY_VLNK], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+ return 1;\r
+ }\r
+\r
+ for (i = 0; i < MAX_PATH && szFile[i]; i++)\r
+ {\r
+ if (szFile[i] == '\\' || szFile[i] == '/')\r
+ {\r
+ Pos = szFile + i;\r
+ }\r
+ }\r
+\r
+ if (!Pos)\r
+ {\r
+ LogA("name part not found\n");\r
+ return 1;\r
+ }\r
+ LogW(L"File Name is <%ls>\n", Pos + 1);\r
+\r
+ memset(UTF8Path, 0, sizeof(UTF8Path));\r
+ WideCharToMultiByte(CP_UTF8, 0, szFile + 2, -1, UTF8Path, MAX_PATH, NULL, 0);\r
+\r
+ for (i = 0; i < MAX_PATH && UTF8Path[i]; i++)\r
+ {\r
+ if (UTF8Path[i] == '\\')\r
+ {\r
+ UTF8Path[i] = '/';\r
+ }\r
+ }\r
+\r
+ len = (int)strlen(UTF8Path);\r
+ if (len >= VLNK_NAME_MAX)\r
+ {\r
+ LogA("File name length %d overflow\n", len);\r
+ MessageBox(hWnd, g_msg_lang[MSGID_FILE_NAME_TOO_LONG], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+ return 1;\r
+ }\r
+\r
+ DiskExtend.StartingOffset.QuadPart = 0;\r
+ if (GetPhyDiskInfo((char)szFile[0], &DiskSig, &DiskExtend))\r
+ {\r
+ MessageBox(hWnd, g_msg_lang[MSGID_DISK_INFO_ERR], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+ return 1;\r
+ }\r
+\r
+ Buf = malloc(VLNK_FILE_LEN);\r
+ if (Buf)\r
+ {\r
+ memset(Buf, 0, VLNK_FILE_LEN);\r
+ vlnk = (ventoy_vlnk *)Buf;\r
+ ventoy_create_vlnk(DiskSig, (uint64_t)DiskExtend.StartingOffset.QuadPart, UTF8Path, vlnk);\r
+\r
+ DefaultVlnkDstFullPath(Pos + 1, Dir, DstFullPath);\r
+ LogW(L"vlnk output file path is <%ls>\n", DstFullPath);\r
+\r
+ if (SaveBuffer2File(DstFullPath, Buf, VLNK_FILE_LEN) == 0)\r
+ {\r
+ WCHAR Msg[1024];\r
+\r
+ swprintf_s(Msg, 1024, L"%ls\r\n\r\n%ls", g_msg_lang[MSGID_VLNK_SUCCESS], DstFullPath + lstrlen(Dir) + 1);\r
+\r
+ LogW(L"Vlnk file create success <%ls>\n", DstFullPath);\r
+ MessageBox(hWnd, Msg, g_msg_lang[MSGID_INFO], MB_OK | MB_ICONINFORMATION);\r
+ }\r
+ else\r
+ {\r
+ LogA("Vlnk file save failed\n");\r
+ MessageBox(hWnd, g_msg_lang[MSGID_CREATE_FILE_ERR], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+ }\r
+\r
+ free(Buf);\r
+ }\r
+\r
+ return 0;\r
+}\r
+\r
+static CHAR GetDriveLetter(UINT32 disksig, UINT64 PartOffset)\r
+{\r
+ CHAR Letter;\r
+ DWORD Drives;\r
+ UINT32 Sig;\r
+ DISK_EXTENT DiskExtent;\r
+\r
+ Letter = 'A';\r
+ Drives = GetLogicalDrives();\r
+ LogA("Logic Drives: 0x%x", Drives);\r
+\r
+ while (Drives)\r
+ {\r
+ if (Drives & 0x01)\r
+ {\r
+ Sig = 0;\r
+ DiskExtent.StartingOffset.QuadPart = 0;\r
+ if (GetPhyDiskInfo(Letter, &Sig, &DiskExtent) == 0)\r
+ {\r
+ if (Sig == disksig && DiskExtent.StartingOffset.QuadPart == PartOffset)\r
+ {\r
+ return Letter;\r
+ }\r
+ }\r
+ }\r
+\r
+ Drives >>= 1;\r
+ Letter++;\r
+ }\r
+\r
+ return 0;\r
+}\r
+\r
+\r
+static int ParseVlnk(HWND hWnd)\r
+{\r
+ int i;\r
+ CHAR Letter;\r
+ ventoy_vlnk vlnk;\r
+ OPENFILENAME ofn = { 0 };\r
+ WCHAR szFile[MAX_PATH] = { 0 };\r
+ WCHAR szDst[MAX_PATH + 2] = { 0 };\r
+ WCHAR Msg[1024];\r
+ CHAR *suffix = NULL;\r
+ HANDLE hFile;\r
+\r
+ ofn.lStructSize = sizeof(ofn);\r
+ ofn.hwndOwner = hWnd;\r
+ ofn.lpstrFile = szFile;\r
+ ofn.nMaxFile = sizeof(szFile);\r
+ ofn.lpstrFilter = L"Vlnk File\0*.vlnk.iso;*.vlnk.img;*.vlnk.wim;*.vlnk.efi;*.vlnk.vhd;*.vlnk.vhdx;*.vlnk.vtoy;*.vlnk.dat\0";\r
+ ofn.nFilterIndex = 1;\r
+ ofn.lpstrFileTitle = NULL;\r
+ ofn.nMaxFileTitle = 0;\r
+ ofn.lpstrInitialDir = NULL;\r
+ ofn.Flags = OFN_PATHMUSTEXIST | OFN_FILEMUSTEXIST;\r
+\r
+ if (GetOpenFileName(&ofn) != TRUE)\r
+ {\r
+ return 1;\r
+ }\r
+\r
+ LogW(L"Parse vlnk for <%ls>\n", szFile);\r
+\r
+ if (!IsVlnkFile(szFile, &vlnk))\r
+ {\r
+ MessageBox(hWnd, g_msg_lang[MSGID_INVALID_VLNK], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+ return 1;\r
+ }\r
+\r
+ for (i = 0; i < sizeof(vlnk.filepath) && vlnk.filepath[i]; i++)\r
+ {\r
+ if (vlnk.filepath[i] == '.')\r
+ {\r
+ suffix = vlnk.filepath + i;\r
+ }\r
+ }\r
+\r
+ if (!IsSupportedImgSuffix(suffix))\r
+ {\r
+ MessageBox(hWnd, g_msg_lang[MSGID_SUFFIX_UNSUPPORTED], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+ return 1;\r
+ }\r
+\r
+ Utf8ToUtf16(vlnk.filepath, szDst + 2);\r
+ for (i = 2; i < MAX_PATH && szDst[i]; i++)\r
+ {\r
+ if (szDst[i] == '/')\r
+ {\r
+ szDst[i] = '\\';\r
+ }\r
+ }\r
+ \r
+\r
+ Letter = GetDriveLetter(vlnk.disk_signature, vlnk.part_offset);\r
+ if (Letter == 0)\r
+ {\r
+ MessageBox(hWnd, g_msg_lang[MSGID_VLNK_NO_DST], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+ return 1;\r
+ }\r
+\r
+ szDst[0] = toupper(Letter);\r
+ szDst[1] = ':';\r
+ LogW(L"vlnk dst is %ls\n", szDst);\r
+\r
+ hFile = CreateFileW(szDst, FILE_READ_EA, FILE_SHARE_READ, 0, OPEN_EXISTING, 0, 0);\r
+ if (INVALID_HANDLE_VALUE == hFile)\r
+ {\r
+ MessageBox(hWnd, g_msg_lang[MSGID_VLNK_NO_DST], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+ return 1;\r
+ }\r
+ CloseHandle(hFile);\r
+\r
+ swprintf_s(Msg, 1024, L"%ls %ls", g_msg_lang[MSGID_VLNK_POINT_TO], szDst);\r
+ MessageBox(hWnd, Msg, g_msg_lang[MSGID_INFO], MB_OK | MB_ICONINFORMATION);\r
+\r
+ return 0;\r
+}\r
+\r
+INT_PTR CALLBACK DialogProc(HWND hWnd, UINT Message, WPARAM wParam, LPARAM lParam)\r
+{\r
+ WORD NotifyCode;\r
+ WORD CtrlID;\r
+\r
+ switch (Message)\r
+ {\r
+ case WM_COMMAND:\r
+ {\r
+ NotifyCode = HIWORD(wParam);\r
+ CtrlID = LOWORD(wParam);\r
+\r
+ if (NotifyCode == BN_CLICKED)\r
+ {\r
+ if (CtrlID == IDC_BUTTON1)\r
+ {\r
+ EnableWindow(g_create_button, FALSE);\r
+ CreateVlnk(hWnd, g_CurDirW);\r
+ EnableWindow(g_create_button, TRUE);\r
+ }\r
+ else if (CtrlID == IDC_BUTTON2)\r
+ {\r
+ EnableWindow(g_parse_button, FALSE);\r
+ ParseVlnk(hWnd);\r
+ EnableWindow(g_parse_button, TRUE);\r
+ }\r
+ }\r
+ break;\r
+ }\r
+ case WM_INITDIALOG:\r
+ {\r
+ InitDialog(hWnd, wParam, lParam);\r
+ break;\r
+ } \r
+ case WM_CLOSE:\r
+ {\r
+ OnDestroyDialog();\r
+ EndDialog(hWnd, 0);\r
+ }\r
+ }\r
+\r
+ return 0;\r
+}\r
+\r
+int APIENTRY WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, INT nCmdShow)\r
+{\r
+ int i;\r
+ HANDLE hMutex;\r
+\r
+ UNREFERENCED_PARAMETER(hPrevInstance);\r
+\r
+ if (GetUserDefaultUILanguage() == 0x0804)\r
+ {\r
+ g_msg_lang = g_msg_cn;\r
+ }\r
+ else\r
+ {\r
+ g_msg_lang = g_msg_en;\r
+ }\r
+\r
+ hMutex = CreateMutexA(NULL, TRUE, "VtoyVlnkMUTEX");\r
+ if ((hMutex != NULL) && (GetLastError() == ERROR_ALREADY_EXISTS))\r
+ {\r
+ MessageBoxW(NULL, g_msg_lang[MSGID_RUNNING_TIP], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);\r
+ return 1;\r
+ }\r
+\r
+ GetCurrentDirectoryA(MAX_PATH, g_CurDirA);\r
+ GetCurrentDirectoryW(MAX_PATH, g_CurDirW);\r
+ sprintf_s(g_LogFile, sizeof(g_LogFile), "%s\\VentoyVlnk.log", g_CurDirA);\r
+\r
+ for (i = 0; i < __argc; i++)\r
+ {\r
+ if (strncmp(__argv[i], "-Q", 2) == 0 ||\r
+ strncmp(__argv[i], "-q", 2) == 0)\r
+ {\r
+ g_LogFile[0] = 0;\r
+ break;\r
+ }\r
+ }\r
+ \r
+\r
+ LogA("========= VentoyVlnk =========\n");\r
+\r
+ g_hInst = hInstance;\r
+ DialogBoxA(hInstance, MAKEINTRESOURCEA(IDD_DIALOG1), NULL, DialogProc);\r
+\r
+ return 0;\r
+}\r
--- /dev/null
+#include <stdio.h>
+#include <stdlib.h>
+#include <stdint.h>
+#include <string.h>
+#include "vlnk.h"
+
+int ventoy_create_vlnk(uint32_t disksig, uint64_t partoffset, const char *path, ventoy_vlnk *vlnk)
+{
+ uint32_t crc;
+ ventoy_guid guid = VENTOY_GUID;
+
+ memcpy(&(vlnk->guid), &guid, sizeof(ventoy_guid));
+ vlnk->disk_signature = disksig;
+ vlnk->part_offset = partoffset;
+
+#ifdef WIN32
+ strcpy_s(vlnk->filepath, sizeof(vlnk->filepath) - 1, path);
+#else
+ strncpy(vlnk->filepath, path, sizeof(vlnk->filepath) - 1);
+#endif
+
+ crc = ventoy_getcrc32c(0, vlnk, sizeof(ventoy_vlnk));
+ vlnk->crc32 = crc;
+
+ return 0;
+}
+
+\r
+int CheckVlnkData(ventoy_vlnk *vlnk)\r
+{\r
+ uint32_t readcrc, calccrc;\r
+ ventoy_guid guid = VENTOY_GUID;\r
+\r
+ if (memcmp(&vlnk->guid, &guid, sizeof(guid)))\r
+ {\r
+ return 0;\r
+ }\r
+\r
+ readcrc = vlnk->crc32;\r
+ vlnk->crc32 = 0;\r
+ calccrc = ventoy_getcrc32c(0, vlnk, sizeof(ventoy_vlnk));\r
+\r
+ if (readcrc != calccrc)\r
+ {\r
+ return 0;\r
+ }\r
+\r
+ return 1;\r
+}\r
+\r
+int IsSupportedImgSuffix(char *suffix)\r
+{\r
+ int i = 0;\r
+ const char *suffixs[] =\r
+ {\r
+ ".iso", ".img", ".wim", ".efi", ".vhd", ".vhdx", ".dat", ".vtoy", NULL\r
+ };\r
+\r
+ if (!suffix)\r
+ {\r
+ return 0;\r
+ }\r
+\r
+ while (suffixs[i])\r
+ {\r
+
+#ifdef WIN32\r
+ if (_stricmp(suffixs[i], suffix) == 0)\r
+#else\r
+ if (strcasecmp(suffixs[i], suffix) == 0)\r
+#endif\r
+ {\r
+ return 1;\r
+ }\r
+\r
+ i++;\r
+ }\r
+\r
+ return 0;\r
+}\r
--- /dev/null
+
+#ifndef __VLNK_H__
+#define __VLNK_H__
+
+#define VLNK_FILE_LEN 32768
+
+#define VLNK_NAME_MAX 384
+
+#define VENTOY_GUID { 0x77772020, 0x2e77, 0x6576, { 0x6e, 0x74, 0x6f, 0x79, 0x2e, 0x6e, 0x65, 0x74 }}
+
+#pragma pack(1)
+
+typedef struct ventoy_guid
+{
+ uint32_t data1;
+ uint16_t data2;
+ uint16_t data3;
+ uint8_t data4[8];
+}ventoy_guid;
+
+typedef struct ventoy_vlnk
+{
+ ventoy_guid guid; // VENTOY_GUID
+ uint32_t crc32; // crc32
+ uint32_t disk_signature;
+ uint64_t part_offset; // in bytes
+ char filepath[VLNK_NAME_MAX];
+ uint8_t reserverd[96];
+}ventoy_vlnk;
+#pragma pack()
+
+uint32_t ventoy_getcrc32c (uint32_t crc, const void *buf, int size);
+int ventoy_create_vlnk(uint32_t disksig, uint64_t partoffset, const char *path, ventoy_vlnk *vlnk);
+int CheckVlnkData(ventoy_vlnk *vlnk);
+int IsSupportedImgSuffix(char *suffix);
+
+#endif
+
--- /dev/null
+\r
+Microsoft Visual Studio Solution File, Format Version 12.00\r
+# Visual Studio 2013\r
+VisualStudioVersion = 12.0.21005.1\r
+MinimumVisualStudioVersion = 10.0.40219.1\r
+Project("{8BC9CEB8-8B4A-11D0-8D11-00A0C91BC942}") = "VentoyVlnk", "VentoyVlnk\VentoyVlnk.vcxproj", "{9987D9FE-1A40-4C5F-835C-D66B0FEADA26}"\r
+EndProject\r
+Global\r
+ GlobalSection(SolutionConfigurationPlatforms) = preSolution\r
+ Debug|Win32 = Debug|Win32\r
+ Release|Win32 = Release|Win32\r
+ EndGlobalSection\r
+ GlobalSection(ProjectConfigurationPlatforms) = postSolution\r
+ {9987D9FE-1A40-4C5F-835C-D66B0FEADA26}.Debug|Win32.ActiveCfg = Debug|Win32\r
+ {9987D9FE-1A40-4C5F-835C-D66B0FEADA26}.Debug|Win32.Build.0 = Debug|Win32\r
+ {9987D9FE-1A40-4C5F-835C-D66B0FEADA26}.Release|Win32.ActiveCfg = Release|Win32\r
+ {9987D9FE-1A40-4C5F-835C-D66B0FEADA26}.Release|Win32.Build.0 = Release|Win32\r
+ EndGlobalSection\r
+ GlobalSection(SolutionProperties) = preSolution\r
+ HideSolutionNode = FALSE\r
+ EndGlobalSection\r
+EndGlobal\r
--- /dev/null
+<?xml version="1.0" encoding="UTF-8" standalone="yes"?>\r
+<assembly xmlns="urn:schemas-microsoft-com:asm.v1" manifestVersion="1.0">\r
+ <dependency> \r
+ <dependentAssembly> \r
+ <assemblyIdentity \r
+ type="win32" \r
+ name="Microsoft.Windows.Common-Controls" \r
+ version="6.0.0.0" \r
+ processorArchitecture="x86" \r
+ publicKeyToken="6595b64144ccf1df" \r
+ language="*" \r
+ /> \r
+ </dependentAssembly> \r
+ </dependency> \r
+ \r
+ <compatibility xmlns="urn:schemas-microsoft-com:compatibility.v1">\r
+ <application>\r
+ <!-- Windows 10 -->\r
+ <supportedOS Id="{8e0f7a12-bfb3-4fe8-b9a5-48fd50a15a9a}"/>\r
+ <!-- Windows 8.1 -->\r
+ <supportedOS Id="{1f676c76-80e1-4239-95bb-83d0f6d0da78}"/>\r
+ <!-- Windows 8 -->\r
+ <supportedOS Id="{4a2f28e3-53b9-4441-ba9c-d69d4a4a6e38}"/>\r
+ <!-- Windows 7 -->\r
+ <supportedOS Id="{35138b9a-5d96-4fbd-8e2d-a2440225f93a}"/>\r
+ <!-- Windows Vista -->\r
+ <supportedOS Id="{e2011457-1546-43c5-a5fe-008deee3d3f0}"/> \r
+ </application>\r
+ </compatibility>\r
+ <trustInfo xmlns="urn:schemas-microsoft-com:asm.v3">\r
+ <security>\r
+ <requestedPrivileges>\r
+ <requestedExecutionLevel level="requireAdministrator" uiAccess="false"></requestedExecutionLevel>\r
+ </requestedPrivileges>\r
+ </security>\r
+ </trustInfo> \r
+</assembly>
\ No newline at end of file
--- /dev/null
+<?xml version="1.0" encoding="UTF-8" standalone="yes"?>\r
+<assembly xmlns="urn:schemas-microsoft-com:asm.v1" manifestVersion="1.0">\r
+ <dependency> \r
+ <dependentAssembly> \r
+ <assemblyIdentity \r
+ type="win32" \r
+ name="Microsoft.Windows.Common-Controls" \r
+ version="6.0.0.0" \r
+ processorArchitecture="amd64" \r
+ publicKeyToken="6595b64144ccf1df" \r
+ language="*" \r
+ /> \r
+ </dependentAssembly> \r
+ </dependency> \r
+ \r
+ <compatibility xmlns="urn:schemas-microsoft-com:compatibility.v1">\r
+ <application>\r
+ <!-- Windows 10 -->\r
+ <supportedOS Id="{8e0f7a12-bfb3-4fe8-b9a5-48fd50a15a9a}"/>\r
+ <!-- Windows 8.1 -->\r
+ <supportedOS Id="{1f676c76-80e1-4239-95bb-83d0f6d0da78}"/>\r
+ <!-- Windows 8 -->\r
+ <supportedOS Id="{4a2f28e3-53b9-4441-ba9c-d69d4a4a6e38}"/>\r
+ <!-- Windows 7 -->\r
+ <supportedOS Id="{35138b9a-5d96-4fbd-8e2d-a2440225f93a}"/>\r
+ <!-- Windows Vista -->\r
+ <supportedOS Id="{e2011457-1546-43c5-a5fe-008deee3d3f0}"/> \r
+ </application>\r
+ </compatibility>\r
+ \r
+</assembly>
\ No newline at end of file
--- /dev/null
+<?xml version="1.0" encoding="UTF-8" standalone="yes"?>\r
+<assembly xmlns="urn:schemas-microsoft-com:asm.v1" manifestVersion="1.0">\r
+ <dependency> \r
+ <dependentAssembly> \r
+ <assemblyIdentity \r
+ type="win32" \r
+ name="Microsoft.Windows.Common-Controls" \r
+ version="6.0.0.0" \r
+ processorArchitecture="arm" \r
+ publicKeyToken="6595b64144ccf1df" \r
+ language="*" \r
+ /> \r
+ </dependentAssembly> \r
+ </dependency> \r
+ \r
+ <compatibility xmlns="urn:schemas-microsoft-com:compatibility.v1">\r
+ <application>\r
+ <!-- Windows 10 -->\r
+ <supportedOS Id="{8e0f7a12-bfb3-4fe8-b9a5-48fd50a15a9a}"/>\r
+ <!-- Windows 8.1 -->\r
+ <supportedOS Id="{1f676c76-80e1-4239-95bb-83d0f6d0da78}"/>\r
+ <!-- Windows 8 -->\r
+ <supportedOS Id="{4a2f28e3-53b9-4441-ba9c-d69d4a4a6e38}"/>\r
+ <!-- Windows 7 -->\r
+ <supportedOS Id="{35138b9a-5d96-4fbd-8e2d-a2440225f93a}"/>\r
+ <!-- Windows Vista -->\r
+ <supportedOS Id="{e2011457-1546-43c5-a5fe-008deee3d3f0}"/> \r
+ </application>\r
+ </compatibility>\r
+ \r
+</assembly>
\ No newline at end of file
--- /dev/null
+<?xml version="1.0" encoding="UTF-8" standalone="yes"?>\r
+<assembly xmlns="urn:schemas-microsoft-com:asm.v1" manifestVersion="1.0">\r
+ <dependency> \r
+ <dependentAssembly> \r
+ <assemblyIdentity \r
+ type="win32" \r
+ name="Microsoft.Windows.Common-Controls" \r
+ version="6.0.0.0" \r
+ processorArchitecture="arm64" \r
+ publicKeyToken="6595b64144ccf1df" \r
+ language="*" \r
+ /> \r
+ </dependentAssembly> \r
+ </dependency> \r
+ \r
+ <compatibility xmlns="urn:schemas-microsoft-com:compatibility.v1">\r
+ <application>\r
+ <!-- Windows 10 -->\r
+ <supportedOS Id="{8e0f7a12-bfb3-4fe8-b9a5-48fd50a15a9a}"/>\r
+ <!-- Windows 8.1 -->\r
+ <supportedOS Id="{1f676c76-80e1-4239-95bb-83d0f6d0da78}"/>\r
+ <!-- Windows 8 -->\r
+ <supportedOS Id="{4a2f28e3-53b9-4441-ba9c-d69d4a4a6e38}"/>\r
+ <!-- Windows 7 -->\r
+ <supportedOS Id="{35138b9a-5d96-4fbd-8e2d-a2440225f93a}"/>\r
+ <!-- Windows Vista -->\r
+ <supportedOS Id="{e2011457-1546-43c5-a5fe-008deee3d3f0}"/> \r
+ </application>\r
+ </compatibility>\r
+ \r
+</assembly>
\ No newline at end of file
--- /dev/null
+<?xml version="1.0" encoding="utf-8"?>\r
+<Project DefaultTargets="Build" ToolsVersion="12.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">\r
+ <ItemGroup Label="ProjectConfigurations">\r
+ <ProjectConfiguration Include="Debug|Win32">\r
+ <Configuration>Debug</Configuration>\r
+ <Platform>Win32</Platform>\r
+ </ProjectConfiguration>\r
+ <ProjectConfiguration Include="Release|Win32">\r
+ <Configuration>Release</Configuration>\r
+ <Platform>Win32</Platform>\r
+ </ProjectConfiguration>\r
+ </ItemGroup>\r
+ <PropertyGroup Label="Globals">\r
+ <ProjectGuid>{9987D9FE-1A40-4C5F-835C-D66B0FEADA26}</ProjectGuid>\r
+ <Keyword>Win32Proj</Keyword>\r
+ <RootNamespace>VentoyVlnk</RootNamespace>\r
+ </PropertyGroup>\r
+ <Import Project="$(VCTargetsPath)\Microsoft.Cpp.Default.props" />\r
+ <PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'" Label="Configuration">\r
+ <ConfigurationType>Application</ConfigurationType>\r
+ <UseDebugLibraries>true</UseDebugLibraries>\r
+ <PlatformToolset>v120</PlatformToolset>\r
+ <CharacterSet>Unicode</CharacterSet>\r
+ </PropertyGroup>\r
+ <PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|Win32'" Label="Configuration">\r
+ <ConfigurationType>Application</ConfigurationType>\r
+ <UseDebugLibraries>false</UseDebugLibraries>\r
+ <PlatformToolset>v120</PlatformToolset>\r
+ <WholeProgramOptimization>true</WholeProgramOptimization>\r
+ <CharacterSet>Unicode</CharacterSet>\r
+ </PropertyGroup>\r
+ <Import Project="$(VCTargetsPath)\Microsoft.Cpp.props" />\r
+ <ImportGroup Label="ExtensionSettings">\r
+ </ImportGroup>\r
+ <ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">\r
+ <Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />\r
+ </ImportGroup>\r
+ <ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">\r
+ <Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />\r
+ </ImportGroup>\r
+ <PropertyGroup Label="UserMacros" />\r
+ <PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">\r
+ <LinkIncremental>true</LinkIncremental>\r
+ <IncludePath>$(ProjectDir);$(SolutionDir)/../../src;$(IncludePath)</IncludePath>\r
+ </PropertyGroup>\r
+ <PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">\r
+ <LinkIncremental>false</LinkIncremental>\r
+ <IncludePath>$(SolutionDir)/../../src;$(ProjectDir);$(IncludePath)</IncludePath>\r
+ </PropertyGroup>\r
+ <ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">\r
+ <ClCompile>\r
+ <PrecompiledHeader>\r
+ </PrecompiledHeader>\r
+ <WarningLevel>Level3</WarningLevel>\r
+ <Optimization>Disabled</Optimization>\r
+ <PreprocessorDefinitions>WIN32;_DEBUG;_WINDOWS;%(PreprocessorDefinitions)</PreprocessorDefinitions>\r
+ <SDLCheck>true</SDLCheck>\r
+ <RuntimeLibrary>MultiThreadedDebug</RuntimeLibrary>\r
+ </ClCompile>\r
+ <Link>\r
+ <SubSystem>Windows</SubSystem>\r
+ <GenerateDebugInformation>true</GenerateDebugInformation>\r
+ <UACExecutionLevel>RequireAdministrator</UACExecutionLevel>\r
+ </Link>\r
+ <Manifest>\r
+ <AdditionalManifestFiles>$(ProjectDir)\Res\Vlnk32.manifest %(AdditionalManifestFiles)</AdditionalManifestFiles>\r
+ </Manifest>\r
+ </ItemDefinitionGroup>\r
+ <ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">\r
+ <ClCompile>\r
+ <WarningLevel>Level3</WarningLevel>\r
+ <PrecompiledHeader>\r
+ </PrecompiledHeader>\r
+ <Optimization>MaxSpeed</Optimization>\r
+ <FunctionLevelLinking>true</FunctionLevelLinking>\r
+ <IntrinsicFunctions>true</IntrinsicFunctions>\r
+ <PreprocessorDefinitions>WIN32;NDEBUG;_WINDOWS;%(PreprocessorDefinitions)</PreprocessorDefinitions>\r
+ <SDLCheck>true</SDLCheck>\r
+ <RuntimeLibrary>MultiThreaded</RuntimeLibrary>\r
+ </ClCompile>\r
+ <Link>\r
+ <SubSystem>Windows</SubSystem>\r
+ <GenerateDebugInformation>true</GenerateDebugInformation>\r
+ <EnableCOMDATFolding>true</EnableCOMDATFolding>\r
+ <OptimizeReferences>true</OptimizeReferences>\r
+ <UACExecutionLevel>RequireAdministrator</UACExecutionLevel>\r
+ </Link>\r
+ <Manifest>\r
+ <AdditionalManifestFiles>$(ProjectDir)\Res\Vlnk32.manifest %(AdditionalManifestFiles)</AdditionalManifestFiles>\r
+ </Manifest>\r
+ </ItemDefinitionGroup>\r
+ <ItemGroup>\r
+ <ClInclude Include="..\..\..\src\vlnk.h" />\r
+ <ClInclude Include="resource.h" />\r
+ </ItemGroup>\r
+ <ItemGroup>\r
+ <ResourceCompile Include="VentoyVlnk.rc" />\r
+ </ItemGroup>\r
+ <ItemGroup>\r
+ <ClCompile Include="..\..\..\src\crc32.c" />\r
+ <ClCompile Include="..\..\..\src\main_windows.c" />\r
+ <ClCompile Include="..\..\..\src\vlnk.c" />\r
+ </ItemGroup>\r
+ <ItemGroup>\r
+ <Image Include="Res\vlnk.ico" />\r
+ </ItemGroup>\r
+ <Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" />\r
+ <ImportGroup Label="ExtensionTargets">\r
+ </ImportGroup>\r
+</Project>
\ No newline at end of file
--- /dev/null
+<?xml version="1.0" encoding="utf-8"?>\r
+<Project ToolsVersion="4.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">\r
+ <ItemGroup>\r
+ <Filter Include="源文件">\r
+ <UniqueIdentifier>{4FC737F1-C7A5-4376-A066-2A32D752A2FF}</UniqueIdentifier>\r
+ <Extensions>cpp;c;cc;cxx;def;odl;idl;hpj;bat;asm;asmx</Extensions>\r
+ </Filter>\r
+ <Filter Include="头文件">\r
+ <UniqueIdentifier>{93995380-89BD-4b04-88EB-625FBE52EBFB}</UniqueIdentifier>\r
+ <Extensions>h;hh;hpp;hxx;hm;inl;inc;xsd</Extensions>\r
+ </Filter>\r
+ <Filter Include="资源文件">\r
+ <UniqueIdentifier>{67DA6AB6-F800-4c08-8B7A-83BB121AAD01}</UniqueIdentifier>\r
+ <Extensions>rc;ico;cur;bmp;dlg;rc2;rct;bin;rgs;gif;jpg;jpeg;jpe;resx;tiff;tif;png;wav;mfcribbon-ms</Extensions>\r
+ </Filter>\r
+ </ItemGroup>\r
+ <ItemGroup>\r
+ <ClInclude Include="resource.h">\r
+ <Filter>头文件</Filter>\r
+ </ClInclude>\r
+ <ClInclude Include="..\..\..\src\vlnk.h">\r
+ <Filter>源文件</Filter>\r
+ </ClInclude>\r
+ </ItemGroup>\r
+ <ItemGroup>\r
+ <ResourceCompile Include="VentoyVlnk.rc">\r
+ <Filter>资源文件</Filter>\r
+ </ResourceCompile>\r
+ </ItemGroup>\r
+ <ItemGroup>\r
+ <ClCompile Include="..\..\..\src\crc32.c">\r
+ <Filter>源文件</Filter>\r
+ </ClCompile>\r
+ <ClCompile Include="..\..\..\src\main_windows.c">\r
+ <Filter>源文件</Filter>\r
+ </ClCompile>\r
+ <ClCompile Include="..\..\..\src\vlnk.c">\r
+ <Filter>源文件</Filter>\r
+ </ClCompile>\r
+ </ItemGroup>\r
+ <ItemGroup>\r
+ <Image Include="Res\vlnk.ico">\r
+ <Filter>资源文件</Filter>\r
+ </Image>\r
+ </ItemGroup>\r
+</Project>
\ No newline at end of file
--- /dev/null
+<?xml version="1.0" encoding="utf-8"?>\r
+<Project ToolsVersion="12.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">\r
+ <PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">\r
+ <LocalDebuggerWorkingDirectory>K:\test\ventoy\ventoy-1.0.57</LocalDebuggerWorkingDirectory>\r
+ <DebuggerFlavor>WindowsLocalDebugger</DebuggerFlavor>\r
+ </PropertyGroup>\r
+</Project>
\ No newline at end of file
#include <sys/stat.h>
#include <sys/types.h>
#include <linux/fs.h>
+#include <dirent.h>
#include "biso.h"
#include "biso_list.h"
#include "biso_util.h"
#include "biso_plat.h"
#include "biso_9660.h"
+#include "vtoytool.h"
#ifndef O_BINARY
#define O_BINARY 0
-static int vtoydm_print_linear_table(const char *img_map_file, const char *diskname)
+static int vtoydm_print_linear_table(const char *img_map_file, const char *diskname, int part, uint64_t offset)
{
int i;
int len;
#else
if (strstr(diskname, "nvme") || strstr(diskname, "mmc") || strstr(diskname, "nbd"))
{
- printf("%u %u linear %sp1 %llu\n",
+ printf("%u %u linear %sp%d %llu\n",
(sector_start << 2), disk_sector_num,
- diskname, (unsigned long long)chunk[i].disk_start_sector - 2048);
+ diskname, part, (unsigned long long)chunk[i].disk_start_sector - offset);
}
else
{
- printf("%u %u linear %s1 %llu\n",
+ printf("%u %u linear %s%d %llu\n",
(sector_start << 2), disk_sector_num,
- diskname, (unsigned long long)chunk[i].disk_start_sector - 2048);
+ diskname, part, (unsigned long long)chunk[i].disk_start_sector - offset);
}
#endif
}
return 0;
}
+static uint64_t vtoydm_get_part_start(const char *diskname, int part)
+{
+ int fd;
+ unsigned long long size = 0;
+ char diskpath[256] = {0};
+ char sizebuf[64] = {0};
+
+ if (strstr(diskname, "nvme") || strstr(diskname, "mmc") || strstr(diskname, "nbd"))
+ {
+ snprintf(diskpath, sizeof(diskpath) - 1, "/sys/class/block/%sp%d/start", diskname, part);
+ }
+ else
+ {
+ snprintf(diskpath, sizeof(diskpath) - 1, "/sys/class/block/%s%d/start", diskname, part);
+ }
+
+ if (access(diskpath, F_OK) >= 0)
+ {
+ debug("get part start from sysfs for %s %d\n", diskname, part);
+
+ fd = open(diskpath, O_RDONLY | O_BINARY);
+ if (fd >= 0)
+ {
+ read(fd, sizebuf, sizeof(sizebuf));
+ size = strtoull(sizebuf, NULL, 10);
+ close(fd);
+ return size;
+ }
+ }
+ else
+ {
+ debug("%s not exist \n", diskpath);
+ }
+
+ return size;
+}
+
+static int vtoydm_vlnk_convert(char *disk, int len, int *part, uint64_t *offset)
+{
+ int rc = 1;
+ int cnt = 0;
+ int rdlen;
+ FILE *fp = NULL;
+ ventoy_os_param param;
+ char diskname[128] = {0};
+
+ fp = fopen("/ventoy/ventoy_os_param", "rb");
+ if (!fp)
+ {
+ debug("dm vlnk convert not exist %d\n", errno);
+ goto end;
+ }
+
+ memset(¶m, 0, sizeof(param));
+ rdlen = (int)fread(¶m, 1, sizeof(param), fp);
+ if (rdlen != (int)sizeof(param))
+ {
+ debug("fread failed %d %d\n", rdlen, errno);
+ goto end;
+ }
+
+ debug("dm vlnk convert vtoy_reserved=%d\n", param.vtoy_reserved[6]);
+
+ if (param.vtoy_reserved[6])
+ {
+ cnt = vtoy_find_disk_by_guid(¶m, diskname);
+ debug("vtoy_find_disk_by_guid cnt=%d\n", cnt);
+ if (cnt == 1)
+ {
+ *part = param.vtoy_disk_part_id;
+ *offset = vtoydm_get_part_start(diskname, *part);
+
+ debug("VLNK <%s> <%s> <P%d> <%llu>\n", disk, diskname, *part, (unsigned long long)(*offset));
+
+ snprintf(disk, len, "/dev/%s", diskname);
+
+ rc = 0;
+ }
+ }
+
+end:
+ if (fp)
+ fclose(fp);
+ return rc;
+}
+
int vtoydm_main(int argc, char **argv)
{
int ch;
int cmd = 0;
+ int part = 1;
+ uint64_t offset = 2048;
unsigned long first_sector = 0;
unsigned long long file_size = 0;
char diskname[128] = {0};
debug("cmd=%d file=<%s> disk=<%s> first_sector=%lu file_size=%llu\n",
cmd, filepath, diskname, first_sector, file_size);
+ vtoydm_vlnk_convert(diskname, sizeof(diskname), &part, &offset);
+
switch (cmd)
{
case CMD_PRINT_TABLE:
{
- return vtoydm_print_linear_table(filepath, diskname);
+ return vtoydm_print_linear_table(filepath, diskname, part, offset);
}
case CMD_CREATE_DM:
{
#include <sys/types.h>
#include <linux/fs.h>
#include <dirent.h>
-
-#define IS_DIGIT(x) ((x) >= '0' && (x) <= '9')
-
-#ifndef USE_DIET_C
-#ifndef __mips__
-typedef unsigned long long uint64_t;
-#endif
-typedef unsigned int uint32_t;
-typedef unsigned short uint16_t;
-typedef unsigned char uint8_t;
-#endif
-
-#define VENTOY_GUID { 0x77772020, 0x2e77, 0x6576, { 0x6e, 0x74, 0x6f, 0x79, 0x2e, 0x6e, 0x65, 0x74 }}
-
-typedef enum ventoy_fs_type
-{
- ventoy_fs_exfat = 0, /* 0: exfat */
- ventoy_fs_ntfs, /* 1: NTFS */
- ventoy_fs_ext, /* 2: ext2/ext3/ext4 */
- ventoy_fs_xfs, /* 3: XFS */
- ventoy_fs_udf, /* 4: UDF */
- ventoy_fs_fat, /* 5: FAT */
-
- ventoy_fs_max
-}ventoy_fs_type;
-
-#pragma pack(1)
-
-typedef struct ventoy_guid
-{
- uint32_t data1;
- uint16_t data2;
- uint16_t data3;
- uint8_t data4[8];
-}ventoy_guid;
-
-
-typedef struct ventoy_image_disk_region
-{
- uint32_t image_sector_count; /* image sectors contained in this region */
- uint32_t image_start_sector; /* image sector start */
- uint64_t disk_start_sector; /* disk sector start */
-}ventoy_image_disk_region;
-
-typedef struct ventoy_image_location
-{
- ventoy_guid guid;
-
- /* image sector size, currently this value is always 2048 */
- uint32_t image_sector_size;
-
- /* disk sector size, normally the value is 512 */
- uint32_t disk_sector_size;
-
- uint32_t region_count;
-
- /*
- * disk region data
- * If the image file has more than one fragments in disk,
- * there will be more than one region data here.
- * You can calculate the region count by
- */
- ventoy_image_disk_region regions[1];
-
- /* ventoy_image_disk_region regions[2~region_count-1] */
-}ventoy_image_location;
-
-typedef struct ventoy_os_param
-{
- ventoy_guid guid; // VENTOY_GUID
- uint8_t chksum; // checksum
-
- uint8_t vtoy_disk_guid[16];
- uint64_t vtoy_disk_size; // disk size in bytes
- uint16_t vtoy_disk_part_id; // begin with 1
- uint16_t vtoy_disk_part_type; // 0:exfat 1:ntfs other: reserved
- char vtoy_img_path[384]; // It seems to be enough, utf-8 format
- uint64_t vtoy_img_size; // image file size in bytes
-
- /*
- * Ventoy will write a copy of ventoy_image_location data into runtime memory
- * this is the physically address and length of that memory.
- * Address 0 means no such data exist.
- * Address will be aligned by 4KB.
- *
- */
- uint64_t vtoy_img_location_addr;
- uint32_t vtoy_img_location_len;
-
- uint64_t vtoy_reserved[4]; // Internal use by ventoy
-
- uint8_t vtoy_disk_signature[4];
-
- uint8_t reserved[27];
-}ventoy_os_param;
-
-#pragma pack()
+#include "vtoytool.h"
#ifndef O_BINARY
#define O_BINARY 0
printf("param->vtoy_img_size = <%llu>\n", (unsigned long long)param->vtoy_img_size);
printf("param->vtoy_img_location_addr = <0x%llx>\n", (unsigned long long)param->vtoy_img_location_addr);
printf("param->vtoy_img_location_len = <%u>\n", param->vtoy_img_location_len);
- printf("param->vtoy_reserved[0] = 0x%llx\n", (unsigned long long)param->vtoy_reserved[0]);
- printf("param->vtoy_reserved[1] = 0x%llx\n", (unsigned long long)param->vtoy_reserved[1]);
+ printf("param->vtoy_reserved = %02x %02x %02x %02x %02x %02x %02x %02x\n",
+ param->vtoy_reserved[0],
+ param->vtoy_reserved[1],
+ param->vtoy_reserved[2],
+ param->vtoy_reserved[3],
+ param->vtoy_reserved[4],
+ param->vtoy_reserved[5],
+ param->vtoy_reserved[6],
+ param->vtoy_reserved[7]
+ );
printf("\n");
}
return rc;
}
-static int vtoy_find_disk_by_guid(ventoy_os_param *param, char *diskname)
+int vtoy_find_disk_by_guid(ventoy_os_param *param, char *diskname)
{
int rc = 0;
int count = 0;
}
memset(vtguid, 0, sizeof(vtguid));
+ memset(vtsig, 0, sizeof(vtsig));
rc = vtoy_get_disk_guid(p->d_name, vtguid, vtsig);
if (rc == 0 && memcmp(vtguid, param->vtoy_disk_guid, 16) == 0 &&
memcmp(vtsig, param->vtoy_disk_signature, 4) == 0)
return count;
}
+static int vtoy_find_disk_by_sig(uint8_t *sig, char *diskname)
+{
+ int rc = 0;
+ int count = 0;
+ DIR* dir = NULL;
+ struct dirent* p = NULL;
+ uint8_t vtguid[16];
+ uint8_t vtsig[16];
+
+ dir = opendir("/sys/block");
+ if (!dir)
+ {
+ return 0;
+ }
+
+ while ((p = readdir(dir)) != NULL)
+ {
+ if (!vtoy_is_possible_blkdev(p->d_name))
+ {
+ debug("disk %s is filted by name\n", p->d_name);
+ continue;
+ }
+
+ memset(vtguid, 0, sizeof(vtguid));
+ memset(vtsig, 0, sizeof(vtsig));
+ rc = vtoy_get_disk_guid(p->d_name, vtguid, vtsig);
+ if (rc == 0 && memcmp(vtsig, sig, 4) == 0)
+ {
+ sprintf(diskname, "%s", p->d_name);
+ count++;
+ }
+ }
+ closedir(dir);
+
+ return count;
+}
+
static int vtoy_printf_iso_path(ventoy_os_param *param)
{
printf("%s\n", param->vtoy_img_path);
return 0;
}
+static int vtoy_vlnk_printf(ventoy_os_param *param, char *diskname)
+{
+ int cnt = 0;
+ uint8_t disk_sig[4];
+ uint8_t mbr[512];
+ int fd = -1;
+ char diskpath[128];
+ uint8_t check[8] = { 0x56, 0x54, 0x00, 0x47, 0x65, 0x00, 0x48, 0x44 };
+
+ memcpy(disk_sig, param->vtoy_reserved + 7, 4);
+
+ debug("vlnk disk sig: %02x %02x %02x %02x \n", disk_sig[0], disk_sig[1], disk_sig[2], disk_sig[3]);
+
+ cnt = vtoy_find_disk_by_sig(disk_sig, diskname);
+ if (cnt == 1)
+ {
+ snprintf(diskpath, sizeof(diskpath), "/dev/%s", diskname);
+ fd = open(diskpath, O_RDONLY | O_BINARY);
+ if (fd >= 0)
+ {
+ memset(mbr, 0, sizeof(mbr));
+ read(fd, mbr, sizeof(mbr));
+ close(fd);
+
+ if (memcmp(mbr + 0x190, check, 8) == 0)
+ {
+ printf("/dev/%s", diskname);
+ return 0;
+ }
+ else
+ {
+ debug("check data failed /dev/%s\n", diskname);
+ }
+ }
+ }
+
+ debug("find count=%d\n", cnt);
+ printf("unknown");
+ return 1;
+}
+
static int vtoy_check_device(ventoy_os_param *param, const char *device)
{
unsigned long long size;
snprintf(diskpath, sizeof(diskpath) - 1, "/sys/class/block/%s2/size", diskname);
}
- if (access(diskpath, F_OK) >= 0)
+ if (param->vtoy_reserved[6] == 0 && access(diskpath, F_OK) >= 0)
{
debug("get part size from sysfs for %s\n", diskpath);
int ch;
int print_path = 0;
int print_fs = 0;
+ int vlnk_print = 0;
char filename[256] = {0};
char diskname[256] = {0};
char device[64] = {0};
ventoy_os_param *param = NULL;
- while ((ch = getopt(argc, argv, "c:f:p:s:v::")) != -1)
+ while ((ch = getopt(argc, argv, "c:f:p:t:s:v::")) != -1)
{
if (ch == 'f')
{
print_path = 1;
strncpy(filename, optarg, sizeof(filename) - 1);
}
+ else if (ch == 't')
+ {
+ vlnk_print = 1;
+ strncpy(filename, optarg, sizeof(filename) - 1);
+ }
else if (ch == 's')
{
print_fs = 1;
{
rc = vtoy_printf_fs(param);
}
+ else if (vlnk_print)
+ {
+ rc = vtoy_vlnk_printf(param, diskname);
+ }
else if (device[0])
{
rc = vtoy_check_device(param, device);
--- /dev/null
+/******************************************************************************
+ * vtoytool.h
+ *
+ * Copyright (c) 2022, 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/>.
+ *
+ */
+
+#ifndef __VTOYTOOL_H__
+#define __VTOYTOOL_H__
+
+#define IS_DIGIT(x) ((x) >= '0' && (x) <= '9')
+
+#ifndef USE_DIET_C
+#ifndef __mips__
+typedef unsigned long long uint64_t;
+#endif
+typedef unsigned int uint32_t;
+typedef unsigned short uint16_t;
+typedef unsigned char uint8_t;
+#endif
+
+#define VENTOY_GUID { 0x77772020, 0x2e77, 0x6576, { 0x6e, 0x74, 0x6f, 0x79, 0x2e, 0x6e, 0x65, 0x74 }}
+
+typedef enum ventoy_fs_type
+{
+ ventoy_fs_exfat = 0, /* 0: exfat */
+ ventoy_fs_ntfs, /* 1: NTFS */
+ ventoy_fs_ext, /* 2: ext2/ext3/ext4 */
+ ventoy_fs_xfs, /* 3: XFS */
+ ventoy_fs_udf, /* 4: UDF */
+ ventoy_fs_fat, /* 5: FAT */
+
+ ventoy_fs_max
+}ventoy_fs_type;
+
+#pragma pack(1)
+
+typedef struct ventoy_guid
+{
+ uint32_t data1;
+ uint16_t data2;
+ uint16_t data3;
+ uint8_t data4[8];
+}ventoy_guid;
+
+
+typedef struct ventoy_image_disk_region
+{
+ uint32_t image_sector_count; /* image sectors contained in this region */
+ uint32_t image_start_sector; /* image sector start */
+ uint64_t disk_start_sector; /* disk sector start */
+}ventoy_image_disk_region;
+
+typedef struct ventoy_image_location
+{
+ ventoy_guid guid;
+
+ /* image sector size, currently this value is always 2048 */
+ uint32_t image_sector_size;
+
+ /* disk sector size, normally the value is 512 */
+ uint32_t disk_sector_size;
+
+ uint32_t region_count;
+
+ /*
+ * disk region data
+ * If the image file has more than one fragments in disk,
+ * there will be more than one region data here.
+ * You can calculate the region count by
+ */
+ ventoy_image_disk_region regions[1];
+
+ /* ventoy_image_disk_region regions[2~region_count-1] */
+}ventoy_image_location;
+
+typedef struct ventoy_os_param
+{
+ ventoy_guid guid; // VENTOY_GUID
+ uint8_t chksum; // checksum
+
+ uint8_t vtoy_disk_guid[16];
+ uint64_t vtoy_disk_size; // disk size in bytes
+ uint16_t vtoy_disk_part_id; // begin with 1
+ uint16_t vtoy_disk_part_type; // 0:exfat 1:ntfs other: reserved
+ char vtoy_img_path[384]; // It seems to be enough, utf-8 format
+ uint64_t vtoy_img_size; // image file size in bytes
+
+ /*
+ * Ventoy will write a copy of ventoy_image_location data into runtime memory
+ * this is the physically address and length of that memory.
+ * Address 0 means no such data exist.
+ * Address will be aligned by 4KB.
+ *
+ */
+ uint64_t vtoy_img_location_addr;
+ uint32_t vtoy_img_location_len;
+
+ uint8_t vtoy_reserved[32]; // Internal use by ventoy
+
+ uint8_t vtoy_disk_signature[4];
+
+ uint8_t reserved[27];
+}ventoy_os_param;
+
+#pragma pack()
+
+int vtoy_find_disk_by_guid(ventoy_os_param *param, char *diskname);
+#endif
+
static CHAR g_prog_dir[MAX_PATH];\r
static CHAR g_prog_name[MAX_PATH];\r
\r
+#define VTOY_PECMD_PATH "X:\\Windows\\system32\\ventoy\\PECMD.EXE"\r
+#define ORG_PECMD_PATH "X:\\Windows\\system32\\PECMD.EXE"\r
+#define ORG_PECMD_BK_PATH "X:\\Windows\\system32\\PECMD.EXE_BACK.EXE"\r
+\r
#define AUTO_RUN_BAT "X:\\VentoyAutoRun.bat"\r
#define AUTO_RUN_LOG "X:\\VentoyAutoRun.log"\r
\r
return bRet;\r
}\r
\r
-static int GetPhyDiskUUID(const char LogicalDrive, UINT8 *UUID, DISK_EXTENT *DiskExtent)\r
+static int GetPhyDiskUUID(const char LogicalDrive, UINT8 *UUID, UINT32 *DiskSig, DISK_EXTENT *DiskExtent)\r
{\r
BOOL Ret;\r
DWORD dwSize;\r
}\r
CloseHandle(Handle);\r
\r
- memcpy(DiskExtent, DiskExtents.Extents, sizeof(DiskExtent));\r
- Log("%C: is in PhysicalDrive%d ", LogicalDrive, DiskExtents.Extents[0].DiskNumber);\r
+ memcpy(DiskExtent, DiskExtents.Extents, sizeof(DISK_EXTENT));\r
+ Log("%C: is in PhysicalDrive%d Offset:%llu", LogicalDrive, DiskExtents.Extents[0].DiskNumber, \r
+ (ULONGLONG)(DiskExtents.Extents[0].StartingOffset.QuadPart));\r
\r
sprintf_s(PhyPath, sizeof(PhyPath), "\\\\.\\PhysicalDrive%d", DiskExtents.Extents[0].DiskNumber);\r
Handle = CreateFileA(PhyPath, GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE, 0, OPEN_EXISTING, 0, 0);\r
}\r
\r
memcpy(UUID, SectorBuf + 0x180, 16);\r
+ if (DiskSig)\r
+ {\r
+ memcpy(DiskSig, SectorBuf + 0x1B8, 4);\r
+ }\r
+\r
CloseHandle(Handle);\r
return 0;\r
}\r
int i;\r
int rc;\r
BOOL find = FALSE;\r
+ BOOL vtoyfind = FALSE;\r
CHAR Letter;\r
CHAR MntLetter;\r
+ CHAR VtoyLetter;\r
DWORD Drives;\r
DWORD NewDrives;\r
+ UINT32 DiskSig;\r
+ UINT32 VtoySig;\r
DISK_EXTENT DiskExtent;\r
+ DISK_EXTENT VtoyDiskExtent;\r
UINT8 UUID[16];\r
CHAR IsoPath[MAX_PATH];\r
\r
if (IsFileExist("%s", IsoPath))\r
{\r
Log("File exist under %C:", Letter);\r
- if (GetPhyDiskUUID(Letter, UUID, &DiskExtent) == 0)\r
+ if (GetPhyDiskUUID(Letter, UUID, NULL, &DiskExtent) == 0)\r
{\r
if (memcmp(UUID, param->vtoy_disk_guid, 16) == 0)\r
{\r
\r
Log("Find ISO file <%s>", IsoPath);\r
\r
+ //Find VtoyLetter in Vlnk Mode\r
+ if (g_os_param_reserved[6] == 1)\r
+ {\r
+ memcpy(&VtoySig, g_os_param_reserved + 7, 4);\r
+ for (i = 0; i < 5; i++)\r
+ {\r
+ VtoyLetter = 'A';\r
+ Drives = GetLogicalDrives();\r
+ Log("Logic Drives: 0x%x VentoySig:%08X", Drives, VtoySig);\r
+\r
+ while (Drives)\r
+ {\r
+ if (Drives & 0x01)\r
+ {\r
+ if (GetPhyDiskUUID(VtoyLetter, UUID, &DiskSig, &VtoyDiskExtent) == 0)\r
+ {\r
+ Log("DiskSig=%08X PartStart=%lld", DiskSig, VtoyDiskExtent.StartingOffset.QuadPart);\r
+ if (DiskSig == VtoySig && VtoyDiskExtent.StartingOffset.QuadPart == SIZE_1MB)\r
+ {\r
+ Log("Ventoy Disk Sig match");\r
+ vtoyfind = TRUE;\r
+ break;\r
+ }\r
+ }\r
+ }\r
+\r
+ Drives >>= 1;\r
+ VtoyLetter++;\r
+ }\r
+\r
+ if (vtoyfind)\r
+ {\r
+ Log("Find Ventoy Letter: %C", VtoyLetter);\r
+ break;\r
+ }\r
+ else\r
+ {\r
+ Log("Now wait and retry ...");\r
+ Sleep(1000);\r
+ }\r
+ }\r
+\r
+ if (vtoyfind == FALSE)\r
+ {\r
+ Log("Failed to find ventoy disk");\r
+ return 1;\r
+ }\r
+ }\r
+ else\r
+ {\r
+ VtoyLetter = Letter;\r
+ Log("No vlnk mode %C", Letter);\r
+ }\r
+\r
Drives = GetLogicalDrives();\r
Log("Drives before mount: 0x%x", Drives);\r
\r
}\r
\r
// for protect\r
- rc = DeleteVentoyPart2MountPoint(DiskExtent.DiskNumber);\r
+ rc = DeleteVentoyPart2MountPoint(VtoyDiskExtent.DiskNumber);\r
Log("Delete ventoy mountpoint: %s", rc == 0 ? "SUCCESS" : "NO NEED");\r
\r
if (g_windows_data.auto_install_script[0])\r
{\r
- sprintf_s(IsoPath, sizeof(IsoPath), "%C:%s", Letter, g_windows_data.auto_install_script);\r
+ sprintf_s(IsoPath, sizeof(IsoPath), "%C:%s", VtoyLetter, g_windows_data.auto_install_script);\r
if (IsFileExist("%s", IsoPath))\r
{\r
Log("use auto install script %s...", IsoPath);\r
\r
if (g_windows_data.injection_archive[0])\r
{\r
- sprintf_s(IsoPath, sizeof(IsoPath), "%C:%s", Letter, g_windows_data.injection_archive);\r
+ sprintf_s(IsoPath, sizeof(IsoPath), "%C:%s", VtoyLetter, g_windows_data.injection_archive);\r
if (IsFileExist("%s", IsoPath))\r
{\r
Log("decompress injection archive %s...", IsoPath);\r
- DecompressInjectionArchive(IsoPath, DiskExtent.DiskNumber);\r
+ DecompressInjectionArchive(IsoPath, VtoyDiskExtent.DiskNumber);\r
\r
if (IsFileExist("%s", AUTO_RUN_BAT))\r
{\r
return 0;\r
}\r
\r
+static void VentoyToUpper(CHAR *str)\r
+{\r
+ int i;\r
+ for (i = 0; str[i]; i++)\r
+ {\r
+ str[i] = (CHAR)toupper(str[i]);\r
+ }\r
+}\r
\r
int main(int argc, char **argv)\r
{\r
}\r
else if (_stricmp(g_prog_name, "PECMD.exe") == 0)\r
{\r
- Log("We need to rejump for pecmd ...");\r
+ strcpy_s(NewArgv0, sizeof(NewArgv0), g_prog_dir);\r
+ VentoyToUpper(NewArgv0);\r
+ \r
+ if (NULL == strstr(NewArgv0, "SYSTEM32") && IsFileExist(ORG_PECMD_BK_PATH))\r
+ {\r
+ Log("Just call original pecmd.exe");\r
+ strcpy_s(CallParam, sizeof(CallParam), ORG_PECMD_PATH);\r
+ }\r
+ else\r
+ {\r
+ Log("We need to rejump for pecmd ...");\r
\r
- ventoy_check_create_directory();\r
- CopyFileA(g_prog_full_path, "ventoy\\WinLogon.exe", TRUE);\r
+ ventoy_check_create_directory();\r
+ CopyFileA(g_prog_full_path, "ventoy\\WinLogon.exe", TRUE);\r
\r
- sprintf_s(CallParam, sizeof(CallParam), "ventoy\\WinLogon.exe %s", g_prog_full_path);\r
+ sprintf_s(CallParam, sizeof(CallParam), "ventoy\\WinLogon.exe %s", g_prog_full_path);\r
+ }\r
+ \r
for (i = 1; i < argc; i++)\r
{\r
strcat_s(CallParam, sizeof(CallParam), " ");\r
projects / Ventoy.git / commitdiff