mirror of
https://github.com/VCMP-SqMod/SqMod.git
synced 2025-04-18 10:17:14 +02:00
4a6bfc086c
Switched to POCO library for unified platform/library interface. Deprecated the external module API. It was creating more problems than solving. Removed most built-in libraries in favor of system libraries for easier maintenance. Cleaned and secured code with help from static analyzers.
285 lines
6.3 KiB
C++
285 lines
6.3 KiB
C++
//
|
|
// ProcessTest.cpp
|
|
//
|
|
// Copyright (c) 2005-2006, Applied Informatics Software Engineering GmbH.
|
|
// and Contributors.
|
|
//
|
|
// SPDX-License-Identifier: BSL-1.0
|
|
//
|
|
|
|
|
|
#include "ProcessTest.h"
|
|
#include "CppUnit/TestCaller.h"
|
|
#include "CppUnit/TestSuite.h"
|
|
#include "Poco/Process.h"
|
|
#include "Poco/Pipe.h"
|
|
#include "Poco/PipeStream.h"
|
|
|
|
|
|
using namespace std::string_literals;
|
|
using Poco::Process;
|
|
using Poco::ProcessHandle;
|
|
using Poco::Pipe;
|
|
using Poco::PipeInputStream;
|
|
using Poco::PipeOutputStream;
|
|
|
|
|
|
ProcessTest::ProcessTest(const std::string& name): CppUnit::TestCase(name)
|
|
{
|
|
}
|
|
|
|
|
|
ProcessTest::~ProcessTest()
|
|
{
|
|
}
|
|
|
|
|
|
void ProcessTest::testEscapeArgs()
|
|
{
|
|
#if defined(_WIN32)
|
|
assertTrue (Poco::ProcessImpl::mustEscapeArg("a b"));
|
|
assertFalse (Poco::ProcessImpl::mustEscapeArg("abc"));
|
|
assertFalse (Poco::ProcessImpl::mustEscapeArg("\"a b \""));
|
|
assertFalse (Poco::ProcessImpl::mustEscapeArg("\"abc\""));
|
|
assertTrue (Poco::ProcessImpl::mustEscapeArg("\"a b "));
|
|
assertFalse (Poco::ProcessImpl::mustEscapeArg("/arg=\"a b c\""));
|
|
|
|
assertEquals ("abc"s, Poco::ProcessImpl::escapeArg("abc"));
|
|
assertEquals ("\"a b c\""s, Poco::ProcessImpl::escapeArg("a b c"));
|
|
assertEquals ("\"a b \\\" c\""s, Poco::ProcessImpl::escapeArg("a b \" c"));
|
|
assertEquals ("/arg=\"a b c\""s, Poco::ProcessImpl::escapeArg("/arg=\"a b c\""));
|
|
#endif
|
|
}
|
|
|
|
|
|
void ProcessTest::testLaunch()
|
|
{
|
|
std::string name("TestApp");
|
|
std::string cmd;
|
|
#if defined(_DEBUG) && (POCO_OS != POCO_OS_ANDROID)
|
|
name += "d";
|
|
#endif
|
|
|
|
#if defined(POCO_OS_FAMILY_UNIX)
|
|
cmd += name;
|
|
#elif defined(_WIN32_WCE)
|
|
cmd = "\\";
|
|
cmd += name;
|
|
cmd += ".EXE";
|
|
#else
|
|
cmd = name;
|
|
#endif
|
|
|
|
std::vector<std::string> args;
|
|
args.push_back("arg1");
|
|
args.push_back("arg2");
|
|
args.push_back("arg3");
|
|
ProcessHandle ph = Process::launch(cmd, args);
|
|
int rc = ph.wait();
|
|
assertTrue (rc == 3);
|
|
}
|
|
|
|
|
|
void ProcessTest::testLaunchRedirectIn()
|
|
{
|
|
#if !defined(_WIN32_WCE)
|
|
std::string name("TestApp");
|
|
std::string cmd;
|
|
#if defined(_DEBUG) && (POCO_OS != POCO_OS_ANDROID)
|
|
name += "d";
|
|
#endif
|
|
|
|
#if defined(POCO_OS_FAMILY_UNIX)
|
|
cmd += name;
|
|
#else
|
|
cmd = name;
|
|
#endif
|
|
|
|
std::vector<std::string> args;
|
|
args.push_back("-count");
|
|
Pipe inPipe;
|
|
ProcessHandle ph = Process::launch(cmd, args, &inPipe, 0, 0);
|
|
PipeOutputStream ostr(inPipe);
|
|
ostr << std::string(100, 'x');
|
|
ostr.close();
|
|
int rc = ph.wait();
|
|
assertTrue (rc == 100);
|
|
#endif // !defined(_WIN32_WCE)
|
|
}
|
|
|
|
|
|
void ProcessTest::testLaunchRedirectOut()
|
|
{
|
|
#if !defined(_WIN32_WCE)
|
|
std::string name("TestApp");
|
|
std::string cmd;
|
|
#if defined(_DEBUG) && (POCO_OS != POCO_OS_ANDROID)
|
|
name += "d";
|
|
#endif
|
|
|
|
#if defined(POCO_OS_FAMILY_UNIX)
|
|
cmd += name;
|
|
#else
|
|
cmd = name;
|
|
#endif
|
|
|
|
std::vector<std::string> args;
|
|
args.push_back("-hello");
|
|
Pipe outPipe;
|
|
ProcessHandle ph = Process::launch(cmd, args, 0, &outPipe, 0);
|
|
PipeInputStream istr(outPipe);
|
|
std::string s;
|
|
int c = istr.get();
|
|
while (c != -1) { s += (char) c; c = istr.get(); }
|
|
assertTrue (s == "Hello, world!");
|
|
int rc = ph.wait();
|
|
assertTrue (rc == 1);
|
|
#endif // !defined(_WIN32_WCE)
|
|
}
|
|
|
|
|
|
void ProcessTest::testLaunchEnv()
|
|
{
|
|
#if !defined(_WIN32_WCE)
|
|
std::string name("TestApp");
|
|
std::string cmd;
|
|
#if defined(_DEBUG) && (POCO_OS != POCO_OS_ANDROID)
|
|
name += "d";
|
|
#endif
|
|
|
|
#if defined(POCO_OS_FAMILY_UNIX)
|
|
cmd += name;
|
|
#else
|
|
cmd = name;
|
|
#endif
|
|
|
|
std::vector<std::string> args;
|
|
args.push_back("-env");
|
|
Pipe outPipe;
|
|
Process::Env env;
|
|
env["TESTENV"] = "test";
|
|
ProcessHandle ph = Process::launch(cmd, args, 0, &outPipe, 0, env);
|
|
PipeInputStream istr(outPipe);
|
|
std::string s;
|
|
int c = istr.get();
|
|
while (c != -1) { s += (char) c; c = istr.get(); }
|
|
assertTrue (s == "test");
|
|
int rc = ph.wait();
|
|
assertTrue (rc == 0);
|
|
#endif // !defined(_WIN32_WCE)
|
|
}
|
|
|
|
|
|
void ProcessTest::testLaunchArgs()
|
|
{
|
|
#if defined (_WIN32) && !defined(_WIN32_WCE)
|
|
std::string name("TestApp");
|
|
#if defined(_DEBUG)
|
|
name += 'd';
|
|
#endif
|
|
std::string cmd = name;
|
|
|
|
std::vector<std::string> args;
|
|
args.push_back("-echo-args");
|
|
args.push_back("simple");
|
|
args.push_back("with space");
|
|
args.push_back("with\ttab");
|
|
args.push_back("with\vverticaltab");
|
|
// can't test newline here because TestApp -echo-args uses newline to separate the echoed args
|
|
//args.push_back("with\nnewline");
|
|
args.push_back("with \" quotes");
|
|
args.push_back("ends with \"quotes\"");
|
|
args.push_back("\"starts\" with quotes");
|
|
args.push_back("\"");
|
|
args.push_back("\\");
|
|
args.push_back("c:\\program files\\ends with backslash\\");
|
|
args.push_back("\"already quoted \\\" \\\\\"");
|
|
Pipe outPipe;
|
|
ProcessHandle ph = Process::launch(cmd, args, 0, &outPipe, 0);
|
|
PipeInputStream istr(outPipe);
|
|
std::string receivedArg;
|
|
int c = istr.get();
|
|
int argNumber = 1;
|
|
while (c != -1)
|
|
{
|
|
if ('\n' == c)
|
|
{
|
|
assertTrue (argNumber < args.size());
|
|
std::string expectedArg = args[argNumber];
|
|
if (expectedArg.npos != expectedArg.find("already quoted")) {
|
|
expectedArg = "already quoted \" \\";
|
|
}
|
|
assertTrue (receivedArg == expectedArg);
|
|
++argNumber;
|
|
receivedArg = "";
|
|
}
|
|
else if ('\r' != c)
|
|
{
|
|
receivedArg += (char)c;
|
|
}
|
|
c = istr.get();
|
|
}
|
|
assertTrue (argNumber == args.size());
|
|
int rc = ph.wait();
|
|
assertTrue (rc == args.size());
|
|
#endif // !defined(_WIN32_WCE)
|
|
}
|
|
|
|
|
|
void ProcessTest::testIsRunning()
|
|
{
|
|
#if !defined(_WIN32_WCE)
|
|
std::string name("TestApp");
|
|
std::string cmd;
|
|
#if defined(_DEBUG) && (POCO_OS != POCO_OS_ANDROID)
|
|
name += "d";
|
|
#endif
|
|
|
|
#if defined(POCO_OS_FAMILY_UNIX)
|
|
cmd += name;
|
|
#else
|
|
cmd = name;
|
|
#endif
|
|
|
|
std::vector<std::string> args;
|
|
args.push_back("-count");
|
|
Pipe inPipe;
|
|
ProcessHandle ph = Process::launch(cmd, args, &inPipe, 0, 0);
|
|
Process::PID id = ph.id();
|
|
assertTrue (Process::isRunning(ph));
|
|
assertTrue (Process::isRunning(id));
|
|
PipeOutputStream ostr(inPipe);
|
|
ostr << std::string(100, 'x');
|
|
ostr.close();
|
|
int POCO_UNUSED rc = ph.wait();
|
|
assertTrue (!Process::isRunning(ph));
|
|
assertTrue (!Process::isRunning(id));
|
|
#endif // !defined(_WIN32_WCE)
|
|
}
|
|
|
|
|
|
void ProcessTest::setUp()
|
|
{
|
|
}
|
|
|
|
|
|
void ProcessTest::tearDown()
|
|
{
|
|
}
|
|
|
|
|
|
CppUnit::Test* ProcessTest::suite()
|
|
{
|
|
CppUnit::TestSuite* pSuite = new CppUnit::TestSuite("ProcessTest");
|
|
|
|
CppUnit_addTest(pSuite, ProcessTest, testEscapeArgs);
|
|
CppUnit_addTest(pSuite, ProcessTest, testLaunch);
|
|
CppUnit_addTest(pSuite, ProcessTest, testLaunchRedirectIn);
|
|
CppUnit_addTest(pSuite, ProcessTest, testLaunchRedirectOut);
|
|
CppUnit_addTest(pSuite, ProcessTest, testLaunchEnv);
|
|
CppUnit_addTest(pSuite, ProcessTest, testLaunchArgs);
|
|
CppUnit_addTest(pSuite, ProcessTest, testIsRunning);
|
|
|
|
return pSuite;
|
|
}
|