lbbs/src/str_process.c

/* SPDX-License-Identifier: GPL-3.0-or-later */
/*
 * str_process
 *   - common string processing features with UTF-8 support
 *
 * Copyright (C) 2004-2025  Leaflet <leaflet@leafok.com>
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include "common.h"
#include "log.h"
#include "str_process.h"
#include <ctype.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <wchar.h>

int UTF8_fixed_width = 1;

int str_length(const char *str, int skip_ctrl_seq)
{
        int str_len;
        char input_str[5];
        wchar_t wcs[2];
        int wc_len;
        int i;
        char c;
        int ret = 0;

        for (i = 0; str[i] != '\0'; i++)
        {
                c = str[i];

                if (c == '\r' || c == '\7') // skip
                {
                        continue;
                }

                if (skip_ctrl_seq && c == '\033' && str[i + 1] == '[') // Skip control sequence
                {
                        for (i = i + 2; isdigit((int)str[i]) || str[i] == ';' || str[i] == '?'; i++)
                                ;

                        if (str[i] == 'm') // valid
                        {
                                // skip
                        }
                        else if (isalpha((int)str[i]))
                        {
                                // unsupported ANSI CSI command
                        }
                        else
                        {
                                i--;
                        }

                        continue;
                }

                // Process UTF-8 Chinese characters
                if (c & 0x80) // head of multi-byte character
                {
                        str_len = 0;
                        c = (char)(c & 0xf0);
                        while (c & 0x80)
                        {
                                input_str[str_len] = str[i + str_len];
                                str_len++;
                                c = (c & 0x7f) << 1;
                        }
                        input_str[str_len] = '\0';

                        if (mbstowcs(wcs, input_str, 1) == (size_t)-1)
                        {
                                log_debug("mbstowcs(%s) error", input_str);
                                wc_len = (UTF8_fixed_width ? 2 : 1); // Fallback
                        }
                        else
                        {
                                wc_len = (UTF8_fixed_width ? 2 : wcwidth(wcs[0]));
                        }

                        i += (str_len - 1);
                        ret += wc_len;
                }
                else
                {
                        ret++;
                }
        }

        return ret;
}

int split_line(const char *buffer, int max_display_len, int *p_eol, int *p_display_len, int skip_ctrl_seq)
{
        int i;
        *p_eol = 0;
        *p_display_len = 0;
        char c;
        int str_len;
        char input_str[5];
        wchar_t wcs[2];
        int wc_len;

        for (i = 0; buffer[i] != '\0'; i++)
        {
                c = buffer[i];

                if (c == '\r' || c == '\7') // skip
                {
                        continue;
                }

                if (skip_ctrl_seq && c == '\033' && buffer[i + 1] == '[') // Skip control sequence
                {
                        i += 2;
                        while (buffer[i] != '\0' && buffer[i] != 'm')
                        {
                                i++;
                        }
                        continue;
                }

                if (c & 0x80) // head of multi-byte character
                {
                        str_len = 0;
                        c = (char)(c & 0xf0);
                        while (c & 0x80)
                        {
                                input_str[str_len] = buffer[i + str_len];
                                str_len++;
                                c = (c & 0x7f) << 1;
                        }
                        input_str[str_len] = '\0';

                        if (mbstowcs(wcs, input_str, 1) == (size_t)-1)
                        {
                                log_debug("mbstowcs(%s) error", input_str);
                                wc_len = (UTF8_fixed_width ? 2 : 1); // Fallback
                        }
                        else
                        {
                                wc_len = (UTF8_fixed_width ? 2 : wcwidth(wcs[0]));
                        }

                        if (*p_display_len + wc_len > max_display_len)
                        {
                                break;
                        }

                        i += (str_len - 1);
                        (*p_display_len) += wc_len;
                }
                else
                {
                        if (*p_display_len + 1 > max_display_len)
                        {
                                break;
                        }
                        (*p_display_len)++;

                        // \n is regarded as 1 character wide in terminal editor, which is different from Web version
                        if (c == '\n')
                        {
                                i++;
                                *p_eol = 1;
                                break;
                        }
                }
        }

        return i;
}

long split_data_lines(const char *p_buf, int max_display_len, long *p_line_offsets, long line_offsets_count,
                                          int skip_ctrl_seq, int *p_line_widths)
{
        int line_cnt = 0;
        int len;
        int end_of_line = 0;
        int display_len = 0;

        p_line_offsets[line_cnt] = 0L;

        do
        {
                len = split_line(p_buf, max_display_len, &end_of_line, &display_len, skip_ctrl_seq);

                if (p_line_widths)
                {
                        p_line_widths[line_cnt] = display_len;
                }

                // Exceed max_line_cnt
                if (line_cnt + 1 >= line_offsets_count)
                {
                        log_debug("Line count %d reaches limit %d", line_cnt + 1, line_offsets_count);
                        return line_cnt;
                }

                p_line_offsets[line_cnt + 1] = p_line_offsets[line_cnt] + len;
                line_cnt++;
                p_buf += len;
        } while (p_buf[0] != '\0' || end_of_line);

        return line_cnt;
}

int str_filter(char *buffer, int skip_ctrl_seq)
{
        int i;
        int j;

        for (i = 0, j = 0; buffer[i] != '\0'; i++)
        {
                if (buffer[i] == '\r' || buffer[i] == '\7') // skip
                {
                        continue;
                }

                if (skip_ctrl_seq && buffer[i] == '\033' && buffer[i + 1] == '[') // Skip control sequence
                {
                        i += 2;
                        while (buffer[i] != '\0' && buffer[i] != 'm')
                        {
                                i++;
                        }
                        continue;
                }

                buffer[j] = buffer[i];
                j++;
        }

        buffer[j] = '\0';

        return j;
}
1	sysadm	1.23	/* SPDX-License-Identifier: GPL-3.0-or-later */
2			/*
3			* str_process
4			* - common string processing features with UTF-8 support
5			*
6	sysadm	1.24	* Copyright (C) 2004-2025 Leaflet <leaflet@leafok.com>
7	sysadm	1.23	*/
8	sysadm	1.1
9	sysadm	1.30	#ifdef HAVE_CONFIG_H
10			#include "config.h"
11			#endif
12
13	sysadm	1.1	#include "common.h"
14			#include "log.h"
15	sysadm	1.18	#include "str_process.h"
16	sysadm	1.22	#include <ctype.h>
17	sysadm	1.1	#include <stdio.h>
18	sysadm	1.28	#include <stdlib.h>
19	sysadm	1.1	#include <string.h>
20	sysadm	1.28	#include <wchar.h>
21
22			int UTF8_fixed_width = 1;
23	sysadm	1.1
24	sysadm	1.20	int str_length(const char *str, int skip_ctrl_seq)
25			{
26	sysadm	1.28	int str_len;
27			char input_str[5];
28			wchar_t wcs[2];
29			int wc_len;
30	sysadm	1.20	int i;
31			char c;
32			int ret = 0;
33
34			for (i = 0; str[i] != '\0'; i++)
35			{
36			c = str[i];
37
38			if (c == '\r' \|\| c == '\7') // skip
39			{
40			continue;
41			}
42
43			if (skip_ctrl_seq && c == '\033' && str[i + 1] == '[') // Skip control sequence
44			{
45	sysadm	1.31	for (i = i + 2; isdigit((int)str[i]) \|\| str[i] == ';' \|\| str[i] == '?'; i++)
46	sysadm	1.22	;
47
48			if (str[i] == 'm') // valid
49			{
50			// skip
51			}
52	sysadm	1.31	else if (isalpha((int)str[i]))
53	sysadm	1.22	{
54			// unsupported ANSI CSI command
55			}
56			else
57	sysadm	1.20	{
58	sysadm	1.22	i--;
59	sysadm	1.20	}
60	sysadm	1.22
61	sysadm	1.20	continue;
62			}
63
64			// Process UTF-8 Chinese characters
65	sysadm	1.21	if (c & 0x80) // head of multi-byte character
66	sysadm	1.20	{
67	sysadm	1.28	str_len = 0;
68			c = (char)(c & 0xf0);
69	sysadm	1.21	while (c & 0x80)
70	sysadm	1.20	{
71	sysadm	1.28	input_str[str_len] = str[i + str_len];
72			str_len++;
73	sysadm	1.21	c = (c & 0x7f) << 1;
74	sysadm	1.20	}
75	sysadm	1.28	input_str[str_len] = '\0';
76	sysadm	1.20
77	sysadm	1.28	if (mbstowcs(wcs, input_str, 1) == (size_t)-1)
78			{
79	sysadm	1.34	log_debug("mbstowcs(%s) error", input_str);
80	sysadm	1.29	wc_len = (UTF8_fixed_width ? 2 : 1); // Fallback
81			}
82			else
83			{
84			wc_len = (UTF8_fixed_width ? 2 : wcwidth(wcs[0]));
85	sysadm	1.28	}
86
87			i += (str_len - 1);
88			ret += wc_len;
89	sysadm	1.20	}
90			else
91			{
92			ret++;
93			}
94			}
95
96			return ret;
97			}
98
99	sysadm	1.17	int split_line(const char buffer, int max_display_len, int p_eol, int *p_display_len, int skip_ctrl_seq)
100	sysadm	1.1	{
101	sysadm	1.7	int i;
102	sysadm	1.1	*p_eol = 0;
103	sysadm	1.4	*p_display_len = 0;
104	sysadm	1.11	char c;
105	sysadm	1.28	int str_len;
106			char input_str[5];
107			wchar_t wcs[2];
108			int wc_len;
109	sysadm	1.1
110	sysadm	1.7	for (i = 0; buffer[i] != '\0'; i++)
111	sysadm	1.1	{
112	sysadm	1.11	c = buffer[i];
113	sysadm	1.1
114			if (c == '\r' \|\| c == '\7') // skip
115			{
116			continue;
117			}
118
119	sysadm	1.17	if (skip_ctrl_seq && c == '\033' && buffer[i + 1] == '[') // Skip control sequence
120	sysadm	1.1	{
121			i += 2;
122	sysadm	1.7	while (buffer[i] != '\0' && buffer[i] != 'm')
123	sysadm	1.1	{
124			i++;
125			}
126			continue;
127			}
128
129	sysadm	1.21	if (c & 0x80) // head of multi-byte character
130	sysadm	1.1	{
131	sysadm	1.28	str_len = 0;
132			c = (char)(c & 0xf0);
133			while (c & 0x80)
134	sysadm	1.1	{
135	sysadm	1.28	input_str[str_len] = buffer[i + str_len];
136			str_len++;
137			c = (c & 0x7f) << 1;
138	sysadm	1.1	}
139	sysadm	1.28	input_str[str_len] = '\0';
140	sysadm	1.20
141	sysadm	1.28	if (mbstowcs(wcs, input_str, 1) == (size_t)-1)
142			{
143	sysadm	1.34	log_debug("mbstowcs(%s) error", input_str);
144	sysadm	1.29	wc_len = (UTF8_fixed_width ? 2 : 1); // Fallback
145	sysadm	1.28	}
146	sysadm	1.29	else
147			{
148			wc_len = (UTF8_fixed_width ? 2 : wcwidth(wcs[0]));
149			}
150
151	sysadm	1.28	if (*p_display_len + wc_len > max_display_len)
152	sysadm	1.20	{
153	sysadm	1.28	break;
154	sysadm	1.20	}
155
156	sysadm	1.28	i += (str_len - 1);
157			(*p_display_len) += wc_len;
158	sysadm	1.1	}
159			else
160			{
161	sysadm	1.6	if (*p_display_len + 1 > max_display_len)
162	sysadm	1.1	{
163			break;
164			}
165	sysadm	1.4	(*p_display_len)++;
166	sysadm	1.12
167	sysadm	1.14	// \n is regarded as 1 character wide in terminal editor, which is different from Web version
168	sysadm	1.12	if (c == '\n')
169			{
170			i++;
171			*p_eol = 1;
172			break;
173			}
174	sysadm	1.1	}
175			}
176
177			return i;
178			}
179
180	sysadm	1.19	long split_data_lines(const char p_buf, int max_display_len, long p_line_offsets, long line_offsets_count,
181			int skip_ctrl_seq, int *p_line_widths)
182	sysadm	1.1	{
183	sysadm	1.7	int line_cnt = 0;
184	sysadm	1.13	int len;
185	sysadm	1.1	int end_of_line = 0;
186	sysadm	1.4	int display_len = 0;
187
188	sysadm	1.1	p_line_offsets[line_cnt] = 0L;
189
190	sysadm	1.13	do
191	sysadm	1.1	{
192	sysadm	1.17	len = split_line(p_buf, max_display_len, &end_of_line, &display_len, skip_ctrl_seq);
193	sysadm	1.8
194	sysadm	1.19	if (p_line_widths)
195			{
196			p_line_widths[line_cnt] = display_len;
197			}
198
199	sysadm	1.8	// Exceed max_line_cnt
200	sysadm	1.9	if (line_cnt + 1 >= line_offsets_count)
201	sysadm	1.1	{
202	sysadm	1.34	log_debug("Line count %d reaches limit %d", line_cnt + 1, line_offsets_count);
203	sysadm	1.8	return line_cnt;
204	sysadm	1.1	}
205	sysadm	1.8
206			p_line_offsets[line_cnt + 1] = p_line_offsets[line_cnt] + len;
207			line_cnt++;
208			p_buf += len;
209	sysadm	1.27	} while (p_buf[0] != '\0' \|\| end_of_line);
210	sysadm	1.1
211			return line_cnt;
212			}
213	sysadm	1.14
214	sysadm	1.17	int str_filter(char *buffer, int skip_ctrl_seq)
215	sysadm	1.14	{
216			int i;
217			int j;
218
219			for (i = 0, j = 0; buffer[i] != '\0'; i++)
220			{
221	sysadm	1.15	if (buffer[i] == '\r' \|\| buffer[i] == '\7') // skip
222	sysadm	1.14	{
223			continue;
224			}
225
226	sysadm	1.17	if (skip_ctrl_seq && buffer[i] == '\033' && buffer[i + 1] == '[') // Skip control sequence
227	sysadm	1.14	{
228			i += 2;
229			while (buffer[i] != '\0' && buffer[i] != 'm')
230			{
231			i++;
232			}
233			continue;
234			}
235
236			buffer[j] = buffer[i];
237			j++;
238			}
239
240			buffer[j] = '\0';
241
242			return j;
243			}
webmaster@leafok.com	ViewVC Help
Powered by ViewVC 1.3.0-beta1