/*
    NUT configuration unit test

    Copyright (C)
        2012	Vaclav Krpec <VaclavKrpec@Eaton.com>
        2024-2025    Jim Klimov <jimklimov+nut@gmail.com>

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 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 General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/

#include "config.h"

#include "nutstream.hpp"
#include "nutconf.hpp"
#include "nutwriter.hpp"

#include <algorithm>

/* Current CPPUnit offends the honor of C++98 and maybe later versions */
#if (defined HAVE_PRAGMA_GCC_DIAGNOSTIC_PUSH_POP) && (defined HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_EXIT_TIME_DESTRUCTORS || defined HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_GLOBAL_CONSTRUCTORS || defined HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_DEPRECATED_DECLARATIONS || defined HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_SUGGEST_OVERRIDE_BESIDEFUNC || defined HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_SUGGEST_DESTRUCTOR_OVERRIDE_BESIDEFUNC || defined HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_WEAK_VTABLES_BESIDEFUNC || defined HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_DEPRECATED_DYNAMIC_EXCEPTION_SPEC_BESIDEFUNC || defined HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_EXTRA_SEMI_BESIDEFUNC || defined HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_OLD_STYLE_CAST_BESIDEFUNC)
#pragma GCC diagnostic push
# ifdef HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_GLOBAL_CONSTRUCTORS
#  pragma GCC diagnostic ignored "-Wglobal-constructors"
# endif
# ifdef HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_EXIT_TIME_DESTRUCTORS
#  pragma GCC diagnostic ignored "-Wexit-time-destructors"
# endif
# ifdef HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_DEPRECATED_DECLARATIONS
#  pragma GCC diagnostic ignored "-Wdeprecated-declarations"
# endif
# ifdef HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_SUGGEST_OVERRIDE_BESIDEFUNC
#  pragma GCC diagnostic ignored "-Wsuggest-override"
# endif
# ifdef HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_SUGGEST_DESTRUCTOR_OVERRIDE_BESIDEFUNC
#  pragma GCC diagnostic ignored "-Wsuggest-destructor-override"
# endif
# ifdef HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_WEAK_VTABLES_BESIDEFUNC
#  pragma GCC diagnostic ignored "-Wweak-vtables"
# endif
# ifdef HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_DEPRECATED_DYNAMIC_EXCEPTION_SPEC_BESIDEFUNC
#  pragma GCC diagnostic ignored "-Wdeprecated-dynamic-exception-spec"
# endif
# ifdef HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_EXTRA_SEMI_BESIDEFUNC
#  pragma GCC diagnostic ignored "-Wextra-semi"
# endif
# ifdef HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_OLD_STYLE_CAST_BESIDEFUNC
#  pragma GCC diagnostic ignored "-Wold-style-cast"
# endif
#endif
#if (defined __clang__) && (defined HAVE_PRAGMA_CLANG_DIAGNOSTIC_IGNORED_DEPRECATED_DECLARATIONS)
# ifdef HAVE_PRAGMA_CLANG_DIAGNOSTIC_IGNORED_DEPRECATED_DECLARATIONS
#  pragma clang diagnostic push "-Wdeprecated-declarations"
# endif
#endif

#include <cppunit/extensions/HelperMacros.h>

/**
 *  \brief  NUT configuration unit test
 */
class NutConfigUnitTest: public CppUnit::TestFixture {
	private:

	CPPUNIT_TEST_SUITE(NutConfigUnitTest);
		CPPUNIT_TEST( testNutConfiguration );
		CPPUNIT_TEST( testUpsmonConfiguration );
		CPPUNIT_TEST( testUpsdConfiguration );
		CPPUNIT_TEST( testUpsConfiguration );
		CPPUNIT_TEST( testUpsdUsersConfiguration );
	CPPUNIT_TEST_SUITE_END();

	/**
	 *  \brief  Load configuration from file
	 *
	 *  \param  config     Configuration object
	 *  \param  file_name  Configuration file name
	 */
	void load(nut::Serialisable * config, const std::string & file_name);

	/**
	 *  \brief  Check configuration serialization contents
	 *
	 *  \param  config   Configuration object
	 *  \param  content  Expected serialization
	 *  \param  quote_sensitive  Flag to allow stripping double-quotes for fallback comparison
	 */
	void check(const nut::Serialisable * config, const std::string & content, const bool quote_sensitive);

	/**
	 *  \brief  Check configuration serialization contents (exact)
	 *
	 *  \param  config   Configuration object
	 *  \param  content  Expected serialization
	 */
	inline void check(const nut::Serialisable * config, const std::string & content) {
		check(config, content, true);
	}

	public:

	/** nut.conf test */
	void testNutConfiguration();

	/** upsmon.conf test */
	void testUpsmonConfiguration();

	/** upsd.conf test */
	void testUpsdConfiguration();

	/** ups.conf test */
	void testUpsConfiguration();

	/** upsd.users test */
	void testUpsdUsersConfiguration();

	inline void setUp() override {}
	inline void tearDown() override {}

	virtual ~NutConfigUnitTest() override;
};  // end of class NutConfigUnitTest

// Register the test suite
CPPUNIT_TEST_SUITE_REGISTRATION(NutConfigUnitTest);


void NutConfigUnitTest::load(nut::Serialisable * config, const std::string & file_name) {
	nut::NutFile file(file_name, nut::NutFile::READ_ONLY);

	CPPUNIT_ASSERT(config->parseFrom(file));
}


void NutConfigUnitTest::check(const nut::Serialisable * config, const std::string & content, const bool quote_sensitive) {
	nut::NutMemory mem;

	CPPUNIT_ASSERT(config->writeTo(mem));

	std::string str;

	nut::NutStream::status_t status = mem.getString(str);

	CPPUNIT_ASSERT(nut::NutStream::NUTS_OK == status);

	if (content == str)
		return;

	if (!quote_sensitive) {
		// Re-check with quotes stripped (we output paths, passwords etc. quoted)
		std::string uqStr = str, uqContent = content;
		uqStr.erase(std::remove(uqStr.begin(), uqStr.end(), '"'), uqStr.end());
		uqContent.erase(std::remove(uqContent.begin(), uqContent.end(), '"'), uqContent.end());

		if (uqContent == uqStr)
			return;
	}

	// Neither expectation was met
	std::cerr << "--- expected ---" << std::endl << content << "--- end ---" << std::endl;
	std::cerr << "--- serialized ---" << std::endl << str << "--- end ---" << std::endl;

	CPPUNIT_ASSERT_MESSAGE("Configuration serialization check failed", 0);
}


void NutConfigUnitTest::testNutConfiguration() {
	nut::NutConfiguration config;

	load(static_cast<nut::Serialisable *>(&config), ABS_TOP_SRCDIR "/conf/nut.conf.sample");

	config.mode = nut::NutConfiguration::MODE_STANDALONE;

	config.upsdOptions   = "-DDD -B";
	config.allowNoDevice = true;
	config.poweroffQuiet = false;

	check(static_cast<nut::Serialisable *>(&config),
		"MODE=standalone\n"
		"ALLOW_NO_DEVICE=true\n"
		"UPSD_OPTIONS='-DDD -B'\n"
		"POWEROFF_QUIET=false\n"
	);
}


void NutConfigUnitTest::testUpsmonConfiguration() {
	nut::UpsmonConfiguration config;
	nut::BoolInt	*tmpPtr = nullptr;

	// Note: this file gets generated from a .in template
	load(static_cast<nut::Serialisable *>(&config), ABS_TOP_BUILDDIR "/conf/upsmon.conf.sample");

	config.shutdownCmd   = "/sbin/shutdown -h +2 'System shutdown in 2 minutes!'";
	config.powerDownFlag = "/run/nut/killpower";
	config.pollFreqAlert = 10;
	config.deadTime      = 30;

	config.debugMin      = 6;

	// Note different bool wording expected below
	// and generally try different input types here.
	// Hm, maybe we need typed constructors to init
	// the values for test? ("new" weirdness below
	// is due to Settable<> formal type mismatch)
	config.shutdownExit  = (*(tmpPtr = new nut::BoolInt()) << "true"); delete tmpPtr;
	config.oblbDuration  = -1;
	config.overDuration  = -1;
	config.certVerify    = (*(tmpPtr = new nut::BoolInt()) << false); delete tmpPtr;
	config.forceSsl      = (*(tmpPtr = new nut::BoolInt()) << "1"); delete tmpPtr;
	config.alarmCritical = (*(tmpPtr = new nut::BoolInt()) << "1"); delete tmpPtr;

	config.certIdent.certName    = "My test cert";
	config.certIdent.certDbPass  = "DbPwd!";

	nut::CertHost certHost;
	certHost.host        = "remote:3493";
	certHost.certName    = "NUT server cert";
	certHost.certVerify  = (*(tmpPtr = new nut::BoolInt()) << "true"); delete tmpPtr;
	certHost.forceSsl    = (*(tmpPtr = new nut::BoolInt()) << 0); delete tmpPtr;
	config.certHosts.push_back(certHost);

	certHost.host        = "localhost:13493";
	certHost.certName    = "My NUT server cert";
	certHost.certVerify  = (*(tmpPtr = new nut::BoolInt()) << "false"); delete tmpPtr;
	certHost.forceSsl    = (*(tmpPtr = new nut::BoolInt()) << false); delete tmpPtr;
	config.certHosts.push_back(certHost);

	// Note: More config data points come from the file loaded above
	check(static_cast<nut::Serialisable *>(&config),
		"DEBUG_MIN 6\n"
		"SHUTDOWNCMD \"/sbin/shutdown -h +2 'System shutdown in 2 minutes!'\"\n"
		"POWERDOWNFLAG \"/run/nut/killpower\"\n"
		"MINSUPPLIES 1\n"
		"POLLFREQ 5\n"
		"POLLFREQALERT 10\n"
		"OFFDURATION 30\n"
		"OBLBDURATION -1\n"
		"OVERDURATION -1\n"
		"SHUTDOWNEXIT yes\n"
		"CERTVERIFY 0\n"
		"FORCESSL 1\n"
		"ALARMCRITICAL 1\n"
		"HOSTSYNC 15\n"
		"DEADTIME 30\n"
		"RBWARNTIME 43200\n"
		"NOCOMMWARNTIME 300\n"
		"FINALDELAY 5\n"
		"CERTIDENT \"My test cert\" \"DbPwd!\"\n"
		"CERTHOST \"remote:3493\" \"NUT server cert\" 1 0\n"
		"CERTHOST \"localhost:13493\" \"My NUT server cert\" 0 0\n"
	);
}


void NutConfigUnitTest::testUpsdConfiguration() {
	nut::UpsdConfiguration config;

	load(static_cast<nut::Serialisable *>(&config), ABS_TOP_SRCDIR "/conf/upsd.conf.sample");

	config.maxAge    = 15;
	config.statePath = "/var/run/nut";
	config.maxConn   = 1024;
	config.certFile  = "/usr/share/ssl-cert/ssleay.cnf";

	nut::UpsdConfiguration::Listen listen;

	listen.address = "127.0.0.1";
	listen.port    = 3493;

	config.listens.push_back(listen);

	listen.address = "::1";

	config.listens.push_back(listen);

	// NOTE: Three-arg check() here to retry (and succeed) with un-quoted paths
	check(static_cast<nut::Serialisable *>(&config),
		"MAXAGE 15\n"
		"MAXCONN 1024\n"
		"STATEPATH \"/var/run/nut\"\n"
		"CERTFILE /usr/share/ssl-cert/ssleay.cnf\n"
		"LISTEN 127.0.0.1 3493\n"
		"LISTEN ::1 3493\n"
	, false);
}


void NutConfigUnitTest::testUpsConfiguration() {
	nut::UpsConfiguration config;

	load(static_cast<nut::Serialisable *>(&config), ABS_TOP_SRCDIR "/conf/ups.conf.sample");

	static const std::string my_ups("powerpal");

	config.setDriver(my_ups, "blazer_ser");
	config.setPort(my_ups, "/dev/ttyS0");
	config.setDescription(my_ups, "Web server");

	// IRL not for a serial-port UPS, but okay for tests;
	// Note NUT v2.8.1 introduced these as "hexnum" values,
	// but the strtoul() underneath knows to strip "0x" for
	// base16 conversions - so can we write them either way:
	config.setUsbConfigIndex(my_ups, 0x81);
	config.setUsbHidDescIndex(my_ups, 0x3f);

	// Note: "maxretry = 3" comes from current ups.conf.sample non-comment lines
	std::string expected =
		"maxretry = 3\n\n"
		"[powerpal]\n"
		"\tdesc = \"Web server\"\n"
		"\tdriver = blazer_ser\n"
		"\tport = /dev/ttyS0\n"
		"\tusb_config_index = 81\n"
		"\tusb_hid_desc_index = 3f\n"
		"\n";

	check(static_cast<nut::Serialisable *>(&config), expected);

	// Make sure we can parse back the "hexnum" entries correctly
	config.parseFromString(expected);
	check(static_cast<nut::Serialisable *>(&config), expected);

	// Check that "0x" prefix (and/or quotes) do not matter on
	// this platform (note that parsing results are strings from
	// original input; for the check we interpret them as numbers
	// and save back):
	std::string input2 =
		"maxretry = 3\n\n"
		"[powerpal]\n"
		"\tdesc = \"Web server\"\n"
		"\tdriver = blazer_ser\n"
		"\tport = /dev/ttyS0\n"
		"\tusb_config_index = 0x81\n"
		"\tusb_hid_desc_index = \"0x3f\"\n"
		"\n";

	config.parseFromString(input2);
	config.setUsbConfigIndex (my_ups, config.getUsbConfigIndex(my_ups));
	config.setUsbHidDescIndex(my_ups, config.getUsbHidDescIndex(my_ups));

	// Should retrieve "-1" for unset value by default in its getter,
	// and auto-destroy the (missing) entry upon set() because < 0
	config.setUsbSetAltInterface (my_ups, config.getUsbSetAltInterface(my_ups));

	check(static_cast<nut::Serialisable *>(&config), expected);

	CPPUNIT_ASSERT_EQUAL(static_cast<long long int>(129),
			config.getUsbConfigIndex(my_ups));
	CPPUNIT_ASSERT_EQUAL(static_cast<long long int>(0x3f),
			config.getUsbHidDescIndex(my_ups));

	// Check default.* and override.* value(!) pre-sets:
	std::string input3 =
		"[powerpal]\n"
		"\tdesc = \"Web server\"\n"
		"\tdriver = blazer_ser\n"
		"\tport = /dev/ttyS0\n"
		"\tdefault.battery.voltage.high = 28.3\n"
		"\n";

	// "high" is inherited from input; "log" added by code:
	std::string expected3 =
		"[powerpal]\n"
		"\tdefault.battery.voltage.high = 28.3\n"
		"\tdesc = \"Web server\"\n"
		"\tdriver = blazer_ser\n"
		"\toverride.battery.voltage.low = 12.4\n"
		"\tport = /dev/ttyS0\n"
		"\n";

	config.parseFromString(input3);
	config.setOverrideDouble(my_ups, "battery.voltage.low", 12.4);

	// "maxretry" is inherited from earlier content of config
	// (not cleared). Maybe a bug, address later.
	check(static_cast<nut::Serialisable *>(&config),
		"maxretry = 3\n\n" + expected3);

	CPPUNIT_ASSERT_DOUBLES_EQUAL(28.3,
			config.getDefaultDouble(my_ups, "battery.voltage.high"), 0.0001);
	CPPUNIT_ASSERT_DOUBLES_EQUAL(12.4,
			config.getOverrideDouble(my_ups, "battery.voltage.low"), 0.0001);

	// TODO: Wipe method? Fix parse*() to clear global section
	// like they do clear and repopulate an UPS section?
	// Re-parse from scratch
	nut::UpsConfiguration config3;
	config3.parseFromString(input3);
	config3.setOverrideDouble(my_ups, "battery.voltage.low", 12.4);
	check(static_cast<nut::Serialisable *>(&config3), expected3);
}


void NutConfigUnitTest::testUpsdUsersConfiguration() {
	nut::UpsdUsersConfiguration config;

	load(static_cast<nut::Serialisable *>(&config), ABS_TOP_SRCDIR "/conf/upsd.users.sample");

	config.setPassword("upsmon", "ytrewq");
	config.setUpsmonMode(nut::UpsdUsersConfiguration::UPSMON_PRIMARY);

	config.setPassword("admin", "qwerty=ui");
	config.setActions("admin", nut::ConfigParamList(1, "SET"));
	config.setInstantCommands("admin", nut::ConfigParamList(1, "ALL"));

	check(static_cast<nut::Serialisable *>(&config),
		"[admin]\n"
		"\tactions = SET\n"
		"\tinstcmds = ALL\n"
		"\tpassword = \"qwerty=ui\"\n"
		"\n"
		"[upsmon]\n"
		"\tpassword = ytrewq\n"
		"\tupsmon primary\n"
		"\n"
	);
}

// Implement out of class declaration to avoid
//   error: 'SomeClass' has no out-of-line virtual method
//   definitions; its vtable will be emitted in every translation unit
//   [-Werror,-Wweak-vtables]
NutConfigUnitTest::~NutConfigUnitTest() {}

#if (defined __clang__) && (defined HAVE_PRAGMA_CLANG_DIAGNOSTIC_IGNORED_DEPRECATED_DECLARATIONS)
# pragma clang diagnostic pop
#endif
#if (defined HAVE_PRAGMA_GCC_DIAGNOSTIC_PUSH_POP) && (defined HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_EXIT_TIME_DESTRUCTORS || defined HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_GLOBAL_CONSTRUCTORS || defined HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_DEPRECATED_DECLARATIONS || defined HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_SUGGEST_OVERRIDE_BESIDEFUNC || defined HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_SUGGEST_DESTRUCTOR_OVERRIDE_BESIDEFUNC || defined HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_WEAK_VTABLES_BESIDEFUNC || defined HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_DEPRECATED_DYNAMIC_EXCEPTION_SPEC_BESIDEFUNC || defined HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_EXTRA_SEMI_BESIDEFUNC || defined HAVE_PRAGMA_GCC_DIAGNOSTIC_IGNORED_OLD_STYLE_CAST_BESIDEFUNC)
# pragma GCC diagnostic pop
#endif