mirror of
https://github.com/EnderIce2/rpc-bridge.git
synced 2025-05-25 22:14:38 +00:00
Add GUI creation if no command line arguments are provided
This commit is contained in:
EnderIce2
parent
aeda375cba
commit
e5a09024ff
GPG Key ID:
EACC3AD603BAB4DD
211
gui.c
Normal file
211
gui.c
Normal file
@ -0,0 +1,211 @@
|
||||
#include <windows.h>
|
||||
#include <winuser.h>
|
||||
#include <assert.h>
|
||||
#include <stdio.h>
|
||||
|
||||
/**
|
||||
* The entire code could be better written, but at least it works.
|
||||
*
|
||||
* This will make installation and removal of the bridge WAY easier.
|
||||
*/
|
||||
|
||||
LPTSTR GetErrorMessage();
|
||||
void print(char const *fmt, ...);
|
||||
void InstallService(int ServiceStartType, LPCSTR Path);
|
||||
void RemoveService();
|
||||
|
||||
VOID HandleStartButton(BOOL Silent)
|
||||
{
|
||||
SC_HANDLE hSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
|
||||
if (hSCManager == NULL)
|
||||
{
|
||||
print("OpenSCManager failed: %s\n", GetErrorMessage());
|
||||
return;
|
||||
}
|
||||
|
||||
SC_HANDLE schService = OpenService(hSCManager, "rpc-bridge",
|
||||
SERVICE_QUERY_CONFIG | SERVICE_CHANGE_CONFIG | SERVICE_START);
|
||||
if (schService == NULL)
|
||||
{
|
||||
print("OpenService failed: %s\n", GetErrorMessage());
|
||||
return;
|
||||
}
|
||||
|
||||
DWORD dwBytesNeeded;
|
||||
QueryServiceConfig(schService, NULL, 0, &dwBytesNeeded);
|
||||
LPQUERY_SERVICE_CONFIG lpqsc = (LPQUERY_SERVICE_CONFIG)LocalAlloc(LPTR, dwBytesNeeded);
|
||||
if (lpqsc == NULL)
|
||||
{
|
||||
print("LocalAlloc failed: %s\n", GetErrorMessage());
|
||||
return;
|
||||
}
|
||||
|
||||
if (!QueryServiceConfig(schService, lpqsc, dwBytesNeeded, &dwBytesNeeded))
|
||||
{
|
||||
print("QueryServiceConfig failed: %s\n", GetErrorMessage());
|
||||
return;
|
||||
}
|
||||
|
||||
if (StartService(schService, 0, NULL) == FALSE)
|
||||
{
|
||||
if (GetLastError() == ERROR_SERVICE_ALREADY_RUNNING)
|
||||
return;
|
||||
print("StartService failed: %s\n", GetErrorMessage());
|
||||
}
|
||||
|
||||
LocalFree(lpqsc);
|
||||
CloseServiceHandle(schService);
|
||||
CloseServiceHandle(hSCManager);
|
||||
if (Silent == FALSE)
|
||||
MessageBox(NULL, "Bridge service started successfully", "Info", MB_OK);
|
||||
}
|
||||
|
||||
VOID HandleInstallButton()
|
||||
{
|
||||
char filename[MAX_PATH];
|
||||
GetModuleFileName(NULL, filename, MAX_PATH);
|
||||
CopyFile(filename, "C:\\windows\\bridge.exe", FALSE);
|
||||
InstallService(SERVICE_AUTO_START, "C:\\windows\\bridge.exe --service");
|
||||
MessageBox(NULL, "Bridge installed successfully", "Info", MB_OK);
|
||||
HandleStartButton(TRUE);
|
||||
ExitProcess(0);
|
||||
}
|
||||
|
||||
VOID HandleRemoveButton()
|
||||
{
|
||||
RemoveService();
|
||||
MessageBox(NULL, "Bridge removed successfully", "Info", MB_OK);
|
||||
ExitProcess(0);
|
||||
}
|
||||
|
||||
LRESULT CALLBACK WndProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
|
||||
{
|
||||
switch (msg)
|
||||
{
|
||||
case WM_COMMAND:
|
||||
{
|
||||
switch (LOWORD(wParam))
|
||||
{
|
||||
case 1:
|
||||
HandleStartButton(FALSE);
|
||||
break;
|
||||
case 2:
|
||||
HandleInstallButton();
|
||||
break;
|
||||
case 3:
|
||||
HandleRemoveButton();
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
break;
|
||||
}
|
||||
case WM_CLOSE:
|
||||
DestroyWindow(hwnd);
|
||||
break;
|
||||
case WM_DESTROY:
|
||||
PostQuitMessage(0);
|
||||
ExitProcess(0);
|
||||
break;
|
||||
default:
|
||||
return DefWindowProc(hwnd, msg, wParam, lParam);
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance,
|
||||
LPSTR lpCmdLine, int nCmdShow)
|
||||
{
|
||||
SC_HANDLE hSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
|
||||
SC_HANDLE schService = OpenService(hSCManager, "rpc-bridge", SERVICE_START | SERVICE_QUERY_STATUS);
|
||||
|
||||
INT btnStartStyle = WS_CHILD | WS_VISIBLE | BS_PUSHBUTTON;
|
||||
INT btnRemoveStyle = WS_CHILD | WS_VISIBLE | BS_PUSHBUTTON;
|
||||
INT btnInstallStyle = WS_CHILD | WS_VISIBLE | BS_PUSHBUTTON;
|
||||
|
||||
if (schService != NULL)
|
||||
{
|
||||
btnInstallStyle |= WS_DISABLED;
|
||||
|
||||
SERVICE_STATUS_PROCESS ssStatus;
|
||||
DWORD dwBytesNeeded;
|
||||
assert(QueryServiceStatusEx(schService, SC_STATUS_PROCESS_INFO, (LPBYTE)&ssStatus,
|
||||
sizeof(SERVICE_STATUS_PROCESS), &dwBytesNeeded));
|
||||
|
||||
if (ssStatus.dwCurrentState == SERVICE_RUNNING ||
|
||||
ssStatus.dwCurrentState == SERVICE_START_PENDING)
|
||||
btnStartStyle |= WS_DISABLED;
|
||||
}
|
||||
else
|
||||
{
|
||||
btnStartStyle |= WS_DISABLED;
|
||||
btnRemoveStyle |= WS_DISABLED;
|
||||
}
|
||||
|
||||
CloseServiceHandle(schService);
|
||||
CloseServiceHandle(hSCManager);
|
||||
|
||||
const char szClassName[] = "BridgeWindowClass";
|
||||
|
||||
WNDCLASSEX wc;
|
||||
wc.cbSize = sizeof(WNDCLASSEX);
|
||||
wc.style = 0;
|
||||
wc.lpfnWndProc = WndProc;
|
||||
wc.cbClsExtra = 0;
|
||||
wc.cbWndExtra = 0;
|
||||
wc.hInstance = hInstance;
|
||||
wc.hIcon = LoadIcon(NULL, IDI_WINLOGO);
|
||||
wc.hCursor = LoadCursor(NULL, IDC_ARROW);
|
||||
wc.hbrBackground = (HBRUSH)(COLOR_WINDOW + 1);
|
||||
wc.lpszMenuName = NULL;
|
||||
wc.lpszClassName = szClassName;
|
||||
wc.hIconSm = LoadIcon(NULL, IDI_WINLOGO);
|
||||
|
||||
assert(RegisterClassEx(&wc));
|
||||
|
||||
HWND hwnd = CreateWindowEx(WS_EX_CLIENTEDGE,
|
||||
szClassName,
|
||||
"Discord RPC Bridge",
|
||||
WS_OVERLAPPEDWINDOW & ~WS_THICKFRAME,
|
||||
(GetSystemMetrics(SM_CXSCREEN) - 400) / 2,
|
||||
(GetSystemMetrics(SM_CYSCREEN) - 150) / 2,
|
||||
400, 150,
|
||||
NULL, NULL, hInstance, NULL);
|
||||
|
||||
CreateWindow("STATIC", "Do you want to start, install or remove the bridge?",
|
||||
WS_CHILD | WS_VISIBLE | SS_CENTER,
|
||||
0, 0, 400, 50,
|
||||
hwnd, NULL, hInstance, NULL);
|
||||
|
||||
CreateWindow("BUTTON", "Start",
|
||||
btnStartStyle,
|
||||
50, 50, 100, 30,
|
||||
hwnd, (HMENU)1, hInstance, NULL);
|
||||
|
||||
CreateWindow("BUTTON", "Install",
|
||||
btnInstallStyle,
|
||||
150, 50, 100, 30,
|
||||
hwnd, (HMENU)2, hInstance, NULL);
|
||||
|
||||
CreateWindow("BUTTON", "Remove",
|
||||
btnRemoveStyle,
|
||||
250, 50, 100, 30,
|
||||
hwnd, (HMENU)3, hInstance, NULL);
|
||||
|
||||
ShowWindow(hwnd, nCmdShow);
|
||||
UpdateWindow(hwnd);
|
||||
|
||||
MSG msg;
|
||||
while (GetMessage(&msg, NULL, 0, 0) > 0)
|
||||
{
|
||||
TranslateMessage(&msg);
|
||||
DispatchMessage(&msg);
|
||||
}
|
||||
return msg.wParam;
|
||||
}
|
||||
|
||||
void CreateGUI()
|
||||
{
|
||||
ShowWindow(GetConsoleWindow(), SW_MINIMIZE);
|
||||
ExitProcess(WinMain(GetModuleHandle(NULL), NULL, GetCommandLine(), SW_SHOWNORMAL));
|
||||
}
|
||||
33
main.c
33
main.c
@ -6,6 +6,7 @@
|
||||
FILE *g_logFile = NULL;
|
||||
BOOL RunningAsService = FALSE;
|
||||
|
||||
void CreateGUI();
|
||||
void CreateBridge();
|
||||
void LaunchGame(int argc, char **argv);
|
||||
void ServiceMain(int argc, char *argv[]);
|
||||
@ -261,17 +262,25 @@ void HandleArguments(int argc, char *argv[])
|
||||
else if (strcmp(argv[1], "--help") == 0)
|
||||
{
|
||||
printf("Usage:\n");
|
||||
printf(" %s [args]\n", argv[0]);
|
||||
printf(" --help - Show this help\n");
|
||||
printf(" --install - Install service\n");
|
||||
printf(" This will copy the binary to C:\\windows\\bridge.exe and register it as a service\n");
|
||||
printf(" --uninstall - Uninstall service\n");
|
||||
printf(" This will remove the service and delete C:\\windows\\bridge.exe\n");
|
||||
printf(" --steam - Reserved for Steam\n");
|
||||
printf(" This will start the service and exit (used with bridge.sh)\n");
|
||||
printf(" --no-service - Do not run as service\n");
|
||||
printf(" (only for --steam)\n");
|
||||
printf(" --service - Reserved for service\n");
|
||||
printf(" %s [args]\n\n", argv[0]);
|
||||
|
||||
printf("Arguments:\n");
|
||||
printf(" --help Show this help\n\n");
|
||||
|
||||
printf(" --install Install service\n");
|
||||
printf(" This will copy the binary to C:\\windows\\bridge.exe and register it as a service\n\n");
|
||||
printf(" --uninstall Uninstall service\n");
|
||||
|
||||
printf(" This will remove the service and delete C:\\windows\\bridge.exe\n\n");
|
||||
printf(" --steam Reserved for Steam\n");
|
||||
|
||||
printf(" This will start the service and exit (used with bridge.sh)\n\n");
|
||||
printf(" --no-service Do not run as service\n");
|
||||
|
||||
printf(" (only for --steam)\n\n");
|
||||
printf(" --service Reserved for service\n\n");
|
||||
|
||||
printf("Note: If no arguments are provided, the GUI will be shown instead\n");
|
||||
ExitProcess(0);
|
||||
}
|
||||
}
|
||||
@ -290,6 +299,8 @@ int main(int argc, char *argv[])
|
||||
|
||||
if (argc > 1)
|
||||
HandleArguments(argc, argv);
|
||||
else
|
||||
CreateGUI();
|
||||
|
||||
CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)CreateBridge,
|
||||
NULL, 0, NULL);
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user