1 ///////////////////////////////////////////////////////////////////////////////
2 // LameXP - Audio Encoder Front-End
3 // Copyright (C) 2004-2012 LoRd_MuldeR <MuldeR2@GMX.de>
5 // This program is free software; you can redistribute it and/or modify
6 // it under the terms of the GNU General Public License as published by
7 // the Free Software Foundation; either version 2 of the License, or
8 // (at your option) any later version.
10 // This program is distributed in the hope that it will be useful,
11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 // GNU General Public License for more details.
15 // You should have received a copy of the GNU General Public License along
16 // with this program; if not, write to the Free Software Foundation, Inc.,
17 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
19 // http://www.gnu.org/licenses/gpl-2.0.txt
20 ///////////////////////////////////////////////////////////////////////////////
22 #include "Tool_Abstract.h"
28 #include <QMutexLocker>
30 #include <QProcessEnvironment>
34 * Win32 API definitions
36 typedef HANDLE (WINAPI *CreateJobObjectFun)(__in_opt LPSECURITY_ATTRIBUTES lpJobAttributes, __in_opt LPCSTR lpName);
37 typedef BOOL (WINAPI *SetInformationJobObjectFun)(__in HANDLE hJob, __in JOBOBJECTINFOCLASS JobObjectInformationClass, __in_bcount(cbJobObjectInformationLength) LPVOID lpJobObjectInformation, __in DWORD cbJobObjectInformationLength);
38 typedef BOOL (WINAPI *AssignProcessToJobObjectFun)(__in HANDLE hJob, __in HANDLE hProcess);
43 quint64 AbstractTool::m_lastLaunchTime = 0ui64;
44 QMutex *AbstractTool::m_mutex_startProcess = NULL;
45 HANDLE AbstractTool::m_handle_jobObject = NULL;
46 unsigned int AbstractTool::m_jobObjRefCount = 0U;
51 static const DWORD START_DELAY = 333; //in milliseconds
52 static const quint64 START_DELAY_NANO = START_DELAY * 1000 * 10; //in 100-nanosecond intervals
57 AbstractTool::AbstractTool(void)
59 static CreateJobObjectFun CreateJobObjectPtr = NULL;
60 static SetInformationJobObjectFun SetInformationJobObjectPtr = NULL;
62 if(!m_mutex_startProcess)
64 m_mutex_startProcess = new QMutex();
67 QMutexLocker lock(m_mutex_startProcess);
69 if(m_jobObjRefCount < 1U)
71 if(!CreateJobObjectPtr || !SetInformationJobObjectPtr)
73 QLibrary Kernel32Lib("kernel32.dll");
74 CreateJobObjectPtr = (CreateJobObjectFun) Kernel32Lib.resolve("CreateJobObjectA");
75 SetInformationJobObjectPtr = (SetInformationJobObjectFun) Kernel32Lib.resolve("SetInformationJobObject");
77 if(CreateJobObjectPtr && SetInformationJobObjectPtr)
79 HANDLE jobObject = CreateJobObjectPtr(NULL, NULL);
80 if((jobObject != NULL) && (jobObject != INVALID_HANDLE_VALUE))
82 JOBOBJECT_EXTENDED_LIMIT_INFORMATION jobExtendedLimitInfo;
83 memset(&jobExtendedLimitInfo, 0, sizeof(JOBOBJECT_EXTENDED_LIMIT_INFORMATION));
84 memset(&jobExtendedLimitInfo.BasicLimitInformation, 0, sizeof(JOBOBJECT_BASIC_LIMIT_INFORMATION));
85 jobExtendedLimitInfo.BasicLimitInformation.LimitFlags = JOB_OBJECT_LIMIT_KILL_ON_JOB_CLOSE | JOB_OBJECT_LIMIT_DIE_ON_UNHANDLED_EXCEPTION;
86 if(SetInformationJobObjectPtr(jobObject, JobObjectExtendedLimitInformation, &jobExtendedLimitInfo, sizeof(JOBOBJECT_EXTENDED_LIMIT_INFORMATION)))
88 m_handle_jobObject = jobObject;
89 m_jobObjRefCount = 1U;
93 qWarning("Failed to set job object information!");
94 CloseHandle(jobObject);
104 m_firstLaunch = true;
110 AbstractTool::~AbstractTool(void)
112 QMutexLocker lock(m_mutex_startProcess);
114 if(m_jobObjRefCount >= 1U)
117 if((m_jobObjRefCount < 1U) && m_handle_jobObject)
119 CloseHandle(m_handle_jobObject);
120 m_handle_jobObject = NULL;
126 * Initialize and launch process object
128 bool AbstractTool::startProcess(QProcess &process, const QString &program, const QStringList &args)
130 static AssignProcessToJobObjectFun AssignProcessToJobObjectPtr = NULL;
132 QMutexLocker lock(m_mutex_startProcess);
134 if(currentTime() <= m_lastLaunchTime)
139 emit messageLogged(commandline2string(program, args) + "\n");
141 QProcessEnvironment env = process.processEnvironment();
142 if(env.isEmpty()) env = QProcessEnvironment::systemEnvironment();
143 env.insert("TEMP", QDir::toNativeSeparators(lamexp_temp_folder2()));
144 env.insert("TMP", QDir::toNativeSeparators(lamexp_temp_folder2()));
145 process.setProcessEnvironment(env);
147 if(!AssignProcessToJobObjectPtr)
149 QLibrary Kernel32Lib("kernel32.dll");
150 AssignProcessToJobObjectPtr = (AssignProcessToJobObjectFun) Kernel32Lib.resolve("AssignProcessToJobObject");
153 process.setProcessChannelMode(QProcess::MergedChannels);
154 process.setReadChannel(QProcess::StandardOutput);
155 process.start(program, args);
157 if(process.waitForStarted())
159 if(AssignProcessToJobObjectPtr && m_handle_jobObject)
161 if(!AssignProcessToJobObjectPtr(m_handle_jobObject, process.pid()->hProcess))
163 qWarning("Failed to assign process to job object!");
166 if(!SetPriorityClass(process.pid()->hProcess, BELOW_NORMAL_PRIORITY_CLASS))
168 SetPriorityClass(process.pid()->hProcess, IDLE_PRIORITY_CLASS);
175 emit statusUpdated(0);
176 m_firstLaunch = false;
179 m_lastLaunchTime = currentTime() + START_DELAY_NANO;
183 emit messageLogged("Process creation has failed :-(");
184 QString errorMsg= process.errorString().trimmed();
185 if(!errorMsg.isEmpty()) emit messageLogged(errorMsg);
188 process.waitForFinished(-1);
190 m_lastLaunchTime = currentTime() + START_DELAY_NANO;
195 * Convert program arguments to single string
197 QString AbstractTool::commandline2string(const QString &program, const QStringList &arguments)
199 QString commandline = (program.contains(' ') ? QString("\"%1\"").arg(program) : program);
201 for(int i = 0; i < arguments.count(); i++)
203 commandline += (arguments.at(i).contains(' ') ? QString(" \"%1\"").arg(arguments.at(i)) : QString(" %1").arg(arguments.at(i)));
210 * Convert long path to short path
212 QString AbstractTool::pathToShort(const QString &longPath)
215 DWORD buffSize = GetShortPathNameW(reinterpret_cast<const wchar_t*>(longPath.utf16()), NULL, NULL);
219 wchar_t *buffer = new wchar_t[buffSize];
220 DWORD result = GetShortPathNameW(reinterpret_cast<const wchar_t*>(longPath.utf16()), buffer, buffSize);
222 if(result > 0 && result < buffSize)
224 shortPath = QString::fromUtf16(reinterpret_cast<const unsigned short*>(buffer));
230 return (shortPath.isEmpty() ? longPath : shortPath);
233 const quint64 AbstractTool::currentTime(void)
236 GetSystemTimeAsFileTime(&fileTime);
239 temp.HighPart = fileTime.dwHighDateTime;
240 temp.LowPart = fileTime.dwLowDateTime;
242 return temp.QuadPart;