#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;
#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
uint64_t vtoy_reserved[4]; // Internal use by ventoy
- uint8_t reserved[31];
+ uint8_t vtoy_disk_signature[4];
+
+ uint8_t reserved[27];
}ventoy_os_param;
#pragma pack()
static ventoy_guid vtoy_guid = VENTOY_GUID;
+static const char *g_ventoy_fs[ventoy_fs_max] =
+{
+ "exfat", "ntfs", "ext*", "xfs", "udf", "fat"
+};
+
static int vtoy_check_os_param(ventoy_os_param *param)
{
uint32_t i;
if (memcmp(¶m->guid, &vtoy_guid, sizeof(ventoy_guid)))
{
+ uint8_t *data1 = (uint8_t *)(¶m->guid);
+ uint8_t *data2 = (uint8_t *)(&vtoy_guid);
+
+ for (i = 0; i < 16; i++)
+ {
+ if (data1[i] != data2[i])
+ {
+ debug("guid not equal i = %u, 0x%02x, 0x%02x\n", i, data1[i], data2[i]);
+ }
+ }
return 1;
}
printf("param->vtoy_disk_guid = %02x %02x %02x %02x\n",
param->vtoy_disk_guid[0], param->vtoy_disk_guid[1],
param->vtoy_disk_guid[2], param->vtoy_disk_guid[3]);
+
+ printf("param->vtoy_disk_signature = %02x %02x %02x %02x\n",
+ param->vtoy_disk_signature[0], param->vtoy_disk_signature[1],
+ param->vtoy_disk_signature[2], param->vtoy_disk_signature[3]);
+
printf("param->vtoy_disk_size = %llu\n", (unsigned long long)param->vtoy_disk_size);
printf("param->vtoy_disk_part_id = %u\n", param->vtoy_disk_part_id);
printf("param->vtoy_disk_part_type = %u\n", param->vtoy_disk_part_type);
printf("\n");
}
-static int vtoy_get_disk_guid(const char *diskname, uint8_t *vtguid)
+static int vtoy_get_disk_guid(const char *diskname, uint8_t *vtguid, uint8_t *vtsig)
{
int i = 0;
int fd = 0;
{
lseek(fd, 0x180, SEEK_SET);
read(fd, vtguid, 16);
+
+ lseek(fd, 0x1b8, SEEK_SET);
+ read(fd, vtsig, 4);
close(fd);
debug("GUID for %s: <", devdisk);
return rc;
}
-static int vtoy_find_disk_by_guid(uint8_t *guid, char *diskname)
+static int vtoy_find_disk_by_guid(ventoy_os_param *param, 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)
}
memset(vtguid, 0, sizeof(vtguid));
- rc = vtoy_get_disk_guid(p->d_name, vtguid);
- if (rc == 0 && memcmp(vtguid, guid, 16) == 0)
+ 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)
{
sprintf(diskname, "%s", p->d_name);
count++;
return 0;
}
+static int vtoy_check_device(ventoy_os_param *param, const char *device)
+{
+ unsigned long long size;
+ uint8_t vtguid[16] = {0};
+ uint8_t vtsig[4] = {0};
+
+ debug("vtoy_check_device for <%s>\n", device);
+
+ size = vtoy_get_disk_size_in_byte(device);
+ vtoy_get_disk_guid(device, vtguid, vtsig);
+
+ debug("param->vtoy_disk_size=%llu size=%llu\n",
+ (unsigned long long)param->vtoy_disk_size, (unsigned long long)size);
+
+ if (memcmp(vtguid, param->vtoy_disk_guid, 16) == 0 &&
+ memcmp(vtsig, param->vtoy_disk_signature, 4) == 0)
+ {
+ debug("<%s> is right ventoy disk\n", device);
+ return 0;
+ }
+ else
+ {
+ debug("<%s> is NOT right ventoy disk\n", device);
+ return 1;
+ }
+}
+
static int vtoy_print_os_param(ventoy_os_param *param, char *diskname)
{
- int cnt = 0;
+ int fd, size;
+ int cnt = 0;
char *path = param->vtoy_img_path;
const char *fs;
-
+ char diskpath[256] = {0};
+ char sizebuf[64] = {0};
+
cnt = vtoy_find_disk_by_size(param->vtoy_disk_size, diskname);
- if (cnt > 1)
+ debug("find disk by size %llu, cnt=%d...\n", (unsigned long long)param->vtoy_disk_size, cnt);
+ if (1 == cnt)
{
- cnt = vtoy_find_disk_by_guid(param->vtoy_disk_guid, diskname);
+ if (vtoy_check_device(param, diskname) != 0)
+ {
+ cnt = 0;
+ }
}
- else if (cnt == 0)
+ else
{
- cnt = vtoy_find_disk_by_guid(param->vtoy_disk_guid, diskname);
- debug("find 0 disk by size, try with guid cnt=%d...\n", cnt);
+ cnt = vtoy_find_disk_by_guid(param, diskname);
+ debug("find disk by guid cnt=%d...\n", cnt);
}
- if (param->vtoy_disk_part_type == 0)
+ if (param->vtoy_disk_part_type < ventoy_fs_max)
{
- fs = "exfat";
- }
- else if (param->vtoy_disk_part_type == 0)
- {
- fs = "ntfs";
+ fs = g_ventoy_fs[param->vtoy_disk_part_type];
}
else
{
if (1 == cnt)
{
- printf("/dev/%s#%s#%s\n", diskname, fs, path);
- return 0;
- }
- else
- {
- return 1;
- }
-}
-
-static int vtoy_check_device(ventoy_os_param *param, const char *device)
-{
- unsigned long long size;
- uint8_t vtguid[16] = {0};
-
- debug("vtoy_check_device for <%s>\n", device);
-
- size = vtoy_get_disk_size_in_byte(device);
- vtoy_get_disk_guid(device, vtguid);
+ if (strstr(diskname, "nvme") || strstr(diskname, "mmc") || strstr(diskname, "nbd"))
+ {
+ snprintf(diskpath, sizeof(diskpath) - 1, "/sys/class/block/%sp2/size", diskname);
+ }
+ else
+ {
+ snprintf(diskpath, sizeof(diskpath) - 1, "/sys/class/block/%s2/size", diskname);
+ }
- debug("param->vtoy_disk_size=%llu size=%llu\n",
- (unsigned long long)param->vtoy_disk_size, (unsigned long long)size);
+ if (access(diskpath, F_OK) >= 0)
+ {
+ debug("get part size from sysfs for %s\n", diskpath);
+
+ fd = open(diskpath, O_RDONLY | O_BINARY);
+ if (fd >= 0)
+ {
+ read(fd, sizebuf, sizeof(sizebuf));
+ size = (int)strtoull(sizebuf, NULL, 10);
+ close(fd);
+ if ((size != (64 * 1024)) && (size != (8 * 1024)))
+ {
+ debug("sizebuf=<%s> size=%d\n", sizebuf, size);
+ return 1;
+ }
+ }
+ }
+ else
+ {
+ debug("%s not exist \n", diskpath);
+ }
- if (param->vtoy_disk_size == size &&
- memcmp(vtguid, param->vtoy_disk_guid, 16) == 0)
- {
- debug("<%s> is right ventoy disk\n", device);
+ printf("/dev/%s#%s#%s\n", diskname, fs, path);
return 0;
}
else
{
- debug("<%s> is NOT right ventoy disk\n", device);
return 1;
}
}
rc = vtoy_os_param_from_file(filename, param);
if (rc)
{
- debug("ventoy os param not found %d\n", rc);
- goto end;
+ debug("ventoy os param not found %d %d\n", rc, ENOENT);
+ if (ENOENT == rc)
+ {
+ debug("now try with file %s\n", "/ventoy/ventoy_os_param");
+ rc = vtoy_os_param_from_file("/ventoy/ventoy_os_param", param);
+ if (rc)
+ {
+ goto end;
+ }
+ }
+ else
+ {
+ goto end;
+ }
}
if (verbose)
projects / Ventoy.git / blobdiff