colosseum-near-rt-ric/setup/e2/RIC-E2-TERMINATION/BuildXml.h
Leonardo Bonati 60dffad583 First commit
2021-12-08 20:17:46 +00:00

135 lines
4.3 KiB
C++

/*
* Copyright 2020 AT&T Intellectual Property
* Copyright 2020 Nokia
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
//
// Created by adi ENZEL on 4/5/20.
//
#ifndef E2_BUILDXML_H
#define E2_BUILDXML_H
#include <iostream>
#include <iosfwd>
#include <vector>
#include "pugixml/src/pugixml.hpp"
#include <string>
#include <sstream>
#include <mdclog/mdclog.h>
#include <cstdlib>
using namespace std;
/*
* Copied from pugixml samples
*/
struct xml_string_writer : pugi::xml_writer {
std::string result;
void write(const void *data, size_t size) override {
result.append(static_cast<const char *>(data), size);
}
};
// end::code[]
//struct xml_memory_writer : pugi::xml_writer {
// char *buffer;
// size_t capacity;
// size_t result;
//
// xml_memory_writer() : buffer(nullptr), capacity(0), result(0) {
// }
//
// xml_memory_writer(char *buffer, size_t capacity) : buffer(buffer), capacity(capacity), result(0) {
// }
//
// [[nodiscard]] size_t written_size() const {
// return result < capacity ? result : capacity;
// }
//
// void write(const void *data, size_t size) override {
// if (result < capacity) {
// size_t chunk = (capacity - result < size) ? capacity - result : size;
//
// memcpy(buffer + result, data, chunk);
// }
// result += size;
// }
//};
std::string node_to_string(pugi::xml_node node) {
xml_string_writer writer;
node.print(writer);
return writer.result;
}
int buildXmlData(const string &messageName, const string &ieName, vector<string> &RANfunctionsAdded, unsigned char *buffer, size_t size) {
pugi::xml_document doc;
doc.reset();
pugi::xml_parse_result result = doc.load_buffer((const char *)buffer, size);
if (result) {
unsigned int index = 0;
for (auto tool : doc.child("E2AP-PDU")
.child("initiatingMessage")
.child("value")
.child(messageName.c_str())
.child("protocolIEs")
.children(ieName.c_str())) {
// there can be many ieName entries in the messageName so we need only the ones that containes E2SM continers
auto node = tool.child("id"); // get the id to identify the type of the contained message
if (node.empty()) {
mdclog_write(MDCLOG_ERR, "Failed to find ID node in the XML. File %s, line %d",
__FILE__, __LINE__);
continue;
}
if (strcmp(node.name(), "id") == 0 && strcmp(node.child_value(), "10") == 0) {
auto nodea = tool.child("value").
child("RANfunctions-List").
children("ProtocolIE-SingleContainer");
for (auto n1 : nodea) {
auto n2 = n1.child("value").child("RANfunction-Item").child("ranFunctionDefinition");
n2.remove_children();
string val = RANfunctionsAdded.at(index++);
// here we get vector with counter
n2.append_child(pugi::node_pcdata).set_value(val.c_str());
if (mdclog_level_get() >= MDCLOG_DEBUG) {
mdclog_write(MDCLOG_DEBUG, "entry %s Replaced with : %s", n2.name(), n2.child_value());
}
}
} else {
if (mdclog_level_get() >= MDCLOG_DEBUG) {
mdclog_write(MDCLOG_DEBUG, "Entry %s = value %s skipped", node.name(), node.child_value());
}
continue;
}
}
auto res = node_to_string(doc);
memcpy(buffer, res.c_str(), res.length());
doc.reset();
} else {
mdclog_write(MDCLOG_ERR, "Error loading xml string");
return -1;
}
return 0;
}
#endif //E2_BUILDXML_H