1 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "content/public/common/page_state.h"
6
7 #include "base/files/file_path.h"
8 #include "base/strings/utf_string_conversions.h"
9 #include "content/common/page_state_serialization.h"
10
11 namespace content {
12 namespace {
13
ToNullableString16(const std::string & utf8)14 base::NullableString16 ToNullableString16(const std::string& utf8) {
15 return base::NullableString16(base::UTF8ToUTF16(utf8), false);
16 }
17
ToFilePath(const base::NullableString16 & s)18 base::FilePath ToFilePath(const base::NullableString16& s) {
19 return base::FilePath::FromUTF16Unsafe(s.string());
20 }
21
ToFilePathVector(const std::vector<base::NullableString16> & input,std::vector<base::FilePath> * output)22 void ToFilePathVector(const std::vector<base::NullableString16>& input,
23 std::vector<base::FilePath>* output) {
24 output->clear();
25 output->reserve(input.size());
26 for (size_t i = 0; i < input.size(); ++i)
27 output->push_back(ToFilePath(input[i]));
28 }
29
ToPageState(const ExplodedPageState & state)30 PageState ToPageState(const ExplodedPageState& state) {
31 std::string encoded_data;
32 if (!EncodePageState(state, &encoded_data))
33 return PageState();
34
35 return PageState::CreateFromEncodedData(encoded_data);
36 }
37
RecursivelyRemovePasswordData(ExplodedFrameState * state)38 void RecursivelyRemovePasswordData(ExplodedFrameState* state) {
39 if (state->http_body.contains_passwords)
40 state->http_body = ExplodedHttpBody();
41 }
42
RecursivelyRemoveScrollOffset(ExplodedFrameState * state)43 void RecursivelyRemoveScrollOffset(ExplodedFrameState* state) {
44 state->scroll_offset = gfx::Point();
45 state->pinch_viewport_scroll_offset = gfx::PointF();
46 }
47
RecursivelyRemoveReferrer(ExplodedFrameState * state)48 void RecursivelyRemoveReferrer(ExplodedFrameState* state) {
49 state->referrer = base::NullableString16();
50 state->referrer_policy = blink::WebReferrerPolicyDefault;
51 for (std::vector<ExplodedFrameState>::iterator it = state->children.begin();
52 it != state->children.end();
53 ++it) {
54 RecursivelyRemoveReferrer(&*it);
55 }
56 }
57
58 } // namespace
59
60 // static
CreateFromEncodedData(const std::string & data)61 PageState PageState::CreateFromEncodedData(const std::string& data) {
62 return PageState(data);
63 }
64
65 // static
CreateFromURL(const GURL & url)66 PageState PageState::CreateFromURL(const GURL& url) {
67 ExplodedPageState state;
68
69 state.top.url_string = ToNullableString16(url.possibly_invalid_spec());
70
71 return ToPageState(state);
72 }
73
74 // static
CreateForTesting(const GURL & url,bool body_contains_password_data,const char * optional_body_data,const base::FilePath * optional_body_file_path)75 PageState PageState::CreateForTesting(
76 const GURL& url,
77 bool body_contains_password_data,
78 const char* optional_body_data,
79 const base::FilePath* optional_body_file_path) {
80 ExplodedPageState state;
81
82 state.top.url_string = ToNullableString16(url.possibly_invalid_spec());
83
84 if (optional_body_data || optional_body_file_path) {
85 state.top.http_body.is_null = false;
86 if (optional_body_data) {
87 ExplodedHttpBodyElement element;
88 element.type = blink::WebHTTPBody::Element::TypeData;
89 element.data = optional_body_data;
90 state.top.http_body.elements.push_back(element);
91 }
92 if (optional_body_file_path) {
93 ExplodedHttpBodyElement element;
94 element.type = blink::WebHTTPBody::Element::TypeFile;
95 element.file_path =
96 ToNullableString16(optional_body_file_path->AsUTF8Unsafe());
97 state.top.http_body.elements.push_back(element);
98 state.referenced_files.push_back(element.file_path);
99 }
100 state.top.http_body.contains_passwords =
101 body_contains_password_data;
102 }
103
104 return ToPageState(state);
105 }
106
PageState()107 PageState::PageState() {
108 }
109
IsValid() const110 bool PageState::IsValid() const {
111 return !data_.empty();
112 }
113
Equals(const PageState & other) const114 bool PageState::Equals(const PageState& other) const {
115 return data_ == other.data_;
116 }
117
ToEncodedData() const118 const std::string& PageState::ToEncodedData() const {
119 return data_;
120 }
121
GetReferencedFiles() const122 std::vector<base::FilePath> PageState::GetReferencedFiles() const {
123 std::vector<base::FilePath> results;
124
125 ExplodedPageState state;
126 if (DecodePageState(data_, &state))
127 ToFilePathVector(state.referenced_files, &results);
128
129 return results;
130 }
131
RemovePasswordData() const132 PageState PageState::RemovePasswordData() const {
133 ExplodedPageState state;
134 if (!DecodePageState(data_, &state))
135 return PageState(); // Oops!
136
137 RecursivelyRemovePasswordData(&state.top);
138
139 return ToPageState(state);
140 }
141
RemoveScrollOffset() const142 PageState PageState::RemoveScrollOffset() const {
143 ExplodedPageState state;
144 if (!DecodePageState(data_, &state))
145 return PageState(); // Oops!
146
147 RecursivelyRemoveScrollOffset(&state.top);
148
149 return ToPageState(state);
150 }
151
RemoveReferrer() const152 PageState PageState::RemoveReferrer() const {
153 if (data_.empty())
154 return *this;
155
156 ExplodedPageState state;
157 if (!DecodePageState(data_, &state))
158 return PageState(); // Oops!
159
160 RecursivelyRemoveReferrer(&state.top);
161
162 return ToPageState(state);
163 }
164
PageState(const std::string & data)165 PageState::PageState(const std::string& data)
166 : data_(data) {
167 // TODO(darin): Enable this DCHECK once tests have been fixed up to not pass
168 // bogus encoded data to CreateFromEncodedData.
169 //DCHECK(IsValid());
170 }
171
172 } // namespace content
173