lbbs/src/lml.c

/***************************************************************************
                                                        lml.h  -  description
                                                         -------------------
        Copyright            : (C) 2004-2025 by Leaflet
        Email                : leaflet@leafok.com
 ***************************************************************************/

/***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 3 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/

#include "common.h"
#include "lml.h"
#include "log.h"
#include <stdio.h>
#include <string.h>
#include <sys/param.h>

#define LML_TAG_PARAM_BUF_LEN 256
#define LML_TAG_OUTPUT_BUF_LEN 1024

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);

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)
{
        if (strcasecmp(tag_name, "color") == 0)
        {
                if (strcasecmp(tag_param_buf, "red") == 0)
                {
                        return snprintf(tag_output_buf, tag_output_buf_len, "\033[1;31m");
                }
                else if (strcasecmp(tag_param_buf, "green") == 0)
                {
                        return snprintf(tag_output_buf, tag_output_buf_len, "\033[1;32m");
                }
                else if (strcasecmp(tag_param_buf, "yellow") == 0)
                {
                        return snprintf(tag_output_buf, tag_output_buf_len, "\033[1;33m");
                }
                else if (strcasecmp(tag_param_buf, "blue") == 0)
                {
                        return snprintf(tag_output_buf, tag_output_buf_len, "\033[1;34m");
                }
                else if (strcasecmp(tag_param_buf, "magenta") == 0)
                {
                        return snprintf(tag_output_buf, tag_output_buf_len, "\033[1;35m");
                }
                else if (strcasecmp(tag_param_buf, "cyan") == 0)
                {
                        return snprintf(tag_output_buf, tag_output_buf_len, "\033[1;36m");
                }
                else if (strcasecmp(tag_param_buf, "black") == 0)
                {
                        return snprintf(tag_output_buf, tag_output_buf_len, "\033[1;30;47m");
                }
        }
        return 0;
}

#define LML_TAG_QUOTE_MAX_LEVEL 10
#define LML_TAG_QUOTE_LEVEL_LOOP 3

static const char *lml_tag_quote_color[] = {
        "\033[33m", // yellow
        "\033[32m", // green
        "\033[35m", // magenta
};

static int lml_tag_quote_level = 0;

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)
{
        if (strcasecmp(tag_name, "quote") == 0)
        {
                if (lml_tag_quote_level <= LML_TAG_QUOTE_MAX_LEVEL)
                {
                        lml_tag_quote_level++;
                }
                return snprintf(tag_output_buf, tag_output_buf_len, "%s",
                                                lml_tag_quote_color[lml_tag_quote_level % LML_TAG_QUOTE_LEVEL_LOOP]);
        }
        else if (strcasecmp(tag_name, "/quote") == 0)
        {
                if (lml_tag_quote_level > 0)
                {
                        lml_tag_quote_level--;
                }
                return snprintf(tag_output_buf, tag_output_buf_len, "%s",
                                                (lml_tag_quote_level > 0 ? lml_tag_quote_color[lml_tag_quote_level % LML_TAG_QUOTE_LEVEL_LOOP] : "\033[m"));
        }

        return 0;
}

static int lml_tag_disabled = 0;

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)
{
        lml_tag_disabled = 1;

        return snprintf(tag_output_buf, tag_output_buf_len, "%s", (quote_mode ? "[plain]" : ""));
}

typedef struct lml_tag_def_t
{
        const char *tag_name;                    // tag name
        const char *tag_output;                  // output string
        const char *default_param;               // default param string
        const char *quote_mode_output;   // output string in quote mode
        lml_tag_filter_cb tag_filter_cb; // tag filter callback
} LML_TAG_DEF;

const LML_TAG_DEF lml_tag_def[] = {
        // Definition of tuple: {lml_tag, lml_output, default_param, quote_mode_output, lml_filter_cb}
        {"plain", NULL, NULL, NULL, lml_tag_disable_filter},
        {"nolml", "", NULL, "", NULL},
        {"lml", "", NULL, "", NULL},
        {"left", "[", "", "[left]", NULL},
        {"right", "]", "", "[right]", NULL},
        {"bold", "\033[1m", "", "", NULL}, // does not work in Fterm
        {"/bold", "\033[22m", NULL, "", NULL},
        {"b", "\033[1m", "", "", NULL},
        {"/b", "\033[22m", NULL, "", NULL},
        {"italic", "\033[5m", "", "", NULL},   // use blink instead
        {"/italic", "\033[m", NULL, "", NULL}, // \033[25m does not work in Fterm
        {"i", "\033[5m", "", "", NULL},
        {"/i", "\033[m", NULL, "", NULL},
        {"underline", "\033[4m", "", "", NULL},
        {"/underline", "\033[m", NULL, "", NULL}, // \033[24m does not work in Fterm
        {"u", "\033[4m", "", "", NULL},
        {"/u", "\033[m", NULL, "", NULL},
        {"color", NULL, NULL, "", lml_tag_color_filter},
        {"/color", "\033[m", NULL, "", NULL},
        {"quote", NULL, NULL, "", lml_tag_quote_filter},
        {"/quote", NULL, NULL, "", lml_tag_quote_filter},
        {"url", "", "", "", NULL},
        {"/url", "（链接: %s）", NULL, "", NULL},
        {"link", "", "", "", NULL},
        {"/link", "（链接: %s）", NULL, "", NULL},
        {"email", "", "", "", NULL},
        {"/email", "（Email: %s）", NULL, "", NULL},
        {"user", "", "", "", NULL},
        {"/user", "（用户: %s）", NULL, "", NULL},
        {"article", "", "", "", NULL},
        {"/article", "（文章: %s）", NULL, "", NULL},
        {"image", "（图片: %s）", "", "%s", NULL},
        {"flash", "（Flash: %s）", "", "", NULL},
        {"bwf", "\033[1;31m****\033[m", "", "****", NULL},
};

#define LML_TAG_COUNT (sizeof(lml_tag_def) / sizeof(LML_TAG_DEF))

static int lml_tag_name_len[LML_TAG_COUNT];
static int lml_ready = 0;

inline static void lml_init(void)
{
        int i;

        if (!lml_ready)
        {
                for (i = 0; i < LML_TAG_COUNT; i++)
                {
                        lml_tag_name_len[i] = (int)strlen(lml_tag_def[i].tag_name);
                }

                lml_ready = 1;
        }
}

int lml_render(const char *str_in, char *str_out, int buf_len, int quote_mode)
{
        char c;
        char tag_param_buf[LML_TAG_PARAM_BUF_LEN];
        char tag_output_buf[LML_TAG_OUTPUT_BUF_LEN];
        int i;
        int j = 0;
        int k;
        int tag_start_pos = -1;
        int tag_name_pos = -1;
        int tag_end_pos = -1;
        int tag_param_pos = -1;
        int tag_output_len;
        int new_line = 1;
        int fb_quote_level = 0;
        int tag_name_found;

        lml_init();

        lml_tag_disabled = 0;
        lml_tag_quote_level = 0;

        for (i = 0; str_in[i] != '\0'; i++)
        {
                if (!quote_mode && !lml_tag_disabled && new_line)
                {
                        if (fb_quote_level > 0)
                        {
                                lml_tag_quote_level -= fb_quote_level;

                                tag_output_len = snprintf(tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, "%s",
                                                                                  (lml_tag_quote_level > 0 ? lml_tag_quote_color[lml_tag_quote_level % LML_TAG_QUOTE_LEVEL_LOOP] : "\033[m"));
                                if (j + tag_output_len >= buf_len)
                                {
                                        log_error("Buffer is not longer enough for output string %d >= %d\n", j + tag_output_len, buf_len);
                                        str_out[j] = '\0';
                                        return j;
                                }
                                memcpy(str_out + j, tag_output_buf, (size_t)tag_output_len);
                                j += tag_output_len;

                                fb_quote_level = 0;
                        }

                        while (str_in[i + fb_quote_level * 2] == ':' && str_in[i + fb_quote_level * 2 + 1] == ' ') // FB2000 quote leading str
                        {
                                fb_quote_level++;
                        }

                        if (fb_quote_level > 0)
                        {
                                lml_tag_quote_level += fb_quote_level;

                                tag_output_len = snprintf(tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, "%s",
                                                                                  lml_tag_quote_color[(lml_tag_quote_level) % LML_TAG_QUOTE_LEVEL_LOOP]);
                                if (j + tag_output_len >= buf_len)
                                {
                                        log_error("Buffer is not longer enough for output string %d >= %d\n", j + tag_output_len, buf_len);
                                        str_out[j] = '\0';
                                        return j;
                                }
                                memcpy(str_out + j, tag_output_buf, (size_t)tag_output_len);
                                j += tag_output_len;
                        }

                        new_line = 0;
                }

                if (str_in[i] == '\033' && str_in[i + 1] == '[') // Escape sequence -- copy directly
                {
                        for (k = i + 2; str_in[k] != '\0' && str_in[k] != 'm'; k++)
                                ;

                        if (str_in[k] == 'm') // Valid
                        {
                                if (j + (k - i + 1) >= buf_len)
                                {
                                        log_error("Buffer is not longer enough for output string %d >= %d\n", j + (k - i + 1), buf_len);
                                        str_out[j] = '\0';
                                        return j;
                                }
                                memcpy(str_out + j, str_in + i, (size_t)(k - i + 1));
                                j += (k - i + 1);
                                i = k;
                                continue;
                        }
                        else // reach end of string
                        {
                                break;
                        }
                }

                if (str_in[i] == '\n')
                {
                        tag_name_pos = -1; // jump out of tag at end of line
                        new_line = 1;
                }
                else if (str_in[i] == '\r')
                {
                        continue; // ignore '\r'
                }

                if (!lml_tag_disabled && str_in[i] == '[')
                {
                        tag_start_pos = i;
                        tag_name_pos = i + 1;
                }
                else if (!lml_tag_disabled && str_in[i] == ']')
                {
                        if (tag_name_pos >= 0)
                        {
                                tag_end_pos = i;

                                // Skip space characters
                                while (str_in[tag_name_pos] == ' ')
                                {
                                        tag_name_pos++;
                                }

                                for (tag_name_found = 0, k = 0; k < LML_TAG_COUNT; k++)
                                {
                                        if (strncasecmp(lml_tag_def[k].tag_name, str_in + tag_name_pos, (size_t)lml_tag_name_len[k]) == 0)
                                        {
                                                tag_param_pos = -1;
                                                switch (str_in[tag_name_pos + lml_tag_name_len[k]])
                                                {
                                                case ' ':
                                                        tag_name_found = 1;
                                                        tag_param_pos = tag_name_pos + lml_tag_name_len[k] + 1;
                                                        while (str_in[tag_param_pos] == ' ')
                                                        {
                                                                tag_param_pos++;
                                                        }
                                                        strncpy(tag_param_buf, str_in + tag_param_pos, (size_t)MIN(tag_end_pos - tag_param_pos, LML_TAG_PARAM_BUF_LEN));
                                                        tag_param_buf[MIN(tag_end_pos - tag_param_pos, LML_TAG_PARAM_BUF_LEN)] = '\0';
                                                case ']':
                                                        tag_name_found = 1;
                                                        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
                                                        {
                                                                strncpy(tag_param_buf, lml_tag_def[k].default_param, LML_TAG_PARAM_BUF_LEN - 1);
                                                                tag_param_buf[LML_TAG_PARAM_BUF_LEN - 1] = '\0';
                                                        }
                                                        if (!quote_mode)
                                                        {
                                                                if (lml_tag_def[k].tag_output != NULL)
                                                                {
                                                                        tag_output_len = snprintf(tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, lml_tag_def[k].tag_output, tag_param_buf);
                                                                }
                                                                else if (lml_tag_def[k].tag_filter_cb != NULL)
                                                                {
                                                                        tag_output_len = lml_tag_def[k].tag_filter_cb(
                                                                                lml_tag_def[k].tag_name, tag_param_buf, tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, 0);
                                                                }
                                                                else
                                                                {
                                                                        tag_output_len = 0;
                                                                }
                                                        }
                                                        else // if (quote_mode)
                                                        {
                                                                if (lml_tag_def[k].quote_mode_output != NULL)
                                                                {
                                                                        tag_output_len = snprintf(tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, lml_tag_def[k].quote_mode_output, tag_param_buf);
                                                                }
                                                                else if (lml_tag_def[k].tag_filter_cb != NULL)
                                                                {
                                                                        tag_output_len = lml_tag_def[k].tag_filter_cb(
                                                                                lml_tag_def[k].tag_name, tag_param_buf, tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, 1);
                                                                }
                                                                else
                                                                {
                                                                        tag_output_len = 0;
                                                                }
                                                        }
                                                        if (j + tag_output_len >= buf_len)
                                                        {
                                                                log_error("Buffer is not longer enough for output string %d >= %d\n", j + tag_output_len, buf_len);
                                                                str_out[j] = '\0';
                                                                return j;
                                                        }
                                                        memcpy(str_out + j, tag_output_buf, (size_t)tag_output_len);
                                                        j += tag_output_len;
                                                        break;
                                                default: // tag_name not match
                                                        continue;
                                                }
                                                break;
                                        }
                                }

                                if (!tag_name_found)
                                {
                                        tag_output_len = tag_end_pos - tag_start_pos + 1;
                                        if (j + tag_output_len >= buf_len)
                                        {
                                                log_error("Buffer is not longer enough for output string %ld >= %d\n", j + tag_output_len, buf_len);
                                                str_out[j] = '\0';
                                                return j;
                                        }

                                        memcpy(str_out + j, str_in + tag_start_pos, (size_t)tag_output_len);
                                        j += tag_output_len;
                                }

                                tag_name_pos = -1;
                        }
                }
                else if (lml_tag_disabled || tag_name_pos == -1) // not in LML tag
                {
                        if (str_in[i] & 0x80) // head of multi-byte character
                        {
                                if (j + 4 >= buf_len) // Assuming UTF-8 CJK characters use 4 bytes, though most of them actually use 3 bytes
                                {
                                        log_error("Buffer is not longer enough for output string %ld >= %d\n", j + 4, buf_len);
                                        str_out[j] = '\0';
                                        return j;
                                }

                                c = (str_in[i] & 0x70) << 1;
                                while (c & 0x80)
                                {
                                        str_out[j++] = str_in[i++];
                                        if (str_in[i] == '\0')
                                        {
                                                str_out[j] = '\0';
                                                return j;
                                        }
                                        c = (c & 0x7f) << 1;
                                }
                        }

                        if (j + 1 >= buf_len)
                        {
                                log_error("Buffer is not longer enough for output string %ld >= %d\n", j + 1, buf_len);
                                str_out[j] = '\0';
                                return j;
                        }
                        str_out[j++] = str_in[i];
                }
                else // in LML tag
                {
                        // Do nothing
                }
        }

        if (tag_start_pos != -1) // tag is not closed
        {
                tag_end_pos = i - 1;
                tag_output_len = tag_end_pos - tag_start_pos + 1;
                if (j + tag_output_len >= buf_len)
                {
                        log_error("Buffer is not longer enough for output string %ld >= %d\n", j + tag_output_len, buf_len);
                        str_out[j] = '\0';
                        return j;
                }

                memcpy(str_out + j, str_in + tag_start_pos, (size_t)tag_output_len);
                j += tag_output_len;
        }

        if (!quote_mode && !lml_tag_disabled && lml_tag_quote_level > 0)
        {
                tag_output_len = snprintf(tag_output_buf, LML_TAG_OUTPUT_BUF_LEN, "\033[m");
                if (j + tag_output_len >= buf_len)
                {
                        log_error("Buffer is not longer enough for output string %d >= %d\n", j + tag_output_len, buf_len);
                        str_out[j] = '\0';
                        return j;
                }
                memcpy(str_out + j, tag_output_buf, (size_t)tag_output_len);
                j += tag_output_len;
        }

        str_out[j] = '\0';

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