mirror of
https://github.com/ventoy/Ventoy.git
synced 2025-10-05 01:36:18 +00:00
VentoyPlugson ---- A GUI ventoy.json configurator
This commit is contained in:
longpanda
527
Plugson/src/main_windows.c
Normal file
527
Plugson/src/main_windows.c
Normal file
@@ -0,0 +1,527 @@
|
||||
#include <Windows.h>
|
||||
#include <Shlobj.h>
|
||||
#include <tlhelp32.h>
|
||||
#include <Psapi.h>
|
||||
#include <commctrl.h>
|
||||
#include <resource.h>
|
||||
#include <ventoy_define.h>
|
||||
#include <ventoy_util.h>
|
||||
#include <ventoy_json.h>
|
||||
#include <ventoy_disk.h>
|
||||
#include <ventoy_http.h>
|
||||
|
||||
static BOOL g_running = FALSE;
|
||||
static HWND g_refresh_button;
|
||||
static HWND g_start_button;
|
||||
static HWND g_openlink_button;
|
||||
static HWND g_exit_button;
|
||||
static HWND g_ComboxHwnd;
|
||||
|
||||
typedef enum MSGID
|
||||
{
|
||||
MSGID_ERROR = 0,
|
||||
MSGID_INFO,
|
||||
MSGID_INVALID_DIR,
|
||||
MSGID_NEW_DIR_FAIL,
|
||||
MSGID_RENAME_VENTOY,
|
||||
MSGID_INTERNAL_ERR,
|
||||
MSGID_BTN_REFRESH,
|
||||
MSGID_BTN_START,
|
||||
MSGID_BTN_STOP,
|
||||
MSGID_BTN_LINK,
|
||||
MSGID_BTN_EXIT,
|
||||
|
||||
MSGID_BTN_STOP_TIP,
|
||||
MSGID_BTN_EXIT_TIP,
|
||||
MSGID_RUNNING_TIP,
|
||||
|
||||
MSGID_BUTT
|
||||
}MSGID;
|
||||
|
||||
|
||||
const WCHAR *g_msg_cn[MSGID_BUTT] =
|
||||
{
|
||||
L"<EFBFBD><EFBFBD><EFBFBD><EFBFBD>",
|
||||
L"<EFBFBD><EFBFBD><EFBFBD><EFBFBD>",
|
||||
L"<EFBFBD><EFBFBD><EFBFBD><EFBFBD> Ventoy <20≯<EFBFBD>Ŀ¼<C4BF><C2BC><EFBFBD><EFBFBD><EFBFBD>б<EFBFBD><D0B1><EFBFBD><EFBFBD><EFBFBD><F2A3A1A3><EFBFBD><EFBFBD><EFBFBD>ISO<53>ļ<EFBFBD><C4BC><EFBFBD>λ<EFBFBD>ã<EFBFBD>",
|
||||
L"<EFBFBD><EFBFBD><EFBFBD><EFBFBD> ventoy Ŀ¼ʧ<C2BC>ܣ<EFBFBD><DCA3><EFBFBD><DEB7><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>",
|
||||
L"ventoy Ŀ¼<C4BF><C2BC><EFBFBD>ڣ<EFBFBD><DAA3><EFBFBD><EFBFBD>Ǵ<EFBFBD>Сд<D0A1><D0B4>ƥ<EFBFBD>䣬<EFBFBD><E4A3AC><EFBFBD>Ƚ<EFBFBD><C8BD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>",
|
||||
L"<EFBFBD>ڲ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>˳<EFBFBD><EFBFBD><EFBFBD>",
|
||||
L"ˢ<EFBFBD><EFBFBD>",
|
||||
L"<EFBFBD><EFBFBD><EFBFBD><EFBFBD>",
|
||||
L"ֹͣ",
|
||||
L"<EFBFBD><EFBFBD><EFBFBD><EFBFBD>",
|
||||
L"<EFBFBD>˳<EFBFBD>",
|
||||
|
||||
L"ֹͣ<EFBFBD><EFBFBD><EFBFBD>к<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ҳ<EFBFBD>潫<EFBFBD><EFBFBD><EFBFBD>رգ<EFBFBD><EFBFBD>Ƿ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>",
|
||||
L"<EFBFBD><EFBFBD>ǰ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>У<EFBFBD><EFBFBD>Ƿ<EFBFBD><EFBFBD>˳<EFBFBD><EFBFBD><EFBFBD>",
|
||||
L"<EFBFBD><EFBFBD><EFBFBD>ȹر<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>е<EFBFBD> VentoyPlugson <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>",
|
||||
};
|
||||
const WCHAR *g_msg_en[MSGID_BUTT] =
|
||||
{
|
||||
L"Error",
|
||||
L"Info",
|
||||
L"Please run me at the root of Ventoy partition.",
|
||||
L"Failed to create ventoy directory!",
|
||||
L"ventoy directory case mismatch, please rename it first!",
|
||||
L"Internal error, the program will exit!",
|
||||
L"Refresh",
|
||||
L"Start",
|
||||
L"Stop",
|
||||
L"Link",
|
||||
L"Exit",
|
||||
|
||||
L"The browser page will close after stop, continue?",
|
||||
L"Service is running, continue?",
|
||||
L"Please close another running VentoyPlugson instance!",
|
||||
};
|
||||
|
||||
const WCHAR **g_msg_lang = NULL;
|
||||
|
||||
HINSTANCE g_hInst;
|
||||
|
||||
char g_log_file[MAX_PATH];
|
||||
char g_cur_dir[MAX_PATH];
|
||||
|
||||
int ventoy_log_init(void);
|
||||
void ventoy_log_exit(void);
|
||||
|
||||
static BOOL OnDestroyDialog()
|
||||
{
|
||||
ventoy_http_exit();
|
||||
ventoy_disk_exit();
|
||||
#ifndef VENTOY_SIM
|
||||
ventoy_www_exit();
|
||||
#endif
|
||||
ventoy_log_exit();
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
|
||||
static void OpenURL(void)
|
||||
{
|
||||
int i;
|
||||
char url[128];
|
||||
const static char * Browsers[] =
|
||||
{
|
||||
"C:\\Program Files (x86)\\Google\\Chrome\\Application\\chrome.exe",
|
||||
"C:\\Program Files\\Google\\Chrome\\Application\\chrome.exe",
|
||||
"C:\\Program Files (x86)\\Mozilla Firefox\\firefox.exe",
|
||||
"C:\\Program Files\\Mozilla Firefox\\firefox.exe",
|
||||
NULL
|
||||
};
|
||||
|
||||
sprintf_s(url, sizeof(url), "http://%s:%s/index.html", g_sysinfo.ip, g_sysinfo.port);
|
||||
|
||||
for (i = 0; Browsers[i] != NULL; i++)
|
||||
{
|
||||
if (ventoy_is_file_exist("%s", Browsers[i]))
|
||||
{
|
||||
ShellExecuteA(NULL, "open", Browsers[i], url, NULL, SW_SHOW);
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
ShellExecuteA(NULL, "open", url, NULL, NULL, SW_SHOW);
|
||||
}
|
||||
|
||||
|
||||
static void FillCombox(HWND hWnd)
|
||||
{
|
||||
int i = 0;
|
||||
int num = 0;
|
||||
const ventoy_disk *list = NULL;
|
||||
CHAR DeviceName[256];
|
||||
|
||||
// delete all items
|
||||
SendMessage(g_ComboxHwnd, CB_RESETCONTENT, 0, 0);
|
||||
|
||||
list = ventoy_get_disk_list(&num);
|
||||
if (NULL == list || num <= 0)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
for (i = 0; i < num; i++)
|
||||
{
|
||||
sprintf_s(DeviceName, sizeof(DeviceName),
|
||||
"%C: [%s] %s",
|
||||
list[i].devname[0],
|
||||
list[i].cur_capacity,
|
||||
list[i].cur_model);
|
||||
SendMessageA(g_ComboxHwnd, CB_ADDSTRING, 0, (LPARAM)DeviceName);
|
||||
}
|
||||
SendMessage(g_ComboxHwnd, CB_SETCURSEL, 0, 0);
|
||||
}
|
||||
|
||||
|
||||
static BOOL InitDialog(HWND hWnd, WPARAM wParam, LPARAM lParam)
|
||||
{
|
||||
HICON hIcon;
|
||||
|
||||
g_ComboxHwnd = GetDlgItem(hWnd, IDC_COMBO1);
|
||||
g_refresh_button = GetDlgItem(hWnd, IDC_BUTTON1);
|
||||
g_start_button = GetDlgItem(hWnd, IDC_BUTTON2);
|
||||
g_openlink_button = GetDlgItem(hWnd, IDC_BUTTON3);
|
||||
g_exit_button = GetDlgItem(hWnd, IDC_BUTTON4);
|
||||
|
||||
hIcon = LoadIcon(g_hInst, MAKEINTRESOURCE(IDI_ICON1));
|
||||
SendMessage(hWnd, WM_SETICON, ICON_BIG, (LPARAM)hIcon);
|
||||
SendMessage(hWnd, WM_SETICON, ICON_SMALL, (LPARAM)hIcon);
|
||||
|
||||
SetWindowTextW(g_refresh_button, g_msg_lang[MSGID_BTN_REFRESH]);
|
||||
SetWindowTextW(g_start_button, g_msg_lang[MSGID_BTN_START]);
|
||||
SetWindowTextW(g_openlink_button, g_msg_lang[MSGID_BTN_LINK]);
|
||||
SetWindowTextW(g_exit_button, g_msg_lang[MSGID_BTN_EXIT]);
|
||||
|
||||
EnableWindow(g_openlink_button, FALSE);
|
||||
|
||||
FillCombox(hWnd);
|
||||
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
static void VentoyStopService()
|
||||
{
|
||||
ventoy_http_stop();
|
||||
}
|
||||
|
||||
static int VentoyStartService(int sel)
|
||||
{
|
||||
int rc;
|
||||
BOOL bRet;
|
||||
char Path[128];
|
||||
char CurDir[MAX_PATH];
|
||||
const ventoy_disk *disk = NULL;
|
||||
|
||||
vlog("VentoyStartService ...\n");
|
||||
|
||||
disk = ventoy_get_disk_node(sel);
|
||||
if (disk == NULL)
|
||||
{
|
||||
return 1;
|
||||
}
|
||||
|
||||
vlog("Start service at %C: %s %s ...\n", disk->devname[0], disk->cur_model, disk->cur_ventoy_ver);
|
||||
|
||||
g_cur_dir[0] = disk->devname[0];
|
||||
g_cur_dir[1] = ':';
|
||||
g_cur_dir[2] = '\\';
|
||||
g_cur_dir[3] = 0;
|
||||
|
||||
g_sysinfo.pathcase = disk->pathcase;
|
||||
g_sysinfo.cur_secureboot = disk->cur_secureboot;
|
||||
g_sysinfo.cur_part_style = disk->cur_part_style;
|
||||
strlcpy(g_sysinfo.cur_fsname, disk->cur_fsname);
|
||||
strlcpy(g_sysinfo.cur_capacity, disk->cur_capacity);
|
||||
strlcpy(g_sysinfo.cur_model, disk->cur_model);
|
||||
strlcpy(g_sysinfo.cur_ventoy_ver, disk->cur_ventoy_ver);
|
||||
|
||||
bRet = SetCurrentDirectoryA(g_cur_dir);
|
||||
vlog("SetCurrentDirectoryA %u <%s>\n", bRet, g_cur_dir);
|
||||
|
||||
CurDir[0] = 0;
|
||||
GetCurrentDirectoryA(sizeof(CurDir), CurDir);
|
||||
vlog("CurDir=<%s>\n", CurDir);
|
||||
|
||||
if (strcmp(g_cur_dir, CurDir))
|
||||
{
|
||||
vlog("Failed to change current directory.");
|
||||
}
|
||||
|
||||
g_cur_dir[2] = 0;
|
||||
|
||||
if (ventoy_is_directory_exist("ventoy"))
|
||||
{
|
||||
if (g_sysinfo.pathcase)
|
||||
{
|
||||
vlog("ventoy directory already exist, check case sensitive.\n");
|
||||
strlcpy(Path, "ventoy");
|
||||
|
||||
rc = ventoy_path_case(Path, 0);
|
||||
vlog("ventoy_path_case actual path is <%s> <count:%d>\n", Path, rc);
|
||||
|
||||
if (rc)
|
||||
{
|
||||
vlog("ventoy directory case mismatch, rename<%s>--><%s>\n", Path, "ventoy");
|
||||
if (MoveFileA(Path, "ventoy"))
|
||||
{
|
||||
vlog("Rename <%s>--><%s> success\n", Path, "ventoy");
|
||||
}
|
||||
else
|
||||
{
|
||||
vlog("Rename <%s>--><%s> failed %u\n", Path, "ventoy", LASTERR);
|
||||
MessageBoxW(NULL, g_msg_lang[MSGID_RENAME_VENTOY], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
vlog("ventoy directory already exist, no need to check case sensitive.\n");
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (CreateDirectoryA("ventoy", NULL))
|
||||
{
|
||||
vlog("Create ventoy directory success.\n");
|
||||
}
|
||||
else
|
||||
{
|
||||
vlog("Create ventoy directory failed %u.\n", LASTERR);
|
||||
MessageBoxW(NULL, g_msg_lang[MSGID_NEW_DIR_FAIL], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
return ventoy_http_start(g_sysinfo.ip, g_sysinfo.port);
|
||||
}
|
||||
|
||||
INT_PTR CALLBACK DialogProc(HWND hWnd, UINT Message, WPARAM wParam, LPARAM lParam)
|
||||
{
|
||||
int rc;
|
||||
int nCurSel;
|
||||
WORD NotifyCode;
|
||||
WORD CtrlID;
|
||||
|
||||
switch (Message)
|
||||
{
|
||||
case WM_NOTIFY:
|
||||
{
|
||||
UINT code = 0;
|
||||
UINT_PTR idFrom = 0;
|
||||
|
||||
if (lParam)
|
||||
{
|
||||
code = ((LPNMHDR)lParam)->code;
|
||||
idFrom = ((LPNMHDR)lParam)->idFrom;
|
||||
}
|
||||
|
||||
if (idFrom == IDC_SYSLINK1 && (NM_CLICK == code || NM_RETURN == code))
|
||||
{
|
||||
OpenURL();
|
||||
}
|
||||
break;
|
||||
}
|
||||
case WM_COMMAND:
|
||||
{
|
||||
NotifyCode = HIWORD(wParam);
|
||||
CtrlID = LOWORD(wParam);
|
||||
|
||||
if (NotifyCode == BN_CLICKED)
|
||||
{
|
||||
if (CtrlID == IDC_BUTTON1)
|
||||
{
|
||||
if (!g_running)
|
||||
{
|
||||
//refresh
|
||||
ventoy_disk_exit();
|
||||
ventoy_disk_init();
|
||||
FillCombox(hWnd);
|
||||
}
|
||||
}
|
||||
else if (CtrlID == IDC_BUTTON2)
|
||||
{
|
||||
if (g_running)
|
||||
{
|
||||
if (IDYES == MessageBoxW(NULL, g_msg_lang[MSGID_BTN_STOP_TIP], g_msg_lang[MSGID_INFO], MB_YESNO | MB_ICONINFORMATION))
|
||||
{
|
||||
VentoyStopService();
|
||||
|
||||
g_running = FALSE;
|
||||
SetWindowTextW(g_start_button, g_msg_lang[MSGID_BTN_START]);
|
||||
EnableWindow(g_ComboxHwnd, TRUE);
|
||||
EnableWindow(g_refresh_button, TRUE);
|
||||
EnableWindow(g_openlink_button, FALSE);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
nCurSel = (int)SendMessage(g_ComboxHwnd, CB_GETCURSEL, 0, 0);
|
||||
if (CB_ERR != nCurSel)
|
||||
{
|
||||
rc = VentoyStartService(nCurSel);
|
||||
if (rc)
|
||||
{
|
||||
vlog("Ventoy failed to start http server, check %s for detail\n", g_log_file);
|
||||
MessageBoxW(NULL, g_msg_lang[MSGID_INTERNAL_ERR], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);
|
||||
}
|
||||
else
|
||||
{
|
||||
g_running = TRUE;
|
||||
SetWindowTextW(g_start_button, g_msg_lang[MSGID_BTN_STOP]);
|
||||
|
||||
EnableWindow(g_ComboxHwnd, FALSE);
|
||||
EnableWindow(g_refresh_button, FALSE);
|
||||
EnableWindow(g_openlink_button, TRUE);
|
||||
|
||||
OpenURL();
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
else if (CtrlID == IDC_BUTTON3)
|
||||
{
|
||||
if (g_running)
|
||||
{
|
||||
OpenURL();
|
||||
}
|
||||
}
|
||||
else if (CtrlID == IDC_BUTTON4)
|
||||
{
|
||||
if (g_running)
|
||||
{
|
||||
if (IDYES != MessageBoxW(NULL, g_msg_lang[MSGID_BTN_EXIT_TIP], g_msg_lang[MSGID_INFO], MB_YESNO | MB_ICONINFORMATION))
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
|
||||
ventoy_http_stop();
|
||||
}
|
||||
|
||||
OnDestroyDialog();
|
||||
EndDialog(hWnd, 0);
|
||||
}
|
||||
}
|
||||
break;
|
||||
}
|
||||
case WM_INITDIALOG:
|
||||
{
|
||||
InitDialog(hWnd, wParam, lParam);
|
||||
break;
|
||||
}
|
||||
case WM_CLOSE:
|
||||
{
|
||||
if (g_running)
|
||||
{
|
||||
if (IDYES != MessageBoxW(NULL, g_msg_lang[MSGID_BTN_EXIT_TIP], g_msg_lang[MSGID_INFO], MB_YESNO | MB_ICONINFORMATION))
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
|
||||
VentoyStopService();
|
||||
}
|
||||
|
||||
OnDestroyDialog();
|
||||
EndDialog(hWnd, 0);
|
||||
}
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int ParseCmdLine(LPSTR lpCmdLine, char *ip, char *port)
|
||||
{
|
||||
int portnum;
|
||||
char *ipstart = ip;
|
||||
char *pos;
|
||||
|
||||
|
||||
if (!lpCmdLine)
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
|
||||
pos = strstr(lpCmdLine, "-H");
|
||||
if (!pos)
|
||||
{
|
||||
pos = strstr(lpCmdLine, "-h");
|
||||
}
|
||||
|
||||
if (pos)
|
||||
{
|
||||
pos += 2;
|
||||
while (*pos == ' ' || *pos == '\t')
|
||||
{
|
||||
pos++;
|
||||
}
|
||||
|
||||
while (isdigit(*pos) || *pos == '.')
|
||||
{
|
||||
*ipstart++ = *pos++;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
pos = strstr(lpCmdLine, "-P");
|
||||
if (!pos)
|
||||
{
|
||||
pos = strstr(lpCmdLine, "-p");
|
||||
}
|
||||
|
||||
if (pos)
|
||||
{
|
||||
portnum = (int)strtol(pos + 3, NULL, 10);
|
||||
sprintf_s(port, 16, "%d", portnum);
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int APIENTRY WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, INT nCmdShow)
|
||||
{
|
||||
int rc;
|
||||
HANDLE hMutex;
|
||||
|
||||
UNREFERENCED_PARAMETER(hPrevInstance);
|
||||
|
||||
if (GetUserDefaultUILanguage() == 0x0804)
|
||||
{
|
||||
g_sysinfo.language = LANGUAGE_CN;
|
||||
g_msg_lang = g_msg_cn;
|
||||
}
|
||||
else
|
||||
{
|
||||
g_sysinfo.language = LANGUAGE_EN;
|
||||
g_msg_lang = g_msg_en;
|
||||
}
|
||||
|
||||
hMutex = CreateMutexA(NULL, TRUE, "PlugsonMUTEX");
|
||||
if ((hMutex != NULL) && (GetLastError() == ERROR_ALREADY_EXISTS))
|
||||
{
|
||||
MessageBoxW(NULL, g_msg_lang[MSGID_RUNNING_TIP], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);
|
||||
return 1;
|
||||
}
|
||||
|
||||
GetCurrentDirectoryA(MAX_PATH, g_cur_dir);
|
||||
sprintf_s(g_log_file, sizeof(g_log_file), "%s\\%s", g_cur_dir, LOG_FILE);
|
||||
ventoy_log_init();
|
||||
|
||||
|
||||
ParseCmdLine(lpCmdLine, g_sysinfo.ip, g_sysinfo.port);
|
||||
if (g_sysinfo.ip[0] == 0)
|
||||
{
|
||||
strlcpy(g_sysinfo.ip, "127.0.0.1");
|
||||
}
|
||||
if (g_sysinfo.port[0] == 0)
|
||||
{
|
||||
strlcpy(g_sysinfo.port, "24681");
|
||||
}
|
||||
|
||||
vlog("===============================================\n");
|
||||
vlog("===== Ventoy Plugson %s:%s =====\n", g_sysinfo.ip, g_sysinfo.port);
|
||||
vlog("===============================================\n");
|
||||
|
||||
|
||||
ventoy_disk_init();
|
||||
#ifndef VENTOY_SIM
|
||||
rc = ventoy_www_init();
|
||||
if (rc)
|
||||
{
|
||||
vlog("Failed to init www\n");
|
||||
MessageBoxW(NULL, g_msg_lang[MSGID_INTERNAL_ERR], g_msg_lang[MSGID_ERROR], MB_OK | MB_ICONERROR);
|
||||
|
||||
ventoy_disk_exit();
|
||||
ventoy_log_exit();
|
||||
return 1;
|
||||
}
|
||||
#endif
|
||||
ventoy_http_init();
|
||||
|
||||
g_hInst = hInstance;
|
||||
DialogBoxA(hInstance, MAKEINTRESOURCE(IDD_DIALOG1), NULL, DialogProc);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user