/[LeafOK_CVS]/lbbs/src/lml.c
ViewVC logotype

Contents of /lbbs/src/lml.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1.35 - (show annotations)
Thu Oct 30 03:18:56 2025 UTC (4 months, 2 weeks ago) by sysadm
Branch: MAIN
Changes since 1.34: +18 -20 lines
Content type: text/x-csrc
Fix bug in handling of invalid ANSI control sequence

1 /***************************************************************************
2 lml.h - description
3 -------------------
4 Copyright : (C) 2004-2025 by Leaflet
5 Email : leaflet@leafok.com
6 ***************************************************************************/
7
8 /***************************************************************************
9 * *
10 * This program is free software; you can redistribute it and/or modify *
11 * it under the terms of the GNU General Public License as published by *
12 * the Free Software Foundation; either version 3 of the License, or *
13 * (at your option) any later version. *
14 * *
15 ***************************************************************************/
16
17 #include "common.h"
18 #include "lml.h"
19 #include "log.h"
20 #include <stdio.h>
21 #include <string.h>
22 #include <sys/param.h>
23
24 #define LML_TAG_PARAM_BUF_LEN 256
25 #define LML_TAG_OUTPUT_BUF_LEN 1024
26
27 typedef int (*lml_tag_filter_cb)(const char *tag_name, const char *tag_param_buf, char *tag_output_buf, size_t tag_output_buf_len, int quote_mode);
28
29 static int lml_tag_color_filter(const char *tag_name, const char *tag_param_buf, char *tag_output_buf, size_t tag_output_buf_len, int quote_mode)
30 {
31 if (strcasecmp(tag_name, "color") == 0)
32 {
33 if (strcasecmp(tag_param_buf, "red") == 0)
34 {
35 return snprintf(tag_output_buf, tag_output_buf_len, "\033[1;31m");
36 }
37 else if (strcasecmp(tag_param_buf, "green") == 0)
38 {
39 return snprintf(tag_output_buf, tag_output_buf_len, "\033[1;32m");
40 }
41 else if (strcasecmp(tag_param_buf, "yellow") == 0)
42 {
43 return snprintf(tag_output_buf, tag_output_buf_len, "\033[1;33m");
44 }
45 else if (strcasecmp(tag_param_buf, "blue") == 0)
46 {
47 return snprintf(tag_output_buf, tag_output_buf_len, "\033[1;34m");
48 }
49 else if (strcasecmp(tag_param_buf, "magenta") == 0)
50 {
51 return snprintf(tag_output_buf, tag_output_buf_len, "\033[1;35m");
52 }
53 else if (strcasecmp(tag_param_buf, "cyan") == 0)
54 {
55 return snprintf(tag_output_buf, tag_output_buf_len, "\033[1;36m");
56 }
57 else if (strcasecmp(tag_param_buf, "black") == 0)
58 {
59 return snprintf(tag_output_buf, tag_output_buf_len, "\033[1;30;47m");
60 }
61 }
62 return 0;
63 }
64
65 #define LML_TAG_QUOTE_MAX_LEVEL 10
66
67 static const char *lml_tag_quote_color[] = {
68 "\033[33m", // yellow
69 "\033[32m", // green
70 "\033[35m", // magenta
71 };
72
73 static const int LML_TAG_QUOTE_LEVEL_LOOP = (int)(sizeof(lml_tag_quote_color) / sizeof(const char *));
74
75 static int lml_tag_quote_level = 0;
76
77 static int lml_tag_quote_filter(const char *tag_name, const char *tag_param_buf, char *tag_output_buf, size_t tag_output_buf_len, int quote_mode)
78 {
79 if (strcasecmp(tag_name, "quote") == 0)
80 {
81 if (lml_tag_quote_level <= LML_TAG_QUOTE_MAX_LEVEL)
82 {
83 lml_tag_quote_level++;
84 }
85 return snprintf(tag_output_buf, tag_output_buf_len, "%s",
86 lml_tag_quote_color[lml_tag_quote_level % LML_TAG_QUOTE_LEVEL_LOOP]);
87 }
88 else if (strcasecmp(tag_name, "/quote") == 0)
89 {
90 if (lml_tag_quote_level > 0)
91 {
92 lml_tag_quote_level--;
93 }
94 return snprintf(tag_output_buf, tag_output_buf_len, "%s",
95 (lml_tag_quote_level > 0 ? lml_tag_quote_color[lml_tag_quote_level % LML_TAG_QUOTE_LEVEL_LOOP] : "\033[m"));
96 }
97
98 return 0;
99 }
100
101 static int lml_tag_disabled = 0;
102
103 static int lml_tag_disable_filter(const char *tag_name, const char *tag_param_buf, char *tag_output_buf, size_t tag_output_buf_len, int quote_mode)
104 {
105 lml_tag_disabled = 1;
106
107 return snprintf(tag_output_buf, tag_output_buf_len, "%s", (quote_mode ? "[plain]" : ""));
108 }
109
110 typedef struct lml_tag_def_t
111 {
112 const char *tag_name; // tag name
113 const char *tag_output; // output string
114 const char *default_param; // default param string
115 const char *quote_mode_output; // output string in quote mode
116 lml_tag_filter_cb tag_filter_cb; // tag filter callback
117 } LML_TAG_DEF;
118
119 const LML_TAG_DEF lml_tag_def[] = {
120 // Definition of tuple: {lml_tag, lml_output, default_param, quote_mode_output, lml_filter_cb}
121 {"plain", NULL, NULL, NULL, lml_tag_disable_filter},
122 {"nolml", "", NULL, "", NULL}, // deprecated
123 {"lml", "", NULL, "", NULL}, // deprecated
124 {"align", "", "", "", NULL}, // N/A
125 {"/align", "", "", "", NULL}, // N/A
126 {"size", "", "", "", NULL}, // N/A
127 {"/size", "", "", "", NULL}, // N/A
128 {"left", "[", "", "[left]", NULL},
129 {"right", "]", "", "[right]", NULL},
130 {"bold", "\033[1m", "", "", NULL}, // does not work in Fterm
131 {"/bold", "\033[22m", NULL, "", NULL},
132 {"b", "\033[1m", "", "", NULL},
133 {"/b", "\033[22m", NULL, "", NULL},
134 {"italic", "\033[5m", "", "", NULL}, // use blink instead
135 {"/italic", "\033[m", NULL, "", NULL}, // \033[25m does not work in Fterm
136 {"i", "\033[5m", "", "", NULL},
137 {"/i", "\033[m", NULL, "", NULL},
138 {"underline", "\033[4m", "", "", NULL},
139 {"/underline", "\033[m", NULL, "", NULL}, // \033[24m does not work in Fterm
140 {"u", "\033[4m", "", "", NULL},
141 {"/u", "\033[m", NULL, "", NULL},
142 {"color", NULL, NULL, "", lml_tag_color_filter},
143 {"/color", "\033[m", NULL, "", NULL},
144 {"quote", NULL, NULL, "", lml_tag_quote_filter},
145 {"/quote", NULL, NULL, "", lml_tag_quote_filter},
146 {"url", "", "", "", NULL},
147 {"/url", "(链接: %s)", NULL, "", NULL},
148 {"link", "", "", "", NULL},
149 {"/link", "(链接: %s)", NULL, "", NULL},
150 {"email", "", "", "", NULL},
151 {"/email", "(Email: %s)", NULL, "", NULL},
152 {"user", "", "", "", NULL},
153 {"/user", "(用户: %s)", NULL, "", NULL},
154 {"article", "", "", "", NULL},
155 {"/article", "(文章: %s)", NULL, "", NULL},
156 {"image", "(图片: %s)", "", "%s", NULL},
157 {"flash", "(Flash: %s)", "", "", NULL},
158 {"bwf", "\033[1;31m****\033[m", "", "****", NULL},
159 };
160
161 #define LML_TAG_COUNT (sizeof(lml_tag_def) / sizeof(LML_TAG_DEF))
162
163 static int lml_tag_name_len[LML_TAG_COUNT];
164 static int lml_ready = 0;
165
166 inline static void lml_init(void)
167 {
168 int i;
169
170 if (!lml_ready)
171 {
172 for (i = 0; i < LML_TAG_COUNT; i++)
173 {
174 lml_tag_name_len[i] = (int)strlen(lml_tag_def[i].tag_name);
175 }
176
177 lml_ready = 1;
178 }
179 }
180
181 #define CHECK_AND_APPEND_OUTPUT(out_buf, out_buf_len, out_buf_offset, tag_out, tag_out_len) \
182 { \
183 if ((out_buf_offset) + (tag_out_len) >= (out_buf_len)) \
184 { \
185 log_error("Buffer is not longer enough for output string %d >= %d\n", (out_buf_offset) + (tag_out_len), (out_buf_len)); \
186 out_buf[out_buf_offset] = '\0'; \
187 return (out_buf_offset); \
188 } \
189 memcpy((out_buf) + (out_buf_offset), (tag_out), (size_t)(tag_out_len)); \
190 (out_buf_offset) += (tag_out_len); \
191 }
192
193 int lml_render(const char *str_in, char *str_out, int buf_len, int quote_mode)
194 {
195 char c;
196 char tag_param_buf[LML_TAG_PARAM_BUF_LEN];
197 char tag_output_buf[LML_TAG_OUTPUT_BUF_LEN];
198 int i;
199 int j = 0;
200 int k;
201 int tag_start_pos = -1;
202 int tag_name_pos = -1;
203 int tag_end_pos = -1;
204 int tag_param_pos = -1;
205 int tag_output_len;
206 int new_line = 1;
207 int fb_quote_level = 0;
208 int tag_name_found;
209
210 lml_init();
211
212 lml_tag_disabled = 0;
213 lml_tag_quote_level = 0;
214
215 for (i = 0; str_in[i] != '\0'; i++)
216 {
217 if (!quote_mode && !lml_tag_disabled && new_line)
218 {
219 fb_quote_level = 0;
220
221 while (str_in[i + fb_quote_level * 2] == ':' && str_in[i + fb_quote_level * 2 + 1] == ' ') // FB2000 quote leading str
222 {
223 fb_quote_level++;
224 }
225
226 lml_tag_quote_level += fb_quote_level;
227
228 if (lml_tag_quote_level > 0)
229 {
230 tag_output_len = snprintf(tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, "%s",
231 lml_tag_quote_color[lml_tag_quote_level % LML_TAG_QUOTE_LEVEL_LOOP]);
232 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, tag_output_buf, tag_output_len);
233 }
234
235 new_line = 0;
236 }
237
238 if (str_in[i] == '\033' && str_in[i + 1] == '[') // Escape sequence -- copy directly
239 {
240 for (k = i + 2; str_in[k] != '\0' && str_in[k] != 'm' && str_in[k] != '\033'; k++)
241 ;
242
243 if (str_in[k] != 'm') // invalid
244 {
245 k--;
246 }
247
248 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, str_in + i, k - i + 1);
249 i = k;
250 continue;
251 }
252
253 if (str_in[i] == '\n') // jump out of tag at end of line
254 {
255 if (tag_start_pos != -1) // tag is not closed
256 {
257 tag_end_pos = i - 1;
258 tag_output_len = tag_end_pos - tag_start_pos + 1;
259 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, str_in + tag_start_pos, tag_output_len);
260 }
261
262 if (fb_quote_level > 0)
263 {
264 lml_tag_quote_level -= fb_quote_level;
265
266 tag_output_len = snprintf(tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, "\033[m");
267 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, tag_output_buf, tag_output_len);
268 }
269
270 tag_start_pos = -1;
271 tag_name_pos = -1;
272 new_line = 1;
273 }
274 else if (str_in[i] == '\r')
275 {
276 continue; // ignore '\r'
277 }
278
279 if (!lml_tag_disabled && str_in[i] == '[')
280 {
281 if (tag_start_pos != -1) // tag is not closed
282 {
283 tag_end_pos = i - 1;
284 tag_output_len = tag_end_pos - tag_start_pos + 1;
285 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, str_in + tag_start_pos, tag_output_len);
286 }
287
288 tag_start_pos = i;
289 tag_name_pos = i + 1;
290 }
291 else if (!lml_tag_disabled && str_in[i] == ']' && tag_name_pos >= 0)
292 {
293 tag_end_pos = i;
294
295 // Skip space characters
296 while (str_in[tag_name_pos] == ' ')
297 {
298 tag_name_pos++;
299 }
300
301 for (tag_name_found = 0, k = 0; k < LML_TAG_COUNT; k++)
302 {
303 if (strncasecmp(lml_tag_def[k].tag_name, str_in + tag_name_pos, (size_t)lml_tag_name_len[k]) == 0)
304 {
305 tag_param_pos = -1;
306 switch (str_in[tag_name_pos + lml_tag_name_len[k]])
307 {
308 case ' ':
309 tag_name_found = 1;
310 tag_param_pos = tag_name_pos + lml_tag_name_len[k] + 1;
311 while (str_in[tag_param_pos] == ' ')
312 {
313 tag_param_pos++;
314 }
315 strncpy(tag_param_buf, str_in + tag_param_pos, (size_t)MIN(tag_end_pos - tag_param_pos, LML_TAG_PARAM_BUF_LEN));
316 tag_param_buf[MIN(tag_end_pos - tag_param_pos, LML_TAG_PARAM_BUF_LEN)] = '\0';
317 case ']':
318 tag_name_found = 1;
319 if (tag_param_pos == -1 && lml_tag_def[k].tag_output != NULL && lml_tag_def[k].default_param != NULL) // Apply default param if not defined
320 {
321 strncpy(tag_param_buf, lml_tag_def[k].default_param, LML_TAG_PARAM_BUF_LEN - 1);
322 tag_param_buf[LML_TAG_PARAM_BUF_LEN - 1] = '\0';
323 }
324 tag_output_len = 0;
325 if (!quote_mode)
326 {
327 if (lml_tag_def[k].tag_output != NULL)
328 {
329 tag_output_len = snprintf(tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, lml_tag_def[k].tag_output, tag_param_buf);
330 }
331 else if (lml_tag_def[k].tag_filter_cb != NULL)
332 {
333 tag_output_len = lml_tag_def[k].tag_filter_cb(
334 lml_tag_def[k].tag_name, tag_param_buf, tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, 0);
335 }
336 }
337 else // if (quote_mode)
338 {
339 if (lml_tag_def[k].quote_mode_output != NULL)
340 {
341 tag_output_len = snprintf(tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, lml_tag_def[k].quote_mode_output, tag_param_buf);
342 }
343 else if (lml_tag_def[k].tag_filter_cb != NULL)
344 {
345 tag_output_len = lml_tag_def[k].tag_filter_cb(
346 lml_tag_def[k].tag_name, tag_param_buf, tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, 1);
347 }
348 }
349 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, tag_output_buf, tag_output_len);
350 break;
351 default: // tag_name not match
352 continue;
353 }
354 break;
355 }
356 }
357
358 if (!tag_name_found)
359 {
360 tag_output_len = tag_end_pos - tag_start_pos + 1;
361 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, str_in + tag_start_pos, tag_output_len);
362 }
363
364 tag_start_pos = -1;
365 tag_name_pos = -1;
366 }
367 else if (lml_tag_disabled || tag_name_pos == -1) // not in LML tag
368 {
369 if (str_in[i] & 0x80) // head of multi-byte character
370 {
371 if (j + 4 >= buf_len) // Assuming UTF-8 CJK characters use 4 bytes, though most of them actually use 3 bytes
372 {
373 log_error("Buffer is not longer enough for output string %ld >= %d\n", j + 4, buf_len);
374 str_out[j] = '\0';
375 return j;
376 }
377
378 c = (str_in[i] & 0x70) << 1;
379 while (c & 0x80)
380 {
381 str_out[j++] = str_in[i++];
382 if (str_in[i] == '\0')
383 {
384 str_out[j] = '\0';
385 return j;
386 }
387 c = (c & 0x7f) << 1;
388 }
389 }
390
391 if (j + 1 >= buf_len)
392 {
393 log_error("Buffer is not longer enough for output string %ld >= %d\n", j + 1, buf_len);
394 str_out[j] = '\0';
395 return j;
396 }
397 str_out[j++] = str_in[i];
398 }
399 else // in LML tag
400 {
401 // Do nothing
402 }
403 }
404
405 if (tag_start_pos != -1) // tag is not closed
406 {
407 tag_end_pos = i - 1;
408 tag_output_len = tag_end_pos - tag_start_pos + 1;
409 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, str_in + tag_start_pos, tag_output_len);
410 }
411
412 if (!quote_mode && !lml_tag_disabled && lml_tag_quote_level > 0)
413 {
414 tag_output_len = snprintf(tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, "\033[m");
415 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, tag_output_buf, tag_output_len);
416 }
417
418 str_out[j] = '\0';
419
420 return j;
421 }

webmaster@leafok.com
ViewVC Help
Powered by ViewVC 1.3.0-beta1