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

Annotation of /lbbs/src/lml.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1.45 - (hide annotations)
Thu Nov 6 13:20:24 2025 UTC (4 months, 1 week ago) by sysadm
Branch: MAIN
Changes since 1.44: +22 -11 lines
Content type: text/x-csrc
Fix bug in handling incomplete LML tag
Use the same implementation of handling unmatched LML tag

1 sysadm 1.42 /* SPDX-License-Identifier: GPL-3.0-or-later */
2     /*
3     * lml
4     * - LML render
5     *
6 sysadm 1.43 * Copyright (C) 2004-2025 Leaflet <leaflet@leafok.com>
7 sysadm 1.42 */
8 sysadm 1.1
9 sysadm 1.12 #include "common.h"
10 sysadm 1.1 #include "lml.h"
11     #include "log.h"
12 sysadm 1.36 #include "str_process.h"
13 sysadm 1.37 #include <ctype.h>
14 sysadm 1.1 #include <stdio.h>
15     #include <string.h>
16     #include <sys/param.h>
17    
18 sysadm 1.44 enum _lml_constant_t
19     {
20     LML_TAG_PARAM_BUF_LEN = 256,
21     LML_TAG_OUTPUT_BUF_LEN = 1024,
22     LML_TAG_QUOTE_MAX_LEVEL = 10,
23     };
24 sysadm 1.1
25 sysadm 1.41 clock_t lml_total_exec_duration = 0;
26    
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 static const char *lml_tag_quote_color[] = {
66 sysadm 1.5 "\033[33m", // yellow
67     "\033[32m", // green
68     "\033[35m", // magenta
69 sysadm 1.4 };
70    
71 sysadm 1.30 static const int LML_TAG_QUOTE_LEVEL_LOOP = (int)(sizeof(lml_tag_quote_color) / sizeof(const char *));
72    
73 sysadm 1.4 static int lml_tag_quote_level = 0;
74    
75 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)
76 sysadm 1.4 {
77     if (strcasecmp(tag_name, "quote") == 0)
78     {
79     if (lml_tag_quote_level <= LML_TAG_QUOTE_MAX_LEVEL)
80     {
81     lml_tag_quote_level++;
82     }
83 sysadm 1.14 return snprintf(tag_output_buf, tag_output_buf_len, "%s",
84     lml_tag_quote_color[lml_tag_quote_level % LML_TAG_QUOTE_LEVEL_LOOP]);
85 sysadm 1.4 }
86     else if (strcasecmp(tag_name, "/quote") == 0)
87     {
88     if (lml_tag_quote_level > 0)
89     {
90     lml_tag_quote_level--;
91     }
92 sysadm 1.14 return snprintf(tag_output_buf, tag_output_buf_len, "%s",
93 sysadm 1.5 (lml_tag_quote_level > 0 ? lml_tag_quote_color[lml_tag_quote_level % LML_TAG_QUOTE_LEVEL_LOOP] : "\033[m"));
94 sysadm 1.4 }
95    
96     return 0;
97     }
98    
99 sysadm 1.20 static int lml_tag_disabled = 0;
100    
101     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)
102     {
103     lml_tag_disabled = 1;
104    
105 sysadm 1.23 return snprintf(tag_output_buf, tag_output_buf_len, "%s", (quote_mode ? "[plain]" : ""));
106 sysadm 1.20 }
107    
108 sysadm 1.22 typedef struct lml_tag_def_t
109     {
110 sysadm 1.26 const char *tag_name; // tag name
111     const char *tag_output; // output string
112     const char *default_param; // default param string
113     const char *quote_mode_output; // output string in quote mode
114 sysadm 1.22 lml_tag_filter_cb tag_filter_cb; // tag filter callback
115     } LML_TAG_DEF;
116    
117     const LML_TAG_DEF lml_tag_def[] = {
118     // Definition of tuple: {lml_tag, lml_output, default_param, quote_mode_output, lml_filter_cb}
119     {"plain", NULL, NULL, NULL, lml_tag_disable_filter},
120 sysadm 1.29 {"nolml", "", NULL, "", NULL}, // deprecated
121 sysadm 1.30 {"lml", "", NULL, "", NULL}, // deprecated
122     {"align", "", "", "", NULL}, // N/A
123     {"/align", "", "", "", NULL}, // N/A
124     {"size", "", "", "", NULL}, // N/A
125     {"/size", "", "", "", NULL}, // N/A
126 sysadm 1.22 {"left", "[", "", "[left]", NULL},
127     {"right", "]", "", "[right]", NULL},
128     {"bold", "\033[1m", "", "", NULL}, // does not work in Fterm
129     {"/bold", "\033[22m", NULL, "", NULL},
130     {"b", "\033[1m", "", "", NULL},
131     {"/b", "\033[22m", NULL, "", NULL},
132 sysadm 1.26 {"italic", "\033[5m", "", "", NULL}, // use blink instead
133 sysadm 1.22 {"/italic", "\033[m", NULL, "", NULL}, // \033[25m does not work in Fterm
134     {"i", "\033[5m", "", "", NULL},
135     {"/i", "\033[m", NULL, "", NULL},
136     {"underline", "\033[4m", "", "", NULL},
137     {"/underline", "\033[m", NULL, "", NULL}, // \033[24m does not work in Fterm
138     {"u", "\033[4m", "", "", NULL},
139     {"/u", "\033[m", NULL, "", NULL},
140     {"color", NULL, NULL, "", lml_tag_color_filter},
141     {"/color", "\033[m", NULL, "", NULL},
142     {"quote", NULL, NULL, "", lml_tag_quote_filter},
143     {"/quote", NULL, NULL, "", lml_tag_quote_filter},
144     {"url", "", "", "", NULL},
145     {"/url", "(链接: %s)", NULL, "", NULL},
146     {"link", "", "", "", NULL},
147     {"/link", "(链接: %s)", NULL, "", NULL},
148     {"email", "", "", "", NULL},
149     {"/email", "(Email: %s)", NULL, "", NULL},
150     {"user", "", "", "", NULL},
151     {"/user", "(用户: %s)", NULL, "", NULL},
152     {"article", "", "", "", NULL},
153     {"/article", "(文章: %s)", NULL, "", NULL},
154     {"image", "(图片: %s)", "", "%s", NULL},
155     {"flash", "(Flash: %s)", "", "", NULL},
156     {"bwf", "\033[1;31m****\033[m", "", "****", NULL},
157 sysadm 1.1 };
158    
159 sysadm 1.44 static const int lml_tag_count = sizeof(lml_tag_def) / sizeof(LML_TAG_DEF);
160     static int lml_tag_name_len[sizeof(lml_tag_def) / sizeof(LML_TAG_DEF)];
161 sysadm 1.19 static int lml_ready = 0;
162 sysadm 1.1
163     inline static void lml_init(void)
164     {
165     int i;
166    
167 sysadm 1.19 if (!lml_ready)
168 sysadm 1.1 {
169 sysadm 1.44 for (i = 0; i < lml_tag_count; i++)
170 sysadm 1.1 {
171 sysadm 1.22 lml_tag_name_len[i] = (int)strlen(lml_tag_def[i].tag_name);
172 sysadm 1.1 }
173    
174 sysadm 1.19 lml_ready = 1;
175 sysadm 1.1 }
176     }
177    
178 sysadm 1.36 #define CHECK_AND_APPEND_OUTPUT(out_buf, out_buf_len, out_buf_offset, tag_out, tag_out_len, line_width) \
179 sysadm 1.35 { \
180     if ((out_buf_offset) + (tag_out_len) >= (out_buf_len)) \
181     { \
182     log_error("Buffer is not longer enough for output string %d >= %d\n", (out_buf_offset) + (tag_out_len), (out_buf_len)); \
183     out_buf[out_buf_offset] = '\0'; \
184     return (out_buf_offset); \
185     } \
186     memcpy((out_buf) + (out_buf_offset), (tag_out), (size_t)(tag_out_len)); \
187 sysadm 1.36 *((out_buf) + (out_buf_offset) + (size_t)(tag_out_len)) = '\0'; \
188     (line_width) += str_length((out_buf) + (out_buf_offset), 1); \
189 sysadm 1.35 (out_buf_offset) += (tag_out_len); \
190     }
191 sysadm 1.34
192 sysadm 1.36 int lml_render(const char *str_in, char *str_out, int buf_len, int width, int quote_mode)
193 sysadm 1.1 {
194 sysadm 1.41 clock_t clock_begin;
195     clock_t clock_end;
196    
197 sysadm 1.13 char c;
198 sysadm 1.2 char tag_param_buf[LML_TAG_PARAM_BUF_LEN];
199 sysadm 1.1 char tag_output_buf[LML_TAG_OUTPUT_BUF_LEN];
200     int i;
201     int j = 0;
202     int k;
203     int tag_start_pos = -1;
204 sysadm 1.26 int tag_name_pos = -1;
205 sysadm 1.1 int tag_end_pos = -1;
206 sysadm 1.2 int tag_param_pos = -1;
207 sysadm 1.1 int tag_output_len;
208 sysadm 1.5 int new_line = 1;
209     int fb_quote_level = 0;
210 sysadm 1.26 int tag_name_found;
211 sysadm 1.36 int line_width = 0;
212 sysadm 1.1
213 sysadm 1.41 clock_begin = clock();
214    
215 sysadm 1.1 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.41 if (!quote_mode && !lml_tag_disabled && 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.41 if (!lml_tag_disabled && tag_start_pos != -1) // tag is not closed
285 sysadm 1.28 {
286 sysadm 1.45 if (line_width + 1 > width)
287 sysadm 1.38 {
288     CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, "\n", 1, line_width);
289     new_line = 1;
290     line_width = 0;
291 sysadm 1.40 i--; // redo at current i
292 sysadm 1.45 continue;
293 sysadm 1.38 }
294 sysadm 1.45
295     CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, "[", 1, line_width);
296     i = tag_start_pos; // restart from tag_start_pos + 1
297     tag_start_pos = -1;
298     tag_name_pos = -1;
299     continue;
300 sysadm 1.28 }
301    
302 sysadm 1.41 if (!lml_tag_disabled && fb_quote_level > 0)
303 sysadm 1.33 {
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 sysadm 1.45 if (line_width + 1 > width)
332     {
333     CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, "\n", 1, line_width);
334     new_line = 1;
335     line_width = 0;
336     i--; // redo at current i
337     continue;
338     }
339    
340     CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, "[", 1, line_width);
341     i = tag_start_pos; // restart from tag_start_pos + 1
342     tag_start_pos = -1;
343     tag_name_pos = -1;
344     continue;
345 sysadm 1.33 }
346    
347 sysadm 1.26 tag_start_pos = i;
348     tag_name_pos = i + 1;
349 sysadm 1.1 }
350 sysadm 1.32 else if (!lml_tag_disabled && str_in[i] == ']' && tag_name_pos >= 0)
351 sysadm 1.1 {
352 sysadm 1.32 tag_end_pos = i;
353    
354     // Skip space characters
355     while (str_in[tag_name_pos] == ' ')
356 sysadm 1.1 {
357 sysadm 1.32 tag_name_pos++;
358     }
359 sysadm 1.1
360 sysadm 1.44 for (tag_name_found = 0, k = 0; k < lml_tag_count; k++)
361 sysadm 1.32 {
362     if (strncasecmp(lml_tag_def[k].tag_name, str_in + tag_name_pos, (size_t)lml_tag_name_len[k]) == 0)
363 sysadm 1.1 {
364 sysadm 1.32 tag_param_pos = -1;
365     switch (str_in[tag_name_pos + lml_tag_name_len[k]])
366 sysadm 1.1 {
367 sysadm 1.32 case ' ':
368     tag_name_found = 1;
369     tag_param_pos = tag_name_pos + lml_tag_name_len[k] + 1;
370     while (str_in[tag_param_pos] == ' ')
371     {
372     tag_param_pos++;
373     }
374     strncpy(tag_param_buf, str_in + tag_param_pos, (size_t)MIN(tag_end_pos - tag_param_pos, LML_TAG_PARAM_BUF_LEN));
375     tag_param_buf[MIN(tag_end_pos - tag_param_pos, LML_TAG_PARAM_BUF_LEN)] = '\0';
376     case ']':
377     tag_name_found = 1;
378     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
379     {
380     strncpy(tag_param_buf, lml_tag_def[k].default_param, LML_TAG_PARAM_BUF_LEN - 1);
381     tag_param_buf[LML_TAG_PARAM_BUF_LEN - 1] = '\0';
382     }
383 sysadm 1.34 tag_output_len = 0;
384 sysadm 1.32 if (!quote_mode)
385     {
386     if (lml_tag_def[k].tag_output != NULL)
387     {
388     tag_output_len = snprintf(tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, lml_tag_def[k].tag_output, tag_param_buf);
389     }
390     else if (lml_tag_def[k].tag_filter_cb != NULL)
391     {
392     tag_output_len = lml_tag_def[k].tag_filter_cb(
393     lml_tag_def[k].tag_name, tag_param_buf, tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, 0);
394     }
395     }
396     else // if (quote_mode)
397 sysadm 1.1 {
398 sysadm 1.32 if (lml_tag_def[k].quote_mode_output != NULL)
399     {
400     tag_output_len = snprintf(tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, lml_tag_def[k].quote_mode_output, tag_param_buf);
401     }
402     else if (lml_tag_def[k].tag_filter_cb != NULL)
403     {
404     tag_output_len = lml_tag_def[k].tag_filter_cb(
405     lml_tag_def[k].tag_name, tag_param_buf, tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, 1);
406     }
407 sysadm 1.1 }
408 sysadm 1.36 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, tag_output_buf, tag_output_len, line_width);
409 sysadm 1.1 break;
410 sysadm 1.32 default: // tag_name not match
411     continue;
412 sysadm 1.1 }
413 sysadm 1.32 break;
414 sysadm 1.1 }
415 sysadm 1.32 }
416 sysadm 1.1
417 sysadm 1.32 if (!tag_name_found)
418     {
419 sysadm 1.40 if (line_width + 1 > width)
420 sysadm 1.38 {
421     CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, "\n", 1, line_width);
422     new_line = 1;
423     line_width = 0;
424 sysadm 1.40 i--; // redo at current i
425 sysadm 1.38 continue;
426     }
427    
428 sysadm 1.40 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, "[", 1, line_width);
429     i = tag_start_pos; // restart from tag_start_pos + 1
430     tag_start_pos = -1;
431     tag_name_pos = -1;
432     continue;
433 sysadm 1.1 }
434 sysadm 1.32
435     tag_start_pos = -1;
436     tag_name_pos = -1;
437 sysadm 1.1 }
438 sysadm 1.26 else if (lml_tag_disabled || tag_name_pos == -1) // not in LML tag
439 sysadm 1.1 {
440 sysadm 1.36 if (line_width + (str_in[i] & 0x80 ? 2 : 1) > width)
441     {
442     CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, "\n", 1, line_width);
443     new_line = 1;
444     line_width = 0;
445 sysadm 1.40 i--; // redo at current i
446 sysadm 1.36 continue;
447     }
448    
449     tag_output_len = 1;
450 sysadm 1.22 if (str_in[i] & 0x80) // head of multi-byte character
451 sysadm 1.1 {
452 sysadm 1.22 c = (str_in[i] & 0x70) << 1;
453     while (c & 0x80)
454 sysadm 1.1 {
455 sysadm 1.36 if (str_in[i + tag_output_len] == '\0')
456 sysadm 1.13 {
457 sysadm 1.36 break;
458 sysadm 1.13 }
459 sysadm 1.36 tag_output_len++;
460 sysadm 1.22 c = (c & 0x7f) << 1;
461 sysadm 1.1 }
462     }
463    
464 sysadm 1.36 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, str_in + i, tag_output_len, line_width);
465     i += (tag_output_len - 1);
466 sysadm 1.1 }
467     else // in LML tag
468     {
469     // Do nothing
470     }
471     }
472    
473 sysadm 1.41 if (!lml_tag_disabled && tag_start_pos != -1) // tag is not closed
474 sysadm 1.26 {
475     tag_end_pos = i - 1;
476     tag_output_len = tag_end_pos - tag_start_pos + 1;
477 sysadm 1.36 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, str_in + tag_start_pos, tag_output_len, line_width);
478 sysadm 1.26 }
479    
480 sysadm 1.23 if (!quote_mode && !lml_tag_disabled && lml_tag_quote_level > 0)
481 sysadm 1.5 {
482     tag_output_len = snprintf(tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, "\033[m");
483 sysadm 1.36 CHECK_AND_APPEND_OUTPUT(str_out, buf_len, j, tag_output_buf, tag_output_len, line_width);
484 sysadm 1.5 }
485    
486 sysadm 1.1 str_out[j] = '\0';
487    
488 sysadm 1.41 clock_end = clock();
489     lml_total_exec_duration += (clock_end - clock_begin);
490    
491 sysadm 1.1 return j;
492 sysadm 1.5 }

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