133 lines
2.7 KiB
C
133 lines
2.7 KiB
C
#include <limits.h>
|
|
#include <stdarg.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include "kstring.h"
|
|
|
|
int nfail = 0;
|
|
|
|
void check(const char *what, const kstring_t *ks, const char *correct)
|
|
{
|
|
if (ks->l != strlen(correct) || strcmp(ks->s, correct) != 0) {
|
|
fprintf(stderr, "%s produced \"%.*s\" (\"%s\" is correct)\tFAIL\n", what, (int)(ks->l), ks->s, correct);
|
|
nfail++;
|
|
}
|
|
}
|
|
|
|
void test_kputw(kstring_t *ks, int n)
|
|
{
|
|
char buf[16];
|
|
|
|
ks->l = 0;
|
|
kputw(n, ks);
|
|
|
|
sprintf(buf, "%d", n);
|
|
check("kputw()", ks, buf);
|
|
}
|
|
|
|
void test_kputl(kstring_t *ks, long n)
|
|
{
|
|
char buf[24];
|
|
|
|
ks->l = 0;
|
|
kputl(n, ks);
|
|
|
|
sprintf(buf, "%ld", n);
|
|
check("kputl()", ks, buf);
|
|
}
|
|
|
|
static char *mem_gets(char *buf, int buflen, void *vtextp)
|
|
{
|
|
const char **textp = (const char **) vtextp;
|
|
|
|
const char *nl = strchr(*textp, '\n');
|
|
size_t n = nl? nl - *textp + 1 : strlen(*textp);
|
|
|
|
if (n == 0) return NULL;
|
|
|
|
if (n > buflen-1) n = buflen-1;
|
|
memcpy(buf, *textp, n);
|
|
buf[n] = '\0';
|
|
*textp += n;
|
|
return buf;
|
|
}
|
|
|
|
void test_kgetline(kstring_t *ks, const char *text, ...)
|
|
{
|
|
const char *exp;
|
|
va_list arg;
|
|
|
|
va_start(arg, text);
|
|
while ((exp = va_arg(arg, const char *)) != NULL) {
|
|
ks->l = 0;
|
|
if (kgetline(ks, mem_gets, &text) != 0) kputs("EOF", ks);
|
|
check("kgetline()", ks, exp);
|
|
}
|
|
va_end(arg);
|
|
|
|
ks->l = 0;
|
|
if (kgetline(ks, mem_gets, &text) == 0) check("kgetline()", ks, "EOF");
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
kstring_t ks;
|
|
|
|
ks.l = ks.m = 0;
|
|
ks.s = NULL;
|
|
|
|
test_kputw(&ks, 0);
|
|
test_kputw(&ks, 1);
|
|
test_kputw(&ks, 37);
|
|
test_kputw(&ks, 12345);
|
|
test_kputw(&ks, -12345);
|
|
test_kputw(&ks, INT_MAX);
|
|
test_kputw(&ks, -INT_MAX);
|
|
test_kputw(&ks, INT_MIN);
|
|
|
|
test_kputl(&ks, 0);
|
|
test_kputl(&ks, 1);
|
|
test_kputl(&ks, 37);
|
|
test_kputl(&ks, 12345);
|
|
test_kputl(&ks, -12345);
|
|
test_kputl(&ks, INT_MAX);
|
|
test_kputl(&ks, -INT_MAX);
|
|
test_kputl(&ks, INT_MIN);
|
|
test_kputl(&ks, LONG_MAX);
|
|
test_kputl(&ks, -LONG_MAX);
|
|
test_kputl(&ks, LONG_MIN);
|
|
|
|
test_kgetline(&ks, "", NULL);
|
|
test_kgetline(&ks, "apple", "apple", NULL);
|
|
test_kgetline(&ks, "banana\n", "banana", NULL);
|
|
test_kgetline(&ks, "carrot\r\n", "carrot", NULL);
|
|
test_kgetline(&ks, "\n", "", NULL);
|
|
test_kgetline(&ks, "\n\n", "", "", NULL);
|
|
test_kgetline(&ks, "foo\nbar", "foo", "bar", NULL);
|
|
test_kgetline(&ks, "foo\nbar\n", "foo", "bar", NULL);
|
|
test_kgetline(&ks,
|
|
"abcdefghijklmnopqrstuvwxyz0123456789\nABCDEFGHIJKLMNOPQRSTUVWXYZ\n",
|
|
"abcdefghijklmnopqrstuvwxyz0123456789",
|
|
"ABCDEFGHIJKLMNOPQRSTUVWXYZ", NULL);
|
|
|
|
if (argc > 1) {
|
|
FILE *f = fopen(argv[1], "r");
|
|
if (f) {
|
|
for (ks.l = 0; kgetline(&ks, (kgets_func *)fgets, f) == 0; ks.l = 0)
|
|
puts(ks.s);
|
|
fclose(f);
|
|
}
|
|
}
|
|
|
|
free(ks.s);
|
|
|
|
if (nfail > 0) {
|
|
fprintf(stderr, "Total failures: %d\n", nfail);
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|