khanat-code-old/code/nel/tools/misc/exec_timeout/exec_timeout.cpp

82 lines
2.1 KiB
C++

// NeL - MMORPG Framework <http://dev.ryzom.com/projects/nel/>
// Copyright (C) 2010 Winch Gate Property Limited
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as
// published by the Free Software Foundation, either version 3 of the
// License, or (at your option) any later version.
//
// This program 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 Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
#include "nel/misc/debug.h"
#include <windows.h>
#include <string>
using namespace std;
int main(int argc, char* argv[])
{
// Help ?
if (argc<3)
{
printf ("exec_timeout [timeout(ms)] [prog.exe] [arg0] [arg1]...\nReturn 0:error, 1:ok, 2:timeout\n");
return 0;
}
else
{
// String
string argu;
// Build the argus
for (int i=2; i<argc; i++)
{
argu += argv[i];
if (i<argc-1)
argu += " ";
}
// Process information
PROCESS_INFORMATION process_info;
// Timeout
DWORD timeout = atoi (argv[1]);
// Startup info
STARTUPINFO startupInfo;
GetStartupInfo (&startupInfo);
// Exec
if (CreateProcess( NULL, (char*)(argu.c_str()), NULL, NULL, TRUE, CREATE_NEW_PROCESS_GROUP, NULL, NULL, &startupInfo, &process_info))
{
// Return message
switch (WaitForSingleObject ( process_info.hProcess, timeout))
{
case WAIT_OBJECT_0:
return 1;
case WAIT_ABANDONED:
return 0;
case WAIT_TIMEOUT:
if (TerminateProcess (process_info.hProcess, 0))
{
nlwarning ("ERROR: Timeout in process %s", (char*)(argu.c_str()));
return 2;
}
else
nlwarning ("ERROR: Error while terminate current process %s", (char*)(argu.c_str()));
return 0;
}
}
else
nlwarning ("ERROR: Error can't exec process %s", (char*)(argu.c_str()));
return 0;
}
// Ok
return 1;
}