This source file includes following definitions.
- basic
- main
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17 #include <config.h>
18 #include <stdbool.h>
19 #include <sys/types.h>
20 #include <sys/stat.h>
21 #include <fcntl.h>
22 #include <string.h>
23 #include <stdio.h>
24 #include <unistd.h>
25
26 #include "readtokens.h"
27 #include "closeout.h"
28 #include "macros.h"
29
30 static void
31 basic (void)
32 {
33 char const *filename = "in.827";
34 int fd = open (filename, O_CREAT | O_WRONLY, 0600);
35 ASSERT (fd >= 0);
36 ASSERT (write (fd, "a|b;c+d", 7) == 7);
37 ASSERT (close (fd) == 0);
38
39 {
40 token_buffer tb;
41 FILE *fp = fopen (filename, "r");
42 ASSERT (fp);
43
44 init_tokenbuffer (&tb);
45 ASSERT (readtoken (fp, "|;", 2, &tb) == 1 && tb.buffer[0] == 'a');
46 ASSERT (readtoken (fp, "|;", 2, &tb) == 1 && tb.buffer[0] == 'b');
47 ASSERT (readtoken (fp, "+", 1, &tb) == 1 && tb.buffer[0] == 'c');
48 ASSERT (readtoken (fp, "-", 1, &tb) == 1 && tb.buffer[0] == 'd');
49 ASSERT (readtoken (fp, "%", 0, &tb) == (size_t) -1);
50 ASSERT ( ! ferror (fp));
51 ASSERT (fclose (fp) == 0);
52 }
53 }
54
55 int
56 main (int argc, char **argv)
57 {
58 token_buffer tb;
59 char const *delim;
60 size_t delim_len;
61
62 atexit (close_stdout);
63
64 if (argc == 1)
65 {
66 basic ();
67 return 0;
68 }
69
70 init_tokenbuffer (&tb);
71
72 if (argc != 2)
73 return 99;
74
75 delim = argv[1];
76 delim_len = strlen (delim);
77
78 if (STREQ (delim, "\\0"))
79 {
80 delim = "";
81 delim_len = 1;
82 }
83
84 while (1)
85 {
86 size_t token_length = readtoken (stdin, delim, delim_len, &tb);
87 if (token_length == (size_t) -1)
88 break;
89 fwrite (tb.buffer, 1, token_length, stdout);
90 putchar (':');
91 }
92 putchar ('\n');
93 free (tb.buffer);
94
95 ASSERT ( ! ferror (stdin));
96
97 return 0;
98 }