71 lines
1.9 KiB
C
71 lines
1.9 KiB
C
|
#include <stdio.h>
|
||
|
|
#include <stdarg.h>
|
||
|
|
|
||
|
|
void log_message(const char *msg, ...) {
|
||
|
|
va_list args;
|
||
|
|
va_start(args, msg);
|
||
|
|
vfprintf(stderr, msg, args);
|
||
|
|
fprintf(stderr, "\n");
|
||
|
|
fflush(stderr);
|
||
|
|
}
|
||
|
|
|
||
|
|
#include "url.c" /* yes, include the source file */
|
||
|
|
|
||
|
|
static int test_parse_content_length(const char *headers, unsigned long expected, int shouldfail) {
|
||
|
|
int err;
|
||
|
|
unsigned long val;
|
||
|
|
err = parse_content_length(headers, &val);
|
||
|
|
if (err < 0 && !shouldfail) {
|
||
|
|
log_message("%s: unexpectedly failed on input '%s'", __func__, headers);
|
||
|
|
return 2;
|
||
|
|
}
|
||
|
|
if (err == 0 && shouldfail) {
|
||
|
|
log_message("%s: unexpectedly passed on invalid input '%s'", __func__, headers);
|
||
|
|
return 3;
|
||
|
|
}
|
||
|
|
if (!shouldfail && val != expected) {
|
||
|
|
log_message("%s: expected %lu, got %lu on input '%s'",
|
||
|
|
__func__, expected, val, headers);
|
||
|
|
return 5;
|
||
|
|
}
|
||
|
|
if (!shouldfail && val == expected) {
|
||
|
|
log_message("%s: OK: '%s'", __func__, headers);
|
||
|
|
}
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
#define ARRAY_SIZE(a) (sizeof(a)/sizeof(a[0]))
|
||
|
|
|
||
|
|
int main(int argc, char **argv) {
|
||
|
|
int i, err = 0;
|
||
|
|
const char *invalid_inputs[] = {
|
||
|
|
"", /* empty */
|
||
|
|
"Abyrvalg: foobar", /* no Content-Length: */
|
||
|
|
"Content-Length: xyz", /* no digits */
|
||
|
|
"Content-Length: 12xyz", /* invalid characters */
|
||
|
|
"Content-Length: 12 34", /* stray symbols */
|
||
|
|
"Content-Length: 123456789012345678901234567890", /* too big */
|
||
|
|
};
|
||
|
|
const char *valid_inputs[] = {
|
||
|
|
"Content-Length: 1234",
|
||
|
|
"Content-Length: 1234",
|
||
|
|
"Content-Length: 1234 ",
|
||
|
|
"Content-Length: 1234 ",
|
||
|
|
"Content-Length: 1234 \r\n\r\n",
|
||
|
|
};
|
||
|
|
|
||
|
|
for (i = 0; i < ARRAY_SIZE(invalid_inputs); i++) {
|
||
|
|
err += test_parse_content_length(invalid_inputs[i], 0,
|
||
|
|
/* shouldfail = */ 1);
|
||
|
|
}
|
||
|
|
for (i = 0; i < ARRAY_SIZE(valid_inputs); i++) {
|
||
|
|
err += test_parse_content_length(valid_inputs[i], 1234,
|
||
|
|
/* shouldfail = */ 0);
|
||
|
|
}
|
||
|
|
return err;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* compile command:
|
||
|
|
gcc -O2 -g -flto -o test_parse_content_length.c test_parse_content_length.c
|
||
|
|
*/
|