[HTML Import] Respect Content Security Policy Model

Source/core/html/HTMLImportsController.h

 /*
  * Copyright (C) 2013 Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 57 matching lines @@
     Document* importedDocument() const;
 
 private:
     RefPtr<HTMLImportLoader> m_loader;
 };
 
 class HTMLImportLoader : public RefCounted<HTMLImportLoader>, public HTMLImport, public CachedRawResourceClient {
 public:
     enum State {
         StateLoading,
+        StateWritten,
         StateError,
         StateReady
     };
 
     static PassRefPtr<HTMLImportLoader> create(HTMLImport* parent, const KURL&, const CachedResourceHandle<CachedScript>&);
     virtual ~HTMLImportLoader();
 
     Document* importedDocument() const;
     const KURL& url() const { return m_url; }
 
     void importDestroyed();
     bool isDone() const { return m_state == StateReady || m_state == StateError; }
 
     // HTMLImport
     virtual HTMLImportsController* controller() OVERRIDE;
     virtual HTMLImport* parent() OVERRIDE;
     virtual Document* document() OVERRIDE;
     virtual void wasDetachedFromDocument() OVERRIDE;
+    virtual void didFinishParsing() OVERRIDE;
 
 private:
     HTMLImportLoader(HTMLImport*, const KURL&, const CachedResourceHandle<CachedScript>&);
 
     // CachedRawResourceClient
     virtual void responseReceived(CachedResource*, const ResourceResponse&) OVERRIDE;
     virtual void dataReceived(CachedResource*, const char* data, int length) OVERRIDE;
     virtual void notifyFinished(CachedResource*) OVERRIDE;
 
-    State startParsing(const ResourceResponse&);
-    State finish();
+    State startWritingAndParsing(const ResourceResponse&);
+    State finishWriting();
+    State finishParsing();
+
     void setState(State);
     void dispose();
 
     HTMLImport* m_parent;
     State m_state;
     KURL m_url;
     CachedResourceHandle<CachedRawResource> m_resource;
     RefPtr<Document> m_importedDocument;
     RefPtr<DocumentWriter> m_writer;
 };
 
 class HTMLImportsController : public HTMLImport, public Supplement<ScriptExecutionContext> {
     WTF_MAKE_FAST_ALLOCATED;
 public:
     static void provideTo(Document*);
 
     explicit HTMLImportsController(Document*);
     virtual ~HTMLImportsController();
 
     // HTMLImport
     virtual HTMLImportsController* controller() OVERRIDE;
     virtual HTMLImport* parent() OVERRIDE;
     virtual Document* document() OVERRIDE;
     virtual void wasDetachedFromDocument() OVERRIDE;
+    virtual void didFinishParsing() OVERRIDE;
 
     void addImport(PassRefPtr<HTMLImportLoader>);
     void showSecurityErrorMessage(const String&);
     PassRefPtr<HTMLImportLoader> findLinkFor(const KURL&) const;
     SecurityOrigin* securityOrigin() const;
     ResourceFetcher* fetcher() const;
     bool haveChildrenLoaded(HTMLImport* parent) const;
     void didLoad(HTMLImportLoader*);
 
 private:
     void clear();
 
     Document* m_master;
 
     // List of import which has been loaded or being loaded.
     typedef Vector<RefPtr<HTMLImportLoader> > ImportList;
     ImportList m_imports;
 };
 
 } // namespace WebCore
 
 #endif // HTMLImportsController_h