rename *.cpp to *.c; renmae API. move benchmarks/* to test/decoder/http; fix TSG-22347
This commit is contained in:
lijia
289
decoders/http/http_decoder_string.c
Normal file
289
decoders/http/http_decoder_string.c
Normal file
@@ -0,0 +1,289 @@
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <assert.h>
|
||||
#include "http_decoder_private.h"
|
||||
|
||||
static const char *string_state_to_desc(enum string_state state)
|
||||
{
|
||||
switch (state)
|
||||
{
|
||||
case STRING_STATE_INIT:
|
||||
return "init";
|
||||
break;
|
||||
case STRING_STATE_REFER:
|
||||
return "refer";
|
||||
break;
|
||||
case STRING_STATE_CACHE:
|
||||
return "cache";
|
||||
break;
|
||||
case STRING_STATE_COMMIT:
|
||||
return "commit";
|
||||
break;
|
||||
default:
|
||||
return "unknown";
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
void http_decoder_string_refer(struct http_decoder_string *rstr, const char *at, size_t length)
|
||||
{
|
||||
if (NULL == rstr)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
switch (rstr->state)
|
||||
{
|
||||
case STRING_STATE_INIT:
|
||||
case STRING_STATE_CACHE:
|
||||
rstr->refer.iov_base = (char *)at;
|
||||
rstr->refer.iov_len = length;
|
||||
break;
|
||||
default:
|
||||
abort();
|
||||
break;
|
||||
}
|
||||
|
||||
rstr->state = STRING_STATE_REFER;
|
||||
}
|
||||
|
||||
static void string_refer2cache(struct http_decoder_string *rstr)
|
||||
{
|
||||
if (0 == rstr->refer.iov_len)
|
||||
{
|
||||
return;
|
||||
}
|
||||
if (rstr->cache.iov_len >= rstr->max_cache_size)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
size_t length = rstr->cache.iov_len + rstr->refer.iov_len;
|
||||
if (length > rstr->max_cache_size)
|
||||
{
|
||||
length = rstr->max_cache_size;
|
||||
}
|
||||
|
||||
if (NULL == rstr->cache.iov_base)
|
||||
{
|
||||
rstr->cache.iov_base = CALLOC(char, length + 1);
|
||||
memcpy(rstr->cache.iov_base, rstr->refer.iov_base, length);
|
||||
}
|
||||
else
|
||||
{
|
||||
rstr->cache.iov_base = REALLOC(char, rstr->cache.iov_base, length + 1);
|
||||
memcpy((char *)rstr->cache.iov_base + rstr->cache.iov_len, rstr->refer.iov_base,
|
||||
(length - rstr->cache.iov_len));
|
||||
}
|
||||
|
||||
rstr->cache.iov_len = length;
|
||||
rstr->refer.iov_base = NULL;
|
||||
rstr->refer.iov_len = 0;
|
||||
}
|
||||
|
||||
static void string_commit2cache(struct http_decoder_string *rstr)
|
||||
{
|
||||
if (rstr->cache.iov_len == rstr->commit.iov_len &&
|
||||
rstr->cache.iov_base == rstr->commit.iov_base)
|
||||
{
|
||||
rstr->commit.iov_base = NULL;
|
||||
rstr->commit.iov_len = 0;
|
||||
return;
|
||||
}
|
||||
|
||||
// Only http header key need to backward to cache
|
||||
size_t length = 0;
|
||||
if (rstr->commit.iov_len > rstr->max_cache_size)
|
||||
{
|
||||
length = rstr->max_cache_size;
|
||||
}
|
||||
else
|
||||
{
|
||||
length = rstr->commit.iov_len;
|
||||
}
|
||||
|
||||
if (length > 0)
|
||||
{
|
||||
if (NULL == rstr->cache.iov_base)
|
||||
{
|
||||
rstr->cache.iov_base = CALLOC(char, length + 1);
|
||||
}
|
||||
else
|
||||
{
|
||||
abort();
|
||||
}
|
||||
memcpy(rstr->cache.iov_base, rstr->commit.iov_base, length);
|
||||
rstr->cache.iov_len = length;
|
||||
|
||||
rstr->commit.iov_base = NULL;
|
||||
rstr->commit.iov_len = 0;
|
||||
}
|
||||
}
|
||||
|
||||
void http_decoder_string_cache(struct http_decoder_string *rstr)
|
||||
{
|
||||
if (NULL == rstr)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
switch (rstr->state)
|
||||
{
|
||||
case STRING_STATE_REFER:
|
||||
string_refer2cache(rstr);
|
||||
break;
|
||||
case STRING_STATE_CACHE:
|
||||
break;
|
||||
case STRING_STATE_COMMIT:
|
||||
// commit backward to cache
|
||||
string_commit2cache(rstr);
|
||||
break;
|
||||
default:
|
||||
abort();
|
||||
break;
|
||||
}
|
||||
rstr->state = STRING_STATE_CACHE;
|
||||
}
|
||||
|
||||
void http_decoder_string_commit(struct http_decoder_string *rstr)
|
||||
{
|
||||
if (NULL == rstr)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
switch (rstr->state)
|
||||
{
|
||||
case STRING_STATE_REFER:
|
||||
if (rstr->cache.iov_len)
|
||||
{
|
||||
http_decoder_string_cache(rstr);
|
||||
|
||||
rstr->commit.iov_base = rstr->cache.iov_base;
|
||||
rstr->commit.iov_len = rstr->cache.iov_len;
|
||||
// not overwrite rstr->cache.iov_base
|
||||
}
|
||||
else
|
||||
{
|
||||
rstr->commit.iov_base = rstr->refer.iov_base;
|
||||
rstr->commit.iov_len = rstr->refer.iov_len;
|
||||
|
||||
rstr->refer.iov_base = NULL;
|
||||
rstr->refer.iov_len = 0;
|
||||
}
|
||||
break;
|
||||
case STRING_STATE_CACHE:
|
||||
rstr->commit.iov_base = rstr->cache.iov_base;
|
||||
rstr->commit.iov_len = rstr->cache.iov_len;
|
||||
// not overwrite rstr->cache.iov_base
|
||||
break;
|
||||
default:
|
||||
// abort();
|
||||
break;
|
||||
}
|
||||
|
||||
rstr->state = STRING_STATE_COMMIT;
|
||||
}
|
||||
|
||||
void http_decoder_string_reset(struct http_decoder_string *rstr)
|
||||
{
|
||||
assert(rstr);
|
||||
|
||||
switch (rstr->state)
|
||||
{
|
||||
case STRING_STATE_INIT:
|
||||
case STRING_STATE_REFER:
|
||||
case STRING_STATE_CACHE:
|
||||
case STRING_STATE_COMMIT:
|
||||
FREE(rstr->cache.iov_base);
|
||||
memset(rstr, 0, sizeof(struct http_decoder_string));
|
||||
break;
|
||||
default:
|
||||
abort();
|
||||
break;
|
||||
}
|
||||
|
||||
rstr->state = STRING_STATE_INIT;
|
||||
}
|
||||
|
||||
void http_decoder_string_init(struct http_decoder_string *rstr, size_t max_cache_size)
|
||||
{
|
||||
rstr->max_cache_size = max_cache_size;
|
||||
}
|
||||
|
||||
void http_decoder_string_reinit(struct http_decoder_string *rstr)
|
||||
{
|
||||
if (rstr->state == STRING_STATE_CACHE)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
if (rstr->state == STRING_STATE_COMMIT &&
|
||||
rstr->cache.iov_base == rstr->commit.iov_base &&
|
||||
rstr->cache.iov_len == rstr->commit.iov_len)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
if (rstr->cache.iov_base != NULL)
|
||||
{
|
||||
FREE(rstr->cache.iov_base);
|
||||
rstr->cache.iov_len = 0;
|
||||
}
|
||||
|
||||
#if 0
|
||||
rstr->refer.iov_base = NULL;
|
||||
rstr->refer.iov_len = 0;
|
||||
rstr->commit.iov_base = NULL;
|
||||
rstr->commit.iov_len = 0;
|
||||
rstr->state = STRING_STATE_INIT;
|
||||
#endif
|
||||
}
|
||||
|
||||
enum string_state http_decoder_string_state(const struct http_decoder_string *rstr)
|
||||
{
|
||||
return rstr->state;
|
||||
}
|
||||
|
||||
int http_decoder_string_get(const struct http_decoder_string *rstr, char **name, size_t *name_len)
|
||||
{
|
||||
if (NULL == rstr || NULL == name || 0 == name_len)
|
||||
{
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (http_decoder_string_state(rstr) == STRING_STATE_COMMIT)
|
||||
{
|
||||
*name = rstr->commit.iov_base;
|
||||
*name_len = rstr->commit.iov_len;
|
||||
}
|
||||
else
|
||||
{
|
||||
*name = NULL;
|
||||
*name_len = 0;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
void http_decoder_string_dump(struct http_decoder_string *rstr, const char *desc)
|
||||
{
|
||||
if (NULL == rstr)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
char *refer_str = http_safe_dup((char *)rstr->refer.iov_base, rstr->refer.iov_len);
|
||||
char *cache_str = http_safe_dup((char *)rstr->cache.iov_base, rstr->cache.iov_len);
|
||||
char *commit_str = http_safe_dup((char *)rstr->commit.iov_base, rstr->commit.iov_len);
|
||||
|
||||
printf("%s: state: %s, refer: {len: %02zu, iov_base: %s}, cache: {len: %02zu, iov_base: %s}, commit: {len: %02zu, iov_base: %s}\n",
|
||||
desc, string_state_to_desc(rstr->state),
|
||||
rstr->refer.iov_len, refer_str,
|
||||
rstr->cache.iov_len, cache_str,
|
||||
rstr->commit.iov_len, commit_str);
|
||||
|
||||
FREE(refer_str);
|
||||
FREE(cache_str);
|
||||
FREE(commit_str);
|
||||
}
|
||||
Reference in New Issue
Block a user