buffet: Add proper XML parsing for XMPP streams
For large command notifications, a single XMPP stanza can be split
into a number of TCP packets. In order to handle large stanzas and
in order to help with implementing TLS support for XMPP, added an
expat-based XML parser on top of XMPP stream, to make sure that
the stanzas are processed when all the data for a complete XML tag
has arrived.
BUG=brillo:458
TEST=`FEATURES=test emerge-link buffet`
Change-Id: I560f40dafb31c6e6b9e645d232453338ee4fbbef
Reviewed-on: https://chromium-review.googlesource.com/271592
Trybot-Ready: Alex Vakulenko <avakulenko@chromium.org>
Tested-by: Alex Vakulenko <avakulenko@chromium.org>
Reviewed-by: Vitaly Buka <vitalybuka@chromium.org>
Commit-Queue: Alex Vakulenko <avakulenko@chromium.org>
diff --git a/buffet/notification/xml_node.cc b/buffet/notification/xml_node.cc
new file mode 100644
index 0000000..f244c3a
--- /dev/null
+++ b/buffet/notification/xml_node.cc
@@ -0,0 +1,122 @@
+// Copyright 2015 The Chromium OS Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "buffet/notification/xml_node.h"
+
+#include <base/strings/stringprintf.h>
+#include <chromeos/strings/string_utils.h>
+
+namespace buffet {
+
+XmlNode::XmlNode(const std::string& name,
+ std::map<std::string, std::string> attributes)
+ : name_{name}, attributes_{std::move(attributes)} {}
+
+const std::string& XmlNode::name() const {
+ return name_;
+}
+
+const std::string& XmlNode::text() const {
+ return text_;
+}
+
+const std::map<std::string, std::string>& XmlNode::attributes() const {
+ return attributes_;
+}
+
+const std::vector<std::unique_ptr<XmlNode>>& XmlNode::children() const {
+ return children_;
+}
+
+bool XmlNode::GetAttribute(const std::string& name, std::string* value) const {
+ auto p = attributes_.find(name);
+ if (p == attributes_.end())
+ return false;
+
+ *value = p->second;
+ return true;
+}
+
+std::string XmlNode::GetAttributeOrEmpty(const std::string& name) const {
+ std::string value;
+ GetAttribute(name, &value);
+ return value;
+}
+
+const XmlNode* XmlNode::FindFirstChild(const std::string& name_path,
+ bool recursive) const {
+ return FindChildHelper(name_path, recursive, nullptr);
+}
+
+std::vector<const XmlNode*> XmlNode::FindChildren(const std::string& name_path,
+ bool recursive) const {
+ std::vector<const XmlNode*> children;
+ FindChildHelper(name_path, recursive, &children);
+ return children;
+}
+
+const XmlNode* XmlNode::FindChildHelper(
+ const std::string& name_path,
+ bool recursive,
+ std::vector<const XmlNode*>* children) const {
+ std::string name;
+ std::string rest_of_path;
+ chromeos::string_utils::SplitAtFirst(name_path, "/", &name, &rest_of_path,
+ false);
+ for (const auto& child : children_) {
+ const XmlNode* found_node = nullptr;
+ if (child->name() == name) {
+ if (rest_of_path.empty()) {
+ found_node = child.get();
+ } else {
+ found_node = child->FindChildHelper(rest_of_path, false, children);
+ }
+ } else if (recursive) {
+ found_node = child->FindChildHelper(name_path, true, children);
+ }
+
+ if (found_node) {
+ if (!children)
+ return found_node;
+ children->push_back(found_node);
+ }
+ }
+ return nullptr;
+}
+
+void XmlNode::SetText(const std::string& text) {
+ text_ = text;
+}
+
+void XmlNode::AppendText(const std::string& text) {
+ text_ += text;
+}
+
+void XmlNode::AddChild(std::unique_ptr<XmlNode> child) {
+ child->parent_ = this;
+ children_.push_back(std::move(child));
+}
+
+std::string XmlNode::ToString() const {
+ std::string xml = base::StringPrintf("<%s", name_.c_str());
+ for (const auto& pair : attributes_) {
+ base::StringAppendF(&xml, " %s=\"%s\"", pair.first.c_str(),
+ pair.second.c_str());
+ }
+ if (text_.empty() && children_.empty()) {
+ xml += "/>";
+ } else {
+ xml += '>';
+ if (!text_.empty()) {
+ xml += text_;
+ }
+ for (const auto& child : children_) {
+ xml += child->ToString();
+ }
+ base::StringAppendF(&xml, "</%s>", name_.c_str());
+ }
+ return xml;
+}
+
+} // namespace buffet
