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

Annotation of /lbbs/src/lml.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1.40 - (hide annotations)
Fri Oct 31 07:57:02 2025 UTC (4 months, 2 weeks ago) by sysadm
Branch: MAIN
Changes since 1.39: +9 -7 lines
Content type: text/x-csrc
Fix bug in handling over-length invalid LML tag

1 sysadm 1.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 sysadm 1.12 #include "common.h"
18 sysadm 1.1 #include "lml.h"
19     #include "log.h"
20 sysadm 1.36 #include "str_process.h"
21 sysadm 1.37 #include <ctype.h>
22 sysadm 1.1 #include <stdio.h>
23     #include <string.h>
24     #include <sys/param.h>
25    
26 sysadm 1.2 #define LML_TAG_PARAM_BUF_LEN 256
27     #define LML_TAG_OUTPUT_BUF_LEN 1024
28 sysadm 1.1
29 sysadm 1.20 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);
30 sysadm 1.3
31 sysadm 1.22 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)
32 sysadm 1.3 {
33     if (strcasecmp(tag_name, "color") == 0)
34     {
35     if (strcasecmp(tag_param_buf, "red") == 0)
36     {
37     return snprintf(tag_output_buf, tag_output_buf_len, "\033[1;31m");
38     }
39     else if (strcasecmp(tag_param_buf, "green") == 0)
40     {
41     return snprintf(tag_output_buf, tag_output_buf_len, "\033[1;32m");
42     }
43     else if (strcasecmp(tag_param_buf, "yellow") == 0)
44     {
45     return snprintf(tag_output_buf, tag_output_buf_len, "\033[1;33m");
46     }
47     else if (strcasecmp(tag_param_buf, "blue") == 0)
48     {
49     return snprintf(tag_output_buf, tag_output_buf_len, "\033[1;34m");
50     }
51     else if (strcasecmp(tag_param_buf, "magenta") == 0)
52     {
53     return snprintf(tag_output_buf, tag_output_buf_len, "\033[1;35m");
54     }
55     else if (strcasecmp(tag_param_buf, "cyan") == 0)
56     {
57     return snprintf(tag_output_buf, tag_output_buf_len, "\033[1;36m");
58     }
59     else if (strcasecmp(tag_param_buf, "black") == 0)
60     {
61     return snprintf(tag_output_buf, tag_output_buf_len, "\033[1;30;47m");
62     }
63     }
64     return 0;
65     }
66 sysadm 1.1
67 sysadm 1.4 #define LML_TAG_QUOTE_MAX_LEVEL 10
68    
69     static const char *lml_tag_quote_color[] = {
70 sysadm 1.5 "\033[33m", // yellow
71     "\033[32m", // green
72     "\033[35m", // magenta
73 sysadm 1.4 };
74    
75 sysadm 1.30 static const int LML_TAG_QUOTE_LEVEL_LOOP = (int)(sizeof(lml_tag_quote_color) / sizeof(const char *));
76    
77 sysadm 1.4 static int lml_tag_quote_level = 0;
78    
79 sysadm 1.20 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)
80 sysadm 1.4 {
81     if (strcasecmp(tag_name, "quote") == 0)
82     {
83     if (lml_tag_quote_level <= LML_TAG_QUOTE_MAX_LEVEL)
84     {
85     lml_tag_quote_level++;
86     }
87 sysadm 1.14 return snprintf(tag_output_buf, tag_output_buf_len, "%s",
88     lml_tag_quote_color[lml_tag_quote_level % LML_TAG_QUOTE_LEVEL_LOOP]);
89 sysadm 1.4 }
90     else if (strcasecmp(tag_name, "/quote") == 0)
91     {
92     if (lml_tag_quote_level > 0)
93     {
94     lml_tag_quote_level--;
95     }
96 sysadm 1.14 return snprintf(tag_output_buf, tag_output_buf_len, "%s",
97 sysadm 1.5 (lml_tag_quote_level > 0 ? lml_tag_quote_color[lml_tag_quote_level % LML_TAG_QUOTE_LEVEL_LOOP] : "\033[m"));
98 sysadm 1.4 }
99    
100     return 0;
101     }
102    
103 sysadm 1.20 static int lml_tag_disabled = 0;
104    
105     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)
106     {
107     lml_tag_disabled = 1;
108    
109 sysadm 1.23 return snprintf(tag_output_buf, tag_output_buf_len, "%s", (quote_mode ? "[plain]" : ""));
110 sysadm 1.20 }
111    
112 sysadm 1.22 typedef struct lml_tag_def_t
113     {
114 sysadm 1.26 const char *tag_name; // tag name
115     const char *tag_output; // output string
116     const char *default_param; // default param string
117     const char *quote_mode_output; // output string in quote mode
118 sysadm 1.22 lml_tag_filter_cb tag_filter_cb; // tag filter callback
119     } LML_TAG_DEF;
120    
121     const LML_TAG_DEF lml_tag_def[] = {
122     // Definition of tuple: {lml_tag, lml_output, default_param, quote_mode_output, lml_filter_cb}
123     {"plain", NULL, NULL, NULL, lml_tag_disable_filter},
124 sysadm 1.29 {"nolml", "", NULL, "", NULL}, // deprecated
125 sysadm 1.30 {"lml", "", NULL, "", NULL}, // deprecated
126     {"align", "", "", "", NULL}, // N/A
127     {"/align", "", "", "", NULL}, // N/A
128     {"size", "", "", "", NULL}, // N/A
129     {"/size", "", "", "", NULL}, // N/A
130 sysadm 1.22 {"left", "[", "", "[left]", NULL},
131     {"right", "]", "", "[right]", NULL},
132     {"bold", "\033[1m", "", "", NULL}, // does not work in Fterm
133     {"/bold", "\033[22m", NULL, "", NULL},
134     {"b", "\033[1m", "", "", NULL},
135     {"/b", "\033[22m", NULL, "", NULL},
136 sysadm 1.26 {"italic", "\033[5m", "", "", NULL}, // use blink instead
137 sysadm 1.22 {"/italic", "\033[m", NULL, "", NULL}, // \033[25m does not work in Fterm
138     {"i", "\033[5m", "", "", NULL},
139     {"/i", "\033[m", NULL, "", NULL},
140     {"underline", "\033[4m", "", "", NULL},
141     {"/underline", "\033[m", NULL, "", NULL}, // \033[24m does not work in Fterm
142     {"u", "\033[4m", "", "", NULL},
143     {"/u", "\033[m", NULL, "", NULL},
144     {"color", NULL, NULL, "", lml_tag_color_filter},
145     {"/color", "\033[m", NULL, "", NULL},
146     {"quote", NULL, NULL, "", lml_tag_quote_filter},
147     {"/quote", NULL, NULL, "", lml_tag_quote_filter},
148     {"url", "", "", "", NULL},
149     {"/url", "(链接: %s)", NULL, "", NULL},
150     {"link", "", "", "", NULL},
151     {"/link", "(链接: %s)", NULL, "", NULL},
152     {"email", "", "", "", NULL},
153     {"/email", "(Email: %s)", NULL, "", NULL},
154     {"user", "", "", "", NULL},
155     {"/user", "(用户: %s)", NULL, "", NULL},
156     {"article", "", "", "", NULL},
157     {"/article", "(文章: %s)", NULL, "", NULL},
158     {"image", "(图片: %s)", "", "%s", NULL},
159     {"flash", "(Flash: %s)", "", "", NULL},
160     {"bwf", "\033[1;31m****\033[m", "", "****", NULL},
161 sysadm 1.1 };
162    
163 sysadm 1.24 #define LML_TAG_COUNT (sizeof(lml_tag_def) / sizeof(LML_TAG_DEF))
164 sysadm 1.3
165 sysadm 1.19 static int lml_tag_name_len[LML_TAG_COUNT];
166     static int lml_ready = 0;
167 sysadm 1.1
168     inline static void lml_init(void)
169     {
170     int i;
171    
172 sysadm 1.19 if (!lml_ready)
173 sysadm 1.1 {
174     for (i = 0; i < LML_TAG_COUNT; i++)
175     {
176 sysadm 1.22 lml_tag_name_len[i] = (int)strlen(lml_tag_def[i].tag_name);
177 sysadm 1.1 }
178    
179 sysadm 1.19 lml_ready = 1;
180 sysadm 1.1 }
181     }
182    
183 sysadm 1.36 #define CHECK_AND_APPEND_OUTPUT(out_buf, out_buf_len, out_buf_offset, tag_out, tag_out_len, line_width) \
184 sysadm 1.35 { \
185     if ((out_buf_offset) + (tag_out_len) >= (out_buf_len)) \
186     { \
187     log_error("Buffer is not longer enough for output string %d >= %d\n", (out_buf_offset) + (tag_out_len), (out_buf_len)); \
188     out_buf[out_buf_offset] = '\0'; \
189     return (out_buf_offset); \
190     } \
191     memcpy((out_buf) + (out_buf_offset), (tag_out), (size_t)(tag_out_len)); \
192 sysadm 1.36 *((out_buf) + (out_buf_offset) + (size_t)(tag_out_len)) = '\0'; \
193     (line_width) += str_length((out_buf) + (out_buf_offset), 1); \
194 sysadm 1.35 (out_buf_offset) += (tag_out_len); \
195     }
196 sysadm 1.34
197 sysadm 1.36 int lml_render(const char *str_in, char *str_out, int buf_len, int width, int quote_mode)
198 sysadm 1.1 {
199 sysadm 1.13 char c;
200 sysadm 1.2 char tag_param_buf[LML_TAG_PARAM_BUF_LEN];
201 sysadm 1.1 char tag_output_buf[LML_TAG_OUTPUT_BUF_LEN];
202     int i;
203     int j = 0;
204     int k;
205     int tag_start_pos = -1;
206 sysadm 1.26 int tag_name_pos = -1;
207 sysadm 1.1 int tag_end_pos = -1;
208 sysadm 1.2 int tag_param_pos = -1;
209 sysadm 1.1 int tag_output_len;
210 sysadm 1.5 int new_line = 1;
211     int fb_quote_level = 0;
212 sysadm 1.26 int tag_name_found;
213 sysadm 1.36 int line_width = 0;
214 sysadm 1.1
215     lml_init();
216    
217 sysadm 1.20 lml_tag_disabled = 0;
218 sysadm 1.4 lml_tag_quote_level = 0;
219    
220 sysadm 1.36 if (width <= 0)
221     {
222     width = INT_MAX;
223     }
224    
225 sysadm 1.1 for (i = 0; str_in[i] != '\0'; i++)
226     {
227 sysadm 1.36 if (!lml_tag_disabled && new_line)
228 sysadm 1.5 {
229 sysadm 1.36 while (str_in[i] == ':' && str_in[i + 1] == ' ') // FB2000 quote leading str
230 sysadm 1.5 {
231     fb_quote_level++;
232 sysadm 1.36 lml_tag_quote_level++;
233     i += 2;
234 sysadm 1.5 }
235    
236 sysadm 1.36 if (!quote_mode && lml_tag_quote_level > 0)
237 sysadm 1.5 {
238 sysadm 1.14 tag_output_len = snprintf(tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, "%s",
239 sysadm 1.31 lml_tag_quote_color[lml_tag_quote_level % LML_TAG_QUOTE_LEVEL_LOOP]);
240 sysadm 1.36 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, tag_output_buf, tag_output_len, line_width);
241 sysadm 1.5 }
242    
243 sysadm 1.36 for (k = 0; k < fb_quote_level; k++)
244     {
245     tag_output_buf[k * 2] = ':';
246     tag_output_buf[k * 2 + 1] = ' ';
247     }
248     tag_output_buf[fb_quote_level * 2] = '\0';
249     tag_output_len = fb_quote_level * 2;
250     CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, tag_output_buf, tag_output_len, line_width);
251    
252 sysadm 1.5 new_line = 0;
253     }
254    
255 sysadm 1.39 if (lml_tag_disabled && new_line)
256     {
257     new_line = 0;
258     }
259    
260 sysadm 1.37 if (str_in[i] == '\033' && str_in[i + 1] == '[') // Escape sequence
261 sysadm 1.8 {
262 sysadm 1.37 for (k = i + 2; isdigit(str_in[k]) || str_in[k] == ';' || str_in[k] == '?'; k++)
263 sysadm 1.8 ;
264    
265 sysadm 1.37 if (str_in[k] == 'm') // valid -- copy directly
266     {
267     CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, str_in + i, k - i + 1, line_width);
268     }
269     else if (isalpha(str_in[k]))
270     {
271     // unsupported ANSI CSI command
272     }
273     else
274 sysadm 1.8 {
275 sysadm 1.35 k--;
276 sysadm 1.8 }
277 sysadm 1.35
278     i = k;
279     continue;
280 sysadm 1.8 }
281    
282 sysadm 1.28 if (str_in[i] == '\n') // jump out of tag at end of line
283 sysadm 1.5 {
284 sysadm 1.28 if (tag_start_pos != -1) // tag is not closed
285     {
286     tag_end_pos = i - 1;
287     tag_output_len = tag_end_pos - tag_start_pos + 1;
288 sysadm 1.38
289     if (line_width + tag_output_len > width)
290     {
291     CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, "\n", 1, line_width);
292     new_line = 1;
293     line_width = 0;
294 sysadm 1.40 i--; // redo at current i
295 sysadm 1.38 }
296     else
297     {
298     CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, str_in + tag_start_pos, tag_output_len, line_width);
299     }
300 sysadm 1.28 }
301    
302 sysadm 1.33 if (fb_quote_level > 0)
303     {
304     lml_tag_quote_level -= fb_quote_level;
305    
306 sysadm 1.36 tag_output_len = snprintf(tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, (quote_mode ? "" : "\033[m"));
307     CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, tag_output_buf, tag_output_len, line_width);
308    
309     fb_quote_level = 0;
310 sysadm 1.33 }
311    
312 sysadm 1.38 if (new_line)
313     {
314     continue;
315     }
316    
317 sysadm 1.28 tag_start_pos = -1;
318     tag_name_pos = -1;
319 sysadm 1.5 new_line = 1;
320 sysadm 1.38 line_width = -1;
321 sysadm 1.5 }
322 sysadm 1.36 else if (str_in[i] == '\r' || str_in[i] == '\7')
323 sysadm 1.5 {
324 sysadm 1.36 continue; // Skip special characters
325 sysadm 1.5 }
326    
327 sysadm 1.20 if (!lml_tag_disabled && str_in[i] == '[')
328 sysadm 1.1 {
329 sysadm 1.33 if (tag_start_pos != -1) // tag is not closed
330     {
331     tag_end_pos = i - 1;
332     tag_output_len = tag_end_pos - tag_start_pos + 1;
333 sysadm 1.36 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, str_in + tag_start_pos, tag_output_len, line_width);
334 sysadm 1.33 }
335    
336 sysadm 1.26 tag_start_pos = i;
337     tag_name_pos = i + 1;
338 sysadm 1.1 }
339 sysadm 1.32 else if (!lml_tag_disabled && str_in[i] == ']' && tag_name_pos >= 0)
340 sysadm 1.1 {
341 sysadm 1.32 tag_end_pos = i;
342    
343     // Skip space characters
344     while (str_in[tag_name_pos] == ' ')
345 sysadm 1.1 {
346 sysadm 1.32 tag_name_pos++;
347     }
348 sysadm 1.1
349 sysadm 1.32 for (tag_name_found = 0, k = 0; k < LML_TAG_COUNT; k++)
350     {
351     if (strncasecmp(lml_tag_def[k].tag_name, str_in + tag_name_pos, (size_t)lml_tag_name_len[k]) == 0)
352 sysadm 1.1 {
353 sysadm 1.32 tag_param_pos = -1;
354     switch (str_in[tag_name_pos + lml_tag_name_len[k]])
355 sysadm 1.1 {
356 sysadm 1.32 case ' ':
357     tag_name_found = 1;
358     tag_param_pos = tag_name_pos + lml_tag_name_len[k] + 1;
359     while (str_in[tag_param_pos] == ' ')
360     {
361     tag_param_pos++;
362     }
363     strncpy(tag_param_buf, str_in + tag_param_pos, (size_t)MIN(tag_end_pos - tag_param_pos, LML_TAG_PARAM_BUF_LEN));
364     tag_param_buf[MIN(tag_end_pos - tag_param_pos, LML_TAG_PARAM_BUF_LEN)] = '\0';
365     case ']':
366     tag_name_found = 1;
367     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
368     {
369     strncpy(tag_param_buf, lml_tag_def[k].default_param, LML_TAG_PARAM_BUF_LEN - 1);
370     tag_param_buf[LML_TAG_PARAM_BUF_LEN - 1] = '\0';
371     }
372 sysadm 1.34 tag_output_len = 0;
373 sysadm 1.32 if (!quote_mode)
374     {
375     if (lml_tag_def[k].tag_output != NULL)
376     {
377     tag_output_len = snprintf(tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, lml_tag_def[k].tag_output, tag_param_buf);
378     }
379     else if (lml_tag_def[k].tag_filter_cb != NULL)
380     {
381     tag_output_len = lml_tag_def[k].tag_filter_cb(
382     lml_tag_def[k].tag_name, tag_param_buf, tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, 0);
383     }
384     }
385     else // if (quote_mode)
386 sysadm 1.1 {
387 sysadm 1.32 if (lml_tag_def[k].quote_mode_output != NULL)
388     {
389     tag_output_len = snprintf(tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, lml_tag_def[k].quote_mode_output, tag_param_buf);
390     }
391     else if (lml_tag_def[k].tag_filter_cb != NULL)
392     {
393     tag_output_len = lml_tag_def[k].tag_filter_cb(
394     lml_tag_def[k].tag_name, tag_param_buf, tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, 1);
395     }
396 sysadm 1.1 }
397 sysadm 1.36 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, tag_output_buf, tag_output_len, line_width);
398 sysadm 1.1 break;
399 sysadm 1.32 default: // tag_name not match
400     continue;
401 sysadm 1.1 }
402 sysadm 1.32 break;
403 sysadm 1.1 }
404 sysadm 1.32 }
405 sysadm 1.1
406 sysadm 1.32 if (!tag_name_found)
407     {
408 sysadm 1.40 if (line_width + 1 > width)
409 sysadm 1.38 {
410     CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, "\n", 1, line_width);
411     new_line = 1;
412     line_width = 0;
413 sysadm 1.40 i--; // redo at current i
414 sysadm 1.38 continue;
415     }
416    
417 sysadm 1.40 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, "[", 1, line_width);
418     i = tag_start_pos; // restart from tag_start_pos + 1
419     tag_start_pos = -1;
420     tag_name_pos = -1;
421     continue;
422 sysadm 1.1 }
423 sysadm 1.32
424     tag_start_pos = -1;
425     tag_name_pos = -1;
426 sysadm 1.1 }
427 sysadm 1.26 else if (lml_tag_disabled || tag_name_pos == -1) // not in LML tag
428 sysadm 1.1 {
429 sysadm 1.36 if (line_width + (str_in[i] & 0x80 ? 2 : 1) > width)
430     {
431     CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, "\n", 1, line_width);
432     new_line = 1;
433     line_width = 0;
434 sysadm 1.40 i--; // redo at current i
435 sysadm 1.36 continue;
436     }
437    
438     tag_output_len = 1;
439 sysadm 1.22 if (str_in[i] & 0x80) // head of multi-byte character
440 sysadm 1.1 {
441 sysadm 1.22 c = (str_in[i] & 0x70) << 1;
442     while (c & 0x80)
443 sysadm 1.1 {
444 sysadm 1.36 if (str_in[i + tag_output_len] == '\0')
445 sysadm 1.13 {
446 sysadm 1.36 break;
447 sysadm 1.13 }
448 sysadm 1.36 tag_output_len++;
449 sysadm 1.22 c = (c & 0x7f) << 1;
450 sysadm 1.1 }
451     }
452    
453 sysadm 1.36 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, str_in + i, tag_output_len, line_width);
454     i += (tag_output_len - 1);
455 sysadm 1.1 }
456     else // in LML tag
457     {
458     // Do nothing
459     }
460     }
461    
462 sysadm 1.26 if (tag_start_pos != -1) // tag is not closed
463     {
464     tag_end_pos = i - 1;
465     tag_output_len = tag_end_pos - tag_start_pos + 1;
466 sysadm 1.36 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, str_in + tag_start_pos, tag_output_len, line_width);
467 sysadm 1.26 }
468    
469 sysadm 1.23 if (!quote_mode && !lml_tag_disabled && lml_tag_quote_level > 0)
470 sysadm 1.5 {
471     tag_output_len = snprintf(tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, "\033[m");
472 sysadm 1.36 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, tag_output_buf, tag_output_len, line_width);
473 sysadm 1.5 }
474    
475 sysadm 1.1 str_out[j] = '\0';
476    
477     return j;
478 sysadm 1.5 }

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