[7885][7872] wlbt : Enhanced logGing support

HAL changes to support enhanced logGing.
SCSC-Bug-Id:SSB-25788

Change-Id: I9573e7f7ccda8f5c22ce857e2dd8ba2fd90cab08
Signed-off-by: Himani Gupta <himani.g2@samsung.com>
diff --git a/Android.mk b/Android.mk
index bb391ce..739ff68 100755
--- a/Android.mk
+++ b/Android.mk
@@ -27,7 +27,8 @@
 	gscan.cpp \
 	link_layer_stats.cpp \
 	wifi_offload.cpp \
-	roam.cpp
+	roam.cpp \
+	wifi_logger.cpp
 
 LOCAL_MODULE := libwifi-hal-slsi
 
diff --git a/common.h b/common.h
index 164db79..48800ac 100755
--- a/common.h
+++ b/common.h
@@ -123,6 +123,10 @@
     ANDROID_NL80211_SUBCMD_LSTATS_RANGE_START = 0x1200,
     ANDROID_NL80211_SUBCMD_LSTATS_RANGE_END   = 0x12FF,
 
+    /* define all Logger related commands between 0x1400 and 0x14FF */
+    ANDROID_NL80211_SUBCMD_DEBUG_RANGE_START = 0x1400,
+    ANDROID_NL80211_SUBCMD_DEBUG_RANGE_END   = 0x14FF,
+
     /* define all wifi offload related commands between 0x1400 and 0x14FF */
     ANDROID_NL80211_SUBCMD_WIFI_OFFLOAD_RANGE_START = 0x1400,
     ANDROID_NL80211_SUBCMD_WIFI_OFFLOAD_RANGE_END   = 0x14FF,
@@ -172,7 +176,9 @@
     WIFI_HANGED_EVENT,
     WIFI_EPNO_EVENT,
     WIFI_HOTSPOT_MATCH,
-    WIFI_RSSI_REPORT_EVENT
+    WIFI_RSSI_REPORT_EVENT,
+    ENHANCE_LOGGER_RING_EVENT,
+    ENHANCE_LOGGER_MEM_DUMP_EVENT
 
 } WIFI_EVENT;
 
@@ -254,5 +260,13 @@
 #define min(x, y)       ((x) < (y) ? (x) : (y))
 #define max(x, y)       ((x) > (y) ? (x) : (y))
 
+#define NULL_CHECK_RETURN(ptr, str, ret) \
+    do { \
+        if (!(ptr)) { \
+            ALOGE("%s(): null pointer - #ptr (%s)\n", __FUNCTION__, str); \
+            return ret; \
+        } \
+    } while (0)
+
 #endif
 
diff --git a/wifi_hal.cpp b/wifi_hal.cpp
index 1c3564a..951bac5 100755
--- a/wifi_hal.cpp
+++ b/wifi_hal.cpp
@@ -62,6 +62,8 @@
 static wifi_error wifi_start_rssi_monitoring(wifi_request_id id, wifi_interface_handle
                         iface, s8 max_rssi, s8 min_rssi, wifi_rssi_event_handler eh);
 static wifi_error wifi_stop_rssi_monitoring(wifi_request_id id, wifi_interface_handle iface);
+wifi_error (*wifi_get_wake_reason_stats)(wifi_interface_handle iface,
+                        WLAN_DRIVER_WAKE_REASON_CNT *wifi_wake_reason_cnt);
 
 /* Initialize/Cleanup */
 
@@ -132,6 +134,20 @@
     fn->wifi_configure_roaming = wifi_configure_roaming;
     fn->wifi_configure_nd_offload = wifi_configure_nd_offload;
     fn->wifi_get_packet_filter_capabilities = wifi_get_packet_filter_capabilities;
+    fn->wifi_start_pkt_fate_monitoring = wifi_start_pkt_fate_monitoring;
+    fn->wifi_get_tx_pkt_fates = wifi_get_tx_pkt_fates;
+    fn->wifi_get_rx_pkt_fates = wifi_get_rx_pkt_fates;
+    fn->wifi_start_logging = wifi_start_logging;
+    fn->wifi_set_log_handler = wifi_set_log_handler;
+    fn->wifi_set_alert_handler= wifi_set_alert_handler;
+    fn->wifi_get_ring_buffers_status = wifi_get_ring_buffers_status;
+    fn->wifi_get_logger_supported_feature_set = wifi_get_logger_supported_feature_set;
+    fn->wifi_get_ring_data = wifi_get_ring_data;
+    fn->wifi_get_driver_version = wifi_get_driver_version;
+    fn->wifi_get_firmware_version = wifi_get_firmware_version;
+    fn->wifi_get_firmware_memory_dump = wifi_get_firmware_memory_dump;
+    fn->wifi_get_driver_memory_dump = wifi_get_driver_memory_dump;
+    fn->wifi_get_wake_reason_stats = wifi_get_wake_reason_stats;
 
     return WIFI_SUCCESS;
 }
diff --git a/wifi_logger.cpp b/wifi_logger.cpp
new file mode 100755
index 0000000..9ca3e13
--- /dev/null
+++ b/wifi_logger.cpp
@@ -0,0 +1,1428 @@
+#include <stdint.h>

+#include <fcntl.h>

+#include <sys/socket.h>

+#include <netlink/genl/genl.h>

+#include <netlink/genl/family.h>

+#include <netlink/genl/ctrl.h>

+#include <linux/rtnetlink.h>

+#include <netpacket/packet.h>

+#include <linux/filter.h>

+#include <linux/errqueue.h>

+

+#include <linux/pkt_sched.h>

+#include <netlink/object-api.h>

+#include <netlink/netlink.h>

+#include <netlink/socket.h>

+#include <netlink-private/object-api.h>

+#include <netlink-private/types.h>

+

+#include "nl80211_copy.h"

+#include "sync.h"

+

+#define LOG_TAG  "WifiHAL"

+

+#include <utils/Log.h>

+

+#include "wifi_hal.h"

+#include "common.h"

+#include "cpp_bindings.h"

+

+using namespace android;

+

+typedef enum {

+    ENHANCE_LOGGER_START_LOGGING = ANDROID_NL80211_SUBCMD_DEBUG_RANGE_START,

+    ENHANCE_LOGGER_TRIGGER_FW_MEM_DUMP,

+    ENHANCE_LOGGER_GET_FW_MEM_DUMP,

+    ENHANCE_LOGGER_GET_VER,

+    ENHANCE_LOGGER_GET_RING_STATUS,

+    ENHANCE_LOGGER_GET_RING_DATA,

+    ENHANCE_LOGGER_GET_FEATURE,

+    ENHANCE_LOGGER_RESET_LOGGING,

+    ENHANCE_LOGGER_TRIGGER_DRIVER_MEM_DUMP,

+    ENHANCE_LOGGER_GET_DRIVER_MEM_DUMP,

+    ENHANCE_LOGGER_START_PKT_FATE_MONITORING,

+    ENHANCE_LOGGER_GET_TX_PKT_FATES,

+    ENHANCE_LOGGER_GET_RX_PKT_FATES,

+    ENHANCE_LOGGER_GET_WAKE_REASON_STATS,

+} DEBUG_SUB_COMMAND;

+

+typedef enum {

+    ENHANCE_LOGGER_ATTRIBUTE_DRIVER_VER,

+    ENHANCE_LOGGER_ATTRIBUTE_FW_VER,

+    ENHANCE_LOGGER_ATTRIBUTE_RING_ID,

+    ENHANCE_LOGGER_ATTRIBUTE_RING_NAME,

+    ENHANCE_LOGGER_ATTRIBUTE_RING_FLAGS,

+    ENHANCE_LOGGER_ATTRIBUTE_LOG_LEVEL,

+    ENHANCE_LOGGER_ATTRIBUTE_LOG_TIME_INTVAL,

+    ENHANCE_LOGGER_ATTRIBUTE_LOG_MIN_DATA_SIZE,

+    ENHANCE_LOGGER_ATTRIBUTE_FW_DUMP_LEN,

+    ENHANCE_LOGGER_ATTRIBUTE_FW_DUMP_DATA,

+    // LOGGER_ATTRIBUTE_FW_ERR_CODE,

+    ENHANCE_LOGGER_ATTRIBUTE_RING_DATA,

+    ENHANCE_LOGGER_ATTRIBUTE_RING_STATUS,

+    ENHANCE_LOGGER_ATTRIBUTE_RING_NUM,

+    ENHANCE_LOGGER_ATTRIBUTE_DRIVER_DUMP_LEN,

+    ENHANCE_LOGGER_ATTRIBUTE_DRIVER_DUMP_DATA,

+    ENHANCE_LOGGER_ATTRIBUTE_PKT_FATE_NUM,

+    ENHANCE_LOGGER_ATTRIBUTE_PKT_FATE_DATA,

+    //Attributes for data used by wake stats subcmd.

+    ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_INVALID = 0,

+    ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_TOTAL_CMD_EVENT_WAKE,

+    ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_CMD_EVENT_WAKE_CNT_PTR,

+    ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_CMD_EVENT_WAKE_CNT_SZ,

+    ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_TOTAL_DRIVER_FW_LOCAL_WAKE,

+    ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_DRIVER_FW_LOCAL_WAKE_CNT_PTR,

+    ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_DRIVER_FW_LOCAL_WAKE_CNT_SZ,

+    ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_TOTAL_RX_DATA_WAKE,

+    ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_RX_UNICAST_CNT,

+    ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_RX_MULTICAST_CNT,

+    ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_RX_BROADCAST_CNT,

+    ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_ICMP_PKT,

+    ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_ICMP6_PKT,

+    ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_ICMP6_RA,

+    ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_ICMP6_NA,

+    ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_ICMP6_NS,

+    ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_ICMP4_RX_MULTICAST_CNT,

+    ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_ICMP6_RX_MULTICAST_CNT,

+    ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_OTHER_RX_MULTICAST_CNT,

+} ENHANCE_LOGGER_ATTRIBUTE;

+

+typedef enum {

+    DEBUG_OFF = 0,

+    DEBUG_NORMAL,

+    DEBUG_VERBOSE,

+    DEBUG_VERY,

+    DEBUG_VERY_VERY,

+} ENHANCE_LOGGER_LEVEL;

+

+typedef enum {

+    GET_FW_VER,

+    GET_DRV_VER,

+    GET_RING_DATA,

+    GET_RING_STATUS,

+    GET_FEATURE,

+    START_RING_LOG,

+    GET_FW_DUMP,

+    GET_DRIVER_DUMP,

+} GetCmdType;

+

+typedef enum {

+    PACKET_MONITOR_START,

+    TX_PACKET_FATE,

+    RX_PACKET_FATE,

+} PktFateReqType;

+

+class DebugCommand : public WifiCommand

+{

+    char *mBuff;

+    int *mBuffSize;

+    u32 *mNumRings;

+    wifi_ring_buffer_status *mStatus;

+    unsigned int *mSupport;

+    u32 mVerboseLevel;

+    u32 mFlags;

+    u32 mMaxIntervalSec;

+    u32 mMinDataSize;

+    char *mRingName;

+    GetCmdType mType;

+

+public:

+

+    // constructor for get version

+    DebugCommand(wifi_interface_handle iface, char *buffer, int *buffer_size,

+            GetCmdType cmdType)

+        : WifiCommand(iface, 0), mBuff(buffer), mBuffSize(buffer_size), mType

+        (cmdType)

+    {

+        memset(mBuff, 0, *mBuffSize);

+    }

+

+    // constructor for ring data

+    DebugCommand(wifi_interface_handle iface, char *ring_name, GetCmdType cmdType)

+        : WifiCommand(iface, 0), mRingName(ring_name), mType(cmdType)

+    { }

+

+    // constructor for ring status

+    DebugCommand(wifi_interface_handle iface, u32 *num_rings,

+            wifi_ring_buffer_status *status, GetCmdType cmdType)

+        : WifiCommand(iface, 0), mNumRings(num_rings), mStatus(status), mType(cmdType)

+    {

+        memset(mStatus, 0, sizeof(wifi_ring_buffer_status) * (*mNumRings));

+    }

+

+    // constructor for feature set

+    DebugCommand(wifi_interface_handle iface, unsigned int *support, GetCmdType cmdType)

+        : WifiCommand(iface, 0), mSupport(support), mType(cmdType)

+    { }

+

+    // constructor for ring params

+    DebugCommand(wifi_interface_handle iface, u32 verbose_level, u32 flags,

+            u32 max_interval_sec, u32 min_data_size, char *ring_name, GetCmdType cmdType)

+        : WifiCommand(iface, 0), mVerboseLevel(verbose_level), mFlags(flags),

+        mMaxIntervalSec(max_interval_sec), mMinDataSize(min_data_size),

+        mRingName(ring_name), mType(cmdType)

+    { }

+

+    int createRingRequest(WifiRequest& request) {

+        int result = request.create(GOOGLE_OUI, ENHANCE_LOGGER_START_LOGGING);

+        if (result != WIFI_SUCCESS) {

+            ALOGE("Failed to create start ring logger request; result = %d", result);

+            return result;

+        }

+

+        nlattr *data = request.attr_start(NL80211_ATTR_VENDOR_DATA);

+

+        result = request.put_u32(ENHANCE_LOGGER_ATTRIBUTE_LOG_LEVEL, mVerboseLevel);

+        if (result != WIFI_SUCCESS) {

+            ALOGE("Failed to put log level; result = %d", result);

+            return result;

+        }

+        result = request.put_u32(ENHANCE_LOGGER_ATTRIBUTE_RING_FLAGS, mFlags);

+        if (result != WIFI_SUCCESS) {

+            ALOGE("Failed to put ring flags; result = %d", result);

+            return result;

+        }

+        result = request.put_u32(ENHANCE_LOGGER_ATTRIBUTE_LOG_TIME_INTVAL, mMaxIntervalSec);

+        if (result != WIFI_SUCCESS) {

+            ALOGE("Failed to put log time interval; result = %d", result);

+            return result;

+        }

+        result = request.put_u32(ENHANCE_LOGGER_ATTRIBUTE_LOG_MIN_DATA_SIZE, mMinDataSize);

+        if (result != WIFI_SUCCESS) {

+            ALOGE("Failed to put min data size; result = %d", result);

+            return result;

+        }

+        result = request.put_string(ENHANCE_LOGGER_ATTRIBUTE_RING_NAME, mRingName);

+        if (result != WIFI_SUCCESS) {

+            ALOGE("Failed to put ringbuffer name; result = %d", result);

+            return result;

+        }

+        request.attr_end(data);

+

+        return WIFI_SUCCESS;

+    }

+

+    int createRequest(WifiRequest &request) {

+        int result;

+

+        switch (mType) {

+            case GET_FW_VER:

+            {

+                result = request.create(GOOGLE_OUI, ENHANCE_LOGGER_GET_VER);

+                if (result != WIFI_SUCCESS) {

+                    ALOGE("Failed to create get fw version request; result = %d", result);

+                    return result;

+                }

+

+                nlattr *data = request.attr_start(NL80211_ATTR_VENDOR_DATA);

+

+                // Driver expecting only attribute type, passing mbuff as data with

+                // length 0 to avoid undefined state

+                result = request.put(ENHANCE_LOGGER_ATTRIBUTE_FW_VER, mBuff, 0);

+                if (result != WIFI_SUCCESS) {

+                    ALOGE("Failed to put get fw version request; result = %d", result);

+                    return result;

+                }

+                request.attr_end(data);

+                break;

+            }

+

+            case GET_DRV_VER:

+            {

+                result = request.create(GOOGLE_OUI, ENHANCE_LOGGER_GET_VER);

+                if (result != WIFI_SUCCESS) {

+                    ALOGE("Failed to create get drv version request; result = %d", result);

+                    return result;

+                }

+

+                nlattr *data = request.attr_start(NL80211_ATTR_VENDOR_DATA);

+

+                // Driver expecting only attribute type, passing mbuff as data with

+                // length 0 to avoid undefined state

+                result = request.put(ENHANCE_LOGGER_ATTRIBUTE_DRIVER_VER, mBuff, 0);

+

+                if (result != WIFI_SUCCESS) {

+                    ALOGE("Failed to put get drv version request; result = %d", result);

+                    return result;

+                }

+                request.attr_end(data);

+                break;

+            }

+

+            case GET_RING_DATA:

+            {

+                result = request.create(GOOGLE_OUI, ENHANCE_LOGGER_GET_RING_DATA);

+                if (result != WIFI_SUCCESS) {

+                    ALOGE("Failed to create get ring data request; result = %d", result);

+                    return result;

+                }

+

+                nlattr *data = request.attr_start(NL80211_ATTR_VENDOR_DATA);

+                result = request.put_string(ENHANCE_LOGGER_ATTRIBUTE_RING_NAME, mRingName);

+                if (result != WIFI_SUCCESS) {

+                    ALOGE("Failed to put ring data request; result = %d", result);

+                    return result;

+                }

+                request.attr_end(data);

+                break;

+            }

+

+            case GET_RING_STATUS:

+            {

+                result = request.create(GOOGLE_OUI, ENHANCE_LOGGER_GET_RING_STATUS);

+                if (result != WIFI_SUCCESS) {

+                    ALOGE("Failed to create get ring status request; result = %d", result);

+                    return result;

+                }

+                break;

+            }

+

+            case GET_FEATURE:

+            {

+                result = request.create(GOOGLE_OUI, ENHANCE_LOGGER_GET_FEATURE);

+                if (result != WIFI_SUCCESS) {

+                    ALOGE("Failed to create get feature request; result = %d", result);

+                    return result;

+                }

+                break;

+            }

+

+            case START_RING_LOG:

+                result = createRingRequest(request);

+                break;

+

+            default:

+                ALOGE("Unknown Debug command");

+                result = WIFI_ERROR_UNKNOWN;

+        }

+        return result;

+    }

+

+    int start() {

+        // ALOGD("Start debug command");

+        WifiRequest request(familyId(), ifaceId());

+        int result = createRequest(request);

+        if (result != WIFI_SUCCESS) {

+            ALOGE("Failed to create debug request; result = %d", result);

+            return result;

+        }

+

+        result = requestResponse(request);

+        if (result != WIFI_SUCCESS) {

+            ALOGE("Failed to register debug response; result = %d", result);

+        }

+        return result;

+    }

+

+    virtual int handleResponse(WifiEvent& reply) {

+        ALOGD("In DebugCommand::handleResponse");

+

+        if (reply.get_cmd() != NL80211_CMD_VENDOR) {

+            ALOGD("Ignoring reply with cmd = %d", reply.get_cmd());

+            return NL_SKIP;

+        }

+

+        switch (mType) {

+            case GET_DRV_VER:

+            case GET_FW_VER:

+            {

+                void *data = reply.get_vendor_data();

+                int len = reply.get_vendor_data_len();

+

+                ALOGD("len = %d, expected len = %d", len, *mBuffSize);

+                memcpy(mBuff, data, min(len, *mBuffSize));

+                if (*mBuffSize < len)

+                    return NL_SKIP;

+                *mBuffSize = len;

+                break;

+            }

+

+            case START_RING_LOG:

+            case GET_RING_DATA:

+                break;

+

+            case GET_RING_STATUS:

+            {

+                nlattr *vendor_data = reply.get_attribute(NL80211_ATTR_VENDOR_DATA);

+                int len = reply.get_vendor_data_len();

+                wifi_ring_buffer_status *status(mStatus);

+

+                if (vendor_data == NULL || len == 0) {

+                    ALOGE("No Debug data found");

+                    return NL_SKIP;

+                }

+

+                nl_iterator it(vendor_data);

+                if (it.get_type() == ENHANCE_LOGGER_ATTRIBUTE_RING_NUM) {

+                    unsigned int num_rings = it.get_u32();

+                    if (*mNumRings < num_rings) {

+                        ALOGE("Not enough status buffers provided, available: %d required: %d",

+                                *mNumRings, num_rings);

+                    } else {

+                        *mNumRings = num_rings;

+                    }

+                } else {

+                    ALOGE("Unknown attribute: %d expecting %d",

+                            it.get_type(), ENHANCE_LOGGER_ATTRIBUTE_RING_NUM);

+                    return NL_SKIP;

+                }

+

+                it.next();

+                if (it.get_type() == ENHANCE_LOGGER_ATTRIBUTE_RING_STATUS) {

+                    memcpy(status, it.get_data(), sizeof(wifi_ring_buffer_status) * (*mNumRings));

+                } else {

+                    ALOGW("Ignoring invalid attribute type = %d, size = %d",

+                                it.get_type(), it.get_len());

+                }

+                break;

+            }

+

+            case GET_FEATURE:

+            {

+                void *data = reply.get_vendor_data();

+                int len = reply.get_vendor_data_len();

+

+                ALOGD("len = %d, expected len = %d", len, sizeof(unsigned int));

+                memcpy(mSupport, data, sizeof(unsigned int));

+                break;

+            }

+

+            default:

+                ALOGW("Unknown Debug command");

+        }

+        return NL_OK;

+    }

+

+    virtual int handleEvent(WifiEvent& event) {

+        /* NO events! */

+        return NL_SKIP;

+    }

+};

+

+/* API to get supportable feature */

+wifi_error wifi_get_logger_supported_feature_set(wifi_interface_handle iface,

+        unsigned int *support)

+{

+    if (support) {

+        DebugCommand *cmd = new DebugCommand(iface, support, GET_FEATURE);

+        NULL_CHECK_RETURN(cmd, "memory allocation failure", WIFI_ERROR_OUT_OF_MEMORY);

+        wifi_error result = (wifi_error)cmd->start();

+        cmd->releaseRef();

+        return result;

+    } else {

+        ALOGE("Get support buffer NULL");

+        return  WIFI_ERROR_INVALID_ARGS;

+    }

+}

+

+/* API to get the status of all ring buffers supported by driver */

+wifi_error wifi_get_ring_buffers_status(wifi_interface_handle iface,

+        u32 *num_rings, wifi_ring_buffer_status *status)

+{

+    if (status && num_rings) {

+        DebugCommand *cmd = new DebugCommand(iface, num_rings, status, GET_RING_STATUS);

+        NULL_CHECK_RETURN(cmd, "memory allocation failure", WIFI_ERROR_OUT_OF_MEMORY);

+        wifi_error result = (wifi_error)cmd->start();

+        cmd->releaseRef();

+        return result;

+    } else {

+        ALOGE("Ring status buffer NULL");

+        return  WIFI_ERROR_INVALID_ARGS;

+    }

+}

+

+/* API to collect driver records */

+wifi_error wifi_get_ring_data(wifi_interface_handle iface, char *ring_name)

+{

+    DebugCommand *cmd = new DebugCommand(iface, ring_name, GET_RING_DATA);

+    NULL_CHECK_RETURN(cmd, "memory allocation failure", WIFI_ERROR_OUT_OF_MEMORY);

+    wifi_error result = (wifi_error)cmd->start();

+    cmd->releaseRef();

+    return result;

+}

+

+wifi_error wifi_start_logging(wifi_interface_handle iface, u32 verbose_level,

+        u32 flags, u32 max_interval_sec, u32 min_data_size, char *ring_name)

+{

+    if (ring_name) {

+        DebugCommand *cmd = new DebugCommand(iface, verbose_level, flags, max_interval_sec,

+                    min_data_size, ring_name, START_RING_LOG);

+        NULL_CHECK_RETURN(cmd, "memory allocation failure", WIFI_ERROR_OUT_OF_MEMORY);

+        wifi_error result = (wifi_error)cmd->start();

+        cmd->releaseRef();

+        return result;

+    } else {

+        ALOGE("Ring name NULL");

+        return  WIFI_ERROR_INVALID_ARGS;

+    }

+}

+

+/* API to collect a firmware version string */

+wifi_error wifi_get_firmware_version(wifi_interface_handle iface, char *buffer,

+        int buffer_size)

+{

+    if (buffer && (buffer_size > 0)) {

+        DebugCommand *cmd = new DebugCommand(iface, buffer, &buffer_size, GET_FW_VER);

+        NULL_CHECK_RETURN(cmd, "memory allocation failure", WIFI_ERROR_OUT_OF_MEMORY);

+        wifi_error result = (wifi_error)cmd->start();

+        cmd->releaseRef();

+        return result;

+    } else {

+        ALOGE("FW version buffer NULL");

+        return  WIFI_ERROR_INVALID_ARGS;

+    }

+}

+

+/* API to collect a driver version string */

+wifi_error wifi_get_driver_version(wifi_interface_handle iface, char *buffer, int buffer_size)

+{

+    if (buffer && (buffer_size > 0)) {

+        DebugCommand *cmd = new DebugCommand(iface, buffer, &buffer_size, GET_DRV_VER);

+        NULL_CHECK_RETURN(cmd, "memory allocation failure", WIFI_ERROR_OUT_OF_MEMORY);

+        wifi_error result = (wifi_error)cmd->start();

+        cmd->releaseRef();

+        return result;

+    } else {

+        ALOGE("Driver version buffer NULL");

+        return  WIFI_ERROR_INVALID_ARGS;

+    }

+}

+

+

+///////////////////////////////////////////////////////////////////////////////

+class SetLogHandler : public WifiCommand

+{

+    wifi_ring_buffer_data_handler mHandler;

+

+public:

+    SetLogHandler(wifi_interface_handle iface, int id, wifi_ring_buffer_data_handler handler)

+        : WifiCommand(iface, id), mHandler(handler)

+    { }

+

+    int start() {

+        ALOGV("Register loghandler");

+        registerVendorHandler(GOOGLE_OUI, ENHANCE_LOGGER_RING_EVENT);

+        return WIFI_SUCCESS;

+    }

+

+    virtual int cancel() {

+        /* Send a command to driver to stop generating logging events */

+        ALOGV("Clear loghandler");

+

+        /* unregister event handler */

+        unregisterVendorHandler(GOOGLE_OUI, ENHANCE_LOGGER_RING_EVENT);

+

+        WifiRequest request(familyId(), ifaceId());

+        int result = request.create(GOOGLE_OUI, ENHANCE_LOGGER_RESET_LOGGING);

+        if (result != WIFI_SUCCESS) {

+            ALOGE("failed to create reset request; result = %d", result);

+            return result;

+        }

+

+        result = requestResponse(request);

+        if (result != WIFI_SUCCESS) {

+            ALOGE("failed to request reset; result = %d", result);

+            return result;

+        }

+

+        ALOGD("Success to clear loghandler");

+        return WIFI_SUCCESS;

+    }

+

+    virtual int handleEvent(WifiEvent& event) {

+        char *buffer = NULL;

+        int buffer_size = 0;

+

+        // ALOGD("In SetLogHandler::handleEvent");

+        nlattr *vendor_data = event.get_attribute(NL80211_ATTR_VENDOR_DATA);

+        int len = event.get_vendor_data_len();

+        int event_id = event.get_vendor_subcmd();

+        // ALOGI("Got Logger event: %d", event_id);

+

+        if (vendor_data == NULL || len == 0) {

+            ALOGE("No Debug data found");

+            return NL_SKIP;

+        }

+

+        if(event_id == ENHANCE_LOGGER_RING_EVENT) {

+            wifi_ring_buffer_status status;

+            memset(&status, 0, sizeof(status));

+

+            for (nl_iterator it(vendor_data); it.has_next(); it.next()) {

+                if (it.get_type() == ENHANCE_LOGGER_ATTRIBUTE_RING_STATUS) {

+                    memcpy(&status, it.get_data(), sizeof(status));

+                } else if (it.get_type() == ENHANCE_LOGGER_ATTRIBUTE_RING_DATA) {

+                    buffer_size = it.get_len();

+                    buffer = (char *)it.get_data();

+                } else {

+                    ALOGW("Ignoring invalid attribute type = %d, size = %d",

+                            it.get_type(), it.get_len());

+                }

+            }

+

+            // ALOGI("Retrieved Debug data");

+            if (mHandler.on_ring_buffer_data) {

+                (*mHandler.on_ring_buffer_data)((char *)status.name, buffer, buffer_size,

+                        &status);

+            }

+        } else {

+            ALOGE("Unknown Event");

+            return NL_SKIP;

+        }

+        return NL_OK;

+    }

+};

+

+wifi_error wifi_set_log_handler(wifi_request_id id, wifi_interface_handle iface,

+        wifi_ring_buffer_data_handler handler)

+{

+    wifi_handle handle = getWifiHandle(iface);

+    ALOGV("Loghandler start, handle = %p", handle);

+

+    SetLogHandler *cmd = new SetLogHandler(iface, id, handler);

+    NULL_CHECK_RETURN(cmd, "memory allocation failure", WIFI_ERROR_OUT_OF_MEMORY);

+    wifi_error result = wifi_register_cmd(handle, id, cmd);

+    if (result != WIFI_SUCCESS) {

+        cmd->releaseRef();

+        return result;

+    }

+    result = (wifi_error)cmd->start();

+    if (result != WIFI_SUCCESS) {

+        wifi_unregister_cmd(handle, id);

+        cmd->releaseRef();

+        return result;

+    }

+    return result;

+}

+

+wifi_error wifi_reset_log_handler(wifi_request_id id, wifi_interface_handle iface)

+{

+    wifi_handle handle = getWifiHandle(iface);

+    ALOGV("Loghandler reset, wifi_request_id = %d, handle = %p", id, handle);

+

+    if (id == -1) {

+        wifi_ring_buffer_data_handler handler;

+        memset(&handler, 0, sizeof(handler));

+

+        SetLogHandler *cmd = new SetLogHandler(iface, id, handler);

+        NULL_CHECK_RETURN(cmd, "memory allocation failure", WIFI_ERROR_OUT_OF_MEMORY);

+        cmd->cancel();

+        cmd->releaseRef();

+        return WIFI_SUCCESS;

+    }

+

+    return wifi_cancel_cmd(id, iface);

+}

+

+///////////////////////////////////////////////////////////////////////////////

+class SetAlertHandler : public WifiCommand

+{

+    wifi_alert_handler mHandler;

+    int mBuffSize;

+    char *mBuff;

+    int mErrCode;

+

+public:

+    SetAlertHandler(wifi_interface_handle iface, int id, wifi_alert_handler handler)

+        : WifiCommand(iface, id), mHandler(handler), mBuffSize(0), mBuff(NULL),

+            mErrCode(0)

+    { }

+

+    int start() {

+        ALOGV("Start Alerting");

+        registerVendorHandler(GOOGLE_OUI, ENHANCE_LOGGER_MEM_DUMP_EVENT);

+        return WIFI_SUCCESS;

+    }

+

+    virtual int cancel() {

+        ALOGV("Clear alerthandler");

+

+        /* unregister alert handler */

+        unregisterVendorHandler(GOOGLE_OUI, ENHANCE_LOGGER_MEM_DUMP_EVENT);

+        wifi_unregister_cmd(wifiHandle(), id());

+        ALOGD("Success to clear alerthandler");

+        return WIFI_SUCCESS;

+    }

+

+    virtual int handleResponse(WifiEvent& reply) {

+        ALOGD("In SetAlertHandler::handleResponse");

+

+        if (reply.get_cmd() != NL80211_CMD_VENDOR) {

+            ALOGD("Ignoring reply with cmd = %d", reply.get_cmd());

+            return NL_SKIP;

+        }

+

+        nlattr *vendor_data = reply.get_attribute(NL80211_ATTR_VENDOR_DATA);

+        int len = reply.get_vendor_data_len();

+

+        ALOGD("len = %d", len);

+        if (vendor_data == NULL || len == 0) {

+            ALOGE("no vendor data in memory dump response; ignoring it");

+            return NL_SKIP;

+        }

+

+        for (nl_iterator it(vendor_data); it.has_next(); it.next()) {

+            if (it.get_type() == ENHANCE_LOGGER_ATTRIBUTE_FW_DUMP_DATA) {

+                ALOGI("Initiating alert callback");

+                if (mHandler.on_alert) {

+                    (*mHandler.on_alert)(id(), mBuff, mBuffSize, mErrCode);

+                }

+                if (mBuff) {

+                    free(mBuff);

+                    mBuff = NULL;

+                }

+            }

+        }

+        return NL_OK;

+    }

+

+    virtual int handleEvent(WifiEvent& event) {

+        wifi_ring_buffer_id ring_id;

+        char *buffer = NULL;

+        int buffer_size = 0;

+

+

+        nlattr *vendor_data = event.get_attribute(NL80211_ATTR_VENDOR_DATA);

+        int len = event.get_vendor_data_len();

+        int event_id = event.get_vendor_subcmd();

+        ALOGI("Got event: %d", event_id);

+

+        if (vendor_data == NULL || len == 0) {

+            ALOGE("No Debug data found");

+            return NL_SKIP;

+        }

+

+        if (event_id == ENHANCE_LOGGER_MEM_DUMP_EVENT) {

+            for (nl_iterator it(vendor_data); it.has_next(); it.next()) {

+                if (it.get_type() == ENHANCE_LOGGER_ATTRIBUTE_FW_DUMP_LEN) {

+                    mBuffSize = it.get_u32();

+                } else if (it.get_type() == ENHANCE_LOGGER_ATTRIBUTE_RING_DATA) {

+                    buffer_size = it.get_len();

+                    buffer = (char *)it.get_data();

+            /*

+                } else if (it.get_type() == ENHANCE_LOGGER_ATTRIBUTE_FW_ERR_CODE) {

+                    mErrCode = it.get_u32();

+            */

+                } else {

+                    ALOGW("Ignoring invalid attribute type = %d, size = %d",

+                            it.get_type(), it.get_len());

+                }

+            }

+            if (mBuffSize) {

+                ALOGD("dump size: %d meta data size: %d", mBuffSize, buffer_size);

+                if (mBuff) free(mBuff);

+                mBuff = (char *)malloc(mBuffSize + buffer_size);

+                if (!mBuff) {

+                    ALOGE("Buffer allocation failed");

+                    return NL_SKIP;

+                }

+                memcpy(mBuff, buffer, buffer_size);

+

+                WifiRequest request(familyId(), ifaceId());

+                int result = request.create(GOOGLE_OUI, ENHANCE_LOGGER_MEM_DUMP_EVENT);

+                if (result != WIFI_SUCCESS) {

+                    ALOGE("Failed to create get memory dump request; result = %d", result);

+                    free(mBuff);

+                    return NL_SKIP;

+                }

+                nlattr *data = request.attr_start(NL80211_ATTR_VENDOR_DATA);

+                result = request.put_u32(ENHANCE_LOGGER_ATTRIBUTE_FW_DUMP_LEN, mBuffSize);

+                if (result != WIFI_SUCCESS) {

+                    ALOGE("Failed to put get memory dump request; result = %d", result);

+                    return result;

+                }

+

+                result = request.put_u64(ENHANCE_LOGGER_ATTRIBUTE_FW_DUMP_DATA,

+                         (uint64_t)(mBuff+buffer_size));

+                if (result != WIFI_SUCCESS) {

+                    ALOGE("Failed to put get memory dump request; result = %d", result);

+                    return result;

+                }

+

+                request.attr_end(data);

+                mBuffSize += buffer_size;

+

+                result = requestResponse(request);

+

+                if (result != WIFI_SUCCESS) {

+                    ALOGE("Failed to register get momory dump response; result = %d", result);

+                }

+            } else {

+                ALOGE("dump event missing dump length attribute");

+                return NL_SKIP;

+            }

+        }

+        return NL_OK;

+    }

+};

+

+wifi_error wifi_set_alert_handler(wifi_request_id id, wifi_interface_handle iface,

+        wifi_alert_handler handler)

+{

+    wifi_handle handle = getWifiHandle(iface);

+    ALOGV("Alerthandler start, handle = %p", handle);

+

+    SetAlertHandler *cmd = new SetAlertHandler(iface, id, handler);

+    NULL_CHECK_RETURN(cmd, "memory allocation failure", WIFI_ERROR_OUT_OF_MEMORY);

+    wifi_error result = wifi_register_cmd(handle, id, cmd);

+    if (result != WIFI_SUCCESS) {

+        cmd->releaseRef();

+        return result;

+    }

+    result = (wifi_error)cmd->start();

+    if (result != WIFI_SUCCESS) {

+        wifi_unregister_cmd(handle, id);

+        cmd->releaseRef();

+        return result;

+    }

+    return result;

+}

+

+wifi_error wifi_reset_alert_handler(wifi_request_id id, wifi_interface_handle iface)

+{

+    wifi_handle handle = getWifiHandle(iface);

+    ALOGV("Alerthandler reset, wifi_request_id = %d, handle = %p", id, handle);

+

+    if (id == -1) {

+        wifi_alert_handler handler;

+        memset(&handler, 0, sizeof(handler));

+

+        SetAlertHandler *cmd = new SetAlertHandler(iface, id, handler);

+        NULL_CHECK_RETURN(cmd, "memory allocation failure", WIFI_ERROR_OUT_OF_MEMORY);

+        cmd->cancel();

+        cmd->releaseRef();

+        return WIFI_SUCCESS;

+    }

+

+    return wifi_cancel_cmd(id, iface);

+}

+

+

+class PacketFateCommand: public WifiCommand

+{

+    void *mReportBufs;

+    size_t mNoReqFates;

+    size_t *mNoProvidedFates;

+    PktFateReqType mReqType;

+

+public:

+    PacketFateCommand(wifi_interface_handle handle)

+        : WifiCommand(handle, 0), mReqType(PACKET_MONITOR_START)

+    { }

+

+    PacketFateCommand(wifi_interface_handle handle, wifi_tx_report *tx_report_bufs,

+            size_t n_requested_fates, size_t *n_provided_fates)

+        : WifiCommand(handle, 0), mReportBufs(tx_report_bufs),

+                  mNoReqFates(n_requested_fates), mNoProvidedFates(n_provided_fates),

+                  mReqType(TX_PACKET_FATE)

+    { }

+

+    PacketFateCommand(wifi_interface_handle handle, wifi_rx_report *rx_report_bufs,

+            size_t n_requested_fates, size_t *n_provided_fates)

+        : WifiCommand(handle, 0), mReportBufs(rx_report_bufs),

+                  mNoReqFates(n_requested_fates), mNoProvidedFates(n_provided_fates),

+                  mReqType(RX_PACKET_FATE)

+    { }

+

+    int createRequest(WifiRequest& request) {

+        if (mReqType == TX_PACKET_FATE) {

+            ALOGD("%s Get Tx packet fate request\n", __FUNCTION__);

+            return createTxPktFateRequest(request);

+        } else if (mReqType == RX_PACKET_FATE) {

+            ALOGD("%s Get Rx packet fate request\n", __FUNCTION__);

+            return createRxPktFateRequest(request);

+        } else if (mReqType == PACKET_MONITOR_START) {

+            ALOGD("%s Monitor packet fate request\n", __FUNCTION__);

+            return createMonitorPktFateRequest(request);

+        } else {

+            ALOGE("%s Unknown packet fate request\n", __FUNCTION__);

+            return WIFI_ERROR_NOT_SUPPORTED;

+        }

+        return WIFI_SUCCESS;

+    }

+

+    int createMonitorPktFateRequest(WifiRequest& request) {

+        int result = request.create(GOOGLE_OUI, ENHANCE_LOGGER_START_PKT_FATE_MONITORING);

+        if (result < 0) {

+            return result;

+        }

+

+        nlattr *data = request.attr_start(NL80211_ATTR_VENDOR_DATA);

+        request.attr_end(data);

+        return result;

+    }

+

+    int createTxPktFateRequest(WifiRequest& request) {

+        int result = request.create(GOOGLE_OUI, ENHANCE_LOGGER_GET_TX_PKT_FATES);

+        if (result < 0) {

+            return result;

+        }

+

+        memset(mReportBufs, 0, (mNoReqFates * sizeof(wifi_tx_report)));

+        nlattr *data = request.attr_start(NL80211_ATTR_VENDOR_DATA);

+        result = request.put_u32(ENHANCE_LOGGER_ATTRIBUTE_PKT_FATE_NUM, mNoReqFates);

+        if (result < 0) {

+            return result;

+        }

+        result = request.put_u64(ENHANCE_LOGGER_ATTRIBUTE_PKT_FATE_DATA, (uint64_t)mReportBufs);

+        if (result < 0) {

+            return result;

+        }

+        request.attr_end(data);

+        return result;

+    }

+

+    int createRxPktFateRequest(WifiRequest& request) {

+        int result = request.create(GOOGLE_OUI, ENHANCE_LOGGER_GET_RX_PKT_FATES);

+        if (result < 0) {

+            return result;

+        }

+

+        memset(mReportBufs, 0, (mNoReqFates * sizeof(wifi_rx_report)));

+        nlattr *data = request.attr_start(NL80211_ATTR_VENDOR_DATA);

+        result = request.put_u32(ENHANCE_LOGGER_ATTRIBUTE_PKT_FATE_NUM, mNoReqFates);

+        if (result < 0) {

+            return result;

+        }

+        result = request.put_u64(ENHANCE_LOGGER_ATTRIBUTE_PKT_FATE_DATA, (uint64_t)mReportBufs);

+        if (result < 0) {

+            return result;

+        }

+        request.attr_end(data);

+        return result;

+    }

+

+    int start() {

+        ALOGD("Start get packet fate command\n");

+        WifiRequest request(familyId(), ifaceId());

+

+        int result = createRequest(request);

+        if (result < 0) {

+            ALOGE("Failed to create get pkt fate request; result = %d\n", result);

+            return result;

+        }

+

+        result = requestResponse(request);

+        if (result != WIFI_SUCCESS) {

+            ALOGE("Failed to register get pkt fate response; result = %d\n", result);

+        }

+        return result;

+    }

+

+    int handleResponse(WifiEvent& reply) {

+        ALOGD("In GetPktFateCommand::handleResponse\n");

+

+        if (reply.get_cmd() != NL80211_CMD_VENDOR) {

+            ALOGI("Ignoring reply with cmd = %d", reply.get_cmd());

+            return NL_SKIP;

+        }

+

+        int id = reply.get_vendor_id();

+        int subcmd = reply.get_vendor_subcmd();

+        nlattr *vendor_data = reply.get_attribute(NL80211_ATTR_VENDOR_DATA);

+        int len = reply.get_vendor_data_len();

+

+        ALOGI("Id = %0x, subcmd = %d, len = %d", id, subcmd, len);

+

+        if (mReqType == TX_PACKET_FATE) {

+            ALOGI("Response recieved for get TX pkt fate command\n");

+        } else if (mReqType == RX_PACKET_FATE) {

+            ALOGI("Response recieved for get RX pkt fate command\n");

+        } else if (mReqType == PACKET_MONITOR_START) {

+            ALOGI("Response recieved for monitor pkt fate command\n");

+            return NL_OK;

+        } else {

+            ALOGE("Response recieved for unknown pkt fate command\n");

+            return NL_SKIP;

+        }

+

+        if (vendor_data == NULL || len == 0) {

+            ALOGE("no vendor data in GetPktFateCommand response; ignoring it\n");

+            return NL_SKIP;

+        }

+

+        for (nl_iterator it(vendor_data); it.has_next(); it.next()) {

+            if (it.get_type() == ENHANCE_LOGGER_ATTRIBUTE_PKT_FATE_NUM) {

+                *mNoProvidedFates = it.get_u32();

+                ALOGI("No: of pkt fates provided is %d\n", *mNoProvidedFates);

+            } else {

+                ALOGE("Ignoring invalid attribute type = %d, size = %d\n",

+                        it.get_type(), it.get_len());

+            }

+        }

+

+        return NL_OK;

+    }

+

+    int handleEvent(WifiEvent& event) {

+        /* NO events to handle here! */

+        return NL_SKIP;

+    }

+};

+

+wifi_error wifi_start_pkt_fate_monitoring(wifi_interface_handle handle)

+{

+    PacketFateCommand *cmd = new PacketFateCommand(handle);

+    NULL_CHECK_RETURN(cmd, "memory allocation failure", WIFI_ERROR_OUT_OF_MEMORY);

+    wifi_error result = (wifi_error)cmd->start();

+    cmd->releaseRef();

+    return result;

+}

+

+wifi_error wifi_get_tx_pkt_fates(wifi_interface_handle handle,

+        wifi_tx_report *tx_report_bufs, size_t n_requested_fates,

+        size_t *n_provided_fates)

+{

+    PacketFateCommand *cmd = new PacketFateCommand(handle, tx_report_bufs,

+                n_requested_fates, n_provided_fates);

+    NULL_CHECK_RETURN(cmd, "memory allocation failure", WIFI_ERROR_OUT_OF_MEMORY);

+    wifi_error result = (wifi_error)cmd->start();

+    cmd->releaseRef();

+    return result;

+}

+

+wifi_error wifi_get_rx_pkt_fates(wifi_interface_handle handle,

+        wifi_rx_report *rx_report_bufs, size_t n_requested_fates,

+        size_t *n_provided_fates)

+{

+    PacketFateCommand *cmd = new PacketFateCommand(handle, rx_report_bufs,

+                n_requested_fates, n_provided_fates);

+    NULL_CHECK_RETURN(cmd, "memory allocation failure", WIFI_ERROR_OUT_OF_MEMORY);

+    wifi_error result = (wifi_error)cmd->start();

+    cmd->releaseRef();

+    return result;

+}

+

+class MemoryDumpCommand: public WifiCommand

+{

+    wifi_firmware_memory_dump_handler mHandler;

+    wifi_driver_memory_dump_callbacks mcallback;

+    int mBuffSize;

+    char *mBuff;

+    GetCmdType mType;

+

+public:

+    MemoryDumpCommand(wifi_interface_handle iface, wifi_firmware_memory_dump_handler handler, GetCmdType cmdtype )

+        : WifiCommand(iface, 0), mHandler(handler), mBuffSize(0), mBuff(NULL), mType(cmdtype)

+    { }

+

+    MemoryDumpCommand(wifi_interface_handle iface, wifi_driver_memory_dump_callbacks callback, GetCmdType cmdtype)

+        : WifiCommand(iface, 0), mcallback(callback), mBuffSize(0), mBuff(NULL), mType(cmdtype)

+    { }

+

+     int createRequest(WifiRequest &request) {

+        int result;

+

+        switch (mType) {

+                case GET_FW_DUMP:

+                    {

+                        result = request.create(GOOGLE_OUI, ENHANCE_LOGGER_TRIGGER_FW_MEM_DUMP);

+                        if (result != WIFI_SUCCESS) {

+                                 ALOGE("Failed to create trigger fw memory dump request; result = %d", result);

+                                 return result;

+                        }

+                        break;

+                    }

+                case GET_DRIVER_DUMP :

+                    {

+                        result = request.create(GOOGLE_OUI, ENHANCE_LOGGER_TRIGGER_DRIVER_MEM_DUMP);

+                        if (result != WIFI_SUCCESS) {

+                                ALOGE("Failed to create trigger driver memory dump request; result = %d", result);

+                                return result;

+                         }

+                        break;

+                      }

+                 default:

+                         ALOGE("Unknown Debug command");

+                         result = WIFI_ERROR_UNKNOWN;

+                     }

+        return result;

+     }

+    int start() {

+        ALOGD("Start memory dump command");

+        WifiRequest request(familyId(), ifaceId());

+

+        int result = createRequest(request);

+        if (result != WIFI_SUCCESS) {

+            ALOGE("Failed to create trigger memory dump request; result = %d", result);

+            return result;

+        }

+

+        result = requestResponse(request);

+        if (result != WIFI_SUCCESS) {

+            ALOGE("Failed to register trigger memory dump response; result = %d", result);

+        }

+        return result;

+    }

+

+    virtual int handleResponse(WifiEvent& reply) {

+        ALOGD("In MemoryDumpCommand::handleResponse");

+

+        if (reply.get_cmd() != NL80211_CMD_VENDOR) {

+            ALOGD("Ignoring reply with cmd = %d", reply.get_cmd());

+            return NL_SKIP;

+        }

+

+        nlattr *vendor_data = reply.get_attribute(NL80211_ATTR_VENDOR_DATA);

+        int len = reply.get_vendor_data_len();

+

+        ALOGD("len = %d", len);

+        if (vendor_data == NULL || len == 0) {

+            ALOGE("no vendor data in memory dump response; ignoring it");

+            return NL_SKIP;

+        }

+        switch(mType) {

+            case GET_FW_DUMP:

+                 for (nl_iterator it(vendor_data); it.has_next(); it.next()) {

+                     if (it.get_type() == ENHANCE_LOGGER_ATTRIBUTE_FW_DUMP_LEN) {

+                         mBuffSize = it.get_u32();

+                         if (mBuff)

+                             free(mBuff);

+                         mBuff = (char *)malloc(mBuffSize);

+                         if (!mBuff) {

+                             ALOGE("Buffer allocation failed");

+                             return NL_SKIP;

+                         }

+                         WifiRequest request(familyId(), ifaceId());

+                         int result = request.create(GOOGLE_OUI, ENHANCE_LOGGER_GET_FW_MEM_DUMP);

+                         if (result != WIFI_SUCCESS) {

+                             ALOGE("Failed to create get fw memory dump request; result = %d", result);

+                             free(mBuff);

+                             return NL_SKIP;

+                          }

+                          nlattr *data = request.attr_start(NL80211_ATTR_VENDOR_DATA);

+                          result = request.put_u32(ENHANCE_LOGGER_ATTRIBUTE_FW_DUMP_LEN, mBuffSize);

+                          if (result != WIFI_SUCCESS) {

+                              ALOGE("Failed to put get fw memory dump request; result = %d", result);

+                              return result;

+                          }

+                          result = request.put_u64(ENHANCE_LOGGER_ATTRIBUTE_FW_DUMP_DATA, (uint64_t)mBuff);

+                          if (result != WIFI_SUCCESS) {

+                              ALOGE("Failed to put get fw memory dump request; result = %d", result);

+                              return result;

+                          }

+                          request.attr_end(data);

+                          result = requestResponse(request);

+                          if (result != WIFI_SUCCESS) {

+                              ALOGE("Failed to register get fw momory dump response; result = %d", result);

+                          }

+                     } else if (it.get_type() == ENHANCE_LOGGER_ATTRIBUTE_FW_DUMP_DATA) {

+                          ALOGI("Initiating memory dump callback");

+                          if (mHandler.on_firmware_memory_dump) {

+                              (*mHandler.on_firmware_memory_dump)(mBuff, mBuffSize);

+                          }

+                          if (mBuff) {

+                              free(mBuff);

+                              mBuff = NULL;

+                          }

+                    } else {

+                           ALOGW("Ignoring invalid attribute type = %d, size = %d",

+                           it.get_type(), it.get_len());

+                     }

+                 }

+            case GET_DRIVER_DUMP :

+                 for (nl_iterator it(vendor_data); it.has_next(); it.next()) {

+                     if (it.get_type() == ENHANCE_LOGGER_ATTRIBUTE_DRIVER_DUMP_LEN) {

+                         mBuffSize = it.get_u32();

+                         if (mBuff)

+                             free(mBuff);

+                         mBuff = (char *)malloc(mBuffSize);

+                         if (!mBuff) {

+                             ALOGE("Buffer allocation failed");

+                             return NL_SKIP;

+                         }

+                         WifiRequest request(familyId(), ifaceId());

+                         int result = request.create(GOOGLE_OUI, ENHANCE_LOGGER_GET_DRIVER_MEM_DUMP);

+                         if (result != WIFI_SUCCESS) {

+                             ALOGE("Failed to create get driver memory dump request; result = %d", result);

+                             free(mBuff);

+                             return NL_SKIP;

+                          }

+                          nlattr *data = request.attr_start(NL80211_ATTR_VENDOR_DATA);

+                          result = request.put_u32(ENHANCE_LOGGER_ATTRIBUTE_DRIVER_DUMP_LEN, mBuffSize);

+                          if (result != WIFI_SUCCESS) {

+                              ALOGE("Failed to put get driver memory dump request; result = %d", result);

+                              return result;

+                          }

+                          result = request.put_u64(ENHANCE_LOGGER_ATTRIBUTE_DRIVER_DUMP_DATA, (uint64_t)mBuff);

+                          if (result != WIFI_SUCCESS) {

+                              ALOGE("Failed to put get driver memory dump request; result = %d", result);

+                              return result;

+                          }

+                          request.attr_end(data);

+                          result = requestResponse(request);

+                          if (result != WIFI_SUCCESS) {

+                              ALOGE("Failed to register get driver momory dump response; result = %d", result);

+                          }

+                     } else if (it.get_type() == ENHANCE_LOGGER_ATTRIBUTE_DRIVER_DUMP_DATA) {

+                          ALOGI("Initiating memory dump callback");

+                          if (mcallback.on_driver_memory_dump) {

+                              (*mcallback.on_driver_memory_dump)(mBuff, mBuffSize);

+                          }

+                          if (mBuff) {

+                              free(mBuff);

+                              mBuff = NULL;

+                          }

+                    } else {

+                           ALOGW("Ignoring invalid attribute type = %d, size = %d",

+                           it.get_type(), it.get_len());

+                     }

+                 }

+            }

+        return NL_OK;

+    }

+

+    virtual int handleEvent(WifiEvent& event) {

+        /* NO events! */

+        return NL_SKIP;

+    }

+};

+

+/* API to collect a firmware memory dump for a given iface */

+wifi_error wifi_get_firmware_memory_dump( wifi_interface_handle iface,

+        wifi_firmware_memory_dump_handler handler)

+{

+    MemoryDumpCommand *cmd = new MemoryDumpCommand(iface, handler, GET_FW_DUMP);

+    NULL_CHECK_RETURN(cmd, "memory allocation failure", WIFI_ERROR_OUT_OF_MEMORY);

+    wifi_error result = (wifi_error)cmd->start();

+    cmd->releaseRef();

+    return result;

+}

+

+wifi_error wifi_get_driver_memory_dump(wifi_interface_handle iface,

+                                wifi_driver_memory_dump_callbacks callbacks)

+{

+    MemoryDumpCommand *cmd = new MemoryDumpCommand(iface, callbacks, GET_DRIVER_DUMP);

+    NULL_CHECK_RETURN(cmd, "memory allocation failure", WIFI_ERROR_OUT_OF_MEMORY);

+    wifi_error result = (wifi_error)cmd->start();

+    cmd->releaseRef();

+    return result;

+}

+class WifiLoggerCommand: public WifiCommand

+{

+     WLAN_DRIVER_WAKE_REASON_CNT *mGetWakeStats;

+

+public:

+    WifiLoggerCommand(wifi_interface_handle handle, WLAN_DRIVER_WAKE_REASON_CNT *wifi_wake_reason_cnt)

+        : WifiCommand(handle, 0), mGetWakeStats(wifi_wake_reason_cnt)

+    { }

+

+    int createRequest(WifiRequest& request) {

+        int result = request.create(GOOGLE_OUI, ENHANCE_LOGGER_GET_WAKE_REASON_STATS);

+        if (result < 0) {

+            return result;

+        }

+

+        nlattr *data = request.attr_start(NL80211_ATTR_VENDOR_DATA);

+

+        result = request.put_u32(ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_CMD_EVENT_WAKE_CNT_SZ, mGetWakeStats->cmd_event_wake_cnt_sz);

+        if (result != WIFI_SUCCESS) {

+            ALOGE("Failed to put wake_cnt_sz; result = %d", result);

+            return result;

+        }

+        result = request.put_u32(ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_DRIVER_FW_LOCAL_WAKE_CNT_SZ, mGetWakeStats->driver_fw_local_wake_cnt_sz);

+        if (result != WIFI_SUCCESS) {

+            ALOGE("Failed to put driver_fw_local_wake_cnt; result = %d", result);

+            return result;

+        }

+        request.attr_end(data);

+        return result;

+    }

+

+    int start() {

+        ALOGD("Start get wake reason stats command\n");

+        WifiRequest request(familyId(), ifaceId());

+        int result = createRequest(request);

+        if (result < 0) {

+                 ALOGE("Failed to create get wake reason stats request; result = %d\n", result);

+                 return result;

+        }

+        result = requestResponse(request);

+        if (result != WIFI_SUCCESS) {

+                ALOGE("Failed to register get wake reason stats response; result = %d\n", result);

+        }

+        return result;

+    }

+

+    int handleResponse(WifiEvent& reply) {

+         int len = 0;

+         ALOGD("In WifiLoggerCommand::handleResponse\n");

+

+         if (reply.get_cmd() != NL80211_CMD_VENDOR) {

+             ALOGI("Ignoring reply with cmd = %d", reply.get_cmd());

+             return NL_SKIP;

+        }

+        nlattr *vendor_data = reply.get_attribute(NL80211_ATTR_VENDOR_DATA);

+        len = reply.get_vendor_data_len();

+        if (vendor_data == NULL || len == 0) {

+            ALOGE("No Debug data found");

+            return NL_SKIP;

+       }

+       nl_iterator it(vendor_data);

+

+       if (it.get_type() != ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_TOTAL_CMD_EVENT_WAKE) {

+            ALOGE("TOTAL_CMD_EVENT_WAKE not found %d", it.get_type());

+            return NL_SKIP;

+       }

+

+       mGetWakeStats->total_cmd_event_wake = it.get_u32();

+       it.next();

+

+       if(mGetWakeStats->total_cmd_event_wake && mGetWakeStats->cmd_event_wake_cnt) {

+            if (it.get_type() != ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_CMD_EVENT_WAKE_CNT_PTR) {

+                 ALOGE("CMD_EVENT_WAKE_CNT_PTR not found %d", it.get_type());

+                 return NL_SKIP;

+            }

+

+            len = it.get_len();

+            mGetWakeStats->cmd_event_wake_cnt_used = (len < mGetWakeStats->cmd_event_wake_cnt_sz) ? len : mGetWakeStats->cmd_event_wake_cnt_sz;

+            memcpy(mGetWakeStats->cmd_event_wake_cnt, it.get_data(), mGetWakeStats->cmd_event_wake_cnt_used * sizeof(int));

+       } else {

+            mGetWakeStats->cmd_event_wake_cnt_used = 0;

+       }

+

+       it.next();

+

+       if (it.get_type() != ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_TOTAL_DRIVER_FW_LOCAL_WAKE) {

+            ALOGE("TOTAL_DRIVER_FW_LOCAL_WAKE not found %d", it.get_type());

+            return NL_SKIP;

+       }

+

+       mGetWakeStats->total_driver_fw_local_wake = it.get_u32();

+       it.next();

+

+       if(mGetWakeStats->total_driver_fw_local_wake && mGetWakeStats->driver_fw_local_wake_cnt) {

+            if (it.get_type() != ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_DRIVER_FW_LOCAL_WAKE_CNT_PTR) {

+                 ALOGE("DRIVER_FW_LOCAL_WAKE_CNT_PTR not found %d", it.get_type());

+                 return NL_SKIP;

+            }

+

+            len = it.get_len();

+            mGetWakeStats->driver_fw_local_wake_cnt_used= (len < mGetWakeStats->driver_fw_local_wake_cnt_sz) ? len : mGetWakeStats->driver_fw_local_wake_cnt_sz;

+            memcpy(mGetWakeStats->driver_fw_local_wake_cnt, it.get_data(), mGetWakeStats->driver_fw_local_wake_cnt_used * sizeof(int));

+       } else {

+            mGetWakeStats->driver_fw_local_wake_cnt_used= 0;

+       }

+

+       it.next();

+

+       if (it.get_type() != ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_TOTAL_RX_DATA_WAKE) {

+            ALOGE("TOTAL_RX_DATA_WAKE not found %d", it.get_type());

+            return NL_SKIP;

+       }

+

+       mGetWakeStats->total_rx_data_wake = it.get_u32();

+       it.next();

+

+       if (it.get_type() != ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_RX_UNICAST_CNT) {

+            ALOGE("RX_UNICAST_CNT not found %d", it.get_type());

+            return NL_SKIP;

+       }

+

+       mGetWakeStats->rx_wake_details.rx_unicast_cnt = it.get_u32();

+       it.next();

+

+       if (it.get_type() != ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_RX_MULTICAST_CNT) {

+            ALOGE("RX_MULTICAST_CNT not found %d", it.get_type());

+            return NL_SKIP;

+       }

+

+       mGetWakeStats->rx_wake_details.rx_multicast_cnt = it.get_u32();

+       it.next();

+

+       if (it.get_type() != ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_RX_BROADCAST_CNT) {

+            ALOGE("RX_BROADCAST_CNT not found %d", it.get_type());

+            return NL_SKIP;

+       }

+

+       mGetWakeStats->rx_wake_details.rx_broadcast_cnt = it.get_u32();

+       it.next();

+

+       if (it.get_type() != ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_ICMP_PKT) {

+            ALOGE("ICMP_PKT not found %d", it.get_type());

+            return NL_SKIP;

+       }

+

+       mGetWakeStats->rx_wake_pkt_classification_info .icmp_pkt = it.get_u32();

+       it.next();

+

+       if (it.get_type() != ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_ICMP6_PKT) {

+            ALOGE("ICMP6_PKT not found %d", it.get_type());

+            return NL_SKIP;

+       }

+

+       mGetWakeStats->rx_wake_pkt_classification_info .icmp6_pkt = it.get_u32();

+       it.next();

+

+       if (it.get_type() != ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_ICMP6_RA) {

+            ALOGE("ICMP6_RA not found %d", it.get_type());

+            return NL_SKIP;

+       }

+

+       mGetWakeStats->rx_wake_pkt_classification_info .icmp6_ra = it.get_u32();

+       it.next();

+

+       if (it.get_type() != ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_ICMP6_NA) {

+            ALOGE("ICMP6_NA not found %d", it.get_type());

+            return NL_SKIP;

+       }

+

+       mGetWakeStats->rx_wake_pkt_classification_info .icmp6_na = it.get_u32();

+       it.next();

+

+       if (it.get_type() != ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_ICMP6_NS) {

+            ALOGE("ICMP6_NS not found %d", it.get_type());

+            return NL_SKIP;

+       }

+

+       mGetWakeStats->rx_wake_pkt_classification_info .icmp6_ns = it.get_u32();

+       it.next();

+

+       if (it.get_type() != ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_ICMP4_RX_MULTICAST_CNT) {

+            ALOGE("ICMP4_RX_MULTICAST_CNT not found %d", it.get_type());

+            return NL_SKIP;

+       }

+

+       mGetWakeStats->rx_multicast_wake_pkt_info.ipv4_rx_multicast_addr_cnt = it.get_u32();

+       it.next();

+

+       if (it.get_type() != ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_ICMP6_RX_MULTICAST_CNT) {

+            ALOGE("ICMP6_RX_MULTICAST_CNT not found %d", it.get_type());

+            return NL_SKIP;

+       }

+

+       mGetWakeStats->rx_multicast_wake_pkt_info.ipv6_rx_multicast_addr_cnt = it.get_u32();

+       it.next();

+

+       if (it.get_type() != ENHANCE_LOGGER_ATTRIBUTE_WAKE_STATS_OTHER_RX_MULTICAST_CNT) {

+            ALOGE("OTHER_RX_MULTICAST_CNT not found %d", it.get_type());

+            return NL_SKIP;

+       }

+

+       mGetWakeStats->rx_multicast_wake_pkt_info.other_rx_multicast_addr_cnt = it.get_u32();

+

+       return NL_OK;

+    }

+

+    int handleEvent(WifiEvent& event) {

+        /* NO events to handle here! */

+        return NL_SKIP;

+    }

+};

+

+ /* Function to get wake lock stats */

+wifi_error wifi_get_wake_reason_stats(wifi_interface_handle iface,

+                             WLAN_DRIVER_WAKE_REASON_CNT *wifi_wake_reason_cnt)

+{

+    if(wifi_wake_reason_cnt) {

+        WifiLoggerCommand *cmd =  new WifiLoggerCommand(iface,wifi_wake_reason_cnt);

+        NULL_CHECK_RETURN(cmd, "memory allocation failure", WIFI_ERROR_OUT_OF_MEMORY);

+        wifi_error result = (wifi_error)cmd->start();

+        cmd->releaseRef();

+        return result;

+    } else {

+        ALOGE("Reason cnt NULL");

+        return  WIFI_ERROR_INVALID_ARGS;

+    }

+}

+