| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/ui/webui/screenshot_source.h" | 5 #include "chrome/browser/ui/webui/screenshot_source.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/callback.h" | 8 #include "base/callback.h" |
| 9 #include "base/file_util.h" | 9 #include "base/file_util.h" |
| 10 #include "base/memory/ref_counted_memory.h" | 10 #include "base/memory/ref_counted_memory.h" |
| (...skipping 145 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 156 content::BrowserThread::FILE, FROM_HERE, | 156 content::BrowserThread::FILE, FROM_HERE, |
| 157 base::Bind(&ScreenshotSource::SendSavedScreenshot, | 157 base::Bind(&ScreenshotSource::SendSavedScreenshot, |
| 158 base::Unretained(this), screenshot_path, request_id, file)); | 158 base::Unretained(this), screenshot_path, request_id, file)); |
| 159 } | 159 } |
| 160 #endif | 160 #endif |
| 161 | 161 |
| 162 void ScreenshotSource::CacheAndSendScreenshot( | 162 void ScreenshotSource::CacheAndSendScreenshot( |
| 163 const std::string& screenshot_path, | 163 const std::string& screenshot_path, |
| 164 int request_id, | 164 int request_id, |
| 165 ScreenshotDataPtr bytes) { | 165 ScreenshotDataPtr bytes) { |
| 166 cached_screenshots_[screenshot_path] = bytes; | 166 // Strip the query from the screenshot path. |
| 167 std::string path = screenshot_path.substr( |
| 168 0, screenshot_path.find_first_of("?")); |
| 169 cached_screenshots_[path] = bytes; |
| 167 SendResponse(request_id, new base::RefCountedBytes(*bytes)); | 170 SendResponse(request_id, new base::RefCountedBytes(*bytes)); |
| 168 } | 171 } |
| OLD | NEW |