]> git.wincent.com - wikitext.git/blob - ext/parser.c
Move some functions around to avoid the forward declarations
[wikitext.git] / ext / parser.c
1 // Copyright 2007-2012 Wincent Colaiuta. All rights reserved.
2 //
3 // Redistribution and use in source and binary forms, with or without
4 // modification, are permitted provided that the following conditions are met:
5 //
6 // 1. Redistributions of source code must retain the above copyright notice,
7 //    this list of conditions and the following disclaimer.
8 // 2. Redistributions in binary form must reproduce the above copyright notice,
9 //    this list of conditions and the following disclaimer in the documentation
10 //    and/or other materials provided with the distribution.
11 //
12 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
13 // AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
14 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
15 // ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR CONTRIBUTORS BE
16 // LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
17 // CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
18 // SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
19 // INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
20 // CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
21 // ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
22 // POSSIBILITY OF SUCH DAMAGE.
23
24 #include <stdbool.h>
25
26 #include "parser.h"
27 #include "ary.h"
28 #include "str.h"
29 #include "wikitext.h"
30 #include "wikitext_ragel.h"
31
32 #define IN(type) ary_includes(parser->scope, type)
33 #define IN_EITHER_OF(type1, type2) ary_includes2(parser->scope, type1, type2)
34 #define IN_ANY_OF(type1, type2, type3) ary_includes3(parser->scope, type1, type2, type3)
35
36 // output styles
37 enum { HTML_OUTPUT, XML_OUTPUT };
38
39 // poor man's object orientation in C:
40 // instead of passing around multiple parameters between functions in the parser
41 // we pack everything into a struct and pass around only a pointer to that
42 typedef struct
43 {
44     str_t   *capture;               // capturing to link_target, link_text, or NULL (direct to output, not capturing)
45     str_t   *output;                // for accumulating output to be returned
46     str_t   *link_target;           // short term "memory" for parsing links
47     str_t   *link_text;             // short term "memory" for parsing links
48     str_t   *line_ending;
49     str_t   *tabulation;            // caching buffer for emitting indentation
50     ary_t   *scope;                 // stack for tracking scope
51     ary_t   *line;                  // stack for tracking scope as implied by current line
52     ary_t   *line_buffer;           // stack for tracking raw tokens (not scope) on current line
53     VALUE   external_link_class;    // CSS class applied to external links
54     VALUE   external_link_rel;      // rel attribute applied to external links
55     VALUE   mailto_class;           // CSS class applied to email (mailto) links
56     VALUE   img_prefix;             // path prepended when emitting img tags
57     int     output_style;           // HTML_OUTPUT (default) or XML_OUTPUT
58     int     base_indent;            // controlled by the :indent option to Wikitext::Parser#parse
59     int     current_indent;         // fluctuates according to currently nested structures
60     int     base_heading_level;
61     bool    pending_crlf;
62     bool    autolink;
63     bool    space_to_underscore;
64 } parser_t;
65
66 const char null_str[]                   = { 0 };
67 const char escaped_no_wiki_start[]      = "&lt;nowiki&gt;";
68 const char escaped_no_wiki_end[]        = "&lt;/nowiki&gt;";
69 const char literal_strong_em[]          = "'''''";
70 const char literal_strong[]             = "'''";
71 const char literal_em[]                 = "''";
72 const char escaped_em_start[]           = "&lt;em&gt;";
73 const char escaped_em_end[]             = "&lt;/em&gt;";
74 const char escaped_strong_start[]       = "&lt;strong&gt;";
75 const char escaped_strong_end[]         = "&lt;/strong&gt;";
76 const char escaped_tt_start[]           = "&lt;tt&gt;";
77 const char escaped_tt_end[]             = "&lt;/tt&gt;";
78 const char pre_start[]                  = "<pre>";
79 const char pre_end[]                    = "</pre>";
80 const char escaped_pre_start[]          = "&lt;pre&gt;";
81 const char escaped_pre_end[]            = "&lt;/pre&gt;";
82 const char blockquote_start[]           = "<blockquote>";
83 const char blockquote_end[]             = "</blockquote>";
84 const char escaped_blockquote_start[]   = "&lt;blockquote&gt;";
85 const char escaped_blockquote_end[]     = "&lt;/blockquote&gt;";
86 const char strong_em_start[]            = "<strong><em>";
87 const char strong_start[]               = "<strong>";
88 const char strong_end[]                 = "</strong>";
89 const char em_start[]                   = "<em>";
90 const char em_end[]                     = "</em>";
91 const char code_start[]                 = "<code>";
92 const char code_end[]                   = "</code>";
93 const char ol_start[]                   = "<ol>";
94 const char ol_end[]                     = "</ol>";
95 const char ul_start[]                   = "<ul>";
96 const char ul_end[]                     = "</ul>";
97 const char li_start[]                   = "<li>";
98 const char li_end[]                     = "</li>";
99 const char h6_start[]                   = "<h6>";
100 const char h6_end[]                     = "</h6>";
101 const char h5_start[]                   = "<h5>";
102 const char h5_end[]                     = "</h5>";
103 const char h4_start[]                   = "<h4>";
104 const char h4_end[]                     = "</h4>";
105 const char h3_start[]                   = "<h3>";
106 const char h3_end[]                     = "</h3>";
107 const char h2_start[]                   = "<h2>";
108 const char h2_end[]                     = "</h2>";
109 const char h1_start[]                   = "<h1>";
110 const char h1_end[]                     = "</h1>";
111 const char p_start[]                    = "<p>";
112 const char p_end[]                      = "</p>";
113 const char space[]                      = " ";
114 const char a_start[]                    = "<a href=\"";
115 const char a_class[]                    = "\" class=\"";
116 const char a_rel[]                      = "\" rel=\"";
117 const char a_start_close[]              = "\">";
118 const char a_end[]                      = "</a>";
119 const char link_start[]                 = "[[";
120 const char link_end[]                   = "]]";
121 const char separator[]                  = "|";
122 const char ext_link_start[]             = "[";
123 const char backtick[]                   = "`";
124 const char quote[]                      = "\"";
125 const char ampersand[]                  = "&";
126 const char quot_entity[]                = "&quot;";
127 const char amp_entity[]                 = "&amp;";
128 const char lt_entity[]                  = "&lt;";
129 const char gt_entity[]                  = "&gt;";
130 const char escaped_blockquote[]         = "&gt; ";
131 const char ext_link_end[]               = "]";
132 const char literal_img_start[]          = "{{";
133 const char img_start[]                  = "<img src=\"";
134 const char img_end_xml[]                = "\" />";
135 const char img_end_html[]               = "\">";
136 const char img_alt[]                    = "\" alt=\"";
137 const char pre_class_start[]            = "<pre class=\"";
138 const char pre_class_end[]              = "-syntax\">";
139
140 // Mark the parser struct designated by ptr as a participant in Ruby's
141 // mark-and-sweep garbage collection scheme. A variable named name is placed on
142 // the C stack to prevent the structure from being prematurely collected.
143 #define GC_WRAP_PARSER(ptr, name) volatile VALUE name __attribute__((unused)) = Data_Wrap_Struct(rb_cObject, 0, parser_free, ptr)
144
145 parser_t *parser_new(void)
146 {
147     parser_t *parser                = ALLOC_N(parser_t, 1);
148     parser->capture                 = NULL; // not a real instance, pointer to other member's instance
149     parser->output                  = str_new();
150     parser->link_target             = str_new();
151     parser->link_text               = str_new();
152     parser->line_ending             = NULL; // caller should set up
153     parser->tabulation              = str_new();
154     parser->scope                   = ary_new();
155     parser->line                    = ary_new();
156     parser->line_buffer             = ary_new();
157     parser->external_link_class     = Qnil; // caller should set up
158     parser->external_link_rel       = Qnil; // caller should set up
159     parser->mailto_class            = Qnil; // caller should set up
160     parser->img_prefix              = Qnil; // caller should set up
161     parser->output_style            = HTML_OUTPUT;
162     parser->base_indent             = 0;
163     parser->current_indent          = 0;
164     parser->base_heading_level      = 0;
165     parser->pending_crlf            = false;
166     parser->autolink                = true;
167     parser->space_to_underscore     = true;
168     return parser;
169 }
170
171 void parser_free(parser_t *parser)
172 {
173     // we don't free parser->capture; it's just a redundant pointer
174     if (parser->output)         str_free(parser->output);
175     if (parser->link_target)    str_free(parser->link_target);
176     if (parser->link_text)      str_free(parser->link_text);
177     if (parser->line_ending)    str_free(parser->line_ending);
178     if (parser->tabulation)     str_free(parser->tabulation);
179     if (parser->scope)          ary_free(parser->scope);
180     if (parser->line)           ary_free(parser->line);
181     if (parser->line_buffer)    ary_free(parser->line_buffer);
182     free(parser);
183 }
184
185 // for testing and debugging only
186 VALUE Wikitext_parser_tokenize(VALUE self, VALUE string)
187 {
188     if (NIL_P(string))
189         return Qnil;
190     string = StringValue(string);
191     VALUE tokens = rb_ary_new();
192     char *p = RSTRING_PTR(string);
193     long len = RSTRING_LEN(string);
194     char *pe = p + len;
195     token_t token;
196     next_token(&token, NULL, p, pe);
197     rb_ary_push(tokens, wiki_token(&token));
198     while (token.type != END_OF_FILE)
199     {
200         next_token(&token, &token, NULL, pe);
201         rb_ary_push(tokens, wiki_token(&token));
202     }
203     return tokens;
204 }
205
206 // for benchmarking raw tokenization speed only
207 VALUE Wikitext_parser_benchmarking_tokenize(VALUE self, VALUE string)
208 {
209     if (NIL_P(string))
210         return Qnil;
211     string = StringValue(string);
212     char *p = RSTRING_PTR(string);
213     long len = RSTRING_LEN(string);
214     char *pe = p + len;
215     token_t token;
216     next_token(&token, NULL, p, pe);
217     while (token.type != END_OF_FILE)
218         next_token(&token, &token, NULL, pe);
219     return Qnil;
220 }
221
222 VALUE Wikitext_parser_fulltext_tokenize(int argc, VALUE *argv, VALUE self)
223 {
224     // process arguments
225     VALUE string, options;
226     if (rb_scan_args(argc, argv, "11", &string, &options) == 1) // 1 mandatory argument, 1 optional argument
227         options = Qnil;
228     if (NIL_P(string))
229         return Qnil;
230     string = StringValue(string);
231     VALUE tokens = rb_ary_new();
232
233     // check instance variables
234     VALUE min = rb_iv_get(self, "@minimum_fulltext_token_length");
235
236     // process options hash (can override instance variables)
237     if (!NIL_P(options) && TYPE(options) == T_HASH)
238     {
239         if (rb_funcall(options, rb_intern("has_key?"), 1, ID2SYM(rb_intern("minimum"))) == Qtrue)
240             min = rb_hash_aref(options, ID2SYM(rb_intern("minimum")));
241     }
242     int min_len = NIL_P(min) ? 3 : NUM2INT(min);
243     if (min_len < 0)
244         min_len = 0;
245
246     // set up scanner
247     char *p = RSTRING_PTR(string);
248     long len = RSTRING_LEN(string);
249     char *pe = p + len;
250     token_t token;
251     token_t *_token = &token;
252     next_token(&token, NULL, p, pe);
253     while (token.type != END_OF_FILE)
254     {
255         switch (token.type)
256         {
257             case URI:
258             case MAIL:
259             case ALNUM:
260                 if (TOKEN_LEN(_token) >= min_len)
261                     rb_ary_push(tokens, TOKEN_TEXT(_token));
262                 break;
263             default:
264                 // ignore everything else
265                 break;
266         }
267         next_token(&token, &token, NULL, pe);
268     }
269     return tokens;
270 }
271
272 // we downcase "in place", overwriting the original contents of the buffer
273 void wiki_downcase_bang(char *ptr, long len)
274 {
275     for (long i = 0; i < len; i++)
276     {
277         if (ptr[i] >= 'A' && ptr[i] <= 'Z')
278             ptr[i] += 32;
279     }
280 }
281
282 void wiki_append_entity_from_utf32_char(str_t *output, uint32_t character)
283 {
284     char hex_string[8]  = { '&', '#', 'x', 0, 0, 0, 0, ';' };
285     char scratch        = (character & 0xf000) >> 12;
286     hex_string[3]       = (scratch <= 9 ? scratch + 48 : scratch + 87);
287     scratch             = (character & 0x0f00) >> 8;
288     hex_string[4]       = (scratch <= 9 ? scratch + 48 : scratch + 87);
289     scratch             = (character & 0x00f0) >> 4;
290     hex_string[5]       = (scratch <= 9 ? scratch + 48 : scratch + 87);
291     scratch             = character & 0x000f;
292     hex_string[6]       = (scratch <= 9 ? scratch + 48 : scratch + 87);
293     str_append(output, hex_string, sizeof(hex_string));
294 }
295
296 // Convert a single UTF-8 codepoint to UTF-32
297 //
298 // Expects an input buffer, src, containing a UTF-8 encoded character (which
299 // may be multi-byte). The end of the input buffer, end, is also passed in to
300 // allow the detection of invalidly truncated codepoints. The number of bytes
301 // in the UTF-8 character (between 1 and 4) is returned by reference in
302 // width_out.
303 //
304 // Raises a RangeError if the supplied character is invalid UTF-8.
305 uint32_t wiki_utf8_to_utf32(char *src, char *end, long *width_out)
306 {
307     uint32_t dest;
308     if ((unsigned char)src[0] <= 0x7f)
309     {
310         // ASCII
311         dest = src[0];
312         *width_out = 1;
313     }
314     else if ((src[0] & 0xe0) == 0xc0)
315     {
316         // byte starts with 110..... : this should be a two-byte sequence
317         if (src + 1 >= end)
318             // no second byte
319             rb_raise(eWikitextParserError, "invalid encoding: truncated byte sequence");
320         else if (((unsigned char)src[0] == 0xc0) ||
321                 ((unsigned char)src[0] == 0xc1))
322             // overlong encoding: lead byte of 110..... but code point <= 127
323             rb_raise(eWikitextParserError, "invalid encoding: overlong encoding");
324         else if ((src[1] & 0xc0) != 0x80 )
325             // should have second byte starting with 10......
326             rb_raise(eWikitextParserError, "invalid encoding: malformed byte sequence");
327
328         dest =
329             ((uint32_t)(src[0] & 0x1f)) << 6 |
330             (src[1] & 0x3f);
331         *width_out = 2;
332     }
333     else if ((src[0] & 0xf0) == 0xe0)
334     {
335         // byte starts with 1110.... : this should be a three-byte sequence
336         if (src + 2 >= end)
337             // missing second or third byte
338             rb_raise(eWikitextParserError, "invalid encoding: truncated byte sequence");
339         else if (((src[1] & 0xc0) != 0x80 ) ||
340                 ((src[2] & 0xc0) != 0x80 ))
341             // should have second and third bytes starting with 10......
342             rb_raise(eWikitextParserError, "invalid encoding: malformed byte sequence");
343
344         dest =
345             ((uint32_t)(src[0] & 0x0f)) << 12 |
346             ((uint32_t)(src[1] & 0x3f)) << 6 |
347             (src[2] & 0x3f);
348         *width_out = 3;
349     }
350     else if ((src[0] & 0xf8) == 0xf0)
351     {
352         // bytes starts with 11110... : this should be a four-byte sequence
353         if (src + 3 >= end)
354             // missing second, third, or fourth byte
355             rb_raise(eWikitextParserError, "invalid encoding: truncated byte sequence");
356         else if ((unsigned char)src[0] >= 0xf5 &&
357                 (unsigned char)src[0] <= 0xf7)
358             // disallowed by RFC 3629 (codepoints above 0x10ffff)
359             rb_raise(eWikitextParserError, "invalid encoding: overlong encoding");
360         else if (((src[1] & 0xc0) != 0x80 ) ||
361                 ((src[2] & 0xc0) != 0x80 ) ||
362                 ((src[3] & 0xc0) != 0x80 ))
363             // should have second and third bytes starting with 10......
364             rb_raise(eWikitextParserError, "invalid encoding: malformed byte sequence");
365
366         dest =
367             ((uint32_t)(src[0] & 0x07)) << 18 |
368             ((uint32_t)(src[1] & 0x3f)) << 12 |
369             ((uint32_t)(src[1] & 0x3f)) << 6 |
370             (src[2] & 0x3f);
371         *width_out = 4;
372     }
373     else
374         rb_raise(eWikitextParserError, "invalid encoding: unexpected byte");
375     return dest;
376 }
377
378 // - non-printable (non-ASCII) characters converted to numeric entities
379 // - QUOT and AMP characters converted to named entities
380 // - if trim is true, leading and trailing whitespace trimmed
381 // - if trim is false, there is no special treatment of spaces
382 void wiki_append_sanitized_link_target(str_t *link_target, str_t *output, bool trim)
383 {
384     char    *src        = link_target->ptr;
385     char    *start      = src;                          // remember this so we can check if we're at the start
386     char    *non_space  = output->ptr + output->len;    // remember last non-space character output
387     char    *end        = src + link_target->len;
388     while (src < end)
389     {
390         // need at most 8 bytes to display each input character (&#x0000;)
391         if (output->ptr + output->len + 8 > output->ptr + output->capacity) // outgrowing buffer, must grow
392         {
393             char *old_ptr = output->ptr;
394             str_grow(output, output->len + (end - src) * 8);    // allocate enough for worst case
395             if (old_ptr != output->ptr) // may have moved
396                 non_space += output->ptr - old_ptr;
397         }
398
399         if (*src == '"')
400         {
401             char quot_entity_literal[] = { '&', 'q', 'u', 'o', 't', ';' };  // no trailing NUL
402             str_append(output, quot_entity_literal, sizeof(quot_entity_literal));
403         }
404         else if (*src == '&')
405         {
406             char amp_entity_literal[] = { '&', 'a', 'm', 'p', ';' };    // no trailing NUL
407             str_append(output, amp_entity_literal, sizeof(amp_entity_literal));
408         }
409         else if (*src == '<' || *src == '>')
410             rb_raise(rb_eRangeError, "invalid link text (\"%c\" may not appear in link text)", *src);
411         else if (*src == ' ' && src == start && trim)
412             start++;                            // we eat leading space
413         else if (*src >= 0x20 && *src <= 0x7e)  // printable ASCII
414         {
415             *(output->ptr + output->len) = *src;
416             output->len++;
417         }
418         else    // all others: must convert to entities
419         {
420             long        width;
421             wiki_append_entity_from_utf32_char(output, wiki_utf8_to_utf32(src, end, &width));
422             src         += width;
423             non_space   = output->ptr + output->len;
424             continue;
425         }
426         if (*src != ' ')
427             non_space = output->ptr + output->len;
428         src++;
429     }
430
431     // trim trailing space if necessary
432     if (trim && output->ptr + output->len != non_space)
433         output->len -= (output->ptr + output->len) - non_space;
434 }
435
436 // prepare hyperlink and append it to parser->output
437 // if check_autolink is true, checks parser->autolink to decide whether to emit a real hyperlink
438 // or merely the literal link target
439 // if link_text is Qnil, the link_target is re-used for the link text
440 void wiki_append_hyperlink(parser_t *parser, VALUE link_prefix, str_t *link_target, str_t *link_text, VALUE link_class, VALUE link_rel, bool check_autolink)
441 {
442     if (check_autolink && !parser->autolink)
443         str_append_str(parser->output, link_target);
444     else
445     {
446         str_append(parser->output, a_start, sizeof(a_start) - 1);               // <a href="
447         if (!NIL_P(link_prefix))
448             str_append_string(parser->output, link_prefix);
449         str_append_str(parser->output, link_target);
450
451         // special handling for mailto URIs
452         const char *mailto = "mailto:";
453         long mailto_len = (long)sizeof(mailto) - 1; // don't count NUL byte
454         if ((link_target->len >= mailto_len &&
455              strncmp(mailto, link_target->ptr, mailto_len) == 0) ||
456             (!NIL_P(link_prefix) &&
457              RSTRING_LEN(link_prefix) >= mailto_len &&
458              strncmp(mailto, RSTRING_PTR(link_prefix), mailto_len) == 0))
459             link_class = parser->mailto_class; // use mailto_class from parser
460         if (link_class != Qnil)
461         {
462             str_append(parser->output, a_class, sizeof(a_class) - 1);           // " class="
463             str_append_string(parser->output, link_class);
464         }
465         if (link_rel != Qnil)
466         {
467             str_append(parser->output, a_rel, sizeof(a_rel) - 1);               // " rel="
468             str_append_string(parser->output, link_rel);
469         }
470         str_append(parser->output, a_start_close, sizeof(a_start_close) - 1);   // ">
471         if (!link_text || link_text->len == 0) // re-use link_target
472             str_append_str(parser->output, link_target);
473         else
474             str_append_str(parser->output, link_text);
475         str_append(parser->output, a_end, sizeof(a_end) - 1);                   // </a>
476     }
477 }
478
479 void wiki_append_img(parser_t *parser, char *token_ptr, int token_len)
480 {
481     str_append(parser->output, img_start, sizeof(img_start) - 1);           // <img src="
482     if (!NIL_P(parser->img_prefix) && *token_ptr != '/')                    // len always > 0
483         str_append_string(parser->output, parser->img_prefix);
484     str_append(parser->output, token_ptr, token_len);
485     str_append(parser->output, img_alt, sizeof(img_alt) - 1);               // " alt="
486     str_append(parser->output, token_ptr, token_len);
487     if (parser->output_style == XML_OUTPUT)
488         str_append(parser->output, img_end_xml, sizeof(img_end_xml) - 1);   // " />
489     else
490         str_append(parser->output, img_end_html, sizeof(img_end_html) - 1); // ">
491 }
492
493 // will emit indentation only if we are about to emit any of:
494 //      <blockquote>, <p>, <ul>, <ol>, <li>, <h1> etc, <pre>
495 // each time we enter one of those spans must ++ the indentation level
496 void wiki_indent(parser_t *parser)
497 {
498     if (parser->base_indent == -1) // indentation disabled
499         return;
500     int space_count = parser->current_indent + parser->base_indent;
501     if (space_count > 0)
502     {
503         char *old_end, *new_end;
504         if (parser->tabulation->len < space_count)
505             str_grow(parser->tabulation, space_count); // reallocates if necessary
506         old_end = parser->tabulation->ptr + parser->tabulation->len;
507         new_end = parser->tabulation->ptr + space_count;
508         while (old_end < new_end)
509             *old_end++ = ' ';
510         if (space_count > parser->tabulation->len)
511             parser->tabulation->len = space_count;
512         str_append(parser->output, parser->tabulation->ptr, space_count);
513     }
514     parser->current_indent += 2;
515 }
516
517 void wiki_append_pre_start(parser_t *parser, token_t *token)
518 {
519     wiki_indent(parser);
520     if ((size_t)TOKEN_LEN(token) > sizeof(pre_start) - 1)
521     {
522         str_append(parser->output, pre_class_start, sizeof(pre_class_start) - 1);   // <pre class="
523         str_append(parser->output, token->start + 11, TOKEN_LEN(token) - 13);       // (the "lang" substring)
524         str_append(parser->output, pre_class_end, sizeof(pre_class_end) - 1);       // -syntax">
525     }
526     else
527         str_append(parser->output, pre_start, sizeof(pre_start) - 1);
528     ary_push(parser->scope, PRE_START);
529     ary_push(parser->line, PRE_START);
530 }
531
532 void wiki_dedent(parser_t *parser, bool emit)
533 {
534     if (parser->base_indent == -1) // indentation disabled
535         return;
536     parser->current_indent -= 2;
537     if (!emit)
538         return;
539     int space_count = parser->current_indent + parser->base_indent;
540     if (space_count > 0)
541         str_append(parser->output, parser->tabulation->ptr, space_count);
542 }
543
544 // Pops a single item off the parser's scope stack.
545 // A corresponding closing tag is written to the target string.
546 // The target string may be the main output buffer, or a substring capturing buffer if a link is being scanned.
547 void wiki_pop_from_stack(parser_t *parser, str_t *target)
548 {
549     int top = ary_entry(parser->scope, -1);
550     if (NO_ITEM(top))
551         return;
552     if (!target)
553         target = parser->output;
554
555     // for headings, take base_heading_level into account
556     if (top >= H1_START && top <= H6_START)
557     {
558         top += parser->base_heading_level;
559         // no need to check for underflow (base_heading_level is never negative)
560         if (top > H6_START)
561             top = H6_START;
562     }
563
564     switch (top)
565     {
566         case PRE:
567         case PRE_START:
568             str_append(target, pre_end, sizeof(pre_end) - 1);
569             str_append_str(target, parser->line_ending);
570             wiki_dedent(parser, false);
571             break;
572
573         case BLOCKQUOTE:
574         case BLOCKQUOTE_START:
575             wiki_dedent(parser, true);
576             str_append(target, blockquote_end, sizeof(blockquote_end) - 1);
577             str_append_str(target, parser->line_ending);
578             break;
579
580         case NO_WIKI_START:
581             // not a real HTML tag; so nothing to pop
582             break;
583
584         case STRONG:
585         case STRONG_START:
586             str_append(target, strong_end, sizeof(strong_end) - 1);
587             break;
588
589         case EM:
590         case EM_START:
591             str_append(target, em_end, sizeof(em_end) - 1);
592             break;
593
594         case TT:
595         case TT_START:
596             str_append(target, code_end, sizeof(code_end) - 1);
597             break;
598
599         case OL:
600             wiki_dedent(parser, true);
601             str_append(target, ol_end, sizeof(ol_end) - 1);
602             str_append_str(target, parser->line_ending);
603             break;
604
605         case UL:
606             wiki_dedent(parser, true);
607             str_append(target, ul_end, sizeof(ul_end) - 1);
608             str_append_str(target, parser->line_ending);
609             break;
610
611         case NESTED_LIST:
612             // next token to pop will be a LI
613             // LI is an interesting token because sometimes we want it to behave like P (ie. do a non-emitting indent)
614             // and other times we want it to behave like BLOCKQUOTE (ie. when it has a nested list inside)
615             // hence this hack: we do an emitting dedent on behalf of the LI that we know must be coming
616             // and then when we pop the actual LI itself (below) we do the standard non-emitting indent
617             wiki_dedent(parser, true);      // we really only want to emit the spaces
618             parser->current_indent += 2;    // we don't want to decrement the actual indent level, so put it back
619             break;
620
621         case LI:
622             str_append(target, li_end, sizeof(li_end) - 1);
623             str_append_str(target, parser->line_ending);
624             wiki_dedent(parser, false);
625             break;
626
627         case H6_START:
628             str_append(target, h6_end, sizeof(h6_end) - 1);
629             str_append_str(target, parser->line_ending);
630             wiki_dedent(parser, false);
631             break;
632
633         case H5_START:
634             str_append(target, h5_end, sizeof(h5_end) - 1);
635             str_append_str(target, parser->line_ending);
636             wiki_dedent(parser, false);
637             break;
638
639         case H4_START:
640             str_append(target, h4_end, sizeof(h4_end) - 1);
641             str_append_str(target, parser->line_ending);
642             wiki_dedent(parser, false);
643             break;
644
645         case H3_START:
646             str_append(target, h3_end, sizeof(h3_end) - 1);
647             str_append_str(target, parser->line_ending);
648             wiki_dedent(parser, false);
649             break;
650
651         case H2_START:
652             str_append(target, h2_end, sizeof(h2_end) - 1);
653             str_append_str(target, parser->line_ending);
654             wiki_dedent(parser, false);
655             break;
656
657         case H1_START:
658             str_append(target, h1_end, sizeof(h1_end) - 1);
659             str_append_str(target, parser->line_ending);
660             wiki_dedent(parser, false);
661             break;
662
663         case LINK_START:
664             // not an HTML tag; so nothing to emit
665             break;
666
667         case EXT_LINK_START:
668             // not an HTML tag; so nothing to emit
669             break;
670
671         case PATH:
672             // not an HTML tag; so nothing to emit
673             break;
674
675         case SPACE:
676             // not an HTML tag (only used to separate an external link target from the link text); so nothing to emit
677             break;
678
679         case SEPARATOR:
680             // not an HTML tag (only used to separate an external link target from the link text); so nothing to emit
681             break;
682
683         case P:
684             str_append(target, p_end, sizeof(p_end) - 1);
685             str_append_str(target, parser->line_ending);
686             wiki_dedent(parser, false);
687             break;
688
689         case END_OF_FILE:
690             // nothing to do
691             break;
692
693         default:
694             // should probably raise an exception here
695             break;
696     }
697     ary_pop(parser->scope);
698 }
699
700 // Pops items off the top of parser's scope stack, accumulating closing tags for them into the target string, until item is reached.
701 // If including is true then the item itself is also popped.
702 // The target string may be the main output buffer, or a substring capturing buffer when scanning links.
703 void wiki_pop_from_stack_up_to(parser_t *parser, str_t *target, int item, bool including)
704 {
705     int continue_looping = 1;
706     do
707     {
708         int top = ary_entry(parser->scope, -1);
709         if (NO_ITEM(top))
710             return;
711         if (top == item)
712         {
713             if (!including)
714                 return;
715             continue_looping = 0;
716         }
717         wiki_pop_from_stack(parser, target);
718     } while (continue_looping);
719 }
720
721 void wiki_pop_all_from_stack(parser_t *parser)
722 {
723     for (int i = 0, max = parser->scope->count; i < max; i++)
724         wiki_pop_from_stack(parser, NULL);
725 }
726
727 void wiki_start_para_if_necessary(parser_t *parser)
728 {
729     if (parser->capture)
730         return;
731
732     // if no block open yet, or top of stack is BLOCKQUOTE/BLOCKQUOTE_START (with nothing in it yet)
733     if (parser->scope->count == 0 ||
734         ary_entry(parser->scope, -1) == BLOCKQUOTE ||
735         ary_entry(parser->scope, -1) == BLOCKQUOTE_START)
736     {
737         wiki_indent(parser);
738         str_append(parser->output, p_start, sizeof(p_start) - 1);
739         ary_push(parser->scope, P);
740         ary_push(parser->line, P);
741     }
742     else if (parser->pending_crlf)
743     {
744         if (IN(P))
745             // already in a paragraph block; convert pending CRLF into a space
746             str_append(parser->output, space, sizeof(space) - 1);
747         else if (IN(PRE))
748             // PRE blocks can have pending CRLF too (helps us avoid emitting the trailing newline)
749             str_append_str(parser->output, parser->line_ending);
750     }
751     parser->pending_crlf = false;
752 }
753
754 void wiki_emit_pending_crlf_if_necessary(parser_t *parser)
755 {
756     if (parser->pending_crlf)
757     {
758         str_append_str(parser->output, parser->line_ending);
759         parser->pending_crlf = false;
760     }
761 }
762
763 // Helper function that pops any excess elements off scope (pushing is already handled in the respective rules).
764 // For example, given input like:
765 //
766 //      > > foo
767 //      bar
768 //
769 // Upon seeing "bar", we want to pop two BLOCKQUOTE elements from the scope.
770 // The reverse case (shown below) is handled from inside the BLOCKQUOTE rule itself:
771 //
772 //      foo
773 //      > > bar
774 //
775 // Things are made slightly more complicated by the fact that there is one block-level tag that can be on the scope
776 // but not on the line scope:
777 //
778 //      <blockquote>foo
779 //      bar</blockquote>
780 //
781 // Here on seeing "bar" we have one item on the scope (BLOCKQUOTE_START) which we don't want to pop, but we have nothing
782 // on the line scope.
783 // Luckily, BLOCKQUOTE_START tokens can only appear at the start of the scope array, so we can check for them first before
784 // entering the for loop.
785 void wiki_pop_excess_elements(parser_t *parser)
786 {
787     if (parser->capture)
788         return;
789     for (int i = parser->scope->count - ary_count(parser->scope, BLOCKQUOTE_START), j = parser->line->count; i > j; i--)
790     {
791         // special case for last item on scope
792         if (i - j == 1)
793         {
794             // don't auto-pop P if it is only item on scope
795             if (ary_entry(parser->scope, -1) == P)
796             {
797                 // add P to the line scope to prevent us entering the loop at all next time around
798                 ary_push(parser->line, P);
799                 continue;
800             }
801         }
802         wiki_pop_from_stack(parser, NULL);
803     }
804 }
805
806 // trim parser->link_text in place
807 void wiki_trim_link_text(parser_t *parser)
808 {
809     char    *src        = parser->link_text->ptr;
810     char    *start      = src;                  // remember this so we can check if we're at the start
811     char    *left       = src;
812     char    *non_space  = src;                  // remember last non-space character output
813     char    *end        = src + parser->link_text->len;
814     while (src < end)
815     {
816         if (*src == ' ')
817         {
818             if (src == left)
819                 left++;
820         }
821         else
822             non_space = src;
823         src++;
824     }
825     if (left != start || non_space + 1 != end)
826     {
827         // TODO: could potentially avoid this memmove by extending the str_t struct with an "offset" or "free" member
828         parser->link_text->len = (non_space + 1) - left;
829         memmove(parser->link_text->ptr, left, parser->link_text->len);
830     }
831 }
832
833 VALUE Wikitext_parser_sanitize_link_target(VALUE self, VALUE string)
834 {
835     str_t *link_target = str_new_from_string(string);
836     GC_WRAP_STR(link_target, link_target_gc);
837     str_t *output = str_new();
838     GC_WRAP_STR(output, output_gc);
839     wiki_append_sanitized_link_target(link_target, output, true);
840     return string_from_str(output);
841 }
842
843 // Encodes the parser link_target member (in-place) according to RFCs 2396 and 2718
844 //
845 // Leading and trailing whitespace trimmed. Spaces are converted to
846 // underscores if the parser space_to_underscore member is true.
847 static void wiki_encode_link_target(parser_t *parser)
848 {
849     char        *src        = parser->link_target->ptr;
850     char        *start      = src;  // remember this so we can check if we're at the start
851     long        len         = parser->link_target->len;
852     if (!(len > 0))
853         return;
854     char        *end        = src + len;
855     long        dest_len    = len * 2;
856     char        *dest       = ALLOC_N(char, dest_len);
857     char        *dest_ptr   = dest; // hang on to this so we can pass it to free() later
858     char        *non_space  = dest; // remember last non-space character output
859     static char hex[]       = { '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'b', 'c', 'd', 'e', 'f' };
860     for (; src < end; src++)
861     {
862         // worst case: a single character may grow to 3 characters once encoded
863         if ((dest + 3) > (dest_ptr + dest_len))
864         {
865             // outgrowing buffer, must reallocate
866             char *old_dest      = dest;
867             char *old_dest_ptr  = dest_ptr;
868             dest_len            += len;
869             dest                = realloc(dest_ptr, dest_len);
870             if (dest == NULL)
871             {
872                 // would have used reallocf, but this has to run on Linux too, not just Darwin
873                 free(dest_ptr);
874                 rb_raise(rb_eNoMemError, "failed to re-allocate temporary storage (memory allocation error)");
875             }
876             dest_ptr    = dest;
877             dest        = dest_ptr + (old_dest - old_dest_ptr);
878             non_space   = dest_ptr + (non_space - old_dest_ptr);
879         }
880
881         // pass through unreserved characters
882         if ((*src >= 'a' && *src <= 'z') ||
883             (*src >= 'A' && *src <= 'Z') ||
884             (*src >= '0' && *src <= '9') ||
885             *src == '-' ||
886             *src == '_' ||
887             *src == '.' ||
888             *src == '~')
889         {
890             *dest++     = *src;
891             non_space   = dest;
892         }
893         else if (*src == ' ' && src == start)
894             start++;                    // we eat leading space
895         else if (*src == ' ' && parser->space_to_underscore)
896             *dest++     = '_';
897         else    // everything else gets URL-encoded
898         {
899             *dest++     = '%';
900             *dest++     = hex[(unsigned char)(*src) / 16];   // left
901             *dest++     = hex[(unsigned char)(*src) % 16];   // right
902             if (*src != ' ')
903                 non_space = dest;
904         }
905     }
906
907     // trim trailing space if necessary
908     if (non_space > dest_ptr && dest != non_space)
909         dest_len = non_space - dest_ptr;
910     else
911         dest_len = dest - dest_ptr;
912     str_clear(parser->link_target);
913     str_append(parser->link_target, dest_ptr, dest_len);
914     free(dest_ptr);
915 }
916
917 VALUE Wikitext_parser_encode_link_target(VALUE self, VALUE in)
918 {
919     parser_t parser;
920     parser.space_to_underscore      = false;
921     parser.link_target              = str_new_from_string(in);
922     GC_WRAP_STR(parser.link_target, link_target_gc);
923     wiki_encode_link_target(&parser);
924     return string_from_str(parser.link_target);
925 }
926
927 // returns 1 (true) if supplied string is blank (nil, empty, or all whitespace)
928 // returns 0 (false) otherwise
929 bool wiki_blank(str_t *str)
930 {
931     if (str->len == 0)
932         return true;
933     for (char *ptr = str->ptr,
934         *end = str->ptr + str->len;
935         ptr < end; ptr++)
936     {
937         if (*ptr != ' ')
938             return false;
939     }
940     return true;
941 }
942
943 void wiki_rollback_failed_internal_link(parser_t *parser)
944 {
945     if (!IN(LINK_START))
946         return; // nothing to do!
947     int scope_includes_separator = IN(SEPARATOR);
948     wiki_pop_from_stack_up_to(parser, NULL, LINK_START, true);
949     str_append(parser->output, link_start, sizeof(link_start) - 1);
950     if (parser->link_target->len > 0)
951     {
952         wiki_append_sanitized_link_target(parser->link_target, parser->output, false);
953         if (scope_includes_separator)
954         {
955             str_append(parser->output, separator, sizeof(separator) - 1);
956             if (parser->link_text->len > 0)
957                 str_append_str(parser->output, parser->link_text);
958         }
959     }
960     parser->capture = NULL;
961     str_clear(parser->link_target);
962     str_clear(parser->link_text);
963 }
964
965 void wiki_rollback_failed_external_link(parser_t *parser)
966 {
967     if (!IN(EXT_LINK_START))
968         return; // nothing to do!
969
970     // store a couple of values before popping
971     int scope_includes_space = IN(SPACE);
972     VALUE link_class = IN(PATH) ? Qnil : parser->external_link_class;
973     VALUE link_rel   = IN(PATH) ? Qnil : parser->external_link_rel;
974     wiki_pop_from_stack_up_to(parser, NULL, EXT_LINK_START, true);
975
976     str_append(parser->output, ext_link_start, sizeof(ext_link_start) - 1);
977     if (parser->link_target->len > 0)
978     {
979         wiki_append_hyperlink(parser, Qnil, parser->link_target, NULL, link_class, link_rel, true);
980         if (scope_includes_space)
981         {
982             str_append(parser->output, space, sizeof(space) - 1);
983             if (parser->link_text->len > 0)
984                 str_append_str(parser->output, parser->link_text);
985         }
986     }
987     parser->capture = NULL;
988     str_clear(parser->link_target);
989     str_clear(parser->link_text);
990 }
991
992 void wiki_rollback_failed_link(parser_t *parser)
993 {
994     wiki_rollback_failed_internal_link(parser);
995     wiki_rollback_failed_external_link(parser);
996 }
997
998 VALUE Wikitext_parser_initialize(int argc, VALUE *argv, VALUE self)
999 {
1000     // process arguments
1001     VALUE options;
1002     if (rb_scan_args(argc, argv, "01", &options) == 0) // 0 mandatory arguments, 1 optional argument
1003         options = Qnil;
1004
1005     // defaults
1006     VALUE autolink                      = Qtrue;
1007     VALUE line_ending                   = rb_str_new2("\n");
1008     VALUE external_link_class           = rb_str_new2("external");
1009     VALUE external_link_rel             = Qnil;
1010     VALUE mailto_class                  = rb_str_new2("mailto");
1011     VALUE internal_link_prefix          = rb_str_new2("/wiki/");
1012     VALUE img_prefix                    = rb_str_new2("/images/");
1013     VALUE output_style                  = ID2SYM(rb_intern("html"));
1014     VALUE space_to_underscore           = Qtrue;
1015     VALUE minimum_fulltext_token_length = INT2NUM(3);
1016     VALUE base_heading_level            = INT2NUM(0);
1017
1018     // process options hash (override defaults)
1019     if (!NIL_P(options) && TYPE(options) == T_HASH)
1020     {
1021 #define OVERRIDE_IF_SET(name)   rb_funcall(options, rb_intern("has_key?"), 1, ID2SYM(rb_intern(#name))) == Qtrue ? \
1022                                 rb_hash_aref(options, ID2SYM(rb_intern(#name))) : name
1023         autolink                        = OVERRIDE_IF_SET(autolink);
1024         line_ending                     = OVERRIDE_IF_SET(line_ending);
1025         external_link_class             = OVERRIDE_IF_SET(external_link_class);
1026         external_link_rel               = OVERRIDE_IF_SET(external_link_rel);
1027         mailto_class                    = OVERRIDE_IF_SET(mailto_class);
1028         internal_link_prefix            = OVERRIDE_IF_SET(internal_link_prefix);
1029         img_prefix                      = OVERRIDE_IF_SET(img_prefix);
1030         output_style                    = OVERRIDE_IF_SET(output_style);
1031         space_to_underscore             = OVERRIDE_IF_SET(space_to_underscore);
1032         minimum_fulltext_token_length   = OVERRIDE_IF_SET(minimum_fulltext_token_length);
1033         base_heading_level              = OVERRIDE_IF_SET(base_heading_level);
1034     }
1035
1036     // no need to call super here; rb_call_super()
1037     rb_iv_set(self, "@autolink",                        autolink);
1038     rb_iv_set(self, "@line_ending",                     line_ending);
1039     rb_iv_set(self, "@external_link_class",             external_link_class);
1040     rb_iv_set(self, "@external_link_rel",               external_link_rel);
1041     rb_iv_set(self, "@mailto_class",                    mailto_class);
1042     rb_iv_set(self, "@internal_link_prefix",            internal_link_prefix);
1043     rb_iv_set(self, "@img_prefix",                      img_prefix);
1044     rb_iv_set(self, "@output_style",                    output_style);
1045     rb_iv_set(self, "@space_to_underscore",             space_to_underscore);
1046     rb_iv_set(self, "@minimum_fulltext_token_length",   minimum_fulltext_token_length);
1047     rb_iv_set(self, "@base_heading_level",              base_heading_level);
1048     return self;
1049 }
1050
1051 VALUE Wikitext_parser_profiling_parse(VALUE self, VALUE string)
1052 {
1053     for (int i = 0; i < 100000; i++)
1054         Wikitext_parser_parse(1, &string, self);
1055     return Qnil;
1056 }
1057
1058 // convert a Ruby object (:xml, :html etc) into an int output style
1059 int Wikitext_output_style(VALUE output)
1060 {
1061     if (TYPE(output) == T_SYMBOL)
1062     {
1063         if (SYM2ID(output) == rb_intern("xml"))
1064             return XML_OUTPUT;
1065     }
1066     return HTML_OUTPUT; // fall back to default
1067 }
1068
1069 VALUE Wikitext_parser_parse(int argc, VALUE *argv, VALUE self)
1070 {
1071     // process arguments
1072     VALUE string, options;
1073     if (rb_scan_args(argc, argv, "11", &string, &options) == 1) // 1 mandatory argument, 1 optional argument
1074         options = Qnil;
1075     if (NIL_P(string))
1076         return Qnil;
1077     string = StringValue(string);
1078
1079     // access these once per parse
1080     VALUE line_ending   = rb_iv_get(self, "@line_ending");
1081     line_ending         = StringValue(line_ending);
1082     VALUE link_class    = rb_iv_get(self, "@external_link_class");
1083     link_class          = NIL_P(link_class) ? Qnil : StringValue(link_class);
1084     VALUE link_rel      = rb_iv_get(self, "@external_link_rel");
1085     link_rel            = NIL_P(link_rel) ? Qnil : StringValue(link_rel);
1086     VALUE mailto_class  = rb_iv_get(self, "@mailto_class");
1087     mailto_class        = NIL_P(mailto_class) ? Qnil : StringValue(mailto_class);
1088     VALUE prefix        = rb_iv_get(self, "@internal_link_prefix");
1089     int output_style    = Wikitext_output_style(rb_iv_get(self, "@output_style"));
1090
1091     // process options hash
1092     int base_indent = 0;
1093     int base_heading_level = NUM2INT(rb_iv_get(self, "@base_heading_level"));
1094     VALUE link_proc = Qnil;
1095     if (!NIL_P(options) && TYPE(options) == T_HASH)
1096     {
1097         // :indent => 0 (or more)
1098         ID has_key = rb_intern("has_key?");
1099         ID id = ID2SYM(rb_intern("indent"));
1100         if (rb_funcall(options, has_key, 1, id) == Qtrue)
1101         {
1102             VALUE indent = rb_hash_aref(options, id);
1103             if (indent == Qfalse)
1104                 base_indent = -1; // indentation disabled
1105             else
1106             {
1107                 base_indent = NUM2INT(indent);
1108                 if (base_indent < 0)
1109                     base_indent = 0;
1110             }
1111         }
1112
1113         // :base_heading_level => 0/1/2/3/4/5/6
1114         id = ID2SYM(rb_intern("base_heading_level"));
1115         if (rb_funcall(options, has_key, 1, id) == Qtrue)
1116             base_heading_level = NUM2INT(rb_hash_aref(options, id));
1117
1118         // :external_link_rel => 'nofollow'
1119         id = ID2SYM(rb_intern("external_link_rel"));
1120         if (rb_funcall(options, has_key, 1, id) == Qtrue)
1121         {
1122             link_rel = rb_hash_aref(options, id);
1123             link_rel = NIL_P(link_rel) ? Qnil : StringValue(link_rel);
1124         }
1125
1126         // :output_style => :html/:xml
1127         id = ID2SYM(rb_intern("output_style"));
1128         if (rb_funcall(options, has_key, 1, id) == Qtrue)
1129             output_style = Wikitext_output_style(rb_hash_aref(options, id));
1130
1131         // :link_proc => lambda { |link_target| ... }
1132         id = ID2SYM(rb_intern("link_proc"));
1133         if (rb_funcall(options, has_key, 1, id) == Qtrue)
1134             link_proc = rb_hash_aref(options, id);
1135     }
1136
1137     // normalize, regardless of whether this came from instance variable or override
1138     if (base_heading_level < 0)
1139         base_heading_level = 0;
1140     if (base_heading_level > 6)
1141         base_heading_level = 6;
1142
1143     // set up scanner
1144     char *p = RSTRING_PTR(string);
1145     long len = RSTRING_LEN(string);
1146     char *pe = p + len;
1147
1148     // set up parser struct to make passing parameters a little easier
1149     parser_t *parser                = parser_new();
1150     GC_WRAP_PARSER(parser, parser_gc);
1151     parser->external_link_class     = link_class;
1152     parser->external_link_rel       = link_rel;
1153     parser->mailto_class            = mailto_class;
1154     parser->img_prefix              = rb_iv_get(self, "@img_prefix");
1155     parser->autolink                = rb_iv_get(self, "@autolink") == Qtrue ? true : false;
1156     parser->space_to_underscore     = rb_iv_get(self, "@space_to_underscore") == Qtrue ? true : false;
1157     parser->line_ending             = str_new_from_string(line_ending);
1158     parser->base_indent             = base_indent;
1159     parser->base_heading_level      = base_heading_level;
1160     parser->output_style            = output_style;
1161
1162     // this simple looping design leads to a single enormous function,
1163     // but it's faster than doing actual recursive descent and also secure in the face of
1164     // malicious input that seeks to overflow the stack
1165     // (with "<blockquote><blockquote><blockquote>..." times by 10,000, for example)
1166     // given that we expect to deal with a lot of malformed input, a recursive descent design is less appropriate
1167     // than a straightforward looping translator like this one anyway
1168     token_t _token;
1169     _token.type = NO_TOKEN;
1170     token_t *token = NULL;
1171     do
1172     {
1173         // note that whenever we grab a token we push it into the line buffer
1174         // this provides us with context-sensitive "memory" of what's been seen so far on this line
1175 #define NEXT_TOKEN()    token = &_token, next_token(token, token, NULL, pe), ary_push(parser->line_buffer, token->type)
1176
1177         // check to see if we have a token hanging around from a previous iteration of this loop
1178         if (token == NULL)
1179         {
1180             if (_token.type == NO_TOKEN)
1181             {
1182                 // first time here (haven't started scanning yet)
1183                 token = &_token;
1184                 next_token(token, NULL, p, pe);
1185                 ary_push(parser->line_buffer, token->type);
1186             }
1187             else
1188                 // already scanning
1189                 NEXT_TOKEN();
1190         }
1191         int type = token->type;
1192
1193         // can't declare new variables inside a switch statement, so predeclare them here
1194         long remove_strong          = -1;
1195         long remove_em              = -1;
1196
1197         // general purpose counters, flags and pointers
1198         long i                      = 0;
1199         long j                      = 0;
1200         long k                      = 0;
1201         str_t *output               = NULL;
1202         str_t _token_str;
1203         str_t *token_str            = &_token_str;
1204
1205         // The following giant switch statement contains cases for all the possible token types.
1206         // In the most basic sense we are emitting the HTML that corresponds to each token,
1207         // but some tokens require context information in order to decide what to output.
1208         // For example, does the STRONG token (''') translate to <strong> or </strong>?
1209         // So when looking at any given token we have three state-maintaining variables which gives us a notion of "where we are":
1210         //
1211         //  - the "scope" stack (indicates what HTML DOM structures we are currently nested inside, similar to a CSS selector)
1212         //  - the line buffer (records tokens seen so far on the current line)
1213         //  - the line "scope" stack (indicates what the scope should be based only on what is visible on the line so far)
1214         //
1215         // Although this is fairly complicated, there is one key simplifying factor:
1216         // The translator continuously performs auto-correction, and this means that we always have a guarantee that the
1217         // scope stack (up to the current token) is valid; our translator can take this as a given.
1218         // Auto-correction basically consists of inserting missing tokens (preventing subsquent HTML from being messed up),
1219         // or converting illegal (unexpected) tokens to their plain text equivalents (providing visual feedback to Wikitext author).
1220         switch (type)
1221         {
1222             case PRE:
1223                 if (IN_EITHER_OF(NO_WIKI_START, PRE_START))
1224                 {
1225                     str_append(parser->output, space, sizeof(space) - 1);
1226                     break;
1227                 }
1228                 else if (IN(BLOCKQUOTE_START))
1229                 {
1230                     // this kind of nesting not allowed (to avoid user confusion)
1231                     wiki_pop_excess_elements(parser);
1232                     wiki_start_para_if_necessary(parser);
1233                     output = parser->capture ? parser->capture : parser->output;
1234                     str_append(output, space, sizeof(space) - 1);
1235                     break;
1236                 }
1237
1238                 // count number of BLOCKQUOTE tokens in line buffer and in scope stack
1239                 ary_push(parser->line, PRE);
1240                 i = ary_count(parser->line, BLOCKQUOTE);
1241                 j = ary_count(parser->scope, BLOCKQUOTE);
1242                 if (i < j)
1243                 {
1244                     // must pop (reduce nesting level)
1245                     for (i = j - i; i > 0; i--)
1246                         wiki_pop_from_stack_up_to(parser, NULL, BLOCKQUOTE, true);
1247                 }
1248
1249                 if (!IN(PRE))
1250                 {
1251                     parser->pending_crlf = false;
1252                     wiki_pop_from_stack_up_to(parser, NULL, BLOCKQUOTE, false);
1253                     wiki_indent(parser);
1254                     str_append(parser->output, pre_start, sizeof(pre_start) - 1);
1255                     ary_push(parser->scope, PRE);
1256                 }
1257                 break;
1258
1259             case PRE_START:
1260                 if (IN_ANY_OF(NO_WIKI_START, PRE, PRE_START))
1261                 {
1262                     wiki_emit_pending_crlf_if_necessary(parser);
1263                     str_append(parser->output, escaped_pre_start, sizeof(escaped_pre_start) - 1);
1264                 }
1265                 else if (IN(BLOCKQUOTE_START))
1266                 {
1267                     wiki_rollback_failed_link(parser); // if any
1268                     wiki_pop_from_stack_up_to(parser, NULL, BLOCKQUOTE_START, false);
1269                     wiki_append_pre_start(parser, token);
1270                 }
1271                 else if (IN(BLOCKQUOTE))
1272                 {
1273                     if (token->column_start == 1) // only allowed in first column
1274                     {
1275                         wiki_rollback_failed_link(parser); // if any
1276                         wiki_pop_all_from_stack(parser);
1277                         wiki_append_pre_start(parser, token);
1278                     }
1279                     else // PRE_START illegal here
1280                     {
1281                         output = parser->capture ? parser->capture : parser->output;
1282                         wiki_pop_excess_elements(parser);
1283                         wiki_start_para_if_necessary(parser);
1284                         str_append(output, escaped_pre_start, sizeof(escaped_pre_start) - 1);
1285                     }
1286                 }
1287                 else
1288                 {
1289                     wiki_rollback_failed_link(parser); // if any
1290                     wiki_pop_from_stack_up_to(parser, NULL, P, true);
1291                     wiki_append_pre_start(parser, token);
1292                 }
1293                 break;
1294
1295             case PRE_END:
1296                 if (IN_EITHER_OF(NO_WIKI_START, PRE))
1297                 {
1298                     wiki_emit_pending_crlf_if_necessary(parser);
1299                     str_append(parser->output, escaped_pre_end, sizeof(escaped_pre_end) - 1);
1300                 }
1301                 else
1302                 {
1303                     if (IN(PRE_START))
1304                         wiki_pop_from_stack_up_to(parser, parser->output, PRE_START, true);
1305                     else
1306                     {
1307                         output = parser->capture ? parser->capture : parser->output;
1308                         wiki_pop_excess_elements(parser);
1309                         wiki_start_para_if_necessary(parser);
1310                         str_append(output, escaped_pre_end, sizeof(escaped_pre_end) - 1);
1311                     }
1312                 }
1313                 break;
1314
1315             case BLOCKQUOTE:
1316                 if (IN_EITHER_OF(NO_WIKI_START, PRE_START))
1317                     // no need to check for <pre>; can never appear inside it
1318                     str_append(parser->output, escaped_blockquote, TOKEN_LEN(token) + 3); // will either emit "&gt;" or "&gt; "
1319                 else if (IN(BLOCKQUOTE_START))
1320                 {
1321                     // this kind of nesting not allowed (to avoid user confusion)
1322                     wiki_pop_excess_elements(parser);
1323                     wiki_start_para_if_necessary(parser);
1324                     output = parser->capture ? parser->capture : parser->output;
1325                     str_append(output, escaped_blockquote, TOKEN_LEN(token) + 3); // will either emit "&gt;" or "&gt; "
1326                     break;
1327                 }
1328                 else
1329                 {
1330                     ary_push(parser->line, BLOCKQUOTE);
1331
1332                     // count number of BLOCKQUOTE tokens in line buffer and in scope stack
1333                     i = ary_count(parser->line, BLOCKQUOTE);
1334                     j = ary_count(parser->scope, BLOCKQUOTE);
1335
1336                     // given that BLOCKQUOTE tokens can be nested, peek ahead and see if there are any more which might affect the decision to push or pop
1337                     while (NEXT_TOKEN(), (token->type == BLOCKQUOTE))
1338                     {
1339                         ary_push(parser->line, BLOCKQUOTE);
1340                         i++;
1341                     }
1342
1343                     // now decide whether to push, pop or do nothing
1344                     if (i > j)
1345                     {
1346                         // must push (increase nesting level)
1347                         wiki_pop_from_stack_up_to(parser, NULL, BLOCKQUOTE, false);
1348                         for (i = i - j; i > 0; i--)
1349                         {
1350                             wiki_indent(parser);
1351                             str_append(parser->output, blockquote_start, sizeof(blockquote_start) - 1);
1352                             str_append_str(parser->output, parser->line_ending);
1353                             ary_push(parser->scope, BLOCKQUOTE);
1354                         }
1355                     }
1356                     else if (i < j)
1357                     {
1358                         // must pop (reduce nesting level)
1359                         for (i = j - i; i > 0; i--)
1360                             wiki_pop_from_stack_up_to(parser, NULL, BLOCKQUOTE, true);
1361                     }
1362
1363                     // jump to top of the loop to process token we scanned during lookahead
1364                     continue;
1365                 }
1366                 break;
1367
1368             case BLOCKQUOTE_START:
1369                 if (IN_ANY_OF(NO_WIKI_START, PRE, PRE_START))
1370                 {
1371                     wiki_emit_pending_crlf_if_necessary(parser);
1372                     str_append(parser->output, escaped_blockquote_start, sizeof(escaped_blockquote_start) - 1);
1373                 }
1374                 else if (IN(BLOCKQUOTE_START))
1375                 {
1376                     // nesting is fine here
1377                     wiki_rollback_failed_link(parser); // if any
1378                     wiki_pop_from_stack_up_to(parser, NULL, BLOCKQUOTE_START, false);
1379                     wiki_indent(parser);
1380                     str_append(parser->output, blockquote_start, sizeof(blockquote_start) - 1);
1381                     str_append_str(parser->output, parser->line_ending);
1382                     ary_push(parser->scope, BLOCKQUOTE_START);
1383                     ary_push(parser->line, BLOCKQUOTE_START);
1384                 }
1385                 else if (IN(BLOCKQUOTE))
1386                 {
1387                     if (token->column_start == 1) // only allowed in first column
1388                     {
1389                         wiki_rollback_failed_link(parser); // if any
1390                         wiki_pop_all_from_stack(parser);
1391                         wiki_indent(parser);
1392                         str_append(parser->output, blockquote_start, sizeof(blockquote_start) - 1);
1393                         str_append_str(parser->output, parser->line_ending);
1394                         ary_push(parser->scope, BLOCKQUOTE_START);
1395                         ary_push(parser->line, BLOCKQUOTE_START);
1396                     }
1397                     else // BLOCKQUOTE_START illegal here
1398                     {
1399                         output = parser->capture ? parser->capture : parser->output;
1400                         wiki_pop_excess_elements(parser);
1401                         wiki_start_para_if_necessary(parser);
1402                         str_append(output, escaped_blockquote_start, sizeof(escaped_blockquote_start) - 1);
1403                     }
1404                 }
1405                 else
1406                 {
1407                     // would be nice to eliminate the repetition here but it's probably the clearest way
1408                     wiki_rollback_failed_link(parser); // if any
1409                     wiki_pop_from_stack_up_to(parser, NULL, P, true);
1410                     wiki_indent(parser);
1411                     str_append(parser->output, blockquote_start, sizeof(blockquote_start) - 1);
1412                     str_append_str(parser->output, parser->line_ending);
1413                     ary_push(parser->scope, BLOCKQUOTE_START);
1414                     ary_push(parser->line, BLOCKQUOTE_START);
1415                 }
1416                 break;
1417
1418             case BLOCKQUOTE_END:
1419                 if (IN_ANY_OF(NO_WIKI_START, PRE, PRE_START))
1420                 {
1421                     wiki_emit_pending_crlf_if_necessary(parser);
1422                     str_append(parser->output, escaped_blockquote_end, sizeof(escaped_blockquote_end) - 1);
1423                 }
1424                 else
1425                 {
1426                     if (IN(BLOCKQUOTE_START))
1427                         wiki_pop_from_stack_up_to(parser, parser->output, BLOCKQUOTE_START, true);
1428                     else
1429                     {
1430                         output = parser->capture ? parser->capture : parser->output;
1431                         wiki_pop_excess_elements(parser);
1432                         wiki_start_para_if_necessary(parser);
1433                         str_append(output, escaped_blockquote_end, sizeof(escaped_blockquote_end) - 1);
1434                     }
1435                 }
1436                 break;
1437
1438             case NO_WIKI_START:
1439                 if (IN_ANY_OF(NO_WIKI_START, PRE, PRE_START))
1440                 {
1441                     wiki_emit_pending_crlf_if_necessary(parser);
1442                     str_append(parser->output, escaped_no_wiki_start, sizeof(escaped_no_wiki_start) - 1);
1443                 }
1444                 else
1445                 {
1446                     wiki_pop_excess_elements(parser);
1447                     wiki_start_para_if_necessary(parser);
1448                     ary_push(parser->scope, NO_WIKI_START);
1449                     ary_push(parser->line, NO_WIKI_START);
1450                 }
1451                 break;
1452
1453             case NO_WIKI_END:
1454                 if (IN(NO_WIKI_START))
1455                     // <nowiki> should always only ever be the last item in the stack, but use the helper routine just in case
1456                     wiki_pop_from_stack_up_to(parser, NULL, NO_WIKI_START, true);
1457                 else
1458                 {
1459                     wiki_pop_excess_elements(parser);
1460                     wiki_start_para_if_necessary(parser);
1461                     str_append(parser->output, escaped_no_wiki_end, sizeof(escaped_no_wiki_end) - 1);
1462                 }
1463                 break;
1464
1465             case STRONG_EM:
1466                 if (IN_ANY_OF(NO_WIKI_START, PRE, PRE_START))
1467                 {
1468                     wiki_emit_pending_crlf_if_necessary(parser);
1469                     str_append(parser->output, literal_strong_em, sizeof(literal_strong_em) - 1);
1470                     break;
1471                 }
1472
1473                 output = parser->capture ? parser->capture : parser->output;
1474                 wiki_pop_excess_elements(parser);
1475
1476                 // if you've seen STRONG/STRONG_START or EM/EM_START, must close them in the reverse order that you saw them!
1477                 // otherwise, must open them
1478                 remove_strong  = -1;
1479                 remove_em      = -1;
1480                 j              = parser->scope->count;
1481                 for (j = j - 1; j >= 0; j--)
1482                 {
1483                     int val = ary_entry(parser->scope, j);
1484                     if (val == STRONG || val == STRONG_START)
1485                     {
1486                         str_append(output, strong_end, sizeof(strong_end) - 1);
1487                         remove_strong = j;
1488                     }
1489                     else if (val == EM || val == EM_START)
1490                     {
1491                         str_append(output, em_end, sizeof(em_end) - 1);
1492                         remove_em = j;
1493                     }
1494                 }
1495
1496                 if (remove_strong > remove_em)      // must remove strong first
1497                 {
1498                     ary_pop(parser->scope);
1499                     if (remove_em > -1)
1500                         ary_pop(parser->scope);
1501                     else    // there was no em to remove!, so consider this an opening em tag
1502                     {
1503                         str_append(output, em_start, sizeof(em_start) - 1);
1504                         ary_push(parser->scope, EM);
1505                         ary_push(parser->line, EM);
1506                     }
1507                 }
1508                 else if (remove_em > remove_strong) // must remove em first
1509                 {
1510                     ary_pop(parser->scope);
1511                     if (remove_strong > -1)
1512                         ary_pop(parser->scope);
1513                     else    // there was no strong to remove!, so consider this an opening strong tag
1514                     {
1515                         str_append(output, strong_start, sizeof(strong_start) - 1);
1516                         ary_push(parser->scope, STRONG);
1517                         ary_push(parser->line, STRONG);
1518                     }
1519                 }
1520                 else    // no strong or em to remove, so this must be a new opening of both
1521                 {
1522                     wiki_start_para_if_necessary(parser);
1523                     str_append(output, strong_em_start, sizeof(strong_em_start) - 1);
1524                     ary_push(parser->scope, STRONG);
1525                     ary_push(parser->line, STRONG);
1526                     ary_push(parser->scope, EM);
1527                     ary_push(parser->line, EM);
1528                 }
1529                 break;
1530
1531             case STRONG:
1532                 if (IN_ANY_OF(NO_WIKI_START, PRE, PRE_START))
1533                 {
1534                     wiki_emit_pending_crlf_if_necessary(parser);
1535                     str_append(parser->output, literal_strong, sizeof(literal_strong) - 1);
1536                 }
1537                 else
1538                 {
1539                     output = parser->capture ? parser->capture : parser->output;
1540                     if (IN(STRONG_START))
1541                         // already in span started with <strong>, no choice but to emit this literally
1542                         str_append(output, literal_strong, sizeof(literal_strong) - 1);
1543                     else if (IN(STRONG))
1544                         // STRONG already seen, this is a closing tag
1545                         wiki_pop_from_stack_up_to(parser, output, STRONG, true);
1546                     else
1547                     {
1548                         // this is a new opening
1549                         wiki_pop_excess_elements(parser);
1550                         wiki_start_para_if_necessary(parser);
1551                         str_append(output, strong_start, sizeof(strong_start) - 1);
1552                         ary_push(parser->scope, STRONG);
1553                         ary_push(parser->line, STRONG);
1554                     }
1555                 }
1556                 break;
1557
1558             case STRONG_START:
1559                 if (IN_ANY_OF(NO_WIKI_START, PRE, PRE_START))
1560                 {
1561                     wiki_emit_pending_crlf_if_necessary(parser);
1562                     str_append(parser->output, escaped_strong_start, sizeof(escaped_strong_start) - 1);
1563                 }
1564                 else
1565                 {
1566                     output = parser->capture ? parser->capture : parser->output;
1567                     if (IN_EITHER_OF(STRONG_START, STRONG))
1568                         str_append(output, escaped_strong_start, sizeof(escaped_strong_start) - 1);
1569                     else
1570                     {
1571                         wiki_pop_excess_elements(parser);
1572                         wiki_start_para_if_necessary(parser);
1573                         str_append(output, strong_start, sizeof(strong_start) - 1);
1574                         ary_push(parser->scope, STRONG_START);
1575                         ary_push(parser->line, STRONG_START);
1576                     }
1577                 }
1578                 break;
1579
1580             case STRONG_END:
1581                 if (IN_ANY_OF(NO_WIKI_START, PRE, PRE_START))
1582                 {
1583                     wiki_emit_pending_crlf_if_necessary(parser);
1584                     str_append(parser->output, escaped_strong_end, sizeof(escaped_strong_end) - 1);
1585                 }
1586                 else
1587                 {
1588                     output = parser->capture ? parser->capture : parser->output;
1589                     if (IN(STRONG_START))
1590                         wiki_pop_from_stack_up_to(parser, output, STRONG_START, true);
1591                     else
1592                     {
1593                         // no STRONG_START in scope, so must interpret the STRONG_END without any special meaning
1594                         wiki_pop_excess_elements(parser);
1595                         wiki_start_para_if_necessary(parser);
1596                         str_append(output, escaped_strong_end, sizeof(escaped_strong_end) - 1);
1597                     }
1598                 }
1599                 break;
1600
1601             case EM:
1602                 if (IN_ANY_OF(NO_WIKI_START, PRE, PRE_START))
1603                 {
1604                     wiki_emit_pending_crlf_if_necessary(parser);
1605                     str_append(parser->output, literal_em, sizeof(literal_em) - 1);
1606                 }
1607                 else
1608                 {
1609                     output = parser->capture ? parser->capture : parser->output;
1610                     if (IN(EM_START))
1611                         // already in span started with <em>, no choice but to emit this literally
1612                         str_append(output, literal_em, sizeof(literal_em) - 1);
1613                     else if (IN(EM))
1614                         // EM already seen, this is a closing tag
1615                         wiki_pop_from_stack_up_to(parser, output, EM, true);
1616                     else
1617                     {
1618                         // this is a new opening
1619                         wiki_pop_excess_elements(parser);
1620                         wiki_start_para_if_necessary(parser);
1621                         str_append(output, em_start, sizeof(em_start) - 1);
1622                         ary_push(parser->scope, EM);
1623                         ary_push(parser->line, EM);
1624                     }
1625                 }
1626                 break;
1627
1628             case EM_START:
1629                 if (IN_ANY_OF(NO_WIKI_START, PRE, PRE_START))
1630                 {
1631                     wiki_emit_pending_crlf_if_necessary(parser);
1632                     str_append(parser->output, escaped_em_start, sizeof(escaped_em_start) - 1);
1633                 }
1634                 else
1635                 {
1636                     output = parser->capture ? parser->capture : parser->output;
1637                     if (IN_EITHER_OF(EM_START, EM))
1638                         str_append(output, escaped_em_start, sizeof(escaped_em_start) - 1);
1639                     else
1640                     {
1641                         wiki_pop_excess_elements(parser);
1642                         wiki_start_para_if_necessary(parser);
1643                         str_append(output, em_start, sizeof(em_start) - 1);
1644                         ary_push(parser->scope, EM_START);
1645                         ary_push(parser->line, EM_START);
1646                     }
1647                 }
1648                 break;
1649
1650             case EM_END:
1651                 if (IN_ANY_OF(NO_WIKI_START, PRE, PRE_START))
1652                 {
1653                     wiki_emit_pending_crlf_if_necessary(parser);
1654                     str_append(parser->output, escaped_em_end, sizeof(escaped_em_end) - 1);
1655                 }
1656                 else
1657                 {
1658                     output = parser->capture ? parser->capture : parser->output;
1659                     if (IN(EM_START))
1660                         wiki_pop_from_stack_up_to(parser, output, EM_START, true);
1661                     else
1662                     {
1663                         // no EM_START in scope, so must interpret the EM_END without any special meaning
1664                         wiki_pop_excess_elements(parser);
1665                         wiki_start_para_if_necessary(parser);
1666                         str_append(output, escaped_em_end, sizeof(escaped_em_end) - 1);
1667                     }
1668                 }
1669                 break;
1670
1671             case TT:
1672                 if (IN_ANY_OF(NO_WIKI_START, PRE, PRE_START))
1673                 {
1674                     wiki_emit_pending_crlf_if_necessary(parser);
1675                     str_append(parser->output, backtick, sizeof(backtick) - 1);
1676                 }
1677                 else
1678                 {
1679                     output = parser->capture ? parser->capture : parser->output;
1680                     if (IN(TT_START))
1681                         // already in span started with <tt>, no choice but to emit this literally
1682                         str_append(output, backtick, sizeof(backtick) - 1);
1683                     else if (IN(TT))
1684                         // TT (`) already seen, this is a closing tag
1685                         wiki_pop_from_stack_up_to(parser, output, TT, true);
1686                     else
1687                     {
1688                         // this is a new opening
1689                         wiki_pop_excess_elements(parser);
1690                         wiki_start_para_if_necessary(parser);
1691                         str_append(output, code_start, sizeof(code_start) - 1);
1692                         ary_push(parser->scope, TT);
1693                         ary_push(parser->line, TT);
1694                     }
1695                 }
1696                 break;
1697
1698             case TT_START:
1699                 if (IN_ANY_OF(NO_WIKI_START, PRE, PRE_START))
1700                 {
1701                     wiki_emit_pending_crlf_if_necessary(parser);
1702                     str_append(parser->output, escaped_tt_start, sizeof(escaped_tt_start) - 1);
1703                 }
1704                 else
1705                 {
1706                     output = parser->capture ? parser->capture : parser->output;
1707                     if (IN_EITHER_OF(TT_START, TT))
1708                         str_append(output, escaped_tt_start, sizeof(escaped_tt_start) - 1);
1709                     else
1710                     {
1711                         wiki_pop_excess_elements(parser);
1712                         wiki_start_para_if_necessary(parser);
1713                         str_append(output, code_start, sizeof(code_start) - 1);
1714                         ary_push(parser->scope, TT_START);
1715                         ary_push(parser->line, TT_START);
1716                     }
1717                 }
1718                 break;
1719
1720             case TT_END:
1721                 if (IN_ANY_OF(NO_WIKI_START, PRE, PRE_START))
1722                 {
1723                     wiki_emit_pending_crlf_if_necessary(parser);
1724                     str_append(parser->output, escaped_tt_end, sizeof(escaped_tt_end) - 1);
1725                 }
1726                 else
1727                 {
1728                     output = parser->capture ? parser->capture : parser->output;
1729                     if (IN(TT_START))
1730                         wiki_pop_from_stack_up_to(parser, output, TT_START, true);
1731                     else
1732                     {
1733                         // no TT_START in scope, so must interpret the TT_END without any special meaning
1734                         wiki_pop_excess_elements(parser);
1735                         wiki_start_para_if_necessary(parser);
1736                         str_append(output, escaped_tt_end, sizeof(escaped_tt_end) - 1);
1737                     }
1738                 }
1739                 break;
1740
1741             case OL:
1742             case UL:
1743                 if (IN_EITHER_OF(NO_WIKI_START, PRE_START))
1744                 {
1745                     // no need to check for PRE; can never appear inside it
1746                     str_append(parser->output, token->start, TOKEN_LEN(token));
1747                     break;
1748                 }
1749
1750                 // count number of tokens in line and scope stacks
1751                 int bq_count = ary_count(parser->scope, BLOCKQUOTE_START);
1752                 i = parser->line->count - ary_count(parser->line, BLOCKQUOTE_START);
1753                 j = parser->scope->count - bq_count;
1754                 k = i;
1755
1756                 // list tokens can be nested so look ahead for any more which might affect the decision to push or pop
1757                 for (;;)
1758                 {
1759                     type = token->type;
1760                     if (type == OL || type == UL)
1761                     {
1762                         token = NULL;
1763                         if (i - k >= 2)                             // already seen at least one OL or UL
1764                         {
1765                             ary_push(parser->line, NESTED_LIST);    // which means this is a nested list
1766                             i += 3;
1767                         }
1768                         else
1769                             i += 2;
1770                         ary_push(parser->line, type);
1771                         ary_push(parser->line, LI);
1772
1773                         // want to compare line with scope but can only do so if scope has enough items on it
1774                         if (j >= i)
1775                         {
1776                             if (ary_entry(parser->scope, i + bq_count - 2) == type &&
1777                                 ary_entry(parser->scope, i + bq_count - 1) == LI)
1778                             {
1779                                 // line and scope match at this point: do nothing yet
1780                             }
1781                             else
1782                             {
1783                                 // item just pushed onto line does not match corresponding slot of scope!
1784                                 for (; j >= i - 2; j--)
1785                                     // must pop back before emitting
1786                                     wiki_pop_from_stack(parser, NULL);
1787
1788                                 // will emit UL or OL, then LI
1789                                 break;
1790                             }
1791                         }
1792                         else        // line stack size now exceeds scope stack size: must increase nesting level
1793                             break;  // will emit UL or OL, then LI
1794                     }
1795                     else
1796                     {
1797                         // not a OL or UL token!
1798                         if (j == i)
1799                             // must close existing LI and re-open new one
1800                             wiki_pop_from_stack(parser, NULL);
1801                         else if (j > i)
1802                         {
1803                             // item just pushed onto line does not match corresponding slot of scope!
1804                             for (; j >= i; j--)
1805                                 // must pop back before emitting
1806                                 wiki_pop_from_stack(parser, NULL);
1807                         }
1808                         break;
1809                     }
1810                     NEXT_TOKEN();
1811                 }
1812
1813                 // will emit
1814                 if (type == OL || type == UL)
1815                 {
1816                     // if LI is at the top of a stack this is the start of a nested list
1817                     if (j > 0 && ary_entry(parser->scope, -1) == LI)
1818                     {
1819                         // so we should precede it with a CRLF, and indicate that it's a nested list
1820                         str_append(parser->output, parser->line_ending->ptr, parser->line_ending->len);
1821                         ary_push(parser->scope, NESTED_LIST);
1822                     }
1823                     else
1824                     {
1825                         // this is a new list
1826                         if (IN(BLOCKQUOTE_START))
1827                             wiki_pop_from_stack_up_to(parser, NULL, BLOCKQUOTE_START, false);
1828                         else
1829                             wiki_pop_from_stack_up_to(parser, NULL, BLOCKQUOTE, false);
1830                     }
1831
1832                     // emit
1833                     wiki_indent(parser);
1834                     if (type == OL)
1835                         str_append(parser->output, ol_start, sizeof(ol_start) - 1);
1836                     else if (type == UL)
1837                         str_append(parser->output, ul_start, sizeof(ul_start) - 1);
1838                     ary_push(parser->scope, type);
1839                     str_append(parser->output, parser->line_ending->ptr, parser->line_ending->len);
1840                 }
1841                 else if (type == SPACE)
1842                     // silently throw away the optional SPACE token after final list marker
1843                     token = NULL;
1844
1845                 wiki_indent(parser);
1846                 str_append(parser->output, li_start, sizeof(li_start) - 1);
1847                 ary_push(parser->scope, LI);
1848
1849                 // any subsequent UL or OL tokens on this line are syntax errors and must be emitted literally
1850                 if (type == OL || type == UL)
1851                 {
1852                     k = 0;
1853                     while (k++, NEXT_TOKEN(), (type = token->type))
1854                     {
1855                         if (type == OL || type == UL)
1856                             str_append(parser->output, token->start, TOKEN_LEN(token));
1857                         else if (type == SPACE && k == 1)
1858                         {
1859                             // silently throw away the optional SPACE token after final list marker
1860                             token = NULL;
1861                             break;
1862                         }
1863                         else
1864                             break;
1865                     }
1866                 }
1867
1868                 // jump to top of the loop to process token we scanned during lookahead
1869                 continue;
1870
1871             case H6_START:
1872             case H5_START:
1873             case H4_START:
1874             case H3_START:
1875             case H2_START:
1876             case H1_START:
1877                 if (IN_EITHER_OF(NO_WIKI_START, PRE_START))
1878                 {
1879                     // no need to check for PRE; can never appear inside it
1880                     str_append(parser->output, token->start, TOKEN_LEN(token));
1881                     break;
1882                 }
1883
1884                 // pop up to but not including the last BLOCKQUOTE on the scope stack
1885                 if (IN(BLOCKQUOTE_START))
1886                     wiki_pop_from_stack_up_to(parser, NULL, BLOCKQUOTE_START, false);
1887                 else
1888                     wiki_pop_from_stack_up_to(parser, NULL, BLOCKQUOTE, false);
1889
1890                 // count number of BLOCKQUOTE tokens in line buffer and in scope stack
1891                 ary_push(parser->line, type);
1892                 i = ary_count(parser->line, BLOCKQUOTE);
1893                 j = ary_count(parser->scope, BLOCKQUOTE);
1894
1895                 // decide whether we need to pop off excess BLOCKQUOTE tokens (will never need to push; that is handled above in the BLOCKQUOTE case itself)
1896                 if (i < j)
1897                 {
1898                     // must pop (reduce nesting level)
1899                     for (i = j - i; i > 0; i--)
1900                         wiki_pop_from_stack_up_to(parser, NULL, BLOCKQUOTE, true);
1901                 }
1902
1903                 // discard any whitespace here (so that "== foo ==" will be translated to "<h2>foo</h2>" rather than "<h2> foo </h2")
1904                 while (NEXT_TOKEN(), (token->type == SPACE))
1905                     ; // discard
1906
1907                 ary_push(parser->scope, type);
1908                 wiki_indent(parser);
1909
1910                 // take base_heading_level into account
1911                 type += base_heading_level;
1912                 if (type > H6_START) // no need to check for underflow (base_heading_level never negative)
1913                     type = H6_START;
1914
1915                 // rather than repeat all that code for each kind of heading, share it and use a conditional here
1916                 if (type == H6_START)
1917                     str_append(parser->output, h6_start, sizeof(h6_start) - 1);
1918                 else if (type == H5_START)
1919                     str_append(parser->output, h5_start, sizeof(h5_start) - 1);
1920                 else if (type == H4_START)
1921                     str_append(parser->output, h4_start, sizeof(h4_start) - 1);
1922                 else if (type == H3_START)
1923                     str_append(parser->output, h3_start, sizeof(h3_start) - 1);
1924                 else if (type == H2_START)
1925                     str_append(parser->output, h2_start, sizeof(h2_start) - 1);
1926                 else if (type == H1_START)
1927                     str_append(parser->output, h1_start, sizeof(h1_start) - 1);
1928
1929                 // jump to top of the loop to process token we scanned during lookahead
1930                 continue;
1931
1932             case H6_END:
1933             case H5_END:
1934             case H4_END:
1935             case H3_END:
1936             case H2_END:
1937             case H1_END:
1938                 if (IN_ANY_OF(NO_WIKI_START, PRE, PRE_START))
1939                 {
1940                     wiki_emit_pending_crlf_if_necessary(parser);
1941                     str_append(parser->output, token->start, TOKEN_LEN(token));
1942                 }
1943                 else
1944                 {
1945                     wiki_rollback_failed_external_link(parser); // if any
1946                     if ((type == H6_END && !IN(H6_START)) ||
1947                         (type == H5_END && !IN(H5_START)) ||
1948                         (type == H4_END && !IN(H4_START)) ||
1949                         (type == H3_END && !IN(H3_START)) ||
1950                         (type == H2_END && !IN(H2_START)) ||
1951                         (type == H1_END && !IN(H1_START)))
1952                     {
1953                         // literal output only if not in appropriate scope (we stay silent in that case)
1954                         wiki_start_para_if_necessary(parser);
1955                         str_append(parser->output, token->start, TOKEN_LEN(token));
1956                     }
1957                 }
1958                 break;
1959
1960             case MAIL:
1961                 if (IN_ANY_OF(NO_WIKI_START, PRE, PRE_START))
1962                 {
1963                     wiki_emit_pending_crlf_if_necessary(parser);
1964                     str_append(parser->output, token->start, TOKEN_LEN(token));
1965                 }
1966                 else if (IN(EXT_LINK_START))
1967                     // must be capturing and this must be part of the link text
1968                     str_append(parser->capture, token->start, TOKEN_LEN(token));
1969                 else
1970                 {
1971                     wiki_pop_excess_elements(parser);
1972                     wiki_start_para_if_necessary(parser);
1973                     token_str->ptr = token->start;
1974                     token_str->len = TOKEN_LEN(token);
1975                     wiki_append_hyperlink(parser, rb_str_new2("mailto:"), token_str, NULL, mailto_class, Qnil, true);
1976                 }
1977                 break;
1978
1979             case URI:
1980                 if (IN(NO_WIKI_START))
1981                     // user can temporarily suppress autolinking by using <nowiki></nowiki>
1982                     // note that unlike MediaWiki, we do allow autolinking inside PRE blocks
1983                     str_append(parser->output, token->start, TOKEN_LEN(token));
1984                 else if (IN(LINK_START))
1985                 {
1986                     // if the URI were allowed it would have been handled already in LINK_START
1987                     wiki_rollback_failed_internal_link(parser);
1988                     token_str->ptr = token->start;
1989                     token_str->len = TOKEN_LEN(token);
1990                     wiki_append_hyperlink(parser, Qnil, token_str, NULL, parser->external_link_class, parser->external_link_rel, true);
1991                 }
1992                 else if (IN(EXT_LINK_START))
1993                 {
1994                     if (parser->link_target->len == 0)
1995                     {
1996                         // this must be our link target: look ahead to make sure we see the space we're expecting to see
1997                         token_str->ptr = token->start;
1998                         token_str->len = TOKEN_LEN(token);
1999                         NEXT_TOKEN();
2000                         if (token->type == SPACE)
2001                         {
2002                             ary_push(parser->scope, SPACE);
2003                             str_append_str(parser->link_target, token_str);
2004                             str_clear(parser->link_text);
2005                             parser->capture     = parser->link_text;
2006                             token               = NULL; // silently consume space
2007                         }
2008                         else
2009                         {
2010                             // didn't see the space! this must be an error
2011                             wiki_pop_from_stack(parser, NULL);
2012                             wiki_pop_excess_elements(parser);
2013                             wiki_start_para_if_necessary(parser);
2014                             str_append(parser->output, ext_link_start, sizeof(ext_link_start) - 1);
2015                             wiki_append_hyperlink(parser, Qnil, token_str, NULL, parser->external_link_class, parser->external_link_rel, true);
2016                             continue;
2017                         }
2018                     }
2019                     else
2020                         str_append(parser->link_text, token->start, TOKEN_LEN(token));
2021                 }
2022                 else
2023                 {
2024                     wiki_pop_excess_elements(parser);
2025                     wiki_start_para_if_necessary(parser);
2026                     token_str->ptr = token->start;
2027                     token_str->len = TOKEN_LEN(token);
2028                     wiki_append_hyperlink(parser, Qnil, token_str, NULL, parser->external_link_class, parser->external_link_rel, true);
2029                 }
2030                 break;
2031
2032             case PATH:
2033                 if (IN_ANY_OF(NO_WIKI_START, PRE, PRE_START))
2034                 {
2035                     wiki_emit_pending_crlf_if_necessary(parser);
2036                     str_append(parser->output, token->start, TOKEN_LEN(token));
2037                 }
2038                 else if (IN(EXT_LINK_START))
2039                 {
2040                     if (parser->link_target->len == 0)
2041                     {
2042                         // this must be our link target: look ahead to make sure we see the space we're expecting to see
2043                         token_str->ptr = token->start;
2044                         token_str->len = TOKEN_LEN(token);
2045                         NEXT_TOKEN();
2046                         if (token->type == SPACE)
2047                         {
2048                             ary_push(parser->scope, PATH);
2049                             ary_push(parser->scope, SPACE);
2050                             str_append_str(parser->link_target, token_str);
2051                             str_clear(parser->link_text);
2052                             parser->capture     = parser->link_text;
2053                             token               = NULL; // silently consume space
2054                         }
2055                         else
2056                         {
2057                             // didn't see the space! this must be an error
2058                             wiki_pop_from_stack(parser, NULL);
2059                             wiki_pop_excess_elements(parser);
2060                             wiki_start_para_if_necessary(parser);
2061                             str_append(parser->output, ext_link_start, sizeof(ext_link_start) - 1);
2062                             str_append_str(parser->output, token_str);
2063                             continue;
2064                         }
2065                     }
2066                     else
2067                         str_append(parser->link_text, token->start, TOKEN_LEN(token));
2068                 }
2069                 else
2070                 {
2071                     output = parser->capture ? parser->capture : parser->output;
2072                     wiki_pop_excess_elements(parser);
2073                     wiki_start_para_if_necessary(parser);
2074                     str_append(output, token->start, TOKEN_LEN(token));
2075                 }
2076                 break;
2077
2078             // internal links (links to other wiki articles) look like this:
2079             //      [[another article]] (would point at, for example, "/wiki/another_article")
2080             //      [[the other article|the link text we'll use for it]]
2081             //      [[the other article | the link text we'll use for it]]
2082             // MediaWiki has strict requirements about what it will accept as a link target:
2083             //      all wikitext markup is disallowed:
2084             //          example [[foo ''bar'' baz]]
2085             //          renders [[foo <em>bar</em> baz]]        (ie. not a link)
2086             //          example [[foo <em>bar</em> baz]]
2087             //          renders [[foo <em>bar</em> baz]]        (ie. not a link)
2088             //          example [[foo <nowiki>''</nowiki> baz]]
2089             //          renders [[foo '' baz]]                  (ie. not a link)
2090             //          example [[foo <bar> baz]]
2091             //          renders [[foo &lt;bar&gt; baz]]         (ie. not a link)
2092             //      HTML entities and non-ASCII, however, make it through:
2093             //          example [[foo &euro;]]
2094             //          renders <a href="/wiki/Foo_%E2%82%AC">foo &euro;</a>
2095             //          example [[foo €]]
2096             //          renders <a href="/wiki/Foo_%E2%82%AC">foo €</a>
2097             // we'll impose similar restrictions here for the link target; allowed tokens will be:
2098             //      SPACE, SPECIAL_URI_CHARS, PRINTABLE, PATH, ALNUM, DEFAULT, QUOT and AMP
2099             // everything else will be rejected
2100             case LINK_START:
2101                 output = parser->capture ? parser->capture : parser->output;
2102                 if (IN_ANY_OF(NO_WIKI_START, PRE, PRE_START))
2103                 {
2104                     wiki_emit_pending_crlf_if_necessary(parser);
2105                     str_append(output, link_start, sizeof(link_start) - 1);
2106                 }
2107                 else if (IN(EXT_LINK_START))
2108                     // already in external link scope! (and in fact, must be capturing link_text right now)
2109                     str_append(output, link_start, sizeof(link_start) - 1);
2110                 else if (IN(LINK_START))
2111                 {
2112                     // already in internal link scope! this is a syntax error
2113                     wiki_rollback_failed_internal_link(parser);
2114                     str_append(parser->output, link_start, sizeof(link_start) - 1);
2115                 }
2116                 else if (IN(SEPARATOR))
2117                 {
2118                     // scanning internal link text
2119                 }
2120                 else // not in internal link scope yet
2121                 {
2122                     // will either emit a link, or the rollback of a failed link, so start the para now
2123                     wiki_pop_excess_elements(parser);
2124                     wiki_start_para_if_necessary(parser);
2125                     ary_push(parser->scope, LINK_START);
2126
2127                     // look ahead and try to gobble up link target
2128                     while (NEXT_TOKEN(), (type = token->type))
2129                     {
2130                         if (type == SPACE               ||
2131                             type == SPECIAL_URI_CHARS   ||
2132                             type == PATH                ||
2133                             type == PRINTABLE           ||
2134                             type == ALNUM               ||
2135                             type == DEFAULT             ||
2136                             type == QUOT                ||
2137                             type == QUOT_ENTITY         ||
2138                             type == AMP                 ||
2139                             type == AMP_ENTITY          ||
2140                             type == IMG_START           ||
2141                             type == IMG_END             ||
2142                             type == LEFT_CURLY          ||
2143                             type == RIGHT_CURLY)
2144                         {
2145                             // accumulate these tokens into link_target
2146                             if (parser->link_target->len == 0)
2147                             {
2148                                 str_clear(parser->link_target);
2149                                 parser->capture = parser->link_target;
2150                             }
2151                             if (type == QUOT_ENTITY)
2152                                 // don't insert the entity, insert the literal quote
2153                                 str_append(parser->link_target, quote, sizeof(quote) - 1);
2154                             else if (type == AMP_ENTITY)
2155                                 // don't insert the entity, insert the literal ampersand
2156                                 str_append(parser->link_target, ampersand, sizeof(ampersand) - 1);
2157                             else
2158                                 str_append(parser->link_target, token->start, TOKEN_LEN(token));
2159                         }
2160                         else if (type == LINK_END)
2161                         {
2162                             if (parser->link_target->len == 0) // bail for inputs like "[[]]"
2163                                 wiki_rollback_failed_internal_link(parser);
2164                             break; // jump back to top of loop (will handle this in LINK_END case below)
2165                         }
2166                         else if (type == SEPARATOR)
2167                         {
2168                             if (parser->link_target->len == 0) // bail for inputs like "[[|"
2169                                 wiki_rollback_failed_internal_link(parser);
2170                             else
2171                             {
2172                                 ary_push(parser->scope, SEPARATOR);
2173                                 str_clear(parser->link_text);
2174                                 parser->capture     = parser->link_text;
2175                                 token               = NULL;
2176                             }
2177                             break;
2178                         }
2179                         else // unexpected token (syntax error)
2180                         {
2181                             wiki_rollback_failed_internal_link(parser);
2182                             break; // jump back to top of loop to handle unexpected token
2183                         }
2184                     }
2185
2186                     // jump to top of the loop to process token we scanned during lookahead (if any)
2187                     continue;
2188                 }
2189                 break;
2190
2191             case LINK_END:
2192                 output = parser->capture ? parser->capture : parser->output;
2193                 if (IN_ANY_OF(NO_WIKI_START, PRE, PRE_START))
2194                 {
2195                     wiki_emit_pending_crlf_if_necessary(parser);
2196                     str_append(output, link_end, sizeof(link_end) - 1);
2197                 }
2198                 else if (IN(EXT_LINK_START))
2199                     // already in external link scope! (and in fact, must be capturing link_text right now)
2200                     str_append(output, link_end, sizeof(link_end) - 1);
2201                 else if (IN(LINK_START)) // in internal link scope!
2202                 {
2203                     if (wiki_blank(parser->link_target))
2204                     {
2205                         // special case for inputs like "[[    ]]"
2206                         wiki_rollback_failed_internal_link(parser);
2207                         str_append(parser->output, link_end, sizeof(link_end) - 1);
2208                         break;
2209                     }
2210                     if (parser->link_text->len == 0 ||
2211                         wiki_blank(parser->link_text))
2212                     {
2213                         // use link target as link text
2214                         str_clear(parser->link_text);
2215                         wiki_append_sanitized_link_target(parser->link_target, parser->link_text, true);
2216                     }
2217                     else
2218                         wiki_trim_link_text(parser);
2219
2220                     // perform "redlink" check before manipulating link_target
2221                     if (NIL_P(link_proc))
2222                         j = Qnil;
2223                     else
2224                     {
2225                         j = rb_funcall(link_proc, rb_intern("call"), 1, string_from_str(parser->link_target));
2226                         if (!NIL_P(j))
2227                         {
2228                             VALUE l = j; // can't cast inside StringValue macro
2229                             j = StringValue(l);
2230                         }
2231                     }
2232                     wiki_encode_link_target(parser);
2233                     wiki_pop_from_stack_up_to(parser, output, LINK_START, true);
2234                     parser->capture = NULL;
2235                     wiki_append_hyperlink(parser, prefix, parser->link_target, parser->link_text, j, Qnil, false);
2236                     str_clear(parser->link_target);
2237                     str_clear(parser->link_text);
2238                 }
2239                 else // wasn't in internal link scope
2240                 {
2241                     wiki_pop_excess_elements(parser);
2242                     wiki_start_para_if_necessary(parser);
2243                     str_append(output, link_end, sizeof(link_end) - 1);
2244                 }
2245                 break;
2246
2247             // external links look like this:
2248             //      [http://google.com/ the link text]
2249             //      [/other/page/on/site see this page]
2250             // strings in square brackets which don't match this syntax get passed through literally; eg:
2251             //      he was very angery [sic] about the turn of events
2252             case EXT_LINK_START:
2253                 output = parser->capture ? parser->capture : parser->output;
2254                 if (IN_ANY_OF(NO_WIKI_START, PRE, PRE_START))
2255                 {
2256                     wiki_emit_pending_crlf_if_necessary(parser);
2257                     str_append(output, ext_link_start, sizeof(ext_link_start) - 1);
2258                 }
2259                 else if (IN(EXT_LINK_START))
2260                     // already in external link scope! (and in fact, must be capturing link_text right now)
2261                     str_append(output, ext_link_start, sizeof(ext_link_start) - 1);
2262                 else if (IN(LINK_START))
2263                 {
2264                     // already in internal link scope!
2265                     if (parser->link_target->len == 0 || !IN(SPACE))
2266                         str_append(parser->link_target, ext_link_start, sizeof(ext_link_start) - 1);
2267                     else // link target has already been scanned
2268                         str_append(parser->link_text, ext_link_start, sizeof(ext_link_start) - 1);
2269                 }
2270                 else // not in external link scope yet
2271                 {
2272                     // will either emit a link, or the rollback of a failed link, so start the para now
2273                     wiki_pop_excess_elements(parser);
2274                     wiki_start_para_if_necessary(parser);
2275
2276                     // look ahead: expect an absolute URI (with protocol) or "relative" (path) URI
2277                     NEXT_TOKEN();
2278                     if (token->type == URI || token->type == PATH)
2279                         ary_push(parser->scope, EXT_LINK_START);    // so far so good, jump back to the top of the loop
2280                     else
2281                         // only get here if there was a syntax error (missing URI)
2282                         str_append(parser->output, ext_link_start, sizeof(ext_link_start) - 1);
2283                     continue; // jump back to top of loop to handle token (either URI or whatever it is)
2284                 }
2285                 break;
2286
2287             case EXT_LINK_END:
2288                 output = parser->capture ? parser->capture : parser->output;
2289                 if (IN_ANY_OF(NO_WIKI_START, PRE, PRE_START))
2290                 {
2291                     wiki_emit_pending_crlf_if_necessary(parser);
2292                     str_append(output, ext_link_end, sizeof(ext_link_end) - 1);
2293                 }
2294                 else if (IN(EXT_LINK_START))
2295                 {
2296                     if (parser->link_text->len == 0)
2297                         // syntax error: external link with no link text
2298                         wiki_rollback_failed_external_link(parser);
2299                     else
2300                     {
2301                         // success!
2302                         j = IN(PATH) ? Qnil : parser->external_link_class;
2303                         k = IN(PATH) ? Qnil : parser->external_link_rel;
2304                         wiki_pop_from_stack_up_to(parser, output, EXT_LINK_START, true);
2305                         parser->capture = NULL;
2306                         wiki_append_hyperlink(parser, Qnil, parser->link_target, parser->link_text, j, k, false);
2307                     }
2308                     str_clear(parser->link_target);
2309                     str_clear(parser->link_text);
2310                 }
2311                 else
2312                 {
2313                     wiki_pop_excess_elements(parser);
2314                     wiki_start_para_if_necessary(parser);
2315                     str_append(parser->output, ext_link_end, sizeof(ext_link_end) - 1);
2316                 }
2317                 break;
2318
2319             case SEPARATOR:
2320                 output = parser->capture ? parser->capture : parser->output;
2321                 wiki_pop_excess_elements(parser);
2322                 wiki_start_para_if_necessary(parser);
2323                 str_append(output, separator, sizeof(separator) - 1);
2324                 break;
2325
2326             case SPACE:
2327                 output = parser->capture ? parser->capture : parser->output;
2328                 if (IN_ANY_OF(NO_WIKI_START, PRE, PRE_START))
2329                 {
2330                     wiki_emit_pending_crlf_if_necessary(parser);
2331                     str_append(output, token->start, TOKEN_LEN(token));
2332                 }
2333                 else
2334                 {
2335                     // peek ahead to see next token
2336                     char    *token_ptr  = token->start;
2337                     int     token_len   = TOKEN_LEN(token);
2338                     NEXT_TOKEN();
2339                     type = token->type;
2340                     if ((type == H6_END && IN(H6_START)) ||
2341                         (type == H5_END && IN(H5_START)) ||
2342                         (type == H4_END && IN(H4_START)) ||
2343                         (type == H3_END && IN(H3_START)) ||
2344                         (type == H2_END && IN(H2_START)) ||
2345                         (type == H1_END && IN(H1_START)))
2346                     {
2347                         // will suppress emission of space (discard) if next token is a H6_END, H5_END etc and we are in the corresponding scope
2348                     }
2349                     else
2350                     {
2351                         // emit the space
2352                         wiki_pop_excess_elements(parser);
2353                         wiki_start_para_if_necessary(parser);
2354                         str_append(output, token_ptr, token_len);
2355                     }
2356
2357                     // jump to top of the loop to process token we scanned during lookahead
2358                     continue;
2359                 }
2360                 break;
2361
2362             case QUOT_ENTITY:
2363             case AMP_ENTITY:
2364             case NAMED_ENTITY:
2365             case DECIMAL_ENTITY:
2366                 // pass these through unaltered as they are case sensitive
2367                 output = parser->capture ? parser->capture : parser->output;
2368                 wiki_pop_excess_elements(parser);
2369                 wiki_start_para_if_necessary(parser);
2370                 str_append(output, token->start, TOKEN_LEN(token));
2371                 break;
2372
2373             case HEX_ENTITY:
2374                 // normalize hex entities (downcase them)
2375                 output = parser->capture ? parser->capture : parser->output;
2376                 wiki_pop_excess_elements(parser);
2377                 wiki_start_para_if_necessary(parser);
2378                 str_append(output, token->start, TOKEN_LEN(token));
2379                 wiki_downcase_bang(output->ptr + output->len - TOKEN_LEN(token), TOKEN_LEN(token));
2380                 break;
2381
2382             case QUOT:
2383                 output = parser->capture ? parser->capture : parser->output;
2384                 wiki_pop_excess_elements(parser);
2385                 wiki_start_para_if_necessary(parser);
2386                 str_append(output, quot_entity, sizeof(quot_entity) - 1);
2387                 break;
2388
2389             case AMP:
2390                 output = parser->capture ? parser->capture : parser->output;
2391                 wiki_pop_excess_elements(parser);
2392                 wiki_start_para_if_necessary(parser);
2393                 str_append(output, amp_entity, sizeof(amp_entity) - 1);
2394                 break;
2395
2396             case LESS:
2397                 output = parser->capture ? parser->capture : parser->output;
2398                 wiki_pop_excess_elements(parser);
2399                 wiki_start_para_if_necessary(parser);
2400                 str_append(output, lt_entity, sizeof(lt_entity) - 1);
2401                 break;
2402
2403             case GREATER:
2404                 output = parser->capture ? parser->capture : parser->output;
2405                 wiki_pop_excess_elements(parser);
2406                 wiki_start_para_if_necessary(parser);
2407                 str_append(output, gt_entity, sizeof(gt_entity) - 1);
2408                 break;
2409
2410             case IMG_START:
2411                 if (IN_ANY_OF(NO_WIKI_START, PRE, PRE_START))
2412                 {
2413                     wiki_emit_pending_crlf_if_necessary(parser);
2414                     str_append(parser->output, token->start, TOKEN_LEN(token));
2415                 }
2416                 else if (parser->capture)
2417                     str_append(parser->capture, token->start, TOKEN_LEN(token));
2418                 else
2419                 {
2420                     // not currently capturing: will be emitting something on success or failure, so get ready
2421                     wiki_pop_excess_elements(parser);
2422                     wiki_start_para_if_necessary(parser);
2423
2424                     // scan ahead consuming PATH, PRINTABLE, ALNUM and SPECIAL_URI_CHARS tokens
2425                     // will cheat here and abuse the link_target capture buffer to accumulate text
2426                     while (NEXT_TOKEN(), (type = token->type))
2427                     {
2428                         if (type == PATH || type == PRINTABLE || type == ALNUM || type == SPECIAL_URI_CHARS)
2429                             str_append(parser->link_target, token->start, TOKEN_LEN(token));
2430                         else if (type == IMG_END && parser->link_target->len > 0)
2431                         {
2432                             // success
2433                             wiki_append_img(parser, parser->link_target->ptr, parser->link_target->len);
2434                             token = NULL;
2435                             break;
2436                         }
2437                         else // unexpected token or zero-length target (syntax error)
2438                         {
2439                             // rollback
2440                             str_append(parser->output, literal_img_start, sizeof(literal_img_start) - 1);
2441                             if (parser->link_target->len > 0)
2442                                 str_append(parser->output, parser->link_target->ptr, parser->link_target->len);
2443                             break;
2444                         }
2445                     }
2446
2447                     // jump to top of the loop to process token we scanned during lookahead
2448                     str_clear(parser->link_target);
2449                     continue;
2450                 }
2451                 break;
2452
2453             case CRLF:
2454                 i = parser->pending_crlf;
2455                 parser->pending_crlf = false;
2456                 wiki_rollback_failed_link(parser); // if any
2457                 if (IN_EITHER_OF(NO_WIKI_START, PRE_START))
2458                 {
2459                     ary_clear(parser->line_buffer);
2460                     str_append_str(parser->output, parser->line_ending);
2461                     break;
2462                 }
2463                 else if (IN(PRE))
2464                 {
2465                     // beware when BLOCKQUOTE on line buffer (not line stack!) prior to CRLF, that must be end of PRE block
2466                     if (ary_entry(parser->line_buffer, -2) == BLOCKQUOTE)
2467                         // don't emit in this case
2468                         wiki_pop_from_stack_up_to(parser, parser->output, PRE, true);
2469                     else
2470                     {
2471                         if (ary_entry(parser->line_buffer, -2) == PRE)
2472                         {
2473                              // only thing on line is the PRE: emit pending line ending (if we had one)
2474                              if (i)
2475                                  str_append_str(parser->output, parser->line_ending);
2476                         }
2477
2478                         // clear these _before_ calling NEXT_TOKEN (NEXT_TOKEN adds to the line_buffer)
2479                         ary_clear(parser->line);
2480                         ary_clear(parser->line_buffer);
2481
2482                         // peek ahead to see if this is definitely the end of the PRE block
2483                         NEXT_TOKEN();
2484                         type = token->type;
2485                         if (type != BLOCKQUOTE && type != PRE)
2486                             // this is definitely the end of the block, so don't emit
2487                             wiki_pop_from_stack_up_to(parser, parser->output, PRE, true);
2488                         else
2489                             // potentially will emit
2490                             parser->pending_crlf = true;
2491
2492                         continue; // jump back to top of loop to handle token grabbed via lookahead
2493                     }
2494                 }
2495                 else
2496                 {
2497                     parser->pending_crlf = true;
2498
2499                     // count number of BLOCKQUOTE tokens in line buffer (can be zero) and pop back to that level
2500                     // as a side effect, this handles any open span-level elements and unclosed blocks
2501                     // (with special handling for P blocks and LI elements)
2502                     i = ary_count(parser->line, BLOCKQUOTE) + ary_count(parser->scope, BLOCKQUOTE_START);
2503                     for (j = parser->scope->count; j > i; j--)
2504                     {
2505                         if (parser->scope->count > 0 && ary_entry(parser->scope, -1) == LI)
2506                         {
2507                             parser->pending_crlf = false;
2508                             break;
2509                         }
2510
2511                         // special handling on last iteration through the loop if the top item on the scope is a P block
2512                         if ((j - i == 1) && ary_entry(parser->scope, -1) == P)
2513                         {
2514                             // if nothing or BLOCKQUOTE on line buffer (not line stack!) prior to CRLF, this must be a paragraph break
2515                             // (note that we have to make sure we're not inside a BLOCKQUOTE_START block
2516                             // because in those blocks BLOCKQUOTE tokens have no special meaning)
2517                             if (NO_ITEM(ary_entry(parser->line_buffer, -2)) ||
2518                                 (ary_entry(parser->line_buffer, -2) == BLOCKQUOTE && !IN(BLOCKQUOTE_START)))
2519                                 // paragraph break
2520                                 parser->pending_crlf = false;
2521                             else
2522                                 // not a paragraph break!
2523                                 continue;
2524                         }
2525                         wiki_pop_from_stack(parser, NULL);
2526                     }
2527                 }
2528
2529                 // delete the entire contents of the line scope stack and buffer
2530                 ary_clear(parser->line);
2531                 ary_clear(parser->line_buffer);
2532                 break;
2533
2534             case SPECIAL_URI_CHARS:
2535             case PRINTABLE:
2536             case ALNUM:
2537             case IMG_END:
2538             case LEFT_CURLY:
2539             case RIGHT_CURLY:
2540                 output = parser->capture ? parser->capture : parser->output;
2541                 wiki_pop_excess_elements(parser);
2542                 wiki_start_para_if_necessary(parser);
2543                 str_append(output, token->start, TOKEN_LEN(token));
2544                 break;
2545
2546             case DEFAULT:
2547                 output = parser->capture ? parser->capture : parser->output;
2548                 wiki_pop_excess_elements(parser);
2549                 wiki_start_para_if_necessary(parser);
2550                 wiki_append_entity_from_utf32_char(output, token->code_point);
2551                 break;
2552
2553             case END_OF_FILE:
2554                 // special case for input like " foo\n " (see pre_spec.rb)
2555                 if (IN(PRE) &&
2556                     ary_entry(parser->line_buffer, -2) == PRE &&
2557                     parser->pending_crlf)
2558                     str_append(parser->output, parser->line_ending->ptr, parser->line_ending->len);
2559
2560                 // close any open scopes on hitting EOF
2561                 wiki_rollback_failed_link(parser); // if any
2562                 wiki_pop_all_from_stack(parser);
2563                 goto return_output; // break not enough here (want to break out of outer while loop, not inner switch statement)
2564
2565             default:
2566                 break;
2567         }
2568
2569         // reset current token; forcing lexer to return another token at the top of the loop
2570         token = NULL;
2571     } while (1);
2572 return_output:
2573     // nasty hack to avoid re-allocating our return value
2574     str_append(parser->output, null_str, 1); // null-terminate
2575     len = parser->output->len - 1; // don't count null termination
2576
2577 #if defined(RUBY_1_9_x)
2578     VALUE out = rb_str_buf_new(RSTRING_EMBED_LEN_MAX + 1);
2579     free(RSTRING_PTR(out));
2580     RSTRING(out)->as.heap.aux.capa = len;
2581     RSTRING(out)->as.heap.ptr = parser->output->ptr;
2582     RSTRING(out)->as.heap.len = len;
2583 #elif defined(RUBY_1_8_x)
2584     VALUE out = rb_str_new2("");
2585     free(RSTRING_PTR(out));
2586     RSTRING(out)->len = len;
2587     RSTRING(out)->aux.capa = len;
2588     RSTRING(out)->ptr = parser->output->ptr;
2589 #else
2590 #error unsupported RUBY_VERSION
2591 #endif
2592     parser->output->ptr = NULL; // don't double-free
2593     return out;
2594 }