teleport/client/tp_assist_win/tp_assist.cpp

#include "stdafx.h"
#include <shellapi.h>
#include <fcntl.h>
#include "resource.h"
#include "dlg_main.h"
#include "ts_http_rpc.h"
#include "ts_ws_client.h"

#ifdef _DEBUG
// #	include <vld.h>
#endif

#pragma comment(lib, "shlwapi.lib")

#define WMU_INSTANCE_EXIT		(WM_USER + 2)
#define WMU_SHOW_EXIST_DLGUI	(WM_USER + 3)

static ExLogger g_ex_logger;

static ATOM MyRegisterClass();
static BOOL InitInstance();
static LRESULT CALLBACK WndProc(HWND, UINT, WPARAM, LPARAM);
void failed(const char* msg);
void failed(const wchar_t* msg);

static HANDLE g_SingleInstanceMutexHandle = NULL;

HINSTANCE g_hInstance = NULL;
ULONG g_ulSingleInstanceMsgId = 0;
static TCHAR szKernalName[MAX_PATH] = { 0 };

HWND g_hwndBase = NULL;
int g_argc = 0;
wchar_t** g_argv = NULL;

ex_astr g_url_protocol;

#define TP_ASSIST_GUID			_T("A6EFE1250C5F4416BFA819FE92CBD4B4")
#define TP_ASSIST_INSTANCE		_T("TS_ASSIST_SINGLE_INSTANCE")
#define TP_ASSIST_WIN_CLASS		_T("TS_ASSIST_WINDOW_CLASS")
#define MAKEDWORD(low, high)	((DWORD)(((WORD)(((DWORD_PTR)(low)) & 0xffff)) | ((DWORD)((WORD)(((DWORD_PTR)(high)) & 0xffff))) << 16))

DWORD WINAPI HttpServerThreadProc(LPVOID lpParam) {
    // http_rpc_main_loop();

	if (!g_http_interface.init()) {
		EXLOGE("[ERROR] can not start HTTP-RPC listener, maybe port is already in use.\n");
		return 0;
	}

	EXLOGW("======================================================\n");
	EXLOGW("[rpc] TeleportAssist-HTTP-RPC ready on 127.0.0.1:%d\n", g_http_interface.get_port());

	g_http_interface.run();

	EXLOGW("[rpc] HTTP-Server main loop end.\n");
    return 0;
}

//DWORD WINAPI HttpsServerThreadProc(LPVOID lpParam) {
//    http_rpc_main_loop(true);
//    return 0;
//}

int APIENTRY wWinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPTSTR lpCmdLine, int nShowCmd) {
	EXLOG_USE_LOGGER(&g_ex_logger);

	WORD wVersionRequested;
	WSADATA wsaData;
	int err;

	wVersionRequested = MAKEWORD(1, 1);

	err = WSAStartup(wVersionRequested, &wsaData);
	if (err != 0) {
		return 0;
	}

	if (LOBYTE(wsaData.wVersion) != 1 ||
		HIBYTE(wsaData.wVersion) != 1) {
		WSACleanup();
		return 0;
	}

	g_env.init();

#ifdef EX_DEBUG
	EXLOG_LEVEL(EX_LOG_LEVEL_DEBUG);
#else
	EXLOG_LEVEL(EX_LOG_LEVEL_VERBOSE);
#endif

	EXLOG_FILE(L"tp_assist.log", g_env.m_log_path.c_str(), EX_LOG_FILE_MAX_SIZE, EX_LOG_FILE_MAX_COUNT);

	if (!g_cfg.init()) {
		MessageBox(NULL, _T("<EFBFBD>޷<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ļ<EFBFBD><EFBFBD><EFBFBD>"), _T("<EFBFBD><EFBFBD><EFBFBD><EFBFBD>"), MB_OK | MB_ICONERROR);
		return FALSE;
	}

	g_hInstance = hInstance;
	_stprintf_s(szKernalName, MAX_PATH, _T("%s_%s"), TP_ASSIST_GUID, TP_ASSIST_INSTANCE);
	g_ulSingleInstanceMsgId = RegisterWindowMessage(szKernalName);
	if (0 == g_ulSingleInstanceMsgId)
		return FALSE;

	LPWSTR szCmdLine = (LPWSTR)::GetCommandLineW(); //<2F><>ȡ<EFBFBD><C8A1><EFBFBD><EFBFBD><EFBFBD>в<EFBFBD><D0B2><EFBFBD><EFBFBD><EFBFBD>
	g_argv = CommandLineToArgvW(szCmdLine, &g_argc); //<2F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>в<EFBFBD><D0B2><EFBFBD><EFBFBD>ַ<EFBFBD><D6B7><EFBFBD><EFBFBD><EFBFBD>

	for (int i = 0; i < g_argc; ++i) {
		ex_wstr arg = g_argv[i];
		if (0 == lstrcmp(g_argv[i], _T("--stop"))) {
			PostMessage(HWND_BROADCAST, g_ulSingleInstanceMsgId, WMU_INSTANCE_EXIT, 0);
			LocalFree(g_argv);
			g_argv = NULL;
			return -1;
		}
		else if (arg.find(L"teleport://", 0) == 0) {
			// url-protocol
			// teleport://register?param={"ws_url":"ws://127.0.0.1:7190/ws/assist/","assist_id":1234,"session_id":"tp_5678"}

			EXLOGV(L"url-protocol: %s\n", arg.c_str());
			ex_wstr2astr(arg, g_url_protocol);

			break;
		}
	}

	// make sure run single instance.
	_stprintf_s(szKernalName, MAX_PATH, _T("%s_%s"), TP_ASSIST_GUID, TP_ASSIST_INSTANCE);
	g_SingleInstanceMutexHandle = CreateMutex(NULL, FALSE, szKernalName);
	if (GetLastError() == ERROR_ALREADY_EXISTS) {
		// if we got url-protocol, send it by WM_COPYDATA, else just bring running instance to front.
		if (g_url_protocol.empty())
		{
			PostMessage(HWND_BROADCAST, g_ulSingleInstanceMsgId, WMU_SHOW_EXIST_DLGUI, 0);
		}
		else
		{
			HWND hwnd = FindWindow(TP_ASSIST_WIN_CLASS, NULL);
			if (hwnd == NULL) {
				MessageBoxW(NULL, _T("<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ѿ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ˣ<EFBFBD><EFBFBD><EFBFBD><EFBFBD>޷<EFBFBD><EFBFBD><EFBFBD>λ<EFBFBD><EFBFBD>"), _T("<EFBFBD><EFBFBD><EFBFBD><EFBFBD>"), MB_OK | MB_ICONERROR);
			}
			else
			{
				COPYDATASTRUCT data;
				data.dwData = NULL;
				data.cbData = g_url_protocol.length() + 1;  // include zero end.
				data.lpData = (void*)g_url_protocol.c_str();
				SendMessage(hwnd, WM_COPYDATA, NULL, (LPARAM)&data);
			}
		}

		CloseHandle(g_SingleInstanceMutexHandle);
		LocalFree(g_argv);
		g_argv = NULL;
		return 0;
	}


	// create dialog-box window.
	MyRegisterClass();

	// Perform application initialization:
	if (!InitInstance()) {
		CloseHandle(g_SingleInstanceMutexHandle);
		LocalFree(g_argv);
		g_argv = NULL;
		return FALSE;
	}

    HANDLE hThreadHttpServer=NULL;
    DWORD dwThreadId=0;
    hThreadHttpServer=CreateThread(NULL, 0, HttpServerThreadProc, NULL, 0, &dwThreadId);

	//HANDLE hThreadHttpsServer=NULL;
	//dwThreadId=0;
	//hThreadHttpsServer=CreateThread(NULL, 0, HttpsServerThreadProc, NULL, 0, &dwThreadId);

	MSG msg;
	while (GetMessage(&msg, NULL, 0, 0)) {
		TranslateMessage(&msg);
		DispatchMessage(&msg);
	}

	g_http_interface.stop();
    WaitForSingleObject(hThreadHttpServer, INFINITE);

	//http_rpc_stop(true);
	//WaitForSingleObject(hThreadHttpsServer, INFINITE);

	CloseHandle(g_SingleInstanceMutexHandle);

	LocalFree(g_argv);
	g_argv = NULL;
	return 0;
}


void failed(const char* msg) {
	OutputDebugStringA(msg);
}
void failed(const wchar_t* msg) {
	OutputDebugStringW(msg);
}

ATOM MyRegisterClass() {
	WNDCLASSEX wcex;

	wcex.cbSize = sizeof(WNDCLASSEX);

	wcex.style = CS_HREDRAW | CS_VREDRAW;
	wcex.lpfnWndProc = WndProc;
	wcex.cbClsExtra = 0;
	wcex.cbWndExtra = 0;
	wcex.hInstance = g_hInstance;
	wcex.hIcon = LoadIcon(g_hInstance, MAKEINTRESOURCE(IDI_NORMAL_BIG));
	wcex.hCursor = LoadCursor(NULL, IDC_ARROW);
	wcex.hbrBackground = (HBRUSH)(COLOR_WINDOW + 1);
	wcex.lpszMenuName = MAKEINTRESOURCE(IDR_ASSIST);
	wcex.lpszClassName = TP_ASSIST_WIN_CLASS;
	wcex.hIconSm = LoadIcon(wcex.hInstance, MAKEINTRESOURCE(IDI_NORMAL_SMALL));

	return RegisterClassEx(&wcex);
}

BOOL InitInstance(void) {
	g_hwndBase = CreateWindow(TP_ASSIST_WIN_CLASS, _T(""), WS_OVERLAPPEDWINDOW, 8, 0, 8, 0, NULL, NULL, g_hInstance, NULL);
	if (!g_hwndBase)
		return FALSE;

	ShowWindow(g_hwndBase, SW_HIDE);

	return TRUE;
}

LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam) {
	if (g_ulSingleInstanceMsgId == message) {
		if (WMU_INSTANCE_EXIT == wParam) {
			PostMessage(g_hDlgMain, WM_COMMAND, MAKEDWORD(IDCANCEL, 0), NULL);
			return 0;
		}
		else if (WMU_SHOW_EXIST_DLGUI == wParam) {
			ShowWindow(g_hDlgMain, SW_SHOW);
			SetWindowPos(g_hDlgMain, HWND_TOPMOST, 0, 0, 0, 0, SWP_NOSIZE | SWP_NOMOVE);
			SetActiveWindow(g_hDlgMain);
			BringWindowToTop(g_hDlgMain);
			SetWindowPos(g_hDlgMain, HWND_NOTOPMOST, 0, 0, 0, 0, SWP_NOSIZE | SWP_NOMOVE);
			return 0;
		}
	}

	switch (message) {
	case WM_CREATE:
		PostMessage(hWnd, WM_COMMAND, MAKEDWORD(IDM_MAIN, 0), NULL);

		if (!g_url_protocol.empty())
		{
			g_ws_client.url_scheme_handler(g_url_protocol);
		}

		return DefWindowProc(hWnd, message, wParam, lParam);
		break;
	case WM_COMMAND:
		if (IDM_MAIN == LOWORD(wParam)) {
			CreateDialog(g_hInstance, MAKEINTRESOURCE(IDD_DLG_MAIN), hWnd, tpDlgMainProc);
			ShowWindow(g_hDlgMain, SW_HIDE);
		}
		break;
	case WM_DESTROY:
		g_ws_client.stop_all_client();
		SendMessage(g_hDlgMain, WMU_DLG_MAIN_EXIT, NULL, NULL);
		PostQuitMessage(0);
		break;
	case WM_COPYDATA:
	{
		COPYDATASTRUCT* data = (COPYDATASTRUCT*)lParam;
		ex_astr url_protocol((char*)data->lpData);
		// MessageBoxA(hWnd, url_protocol.c_str(), "url-protocol", MB_OK);
		g_ws_client.url_scheme_handler(url_protocol);
		break;
	}
	default:
		return DefWindowProc(hWnd, message, wParam, lParam);
	}
	return 0;
}