blob: 670c70d61814b78192b0282706790f5c60e4359a (
plain)
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
|
// 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.
#include "net/ftp/ftp_ctrl_response_buffer.h"
#include "base/logging.h"
#include "base/string_number_conversions.h"
#include "base/string_piece.h"
//#include "base/string_util.h"
#include "net/base/net_errors.h"
namespace net {
// static
const int FtpCtrlResponse::kInvalidStatusCode = -1;
FtpCtrlResponse::FtpCtrlResponse() : status_code(kInvalidStatusCode) {}
FtpCtrlResponse::~FtpCtrlResponse() {}
FtpCtrlResponseBuffer::FtpCtrlResponseBuffer() : multiline_(false) {}
FtpCtrlResponseBuffer::~FtpCtrlResponseBuffer() {}
int FtpCtrlResponseBuffer::ConsumeData(const char* data, int data_length) {
buffer_.append(data, data_length);
ExtractFullLinesFromBuffer();
while (!lines_.empty()) {
ParsedLine line = lines_.front();
lines_.pop();
if (multiline_) {
if (!line.is_complete || line.status_code != response_buf_.status_code) {
line_buf_.append(line.raw_text);
continue;
}
response_buf_.lines.push_back(line_buf_);
line_buf_ = line.status_text;
DCHECK_EQ(line.status_code, response_buf_.status_code);
if (!line.is_multiline) {
response_buf_.lines.push_back(line_buf_);
responses_.push(response_buf_);
// Prepare to handle following lines.
response_buf_ = FtpCtrlResponse();
line_buf_.clear();
multiline_ = false;
}
} else {
if (!line.is_complete)
return ERR_INVALID_RESPONSE;
response_buf_.status_code = line.status_code;
if (line.is_multiline) {
line_buf_ = line.status_text;
multiline_ = true;
} else {
response_buf_.lines.push_back(line.status_text);
responses_.push(response_buf_);
// Prepare to handle following lines.
response_buf_ = FtpCtrlResponse();
line_buf_.clear();
}
}
}
return OK;
}
FtpCtrlResponse FtpCtrlResponseBuffer::PopResponse() {
FtpCtrlResponse result = responses_.front();
responses_.pop();
return result;
}
FtpCtrlResponseBuffer::ParsedLine::ParsedLine()
: has_status_code(false),
is_multiline(false),
is_complete(false),
status_code(FtpCtrlResponse::kInvalidStatusCode) {
}
// static
FtpCtrlResponseBuffer::ParsedLine FtpCtrlResponseBuffer::ParseLine(
const std::string& line) {
ParsedLine result;
if (line.length() >= 3) {
if (base::StringToInt(base::StringPiece(line.begin(), line.begin() + 3),
&result.status_code))
result.has_status_code = (100 <= result.status_code &&
result.status_code <= 599);
if (result.has_status_code && line.length() >= 4 && line[3] == ' ') {
result.is_complete = true;
} else if (result.has_status_code && line.length() >= 4 && line[3] == '-') {
result.is_complete = true;
result.is_multiline = true;
}
}
if (result.is_complete) {
result.status_text = line.substr(4);
} else {
result.status_text = line;
}
result.raw_text = line;
return result;
}
void FtpCtrlResponseBuffer::ExtractFullLinesFromBuffer() {
int cut_pos = 0;
for (size_t i = 0; i < buffer_.length(); i++) {
if (i >= 1 && buffer_[i - 1] == '\r' && buffer_[i] == '\n') {
lines_.push(ParseLine(buffer_.substr(cut_pos, i - cut_pos - 1)));
cut_pos = i + 1;
}
}
buffer_.erase(0, cut_pos);
}
} // namespace net
|
