aisl-sdk/components/query.c

155 lines
2.7 KiB
C

/******************************************************************************
*
* Copyright (c) 2017-2019 by Löwenware Ltd
* Please, refer LICENSE file for legal information
*
******************************************************************************/
/**
* @file query.c
* @author Ilja Kartašov <ik@lowenware.com>
* @brief HTTP POST/GET query stream parser source file
*
* @see https://lowenware.com/aisl/
*/
#include <stdlib.h>
#include <stdarg.h>
#include <string.h>
#include "log.h"
#include "query.h"
AislStatus
ax_query_init(AxQuery query, size_t total, AxQueryHandler on_var, void *p_ctx)
{
query->on_var = on_var;
query->p_ctx = p_ctx;
query->data = NULL;
query->size = 0;
query->total = total;
query->separator = 0;
return AISL_SUCCESS;
}
void
ax_query_release(AxQuery query)
{
if (query->data)
free(query->data);
}
static void
ax_query_notify(const char *key, uint32_t k_len,
const char *val, uint32_t v_len,
AxQuery query)
{
if (query->on_var) {
if (query->on_var(key, k_len, val, v_len, query->p_ctx) != 0)
query->on_var = NULL;
}
}
static int32_t
ax_query_process(AxQuery query, const char *data, int32_t length)
{
const char *i = data,
*key = data,
*val = NULL;
uint32_t k_len = 0;
size_t result;
for (result=0; result < length; result++) {
char c;
switch( (c = *i) ) {
case '=':
if (val)
return -1;
k_len = i-key;
val = i+1;
break;
case '&':
case ';':
if (!val)
return -1;
if (!query->separator)
query->separator = c;
if (query->separator == c) {
ax_query_notify(key, k_len, val, i-val, query);
/* reset parser */
key = (c == ';') ? i+2 : i+1;
k_len = 0;
val = NULL;
break;
}
}
i++;
}
if (query->total == result) {
ax_query_notify(key, k_len, val, i-val, query);
} else {
result = (key > i) ? i - data : key - data;
}
query->total -= result;
return result;
}
AislStatus
ax_query_feed(AxQuery query, const char *data, int32_t length)
{
size_t processed;
const char *source;
if (query->data) {
if (!(source = realloc(query->data, length + query->size)))
return AISL_MALLOC_ERROR;
length += query->size;
query->data = (char*)source;
query->size = length;
memcpy((char*)&source[query->size], data, length);
} else {
source = data;
}
processed = ax_query_process(query, source, length);
if (processed < 0)
return AISL_INPUT_ERROR;
if (processed < length) {
char *buffer;
length -= processed;
if (!(buffer = malloc(length)))
return AISL_MALLOC_ERROR;
memcpy(buffer, &source[processed], length);
if (query->data)
free(query->data);
query->data = buffer;
query->size = length;
}
return AISL_SUCCESS;
}