From 9f4de95e623be2e71ac97d7e0e04ad1126b78272 Mon Sep 17 00:00:00 2001
From: iska <iska.gp@gmail.com>
Date: Sun, 15 Mar 2015 21:04:25 +0100
Subject: [PATCH] Add HTML Tokenizer property to set the associated parser
 object

---
 HTMLKit/HTMLParser.m    | 4 ++++
 HTMLKit/HTMLTokenizer.h | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/HTMLKit/HTMLParser.m b/HTMLKit/HTMLParser.m
index 76fbab9..2c18460 100644
--- a/HTMLKit/HTMLParser.m
+++ b/HTMLKit/HTMLParser.m
@@ -17,6 +17,9 @@
 #import "HTMLMarker.h"
 #import "NSString+HTMLKit.h"
 
+@interface HTMLTokenizer (Private)
+@property (nonatomic, weak) HTMLParser *parser;
+@end
 
 @interface HTMLDocument (Private)
 @property (nonatomic, assign) HTMLDocumentReadyState readyState;
@@ -70,6 +73,7 @@
 		_stackOfOpenElements = [HTMLStackOfOpenElements new];
 		_listOfActiveFormattingElements = [NSMutableArray new];
 		_tokenizer = [[HTMLTokenizer alloc] initWithString:string];
+		_tokenizer.parser = self;
 
 		_pendingTableCharacterTokens = [[HTMLCharacterToken alloc] initWithString:@""];
 
diff --git a/HTMLKit/HTMLTokenizer.h b/HTMLKit/HTMLTokenizer.h
index c2c82dc..b9be8de 100644
--- a/HTMLKit/HTMLTokenizer.h
+++ b/HTMLKit/HTMLTokenizer.h
@@ -14,9 +14,13 @@
  * HTML Tokenizer
  * https://html.spec.whatwg.org/multipage/syntax.html#tokenization
  */
+
+@class HTMLParser;
+
 @interface HTMLTokenizer : NSObject <NSFastEnumeration>
 
 @property (nonatomic, assign) HTMLTokenizerState state;
+@property (nonatomic, weak, readonly) HTMLParser *parser;
 
 - (instancetype)initWithString:(NSString *)string;