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
154
|
// 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 <stddef.h>
#include <stdint.h>
#include <map>
#include <string>
#include "base/compiler_specific.h"
#include "base/macros.h"
#include "base/memory/scoped_ptr.h"
#include "net/tools/balsa/balsa_headers.h"
#include "net/tools/balsa/balsa_visitor_interface.h"
#include "net/tools/flip_server/constants.h"
namespace net {
class StoreBodyAndHeadersVisitor : public BalsaVisitorInterface {
public:
void HandleError() { error_ = true; }
// BalsaVisitorInterface:
void ProcessBodyInput(const char* input, size_t size) override {}
void ProcessBodyData(const char* input, size_t size) override;
void ProcessHeaderInput(const char* input, size_t size) override {}
void ProcessTrailerInput(const char* input, size_t size) override {}
void ProcessHeaders(const BalsaHeaders& headers) override {
// nothing to do here-- we're assuming that the BalsaFrame has
// been handed our headers.
}
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 {}
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 {}
void ProcessChunkLength(size_t chunk_length) override {}
void ProcessChunkExtensions(const char* input, size_t size) override {}
void HeaderDone() override {}
void MessageDone() override {}
void HandleHeaderError(BalsaFrame* framer) override;
void HandleHeaderWarning(BalsaFrame* framer) override;
void HandleChunkingError(BalsaFrame* framer) override;
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_t stream_id;
uint32_t max_segment_size;
size_t bytes_sent;
};
class MemoryCache {
public:
using Files = std::map<std::string, scoped_ptr<FileData>>;
public:
MemoryCache();
virtual ~MemoryCache();
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);
// For unittests
void InsertFile(const BalsaHeaders* headers,
const std::string& filename,
const std::string& body);
private:
void InsertFile(FileData* file_data);
Files files_;
std::string cwd_;
};
class NotifierInterface {
public:
virtual ~NotifierInterface() {}
virtual void Notify() = 0;
};
} // namespace net
#endif // NET_TOOLS_FLIP_SERVER_MEM_CACHE_H_
|