/******************************************************************************
* Icinga 2 *
- * Copyright (C) 2012-2014 Icinga Development Team (http://www.icinga.org) *
+ * Copyright (C) 2012-2015 Icinga Development Team (http://www.icinga.org) *
* *
* This program is free software; you can redistribute it and/or *
* modify it under the terms of the GNU General Public License *
#include <wuapi.h>
#include <wuerror.h>
-#include "thresholds.h"
-
-#include "boost/program_options.hpp"
+#include "check_update.h"
#define VERSION 1.0
namespace po = boost::program_options;
-using std::wcout; using std::endl;
-using std::wstring; using std::cout;
+static BOOL debug = FALSE;
-struct printInfoStruct
+INT wmain(INT argc, WCHAR **argv)
{
- BOOL warn, crit;
- LONG numUpdates;
- BOOL important, reboot, careForCanRequest;
-};
-
-static int parseArguments(int, wchar_t **, po::variables_map&, printInfoStruct&);
-static int printOutput(const printInfoStruct&);
-static int check_update(printInfoStruct&);
-
-int main(int argc, wchar_t **argv)
-{
- po::variables_map vm;
printInfoStruct printInfo = { FALSE, FALSE, 0, FALSE, FALSE, FALSE };
+ po::variables_map vm;
- int ret = parseArguments(argc, argv, vm, printInfo);
+ INT ret = parseArguments(argc, argv, vm, printInfo);
if (ret != -1)
return ret;
return printOutput(printInfo);
}
-int printOutput(const printInfoStruct& printInfo)
-{
- state state = OK;
- wstring output = L"UPDATE ";
-
- if (printInfo.important)
- state = WARNING;
-
- if (printInfo.reboot)
- state = CRITICAL;
-
- switch (state) {
- case OK:
- output.append(L"OK ");
- break;
- case WARNING:
- output.append(L"WARNING ");
- break;
- case CRITICAL:
- output.append(L"CRITICAL ");
- break;
- }
-
- wcout << output << printInfo.numUpdates << L" | update=" << printInfo.numUpdates << L";"
- << printInfo.warn << L";" << printInfo.crit << L";0;" << endl;
-
- return state;
-}
-
-int parseArguments(int ac, wchar_t **av, po::variables_map& vm, printInfoStruct& printInfo)
+INT parseArguments(INT ac, WCHAR **av, po::variables_map& vm, printInfoStruct& printInfo)
{
- wchar_t namePath[MAX_PATH];
+ WCHAR namePath[MAX_PATH];
GetModuleFileName(NULL, namePath, MAX_PATH);
- wchar_t *progName = PathFindFileName(namePath);
+ WCHAR *progName = PathFindFileName(namePath);
po::options_description desc;
desc.add_options()
- ("help,h", "print help message and exit")
- ("version,V", "print version and exit")
- ("warning,w", "warn if there are important updates available")
- ("critical,c", "critical if there are important updates that require a reboot")
- ("possible-reboot", "treat \"update may need to reboot\" as \"update needs to reboot\"")
+ ("help,h", "Print help message and exit")
+ ("version,V", "Print version and exit")
+ ("debug,d", "Verbose/Debug output")
+ ("warning,w", "Warn if there are important updates available")
+ ("critical,c", "Critical if there are important updates that require a reboot")
+ ("possible-reboot", "Treat \"update may need reboot\" as \"update needs reboot\"")
;
- po::basic_command_line_parser<wchar_t> parser(ac, av);
+ po::basic_command_line_parser<WCHAR> parser(ac, av);
try {
po::store(
vm);
vm.notify();
} catch (std::exception& e) {
- cout << e.what() << endl << desc << endl;
+ std::cout << e.what() << '\n' << desc << '\n';
return 3;
}
-
+
if (vm.count("help")) {
- wcout << progName << " Help\n\tVersion: " << VERSION << endl;
+ std::wcout << progName << " Help\n\tVersion: " << VERSION << '\n';
wprintf(
L"%s is a simple program to check a machines required updates.\n"
L"You can use the following options to define its behaviour:\n\n", progName);
- cout << desc;
+ std::cout << desc;
wprintf(
L"\nAfter some time, it will then output a string like this one:\n\n"
- L"\tUPDATE WARNING 8|updates=8;1;1;0\n\n"
+ L"\tUPDATE WARNING 8 | updates=8;1;1;0\n\n"
L"\"UPDATE\" being the type of the check, \"WARNING\" the returned status\n"
L"and \"8\" is the number of important updates updates.\n"
L"The performance data is found behind the \"|\", in order:\n"
L"The \"possible-reboot\" option is not recommended since this true for nearly\n"
L"every update."
, progName, progName);
- cout << endl;
+ std::cout << '\n';
return 0;
- } if (vm.count("version")) {
- cout << "Version: " << VERSION << endl;
+ } if (vm.count("version")) {
+ std::cout << "Version: " << VERSION << '\n';
return 0;
}
if (vm.count("possible-reboot"))
printInfo.careForCanRequest = TRUE;
+ if (vm.count("debug"))
+ debug = TRUE;
+
return -1;
}
-int check_update(printInfoStruct& printInfo)
+INT printOutput(const printInfoStruct& printInfo)
{
+ if (debug)
+ std::wcout << L"Constructing output string" << '\n';
+
+ state state = OK;
+ std::wstring output = L"UPDATE ";
+
+ if (printInfo.important)
+ state = WARNING;
+
+ if (printInfo.reboot)
+ state = CRITICAL;
+
+ switch (state) {
+ case OK:
+ output.append(L"OK ");
+ break;
+ case WARNING:
+ output.append(L"WARNING ");
+ break;
+ case CRITICAL:
+ output.append(L"CRITICAL ");
+ break;
+ }
+
+ std::wcout << output << printInfo.numUpdates << L" | update=" << printInfo.numUpdates << L";"
+ << printInfo.warn << L";" << printInfo.crit << L";0;" << '\n';
+
+ return state;
+}
+
+INT check_update(printInfoStruct& printInfo)
+{
+ if (debug)
+ std::wcout << "Initializing COM library" << '\n';
CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
ISearchResult *pResult;
IUpdateSession *pSession;
IUpdateSearcher *pSearcher;
+ BSTR criteria = NULL;
HRESULT err;
-
+ if (debug)
+ std::wcout << "Creating UpdateSession and UpdateSearcher" << '\n';
CoCreateInstance(CLSID_UpdateSession, NULL, CLSCTX_INPROC_SERVER, IID_IUpdateSession, (LPVOID*)&pSession);
pSession->CreateUpdateSearcher(&pSearcher);
RebootRequired = 1: Reboot required
*/
- BSTR criteria = SysAllocString(CRITERIA);
+ criteria = SysAllocString(CRITERIA);
// http://msdn.microsoft.com/en-us/library/windows/desktop/aa386526%28v=vs.85%29.aspx
// http://msdn.microsoft.com/en-us/library/ff357803%28v=vs.85%29.aspx
+ if (debug)
+ std::wcout << L"Querrying updates from server" << '\n';
+
err = pSearcher->Search(criteria, &pResult);
if (!SUCCEEDED(err))
goto die;
return -1;
printInfo.numUpdates = updateSize;
- printInfo.important = printInfo.warn;
-
- if (!printInfo.crit)
- return -1;
+// printInfo.important = printInfo.warn;
IInstallationBehavior *pIbehav;
InstallationRebootBehavior updateReboot;
for (LONG i = 0; i < updateSize; i++) {
pCollection->get_Item(i, &pUpdate);
+ if (debug) {
+ std::wcout << L"Checking reboot behaviour of update number " << i << '\n';
+ }
pUpdate->get_InstallationBehavior(&pIbehav);
pIbehav->get_RebootBehavior(&updateReboot);
if (updateReboot == irbAlwaysRequiresReboot) {
printInfo.reboot = TRUE;
+ if (debug)
+ std::wcout << L"It requires reboot" << '\n';
continue;
}
if (printInfo.careForCanRequest && updateReboot == irbCanRequestReboot)
+ if (debug)
+ std::wcout << L"It requires reboot" << '\n';
printInfo.reboot = TRUE;
}
- return 0;
+ if (debug)
+ std::wcout << L"Cleaning up and returning" << '\n';
+
+ SysFreeString(criteria);
+ CoUninitialize();
+ return -1;
die:
die(err);
+ CoUninitialize();
if (criteria)
SysFreeString(criteria);
return 3;
-}
\ No newline at end of file
+}