172a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen// Copyright (c) 2011 The Chromium Authors. All rights reserved. 2bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen// Use of this source code is governed by a BSD-style license that can be 3bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen// found in the LICENSE file. 4bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen 5dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen#include "chrome/browser/ui/webui/screenshot_source.h" 6bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen 7bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen#include "base/file_util.h" 8ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen#include "base/memory/ref_counted_memory.h" 9bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen#include "base/path_service.h" 103f50c38dc070f4bb515c1b64450dae14f316474eKristian Monsen#include "base/synchronization/waitable_event.h" 1172a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen#include "base/task.h" 12bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen#include "chrome/common/chrome_paths.h" 13bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen#include "chrome/common/url_constants.h" 14dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen#include "content/browser/browser_thread.h" 15bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen 16bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsenstatic const char kCurrentScreenshot[] = "current"; 17bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen#if defined(OS_CHROMEOS) 18bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsenstatic const char kSavedScreenshots[] = "saved/"; 19bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen#endif 20bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen 21bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsenstatic const char kScreenshotsRelativePath[] = "/Screenshots/"; 22bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen 23bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen#if defined(OS_CHROMEOS) 24bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen// Read the file from the screenshots directory into the read_bytes vector. 25bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsenvoid ReadScreenshot(const std::string& filename, 26bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen std::vector<unsigned char>* read_bytes, 27bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen base::WaitableEvent* read_complete) { 28bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen read_bytes->clear(); 29bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen 30bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen FilePath fileshelf_path; 31bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen if (!PathService::Get(chrome::DIR_DEFAULT_DOWNLOADS, &fileshelf_path)) { 32bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen read_complete->Signal(); 33bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen return; 34bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen } 35bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen 36bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen FilePath file(fileshelf_path.value() + std::string(kScreenshotsRelativePath) + 37bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen filename); 38bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen 39bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen int64 file_size = 0; 40bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen if (!file_util::GetFileSize(file, &file_size)) { 41bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen read_complete->Signal(); 42bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen return; 43bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen } 44bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen 45bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen // expand vector to file size 46bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen read_bytes->resize(file_size); 47bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen // read file into the vector 48bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen int bytes_read = 0; 49bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen if (!(bytes_read = file_util::ReadFile(file, 50bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen reinterpret_cast<char*>( 51bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen &read_bytes->front()), 52bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen static_cast<int>(file_size)))) 53bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen read_bytes->clear(); 54bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen 55bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen // We're done, if successful, read_bytes will have the data 56bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen // otherwise, it'll be empty. 57bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen read_complete->Signal(); 58bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen} 59bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen 60bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen// Get a saved screenshot - read on the FILE thread. 61bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsenstd::vector<unsigned char> GetSavedScreenshot(std::string filename) { 62bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen base::WaitableEvent read_complete(true, false); 63bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen std::vector<unsigned char> bytes; 64731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick BrowserThread::PostTask(BrowserThread::FILE, FROM_HERE, 65731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick NewRunnableFunction(&ReadScreenshot, filename, 66731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick &bytes, &read_complete)); 67bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen read_complete.Wait(); 68bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen return bytes; 69bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen} 70bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen#endif 71bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen 72dc0f95d653279beabeb9817299e2902918ba123eKristian Monsenstd::vector<unsigned char> ScreenshotSource::GetScreenshot( 73bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen const std::string& full_path) { 74bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen // Strip the query param value - we only use it as a hack to ensure our 75bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen // image gets reloaded instead of being pulled from the browser cache 76bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen std::string path = full_path.substr(0, full_path.find_first_of("?")); 77bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen if (path == kCurrentScreenshot) { 78bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen return current_screenshot_; 79bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen#if defined(OS_CHROMEOS) 80bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen } else if (path.compare(0, strlen(kSavedScreenshots), 81bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen kSavedScreenshots) == 0) { 82bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen // Split the saved screenshot filename from the path 83bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen std::string filename = path.substr(strlen(kSavedScreenshots)); 84bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen 85bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen return GetSavedScreenshot(filename); 86bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen#endif 87bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen } else { 88bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen std::vector<unsigned char> ret; 89bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen // TODO(rkc): Weird vc bug, return std::vector<unsigned char>() causes 90bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen // the object assigned to the return value of this function magically 91bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen // change it's address 0x0; look into this eventually. 92bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen return ret; 93bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen } 94bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen} 95bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen 96dc0f95d653279beabeb9817299e2902918ba123eKristian MonsenScreenshotSource::ScreenshotSource( 97bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen std::vector<unsigned char>* current_screenshot) 98bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen : DataSource(chrome::kChromeUIScreenshotPath, MessageLoop::current()) { 99bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen // Setup the last screenshot taken. 100bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen if (current_screenshot) 101bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen current_screenshot_ = *current_screenshot; 102bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen else 103bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen current_screenshot_.clear(); 104bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen} 105bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen 106dc0f95d653279beabeb9817299e2902918ba123eKristian MonsenScreenshotSource::~ScreenshotSource() {} 107731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick 108dc0f95d653279beabeb9817299e2902918ba123eKristian Monsenvoid ScreenshotSource::StartDataRequest(const std::string& path, 109ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen bool is_incognito, 110bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen int request_id) { 111bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen SendResponse(request_id, new RefCountedBytes(GetScreenshot(path))); 112bda42a81ee5f9b20d2bebedcf0bbef1e30e5b293Kristian Monsen} 11321d179b334e59e9a3bfcaed4c4430bef1bc5759dKristian Monsen 114dc0f95d653279beabeb9817299e2902918ba123eKristian Monsenstd::string ScreenshotSource::GetMimeType(const std::string&) const { 11521d179b334e59e9a3bfcaed4c4430bef1bc5759dKristian Monsen // We need to explicitly return a mime type, otherwise if the user tries to 11621d179b334e59e9a3bfcaed4c4430bef1bc5759dKristian Monsen // drag the image they get no extension. 11721d179b334e59e9a3bfcaed4c4430bef1bc5759dKristian Monsen return "image/png"; 11821d179b334e59e9a3bfcaed4c4430bef1bc5759dKristian Monsen} 119