version 4.0.8
[ric-plt/e2.git] / RIC-E2-TERMINATION / BuildXml.h
1 /*
2  * Copyright 2020 AT&T Intellectual Property
3  * Copyright 2020 Nokia
4  *
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at
8  *
9  *      http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  */
17
18 //
19 // Created by adi ENZEL on 4/5/20.
20 //
21
22 #ifndef E2_BUILDXML_H
23 #define E2_BUILDXML_H
24 #include <iostream>
25 #include <iosfwd>
26 #include <vector>
27 #include "pugixml/src/pugixml.hpp"
28 #include <string>
29 #include <sstream>
30 #include <mdclog/mdclog.h>
31 #include <cstdlib>
32
33 using namespace std;
34
35 /*
36  * Copied from pugixml samples
37  */
38 struct xml_string_writer : pugi::xml_writer {
39     std::string result;
40
41     void write(const void *data, size_t size) override {
42         result.append(static_cast<const char *>(data), size);
43     }
44 };
45 // end::code[]
46
47 //struct xml_memory_writer : pugi::xml_writer {
48 //    char *buffer;
49 //    size_t capacity;
50 //    size_t result;
51 //
52 //    xml_memory_writer() : buffer(nullptr), capacity(0), result(0) {
53 //    }
54 //
55 //    xml_memory_writer(char *buffer, size_t capacity) : buffer(buffer), capacity(capacity), result(0) {
56 //    }
57 //
58 //    [[nodiscard]] size_t written_size() const {
59 //        return result < capacity ? result : capacity;
60 //    }
61 //
62 //    void write(const void *data, size_t size) override {
63 //        if (result < capacity) {
64 //            size_t chunk = (capacity - result < size) ? capacity - result : size;
65 //
66 //            memcpy(buffer + result, data, chunk);
67 //        }
68 //        result += size;
69 //    }
70 //};
71
72 std::string node_to_string(pugi::xml_node node) {
73     xml_string_writer writer;
74     node.print(writer);
75
76     return writer.result;
77 }
78
79
80 int buildXmlData(const string &messageName, const string &ieName, vector<string> &RANfunctionsAdded, unsigned char *buffer, size_t size) {
81     pugi::xml_document doc;
82
83     doc.reset();
84     pugi::xml_parse_result result = doc.load_buffer((const char *)buffer, size);
85     if (result) {
86         unsigned int index = 0;
87         for (auto tool : doc.child("E2AP-PDU")
88                 .child("initiatingMessage")
89                 .child("value")
90                 .child(messageName.c_str())
91                 .child("protocolIEs")
92                 .children(ieName.c_str())) {
93             auto node = tool.child("id");
94             if (strcmp(node.name(), "id") == 0 && strcmp(node.child_value(), "10") == 0) {
95                 auto nodea = tool.child("value").
96                         child("RANfunctions-List").
97                         children("ProtocolIE-SingleContainer");
98                 for (auto n1 : nodea) {
99                     auto n2 = n1.child("value").child("RANfunction-Item").child("ranFunctionDefinition");
100                     n2.remove_children();
101                     string val = RANfunctionsAdded.at(index++);
102                     // here we get vector with counter
103                     n2.append_child(pugi::node_pcdata).set_value(val.c_str());
104                     if (mdclog_level_get() >= MDCLOG_DEBUG) {
105                         mdclog_write(MDCLOG_DEBUG, "entry %s Replaced with : %s", n2.name(), n2.child_value());
106                     }
107                 }
108             } else {
109                 if (mdclog_level_get() >= MDCLOG_DEBUG) {
110                     mdclog_write(MDCLOG_DEBUG, "Entry %s = value %s skipped", node.name(), node.child_value());
111                 }
112                 continue;
113             }
114         }
115
116         auto res = node_to_string(doc);
117         memcpy(buffer, res.c_str(), res.length());
118         doc.reset();
119     } else {
120         mdclog_write(MDCLOG_ERR, "Error loading xml string");
121         return -1;
122     }
123     return 0;
124
125 }
126
127 #endif //E2_BUILDXML_H