escape html reserved characters in text

1 files changed, 8 insertions, 7 deletions

diff --git a/fsm.py b/fsm.py
index b7439d1..eccfc6e 100644
--- a/fsm.py
+++ b/fsm.py
@@ -1,3 +1,4 @@
+import html
 import sys
 import urllib.parse
 
@@ -62,18 +63,18 @@ class Parser(object):
             self._output.write('<ul>\n')
         else:
             if line.startswith('# '):
-                self._output.write('<h1>{}</h1>\n'.format(line[2:]))
+                self._output.write('<h1>{}</h1>\n'.format(html.escape(line[2:])))
             elif line.startswith('## '):
-                self._output.write('<h2>{}</h2>\n'.format(line[3:]))
+                self._output.write('<h2>{}</h2>\n'.format(html.escape(line[3:])))
             elif line.startswith('### '):
-                self._output.write('<h3>{}</h3>\n'.format(line[4:]))
+                self._output.write('<h3>{}</h3>\n'.format(html.escape(line[4:])))
             elif line.startswith('> '):
-                self._output.write('<blockquote>{}</blockquote>\n'.format(line[2:]))
+                self._output.write('<blockquote>{}</blockquote>\n'.format(html.escape(line[2:])))
             elif line.strip() == '':
                 if self._blanks > 1:
                     self._output.write('<br/>\n')
             else:
-                self._output.write('<p>{}</p>\n'.format(line))
+                self._output.write('<p>{}</p>\n'.format(html.escape(line)))
             self._offset += 1
 
     def pre_state(self):
@@ -89,7 +90,7 @@ class Parser(object):
     def list_state(self):
         line = self._document[self._offset]
         if line.startswith('* '):
-            self._output.write('<li>{}</li>\n'.format(line[2:]))
+            self._output.write('<li>{}</li>\n'.format(html.escape(line[2:])))
             self._offset += 1
         else:
             self._fsm.pop_state()
@@ -108,7 +109,7 @@ class Parser(object):
             if len(parts) == 1:
                 text = url
             else:
-                text = parts[1]
+                text = html.escape(parts[1])
             self._output.write('<li class="link"><a href="{}">{}</a>{}</li>\n'.format(url, text, external))
             self._offset += 1
         else: