From 981f8b3acf7e76f27c02f4ced80d55b424cc7497 Mon Sep 17 00:00:00 2001 From: Tannin Date: Sun, 3 Feb 2013 12:49:25 +0100 Subject: initial commit to mercurial repository. Corresponds to MO version 0.12.6 --- src/helper.cpp | 118 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 118 insertions(+) create mode 100644 src/helper.cpp (limited to 'src/helper.cpp') diff --git a/src/helper.cpp b/src/helper.cpp new file mode 100644 index 00000000..55b21a4b --- /dev/null +++ b/src/helper.cpp @@ -0,0 +1,118 @@ +/* +Copyright (C) 2012 Sebastian Herbord. All rights reserved. + +This file is part of Mod Organizer. + +Mod Organizer is free software: you can redistribute it and/or modify +it under the terms of the GNU General Public License as published by +the Free Software Foundation, either version 3 of the License, or +(at your option) any later version. + +Mod Organizer is distributed in the hope that it will be useful, +but WITHOUT ANY WARRANTY; without even the implied warranty of +MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +GNU General Public License for more details. + +You should have received a copy of the GNU General Public License +along with Mod Organizer. If not, see . +*/ + +#include "helper.h" +#include "report.h" +#include "utility.h" +#include + +#define WIN32_LEAN_AND_MEAN +#include + +#include + + +namespace Helper { + + +bool helperExec(LPCWSTR moDirectory, LPCWSTR commandLine) +{ + wchar_t fileName[MAX_PATH]; + _snwprintf(fileName, MAX_PATH, L"%ls\\helper.exe", moDirectory); + + SHELLEXECUTEINFOW execInfo = {0}; + + execInfo.cbSize = sizeof(SHELLEXECUTEINFOW); + execInfo.fMask = SEE_MASK_NOCLOSEPROCESS; + execInfo.hwnd = NULL; + execInfo.lpVerb = L"runas"; + execInfo.lpFile = fileName; + execInfo.lpParameters = commandLine; + execInfo.lpDirectory = moDirectory; + execInfo.nShow = SW_SHOW; + + ::ShellExecuteExW(&execInfo); + + if (::WaitForSingleObject(execInfo.hProcess, INFINITE) != WAIT_OBJECT_0) { + reportError(QObject::tr("helper failed")); + return false; + } + + DWORD exitCode; + GetExitCodeProcess(execInfo.hProcess, &exitCode); + return exitCode == NOERROR; +} + + +bool init(const std::wstring &moDirectory) +{ + DWORD userNameLen = UNLEN + 1; + wchar_t userName[UNLEN + 1]; + + if (!GetUserName(userName, &userNameLen)) { + reportError(QObject::tr("failed to determine account name")); + return false; + } + wchar_t *commandLine = new wchar_t[32768]; + + _snwprintf(commandLine, 32768, L"init \"%ls\" \"%ls\"", + moDirectory.c_str(), userName); + + bool res = helperExec(moDirectory.c_str(), commandLine); + delete [] commandLine; + + return res; +} + + +bool setNXMHandler(const std::wstring &moDirectory, bool enable) +{ + DWORD userNameLen = UNLEN + 1; + wchar_t userName[UNLEN + 1]; + + if (!GetUserName(userName, &userNameLen)) { + reportError(QObject::tr("failed to determine account name")); + return false; + } + + wchar_t *commandLine = new wchar_t[32768]; + _snwprintf(commandLine, 32768, L"handleNXM \"%ls\" %d", + moDirectory.c_str(), enable ? 1 : 0); + + bool res = helperExec(moDirectory.c_str(), commandLine); + delete [] commandLine; + + return res; +} + + +bool backdateBSAs(const std::wstring &moDirectory, const std::wstring &dataPath) +{ + wchar_t *commandLine = new wchar_t[32768]; + _snwprintf(commandLine, 32768, L"backdateBSA \"%ls\"", + dataPath.c_str()); + + bool res = helperExec(moDirectory.c_str(), commandLine); + delete [] commandLine; + + return res; +} + + +} // namespace -- cgit v1.3.1