ntsimulator/ntsim-ng/features/ves_file_ready/ves_file_ready.c

   1 /*************************************************************************
   2 *
   3 * Copyright 2020 highstreet technologies GmbH and others
   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 #define _GNU_SOURCE
  19
  20 #include "ves_file_ready.h"
  21 #include "utils/log_utils.h"
  22 #include "utils/sys_utils.h"
  23 #include "utils/http_client.h"
  24 #include "utils/nts_utils.h"
  25 #include <stdio.h>
  26 #include <assert.h>
  27
  28 #include <sysrepo.h>
  29 #include <sysrepo/values.h>
  30
  31 #include "core/framework.h"
  32 #include "core/session.h"
  33 #include "core/xpath.h"
  34
  35 static int ves_file_ready_invoke_pm_cb(sr_session_ctx_t *session, const char *path, const sr_val_t *input, const size_t input_cnt, sr_event_t event, uint32_t request_id, sr_val_t **output, size_t *output_cnt, void *private_data);
  36 static int ves_file_ready_send_message(sr_session_ctx_t *session, const char *file_location, int port);
  37 static cJSON* ves_create_file_ready_fields(const char* file_location);
  38
  39 static sr_subscription_ctx_t *ves_file_ready_subscription = 0;
  40
  41 int ves_file_ready_feature_get_status(void) {
  42     return (ves_file_ready_subscription != 0);
  43 }
  44
  45 int ves_file_ready_feature_start(sr_session_ctx_t *current_session) {
  46     assert(current_session);
  47     assert_session();
  48
  49     if(ves_file_ready_subscription == 0) {
  50         int rc = sr_rpc_subscribe(current_session, NTS_NF_RPC_FILE_READY_SCHEMA_XPATH, ves_file_ready_invoke_pm_cb, 0, 0, SR_SUBSCR_CTX_REUSE, &ves_file_ready_subscription);
  51         if(rc != SR_ERR_OK) {
  52             log_error("error from sr_rpc_subscribe: %s\n", sr_strerror(rc));
  53             return NTS_ERR_FAILED;
  54         }
  55
  56         sftp_daemon_init();
  57         vsftp_daemon_init();
  58     }
  59
  60     return NTS_ERR_OK;
  61 }
  62
  63 int ves_file_ready_feature_stop(void) {
  64     assert_session();
  65
  66     if(ves_file_ready_subscription) {
  67         int rc = sr_unsubscribe(ves_file_ready_subscription);
  68         if(rc != SR_ERR_OK) {
  69             log_error("error from sr_rpc_subscribe: %s\n", sr_strerror(rc));
  70             return NTS_ERR_FAILED;
  71         }
  72
  73         vsftp_daemon_deinit();
  74         sftp_daemon_deinit();
  75         ves_file_ready_subscription = 0;
  76     }
  77
  78     return NTS_ERR_OK;
  79 }
  80
  81 static int ves_file_ready_invoke_pm_cb(sr_session_ctx_t *session, const char *path, const sr_val_t *input, const size_t input_cnt, sr_event_t event, uint32_t request_id, sr_val_t **output, size_t *output_cnt, void *private_data) {
  82     int ssh_base_port = 0;
  83     int tls_base_port = 0;
  84     sr_session_ctx_t *current_session = 0;
  85
  86     int rc = sr_session_start(session_connection, SR_DS_RUNNING, &current_session);
  87     if(rc != SR_ERR_OK) {
  88         log_error("could not start sysrepo session\n");
  89         return NTS_ERR_FAILED;
  90     }
  91
  92     nts_mount_point_addressing_method_t mp = nts_mount_point_addressing_method_get(current_session);
  93     if(mp == UNKNOWN_MAPPING) {
  94         log_error("mount-point-addressing-method failed\n");
  95         sr_session_stop(current_session);
  96         return NTS_ERR_FAILED;
  97     }
  98     else if(mp == DOCKER_MAPPING) {
  99         ssh_base_port = STANDARD_NETCONF_PORT;
 100         tls_base_port = ssh_base_port + framework_environment.settings.ssh_connections;
 101     }
 102     else {
 103         ssh_base_port = framework_environment.host.ssh_base_port;
 104         tls_base_port = framework_environment.host.tls_base_port;
 105     }
 106
 107     int failed = 0;
 108
 109     if((framework_environment.settings.ssh_connections + framework_environment.settings.tls_connections) > 1) {
 110         for(int port = ssh_base_port; port < ssh_base_port + framework_environment.settings.ssh_connections; port++) {
 111             int rc = ves_file_ready_send_message(current_session, input[0].data.string_val, port);
 112             if(rc != NTS_ERR_OK) {
 113                 log_error("ves_file_ready_send_message failed\n");
 114                 failed++;
 115             }
 116         }
 117
 118         for(int port = tls_base_port; port < tls_base_port + framework_environment.settings.tls_connections; port++) {
 119             int rc = ves_file_ready_send_message(current_session, input[0].data.string_val, port);
 120             if(rc != NTS_ERR_OK) {
 121                 log_error("ves_file_ready_send_message failed\n");
 122                 failed++;
 123             }
 124         }
 125     }
 126     else {
 127         int rc = ves_file_ready_send_message(current_session, input[0].data.string_val, 0);
 128         if(rc != NTS_ERR_OK) {
 129             log_error("ves_file_ready_send_message failed\n");
 130             failed++;
 131         }
 132     }
 133
 134     rc = sr_session_stop(current_session);
 135     if(rc != SR_ERR_OK) {
 136         log_error("could not stop sysrepo session\n");
 137         return NTS_ERR_FAILED;
 138     }
 139
 140     *output_cnt = 1;
 141     rc = sr_new_values(*output_cnt, output);
 142     if(SR_ERR_OK != rc) {
 143         return rc;
 144     }
 145
 146     rc = sr_val_set_xpath(output[0], NTS_NF_RPC_FILE_READY_SCHEMA_XPATH"/status");
 147     if(SR_ERR_OK != rc) {
 148         return rc;
 149     }
 150
 151     if(failed != 0) {
 152         rc = sr_val_build_str_data(output[0], SR_ENUM_T, "%s", "ERROR");
 153     }
 154     else {
 155         rc = sr_val_build_str_data(output[0], SR_ENUM_T, "%s", "SUCCESS");
 156     }
 157
 158     return rc;
 159 }
 160
 161 static int ves_file_ready_send_message(sr_session_ctx_t *session, const char *file_location, int port) {
 162     assert(session);
 163     assert(file_location);
 164
 165     int rc;
 166     static int sequence_number = 0;
 167
 168     cJSON *post_data_json = cJSON_CreateObject();
 169     if(post_data_json == 0) {
 170         log_error("could not create cJSON object\n");
 171         return NTS_ERR_FAILED;
 172     }
 173
 174     cJSON *event = cJSON_CreateObject();
 175     if(event == 0) {
 176         log_error("could not create cJSON object\n");
 177         cJSON_Delete(post_data_json);
 178         return NTS_ERR_FAILED;
 179     }
 180
 181     if(cJSON_AddItemToObject(post_data_json, "event", event) == 0) {
 182         log_error("cJSON_AddItemToObject failed\n");
 183         cJSON_Delete(post_data_json);
 184         return NTS_ERR_FAILED;
 185     }
 186
 187     cJSON *common_event_header = ves_create_common_event_header("notification", "Notification-gnb_Nokia-FileReady", framework_environment.settings.hostname, port, "Normal", sequence_number++);
 188     if(common_event_header == 0) {
 189         log_error("could not create cJSON object\n");
 190         cJSON_Delete(post_data_json);
 191         return NTS_ERR_FAILED;
 192     }
 193
 194     if(cJSON_AddItemToObject(event, "commonEventHeader", common_event_header) == 0) {
 195         log_error("cJSON_AddItemToObject failed\n");
 196         cJSON_Delete(post_data_json);
 197         return NTS_ERR_FAILED;
 198     }
 199
 200     cJSON *file_ready_fields = ves_create_file_ready_fields(file_location);
 201     if(file_ready_fields == 0) {
 202         log_error("could not create cJSON object\n");
 203         cJSON_Delete(post_data_json);
 204         return NTS_ERR_FAILED;
 205     }
 206
 207     if(cJSON_AddItemToObject(event, "notificationFields", file_ready_fields) == 0) {
 208         log_error("cJSON_AddItemToObject failed\n");
 209         cJSON_Delete(post_data_json);
 210         return NTS_ERR_FAILED;
 211     }
 212
 213     char *post_data = cJSON_PrintUnformatted(post_data_json);
 214     cJSON_Delete(post_data_json);
 215     if(post_data == 0) {
 216         log_error("cJSON_PrintUnformatted failed\n");
 217         return NTS_ERR_FAILED;
 218     }
 219
 220
 221     ves_details_t *ves_details = ves_endpoint_details_get(session, 0);
 222     if(!ves_details) {
 223         log_error("ves_endpoint_details_get failed\n");
 224         free(post_data);
 225         return NTS_ERR_FAILED;
 226     }
 227
 228     rc = http_request(ves_details->url, ves_details->username, ves_details->password, "POST", post_data, 0, 0);
 229     ves_details_free(ves_details);
 230     free(post_data);
 231
 232     if(rc != NTS_ERR_OK) {
 233         log_error("http_request failed\n");
 234         return NTS_ERR_FAILED;
 235     }
 236
 237     return NTS_ERR_OK;
 238 }
 239
 240 static cJSON* ves_create_file_ready_fields(const char* file_location) {
 241     assert(file_location);
 242
 243     cJSON *file_ready_fields = cJSON_CreateObject();
 244     if(file_ready_fields == 0) {
 245         log_error("could not create JSON object\n");
 246         return 0;
 247     }
 248
 249     if(cJSON_AddStringToObject(file_ready_fields, "changeIdentifier", "PM_MEAS_FILES") == 0) {
 250         log_error("cJSON_AddStringToObject failed\n");
 251         cJSON_Delete(file_ready_fields);
 252         return 0;
 253     }
 254
 255     if(cJSON_AddStringToObject(file_ready_fields, "changeType", "FileReady") == 0) {
 256         log_error("cJSON_AddStringToObject failed\n");
 257         cJSON_Delete(file_ready_fields);
 258         return 0;
 259     }
 260
 261     if(cJSON_AddStringToObject(file_ready_fields, "notificationFieldsVersion", "2.0") == 0) {
 262         log_error("cJSON_AddStringToObject failed\n");
 263         cJSON_Delete(file_ready_fields);
 264         return 0;
 265     }
 266
 267     cJSON *array_of_named_hash_map = cJSON_CreateArray();
 268     if(array_of_named_hash_map == 0) {
 269         log_error("could not create JSON object\n");
 270         cJSON_Delete(file_ready_fields);
 271         return 0;
 272     }
 273
 274     if(cJSON_AddItemToObject(file_ready_fields, "arrayOfNamedHashMap", array_of_named_hash_map) == 0) {
 275         log_error("cJSON_AddStringToObject failed\n");
 276         cJSON_Delete(file_ready_fields);
 277         return 0;
 278     }
 279
 280     cJSON *additional_fields_entry = cJSON_CreateObject();
 281     if(additional_fields_entry == 0) {
 282         log_error("could not create JSON object\n");
 283         cJSON_Delete(file_ready_fields);
 284         return 0;
 285     }
 286
 287     char *filename = strrchr(file_location, '/');
 288
 289     if(filename == 0) {
 290         if(cJSON_AddStringToObject(additional_fields_entry, "name", "dummy_file.tar.gz") == 0) {
 291             log_error("cJSON_AddStringToObject failed\n");
 292             cJSON_Delete(file_ready_fields);
 293             return 0;
 294         }
 295     }
 296     else {
 297         if(cJSON_AddStringToObject(additional_fields_entry, "name", filename + 1) == 0) {
 298             log_error("cJSON_AddStringToObject failed\n");
 299             cJSON_Delete(file_ready_fields);
 300             return 0;
 301         }
 302     }
 303
 304     cJSON *hash_map = cJSON_CreateObject();
 305     if(hash_map == 0) {
 306         log_error("could not create JSON object\n");
 307         cJSON_Delete(file_ready_fields);
 308         return 0;
 309     }
 310
 311     if(cJSON_AddItemToObject(additional_fields_entry, "hashMap", hash_map) == 0) {
 312         log_error("cJSON_AddStringToObject failed\n");
 313         cJSON_Delete(file_ready_fields);
 314         return 0;
 315     }
 316
 317     if(cJSON_AddStringToObject(hash_map, "location", file_location) == 0) {
 318         log_error("cJSON_AddStringToObject failed\n");
 319         cJSON_Delete(file_ready_fields);
 320         return 0;
 321     }
 322
 323     if(cJSON_AddStringToObject(hash_map, "compression", "gzip") == 0) {
 324         log_error("cJSON_AddStringToObject failed\n");
 325         cJSON_Delete(file_ready_fields);
 326         return 0;
 327     }
 328
 329     if(cJSON_AddStringToObject(hash_map, "fileFormatType", "org.3GPP.32.435#measCollec") == 0) {
 330         log_error("cJSON_AddStringToObject failed\n");
 331         cJSON_Delete(file_ready_fields);
 332         return 0;
 333     }
 334
 335     if(cJSON_AddStringToObject(hash_map, "fileFormatVersion", "V5") == 0) {
 336         log_error("cJSON_AddStringToObject failed\n");
 337         cJSON_Delete(file_ready_fields);
 338         return 0;
 339     }
 340
 341     if(cJSON_AddItemToArray(array_of_named_hash_map, additional_fields_entry) == 0) {
 342         log_error("cJSON_AddStringToObject failed\n");
 343         cJSON_Delete(file_ready_fields);
 344         return 0;
 345     }
 346
 347     return file_ready_fields;
 348 }