Fix document toc. (#2510)

Co-authored-by: Yong He <yhe@nvidia.com>

1 files changed, 21 insertions, 1 deletions

diff --git a/docs/scripts/Program.cs b/docs/scripts/Program.cs
index 8a7b6c1b6..d7f4ef623 100644
--- a/docs/scripts/Program.cs
+++ b/docs/scripts/Program.cs
@@ -95,7 +95,7 @@ namespace toc
         }
         const string shortTitlePrefix = "[//]: # (ShortTitle: ";
 
-        public static string maybeGetShortTitle(string originalTitle, string[] lines, int line)
+        public static string maybeGetShortTitleImpl(string originalTitle, string[] lines, int line)
         {
             string nextLine = getNextNonEmptyLine(lines, line);
             if (nextLine.StartsWith(shortTitlePrefix))
@@ -104,6 +104,26 @@ namespace toc
             }
             return originalTitle;
         }
+
+        public static string escapeString(string input)
+        {
+            StringBuilder sb = new StringBuilder();
+            foreach (var ch in input)
+            {
+                if (ch == '<')
+                    sb.Append("&lt;");
+                else if (ch == '>')
+                    sb.Append("&gt;");
+                else
+                    sb.Append(ch);
+            }
+            return sb.ToString();
+        }
+        public static string maybeGetShortTitle(string originalTitle, string[] lines, int line)
+        {
+            string title = maybeGetShortTitleImpl(originalTitle, lines, line);
+            return escapeString(title);
+        }
         public static string Run(string path)
         {
             StringBuilder outputSB = new StringBuilder();