| OLD | NEW |
| 1 // Copyright (c) 2006-2008 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2006-2008 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/download/save_file.h" | 5 #include "chrome/browser/download/save_file.h" |
| 6 | 6 |
| 7 #include "base/basictypes.h" |
| 7 #include "base/file_util.h" | 8 #include "base/file_util.h" |
| 8 #include "base/logging.h" | 9 #include "base/logging.h" |
| 9 #include "base/path_service.h" | 10 #include "base/path_service.h" |
| 11 #include "base/string_util.h" |
| 10 #include "chrome/browser/download/save_types.h" | 12 #include "chrome/browser/download/save_types.h" |
| 13 #if defined(OS_WIN) |
| 11 #include "chrome/common/win_util.h" | 14 #include "chrome/common/win_util.h" |
| 12 #include "chrome/common/win_safe_util.h" | 15 #include "chrome/common/win_safe_util.h" |
| 16 #endif |
| 13 | 17 |
| 14 SaveFile::SaveFile(const SaveFileCreateInfo* info) | 18 SaveFile::SaveFile(const SaveFileCreateInfo* info) |
| 15 : info_(info), | 19 : info_(info), |
| 16 file_(NULL), | 20 file_(NULL), |
| 17 bytes_so_far_(0), | 21 bytes_so_far_(0), |
| 18 path_renamed_(false), | 22 path_renamed_(false), |
| 19 in_progress_(true) { | 23 in_progress_(true) { |
| 20 DCHECK(info); | 24 DCHECK(info); |
| 21 DCHECK(info->path.empty()); | 25 DCHECK(info->path.empty()); |
| 22 if (file_util::CreateTemporaryFileName(&full_path_)) | 26 if (file_util::CreateTemporaryFileName(&full_path_)) |
| 23 Open(L"wb"); | 27 Open("wb"); |
| 24 } | 28 } |
| 25 | 29 |
| 26 SaveFile::~SaveFile() { | 30 SaveFile::~SaveFile() { |
| 27 Close(); | 31 Close(); |
| 28 } | 32 } |
| 29 | 33 |
| 30 // Return false indicate that we got disk error, save file manager will tell | 34 // Return false indicate that we got disk error, save file manager will tell |
| 31 // SavePackage this error, then SavePackage will call its Cancel() method to | 35 // SavePackage this error, then SavePackage will call its Cancel() method to |
| 32 // cancel whole save job. | 36 // cancel whole save job. |
| 33 bool SaveFile::AppendDataToFile(const char* data, int data_len) { | 37 bool SaveFile::AppendDataToFile(const char* data, size_t data_len) { |
| 34 if (file_) { | 38 if (file_) { |
| 35 if (data_len == fwrite(data, 1, data_len, file_)) { | 39 if (data_len == fwrite(data, 1, data_len, file_)) { |
| 36 bytes_so_far_ += data_len; | 40 bytes_so_far_ += data_len; |
| 37 return true; | 41 return true; |
| 38 } else { | 42 } else { |
| 39 Close(); | 43 Close(); |
| 40 return false; | 44 return false; |
| 41 } | 45 } |
| 42 } | 46 } |
| 43 // No file_, treat it as disk error. | 47 // No file_, treat it as disk error. |
| 44 return false; | 48 return false; |
| 45 } | 49 } |
| 46 | 50 |
| 47 void SaveFile::Cancel() { | 51 void SaveFile::Cancel() { |
| 48 Close(); | 52 Close(); |
| 49 // If this job has been canceled, and it has created file, | 53 // If this job has been canceled, and it has created file, |
| 50 // We need to delete this created file. | 54 // We need to delete this created file. |
| 51 if (!full_path_.empty()) { | 55 if (!full_path_.empty()) { |
| 52 DeleteFile(full_path_.c_str()); | 56 file_util::Delete(full_path_, false); |
| 53 } | 57 } |
| 54 } | 58 } |
| 55 | 59 |
| 56 // Rename the file when we have final name. | 60 // Rename the file when we have final name. |
| 57 bool SaveFile::Rename(const std::wstring& new_path) { | 61 bool SaveFile::Rename(const std::wstring& new_path) { |
| 58 Close(); | 62 Close(); |
| 59 | 63 |
| 60 DCHECK(!path_renamed()); | 64 DCHECK(!path_renamed()); |
| 61 // We cannot rename because rename will keep the same security descriptor | 65 // We cannot rename because rename will keep the same security descriptor |
| 62 // on the destination file. We want to recreate the security descriptor | 66 // on the destination file. We want to recreate the security descriptor |
| 63 // with the security that makes sense in the new path. If the last parameter | 67 // with the security that makes sense in the new path. |
| 64 // is FALSE and the new file already exists, the function overwrites the | 68 if (!file_util::CopyFile(full_path_, new_path)) |
| 65 // existing file and succeeds. | |
| 66 if (!CopyFile(full_path_.c_str(), new_path.c_str(), FALSE)) | |
| 67 return false; | 69 return false; |
| 68 | 70 |
| 69 DeleteFile(full_path_.c_str()); | 71 file_util::Delete(full_path_, false); |
| 70 | 72 |
| 71 full_path_ = new_path; | 73 full_path_ = new_path; |
| 72 path_renamed_ = true; | 74 path_renamed_ = true; |
| 73 | 75 |
| 74 // Still in saving process, reopen the file. | 76 // Still in saving process, reopen the file. |
| 75 if (in_progress_ && !Open(L"a+b")) | 77 if (in_progress_ && !Open("a+b")) |
| 76 return false; | 78 return false; |
| 77 return true; | 79 return true; |
| 78 } | 80 } |
| 79 | 81 |
| 80 void SaveFile::Finish() { | 82 void SaveFile::Finish() { |
| 81 Close(); | 83 Close(); |
| 82 in_progress_ = false; | 84 in_progress_ = false; |
| 83 } | 85 } |
| 84 | 86 |
| 85 void SaveFile::Close() { | 87 void SaveFile::Close() { |
| 86 if (file_) { | 88 if (file_) { |
| 87 fclose(file_); | 89 file_util::CloseFile(file_); |
| 88 file_ = NULL; | 90 file_ = NULL; |
| 89 } | 91 } |
| 90 } | 92 } |
| 91 | 93 |
| 92 bool SaveFile::Open(const wchar_t* open_mode) { | 94 bool SaveFile::Open(const char* open_mode) { |
| 93 DCHECK(!full_path_.empty()); | 95 DCHECK(!full_path_.empty()); |
| 94 if (_wfopen_s(&file_, full_path_.c_str(), open_mode)) { | 96 file_ = file_util::OpenFile(full_path_, open_mode); |
| 95 file_ = NULL; | 97 if (!file_) { |
| 96 return false; | 98 return false; |
| 97 } | 99 } |
| 100 #if defined(OS_WIN) |
| 98 // Sets the zone to tell Windows that this file comes from the Internet. | 101 // Sets the zone to tell Windows that this file comes from the Internet. |
| 99 // We ignore the return value because a failure is not fatal. | 102 // We ignore the return value because a failure is not fatal. |
| 103 // TODO(port): Similarly mark on Mac. |
| 100 win_util::SetInternetZoneIdentifier(full_path_); | 104 win_util::SetInternetZoneIdentifier(full_path_); |
| 105 #endif |
| 101 return true; | 106 return true; |
| 102 } | 107 } |
| 103 | 108 |
| OLD | NEW |