Fix hanging on incomplete directives. #638

aaubry · Jul 13, 2022 · 898dea8 · 898dea8
1 parent 43a17b9
commit 898dea8
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/YamlDotNet/Core/Scanner.cs b/YamlDotNet/Core/Scanner.cs
@@ -796,7 +796,7 @@ private void FetchDirective()
 
                 default:
                     // warning: skipping reserved directive line
-                    while (!analyzer.Check('#') && !analyzer.IsBreak())
+                    while (!analyzer.EndOfInput && !analyzer.Check('#') && !analyzer.IsBreak())
                     {
                         Skip();
                     }
@@ -2284,6 +2284,13 @@ private string ScanDirectiveName(in Mark start)
                 throw new SyntaxErrorException(start, cursor.Mark(), "While scanning a directive, could not find expected directive name.");
             }
 
+            // Check for end of stream
+
+            if (analyzer.EndOfInput)
+            {
+                throw new SyntaxErrorException(start, cursor.Mark(), "While scanning a directive, found unexpected end of stream.");
+            }
+
             // Check for an blank character after the name.
 
             if (!analyzer.IsWhiteBreakOrZero())