Add ELEMENT_TYPE_WEBSOCKET to subresource filter

components/subresource_filter/content/renderer/web_document_subresource_filter_impl.h

Index: components/subresource_filter/content/renderer/web_document_subresource_filter_impl.h
diff --git a/components/subresource_filter/content/renderer/web_document_subresource_filter_impl.h b/components/subresource_filter/content/renderer/web_document_subresource_filter_impl.h
index 32568bbc0eaf3129d35ec2db51f5eaabc0048c1a..fe0a120336daf23c2becfac1de57d9402f1fb85c 100644
--- a/components/subresource_filter/content/renderer/web_document_subresource_filter_impl.h
+++ b/components/subresource_filter/content/renderer/web_document_subresource_filter_impl.h
@@ -9,6 +9,7 @@
 #include "base/macros.h"
 #include "base/memory/weak_ptr.h"
 #include "components/subresource_filter/core/common/document_subresource_filter.h"
+#include "components/subresource_filter/core/common/proto/rules.pb.h"
 #include "third_party/WebKit/public/platform/WebDocumentSubresourceFilter.h"
 
 namespace subresource_filter {
@@ -35,12 +36,16 @@ class WebDocumentSubresourceFilterImpl
   const DocumentSubresourceFilter& filter() const { return filter_; }
 
   // blink::WebDocumentSubresourceFilter:
-  blink::WebDocumentSubresourceFilter::LoadPolicy getLoadPolicy(
-      const blink::WebURL& resourceUrl,
-      blink::WebURLRequest::RequestContext) override;
+  LoadPolicy getLoadPolicy(const blink::WebURL& resourceUrl,
+                           blink::WebURLRequest::RequestContext) override;
+  LoadPolicy getLoadPolicyForWebSocketConnect(
+      const blink::WebURL& url) override;
   void reportDisallowedLoad() override;
 
  private:
+  LoadPolicy getLoadPolicyImpl(const blink::WebURL& url,
+                               proto::ElementType element_type);
+
   DocumentSubresourceFilter filter_;
   base::OnceClosure first_disallowed_load_callback_;