MeshCentral/MeshAgent
1
0
Fork 0
mirror of https://github.com/Ylianst/MeshAgent synced 2025-12-16 00:03:45 +00:00
Code Issues Releases Wiki Activity

Updated new windows self-update to support UTF8

Browse Source
This commit is contained in:
Bryan Roe
2021-01-14 14:58:14 -08:00
parent 4eb80a4f22
commit 3a6e1f036c
1 changed files with 16 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
meshcore/agentcore.c
View File

@@ -2606,19 +2606,27 @@ void MeshServer_selfupdate_continue(MeshAgentHostContainer *agent)
}
else
{
char parms[65535];
WCHAR w_meshservicename[4096] = { 0 };
WCHAR w_updatefile[4096] = { 0 };
WCHAR w_exepath[4096] = { 0 };
WCHAR parms[65535] = { 0 };
char *updatefile = MeshAgent_MakeAbsolutePathEx(agent->exePath, ".update.exe", 0);
char cmd[MAX_PATH];
char env[MAX_PATH];
WCHAR cmd[MAX_PATH] = { 0 };
WCHAR env[MAX_PATH] = { 0 };
size_t envlen = sizeof(env);
if (getenv_s(&envlen, env, sizeof(env), "windir") == 0)
if (_wgetenv_s(&envlen, env, MAX_PATH, L"windir") == 0)
{
sprintf_s(cmd, sizeof(cmd), "%s\\system32\\cmd.exe", env);
sprintf_s(parms, sizeof(parms), "/C wmic service \"%s\" call stopservice && copy \"%s\" \"%s\" && wmic service \"%s\" call startservice && erase \"%s\"",
agent->meshServiceName, updatefile, agent->exePath, agent->meshServiceName, updatefile);
ILibUTF8ToWideEx(agent->meshServiceName, (int)strnlen_s(agent->meshServiceName, 255), w_meshservicename, 4096);
ILibUTF8ToWideEx(updatefile, (int)strnlen_s(updatefile, 4096), w_updatefile, 4096);
ILibUTF8ToWideEx(agent->exePath, (int)strnlen_s(agent->exePath, 4096), w_exepath, 4096);
swprintf_s(cmd, MAX_PATH, L"%s\\system32\\cmd.exe", env);
swprintf_s(parms, 65535, L"/C wmic service \"%s\" call stopservice && copy \"%s\" \"%s\" && wmic service \"%s\" call startservice && erase \"%s\"",
w_meshservicename, w_updatefile, w_exepath, w_meshservicename, w_updatefile);
ILIBLOGMESSAGEX("SelfUpdate -> Updating and restarting service...");
_execve(cmd, (char*[]) { "cmd", parms, NULL }, NULL);
_wexecve(cmd, (WCHAR*[]) { L"cmd", parms, NULL }, NULL);
}
ILIBLOGMESSAGEX("SelfUpdate -> FAILED");
return;