7026d2c9efef94671940a39dcc7b090fe28bced2
[WebKit-https.git] / Source / WebCore / inspector / NetworkResourcesData.h
1 /*
2  * Copyright (C) 2011 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  * 1. Redistributions of source code must retain the above copyright
9  * notice, this list of conditions and the following disclaimer.
10  *
11  * 2. Redistributions in binary form must reproduce the above
12  * copyright notice, this list of conditions and the following disclaimer
13  * in the documentation and/or other materials provided with the
14  * distribution.
15  *
16  * THIS SOFTWARE IS PROVIDED BY GOOGLE INC. AND ITS CONTRIBUTORS
17  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
18  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
19  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GOOGLE INC.
20  * OR ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
21  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
22  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
26  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27  */
28
29 #ifndef NetworkResourcesData_h
30 #define NetworkResourcesData_h
31
32 #include "InspectorPageAgent.h"
33 #include "TextResourceDecoder.h"
34
35 #include <wtf/Deque.h>
36 #include <wtf/HashMap.h>
37 #include <wtf/RefCounted.h>
38 #include <wtf/text/StringBuilder.h>
39 #include <wtf/text/WTFString.h>
40
41 #if ENABLE(INSPECTOR)
42
43 namespace WebCore {
44
45 class CachedResource;
46 class SharedBuffer;
47 class TextResourceDecoder;
48
49 class NetworkResourcesData {
50 public:
51     class ResourceData {
52         friend class NetworkResourcesData;
53     public:
54         ResourceData(const String& requestId, const String& loaderId);
55
56         String requestId() const { return m_requestId; }
57         String loaderId() const { return m_loaderId; }
58
59         String frameId() const { return m_frameId; }
60         void setFrameId(const String& frameId) { m_frameId = frameId; }
61
62         String url() const { return m_url; }
63         void setUrl(const String& url) { m_url = url; }
64
65         bool hasContent() const { return !m_content.isNull(); }
66         String content() const { return m_content; }
67         void setContent(const String&, bool base64Encoded);
68
69         bool base64Encoded() const { return m_base64Encoded; }
70
71         unsigned removeContent();
72         bool isContentPurged() const { return m_isContentPurged; }
73         unsigned purgeContent();
74
75         InspectorPageAgent::ResourceType type() const { return m_type; }
76         void setType(InspectorPageAgent::ResourceType type) { m_type = type; }
77
78         int httpStatusCode() const { return m_httpStatusCode; }
79         void setHTTPStatusCode(int httpStatusCode) { m_httpStatusCode = httpStatusCode; }
80
81         String textEncodingName() const { return m_textEncodingName; }
82         void setTextEncodingName(const String& textEncodingName) { m_textEncodingName = textEncodingName; }
83
84         PassRefPtr<TextResourceDecoder> decoder() const { return m_decoder; }
85         void setDecoder(PassRefPtr<TextResourceDecoder> decoder) { m_decoder = decoder; }
86
87         PassRefPtr<SharedBuffer> buffer() const { return m_buffer; }
88         void setBuffer(PassRefPtr<SharedBuffer> buffer) { m_buffer = buffer; }
89
90         CachedResource* cachedResource() const { return m_cachedResource; }
91         void setCachedResource(CachedResource* cachedResource) { m_cachedResource = cachedResource; }
92
93     private:
94         bool hasData() const { return m_dataBuffer; }
95         size_t dataLength() const;
96         void appendData(const char* data, size_t dataLength);
97         size_t decodeDataToContent();
98
99         String m_requestId;
100         String m_loaderId;
101         String m_frameId;
102         String m_url;
103         String m_content;
104         bool m_base64Encoded;
105         RefPtr<SharedBuffer> m_dataBuffer;
106         bool m_isContentPurged;
107         InspectorPageAgent::ResourceType m_type;
108         int m_httpStatusCode;
109
110         String m_textEncodingName;
111         RefPtr<TextResourceDecoder> m_decoder;
112
113         RefPtr<SharedBuffer> m_buffer;
114         CachedResource* m_cachedResource;
115     };
116
117     NetworkResourcesData();
118
119     ~NetworkResourcesData();
120
121     void resourceCreated(const String& requestId, const String& loaderId);
122     void responseReceived(const String& requestId, const String& frameId, const ResourceResponse&);
123     void setResourceType(const String& requestId, InspectorPageAgent::ResourceType);
124     InspectorPageAgent::ResourceType resourceType(const String& requestId);
125     void setResourceContent(const String& requestId, const String& content, bool base64Encoded = false);
126     void maybeAddResourceData(const String& requestId, const char* data, size_t dataLength);
127     void maybeDecodeDataToContent(const String& requestId);
128     void addCachedResource(const String& requestId, CachedResource*);
129     void addResourceSharedBuffer(const String& requestId, PassRefPtr<SharedBuffer>, const String& textEncodingName);
130     ResourceData const* data(const String& requestId);
131     Vector<String> removeCachedResource(CachedResource*);
132     void clear(const String& preservedLoaderId = String());
133
134     void setResourcesDataSizeLimits(size_t maximumResourcesContentSize, size_t maximumSingleResourceContentSize);
135
136 private:
137     void ensureNoDataForRequestId(const String& requestId);
138     bool ensureFreeSpace(size_t size);
139
140     Deque<String> m_requestIdsDeque;
141
142     typedef HashMap<String, ResourceData*> ResourceDataMap;
143     ResourceDataMap m_requestIdToResourceDataMap;
144     size_t m_contentSize;
145     size_t m_maximumResourcesContentSize;
146     size_t m_maximumSingleResourceContentSize;
147 };
148
149 } // namespace WebCore
150
151 #endif // ENABLE(INSPECTOR)
152
153 #endif // !defined(NetworkResourcesData_h)