log_mqtt.c

/*
 * Tencent is pleased to support the open source community by making IoT Hub available.
 * Copyright (C) 2016 THL A29 Limited, a Tencent company. All rights reserved.

 * Licensed under the MIT License (the "License"); you may not use this file except in
 * compliance with the License. You may obtain a copy of the License at
 * http://opensource.org/licenses/MIT

 * 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.
 *
 */


#ifdef __cplusplus
extern "C" {
#endif

#include "config.h"
#ifdef LOG_UPLOAD

#include <string.h>

#include "mqtt_client.h"
#include "lite-utils.h"
#include "qcloud_iot_device.h"
#include "log_upload.h"

typedef struct _log_mqtt_state {
    bool topic_sub_ok;
    bool result_recv_ok;
    int  log_level;    
}LogMQTTState;

static LogMQTTState sg_state = {
            .topic_sub_ok = false, 
            .result_recv_ok = false,
            .log_level = eLOG_ERROR};
            

static bool _get_json_log_level(char *json, int32_t* res)
{
	char *v = LITE_json_value_of("log_level", json);
	if (v == NULL) {
        Log_e("Invalid log level from JSON: %s", json);
		return false;
	}
	if (LITE_get_int32(res, v) != QCLOUD_RET_SUCCESS) {
        Log_e("Invalid log level from JSON: %s", json);
        HAL_Free(v);
		return false;
	}
    HAL_Free(v);
	return true;
}

static void _log_level_sub_cb(void *pClient, MQTTMessage *message, void *pUserData)
{
#define LOG_JSON_LENGTH 128        
    char json_buf[LOG_JSON_LENGTH] = {0};
    int32_t json_buf_len = 0; 
	if (message == NULL) {
		return;
	}
	LogMQTTState *state = (LogMQTTState *)pUserData;

    json_buf_len = Min(LOG_JSON_LENGTH - 1, message->payload_len);
	memcpy(json_buf, message->payload, json_buf_len);
	json_buf[json_buf_len] = '\0';    // json_parse relies on a string

	Log_d("Recv Msg Topic:%s, payload:%s", message->ptopic, json_buf);
	int log_level;
    if (!_get_json_log_level(json_buf, &log_level)) {        
        return ;
    }

    switch(log_level) {
        case eLOG_DISABLE:
            Log_w("Upload log level change to: %d", log_level);
            clear_upload_buffer();
            set_log_upload_in_comm_err(true);
            IOT_Log_Set_Upload_Level(eLOG_ERROR);
            break;
        case eLOG_ERROR:            
        case eLOG_WARN:
        case eLOG_INFO:
        case eLOG_DEBUG:            
            if (log_level < IOT_Log_Get_Upload_Level())
                clear_upload_buffer();
            IOT_Log_Set_Upload_Level((LOG_LEVEL)log_level);
            Log_w("Upload log level change to: %d", log_level);
            set_log_upload_in_comm_err(false);                            
            break;
        default:
            Log_e("Invalid log level: %d", log_level);
            break;
    }

    state->log_level = log_level;
    state->result_recv_ok = true;
    
}

static void _log_mqtt_sub_event_handler(void *pclient, MQTTEventType event_type, void *pUserData)
{
	LogMQTTState *state = (LogMQTTState *)pUserData;

	switch(event_type) {
		case MQTT_EVENT_SUBCRIBE_SUCCESS:
			Log_d("mqtt log topic subscribe success");
            state->topic_sub_ok = true;			
			break;

		case MQTT_EVENT_SUBCRIBE_TIMEOUT:
			Log_i("mqtt log topic subscribe timeout");
            state->topic_sub_ok = false;			
			break;

		case MQTT_EVENT_SUBCRIBE_NACK:
			Log_i("mqtt log topic subscribe NACK");
            state->topic_sub_ok = false;			
			break;
		case MQTT_EVENT_UNSUBSCRIBE:
			Log_i("mqtt log topic has been unsubscribed");
            state->topic_sub_ok = false;;			
			break;
		case MQTT_EVENT_CLIENT_DESTROY:
			Log_i("mqtt client has been destroyed");
            state->topic_sub_ok = false;;			
			break;
		default:
            return;
	}
}

static int _iot_log_level_get_publish(void *pClient)
{
    POINTER_SANITY_CHECK(pClient, QCLOUD_ERR_INVAL);
    static unsigned int sg_client_token = 1;

    Qcloud_IoT_Client   *mqtt_client = (Qcloud_IoT_Client *)pClient;
    DeviceInfo          *dev_info = iot_device_info_get();
    POINTER_SANITY_CHECK(dev_info, QCLOUD_ERR_INVAL);

    char topic_name[128] = {0};
    char payload_content[128] = {0};

    HAL_Snprintf(topic_name, sizeof(topic_name), "$log/operation/%s/%s", dev_info->product_id, dev_info->device_name);
    HAL_Snprintf(payload_content, sizeof(payload_content), "{\"type\": \"get_log_level\", "
                        "\"clientToken\": \"%s-%u\"}", dev_info->product_id, sg_client_token++);

    PublishParams pub_params = DEFAULT_PUB_PARAMS;
    pub_params.qos = QOS0;
    pub_params.payload = payload_content;
    pub_params.payload_len = strlen(payload_content);

	return IOT_MQTT_Publish(mqtt_client, topic_name, &pub_params);
}

int qcloud_log_topic_subscribe(void *client)
{
    /* subscribe the log topic: "$log/operation/result/${productId}/${deviceName}" */
    char topic_name[128] = {0};
    int size = HAL_Snprintf(topic_name, sizeof(topic_name), "$log/operation/result/%s/%s", 
            iot_device_info_get()->product_id, iot_device_info_get()->device_name);
    if (size < 0 || size > sizeof(topic_name) - 1)
    {
        Log_e("topic content buf not enough! content size:%d buf size:%d", size, (int)sizeof(topic_name));
        return QCLOUD_ERR_FAILURE;
    }

    SubscribeParams sub_params = DEFAULT_SUB_PARAMS;
    sub_params.on_message_handler = _log_level_sub_cb;
    sub_params.on_sub_event_handler = _log_mqtt_sub_event_handler;
    sub_params.user_data = (void *)&sg_state;
    sub_params.qos = QOS0;    

    return IOT_MQTT_Subscribe(client, topic_name, &sub_params);

}


int qcloud_get_log_level(void* pClient, int *log_level)
{
    int ret = 0;
    int cntSub = 0;
    int cntRev = 0;

    POINTER_SANITY_CHECK(pClient, QCLOUD_ERR_INVAL);
    Qcloud_IoT_Client   *mqtt_client = (Qcloud_IoT_Client *)pClient;

    // subscribe log topic: $log/operation/get/${productid}/${devicename}
    // skip this if the subscription is done and valid
    if(!sg_state.topic_sub_ok){
        for(cntSub = 0; cntSub < 3; cntSub++){
            ret = qcloud_log_topic_subscribe(mqtt_client);
            if (ret < 0) {
                Log_w("qcloud_log_topic_subscribe failed: %d, cnt: %d", ret, cntSub);
                continue;
            }

            /* wait for sub ack */
            ret = qcloud_iot_mqtt_yield((Qcloud_IoT_Client *)pClient, 100);
            if(sg_state.topic_sub_ok) {                
                break;
            }
        }
    }

    // return failure if subscribe failed
    if(!sg_state.topic_sub_ok){
        Log_e("Subscribe log topic failed!");
        return QCLOUD_ERR_FAILURE;
    }

    sg_state.result_recv_ok = false;
    // publish msg to get log level
	ret = _iot_log_level_get_publish(mqtt_client);
	if (ret < 0) {
		Log_e("client publish log topic failed :%d", ret);
        return ret;
	}

    do {
        ret = qcloud_iot_mqtt_yield((Qcloud_IoT_Client *)pClient, 100);
        cntRev++;
    } while(!sg_state.result_recv_ok && cntRev < 20);

    *log_level = sg_state.log_level;
    if (sg_state.result_recv_ok)
        ret = QCLOUD_RET_SUCCESS;
    else
        ret = QCLOUD_ERR_FAILURE;

    return ret;
}

#endif

#ifdef __cplusplus
}
#endif