1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
/* This is test for grab_file() function
*/
#include <ccan/grab_file/grab_file.h>
#include <stdlib.h>
#include <stdio.h>
#include <err.h>
#include <sys/stat.h>
#include <ccan/grab_file/grab_file.c>
#include <ccan/tap/tap.h>
#include <string.h>
static char **strsplit(const void *ctx, const char *string, const char *delims)
{
char **lines = NULL;
unsigned int max = 64, num = 0;
lines = talloc_array(ctx, char *, max+1);
while (*string != '\0') {
unsigned int len = strcspn(string, delims);
lines[num] = talloc_array(lines, char, len + 1);
memcpy(lines[num], string, len);
lines[num][len] = '\0';
string += len;
string += strspn(string, delims) ? 1 : 0;
if (++num == max)
lines = talloc_realloc(ctx, lines, char *, max*=2 + 1);
}
lines[num] = NULL;
/* Shrink, so talloc_get_size works */
return talloc_realloc(ctx, lines, char *, num+1);
}
int
main(int argc, char *argv[])
{
unsigned int i;
char **split, *str;
int length;
struct stat st;
str = grab_file(NULL, "test/run-grab.c", NULL);
split = strsplit(str, str, "\n");
length = strlen(split[0]);
ok1(!strcmp(split[0], "/* This is test for grab_file() function"));
for (i = 1; split[i]; i++)
length += strlen(split[i]);
ok1(!strcmp(split[i-1], "/* End of grab_file() test */"));
if (stat("test/run-grab.c", &st) != 0)
/* FIXME: ditto */
if (stat("ccan/grab_file/test/run-grab.c", &st) != 0)
err(1, "Could not stat self");
ok1(st.st_size == length + i);
talloc_free(str);
return 0;
}
/* End of grab_file() test */
|