浏览代码

Mejorado log

Guillermo Espinoza 6 年之前
父节点
当前提交
0a6281ca41
共有 1 个文件被更改,包括 99 次插入99 次删除
  1. 99 99
      hooks/kea-cm-hook/src/callouts.cc

+ 99 - 99
hooks/kea-cm-hook/src/callouts.cc

@@ -22,7 +22,7 @@ extern "C" {
 
 #include <numeric>    //inner_product
 #include <functional> //plus, equal_to, not2
-#include <string>   
+#include <string>
 #include <stdexcept>
 
 std::string getMacFromPacket(Pkt4Ptr);
@@ -30,60 +30,60 @@ std::string getMacFromPacket(Pkt4Ptr);
 enum HostType{ CABLEMODEM, EMTA, CPE};
 std::string HostClass[] = { "cm", "mta", "cpe"};
 
-HostType getHostType(Pkt4Ptr pkt){
+HostType getHostType(Pkt4Ptr pkt)
+{
     HostType rtr = CPE;
     std::string cm_mac = getMacFromPacket(pkt);
 
     //Non Empty MAC
-    if(cm_mac.size() != 0){
-
-	//std::string chwdr = pkt->getMAC(HWAddr::HWADDR_SOURCE_RAW)->toText(false);
-	std::string chwdr = pkt->getHWAddr()->toText(false);
-	isc::util::str::uppercase(chwdr);
-	chwdr.erase(std::remove(chwdr.begin(), chwdr.end(), ':'), chwdr.end());
-
-	std::string opt82mac = "";
-
-        if(pkt->getOption(DHO_DHCP_AGENT_OPTIONS)){
-	    opt82mac = pkt->getOption(DHO_DHCP_AGENT_OPTIONS)->toHexString(false).substr(6,12);
-
-	    std::vector<uint8_t> binary;
-	    isc::util::str::decodeFormattedHexString(opt82mac, binary);
-	}
-		LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("hwaddr: " + chwdr);
-		LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("remoteid: "+opt82mac);
-
-	if(chwdr == opt82mac){
-		rtr = CABLEMODEM;
-
-		LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("option 82 mac == chwdr ==> CABLEMODEM");
-	}else {
-		LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("dou't " + chwdr + " !== " + opt82mac);
-
-		if(pkt->getOption(DHO_VENDOR_CLASS_IDENTIFIER)){
-			std::string docsis_vendor_class = pkt->getOption(DHO_VENDOR_CLASS_IDENTIFIER)->toString().substr(0,6);
-			std::string pktc_vendor_class = pkt->getOption(DHO_VENDOR_CLASS_IDENTIFIER)->toString().substr(0,4);
-
-			LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("Checking Vendor Class ID");
-			if(docsis_vendor_class.compare("docsis") == 0){
-				rtr = CABLEMODEM;
-				LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("docsis en vendor class ==> CABLEMODEM");
-			} else {
-				if(pktc_vendor_class.compare("pktc") == 0){
-					rtr = EMTA;
-					LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("pktc en vendor class ==> EMTA");
-				} else {
-					LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("vendor class unknown");
-				}
-			}
-		}else{
-					LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("no vendor class");
-		}
-	}
+    if (cm_mac.size() != 0) {
+    	//std::string chwdr = pkt->getMAC(HWAddr::HWADDR_SOURCE_RAW)->toText(false);
+    	std::string chwdr = pkt->getHWAddr()->toText(false);
+    	isc::util::str::uppercase(chwdr);
+    	chwdr.erase(std::remove(chwdr.begin(), chwdr.end(), ':'), chwdr.end());
+
+    	std::string opt82mac = "";
+
+        if (pkt->getOption(DHO_DHCP_AGENT_OPTIONS)) {
+    	    opt82mac = pkt->getOption(DHO_DHCP_AGENT_OPTIONS)->toHexString(false).substr(6,12);
+
+    	    std::vector<uint8_t> binary;
+    	    isc::util::str::decodeFormattedHexString(opt82mac, binary);
+    	}
+		LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("getHostType -> hwaddr: " + chwdr);
+		LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("getHostType -> remoteid: " + opt82mac);
+
+    	if (chwdr == opt82mac) {
+    		rtr = CABLEMODEM;
+
+    		LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("getHostType -> option 82 mac == chwdr ==> CABLEMODEM");
+    	} else {
+    		LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("getHostType -> dou't " + chwdr + " !== " + opt82mac);
+
+    		if (pkt->getOption(DHO_VENDOR_CLASS_IDENTIFIER)) {
+    			std::string docsis_vendor_class = pkt->getOption(DHO_VENDOR_CLASS_IDENTIFIER)->toString().substr(0,6);
+    			std::string pktc_vendor_class = pkt->getOption(DHO_VENDOR_CLASS_IDENTIFIER)->toString().substr(0,4);
+
+    			LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("getHostType -> Checking Vendor Class ID");
+    			if (docsis_vendor_class.compare("docsis") == 0) {
+    				rtr = CABLEMODEM;
+    				LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("getHostType -> docsis en vendor class ==> CABLEMODEM");
+    			} else {
+    				if (pktc_vendor_class.compare("pktc") == 0) {
+    					rtr = EMTA;
+    					LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("getHostType -> pktc en vendor class ==> EMTA");
+    				} else {
+    					LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("getHostType -> vendor class unknown");
+    				}
+    			}
+    		} else {
+                LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("getHostType -> no vendor class");
+    		}
+    	}
     } else {
-	    LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("Empty MAC fomr getMacFromPacket ==> CPE " + std::to_string(cm_mac.size()));
-	    if(pkt->isRelayed()){
-		    LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("Pakcet Relayed");
+	    LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("getHostType -> Empty MAC fomr getMacFromPacket ==> CPE " + std::to_string(cm_mac.size()));
+	    if (pkt->isRelayed()) {
+		    LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("getHostType -> Packet Relayed");
 	    }
 	    //LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("HWAddr " + pkt->getMAC(HWAddr::HWADDR_SOURCE_ANY)->toText());
     }
@@ -94,67 +94,69 @@ HostType getHostType(Pkt4Ptr pkt){
 /**
  * 	Get the cablemodem mac from (option 82) or if vendor_class is *docsis* get the c_hwaddr
  */
-std::string getMacFromPacket(Pkt4Ptr pkt){
+std::string getMacFromPacket(Pkt4Ptr pkt)
+{
     std::string rtr = "";
-    if(pkt->getOption(DHO_DHCP_AGENT_OPTIONS)){
-	    //rtr = pkt->getOption(DHO_DHCP_AGENT_OPTIONS)->toHexString(false).substr(26,12);
+    if (pkt->getOption(DHO_DHCP_AGENT_OPTIONS)) {
+        //rtr = pkt->getOption(DHO_DHCP_AGENT_OPTIONS)->toHexString(false).substr(26,12);
 	    rtr = pkt->getOption(DHO_DHCP_AGENT_OPTIONS)->toHexString(false).substr(6,12);
 
 	    std::vector<uint8_t> binary;
 	    isc::util::str::decodeFormattedHexString(rtr, binary);
 
-	    LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("getMacFromPacket... from option 82 " + rtr);
-    }else if(pkt->getOption(DHO_VENDOR_CLASS_IDENTIFIER)){
-	std::string docsis_vendor_class = pkt->getOption(DHO_VENDOR_CLASS_IDENTIFIER)->toString().substr(2,6);
-	if(docsis_vendor_class.compare("docsis") == 0){
-		rtr = pkt->getMAC(HWAddr::HWADDR_SOURCE_RAW)->toText(false);
-		isc::util::str::uppercase(rtr);
-		rtr.erase(std::remove(rtr.begin(), rtr.end(), ':'), rtr.end());
-		LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("getMacFromPacket is vendor class 'docsis' chwaddr..." + rtr);
-	}
+	    LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("getMacFromPacket -> from option 82 " + rtr);
+    } else if(pkt->getOption(DHO_VENDOR_CLASS_IDENTIFIER)) {
+        std::string docsis_vendor_class = pkt->getOption(DHO_VENDOR_CLASS_IDENTIFIER)->toString().substr(2,6);
+        if (docsis_vendor_class.compare("docsis") == 0) {
+            rtr = pkt->getMAC(HWAddr::HWADDR_SOURCE_RAW)->toText(false);
+            isc::util::str::uppercase(rtr);
+            rtr.erase(std::remove(rtr.begin(), rtr.end(), ':'), rtr.end());
+            LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("getMacFromPacket -> is vendor class 'docsis' chwaddr..." + rtr);
+        }
     }
 
     return rtr;
 }
 
 /* IPv4 callouts */
-int pkt4_receive(CalloutHandle& handle) {
+int pkt4_receive(CalloutHandle& handle)
+{
     Pkt4Ptr query;
     handle.getArgument("query4", query);
 
     std::vector<std::string> list_class;
 
-
     std::string mac = getMacFromPacket(query);
     HWAddrPtr remote_id;
-   
+
     HostType ht = getHostType(query);
-	
+
     list_class.push_back(HostClass[ht]);
     std::map<std::string,std::string>::iterator it = cm_map.find(mac);
-    if (it != cm_map.end()){
-	list_class.push_back(cm_map[mac]);
+    if (it != cm_map.end()) {
+    	list_class.push_back(cm_map[mac]);
     }
 
     std::string me, me2;
 
     BOOST_FOREACH(me, list_class) {
-         LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("class  : " + me);
+        LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("pkt4_receive -> class  : " + me);
 
-         BOOST_FOREACH(me2, list_class) {
-         	if(me.compare(me2) != 0){
-         		LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("class  : " + me + "-" + me2);
+        BOOST_FOREACH(me2, list_class) {
+         	if (me.compare(me2) != 0) {
+                LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("pkt4_receive -> class  : " + me + "-" + me2);
          		query->addClass(me + "-" + me2);
          	}
-         }
+        }
 
-         query->addClass(me);
+        query->addClass(me);
     }
 
     return 0;
 }
 
-int pkt4_send(CalloutHandle& handle) {
+int pkt4_send(CalloutHandle& handle)
+{
     Pkt4Ptr response;
     Pkt4Ptr query;
     handle.getArgument("response4", response);
@@ -164,10 +166,10 @@ int pkt4_send(CalloutHandle& handle) {
     std::string mac;
 
     HostType ht = getHostType(query);
-    LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("ht : " + std::to_string(ht));
+    LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("pkt4_send -> HostType : " + std::to_string(ht));
 
-    if(ht == CABLEMODEM or ht == EMTA){
-	mac = getMacFromPacket(query);
+    if (ht == CABLEMODEM or ht == EMTA) {
+        mac = getMacFromPacket(query);
     }
 
     //OptionStringPtr asc1(new OptionString(Option::V4, 43, "http://asc-test.org:7547/"));
@@ -177,53 +179,51 @@ int pkt4_send(CalloutHandle& handle) {
     //response->addOption(asc2);
 
 
-    if(ht == CABLEMODEM){
+    if (ht == CABLEMODEM) {
 	    std::string file = mac + ".bin";
 	    isc::util::str::lowercase(file);
 
-	    LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("file : " + file);
+	    LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("pkt4_send -> CM file : " + file);
 	    strcpy((char*)ufile_name, file.c_str());
-    	    response->setFile(ufile_name, strlen((char *)ufile_name));
+        response->setFile(ufile_name, strlen((char *)ufile_name));
 
 	    std::map<std::string,std::string>::iterator it = option122_map.find(mac);
 
-	    if (it != option122_map.end()){
-                  LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("cablemodem " + mac + " found for MTA Activation on " + option122_map[mac]);
-		  OptionPtr option1(new Option(Option::V4, 122));
-
+	    if (it != option122_map.end()) {
+            LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("pkt4_send -> cablemodem " + mac + " found for MTA Activation on " + option122_map[mac]);
+            OptionPtr option1(new Option(Option::V4, 122));
 
- 		  Option4AddrLstPtr opt_server(new Option4AddrLst(1));
-		  opt_server->setAddress(isc::asiolink::IOAddress("255.255.255.255"));//option122_map[mac]));
-	
- 		  OptionStringPtr opt_basic1(new OptionString(Option::V4, 6, "BASIC.1"));
+            Option4AddrLstPtr opt_server(new Option4AddrLst(1));
+            opt_server->setAddress(isc::asiolink::IOAddress("255.255.255.255"));//option122_map[mac]));
 
-		  option1->addOption(opt_server);
-		  option1->addOption(opt_basic1);
+            OptionStringPtr opt_basic1(new OptionString(Option::V4, 6, "BASIC.1"));
 
-		  response->addOption(option1);
-	    }else{
+            option1->addOption(opt_server);
+            option1->addOption(opt_basic1);
 
-            	  LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("cablemodem " + mac + " not found for MTA Activation");
+            response->addOption(option1);
+	    } else {
+            LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("pkt4_send -> cablemodem " + mac + " not found for MTA Activation");
 	    }
     }
 
-    if(ht == EMTA){
+    if (ht == EMTA) {
 	    std::string file = mac + ".mta.bin";
 	    isc::util::str::lowercase(file);
 
 	    std::map<std::string,std::string>::iterator it = option122_map.find(mac);
 
-	    if (it != option122_map.end()){
+	    if (it != option122_map.end()) {
 	        std::string file = mac + ".mta.bin";
 	        isc::util::str::lowercase(file);
 
-	        LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("file : " + file);
+	        LOG_INFO(runscript_logger, FLOWDAT_DEBUG_STRING).arg("pkt4_send -> EMTA file : " + file);
 	        strcpy((char*)ufile_name, file.c_str());
 	        response->setFile(ufile_name, strlen((char *)ufile_name));
 
 	        isc::util::str::lowercase(mac);
- 		OptionStringPtr opt_domain(new OptionString(Option::V4, DHO_HOST_NAME, mac));
-		response->addOption(opt_domain);
+            OptionStringPtr opt_domain(new OptionString(Option::V4, DHO_HOST_NAME, mac));
+            response->addOption(opt_domain);
 	    }
     }