lib/fold: fix warning with -DDEBUG
[petitboot] / lib / fold / fold.c
1
2 #define _GNU_SOURCE
3
4 #include <assert.h>
5 #include <string.h>
6 #include <stdio.h>
7 #include <wchar.h>
8 #include <wctype.h>
9
10 #include "fold/fold.h"
11
12 void fold_text(const char *text,
13                 int linelen,
14                 int line_cb(void *arg, const char *start, int len),
15                 void *arg)
16 {
17         const char *start, *end, *sep;
18         size_t sep_bytes, len;
19         int col, rc = 0;
20         mbstate_t ps;
21
22         /* start, end and sep are byte-positions in the string, and should always
23          * lie on the start of a multibyte sequence */
24         start = end = sep = text;
25         sep_bytes = 0;
26         col = 0;
27         len = strlen(text);
28         memset(&ps, 0, sizeof(ps));
29
30         while (!rc) {
31                 size_t bytes;
32                 wchar_t wc;
33                 int width;
34
35                 bytes = mbrtowc(&wc, end, len - (end - text), &ps);
36
37                 assert(bytes != (size_t)-1);
38
39                 /* we'll get a zero size for the nul terminator */
40                 if (!bytes) {
41                         line_cb(arg, start, end - start);
42                         break;
43                 }
44
45                 if (wc == L'\n') {
46                         rc = line_cb(arg, start, end - start);
47                         start = sep = end += bytes;
48                         sep_bytes = 0;
49                         col = 0;
50                         continue;
51                 }
52
53                 width = wcwidth(wc);
54
55                 /* we should have caught this in the !bytes check... */
56                 if (width == 0) {
57                         line_cb(arg, start, end - start);
58                         break;
59                 }
60
61                 /* unprintable character? just add it to the current line */
62                 if (width < 0) {
63                         end += bytes;
64                         continue;
65                 }
66
67                 col += width;
68
69                 if (col > linelen) {
70                         if (sep != start) {
71                                 /* split on a previous word boundary, if
72                                  * possible */
73                                 rc = line_cb(arg, start, sep - start);
74                                 end = sep + sep_bytes;
75                         } else {
76                                 /* otherwise, break the word */
77                                 rc = line_cb(arg, start, end - start);
78                         }
79                         sep_bytes = 0;
80                         start = sep = end;
81                         col = 0;
82
83                 } else {
84                         /* record our last separator */
85                         if (wc == L' ') {
86                                 sep = end;
87                                 sep_bytes = bytes;
88                         }
89                         end += bytes;
90                 }
91         }
92 }