blackfriday/v2/esc.go

package blackfriday

import (
        "html"
        "io"
)

var htmlEscaper = [256][]byte{
        '&': []byte("&amp;"),
        '<': []byte("&lt;"),
        '>': []byte("&gt;"),
        '"': []byte("&quot;"),
}

func escapeHTML(w io.Writer, s []byte) {
        escapeEntities(w, s, false)
}

func escapeAllHTML(w io.Writer, s []byte) {
        escapeEntities(w, s, true)
}

func escapeEntities(w io.Writer, s []byte, escapeValidEntities bool) {
        var start, end int
        for end < len(s) {
                escSeq := htmlEscaper[s[end]]
                if escSeq != nil {
                        isEntity, entityEnd := nodeIsEntity(s, end)
                        if isEntity && !escapeValidEntities {
                                w.Write(s[start : entityEnd+1])
                                start = entityEnd + 1
                        } else {
                                w.Write(s[start:end])
                                w.Write(escSeq)
                                start = end + 1
                        }
                }
                end++
        }
        if start < len(s) && end <= len(s) {
                w.Write(s[start:end])
        }
}

func nodeIsEntity(s []byte, end int) (isEntity bool, endEntityPos int) {
        isEntity = false
        endEntityPos = end + 1

        if s[end] == '&' {
                for endEntityPos < len(s) {
                        if s[endEntityPos] == ';' {
                                if entities[string(s[end:endEntityPos+1])] {
                                        isEntity = true
                                        break
                                }
                        }
                        if !isalnum(s[endEntityPos]) && s[endEntityPos] != '&' && s[endEntityPos] != '#' {
                                break
                        }
                        endEntityPos++
                }
        }

        return isEntity, endEntityPos
}

func escLink(w io.Writer, text []byte) {
        unesc := html.UnescapeString(string(text))
        escapeHTML(w, []byte(unesc))
}
1	package blackfriday
2
3	import (
4	"html"
5	"io"
6	)
7
8	var htmlEscaper = [256][]byte{
9	'&': []byte("&"),
10	'<': []byte("<"),
11	'>': []byte(">"),
12	'"': []byte("""),
13	}
14
15	func escapeHTML(w io.Writer, s []byte) {
16	escapeEntities(w, s, false)
17	}
18
19	func escapeAllHTML(w io.Writer, s []byte) {
20	escapeEntities(w, s, true)
21	}
22
23	func escapeEntities(w io.Writer, s []byte, escapeValidEntities bool) {
24	var start, end int
25	for end < len(s) {
26	escSeq := htmlEscaper[s[end]]
27	if escSeq != nil {
28	isEntity, entityEnd := nodeIsEntity(s, end)
29	if isEntity && !escapeValidEntities {
30	w.Write(s[start : entityEnd+1])
31	start = entityEnd + 1
32	} else {
33	w.Write(s[start:end])
34	w.Write(escSeq)
35	start = end + 1
36	}
37	}
38	end++
39	}
40	if start < len(s) && end <= len(s) {
41	w.Write(s[start:end])
42	}
43	}
44
45	func nodeIsEntity(s []byte, end int) (isEntity bool, endEntityPos int) {
46	isEntity = false
47	endEntityPos = end + 1
48
49	if s[end] == '&' {
50	for endEntityPos < len(s) {
51	if s[endEntityPos] == ';' {
52	if entities[string(s[end:endEntityPos+1])] {
53	isEntity = true
54	break
55	}
56	}
57	if !isalnum(s[endEntityPos]) && s[endEntityPos] != '&' && s[endEntityPos] != '#' {
58	break
59	}
60	endEntityPos++
61	}
62	}
63
64	return isEntity, endEntityPos
65	}
66
67	func escLink(w io.Writer, text []byte) {
68	unesc := html.UnescapeString(string(text))
69	escapeHTML(w, []byte(unesc))
70	}
nishi@chaotic.ninja	ViewVC Help
Powered by ViewVC 1.3.0-dev