#
SUBDIRS = snapper dbus server client scripts pam data doc po examples \
- testsuite testsuite-real testsuite-cmp zypp-plugin
+ testsuite testsuite-real testsuite-cmp stomp zypp-plugin
AUTOMAKE_OPTIONS = foreign dist-xz no-dist-gzip
testsuite/Makefile
testsuite-real/Makefile
testsuite-cmp/Makefile
+ stomp/Makefile
+ stomp/testsuite/Makefile
zypp-plugin/Makefile
zypp-plugin/testsuite/Makefile
package/snapper.spec:snapper.spec.in
+-------------------------------------------------------------------
+Fri Jul 05 16:03:42 CEST 2024 - aschnell@suse.com
+
+- handle content-length of stomp in zypper plugin
+ (gh#openSUSE/snapper#918)
+
-------------------------------------------------------------------
Wed May 22 17:21:18 CEST 2024 - aschnell@suse.com
--- /dev/null
+*.lo
+*.la
--- /dev/null
+#
+# Makefile.am for snapper/stomp
+#
+
+SUBDIRS = . testsuite
+
+noinst_LTLIBRARIES = libstomp.la
+
+libstomp_la_SOURCES = \
+ Stomp.h Stomp.cc
--- /dev/null
+/*
+ * Copyright (c) [2019-2024] SUSE LLC
+ *
+ * All Rights Reserved.
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms of version 2 of the GNU General Public License as published
+ * by the Free Software Foundation.
+ *
+ * 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, contact SUSE LLC.
+ *
+ * To contact SUSE about this file by physical or electronic mail, you may
+ * find current contact information at www.suse.com.
+ */
+
+
+#include <iostream>
+#include <regex>
+
+
+using namespace std;
+
+
+#include "Stomp.h"
+
+
+namespace Stomp
+{
+
+ Message
+ read_message(istream& is)
+ {
+ static const regex rx_command("[A-Za-z0-9_]+", regex::extended);
+
+ enum class State { Start, Headers, Body } state = State::Start;
+ bool has_content_length = false;
+ ssize_t content_length = 0;
+
+ Message msg;
+
+ while (!is.eof())
+ {
+ string line;
+ getline(is, line);
+
+ if (state == State::Start)
+ {
+ if (is.eof())
+ return msg; // empty
+
+ if (line.empty())
+ continue;
+
+ if (regex_match(line, rx_command))
+ {
+ msg = Message();
+ msg.command = line;
+ state = State::Headers;
+ }
+ else
+ {
+ throw runtime_error("stomp error: expected a command, got '" + line + "'");
+ }
+ }
+ else if (state == State::Headers)
+ {
+ if (line.empty())
+ {
+ state = State::Body;
+
+ if (has_content_length)
+ {
+ if (content_length > 0)
+ {
+ vector<char> buf(content_length);
+ is.read(buf.data(), content_length);
+ msg.body.assign(buf.data(), content_length);
+ }
+
+ // still read the \0 that terminates the frame
+ char buf2 = '-';
+ is.read(&buf2, 1);
+ if (buf2 != '\0')
+ throw runtime_error("stomp error: missing \\0 at frame end");
+ }
+ else
+ {
+ getline(is, msg.body, '\0');
+ }
+
+ return msg;
+ }
+ else
+ {
+ string::size_type pos = line.find(':');
+ if (pos == string::npos)
+ throw runtime_error("stomp error: expected a header or new line, got '" + line + "'");
+
+ string key = unescape_header(line.substr(0, pos));
+ string value = unescape_header(line.substr(pos + 1));
+
+ if (key == "content-length")
+ {
+ has_content_length = true;
+ content_length = std::stol(value.c_str());
+ }
+
+ msg.headers[key] = value;
+ }
+ }
+ }
+
+ throw runtime_error("stomp error: expected a message, got a part of it");
+ }
+
+
+ void
+ write_message(ostream& os, const Message& msg)
+ {
+ os << msg.command << '\n';
+ for (auto it : msg.headers)
+ os << escape_header(it.first) << ':' << escape_header(it.second) << '\n';
+ os << '\n';
+ os << msg.body << '\0';
+ os.flush();
+ }
+
+
+ Message
+ ack()
+ {
+ Message msg;
+ msg.command = "ACK";
+ return msg;
+ }
+
+
+ Message
+ nack()
+ {
+ Message msg;
+ msg.command = "NACK";
+ return msg;
+ }
+
+
+ std::string
+ escape_header(const std::string& in)
+ {
+ string out;
+
+ for (const char c : in)
+ {
+ switch (c)
+ {
+ case '\r':
+ out += "\\r"; break;
+ case '\n':
+ out += "\\n"; break;
+ case ':':
+ out += "\\c"; break;
+ case '\\':
+ out += "\\\\"; break;
+
+ default:
+ out += c;
+ }
+ }
+
+ return out;
+ }
+
+
+ std::string
+ unescape_header(const std::string& in)
+ {
+ string out;
+
+ for (string::const_iterator it = in.begin(); it != in.end(); ++it)
+ {
+ if (*it == '\\')
+ {
+ if (++it == in.end())
+ throw runtime_error("stomp error: invalid start of escape sequence");
+
+ switch (*it)
+ {
+ case 'r':
+ out += '\r'; break;
+ case 'n':
+ out += '\n'; break;
+ case 'c':
+ out += ':'; break;
+ case '\\':
+ out += '\\'; break;
+
+ default:
+ throw runtime_error("stomp error: unknown escape sequence");
+ }
+ }
+ else
+ {
+ out += *it;
+ }
+ }
+
+ return out;
+ }
+
+}
--- /dev/null
+/*
+ * Copyright (c) [2019-2024] SUSE LLC
+ *
+ * All Rights Reserved.
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms of version 2 of the GNU General Public License as published
+ * by the Free Software Foundation.
+ *
+ * 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, contact SUSE LLC.
+ *
+ * To contact SUSE about this file by physical or electronic mail, you may
+ * find current contact information at www.suse.com.
+ */
+
+
+#ifndef SNAPPER_STOMP_H
+#define SNAPPER_STOMP_H
+
+
+#include <istream>
+#include <ostream>
+#include <string>
+#include <map>
+
+
+/**
+ * A tiny STOMP (https://stomp.github.io/) implementation.
+ */
+
+namespace Stomp
+{
+
+ struct Message
+ {
+ std::string command;
+ std::map<std::string, std::string> headers;
+ std::string body;
+ };
+
+
+ Message read_message(std::istream& is);
+ void write_message(std::ostream& os, const Message& msg);
+
+ Message ack();
+ Message nack();
+
+ std::string escape_header(const std::string& in);
+ std::string unescape_header(const std::string& in);
+
+}
+
+
+#endif
--- /dev/null
+*.log
+*.o
+*.test
+*.trs
+test-suite.log
--- /dev/null
+#
+# Makefile.am for snapper/stomp/testsuite
+#
+
+SUBDIRS = .
+
+LDADD = \
+ ../libstomp.la \
+ -lboost_unit_test_framework
+
+check_PROGRAMS = \
+ read1.test write1.test escape.test
+
+AM_DEFAULT_SOURCE_EXT = .cc
+
+TESTS = $(check_PROGRAMS)
+
--- /dev/null
+
+#define BOOST_TEST_DYN_LINK
+#define BOOST_TEST_MODULE snapper
+
+#include <boost/test/unit_test.hpp>
+
+#include "../Stomp.h"
+
+
+using namespace std;
+using namespace Stomp;
+
+
+BOOST_AUTO_TEST_CASE(escape)
+{
+ BOOST_CHECK_EQUAL(Stomp::escape_header("hello"), "hello");
+
+ BOOST_CHECK_EQUAL(Stomp::escape_header("hello\nworld"), "hello\\nworld");
+ BOOST_CHECK_EQUAL(Stomp::escape_header("hello:world"), "hello\\cworld");
+}
+
+
+BOOST_AUTO_TEST_CASE(unescape)
+{
+ BOOST_CHECK_EQUAL(Stomp::unescape_header("hello"), "hello");
+
+ BOOST_CHECK_EQUAL(Stomp::unescape_header("hello\\nworld"), "hello\nworld");
+ BOOST_CHECK_EQUAL(Stomp::unescape_header("hello\\cworld"), "hello:world");
+}
--- /dev/null
+
+#define BOOST_TEST_DYN_LINK
+#define BOOST_TEST_MODULE snapper
+
+#include <boost/test/unit_test.hpp>
+
+#include "../Stomp.h"
+
+
+using namespace std;
+using namespace Stomp;
+
+
+const string null("\0", 1);
+
+
+BOOST_AUTO_TEST_CASE(test1)
+{
+ // no optional content-lenght
+
+ istringstream s1("HELLO\nkey:value\n\nWORLD" + null);
+ istream s2(s1.rdbuf());
+
+ Message msg = read_message(s2);
+
+ BOOST_CHECK_EQUAL(s2.peek(), -1);
+
+ BOOST_CHECK_EQUAL(msg.command, "HELLO");
+
+ BOOST_CHECK_EQUAL(msg.headers.size(), 1);
+ BOOST_CHECK_EQUAL(msg.headers["key"], "value");
+
+ BOOST_CHECK_EQUAL(msg.body, "WORLD");
+}
+
+
+BOOST_AUTO_TEST_CASE(test2)
+{
+ // optional content-lenght
+
+ istringstream s1("HELLO\nkey:value\ncontent-length:5\n\nWORLD" + null);
+ istream s2(s1.rdbuf());
+
+ Message msg = read_message(s2);
+
+ BOOST_CHECK_EQUAL(s2.peek(), -1);
+
+ BOOST_CHECK_EQUAL(msg.command, "HELLO");
+
+ BOOST_CHECK_EQUAL(msg.headers.size(), 2);
+ BOOST_CHECK_EQUAL(msg.headers["key"], "value");
+ BOOST_CHECK_EQUAL(msg.headers["content-length"], "5");
+
+ BOOST_CHECK_EQUAL(msg.body, "WORLD");
+}
+
+
+BOOST_AUTO_TEST_CASE(escape1)
+{
+ // special characters in header
+
+ istringstream s1("HELLO\nGermany\\cSpain:2\\c1\n\nWORLD" + null);
+ istream s2(s1.rdbuf());
+
+ Message msg = read_message(s2);
+
+ BOOST_CHECK_EQUAL(msg.command, "HELLO");
+
+ BOOST_CHECK_EQUAL(msg.headers.size(), 1);
+ BOOST_CHECK_EQUAL(msg.headers["Germany:Spain"], "2:1");
+
+ BOOST_CHECK_EQUAL(msg.body, "WORLD");
+}
--- /dev/null
+
+#define BOOST_TEST_DYN_LINK
+#define BOOST_TEST_MODULE snapper
+
+#include <boost/test/unit_test.hpp>
+
+#include "../Stomp.h"
+
+
+using namespace std;
+using namespace Stomp;
+
+
+const string null("\0", 1);
+
+
+BOOST_AUTO_TEST_CASE(test1)
+{
+ Message msg;
+ msg.command = "HELLO";
+ msg.headers["key"] = "value";
+ msg.body = "WORLD";
+
+ ostringstream s1;
+ write_message(s1, msg);
+ string s2 = s1.str();
+
+ BOOST_CHECK_EQUAL(s2, "HELLO\nkey:value\n\nWORLD" + null);
+}
+
+
+BOOST_AUTO_TEST_CASE(escape1)
+{
+ // special characters in header
+
+ Message msg;
+ msg.command = "HELLO";
+ msg.headers["Germany:Spain"] = "2:1";
+ msg.body = "WORLD";
+
+ ostringstream s1;
+ write_message(s1, msg);
+ string s2 = s1.str();
+
+ BOOST_CHECK_EQUAL(s2, "HELLO\nGermany\\cSpain:2\\c1\n\nWORLD" + null);
+}
../client/libclient.la \
../snapper/libsnapper.la \
../dbus/libdbus.la \
+ ../stomp/libstomp.la \
$(JSONC_LIBS)
check_PROGRAMS = solvable-matcher.test forwarding-zypp-plugin
forwarding_zypp_plugin_LDADD = \
../snapper/libsnapper.la \
+ ../stomp/libstomp.la \
-lboost_system \
-lpthread
solvable_matcher_test_LDADD = \
../snapper/libsnapper.la \
+ ../stomp/libstomp.la \
$(XML2_LIBS) \
-lboost_unit_test_framework
}
-ZyppPlugin::Message
-ZyppPlugin::read_message(istream& is) const
-{
- enum class State { Start, Headers, Body } state = State::Start;
-
- Message msg;
-
- while (!is.eof())
- {
- string line;
-
- getline(is, line);
- boost::trim_right(line);
-
- if (state == State::Start)
- {
- if (is.eof())
- return msg; //empty
-
- if (line.empty())
- continue;
-
- static const regex rx_word("[A-Za-z0-9_]+", regex::extended);
- if (regex_match(line, rx_word))
- {
- msg = Message();
- msg.command = line;
- state = State::Headers;
- }
- else
- {
- throw runtime_error("Plugin protocol error: expected a command. Got '" + line + "'");
- }
- }
- else if (state == State::Headers)
- {
- if (line.empty())
- {
- state = State::Body;
- getline(is, msg.body, '\0');
-
- return msg;
- }
- else
- {
- static const regex rx_header("([A-Za-z0-9_]+):[ \t]*(.+)", regex::extended);
- smatch match;
-
- if (regex_match(line, match, rx_header))
- {
- string key = match[1];
- string value = match[2];
- msg.headers[key] = value;
- }
- else
- {
- throw runtime_error("Plugin protocol error: expected a header or new line. Got '" + line + "'");
- }
- }
- }
- }
-
- throw runtime_error("Plugin protocol error: expected a message, got a part of it");
-}
-
-
-void
-ZyppPlugin::write_message(ostream& os, const Message& msg) const
-{
- os << msg.command << endl;
- for (auto it : msg.headers)
- os << it.first << ':' << it.second << endl;
- os << endl;
- os << msg.body << '\0';
- os.flush();
-}
-
-
ZyppPlugin::Message
ZyppPlugin::dispatch(const Message& msg)
{
#include <iostream>
-#include <map>
-#include <string>
+
+#include "../stomp/Stomp.h"
class ZyppPlugin
// Plugin message aka frame
// https://doc.opensuse.org/projects/libzypp/SLE12SP2/zypp-plugins.html
- struct Message
- {
- std::string command;
- std::map<std::string, std::string> headers;
- std::string body;
- };
+ using Message = Stomp::Message;
/// Where the protocol reads from
std::istream& pin;
: pin(in), pout(out)
{}
- virtual ~ZyppPlugin() {}
+ virtual ~ZyppPlugin() = default;
virtual int main();
protected:
- Message read_message(std::istream& is) const;
- void write_message(std::ostream& os, const Message& msg) const;
+ Message read_message(std::istream& is) const { return Stomp::read_message(is); }
+ void write_message(std::ostream& os, const Message& msg) const { Stomp::write_message(os, msg); }
/// Handle a message and return a reply.
// Derived classes must override it.
// The base acks a _DISCONNECT and replies _ENOMETHOD to everything else.
virtual Message dispatch(const Message& msg) = 0;
- Message ack() const
- {
- Message a;
- a.command = "ACK";
- return a;
- }
+ Message ack() const { return Stomp::ack(); }
+
};
#endif
projects / thirdparty / snapper.git / commitdiff
