Initial KEA DHCP Hook

Change-Id: I21c8184c8845e620400d0c2419b6ff16c7ffecc9
Signed-off-by: Brian Freeman <bf1936@att.com>

5 files changed, 317 insertions, 0 deletions

diff --git a/vnfs/vCPE/kea-sdnc-notify-mod/src/assembly/dep.xml b/vnfs/vCPE/kea-sdnc-notify-mod/src/assembly/dep.xml
new file mode 100644
index 00000000..d938e10a
--- /dev/null
+++ b/vnfs/vCPE/kea-sdnc-notify-mod/src/assembly/dep.xml
@@ -0,0 +1,28 @@
+<assembly xmlns="http://maven.apache.org/plugins/maven-assembly-plugin/assembly/1.1.2" 
+  xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+  xsi:schemaLocation="http://maven.apache.org/plugins/maven-assembly-plugin/assembly/1.1.2 http://maven.apache.org/xsd/assembly-1.1.2.xsd">
+  <id>demo</id>
+  <formats>
+    <format>tar.gz</format>
+  </formats>
+  <fileSets>
+
+    <fileSet>
+      <directory>${project.basedir}</directory>
+      <outputDirectory>/</outputDirectory>
+      <includes>
+	<include>**/*.md</include>
+	<include>**/*.cc</include>
+	<include>**/*.h</include>
+	<include>**/*.conf</include>
+	<include>**/*.sh</include>
+	<include>**/*.TXT</include>
+      </includes>
+      <excludes>
+	<exclude>build/*</exclude>
+	<exclude>test/*</exclude>
+      </excludes>
+    </fileSet>
+
+  </fileSets>
+</assembly>
diff --git a/vnfs/vCPE/kea-sdnc-notify-mod/src/library_common.h b/vnfs/vCPE/kea-sdnc-notify-mod/src/library_common.h
new file mode 100644
index 00000000..9b33b95e
--- /dev/null
+++ b/vnfs/vCPE/kea-sdnc-notify-mod/src/library_common.h
@@ -0,0 +1,35 @@
+/**************************************************************************//**
+* @file
+* common headers 
+*
+*
+* License
+* -------
+*
+* 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.
+*
+****************************************************************************/
+
+
+
+#ifndef LIBRARY_COMMON_H
+#define LIBRARY_COMMON_H
+#include <string>
+#include <iostream>
+#include <hooks/hooks.h>
+#include <boost/property_tree/ptree.hpp>
+#include <boost/property_tree/json_parser.hpp>
+
+extern std::string json_params[3];
+#endif 
diff --git a/vnfs/vCPE/kea-sdnc-notify-mod/src/load_unload.cc b/vnfs/vCPE/kea-sdnc-notify-mod/src/load_unload.cc
new file mode 100644
index 00000000..2296379c
--- /dev/null
+++ b/vnfs/vCPE/kea-sdnc-notify-mod/src/load_unload.cc
@@ -0,0 +1,47 @@
+/**************************************************************************//**
+* @file
+* load_unload the module in kea 
+*
+*
+* License
+* -------
+*
+* 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 "library_common.h"
+
+using namespace isc::hooks;
+namespace pt = boost::property_tree;
+
+std::string param_url;
+std::string json_params[3];
+
+extern "C" {
+int load(LibraryHandle& handle) {
+    pt::ptree root;
+    pt::read_json("/etc/kea/kea-sdnc-notify.conf", root);
+
+    json_params[0] = (root.get<std::string>("url"));
+    json_params[1] = (root.get<std::string>("siaddr"));
+
+    std::cout << "kea-sdnc-notify.so loaded\n" << std::endl;
+    return 0;
+}
+int unload() {
+    return (0);
+}
+}
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);
+}
+}
diff --git a/vnfs/vCPE/kea-sdnc-notify-mod/src/version.cc b/vnfs/vCPE/kea-sdnc-notify-mod/src/version.cc
new file mode 100644
index 00000000..ce71d758
--- /dev/null
+++ b/vnfs/vCPE/kea-sdnc-notify-mod/src/version.cc
@@ -0,0 +1,26 @@
+/****************************************************************************
+* @file
+* add hook
+* -------
+*
+* 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 <hooks/hooks.h>
+extern "C" {
+int version() {
+    return (KEA_HOOKS_VERSION);
+}
+}