fix: return early when '<' character found in tag name

ericnorris · Nov 30, 2020 · 9a4e2e8 · 9a4e2e8
1 parent e8693e7
commit 9a4e2e8
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,2 +1,3 @@
+/coverage/
 /dist/
 /node_modules/
diff --git a/src/states.ts b/src/states.ts
@@ -99,6 +99,8 @@ export class InTagNameState implements State {
 
         if (character == TAG_START) {
             this.nameBuffer += ENCODED_TAG_START;
+
+            return "";
         }
 
         if (character == TAG_END) {

diff --git a/tests/states.test.ts b/tests/states.test.ts
@@ -227,6 +227,21 @@ describe("InTagNameState", () => {
         expect(got).toEqual(want);
         expect(endState).toBeInstanceOf(InCommentState);
     });
+
+    it("should encode '<' characters within the tag name", () => {
+        const start = new InTagNameState({
+            ...OptionsWithEncodingEnabled,
+            allowedTags: new Set(["&lt;&lt;&lt;"]),
+        });
+
+        const text = "<<<>";
+        const want = `<&lt;&lt;&lt;>`;
+
+        const [got, endState] = consumeStringUntilTransitionOrEOF(start, text);
+
+        expect(got).toEqual(want);
+        expect(endState).toBeInstanceOf(InPlaintextState);
+    });
 });
 
 describe("InTagState", () => {