mirror of
https://github.com/VCMP-SqMod/SqMod.git
synced 2025-01-31 18:07:14 +01: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.
104 lines
2.4 KiB
C++
104 lines
2.4 KiB
C++
//
|
|
// pkill.cpp
|
|
//
|
|
// Process killer utility application.
|
|
//
|
|
// Copyright (c) 2010, Applied Informatics Software Engineering GmbH.
|
|
// and Contributors.
|
|
//
|
|
// SPDX-License-Identifier: BSL-1.0
|
|
//
|
|
|
|
|
|
#include "Poco/Util/Application.h"
|
|
#include "Poco/Util/Option.h"
|
|
#include "Poco/Util/OptionSet.h"
|
|
#include "Poco/Util/HelpFormatter.h"
|
|
#include "Poco/Process.h"
|
|
#include "Poco/NumberParser.h"
|
|
#include <iostream>
|
|
|
|
|
|
using Poco::Util::Application;
|
|
using Poco::Util::Option;
|
|
using Poco::Util::OptionSet;
|
|
using Poco::Util::HelpFormatter;
|
|
using Poco::Util::OptionCallback;
|
|
|
|
|
|
class ProcessKillerApp: public Application
|
|
{
|
|
public:
|
|
ProcessKillerApp():
|
|
_helpRequested(false),
|
|
_friendly(false)
|
|
{
|
|
}
|
|
|
|
protected:
|
|
void defineOptions(OptionSet& options)
|
|
{
|
|
Application::defineOptions(options);
|
|
|
|
options.addOption(
|
|
Option("help", "h", "Display help information on command line arguments.")
|
|
.required(false)
|
|
.repeatable(false)
|
|
.callback(OptionCallback<ProcessKillerApp>(this, &ProcessKillerApp::handleHelp)));
|
|
|
|
options.addOption(
|
|
Option("friendly", "f", "Kindly ask application to shut down.")
|
|
.required(false)
|
|
.repeatable(false)
|
|
.callback(OptionCallback<ProcessKillerApp>(this, &ProcessKillerApp::handleFriendly)));
|
|
}
|
|
|
|
void handleHelp(const std::string& name, const std::string& value)
|
|
{
|
|
_helpRequested = true;
|
|
stopOptionsProcessing();
|
|
}
|
|
|
|
void handleFriendly(const std::string& name, const std::string& value)
|
|
{
|
|
_friendly = true;
|
|
}
|
|
|
|
void displayHelp()
|
|
{
|
|
HelpFormatter helpFormatter(options());
|
|
helpFormatter.setCommand(commandName());
|
|
helpFormatter.setUsage("[options] <pid> ...");
|
|
helpFormatter.setHeader("A utility application to kill processes.");
|
|
helpFormatter.setFooter("Note that the friendly option only works with applications using Poco::Util::ServerApplication::waitForTerminationRequest().");
|
|
helpFormatter.format(std::cout);
|
|
}
|
|
|
|
int main(const std::vector<std::string>& args)
|
|
{
|
|
if (_helpRequested || args.empty())
|
|
{
|
|
displayHelp();
|
|
}
|
|
else
|
|
{
|
|
for (std::vector<std::string>::const_iterator it = args.begin(); it != args.end(); ++it)
|
|
{
|
|
Poco::Process::PID pid = Poco::NumberParser::parseUnsigned(*it);
|
|
if (_friendly)
|
|
Poco::Process::requestTermination(pid);
|
|
else
|
|
Poco::Process::kill(pid);
|
|
}
|
|
}
|
|
return Application::EXIT_OK;
|
|
}
|
|
|
|
private:
|
|
bool _helpRequested;
|
|
bool _friendly;
|
|
};
|
|
|
|
|
|
POCO_APP_MAIN(ProcessKillerApp)
|