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)#include "content/browser/download/save_item.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h" 8868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/string_util.h" 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/browser/download/save_file.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/browser/download/save_file_manager.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/browser/download/save_package.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace content { 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Constructor for SaveItem when creating each saving job. 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)SaveItem::SaveItem(const GURL& url, 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const Referrer& referrer, 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SavePackage* package, 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SaveFileCreateInfo::SaveFileSource save_source) 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : save_id_(-1), 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) url_(url), 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) referrer_(referrer), 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) total_bytes_(0), 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) received_bytes_(0), 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_(WAIT_START), 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) has_final_name_(false), 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) is_success_(false), 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) save_source_(save_source), 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) package_(package) { 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(package); 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)SaveItem::~SaveItem() { 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Set start state for save item. 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void SaveItem::Start() { 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(state_ == WAIT_START); 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_ = IN_PROGRESS; 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// If we've received more data than we were expecting (bad server info?), 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// revert to 'unknown size mode'. 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void SaveItem::UpdateSize(int64 bytes_so_far) { 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) received_bytes_ = bytes_so_far; 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (received_bytes_ >= total_bytes_) 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) total_bytes_ = 0; 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Updates from the file thread may have been posted while this saving job 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// was being canceled in the UI thread, so we'll accept them unless we're 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// complete. 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void SaveItem::Update(int64 bytes_so_far) { 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (state_ != IN_PROGRESS) { 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTREACHED(); 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UpdateSize(bytes_so_far); 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Cancel this saving item job. If the job is not in progress, ignore 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// this command. The SavePackage will each in-progress SaveItem's cancel 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// when canceling whole saving page job. 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void SaveItem::Cancel() { 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If item is in WAIT_START mode, which means no request has been sent. 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // So we need not to cancel it. 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (state_ != IN_PROGRESS) { 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Small downloads might be complete before method has a chance to run. 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_ = CANCELED; 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) is_success_ = false; 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Finish(received_bytes_, false); 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) package_->SaveCanceled(this); 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Set finish state for a save item 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void SaveItem::Finish(int64 size, bool is_success) { 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // When this function is called, the SaveItem should be one of following 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // three situations. 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // a) The data of this SaveItem is finished saving. So it should have 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // generated final name. 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // b) Error happened before the start of saving process. So no |save_id_| is 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // generated for this SaveItem and the |is_success_| should be false. 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // c) Error happened in the start of saving process, the SaveItem has a save 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // id, |is_success_| should be false, and the |size| should be 0. 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(has_final_name() || (save_id_ == -1 && !is_success_) || 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) (save_id_ != -1 && !is_success_ && !size)); 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) state_ = COMPLETE; 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) is_success_ = is_success; 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UpdateSize(size); 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Calculate the percentage of the save item 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int SaveItem::PercentComplete() const { 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) switch (state_) { 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) case COMPLETE: 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) case CANCELED: 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 100; 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) case WAIT_START: 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0; 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) case IN_PROGRESS: { 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int percent = 0; 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (total_bytes_ > 0) 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) percent = static_cast<int>(received_bytes_ * 100.0 / total_bytes_); 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return percent; 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) default: { 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTREACHED(); 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return -1; 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Rename the save item with new path. 1162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void SaveItem::Rename(const base::FilePath& full_path) { 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(!full_path.empty() && !has_final_name()); 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) full_path_ = full_path; 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) file_name_ = full_path_.BaseName(); 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) has_final_name_ = true; 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void SaveItem::SetSaveId(int32 save_id) { 124a93a17c8d99d686bd4a1511e5504e5e6cc9fcadfTorne (Richard Coles) DCHECK_EQ(-1, save_id_); 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) save_id_ = save_id; 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void SaveItem::SetTotalBytes(int64 total_bytes) { 129a93a17c8d99d686bd4a1511e5504e5e6cc9fcadfTorne (Richard Coles) DCHECK_EQ(0, total_bytes_); 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) total_bytes_ = total_bytes; 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace content 134