From 0915b3ef56dfac3113cce55a59a5765dc94976be Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Sun, 28 Apr 2024 14:34:54 +0200
Subject: Adding upstream version 2.13.6.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 lib/livestatus/attributefilter.cpp | 121 +++++++++++++++++++++++++++++++++++++
 1 file changed, 121 insertions(+)
 create mode 100644 lib/livestatus/attributefilter.cpp

(limited to 'lib/livestatus/attributefilter.cpp')

diff --git a/lib/livestatus/attributefilter.cpp b/lib/livestatus/attributefilter.cpp
new file mode 100644
index 0000000..50d7244
--- /dev/null
+++ b/lib/livestatus/attributefilter.cpp
@@ -0,0 +1,121 @@
+/* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */
+
+#include "livestatus/attributefilter.hpp"
+#include "base/convert.hpp"
+#include "base/array.hpp"
+#include "base/objectlock.hpp"
+#include "base/logger.hpp"
+#include <boost/regex.hpp>
+#include <boost/algorithm/string/predicate.hpp>
+
+using namespace icinga;
+
+AttributeFilter::AttributeFilter(String column, String op, String operand)
+	: m_Column(std::move(column)), m_Operator(std::move(op)), m_Operand(std::move(operand))
+{ }
+
+bool AttributeFilter::Apply(const Table::Ptr& table, const Value& row)
+{
+	Column column = table->GetColumn(m_Column);
+
+	Value value = column.ExtractValue(row);
+
+	if (value.IsObjectType<Array>()) {
+		Array::Ptr array = value;
+
+		if (m_Operator == ">=" || m_Operator == "<") {
+			bool negate = (m_Operator == "<");
+
+			ObjectLock olock(array);
+			for (const String& item : array) {
+				if (item == m_Operand)
+					return !negate; /* Item found in list. */
+			}
+
+			return negate; /* Item not found in list. */
+		} else if (m_Operator == "=") {
+			return (array->GetLength() == 0);
+		} else {
+			BOOST_THROW_EXCEPTION(std::invalid_argument("Invalid operator for column '" + m_Column + "': " + m_Operator + " (expected '>=' or '=')."));
+		}
+	} else {
+		if (m_Operator == "=") {
+			if (value.GetType() == ValueNumber || value.GetType() == ValueBoolean)
+				return (static_cast<double>(value) == Convert::ToDouble(m_Operand));
+			else
+				return (static_cast<String>(value) == m_Operand);
+		} else if (m_Operator == "~") {
+			bool ret;
+			try {
+				boost::regex expr(m_Operand.GetData());
+				String operand = value;
+				boost::smatch what;
+				ret = boost::regex_search(operand.GetData(), what, expr);
+			} catch (boost::exception&) {
+				Log(LogWarning, "AttributeFilter")
+					<< "Regex '" << m_Operand << " " << m_Operator << " " << value << "' error.";
+				ret = false;
+			}
+
+			//Log(LogDebug, "LivestatusListener/AttributeFilter")
+			//    << "Attribute filter '" << m_Operand + " " << m_Operator << " "
+			//    << value << "' " << (ret ? "matches" : "doesn't match") << ".";
+
+			return ret;
+		} else if (m_Operator == "=~") {
+			bool ret;
+			try {
+				String operand = value;
+				ret = boost::iequals(operand, m_Operand.GetData());
+			} catch (boost::exception&) {
+				Log(LogWarning, "AttributeFilter")
+					<< "Case-insensitive equality '" << m_Operand << " " << m_Operator << " " << value << "' error.";
+				ret = false;
+			}
+
+			return ret;
+		} else if (m_Operator == "~~") {
+			bool ret;
+			try {
+				boost::regex expr(m_Operand.GetData(), boost::regex::icase);
+				String operand = value;
+				boost::smatch what;
+				ret = boost::regex_search(operand.GetData(), what, expr);
+			} catch (boost::exception&) {
+				Log(LogWarning, "AttributeFilter")
+					<< "Regex '" << m_Operand << " " << m_Operator << " " << value << "' error.";
+				ret = false;
+			}
+
+			//Log(LogDebug, "LivestatusListener/AttributeFilter")
+			//    << "Attribute filter '" << m_Operand << " " << m_Operator << " "
+			//    << value << "' " << (ret ? "matches" : "doesn't match") << ".";
+
+			return ret;
+		} else if (m_Operator == "<") {
+			if (value.GetType() == ValueNumber)
+				return (static_cast<double>(value) < Convert::ToDouble(m_Operand));
+			else
+				return (static_cast<String>(value) < m_Operand);
+		} else if (m_Operator == ">") {
+			if (value.GetType() == ValueNumber)
+				return (static_cast<double>(value) > Convert::ToDouble(m_Operand));
+			else
+				return (static_cast<String>(value) > m_Operand);
+		} else if (m_Operator == "<=") {
+			if (value.GetType() == ValueNumber)
+				return (static_cast<double>(value) <= Convert::ToDouble(m_Operand));
+			else
+				return (static_cast<String>(value) <= m_Operand);
+		} else if (m_Operator == ">=") {
+			if (value.GetType() == ValueNumber)
+				return (static_cast<double>(value) >= Convert::ToDouble(m_Operand));
+			else
+				return (static_cast<String>(value) >= m_Operand);
+		} else {
+			BOOST_THROW_EXCEPTION(std::invalid_argument("Unknown operator for column '" + m_Column + "': " + m_Operator));
+		}
+	}
+
+	return false;
+}
-- 
cgit v1.2.3