Chromium Code Reviews| Index: chrome/browser/extensions/api/image_writer_private/destroy_partitions_operation.cc |
| diff --git a/chrome/browser/extensions/api/image_writer_private/destroy_partitions_operation.cc b/chrome/browser/extensions/api/image_writer_private/destroy_partitions_operation.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..a5f1b0fe7144a9af68e8a41d49495cad574eec98 |
| --- /dev/null |
| +++ b/chrome/browser/extensions/api/image_writer_private/destroy_partitions_operation.cc |
| @@ -0,0 +1,60 @@ |
| +// Copyright 2013 The Chromium Authors. All rights reserved. |
| +// Use of this source code is governed by a BSD-style license that can be |
| +// found in the LICENSE file. |
| + |
| +#include "base/file_util.h" |
| +#include "chrome/browser/extensions/api/image_writer_private/destroy_partitions_operation.h" |
| +#include "chrome/browser/extensions/api/image_writer_private/error_messages.h" |
| + |
| +namespace extensions { |
| +namespace image_writer { |
| + |
| +// Number of bytes for the maximum partition table size. By wiping this many |
| +// bytes we can essentially guarantee the header and associated information will |
| +// be wiped. See http://crbug.com/328246 for more information. |
| +const int kPartitionTableSize = 32 * 1024; |
| + |
| +namespace { |
| + |
| +// Small wrapper to remove the temporary file and display an error if it fails. |
|
tbarzic
2013/12/14 01:01:04
display error where?
If you meant DLOG part, I don
Drew Haven
2013/12/16 20:15:33
Good point. It's such a trivial function, but I f
|
| +void RemoveTempFile(const base::FilePath& path) { |
| + if (!DeleteFile(path, false)) |
| + DLOG(ERROR) << "Unable to delete temporary file: " << path.value(); |
| +} |
| + |
| +} |
|
tbarzic
2013/12/14 01:01:04
// namespace
Drew Haven
2013/12/16 20:15:33
Done.
|
| + |
| +DestroyPartitionsOperation::DestroyPartitionsOperation( |
| + base::WeakPtr<OperationManager> manager, |
| + const ExtensionId& extension_id, |
| + const std::string& storage_unit_id) |
| + : Operation(manager, extension_id, storage_unit_id) { |
|
tbarzic
2013/12/14 01:01:04
indent+=2
Drew Haven
2013/12/16 20:15:33
Done.
|
| + // Turn off the verification step. |
| + verify_write_ = false; |
|
tbarzic
2013/12/14 01:01:04
I'd add this to initializer list
Drew Haven
2013/12/16 20:15:33
It's a protected variable of the superclass, so I
tbarzic
2013/12/16 20:33:45
ok. Though, it may make sense to pass verify_write
Drew Haven
2013/12/16 22:05:25
I think we'll set that up when/if we need it.
|
| +} |
| + |
| +DestroyPartitionsOperation::~DestroyPartitionsOperation() { |
| +} |
|
tbarzic
2013/12/14 01:01:04
no need for a nl here
Drew Haven
2013/12/16 20:15:33
Done.
|
| + |
| +void DestroyPartitionsOperation::Start() { |
| + if (!base::CreateTemporaryFile(&image_path_)) { |
| + Error(error::kTempFile); |
| + return; |
| + } |
| + |
| + AddCleanUpFunction(base::Bind(&RemoveTempFile, image_path_)); |
| + |
| + char buffer[kPartitionTableSize]; |
|
tbarzic
2013/12/14 01:01:04
I'd use scoped_array<char> here (32kB is not that
Drew Haven
2013/12/16 20:15:33
I'm not finding scoped_arrays in base. Though the
|
| + memset(buffer, 0, kPartitionTableSize); |
| + |
| + if (file_util::WriteFile(image_path_, buffer, kPartitionTableSize) != |
| + kPartitionTableSize) { |
| + Error(error::kTempFile); |
| + return; |
| + } |
| + |
| + WriteStart(); |
| +} |
| + |
| +} // namespace image_writer |
| +} // namespace extensions |