kolibrios/contrib/network/netsurf/libhubbub/test/csdetect.c

124 lines
2.5 KiB
C
Raw Normal View History

#include <inttypes.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <parserutils/charset/mibenum.h>
#include <hubbub/hubbub.h>
#include "charset/detect.h"
#include "utils/utils.h"
#include "testutils.h"
typedef struct line_ctx {
size_t buflen;
size_t bufused;
uint8_t *buf;
char enc[64];
bool indata;
bool inenc;
} line_ctx;
static bool handle_line(const char *data, size_t datalen, void *pw);
static void run_test(const uint8_t *data, size_t len, char *expected);
int main(int argc, char **argv)
{
line_ctx ctx;
if (argc != 2) {
printf("Usage: %s <filename>\n", argv[0]);
return 1;
}
ctx.buflen = parse_filesize(argv[1]);
if (ctx.buflen == 0)
return 1;
ctx.buf = malloc(ctx.buflen);
if (ctx.buf == NULL) {
printf("Failed allocating %u bytes\n",
(unsigned int) ctx.buflen);
return 1;
}
ctx.buf[0] = '\0';
ctx.enc[0] = '\0';
ctx.bufused = 0;
ctx.indata = false;
ctx.inenc = false;
assert(parse_testfile(argv[1], handle_line, &ctx) == true);
/* and run final test */
if (ctx.bufused > 0 && ctx.buf[ctx.bufused - 1] == '\n')
ctx.bufused -= 1;
run_test(ctx.buf, ctx.bufused, ctx.enc);
free(ctx.buf);
printf("PASS\n");
return 0;
}
bool handle_line(const char *data, size_t datalen, void *pw)
{
line_ctx *ctx = (line_ctx *) pw;
if (data[0] == '#') {
if (ctx->inenc) {
/* This marks end of testcase, so run it */
if (ctx->buf[ctx->bufused - 1] == '\n')
ctx->bufused -= 1;
run_test(ctx->buf, ctx->bufused, ctx->enc);
ctx->buf[0] = '\0';
ctx->enc[0] = '\0';
ctx->bufused = 0;
}
ctx->indata = (strncasecmp(data+1, "data", 4) == 0);
ctx->inenc = (strncasecmp(data+1, "encoding", 8) == 0);
} else {
if (ctx->indata) {
memcpy(ctx->buf + ctx->bufused, data, datalen);
ctx->bufused += datalen;
}
if (ctx->inenc) {
strcpy(ctx->enc, data);
if (ctx->enc[strlen(ctx->enc) - 1] == '\n')
ctx->enc[strlen(ctx->enc) - 1] = '\0';
}
}
return true;
}
void run_test(const uint8_t *data, size_t len, char *expected)
{
uint16_t mibenum = 0;
hubbub_charset_source source = HUBBUB_CHARSET_UNKNOWN;
static int testnum;
assert(hubbub_charset_extract(data, len,
&mibenum, &source) == HUBBUB_OK);
assert(mibenum != 0);
printf("%d: Detected charset %s (%d) Source %d Expected %s (%d)\n",
++testnum, parserutils_charset_mibenum_to_name(mibenum),
mibenum, source, expected,
parserutils_charset_mibenum_from_name(
expected, strlen(expected)));
assert(mibenum == parserutils_charset_mibenum_from_name(
expected, strlen(expected)));
}