1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
|
// Copyright (c) 2011 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef NET_TOOLS_FLIP_SERVER_MEM_CACHE_H_
#define NET_TOOLS_FLIP_SERVER_MEM_CACHE_H_
#include <map>
#include <string>
#include "base/compiler_specific.h"
#include "base/memory/scoped_ptr.h"
#include "net/tools/flip_server/balsa_headers.h"
#include "net/tools/flip_server/balsa_visitor_interface.h"
#include "net/tools/flip_server/constants.h"
namespace net {
class StoreBodyAndHeadersVisitor: public BalsaVisitorInterface {
public:
void HandleError() { error_ = true; }
// BalsaVisitorInterface:
virtual void ProcessBodyInput(const char *input, size_t size) OVERRIDE {}
virtual void ProcessBodyData(const char *input, size_t size) OVERRIDE;
virtual void ProcessHeaderInput(const char *input, size_t size) OVERRIDE {}
virtual void ProcessTrailerInput(const char *input, size_t size) OVERRIDE {}
virtual void ProcessHeaders(const BalsaHeaders& headers) OVERRIDE {
// nothing to do here-- we're assuming that the BalsaFrame has
// been handed our headers.
}
virtual void ProcessRequestFirstLine(const char* line_input,
size_t line_length,
const char* method_input,
size_t method_length,
const char* request_uri_input,
size_t request_uri_length,
const char* version_input,
size_t version_length) OVERRIDE {}
virtual void ProcessResponseFirstLine(const char *line_input,
size_t line_length,
const char *version_input,
size_t version_length,
const char *status_input,
size_t status_length,
const char *reason_input,
size_t reason_length) OVERRIDE {}
virtual void ProcessChunkLength(size_t chunk_length) OVERRIDE {}
virtual void ProcessChunkExtensions(const char *input,
size_t size) OVERRIDE {}
virtual void HeaderDone() OVERRIDE {}
virtual void MessageDone() OVERRIDE {}
virtual void HandleHeaderError(BalsaFrame* framer) OVERRIDE;
virtual void HandleHeaderWarning(BalsaFrame* framer) OVERRIDE;
virtual void HandleChunkingError(BalsaFrame* framer) OVERRIDE;
virtual void HandleBodyError(BalsaFrame* framer) OVERRIDE;
BalsaHeaders headers;
std::string body;
bool error_;
};
////////////////////////////////////////////////////////////////////////////////
class FileData {
public:
FileData();
FileData(const BalsaHeaders* headers,
const std::string& filename,
const std::string& body);
~FileData();
BalsaHeaders* headers() { return headers_.get(); }
const BalsaHeaders* headers() const { return headers_.get(); }
const std::string& filename() { return filename_; }
const std::string& body() { return body_; }
private:
scoped_ptr<BalsaHeaders> headers_;
std::string filename_;
std::string body_;
DISALLOW_COPY_AND_ASSIGN(FileData);
};
////////////////////////////////////////////////////////////////////////////////
class MemCacheIter {
public:
MemCacheIter() :
file_data(NULL),
priority(0),
transformed_header(false),
body_bytes_consumed(0),
stream_id(0),
max_segment_size(kInitialDataSendersThreshold),
bytes_sent(0) {}
explicit MemCacheIter(FileData* fd) :
file_data(fd),
priority(0),
transformed_header(false),
body_bytes_consumed(0),
stream_id(0),
max_segment_size(kInitialDataSendersThreshold),
bytes_sent(0) {}
FileData* file_data;
int priority;
bool transformed_header;
size_t body_bytes_consumed;
uint32 stream_id;
uint32 max_segment_size;
size_t bytes_sent;
};
////////////////////////////////////////////////////////////////////////////////
class MemoryCache {
public:
typedef std::map<std::string, FileData*> Files;
public:
MemoryCache();
virtual ~MemoryCache();
void CloneFrom(const MemoryCache& mc);
void AddFiles();
// virtual for unittests
virtual void ReadToString(const char* filename, std::string* output);
void ReadAndStoreFileContents(const char* filename);
FileData* GetFileData(const std::string& filename);
bool AssignFileData(const std::string& filename, MemCacheIter* mci);
private:
void ClearFiles();
Files files_;
std::string cwd_;
};
class NotifierInterface {
public:
virtual ~NotifierInterface() {}
virtual void Notify() = 0;
};
} // namespace net
#endif // NET_TOOLS_FLIP_SERVER_MEM_CACHE_H_
|
