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

Annotation of /lbbs/src/lml.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1.35 - (hide 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 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     #include <stdio.h>
21     #include <string.h>
22     #include <sys/param.h>
23    
24 sysadm 1.2 #define LML_TAG_PARAM_BUF_LEN 256
25     #define LML_TAG_OUTPUT_BUF_LEN 1024
26 sysadm 1.1
27 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);
28 sysadm 1.3
29 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)
30 sysadm 1.3 {
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 sysadm 1.1
65 sysadm 1.4 #define LML_TAG_QUOTE_MAX_LEVEL 10
66    
67     static const char *lml_tag_quote_color[] = {
68 sysadm 1.5 "\033[33m", // yellow
69     "\033[32m", // green
70     "\033[35m", // magenta
71 sysadm 1.4 };
72    
73 sysadm 1.30 static const int LML_TAG_QUOTE_LEVEL_LOOP = (int)(sizeof(lml_tag_quote_color) / sizeof(const char *));
74    
75 sysadm 1.4 static int lml_tag_quote_level = 0;
76    
77 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)
78 sysadm 1.4 {
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 sysadm 1.14 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 sysadm 1.4 }
88     else if (strcasecmp(tag_name, "/quote") == 0)
89     {
90     if (lml_tag_quote_level > 0)
91     {
92     lml_tag_quote_level--;
93     }
94 sysadm 1.14 return snprintf(tag_output_buf, tag_output_buf_len, "%s",
95 sysadm 1.5 (lml_tag_quote_level > 0 ? lml_tag_quote_color[lml_tag_quote_level % LML_TAG_QUOTE_LEVEL_LOOP] : "\033[m"));
96 sysadm 1.4 }
97    
98     return 0;
99     }
100    
101 sysadm 1.20 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 sysadm 1.23 return snprintf(tag_output_buf, tag_output_buf_len, "%s", (quote_mode ? "[plain]" : ""));
108 sysadm 1.20 }
109    
110 sysadm 1.22 typedef struct lml_tag_def_t
111     {
112 sysadm 1.26 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 sysadm 1.22 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 sysadm 1.29 {"nolml", "", NULL, "", NULL}, // deprecated
123 sysadm 1.30 {"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 sysadm 1.22 {"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 sysadm 1.26 {"italic", "\033[5m", "", "", NULL}, // use blink instead
135 sysadm 1.22 {"/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 sysadm 1.1 };
160    
161 sysadm 1.24 #define LML_TAG_COUNT (sizeof(lml_tag_def) / sizeof(LML_TAG_DEF))
162 sysadm 1.3
163 sysadm 1.19 static int lml_tag_name_len[LML_TAG_COUNT];
164     static int lml_ready = 0;
165 sysadm 1.1
166     inline static void lml_init(void)
167     {
168     int i;
169    
170 sysadm 1.19 if (!lml_ready)
171 sysadm 1.1 {
172     for (i = 0; i < LML_TAG_COUNT; i++)
173     {
174 sysadm 1.22 lml_tag_name_len[i] = (int)strlen(lml_tag_def[i].tag_name);
175 sysadm 1.1 }
176    
177 sysadm 1.19 lml_ready = 1;
178 sysadm 1.1 }
179     }
180    
181 sysadm 1.35 #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 sysadm 1.34
193 sysadm 1.20 int lml_render(const char *str_in, char *str_out, int buf_len, int quote_mode)
194 sysadm 1.1 {
195 sysadm 1.13 char c;
196 sysadm 1.2 char tag_param_buf[LML_TAG_PARAM_BUF_LEN];
197 sysadm 1.1 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 sysadm 1.26 int tag_name_pos = -1;
203 sysadm 1.1 int tag_end_pos = -1;
204 sysadm 1.2 int tag_param_pos = -1;
205 sysadm 1.1 int tag_output_len;
206 sysadm 1.5 int new_line = 1;
207     int fb_quote_level = 0;
208 sysadm 1.26 int tag_name_found;
209 sysadm 1.1
210     lml_init();
211    
212 sysadm 1.20 lml_tag_disabled = 0;
213 sysadm 1.4 lml_tag_quote_level = 0;
214    
215 sysadm 1.1 for (i = 0; str_in[i] != '\0'; i++)
216     {
217 sysadm 1.23 if (!quote_mode && !lml_tag_disabled && new_line)
218 sysadm 1.5 {
219 sysadm 1.33 fb_quote_level = 0;
220 sysadm 1.5
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 sysadm 1.33 lml_tag_quote_level += fb_quote_level;
227    
228     if (lml_tag_quote_level > 0)
229 sysadm 1.5 {
230 sysadm 1.14 tag_output_len = snprintf(tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, "%s",
231 sysadm 1.31 lml_tag_quote_color[lml_tag_quote_level % LML_TAG_QUOTE_LEVEL_LOOP]);
232 sysadm 1.34 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, tag_output_buf, tag_output_len);
233 sysadm 1.5 }
234    
235     new_line = 0;
236     }
237    
238 sysadm 1.8 if (str_in[i] == '\033' && str_in[i + 1] == '[') // Escape sequence -- copy directly
239     {
240 sysadm 1.35 for (k = i + 2; str_in[k] != '\0' && str_in[k] != 'm' && str_in[k] != '\033'; k++)
241 sysadm 1.8 ;
242    
243 sysadm 1.35 if (str_in[k] != 'm') // invalid
244 sysadm 1.8 {
245 sysadm 1.35 k--;
246 sysadm 1.8 }
247 sysadm 1.35
248     CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, str_in + i, k - i + 1);
249     i = k;
250     continue;
251 sysadm 1.8 }
252    
253 sysadm 1.28 if (str_in[i] == '\n') // jump out of tag at end of line
254 sysadm 1.5 {
255 sysadm 1.28 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 sysadm 1.34 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, str_in + tag_start_pos, tag_output_len);
260 sysadm 1.28 }
261    
262 sysadm 1.33 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 sysadm 1.34 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, tag_output_buf, tag_output_len);
268 sysadm 1.33 }
269    
270 sysadm 1.28 tag_start_pos = -1;
271     tag_name_pos = -1;
272 sysadm 1.5 new_line = 1;
273     }
274     else if (str_in[i] == '\r')
275     {
276     continue; // ignore '\r'
277     }
278    
279 sysadm 1.20 if (!lml_tag_disabled && str_in[i] == '[')
280 sysadm 1.1 {
281 sysadm 1.33 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 sysadm 1.34 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, str_in + tag_start_pos, tag_output_len);
286 sysadm 1.33 }
287    
288 sysadm 1.26 tag_start_pos = i;
289     tag_name_pos = i + 1;
290 sysadm 1.1 }
291 sysadm 1.32 else if (!lml_tag_disabled && str_in[i] == ']' && tag_name_pos >= 0)
292 sysadm 1.1 {
293 sysadm 1.32 tag_end_pos = i;
294    
295     // Skip space characters
296     while (str_in[tag_name_pos] == ' ')
297 sysadm 1.1 {
298 sysadm 1.32 tag_name_pos++;
299     }
300 sysadm 1.1
301 sysadm 1.32 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 sysadm 1.1 {
305 sysadm 1.32 tag_param_pos = -1;
306     switch (str_in[tag_name_pos + lml_tag_name_len[k]])
307 sysadm 1.1 {
308 sysadm 1.32 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 sysadm 1.34 tag_output_len = 0;
325 sysadm 1.32 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 sysadm 1.1 {
339 sysadm 1.32 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 sysadm 1.1 }
349 sysadm 1.34 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, tag_output_buf, tag_output_len);
350 sysadm 1.1 break;
351 sysadm 1.32 default: // tag_name not match
352     continue;
353 sysadm 1.1 }
354 sysadm 1.32 break;
355 sysadm 1.1 }
356 sysadm 1.32 }
357 sysadm 1.1
358 sysadm 1.32 if (!tag_name_found)
359     {
360     tag_output_len = tag_end_pos - tag_start_pos + 1;
361 sysadm 1.34 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, str_in + tag_start_pos, tag_output_len);
362 sysadm 1.1 }
363 sysadm 1.32
364     tag_start_pos = -1;
365     tag_name_pos = -1;
366 sysadm 1.1 }
367 sysadm 1.26 else if (lml_tag_disabled || tag_name_pos == -1) // not in LML tag
368 sysadm 1.1 {
369 sysadm 1.22 if (str_in[i] & 0x80) // head of multi-byte character
370 sysadm 1.1 {
371 sysadm 1.13 if (j + 4 >= buf_len) // Assuming UTF-8 CJK characters use 4 bytes, though most of them actually use 3 bytes
372 sysadm 1.1 {
373 sysadm 1.13 log_error("Buffer is not longer enough for output string %ld >= %d\n", j + 4, buf_len);
374 sysadm 1.1 str_out[j] = '\0';
375     return j;
376     }
377 sysadm 1.13
378 sysadm 1.22 c = (str_in[i] & 0x70) << 1;
379     while (c & 0x80)
380 sysadm 1.1 {
381 sysadm 1.13 str_out[j++] = str_in[i++];
382     if (str_in[i] == '\0')
383     {
384     str_out[j] = '\0';
385     return j;
386     }
387 sysadm 1.22 c = (c & 0x7f) << 1;
388 sysadm 1.1 }
389     }
390    
391 sysadm 1.8 if (j + 1 >= buf_len)
392 sysadm 1.1 {
393 sysadm 1.8 log_error("Buffer is not longer enough for output string %ld >= %d\n", j + 1, buf_len);
394 sysadm 1.1 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 sysadm 1.26 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 sysadm 1.34 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, str_in + tag_start_pos, tag_output_len);
410 sysadm 1.26 }
411    
412 sysadm 1.23 if (!quote_mode && !lml_tag_disabled && lml_tag_quote_level > 0)
413 sysadm 1.5 {
414     tag_output_len = snprintf(tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, "\033[m");
415 sysadm 1.34 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, tag_output_buf, tag_output_len);
416 sysadm 1.5 }
417    
418 sysadm 1.1 str_out[j] = '\0';
419    
420     return j;
421 sysadm 1.5 }

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