forked from qt-creator/qt-creator
move qtcbuildhelper where it belongs and rename it accordingly
this is not a creator tool, but a helper of qtcprocess, just like the existing stub. Change-Id: Ice7c60a1dee0b132c6063747f82ea35699f4127d Reviewed-on: http://codereview.qt.nokia.com/2918 Reviewed-by: Qt Sanity Bot <qt_sanity_bot@ovi.com> Reviewed-by: Joerg Bornemann <joerg.bornemann@nokia.com>
This commit is contained in:
committed by
Joerg Bornemann
parent
9e20d6df04
commit
65317deaf9
192
src/libs/utils/process_ctrlc_stub.cpp
Normal file
192
src/libs/utils/process_ctrlc_stub.cpp
Normal file
@@ -0,0 +1,192 @@
|
||||
/**************************************************************************
|
||||
**
|
||||
** This file is part of Qt Creator
|
||||
**
|
||||
** Copyright (c) 2011 Nokia Corporation and/or its subsidiary(-ies).
|
||||
**
|
||||
** Contact: Nokia Corporation (info@qt.nokia.com)
|
||||
**
|
||||
**
|
||||
** GNU Lesser General Public License Usage
|
||||
**
|
||||
** This file may be used under the terms of the GNU Lesser General Public
|
||||
** License version 2.1 as published by the Free Software Foundation and
|
||||
** appearing in the file LICENSE.LGPL included in the packaging of this file.
|
||||
** Please review the following information to ensure the GNU Lesser General
|
||||
** Public License version 2.1 requirements will be met:
|
||||
** http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
|
||||
**
|
||||
** In addition, as a special exception, Nokia gives you certain additional
|
||||
** rights. These rights are described in the Nokia Qt LGPL Exception
|
||||
** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
|
||||
**
|
||||
** Other Usage
|
||||
**
|
||||
** Alternatively, this file may be used in accordance with the terms and
|
||||
** conditions contained in a signed written agreement between you and Nokia.
|
||||
**
|
||||
** If you have questions regarding the use of this file, please contact
|
||||
** Nokia at info@qt.nokia.com.
|
||||
**
|
||||
**************************************************************************/
|
||||
|
||||
/* A stub for Windows console processes (like nmake) that is able to terminate
|
||||
* its child process via a generated Ctrl-C event.
|
||||
* The termination is triggered by sending a custom message to the HWND of
|
||||
* this process. */
|
||||
|
||||
#ifndef WINVER
|
||||
#define WINVER 0x0501
|
||||
#endif
|
||||
|
||||
#ifndef _WIN32_WINNT
|
||||
#define _WIN32_WINNT 0x0501
|
||||
#endif
|
||||
|
||||
#define WIN32_LEAN_AND_MEAN
|
||||
#include <windows.h>
|
||||
#include <shellapi.h>
|
||||
#include <wchar.h>
|
||||
#include <cstdlib>
|
||||
#include <cstdio>
|
||||
|
||||
const wchar_t szTitle[] = L"qtcbuildhelper";
|
||||
const wchar_t szWindowClass[] = L"wcqtcbuildhelper";
|
||||
UINT uiShutDownWindowMessage;
|
||||
HWND hwndMain = 0;
|
||||
|
||||
LRESULT CALLBACK WndProc(HWND, UINT, WPARAM, LPARAM);
|
||||
BOOL WINAPI ctrlHandler(DWORD dwCtrlType);
|
||||
bool findFirst(const wchar_t *str, const size_t strLength, const size_t startPos, const wchar_t *chars, size_t &pos);
|
||||
bool startProcess(wchar_t pCommandLine[]);
|
||||
|
||||
int main(int argc, char **)
|
||||
{
|
||||
if (argc < 2) {
|
||||
fprintf(stderr, "This is an internal helper of Qt Creator. Do not run it manually.\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
SetConsoleCtrlHandler(ctrlHandler, TRUE);
|
||||
uiShutDownWindowMessage = RegisterWindowMessage(L"qtcbuildhelper_shutdown");
|
||||
|
||||
WNDCLASSEX wcex;
|
||||
ZeroMemory(&wcex, sizeof(wcex));
|
||||
wcex.cbSize = sizeof(wcex);
|
||||
wcex.lpfnWndProc = WndProc;
|
||||
wcex.hInstance = GetModuleHandle(0);
|
||||
wcex.lpszClassName = szWindowClass;
|
||||
if (!RegisterClassEx(&wcex))
|
||||
return 1;
|
||||
|
||||
hwndMain = CreateWindow(szWindowClass, szTitle, WS_DISABLED,
|
||||
0, 0, 0, 0, NULL, NULL, wcex.hInstance, NULL);
|
||||
if (!hwndMain)
|
||||
return FALSE;
|
||||
|
||||
// Get the command line and remove the call to this executable.
|
||||
// Note: We trust Qt Creator at this point to quote the call to this tool in a sensible way.
|
||||
// Strange things like C:\Q"t Crea"tor\bin\qtcbuildhelper.exe are not supported.
|
||||
wchar_t *strCommandLine = _wcsdup(GetCommandLine());
|
||||
const size_t strCommandLineLength = wcslen(strCommandLine);
|
||||
size_t pos = 1;
|
||||
if (strCommandLine[0] == L'"')
|
||||
if (!findFirst(strCommandLine, strCommandLineLength, pos, L"\"", pos))
|
||||
return 1;
|
||||
if (!findFirst(strCommandLine, strCommandLineLength, pos, L" \t", pos))
|
||||
return 1;
|
||||
#ifdef Q_CC_MSVC
|
||||
wmemmove_s(strCommandLine, strCommandLineLength, strCommandLine + pos + 1, strCommandLineLength - pos);
|
||||
#else
|
||||
wmemmove(strCommandLine, strCommandLine + pos + 1, strCommandLineLength - pos);
|
||||
#endif
|
||||
bool bSuccess = startProcess(strCommandLine);
|
||||
free(strCommandLine);
|
||||
|
||||
if (!bSuccess)
|
||||
return 1;
|
||||
|
||||
MSG msg;
|
||||
while (GetMessage(&msg, NULL, 0, 0))
|
||||
{
|
||||
TranslateMessage(&msg);
|
||||
DispatchMessage(&msg);
|
||||
}
|
||||
|
||||
return (int) msg.wParam;
|
||||
}
|
||||
|
||||
LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
|
||||
{
|
||||
if (message == uiShutDownWindowMessage) {
|
||||
GenerateConsoleCtrlEvent(CTRL_C_EVENT, 0);
|
||||
PostQuitMessage(0);
|
||||
return 0;
|
||||
}
|
||||
|
||||
switch (message)
|
||||
{
|
||||
case WM_DESTROY:
|
||||
PostQuitMessage(0);
|
||||
break;
|
||||
default:
|
||||
return DefWindowProc(hWnd, message, wParam, lParam);
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
bool findFirst(const wchar_t *str, const size_t strLength, const size_t startPos, const wchar_t *chars, size_t &pos)
|
||||
{
|
||||
for (size_t i=startPos; i < strLength; ++i) {
|
||||
for (size_t j=0; chars[j]; ++j) {
|
||||
if (str[i] == chars[j]) {
|
||||
pos = i;
|
||||
return true;
|
||||
}
|
||||
}
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
BOOL WINAPI ctrlHandler(DWORD /*dwCtrlType*/)
|
||||
{
|
||||
PostMessage(hwndMain, WM_DESTROY, 0, 0);
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
DWORD WINAPI processWatcherThread(LPVOID lpParameter)
|
||||
{
|
||||
HANDLE hProcess = reinterpret_cast<HANDLE>(lpParameter);
|
||||
WaitForSingleObject(hProcess, INFINITE);
|
||||
PostMessage(hwndMain, WM_DESTROY, 0, 0);
|
||||
return 0;
|
||||
}
|
||||
|
||||
bool startProcess(wchar_t *pCommandLine)
|
||||
{
|
||||
SECURITY_ATTRIBUTES sa;
|
||||
ZeroMemory(&sa, sizeof(sa));
|
||||
sa.nLength = sizeof(sa);
|
||||
sa.bInheritHandle = TRUE;
|
||||
|
||||
STARTUPINFO si;
|
||||
ZeroMemory(&si, sizeof(si));
|
||||
si.cb = sizeof(si);
|
||||
|
||||
PROCESS_INFORMATION pi;
|
||||
DWORD dwCreationFlags = 0;
|
||||
BOOL bSuccess = CreateProcess(NULL, pCommandLine, &sa, &sa, TRUE, dwCreationFlags, NULL, NULL, &si, &pi);
|
||||
if (!bSuccess) {
|
||||
fwprintf(stderr, L"qtcbuildhelper: Command line failed: %s\n", pCommandLine);
|
||||
return false;
|
||||
}
|
||||
|
||||
HANDLE hThread = CreateThread(NULL, 0, processWatcherThread, reinterpret_cast<void*>(pi.hProcess), 0, NULL);
|
||||
if (!hThread) {
|
||||
fwprintf(stderr, L"qtcbuildhelper: The watch dog thread cannot be started.\n");
|
||||
return false;
|
||||
}
|
||||
CloseHandle(hThread);
|
||||
return true;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user