1/*
2 * Copyright (C) 2006 Apple Inc.  All rights reserved.
3 * Copyright (C) 2006 Michael Emmel mike.emmel@gmail.com
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 * 1. Redistributions of source code must retain the above copyright
10 *    notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 *    notice, this list of conditions and the following disclaimer in the
13 *    documentation and/or other materials provided with the distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
16 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
18 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
19 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
20 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
21 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
22 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
23 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
25 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26 */
27
28#ifndef ResourceHandleManager_h
29#define ResourceHandleManager_h
30
31#include "Frame.h"
32#include "Timer.h"
33#include "ResourceHandleClient.h"
34
35#if PLATFORM(WIN)
36#include <winsock2.h>
37#include <windows.h>
38#endif
39
40#include <curl/curl.h>
41#include <wtf/Vector.h>
42#include <wtf/text/CString.h>
43#include <wtf/text/WTFString.h>
44
45namespace WebCore {
46
47class ResourceHandleManager {
48public:
49    enum ProxyType {
50        HTTP = CURLPROXY_HTTP,
51        Socks4 = CURLPROXY_SOCKS4,
52        Socks4A = CURLPROXY_SOCKS4A,
53        Socks5 = CURLPROXY_SOCKS5,
54        Socks5Hostname = CURLPROXY_SOCKS5_HOSTNAME
55    };
56    static ResourceHandleManager* sharedInstance();
57    void add(ResourceHandle*);
58    void cancel(ResourceHandle*);
59
60    CURLSH* getCurlShareHandle() const;
61
62    void setCookieJarFileName(const char* cookieJarFileName);
63    const char* getCookieJarFileName() const;
64
65    void dispatchSynchronousJob(ResourceHandle*);
66
67    void setupPOST(ResourceHandle*, struct curl_slist**);
68    void setupPUT(ResourceHandle*, struct curl_slist**);
69
70    void setProxyInfo(const String& host = "",
71                      unsigned long port = 0,
72                      ProxyType type = HTTP,
73                      const String& username = "",
74                      const String& password = "");
75
76private:
77    ResourceHandleManager();
78    ~ResourceHandleManager();
79    void downloadTimerCallback(Timer<ResourceHandleManager>*);
80    void removeFromCurl(ResourceHandle*);
81    bool removeScheduledJob(ResourceHandle*);
82    void startJob(ResourceHandle*);
83    bool startScheduledJobs();
84    void applyAuthenticationToRequest(ResourceHandle*, ResourceRequest&);
85
86    void initializeHandle(ResourceHandle*);
87
88    void initCookieSession();
89
90    Timer<ResourceHandleManager> m_downloadTimer;
91    CURLM* m_curlMultiHandle;
92    CURLSH* m_curlShareHandle;
93    char* m_cookieJarFileName;
94    char m_curlErrorBuffer[CURL_ERROR_SIZE];
95    Vector<ResourceHandle*> m_resourceHandleList;
96    const CString m_certificatePath;
97    int m_runningJobs;
98
99    String m_proxy;
100    ProxyType m_proxyType;
101};
102
103}
104
105#endif
106