myvtm/spshell/svc.cpp

#include "precompile.h"

#include "sp_mod.h"
#include "svc.h"
#include "app.h"
#include "def.h"

#include "SpBase.h"
#include "sp_btr.h"
#include "sp_gui.h"
#include "sp_dbg_export.h"
#include "sp_iom.h"
#include "sp_svc.h"
#include "sp_def.h"
#include "sp_cfg.h"
#include "sp_rpc.h"

void on_bluesceen_display(sp_rpc_server_t *server, int epid, int svc_id, int call_type, iobuffer_t **info_pkt)
{
	app_t *app = get_app_instance();
	char *msg = NULL;
	//int iBlueScreen=0;
	//iobuffer_format_read(*info_pkt, "s4", &msg, &iBlueScreen);
	iobuffer_format_read(*info_pkt, "s", &msg);
	sp_gui_show_running_info(app->bsc_gui, msg, 1);
	toolkit_free(msg);
}

void on_fatal_error_display(sp_rpc_server_t *server, int epid, int svc_id, int call_type, iobuffer_t **info_pkt)
{
	app_t *app = get_app_instance();
	char *msg = NULL;
	//int iBlueScreen=0;
	//iobuffer_format_read(*info_pkt, "s4", &msg, &iBlueScreen);
	iobuffer_format_read(*info_pkt, "s", &msg);
	sp_gui_show_running_info(app->bsc_gui, msg, 2);
	toolkit_free(msg);
}

void on_startup_info_display(sp_rpc_server_t *server, int epid, int svc_id, int call_type, iobuffer_t **info_pkt)
{
	app_t *app = get_app_instance();
	char *msg = NULL;
	iobuffer_format_read(*info_pkt, "s", &msg);
	sp_gui_show_running_info(app->bsc_gui, msg, 3);
	toolkit_free(msg);
}


void on_bluesceen_undisplay(sp_rpc_server_t *server, int epid, int svc_id, int call_type, iobuffer_t **info_pkt)
{
	app_t *app = get_app_instance();
	sp_gui_undisplay(app->bsc_gui);
}

static BOOL RestartWindows()
{
	HANDLE hToken;              // handle to process token 
	TOKEN_PRIVILEGES tkp;       // pointer to token structure 
	BOOL fResult;               // system shutdown flag 

	// Get the current process token handle so we can get shutdown privilege. 
	if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken))
	{
		Dbg("get proc token fail");
		return FALSE;
	}

	// Get the LUID for shutdown privilege. 
	LookupPrivilegeValue(NULL, SE_SHUTDOWN_NAME, &tkp.Privileges[0].Luid);

	tkp.PrivilegeCount = 1;  // one privilege to set    
	tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;

	// Get shutdown privilege for this process. 
	AdjustTokenPrivileges(hToken, FALSE, &tkp, 0, (PTOKEN_PRIVILEGES)NULL, 0);

	// Cannot test the return value of AdjustTokenPrivileges. 
	if (GetLastError() != ERROR_SUCCESS)
	{
		Dbg("adjust proc token privilege fail");
		return FALSE;
	}

	// Display the shutdown dialog box and start the countdown. 
	fResult = InitiateSystemShutdown(
		NULL,    // shut down local computer 
		NULL,   // message for user
		0,			// time-out period, in seconds 
		FALSE,   // ask user to close apps 
		TRUE);   // reboot after shutdown 

	if (!fResult)
	{
		Dbg("request windows reboot fail");
		return FALSE;
	}

	// Disable shutdown privilege. 
	tkp.Privileges[0].Attributes = 0;
	AdjustTokenPrivileges(hToken, FALSE, &tkp, 0, (PTOKEN_PRIVILEGES)NULL, 0);
	return TRUE;
}

DWORD getExePath(char *exePath)
{
	char pathbuf[MAX_PATH] = "";
	int  pathlen = ::GetModuleFileName(NULL, pathbuf, MAX_PATH);

	// 替换掉单杠
	while (pathlen > 0)
	{
		if (pathbuf[pathlen--] == '\\')
			break;
	}
	pathbuf[++pathlen] = '\0';
	sprintf(exePath, "%s", pathbuf);
	return strlen(exePath);
}

int KickoffSpRestart(bool bRestart)
{
	STARTUPINFO si;
	PROCESS_INFORMATION pi;

	ZeroMemory(&si, sizeof(si));
	si.cb = sizeof(si);
	ZeroMemory(&pi, sizeof(pi));

	// LPTSTR szCmdline[] = _tcsdup(TEXT("\"C:\\Program Files\\MyApp\" -L -S"));
	//CreateProcess(NULL, szCmdline, /*...*/);

	// Start the child process.
	char exepath[MAX_PATH] = "";
	getExePath(exepath);

	CSimpleStringA csRestart, csVerPath, csAll, csSep("\""), csBlank(" "), csScript("wscript.exe"), csReFlag("r");
	csRestart = CSimpleStringA(exepath) + "\\sprestart.exe";

	sp_env_t *env = sp_get_env();
	csVerPath = env->dir->root_ver_path;
	csVerPath += "\\spexplorerfast.vbs";
	if (!bRestart)
		csReFlag = "n";
	
	csAll = csRestart + csBlank + csSep + csScript + csBlank + csVerPath + csSep
		+ csBlank + csSep + csReFlag + csSep;
	Dbg("allpath[%s]", (LPCTSTR)csAll);
	LPTSTR szCmdline = _strdup(csAll);
	if (!CreateProcess(NULL, szCmdline, NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi))
	{
		Dbg("CreateProcess failed (%d).\n", GetLastError());
		return -1;
	}
	
	Dbg("CreateProcess Success PID:%d.\n", pi.dwProcessId);
	DWORD dwErr = GetLastError();

	// Close process and thread handles. 
	CloseHandle(pi.hProcess);
	CloseHandle(pi.hThread);
	return 0;
}


void on_machine_reboot(sp_rpc_server_t *server, int epid, int svc_id, int call_type, iobuffer_t **info_pkt)
{
	int reason(0), way(0);
	sp_env_t *env = sp_get_env();
	sp_cfg_t *cfg = env->cfg;
	iobuffer_read(*info_pkt, IOBUF_T_I4, &reason, 0);
	iobuffer_read(*info_pkt, IOBUF_T_I4, &way, 0);
	Dbg("now reboot: reason = %d, way = %d", reason, way);

	CSimpleStringA strPath = env->dir->root_runinfo_path;
	strPath += "\\BootLog";	
	sp_btr_write_on_shutdown(strPath, env->btr_ctx, reason, way);

	RebootWayEnum eRebootWay = (RebootWayEnum)way;
	



	// 只处理框架重启及操作系统重启，电源重启由健康实体处理
	if (eRebootWay == RebootWay_Framework)
	{
		Dbg("reboot framework now");
		KickoffSpRestart(true);
	}
	else if (eRebootWay == RebootWay_OS)
	{
		Dbg("reboot windows now");
		RestartWindows();
	}
	app_t *app = get_app_instance();
	sp_iom_post_quit(app->iom);
}

void on_entity_quit(sp_rpc_server_t *server, int epid, int svc_id, int call_type, iobuffer_t **info_pkt)
{
	int entity_id;
	sp_env_t *env = sp_get_env();
	iobuffer_read(*info_pkt, IOBUF_T_I4, &entity_id, NULL);
	sp_mod_mgr_stop_entity(env->mod_mgr, entity_id, svc_id, CloseCause_Self);
}

void on_output_console_on(sp_rpc_server_t *server, int epid, int svc_id, int call_type, iobuffer_t **info_pkt)
{
	Dbg("set output console on");
	//sp_dbg_start_console("可视柜台终端运行监控");	

	auto app = get_app_instance();
	sp_gui_display(app->bsc_gui);
}

void on_output_console_off(sp_rpc_server_t *server, int epid, int svc_id, int call_type, iobuffer_t **info_pkt)
{
	Dbg("set output console off");
	//sp_dbg_close_console();

	auto app = get_app_instance();
	sp_gui_undisplay(app->bsc_gui);
}

iobuffer_t *on_entity_start(sp_rpc_server_t *server, int epid, int svc_id, int rpc_id, int call_type, iobuffer_t **req_pkt)
{
	sp_env_t *env = sp_get_env();
	int entity_id;
	int result;
	char *cmdline = NULL;
	iobuffer_t *ans_pkt = NULL;
	iobuffer_read(*req_pkt, IOBUF_T_I4, &entity_id, NULL);
	sp_dbg_debug("handle start entity req: %d", entity_id);

	iobuffer_format_read(*req_pkt, "s", &cmdline);
	result = sp_mod_mgr_start_entity(env->mod_mgr, entity_id, cmdline, svc_id);
	toolkit_free(cmdline);
	ans_pkt = iobuffer_create(-1, -1);
	iobuffer_write(ans_pkt, IOBUF_T_I4, &result, 0);
	return ans_pkt;
}

iobuffer_t *on_entity_stop(sp_rpc_server_t *server, int epid, int svc_id, int rpc_id, int call_type, iobuffer_t **req_pkt)
{
	sp_env_t *env = sp_get_env();
	int entity_id;
	int result;
	int cause_code;
	iobuffer_t *ans_pkt = NULL;
	iobuffer_read(*req_pkt, IOBUF_T_I4, &entity_id, NULL);
	if (svc_id == SP_SHELL_SVC_ID) {
		cause_code = CloseCause_ShellQuit;
	} else if (svc_id == SP_INVALID_SVC_ID) {
		cause_code = CloseCause_Lost;
	} else if (svc_id == entity_id) {
		cause_code = CloseCause_Self;
	} else {
		cause_code = CloseCause_Other;
	}

	sp_dbg_debug("handle stop entity req: %d", entity_id);
	result = sp_mod_mgr_stop_entity(env->mod_mgr, entity_id, svc_id, cause_code);
	ans_pkt = iobuffer_create(-1, -1);
	iobuffer_write(ans_pkt, IOBUF_T_I4, &result, 0);
	return ans_pkt;
}

iobuffer_t *on_entity_continue(sp_rpc_server_t *server, int epid, int svc_id, int rpc_id, int call_type, iobuffer_t **req_pkt)
{
	sp_env_t *env = sp_get_env();
	int entity_id;
	int result;
	iobuffer_t *ans_pkt = NULL;
	iobuffer_read(*req_pkt, IOBUF_T_I4, &entity_id, NULL);

	sp_dbg_debug("handle continue entity req: %d", entity_id);
	result = sp_mod_mgr_continue_entity(env->mod_mgr, entity_id, svc_id);
	ans_pkt = iobuffer_create(-1, -1);
	iobuffer_write(ans_pkt, IOBUF_T_I4, &result, 0);
	return ans_pkt;
}

iobuffer_t *on_entity_pause(sp_rpc_server_t *server, int epid, int svc_id, int rpc_id, int call_type, iobuffer_t **req_pkt)
{
	sp_env_t *env = sp_get_env();
	int entity_id;
	int result;
	iobuffer_t *ans_pkt = NULL;
	iobuffer_read(*req_pkt, IOBUF_T_I4, &entity_id, NULL);

	sp_dbg_debug("handle pause entity req: %d", entity_id);
	result = sp_mod_mgr_pause_entity(env->mod_mgr, entity_id, svc_id);
	ans_pkt = iobuffer_create(-1, -1);
	iobuffer_write(ans_pkt, IOBUF_T_I4, &result, 0);
	return ans_pkt;
}

iobuffer_t *on_entity_test(sp_rpc_server_t *server, int epid, int svc_id, int rpc_id, int call_type, iobuffer_t **req_pkt)
{
	sp_env_t *env = sp_get_env();
	int entity_id;
	int test_type;
	int result;
	iobuffer_t *ans_pkt = NULL;
	iobuffer_read(*req_pkt, IOBUF_T_I4, &entity_id, NULL);
	iobuffer_read(*req_pkt, IOBUF_T_I4, &test_type, NULL);

	//sp_dbg_debug("handle test entity req: %d", entity_id);
	result = sp_mod_mgr_test_entity(env->mod_mgr, entity_id, svc_id, test_type);
	ans_pkt = iobuffer_create(-1, -1);
	iobuffer_write(ans_pkt, IOBUF_T_I4, &result, 0);
	return ans_pkt;
}

iobuffer_t *on_entity_termination(sp_rpc_server_t *server, int epid, int svc_id, int rpc_id, int call_type, iobuffer_t **req_pkt)
{
	sp_env_t *env = sp_get_env();
	int entity_id;
	int result;
	iobuffer_t *ans_pkt = NULL;
	iobuffer_read(*req_pkt, IOBUF_T_I4, &entity_id, NULL);

	sp_dbg_debug("handle terminate entity req: %d", entity_id);
	result = sp_mod_mgr_terminate_entity(env->mod_mgr, entity_id, svc_id);
	ans_pkt = iobuffer_create(-1, -1);
	iobuffer_write(ans_pkt, IOBUF_T_I4, &result, 0);
	return ans_pkt;
}

iobuffer_t *on_get_bcast_receiver(sp_rpc_server_t *server, int epid, int svc_id, int rpc_id, int call_type, iobuffer_t **req_pkt)
{
	sp_env_t *env = sp_get_env();
	iobuffer_t *ans_pkt = NULL;
	int result = 0;
	int i;
	array_header_t *arr = sp_bcm_daemon_get_receiver(get_app_instance()->bcm_daemon, svc_id);
	ans_pkt = iobuffer_create(-1, -1);
	iobuffer_write(ans_pkt, IOBUF_T_I4, &result, 0);
	iobuffer_write(ans_pkt, IOBUF_T_I4, &arr->nelts, 0);
	for (i = 0; i < arr->nelts; ++i) {
		sp_bcm_receiver_t *recver = ARRAY_IDX(arr, i, sp_bcm_receiver_t*);
		iobuffer_write(ans_pkt, IOBUF_T_I8, &recver->id, 0);
		iobuffer_write(ans_pkt, IOBUF_T_I4, &recver->receiver_id, 0);
		iobuffer_write(ans_pkt, IOBUF_T_STR, &recver->param, -1);
	}
	sp_bcm_daemon_free_receiver_array(arr);
	return ans_pkt;
}