mirror of
https://github.com/VCMP-SqMod/SqMod.git
synced 2025-02-12 07:47:12 +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.
271 lines
7.2 KiB
C++
271 lines
7.2 KiB
C++
//
|
|
// SyslogTest.cpp
|
|
//
|
|
// Copyright (c) 2006, Applied Informatics Software Engineering GmbH.
|
|
// and Contributors.
|
|
//
|
|
// SPDX-License-Identifier: BSL-1.0
|
|
//
|
|
|
|
|
|
#include "SyslogTest.h"
|
|
#include "CppUnit/TestCaller.h"
|
|
#include "CppUnit/TestSuite.h"
|
|
#include "Poco/Net/RemoteSyslogChannel.h"
|
|
#include "Poco/Net/RemoteSyslogListener.h"
|
|
#include "Poco/Net/DNS.h"
|
|
#include "Poco/Thread.h"
|
|
#include "Poco/Message.h"
|
|
#include "Poco/AutoPtr.h"
|
|
#include <list>
|
|
|
|
|
|
using namespace Poco::Net;
|
|
|
|
|
|
class CachingChannel: public Poco::Channel
|
|
/// Caches the last n Messages in memory
|
|
{
|
|
public:
|
|
typedef std::list<Poco::Message> Messages;
|
|
|
|
CachingChannel(std::size_t n = 100);
|
|
/// Creates the CachingChannel. Caches n messages in memory
|
|
|
|
~CachingChannel();
|
|
/// Destroys the CachingChannel.
|
|
|
|
void log(const Poco::Message& msg);
|
|
/// Writes the log message to the cache
|
|
|
|
void getMessages(std::vector<Poco::Message>& msg, int offset, int numEntries) const;
|
|
/// Retrieves numEntries Messages starting with position offset. Most recent messages are first.
|
|
|
|
std::size_t getMaxSize() const;
|
|
|
|
std::size_t getCurrentSize() const;
|
|
|
|
private:
|
|
CachingChannel(const CachingChannel&);
|
|
|
|
Messages _cache;
|
|
std::size_t _size;
|
|
std::size_t _maxSize;
|
|
mutable Poco::FastMutex _mutex;
|
|
};
|
|
|
|
|
|
std::size_t CachingChannel::getMaxSize() const
|
|
{
|
|
return _maxSize;
|
|
}
|
|
|
|
|
|
std::size_t CachingChannel::getCurrentSize() const
|
|
{
|
|
return _size;
|
|
}
|
|
|
|
|
|
CachingChannel::CachingChannel(std::size_t n):
|
|
_cache(),
|
|
_size(0),
|
|
_maxSize(n),
|
|
_mutex()
|
|
{
|
|
}
|
|
|
|
|
|
CachingChannel::~CachingChannel()
|
|
{
|
|
}
|
|
|
|
|
|
void CachingChannel::log(const Poco::Message& msg)
|
|
{
|
|
Poco::FastMutex::ScopedLock lock(_mutex);
|
|
_cache.push_front(msg);
|
|
if (_size == _maxSize)
|
|
{
|
|
_cache.pop_back();
|
|
}
|
|
else
|
|
++_size;
|
|
}
|
|
|
|
|
|
void CachingChannel::getMessages(std::vector<Poco::Message>& msg, int offset, int numEntries) const
|
|
{
|
|
msg.clear();
|
|
Messages::const_iterator it = _cache.begin();
|
|
|
|
while (offset > 0 && it != _cache.end())
|
|
++it;
|
|
|
|
while (numEntries > 0 && it != _cache.end())
|
|
{
|
|
msg.push_back(*it);
|
|
++it;
|
|
}
|
|
}
|
|
|
|
|
|
SyslogTest::SyslogTest(const std::string& name): CppUnit::TestCase(name)
|
|
{
|
|
}
|
|
|
|
|
|
SyslogTest::~SyslogTest()
|
|
{
|
|
}
|
|
|
|
|
|
void SyslogTest::testListener()
|
|
{
|
|
Poco::AutoPtr<RemoteSyslogChannel> channel = new RemoteSyslogChannel();
|
|
channel->setProperty("loghost", "127.0.0.1:51400");
|
|
channel->open();
|
|
Poco::AutoPtr<RemoteSyslogListener> listener = new RemoteSyslogListener(51400);
|
|
listener->open();
|
|
auto pCL = Poco::makeAuto<CachingChannel>();
|
|
listener->addChannel(pCL);
|
|
assertTrue (pCL->getCurrentSize() == 0);
|
|
Poco::Message msg("asource", "amessage", Poco::Message::PRIO_CRITICAL);
|
|
channel->log(msg);
|
|
Poco::Thread::sleep(1000);
|
|
listener->close();
|
|
channel->close();
|
|
assertTrue (pCL->getCurrentSize() == 1);
|
|
std::vector<Poco::Message> msgs;
|
|
pCL->getMessages(msgs, 0, 10);
|
|
assertTrue (msgs.size() == 1);
|
|
assertTrue (msgs[0].getSource() == "asource");
|
|
assertTrue (msgs[0].getText() == "amessage");
|
|
assertTrue (msgs[0].getPriority() == Poco::Message::PRIO_CRITICAL);
|
|
}
|
|
|
|
|
|
void SyslogTest::testChannelOpenClose()
|
|
{
|
|
Poco::AutoPtr<RemoteSyslogChannel> channel = new RemoteSyslogChannel();
|
|
channel->setProperty("loghost", "127.0.0.1:51400");
|
|
channel->open();
|
|
Poco::AutoPtr<RemoteSyslogListener> listener = new RemoteSyslogListener(51400);
|
|
listener->open();
|
|
auto pCL = Poco::makeAuto<CachingChannel>();
|
|
listener->addChannel(pCL);
|
|
|
|
assertTrue (pCL->getCurrentSize() == 0);
|
|
Poco::Message msg1("source1", "message1", Poco::Message::PRIO_CRITICAL);
|
|
channel->log(msg1);
|
|
Poco::Thread::sleep(1000);
|
|
assertTrue (pCL->getCurrentSize() == 1);
|
|
|
|
channel->close(); // close and re-open channel
|
|
channel->open();
|
|
|
|
Poco::Message msg2("source2", "message2", Poco::Message::PRIO_ERROR);
|
|
channel->log(msg2);
|
|
Poco::Thread::sleep(1000);
|
|
assertTrue (pCL->getCurrentSize() == 2);
|
|
|
|
listener->close();
|
|
std::vector<Poco::Message> msgs;
|
|
pCL->getMessages(msgs, 0, 10);
|
|
assertTrue (msgs.size() == 2);
|
|
|
|
assertTrue (msgs[1].getSource() == "source1");
|
|
assertTrue (msgs[1].getText() == "message1");
|
|
assertTrue (msgs[1].getPriority() == Poco::Message::PRIO_CRITICAL);
|
|
|
|
assertTrue (msgs[0].getSource() == "source2");
|
|
assertTrue (msgs[0].getText() == "message2");
|
|
assertTrue (msgs[0].getPriority() == Poco::Message::PRIO_ERROR);
|
|
}
|
|
|
|
|
|
void SyslogTest::testOldBSD()
|
|
{
|
|
Poco::AutoPtr<RemoteSyslogChannel> channel = new RemoteSyslogChannel();
|
|
channel->setProperty("loghost", "127.0.0.1:51400");
|
|
channel->setProperty("format", "bsd");
|
|
channel->open();
|
|
Poco::AutoPtr<RemoteSyslogListener> listener = new RemoteSyslogListener(51400);
|
|
listener->open();
|
|
auto pCL = Poco::makeAuto<CachingChannel>();
|
|
listener->addChannel(pCL);
|
|
assertTrue (pCL->getCurrentSize() == 0);
|
|
Poco::Message msg("asource", "amessage", Poco::Message::PRIO_CRITICAL);
|
|
channel->log(msg);
|
|
Poco::Thread::sleep(1000);
|
|
listener->close();
|
|
channel->close();
|
|
assertTrue (pCL->getCurrentSize() == 1);
|
|
std::vector<Poco::Message> msgs;
|
|
pCL->getMessages(msgs, 0, 10);
|
|
assertTrue (msgs.size() == 1);
|
|
// the source is lost with old BSD messages: we only send the local host name!
|
|
assertTrue (msgs[0].getSource() == Poco::Net::DNS::thisHost().name());
|
|
assertTrue (msgs[0].getText() == "amessage");
|
|
assertTrue (msgs[0].getPriority() == Poco::Message::PRIO_CRITICAL);
|
|
}
|
|
|
|
|
|
void SyslogTest::testStructuredData()
|
|
{
|
|
Poco::AutoPtr<RemoteSyslogChannel> channel = new RemoteSyslogChannel();
|
|
channel->setProperty("loghost", "127.0.0.1:51400");
|
|
channel->open();
|
|
Poco::AutoPtr<RemoteSyslogListener> listener = new RemoteSyslogListener(51400);
|
|
listener->open();
|
|
auto pCL = Poco::makeAuto<CachingChannel>();
|
|
listener->addChannel(pCL);
|
|
assertTrue (pCL->getCurrentSize() == 0);
|
|
Poco::Message msg1("asource", "amessage", Poco::Message::PRIO_CRITICAL);
|
|
msg1.set("structured-data", "[exampleSDID@32473 iut=\"3\" eventSource=\"Application\" eventID=\"1011\"]");
|
|
channel->log(msg1);
|
|
Poco::Message msg2("asource", "amessage", Poco::Message::PRIO_CRITICAL);
|
|
msg2.set("structured-data", "[exampleSDID@32473 iut=\"3\" eventSource=\"Application\" eventID=\"1011\"][examplePriority@32473 class=\"high\"]");
|
|
channel->log(msg2);
|
|
Poco::Thread::sleep(1000);
|
|
listener->close();
|
|
channel->close();
|
|
assertTrue (pCL->getCurrentSize() == 2);
|
|
std::vector<Poco::Message> msgs;
|
|
pCL->getMessages(msgs, 0, 10);
|
|
assertTrue (msgs.size() == 2);
|
|
|
|
assertTrue (msgs[0].getSource() == "asource");
|
|
assertTrue (msgs[0].getText() == "amessage");
|
|
assertTrue (msgs[0].getPriority() == Poco::Message::PRIO_CRITICAL);
|
|
assertTrue (msgs[0].get("structured-data") == "[exampleSDID@32473 iut=\"3\" eventSource=\"Application\" eventID=\"1011\"][examplePriority@32473 class=\"high\"]");
|
|
|
|
assertTrue (msgs[1].getSource() == "asource");
|
|
assertTrue (msgs[1].getText() == "amessage");
|
|
assertTrue (msgs[1].getPriority() == Poco::Message::PRIO_CRITICAL);
|
|
assertTrue (msgs[1].get("structured-data") == "[exampleSDID@32473 iut=\"3\" eventSource=\"Application\" eventID=\"1011\"]");
|
|
}
|
|
|
|
|
|
void SyslogTest::setUp()
|
|
{
|
|
}
|
|
|
|
|
|
void SyslogTest::tearDown()
|
|
{
|
|
}
|
|
|
|
|
|
CppUnit::Test* SyslogTest::suite()
|
|
{
|
|
CppUnit::TestSuite* pSuite = new CppUnit::TestSuite("SyslogTest");
|
|
|
|
CppUnit_addTest(pSuite, SyslogTest, testListener);
|
|
CppUnit_addTest(pSuite, SyslogTest, testChannelOpenClose);
|
|
CppUnit_addTest(pSuite, SyslogTest, testOldBSD);
|
|
CppUnit_addTest(pSuite, SyslogTest, testStructuredData);
|
|
|
|
return pSuite;
|
|
}
|