diff options
Diffstat (limited to 'vnfs/vCPE/kea-sdnc-notify-mod/src/pkt4_send.cc')
| -rw-r--r-- | vnfs/vCPE/kea-sdnc-notify-mod/src/pkt4_send.cc | 181 |
1 files changed, 181 insertions, 0 deletions
diff --git a/vnfs/vCPE/kea-sdnc-notify-mod/src/pkt4_send.cc b/vnfs/vCPE/kea-sdnc-notify-mod/src/pkt4_send.cc new file mode 100644 index 00000000..231a0a8b --- /dev/null +++ b/vnfs/vCPE/kea-sdnc-notify-mod/src/pkt4_send.cc @@ -0,0 +1,181 @@ +/***************************************************************************** +* @file +* main function to process ipv4 packets +* ------- +* +* Copyright (C) 2017 AT&T Intellectual Property. All rights reserved. +* +* 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. +* +****************************************************************************/ + +#include <dhcp/pkt4.h> +#include <log/logger.h> +#include <log/macros.h> +#include <log/message_initializer.h> +#include "library_common.h" +#include <curl/curl.h> +#include <sstream> + +using namespace isc::dhcp; +using namespace isc::hooks; +using namespace std; +using namespace isc::log; +namespace pt = boost::property_tree; + + +isc::log::Logger logger("sdnc-notify-logger"); +const char* log_messages[] = { + "SNL_BASE", "message: %1", + "SNL_PKT_SEND", "Outgoing packet: \n%1", + "SNL_CURL_RECEIVED", "Received: \n%1", + "SNL_CURL_FAILED", "Unable to receive data from %1", + NULL +}; + +/// Initializer for log messages. +const MessageInitializer message_initializer(log_messages); + +extern "C" { +int pkt4_send(CalloutHandle& handle) { + + // Curl variables + CURL *curl; + CURLcode res; + struct curl_slist *list=NULL; + bool perform_updates = 0; + int curl_opt_res = 0; + + // "C" variables. + char *bp; + size_t size; + FILE *response_memfile; + + // String variables + string hwaddr; + string final_url; + string msg_name; + string yiaddr; + string post_data; + + + // Pkt4 variables. + Pkt4Ptr response4_ptr; + HWAddrPtr hwaddr_ptr; + + // Boost / json related variables. + pt::ptree root; + std::stringstream ss; + boost::optional<std::string> siaddr_json_field; + + /* Begin Code */ + handle.getArgument("response4", response4_ptr); + hwaddr_ptr = response4_ptr->getHWAddr(); + isc::asiolink::IOAddress new_yiaddr(response4_ptr->getYiaddr()); + hwaddr = hwaddr_ptr->toText(false); + yiaddr = new_yiaddr.toText(); + msg_name = response4_ptr->getName(); + post_data = "{ macaddr=" + hwaddr + "&yiaddr=" + yiaddr + "&msg=" + msg_name + "}"; + final_url = json_params[0] + hwaddr; + LOG_DEBUG(logger, 0, "SNL_BASE").arg(final_url); + LOG_DEBUG(logger, 0, "SNL_BASE").arg(yiaddr); + + + if ( msg_name == "DHCPACK") + perform_updates = 1; + + if (!perform_updates) { + LOG_WARN(logger, "SNL_BASE").arg("Nothing to update."); + return(0); + } + + curl_global_init(CURL_GLOBAL_DEFAULT); + curl = curl_easy_init(); + if(!curl) { + curl_global_cleanup(); + LOG_ERROR(logger, "SNL_BASE").arg("Could not initialize curl"); + return(1); + } + + list = curl_slist_append(list, "Accept: application/json"); + if (list == NULL) { + curl_easy_cleanup(curl); + curl_global_cleanup(); + LOG_ERROR(logger, "SNL_BASE").arg("Could not create curl slist."); + return(1); + } + + response_memfile = open_memstream (&bp, &size); + if (response_memfile == NULL) { + curl_easy_cleanup(curl); + curl_global_cleanup(); + LOG_ERROR(logger, "SNL_BASE").arg("Could not create memfile."); + return(1); + } + + // DEBUGGING + curl_opt_res += curl_easy_setopt(curl, CURLOPT_VERBOSE, 1L); + curl_opt_res += curl_easy_setopt(curl, CURLOPT_HEADER, 1L); + + // If we don't set a timeout, curl will try for 300 seconds by default. + curl_opt_res += curl_easy_setopt(curl, CURLOPT_TIMEOUT, 1L); + curl_opt_res += curl_easy_setopt(curl, CURLOPT_CONNECTTIMEOUT, 1L); + // libcurl's docs say to cast as void, don't blame me. + curl_opt_res += curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void *)response_memfile); + // CURLOPT_URL takes a char* + curl_opt_res += curl_easy_setopt(curl, CURLOPT_URL, (final_url).c_str()); + + // let curl use strlen + curl_opt_res += curl_easy_setopt(curl, CURLOPT_POSTFIELDSIZE, -1L ); + curl_opt_res += curl_easy_setopt(curl, CURLOPT_POSTFIELDS, post_data.c_str() ); + curl_opt_res += curl_easy_setopt(curl, CURLOPT_HTTPHEADER, list); + + #ifdef SKIP_PEER_VERIFICATION + curl_opt_res += curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, 0L); + #endif + + #ifdef SKIP_HOSTNAME_VERIFICATION + curl_opt_res += curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, 0L); + #endif + if (curl_opt_res > 0) { + fclose(response_memfile); + free(bp); + curl_easy_cleanup(curl); + curl_global_cleanup(); + LOG_ERROR(logger, "SNL_BASE").arg("Error setting curl options."); + return(1); + } + res = curl_easy_perform(curl); + /* Check for errors */ + if(res != CURLE_OK) { + fclose(response_memfile); + free(bp); + curl_easy_cleanup(curl); + curl_global_cleanup(); + LOG_ERROR(logger, "SNL_CURL_FAILED").arg(ss.str()); + return(1); + } + // make bp available for reading. + fclose(response_memfile); + curl_easy_cleanup(curl); + curl_global_cleanup(); + ss << bp; + free(bp); + LOG_DEBUG(logger, 0, "SNL_CURL_RECEIVED").arg(ss.str()); + // Load the json file in this ptree + + + LOG_DEBUG(logger, 0, "SNL_PKT_SEND").arg(response4_ptr->toText()); + + return(0); +} +} |