1 /*

     2  * Copyright (C) 2009 Google Inc. All rights reserved.

     3  *

     4  * Redistribution and use in source and binary forms, with or without

     5  * modification, are permitted provided that the following conditions are

     6  * met:

     7  *

     8  *     * Redistributions of source code must retain the above copyright

     9  * notice, this list of conditions and the following disclaimer.

    10  *     * Redistributions in binary form must reproduce the above

    11  * copyright notice, this list of conditions and the following disclaimer

    12  * in the documentation and/or other materials provided with the

    13  * distribution.

    14  *     * Neither the name of Google Inc. nor the names of its

    15  * contributors may be used to endorse or promote products derived from

    16  * this software without specific prior written permission.

    17  *

    18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS

    19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT

    20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR

    21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT

    22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

    23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

    24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,

    25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY

    26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

    27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE

    28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

    29  */

    30 

    31 #ifndef DocumentThreadableLoader_h

    32 #define DocumentThreadableLoader_h

    33 

    34 #include "FrameLoaderTypes.h"

    35 #include "SubresourceLoaderClient.h"

    36 #include "ThreadableLoader.h"

    37 #include <wtf/OwnPtr.h>

    38 #include <wtf/PassRefPtr.h>

    39 #include <wtf/RefCounted.h>

    40 #include <wtf/RefPtr.h>

    41 

    42 namespace WebCore {

    43     class Document;

    44     class KURL;

    45     class ResourceRequest;

    46     class String;

    47     class ThreadableLoaderClient;

    48 

    49     class DocumentThreadableLoader : public RefCounted<DocumentThreadableLoader>, public ThreadableLoader, private SubresourceLoaderClient  {

    50     public:

    51         static void loadResourceSynchronously(Document*, const ResourceRequest&, ThreadableLoaderClient&, const ThreadableLoaderOptions&);

    52         static PassRefPtr<DocumentThreadableLoader> create(Document*, ThreadableLoaderClient*, const ResourceRequest&, const ThreadableLoaderOptions&);

    53         virtual ~DocumentThreadableLoader();

    54 

    55         virtual void cancel();

    56 

    57         using RefCounted<DocumentThreadableLoader>::ref;

    58         using RefCounted<DocumentThreadableLoader>::deref;

    59 

    60     protected:

    61         virtual void refThreadableLoader() { ref(); }

    62         virtual void derefThreadableLoader() { deref(); }

    63 

    64     private:

    65         enum BlockingBehavior {

    66             LoadSynchronously,

    67             LoadAsynchronously

    68         };

    69 

    70         DocumentThreadableLoader(Document*, ThreadableLoaderClient*, BlockingBehavior blockingBehavior, const ResourceRequest&, const ThreadableLoaderOptions& options);

    71 

    72         virtual void willSendRequest(SubresourceLoader*, ResourceRequest&, const ResourceResponse& redirectResponse);

    73         virtual void didSendData(SubresourceLoader*, unsigned long long bytesSent, unsigned long long totalBytesToBeSent);

    74 

    75         virtual void didReceiveResponse(SubresourceLoader*, const ResourceResponse&);

    76         virtual void didReceiveData(SubresourceLoader*, const char*, int lengthReceived);

    77         virtual void didFinishLoading(SubresourceLoader*);

    78         virtual void didFail(SubresourceLoader*, const ResourceError&);

    79 

    80         virtual bool getShouldUseCredentialStorage(SubresourceLoader*, bool& shouldUseCredentialStorage);

    81         virtual void didReceiveAuthenticationChallenge(SubresourceLoader*, const AuthenticationChallenge&);

    82         virtual void receivedCancellation(SubresourceLoader*, const AuthenticationChallenge&);

    83 

    84         void didFinishLoading(unsigned long identifier);

    85         void makeSimpleCrossOriginAccessRequest(const ResourceRequest& request);

    86         void makeCrossOriginAccessRequestWithPreflight(const ResourceRequest& request);

    87         void preflightSuccess();

    88         void preflightFailure(const String& url, const String& errorDescription);

    89 

    90         void loadRequest(const ResourceRequest&, SecurityCheckPolicy);

    91         bool isAllowedRedirect(const KURL&);

    92 

    93         RefPtr<SubresourceLoader> m_loader;

    94         ThreadableLoaderClient* m_client;

    95         Document* m_document;

    96         ThreadableLoaderOptions m_options;

    97         bool m_sameOriginRequest;

    98         bool m_async;

    99         OwnPtr<ResourceRequest> m_actualRequest;  // non-null during Access Control preflight checks

   100     };

   101 

   102 } // namespace WebCore

   103 

   104 #endif // DocumentThreadableLoader_h