15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Objects that handle file operations for saving files, on the file thread. 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The SaveFileManager owns a set of SaveFile objects, each of which connects 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// with a SaveItem object which belongs to one SavePackage and runs on the file 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// thread for saving data in order to avoid disk activity on either network IO 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// thread or the UI thread. It coordinates the notifications from the network 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// and UI. 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The SaveFileManager itself is a singleton object owned by the 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// ResourceDispatcherHostImpl. 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The data sent to SaveFileManager have 2 sources, one is from 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// ResourceDispatcherHostImpl, run in network IO thread, the all sub-resources 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// and save-only-HTML pages will be got from network IO. The second is from 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// render process, those html pages which are serialized from DOM will be 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// composed in render process and encoded to its original encoding, then sent 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// to UI loop in browser process, then UI loop will dispatch the data to 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// SaveFileManager on the file thread. SaveFileManager will directly 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// call SaveFile's method to persist data. 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// A typical saving job operation involves multiple threads: 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Updating an in progress save file 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// io_thread 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// |----> data from net ---->| 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// | 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// | 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// |----> data from ---->| | 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// | render process | | 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// ui_thread | | 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// file_thread (writes to disk) 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// |----> stats ---->| 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// ui_thread (feedback for user) 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Cancel operations perform the inverse order when triggered by a user action: 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// ui_thread (user click) 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// |----> cancel command ---->| 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// | | file_thread (close file) 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// | |---------------------> cancel command ---->| 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// | io_thread (stops net IO 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// ui_thread (user close contents) for saving) 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// |----> cancel command ---->| 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Render process(stop serializing DOM and sending 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// data) 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The SaveFileManager tracks saving requests, mapping from a save ID (unique 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// integer created in the IO thread) to the SavePackage for the contents where 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// the saving job was initiated. In the event of a contents closure during 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// saving, the SavePackage will notify the SaveFileManage to cancel all SaveFile 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// jobs. 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifndef CONTENT_BROWSER_DOWNLOAD_SAVE_FILE_MANAGER_H_ 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define CONTENT_BROWSER_DOWNLOAD_SAVE_FILE_MANAGER_H_ 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <string> 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/basictypes.h" 647d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)#include "base/containers/hash_tables.h" 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/ref_counted.h" 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/browser/download/save_types.h" 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/common/content_export.h" 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class GURL; 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)namespace base { 722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)class FilePath; 732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace net { 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class IOBuffer; 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace content { 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class ResourceContext; 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class SaveFile; 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class SavePackage; 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)struct Referrer; 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class SaveFileManager : public base::RefCountedThreadSafe<SaveFileManager> { 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SaveFileManager(); 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Lifetime management. 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CONTENT_EXPORT void Shutdown(); 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Called on the IO thread. This generates unique IDs for 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // SaveFileResourceHandler objects (there's one per file in a SavePackage). 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Note that this is different from the SavePackage's id. 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int GetNextId(); 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Save the specified URL. Called on the UI thread and forwarded to the 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // ResourceDispatcherHostImpl on the IO thread. 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SaveURL(const GURL& url, 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const Referrer& referrer, 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int render_process_host_id, 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int render_view_id, 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SaveFileCreateInfo::SaveFileSource save_source, 1042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& file_full_path, 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ResourceContext* context, 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SavePackage* save_package); 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Notifications sent from the IO thread and run on the file thread: 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void StartSave(SaveFileCreateInfo* info); 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void UpdateSaveProgress(int save_id, net::IOBuffer* data, int size); 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SaveFinished(int save_id, 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const GURL& save_url, 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int render_process_id, 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool is_success); 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Notifications sent from the UI thread and run on the file thread. 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Cancel a SaveFile instance which has specified save id. 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void CancelSave(int save_id); 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Called on the UI thread to remove a save package from SaveFileManager's 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // tracking map. 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void RemoveSaveFile(int save_id, const GURL& save_url, 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SavePackage* package); 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Helper function for deleting specified file. 1262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) void DeleteDirectoryOrFile(const base::FilePath& full_path, bool is_dir); 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Runs on file thread to save a file by copying from file system when 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // original url is using file scheme. 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SaveLocalFile(const GURL& original_file_url, 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int save_id, 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int render_process_id); 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Renames all the successfully saved files. 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |final_names| points to a vector which contains pairs of save ids and 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // final names of successfully saved files. 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void RenameAllFiles( 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const FinalNameList& final_names, 1392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& resource_dir, 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int render_process_id, 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int render_view_id, 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int save_package_id); 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // When the user cancels the saving, we need to remove all remaining saved 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // files of this page saving job from save_file_map_. 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void RemoveSavedFileFromFileMap(const SaveIDList & save_ids); 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) friend class base::RefCountedThreadSafe<SaveFileManager>; 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ~SaveFileManager(); 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // A cleanup helper that runs on the file thread. 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void OnShutdown(); 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Called only on UI thread to get the SavePackage for a contents's browser 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // context. 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static SavePackage* GetSavePackageFromRenderIds(int render_process_id, 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int review_view_id); 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Register a starting request. Associate the save URL with a 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // SavePackage for further matching. 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void RegisterStartingRequest(const GURL& save_url, 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SavePackage* save_package); 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Unregister a start request according save URL, disassociate 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // the save URL and SavePackage. 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SavePackage* UnregisterStartingRequest(const GURL& save_url, 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int contents_id); 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Look up the SavePackage according to save id. 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SavePackage* LookupPackage(int save_id); 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Called only on the file thread. 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Look up one in-progress saving item according to save id. 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SaveFile* LookupSaveFile(int save_id); 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Help function for sending notification of canceling specific request. 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SendCancelRequest(int save_id); 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Notifications sent from the file thread and run on the UI thread. 1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Lookup the SaveManager for this WebContents' saving browser context and 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // inform it the saving job has been started. 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void OnStartSave(const SaveFileCreateInfo* info); 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Update the SavePackage with the current state of a started saving job. 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If the SavePackage for this saving job is gone, cancel the request. 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void OnUpdateSaveProgress(int save_id, 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int64 bytes_so_far, 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool write_success); 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Update the SavePackage with the finish state, and remove the request 1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // tracking entries. 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void OnSaveFinished(int save_id, int64 bytes_so_far, bool is_success); 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // For those requests that do not have valid save id, use 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // map:(url, SavePackage) to find the request and remove it. 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void OnErrorFinished(const GURL& save_url, int contents_id); 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Notifies SavePackage that the whole page saving job is finished. 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void OnFinishSavePageJob(int render_process_id, 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int render_view_id, 1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int save_package_id); 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Notifications sent from the UI thread and run on the file thread. 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Deletes a specified file on the file thread. 2042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) void OnDeleteDirectoryOrFile(const base::FilePath& full_path, bool is_dir); 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Notifications sent from the UI thread and run on the IO thread 2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Initiates a request for URL to be saved. 2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void OnSaveURL(const GURL& url, 2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const Referrer& referrer, 2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int render_process_host_id, 2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int render_view_id, 2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ResourceContext* context); 2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Handler for a notification sent to the IO thread for generating save id. 2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void OnRequireSaveJobFromOtherSource(SaveFileCreateInfo* info); 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Call ResourceDispatcherHostImpl's CancelRequest method to execute cancel 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // action in the IO thread. 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void ExecuteCancelSaveRequest(int render_process_id, int request_id); 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Unique ID for the next SaveFile object. 2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int next_id_; 2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // A map of all saving jobs by using save id. 2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) typedef base::hash_map<int, SaveFile*> SaveFileMap; 2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SaveFileMap save_file_map_; 2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Tracks which SavePackage to send data to, called only on UI thread. 2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // SavePackageMap maps save IDs to their SavePackage. 2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) typedef base::hash_map<int, SavePackage*> SavePackageMap; 2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SavePackageMap packages_; 2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // There is a gap between after calling SaveURL() and before calling 2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // StartSave(). In this gap, each request does not have save id for tracking. 2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // But sometimes users might want to stop saving job or ResourceDispatcherHost 2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // calls SaveFinished with save id -1 for network error. We name the requests 2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // as starting requests. For tracking those starting requests, we need to 2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // have some data structure. 2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // First we use a hashmap to map the request URL to SavePackage, then we use a 2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // hashmap to map the contents id (we actually use render_process_id) to the 2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // hashmap since it is possible to save the same URL in different contents at 2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // same time. 2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) typedef base::hash_map<std::string, SavePackage*> StartingRequestsMap; 2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) typedef base::hash_map<int, StartingRequestsMap> 2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ContentsToStartingRequestsMap; 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ContentsToStartingRequestsMap contents_starting_requests_; 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(SaveFileManager); 2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace content 2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif // CONTENT_BROWSER_DOWNLOAD_SAVE_FILE_MANAGER_H_ 253