Component/bt :

Add handling of osi memory calloc failure.
Add length check for snprintf in osi config.
This commit is contained in:
gengyuchao 2020-02-20 19:38:53 +08:00
parent f1ac73f496
commit be5e315282
2 changed files with 30 additions and 5 deletions

View file

@ -387,7 +387,7 @@ bool config_save(const config_t *config, const char *filename)
char *line = osi_calloc(1024); char *line = osi_calloc(1024);
char *keyname = osi_calloc(sizeof(CONFIG_KEY) + 1); char *keyname = osi_calloc(sizeof(CONFIG_KEY) + 1);
int config_size = get_config_size(config); int config_size = get_config_size(config);
char *buf = osi_calloc(config_size + 100); char *buf = osi_calloc(config_size);
if (!line || !buf || !keyname) { if (!line || !buf || !keyname) {
err_code |= 0x01; err_code |= 0x01;
goto error; goto error;
@ -407,6 +407,16 @@ bool config_save(const config_t *config, const char *filename)
for (const list_node_t *node = list_begin(config->sections); node != list_end(config->sections); node = list_next(node)) { for (const list_node_t *node = list_begin(config->sections); node != list_end(config->sections); node = list_next(node)) {
const section_t *section = (const section_t *)list_node(node); const section_t *section = (const section_t *)list_node(node);
w_cnt = snprintf(line, 1024, "[%s]\n", section->name); w_cnt = snprintf(line, 1024, "[%s]\n", section->name);
if(w_cnt < 0) {
OSI_TRACE_ERROR("snprintf error w_cnt %d.",w_cnt);
err_code |= 0x10;
goto error;
}
if(w_cnt_total + w_cnt > config_size) {
OSI_TRACE_ERROR("%s, memcpy size (w_cnt + w_cnt_total = %d) is larger than buffer size (config_size = %d).", __func__, (w_cnt + w_cnt_total), config_size);
err_code |= 0x20;
goto error;
}
OSI_TRACE_DEBUG("section name: %s, w_cnt + w_cnt_total = %d\n", section->name, w_cnt + w_cnt_total); OSI_TRACE_DEBUG("section name: %s, w_cnt + w_cnt_total = %d\n", section->name, w_cnt + w_cnt_total);
memcpy(buf + w_cnt_total, line, w_cnt); memcpy(buf + w_cnt_total, line, w_cnt);
w_cnt_total += w_cnt; w_cnt_total += w_cnt;
@ -415,6 +425,16 @@ bool config_save(const config_t *config, const char *filename)
const entry_t *entry = (const entry_t *)list_node(enode); const entry_t *entry = (const entry_t *)list_node(enode);
OSI_TRACE_DEBUG("(key, val): (%s, %s)\n", entry->key, entry->value); OSI_TRACE_DEBUG("(key, val): (%s, %s)\n", entry->key, entry->value);
w_cnt = snprintf(line, 1024, "%s = %s\n", entry->key, entry->value); w_cnt = snprintf(line, 1024, "%s = %s\n", entry->key, entry->value);
if(w_cnt < 0) {
OSI_TRACE_ERROR("snprintf error w_cnt %d.",w_cnt);
err_code |= 0x10;
goto error;
}
if(w_cnt_total + w_cnt > config_size) {
OSI_TRACE_ERROR("%s, memcpy size (w_cnt + w_cnt_total = %d) is larger than buffer size.(config_size = %d)", __func__, (w_cnt + w_cnt_total), config_size);
err_code |= 0x20;
goto error;
}
OSI_TRACE_DEBUG("%s, w_cnt + w_cnt_total = %d", __func__, w_cnt + w_cnt_total); OSI_TRACE_DEBUG("%s, w_cnt + w_cnt_total = %d", __func__, w_cnt + w_cnt_total);
memcpy(buf + w_cnt_total, line, w_cnt); memcpy(buf + w_cnt_total, line, w_cnt);
w_cnt_total += w_cnt; w_cnt_total += w_cnt;
@ -520,7 +540,10 @@ static void config_parse(nvs_handle fp, config_t *config)
char *section = osi_calloc(1024); char *section = osi_calloc(1024);
char *keyname = osi_calloc(sizeof(CONFIG_KEY) + 1); char *keyname = osi_calloc(sizeof(CONFIG_KEY) + 1);
int buf_size = get_config_size_from_flash(fp); int buf_size = get_config_size_from_flash(fp);
char *buf = osi_calloc(buf_size + 100); char *buf = osi_calloc(buf_size);
if(buf_size == 0) { //First use nvs
goto error;
}
if (!line || !section || !buf || !keyname) { if (!line || !section || !buf || !keyname) {
err_code |= 0x01; err_code |= 0x01;
goto error; goto error;

View file

@ -131,17 +131,19 @@ size_t fixed_queue_capacity(fixed_queue_t *queue)
void fixed_queue_enqueue(fixed_queue_t *queue, void *data) void fixed_queue_enqueue(fixed_queue_t *queue, void *data)
{ {
bool status=false; //Flag whether enqueued success
assert(queue != NULL); assert(queue != NULL);
assert(data != NULL); assert(data != NULL);
osi_sem_take(&queue->enqueue_sem, OSI_SEM_MAX_TIMEOUT); osi_sem_take(&queue->enqueue_sem, OSI_SEM_MAX_TIMEOUT);
osi_mutex_lock(&queue->lock, OSI_MUTEX_MAX_TIMEOUT); osi_mutex_lock(&queue->lock, OSI_MUTEX_MAX_TIMEOUT);
status = list_append(queue->list, data); //Check whether enqueued success
list_append(queue->list, data);
osi_mutex_unlock(&queue->lock); osi_mutex_unlock(&queue->lock);
osi_sem_give(&queue->dequeue_sem); if(status == true)
osi_sem_give(&queue->dequeue_sem);
} }
void *fixed_queue_dequeue(fixed_queue_t *queue) void *fixed_queue_dequeue(fixed_queue_t *queue)