Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(291)

Side by Side Diff: webkit/browser/fileapi/copy_or_move_operation_delegate.cc

Issue 442383002: Move storage-related files from webkit/ to new top-level directory storage/ (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: . Created 6 years, 4 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
(Empty)
1 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "webkit/browser/fileapi/copy_or_move_operation_delegate.h"
6
7 #include "base/bind.h"
8 #include "base/files/file_path.h"
9 #include "net/base/io_buffer.h"
10 #include "net/base/net_errors.h"
11 #include "webkit/browser/blob/file_stream_reader.h"
12 #include "webkit/browser/fileapi/copy_or_move_file_validator.h"
13 #include "webkit/browser/fileapi/file_stream_writer.h"
14 #include "webkit/browser/fileapi/file_system_context.h"
15 #include "webkit/browser/fileapi/file_system_operation_runner.h"
16 #include "webkit/browser/fileapi/file_system_url.h"
17 #include "webkit/browser/fileapi/recursive_operation_delegate.h"
18 #include "webkit/common/blob/shareable_file_reference.h"
19 #include "webkit/common/fileapi/file_system_util.h"
20
21 namespace fileapi {
22
23 const int64 kFlushIntervalInBytes = 10 << 20; // 10MB.
24
25 class CopyOrMoveOperationDelegate::CopyOrMoveImpl {
26 public:
27 virtual ~CopyOrMoveImpl() {}
28 virtual void Run(
29 const CopyOrMoveOperationDelegate::StatusCallback& callback) = 0;
30 virtual void Cancel() = 0;
31
32 protected:
33 CopyOrMoveImpl() {}
34
35 private:
36 DISALLOW_COPY_AND_ASSIGN(CopyOrMoveImpl);
37 };
38
39 namespace {
40
41 // Copies a file on a (same) file system. Just delegate the operation to
42 // |operation_runner|.
43 class CopyOrMoveOnSameFileSystemImpl
44 : public CopyOrMoveOperationDelegate::CopyOrMoveImpl {
45 public:
46 CopyOrMoveOnSameFileSystemImpl(
47 FileSystemOperationRunner* operation_runner,
48 CopyOrMoveOperationDelegate::OperationType operation_type,
49 const FileSystemURL& src_url,
50 const FileSystemURL& dest_url,
51 CopyOrMoveOperationDelegate::CopyOrMoveOption option,
52 const FileSystemOperation::CopyFileProgressCallback&
53 file_progress_callback)
54 : operation_runner_(operation_runner),
55 operation_type_(operation_type),
56 src_url_(src_url),
57 dest_url_(dest_url),
58 option_(option),
59 file_progress_callback_(file_progress_callback) {
60 }
61
62 virtual void Run(
63 const CopyOrMoveOperationDelegate::StatusCallback& callback) OVERRIDE {
64 if (operation_type_ == CopyOrMoveOperationDelegate::OPERATION_MOVE) {
65 operation_runner_->MoveFileLocal(src_url_, dest_url_, option_, callback);
66 } else {
67 operation_runner_->CopyFileLocal(
68 src_url_, dest_url_, option_, file_progress_callback_, callback);
69 }
70 }
71
72 virtual void Cancel() OVERRIDE {
73 // We can do nothing for the copy/move operation on a local file system.
74 // Assuming the operation is quickly done, it should be ok to just wait
75 // for the completion.
76 }
77
78 private:
79 FileSystemOperationRunner* operation_runner_;
80 CopyOrMoveOperationDelegate::OperationType operation_type_;
81 FileSystemURL src_url_;
82 FileSystemURL dest_url_;
83 CopyOrMoveOperationDelegate::CopyOrMoveOption option_;
84 FileSystemOperation::CopyFileProgressCallback file_progress_callback_;
85 DISALLOW_COPY_AND_ASSIGN(CopyOrMoveOnSameFileSystemImpl);
86 };
87
88 // Specifically for cross file system copy/move operation, this class creates
89 // a snapshot file, validates it if necessary, runs copying process,
90 // validates the created file, and removes source file for move (noop for
91 // copy).
92 class SnapshotCopyOrMoveImpl
93 : public CopyOrMoveOperationDelegate::CopyOrMoveImpl {
94 public:
95 SnapshotCopyOrMoveImpl(
96 FileSystemOperationRunner* operation_runner,
97 CopyOrMoveOperationDelegate::OperationType operation_type,
98 const FileSystemURL& src_url,
99 const FileSystemURL& dest_url,
100 CopyOrMoveOperationDelegate::CopyOrMoveOption option,
101 CopyOrMoveFileValidatorFactory* validator_factory,
102 const FileSystemOperation::CopyFileProgressCallback&
103 file_progress_callback)
104 : operation_runner_(operation_runner),
105 operation_type_(operation_type),
106 src_url_(src_url),
107 dest_url_(dest_url),
108 option_(option),
109 validator_factory_(validator_factory),
110 file_progress_callback_(file_progress_callback),
111 cancel_requested_(false),
112 weak_factory_(this) {
113 }
114
115 virtual void Run(
116 const CopyOrMoveOperationDelegate::StatusCallback& callback) OVERRIDE {
117 file_progress_callback_.Run(0);
118 operation_runner_->CreateSnapshotFile(
119 src_url_,
120 base::Bind(&SnapshotCopyOrMoveImpl::RunAfterCreateSnapshot,
121 weak_factory_.GetWeakPtr(), callback));
122 }
123
124 virtual void Cancel() OVERRIDE {
125 cancel_requested_ = true;
126 }
127
128 private:
129 void RunAfterCreateSnapshot(
130 const CopyOrMoveOperationDelegate::StatusCallback& callback,
131 base::File::Error error,
132 const base::File::Info& file_info,
133 const base::FilePath& platform_path,
134 const scoped_refptr<webkit_blob::ShareableFileReference>& file_ref) {
135 if (cancel_requested_)
136 error = base::File::FILE_ERROR_ABORT;
137
138 if (error != base::File::FILE_OK) {
139 callback.Run(error);
140 return;
141 }
142
143 // For now we assume CreateSnapshotFile always return a valid local file
144 // path.
145 DCHECK(!platform_path.empty());
146
147 if (!validator_factory_) {
148 // No validation is needed.
149 RunAfterPreWriteValidation(platform_path, file_info, file_ref, callback,
150 base::File::FILE_OK);
151 return;
152 }
153
154 // Run pre write validation.
155 PreWriteValidation(
156 platform_path,
157 base::Bind(&SnapshotCopyOrMoveImpl::RunAfterPreWriteValidation,
158 weak_factory_.GetWeakPtr(),
159 platform_path, file_info, file_ref, callback));
160 }
161
162 void RunAfterPreWriteValidation(
163 const base::FilePath& platform_path,
164 const base::File::Info& file_info,
165 const scoped_refptr<webkit_blob::ShareableFileReference>& file_ref,
166 const CopyOrMoveOperationDelegate::StatusCallback& callback,
167 base::File::Error error) {
168 if (cancel_requested_)
169 error = base::File::FILE_ERROR_ABORT;
170
171 if (error != base::File::FILE_OK) {
172 callback.Run(error);
173 return;
174 }
175
176 // |file_ref| is unused but necessary to keep the file alive until
177 // CopyInForeignFile() is completed.
178 operation_runner_->CopyInForeignFile(
179 platform_path, dest_url_,
180 base::Bind(&SnapshotCopyOrMoveImpl::RunAfterCopyInForeignFile,
181 weak_factory_.GetWeakPtr(), file_info, file_ref, callback));
182 }
183
184 void RunAfterCopyInForeignFile(
185 const base::File::Info& file_info,
186 const scoped_refptr<webkit_blob::ShareableFileReference>& file_ref,
187 const CopyOrMoveOperationDelegate::StatusCallback& callback,
188 base::File::Error error) {
189 if (cancel_requested_)
190 error = base::File::FILE_ERROR_ABORT;
191
192 if (error != base::File::FILE_OK) {
193 callback.Run(error);
194 return;
195 }
196
197 file_progress_callback_.Run(file_info.size);
198
199 if (option_ == FileSystemOperation::OPTION_NONE) {
200 RunAfterTouchFile(callback, base::File::FILE_OK);
201 return;
202 }
203
204 operation_runner_->TouchFile(
205 dest_url_, base::Time::Now() /* last_access */,
206 file_info.last_modified,
207 base::Bind(&SnapshotCopyOrMoveImpl::RunAfterTouchFile,
208 weak_factory_.GetWeakPtr(), callback));
209 }
210
211 void RunAfterTouchFile(
212 const CopyOrMoveOperationDelegate::StatusCallback& callback,
213 base::File::Error error) {
214 // Even if TouchFile is failed, just ignore it.
215
216 if (cancel_requested_) {
217 callback.Run(base::File::FILE_ERROR_ABORT);
218 return;
219 }
220
221 // |validator_| is NULL when the destination filesystem does not do
222 // validation.
223 if (!validator_) {
224 // No validation is needed.
225 RunAfterPostWriteValidation(callback, base::File::FILE_OK);
226 return;
227 }
228
229 PostWriteValidation(
230 base::Bind(&SnapshotCopyOrMoveImpl::RunAfterPostWriteValidation,
231 weak_factory_.GetWeakPtr(), callback));
232 }
233
234 void RunAfterPostWriteValidation(
235 const CopyOrMoveOperationDelegate::StatusCallback& callback,
236 base::File::Error error) {
237 if (cancel_requested_) {
238 callback.Run(base::File::FILE_ERROR_ABORT);
239 return;
240 }
241
242 if (error != base::File::FILE_OK) {
243 // Failed to validate. Remove the destination file.
244 operation_runner_->Remove(
245 dest_url_, true /* recursive */,
246 base::Bind(&SnapshotCopyOrMoveImpl::DidRemoveDestForError,
247 weak_factory_.GetWeakPtr(), error, callback));
248 return;
249 }
250
251 if (operation_type_ == CopyOrMoveOperationDelegate::OPERATION_COPY) {
252 callback.Run(base::File::FILE_OK);
253 return;
254 }
255
256 DCHECK_EQ(CopyOrMoveOperationDelegate::OPERATION_MOVE, operation_type_);
257
258 // Remove the source for finalizing move operation.
259 operation_runner_->Remove(
260 src_url_, true /* recursive */,
261 base::Bind(&SnapshotCopyOrMoveImpl::RunAfterRemoveSourceForMove,
262 weak_factory_.GetWeakPtr(), callback));
263 }
264
265 void RunAfterRemoveSourceForMove(
266 const CopyOrMoveOperationDelegate::StatusCallback& callback,
267 base::File::Error error) {
268 if (cancel_requested_)
269 error = base::File::FILE_ERROR_ABORT;
270
271 if (error == base::File::FILE_ERROR_NOT_FOUND)
272 error = base::File::FILE_OK;
273 callback.Run(error);
274 }
275
276 void DidRemoveDestForError(
277 base::File::Error prior_error,
278 const CopyOrMoveOperationDelegate::StatusCallback& callback,
279 base::File::Error error) {
280 if (error != base::File::FILE_OK) {
281 VLOG(1) << "Error removing destination file after validation error: "
282 << error;
283 }
284 callback.Run(prior_error);
285 }
286
287 // Runs pre-write validation.
288 void PreWriteValidation(
289 const base::FilePath& platform_path,
290 const CopyOrMoveOperationDelegate::StatusCallback& callback) {
291 DCHECK(validator_factory_);
292 validator_.reset(
293 validator_factory_->CreateCopyOrMoveFileValidator(
294 src_url_, platform_path));
295 validator_->StartPreWriteValidation(callback);
296 }
297
298 // Runs post-write validation.
299 void PostWriteValidation(
300 const CopyOrMoveOperationDelegate::StatusCallback& callback) {
301 operation_runner_->CreateSnapshotFile(
302 dest_url_,
303 base::Bind(
304 &SnapshotCopyOrMoveImpl::PostWriteValidationAfterCreateSnapshotFile,
305 weak_factory_.GetWeakPtr(), callback));
306 }
307
308 void PostWriteValidationAfterCreateSnapshotFile(
309 const CopyOrMoveOperationDelegate::StatusCallback& callback,
310 base::File::Error error,
311 const base::File::Info& file_info,
312 const base::FilePath& platform_path,
313 const scoped_refptr<webkit_blob::ShareableFileReference>& file_ref) {
314 if (cancel_requested_)
315 error = base::File::FILE_ERROR_ABORT;
316
317 if (error != base::File::FILE_OK) {
318 callback.Run(error);
319 return;
320 }
321
322 DCHECK(validator_);
323 // Note: file_ref passed here to keep the file alive until after
324 // the StartPostWriteValidation operation finishes.
325 validator_->StartPostWriteValidation(
326 platform_path,
327 base::Bind(&SnapshotCopyOrMoveImpl::DidPostWriteValidation,
328 weak_factory_.GetWeakPtr(), file_ref, callback));
329 }
330
331 // |file_ref| is unused; it is passed here to make sure the reference is
332 // alive until after post-write validation is complete.
333 void DidPostWriteValidation(
334 const scoped_refptr<webkit_blob::ShareableFileReference>& file_ref,
335 const CopyOrMoveOperationDelegate::StatusCallback& callback,
336 base::File::Error error) {
337 callback.Run(error);
338 }
339
340 FileSystemOperationRunner* operation_runner_;
341 CopyOrMoveOperationDelegate::OperationType operation_type_;
342 FileSystemURL src_url_;
343 FileSystemURL dest_url_;
344
345 CopyOrMoveOperationDelegate::CopyOrMoveOption option_;
346 CopyOrMoveFileValidatorFactory* validator_factory_;
347 scoped_ptr<CopyOrMoveFileValidator> validator_;
348 FileSystemOperation::CopyFileProgressCallback file_progress_callback_;
349 bool cancel_requested_;
350 base::WeakPtrFactory<SnapshotCopyOrMoveImpl> weak_factory_;
351 DISALLOW_COPY_AND_ASSIGN(SnapshotCopyOrMoveImpl);
352 };
353
354 // The size of buffer for StreamCopyHelper.
355 const int kReadBufferSize = 32768;
356
357 // To avoid too many progress callbacks, it should be called less
358 // frequently than 50ms.
359 const int kMinProgressCallbackInvocationSpanInMilliseconds = 50;
360
361 // Specifically for cross file system copy/move operation, this class uses
362 // stream reader and writer for copying. Validator is not supported, so if
363 // necessary SnapshotCopyOrMoveImpl should be used.
364 class StreamCopyOrMoveImpl
365 : public CopyOrMoveOperationDelegate::CopyOrMoveImpl {
366 public:
367 StreamCopyOrMoveImpl(
368 FileSystemOperationRunner* operation_runner,
369 CopyOrMoveOperationDelegate::OperationType operation_type,
370 const FileSystemURL& src_url,
371 const FileSystemURL& dest_url,
372 CopyOrMoveOperationDelegate::CopyOrMoveOption option,
373 scoped_ptr<webkit_blob::FileStreamReader> reader,
374 scoped_ptr<FileStreamWriter> writer,
375 const FileSystemOperation::CopyFileProgressCallback&
376 file_progress_callback)
377 : operation_runner_(operation_runner),
378 operation_type_(operation_type),
379 src_url_(src_url),
380 dest_url_(dest_url),
381 option_(option),
382 reader_(reader.Pass()),
383 writer_(writer.Pass()),
384 file_progress_callback_(file_progress_callback),
385 cancel_requested_(false),
386 weak_factory_(this) {
387 }
388
389 virtual void Run(
390 const CopyOrMoveOperationDelegate::StatusCallback& callback) OVERRIDE {
391 // Reader can be created even if the entry does not exist or the entry is
392 // a directory. To check errors before destination file creation,
393 // check metadata first.
394 operation_runner_->GetMetadata(
395 src_url_,
396 base::Bind(&StreamCopyOrMoveImpl::RunAfterGetMetadataForSource,
397 weak_factory_.GetWeakPtr(), callback));
398 }
399
400 virtual void Cancel() OVERRIDE {
401 cancel_requested_ = true;
402 if (copy_helper_)
403 copy_helper_->Cancel();
404 }
405
406 private:
407 void RunAfterGetMetadataForSource(
408 const CopyOrMoveOperationDelegate::StatusCallback& callback,
409 base::File::Error error,
410 const base::File::Info& file_info) {
411 if (cancel_requested_)
412 error = base::File::FILE_ERROR_ABORT;
413
414 if (error != base::File::FILE_OK) {
415 callback.Run(error);
416 return;
417 }
418
419 if (file_info.is_directory) {
420 // If not a directory, failed with appropriate error code.
421 callback.Run(base::File::FILE_ERROR_NOT_A_FILE);
422 return;
423 }
424
425 // To use FileStreamWriter, we need to ensure the destination file exists.
426 operation_runner_->CreateFile(
427 dest_url_, false /* exclusive */,
428 base::Bind(&StreamCopyOrMoveImpl::RunAfterCreateFileForDestination,
429 weak_factory_.GetWeakPtr(),
430 callback, file_info.last_modified));
431 }
432
433 void RunAfterCreateFileForDestination(
434 const CopyOrMoveOperationDelegate::StatusCallback& callback,
435 const base::Time& last_modified,
436 base::File::Error error) {
437 if (cancel_requested_)
438 error = base::File::FILE_ERROR_ABORT;
439
440 if (error != base::File::FILE_OK) {
441 callback.Run(error);
442 return;
443 }
444
445 const bool need_flush = dest_url_.mount_option().copy_sync_option() ==
446 fileapi::COPY_SYNC_OPTION_SYNC;
447
448 DCHECK(!copy_helper_);
449 copy_helper_.reset(
450 new CopyOrMoveOperationDelegate::StreamCopyHelper(
451 reader_.Pass(), writer_.Pass(),
452 need_flush,
453 kReadBufferSize,
454 file_progress_callback_,
455 base::TimeDelta::FromMilliseconds(
456 kMinProgressCallbackInvocationSpanInMilliseconds)));
457 copy_helper_->Run(
458 base::Bind(&StreamCopyOrMoveImpl::RunAfterStreamCopy,
459 weak_factory_.GetWeakPtr(), callback, last_modified));
460 }
461
462 void RunAfterStreamCopy(
463 const CopyOrMoveOperationDelegate::StatusCallback& callback,
464 const base::Time& last_modified,
465 base::File::Error error) {
466 if (cancel_requested_)
467 error = base::File::FILE_ERROR_ABORT;
468
469 if (error != base::File::FILE_OK) {
470 callback.Run(error);
471 return;
472 }
473
474 if (option_ == FileSystemOperation::OPTION_NONE) {
475 RunAfterTouchFile(callback, base::File::FILE_OK);
476 return;
477 }
478
479 operation_runner_->TouchFile(
480 dest_url_, base::Time::Now() /* last_access */, last_modified,
481 base::Bind(&StreamCopyOrMoveImpl::RunAfterTouchFile,
482 weak_factory_.GetWeakPtr(), callback));
483 }
484
485 void RunAfterTouchFile(
486 const CopyOrMoveOperationDelegate::StatusCallback& callback,
487 base::File::Error error) {
488 // Even if TouchFile is failed, just ignore it.
489 if (cancel_requested_) {
490 callback.Run(base::File::FILE_ERROR_ABORT);
491 return;
492 }
493
494 if (operation_type_ == CopyOrMoveOperationDelegate::OPERATION_COPY) {
495 callback.Run(base::File::FILE_OK);
496 return;
497 }
498
499 DCHECK_EQ(CopyOrMoveOperationDelegate::OPERATION_MOVE, operation_type_);
500
501 // Remove the source for finalizing move operation.
502 operation_runner_->Remove(
503 src_url_, false /* recursive */,
504 base::Bind(&StreamCopyOrMoveImpl::RunAfterRemoveForMove,
505 weak_factory_.GetWeakPtr(), callback));
506 }
507
508 void RunAfterRemoveForMove(
509 const CopyOrMoveOperationDelegate::StatusCallback& callback,
510 base::File::Error error) {
511 if (cancel_requested_)
512 error = base::File::FILE_ERROR_ABORT;
513 if (error == base::File::FILE_ERROR_NOT_FOUND)
514 error = base::File::FILE_OK;
515 callback.Run(error);
516 }
517
518 FileSystemOperationRunner* operation_runner_;
519 CopyOrMoveOperationDelegate::OperationType operation_type_;
520 FileSystemURL src_url_;
521 FileSystemURL dest_url_;
522 CopyOrMoveOperationDelegate::CopyOrMoveOption option_;
523 scoped_ptr<webkit_blob::FileStreamReader> reader_;
524 scoped_ptr<FileStreamWriter> writer_;
525 FileSystemOperation::CopyFileProgressCallback file_progress_callback_;
526 scoped_ptr<CopyOrMoveOperationDelegate::StreamCopyHelper> copy_helper_;
527 bool cancel_requested_;
528 base::WeakPtrFactory<StreamCopyOrMoveImpl> weak_factory_;
529 DISALLOW_COPY_AND_ASSIGN(StreamCopyOrMoveImpl);
530 };
531
532 } // namespace
533
534 CopyOrMoveOperationDelegate::StreamCopyHelper::StreamCopyHelper(
535 scoped_ptr<webkit_blob::FileStreamReader> reader,
536 scoped_ptr<FileStreamWriter> writer,
537 bool need_flush,
538 int buffer_size,
539 const FileSystemOperation::CopyFileProgressCallback&
540 file_progress_callback,
541 const base::TimeDelta& min_progress_callback_invocation_span)
542 : reader_(reader.Pass()),
543 writer_(writer.Pass()),
544 need_flush_(need_flush),
545 file_progress_callback_(file_progress_callback),
546 io_buffer_(new net::IOBufferWithSize(buffer_size)),
547 num_copied_bytes_(0),
548 previous_flush_offset_(0),
549 min_progress_callback_invocation_span_(
550 min_progress_callback_invocation_span),
551 cancel_requested_(false),
552 weak_factory_(this) {
553 }
554
555 CopyOrMoveOperationDelegate::StreamCopyHelper::~StreamCopyHelper() {
556 }
557
558 void CopyOrMoveOperationDelegate::StreamCopyHelper::Run(
559 const StatusCallback& callback) {
560 file_progress_callback_.Run(0);
561 last_progress_callback_invocation_time_ = base::Time::Now();
562 Read(callback);
563 }
564
565 void CopyOrMoveOperationDelegate::StreamCopyHelper::Cancel() {
566 cancel_requested_ = true;
567 }
568
569 void CopyOrMoveOperationDelegate::StreamCopyHelper::Read(
570 const StatusCallback& callback) {
571 int result = reader_->Read(
572 io_buffer_.get(), io_buffer_->size(),
573 base::Bind(&StreamCopyHelper::DidRead,
574 weak_factory_.GetWeakPtr(), callback));
575 if (result != net::ERR_IO_PENDING)
576 DidRead(callback, result);
577 }
578
579 void CopyOrMoveOperationDelegate::StreamCopyHelper::DidRead(
580 const StatusCallback& callback, int result) {
581 if (cancel_requested_) {
582 callback.Run(base::File::FILE_ERROR_ABORT);
583 return;
584 }
585
586 if (result < 0) {
587 callback.Run(NetErrorToFileError(result));
588 return;
589 }
590
591 if (result == 0) {
592 // Here is the EOF.
593 if (need_flush_)
594 Flush(callback, true /* is_eof */);
595 else
596 callback.Run(base::File::FILE_OK);
597 return;
598 }
599
600 Write(callback, new net::DrainableIOBuffer(io_buffer_.get(), result));
601 }
602
603 void CopyOrMoveOperationDelegate::StreamCopyHelper::Write(
604 const StatusCallback& callback,
605 scoped_refptr<net::DrainableIOBuffer> buffer) {
606 DCHECK_GT(buffer->BytesRemaining(), 0);
607
608 int result = writer_->Write(
609 buffer.get(), buffer->BytesRemaining(),
610 base::Bind(&StreamCopyHelper::DidWrite,
611 weak_factory_.GetWeakPtr(), callback, buffer));
612 if (result != net::ERR_IO_PENDING)
613 DidWrite(callback, buffer, result);
614 }
615
616 void CopyOrMoveOperationDelegate::StreamCopyHelper::DidWrite(
617 const StatusCallback& callback,
618 scoped_refptr<net::DrainableIOBuffer> buffer,
619 int result) {
620 if (cancel_requested_) {
621 callback.Run(base::File::FILE_ERROR_ABORT);
622 return;
623 }
624
625 if (result < 0) {
626 callback.Run(NetErrorToFileError(result));
627 return;
628 }
629
630 buffer->DidConsume(result);
631 num_copied_bytes_ += result;
632
633 // Check the elapsed time since last |file_progress_callback_| invocation.
634 base::Time now = base::Time::Now();
635 if (now - last_progress_callback_invocation_time_ >=
636 min_progress_callback_invocation_span_) {
637 file_progress_callback_.Run(num_copied_bytes_);
638 last_progress_callback_invocation_time_ = now;
639 }
640
641 if (buffer->BytesRemaining() > 0) {
642 Write(callback, buffer);
643 return;
644 }
645
646 if (need_flush_ &&
647 (num_copied_bytes_ - previous_flush_offset_) > kFlushIntervalInBytes) {
648 Flush(callback, false /* not is_eof */);
649 } else {
650 Read(callback);
651 }
652 }
653
654 void CopyOrMoveOperationDelegate::StreamCopyHelper::Flush(
655 const StatusCallback& callback, bool is_eof) {
656 int result = writer_->Flush(
657 base::Bind(&StreamCopyHelper::DidFlush,
658 weak_factory_.GetWeakPtr(), callback, is_eof));
659 if (result != net::ERR_IO_PENDING)
660 DidFlush(callback, is_eof, result);
661 }
662
663 void CopyOrMoveOperationDelegate::StreamCopyHelper::DidFlush(
664 const StatusCallback& callback, bool is_eof, int result) {
665 if (cancel_requested_) {
666 callback.Run(base::File::FILE_ERROR_ABORT);
667 return;
668 }
669
670 previous_flush_offset_ = num_copied_bytes_;
671 if (is_eof)
672 callback.Run(NetErrorToFileError(result));
673 else
674 Read(callback);
675 }
676
677 CopyOrMoveOperationDelegate::CopyOrMoveOperationDelegate(
678 FileSystemContext* file_system_context,
679 const FileSystemURL& src_root,
680 const FileSystemURL& dest_root,
681 OperationType operation_type,
682 CopyOrMoveOption option,
683 const CopyProgressCallback& progress_callback,
684 const StatusCallback& callback)
685 : RecursiveOperationDelegate(file_system_context),
686 src_root_(src_root),
687 dest_root_(dest_root),
688 operation_type_(operation_type),
689 option_(option),
690 progress_callback_(progress_callback),
691 callback_(callback),
692 weak_factory_(this) {
693 same_file_system_ = src_root_.IsInSameFileSystem(dest_root_);
694 }
695
696 CopyOrMoveOperationDelegate::~CopyOrMoveOperationDelegate() {
697 STLDeleteElements(&running_copy_set_);
698 }
699
700 void CopyOrMoveOperationDelegate::Run() {
701 // Not supported; this should never be called.
702 NOTREACHED();
703 }
704
705 void CopyOrMoveOperationDelegate::RunRecursively() {
706 // Perform light-weight checks first.
707
708 // It is an error to try to copy/move an entry into its child.
709 if (same_file_system_ && src_root_.path().IsParent(dest_root_.path())) {
710 callback_.Run(base::File::FILE_ERROR_INVALID_OPERATION);
711 return;
712 }
713
714 if (same_file_system_ && src_root_.path() == dest_root_.path()) {
715 // In JS API this should return error, but we return success because Pepper
716 // wants to return success and we have a code path that returns error in
717 // Blink for JS (http://crbug.com/329517).
718 callback_.Run(base::File::FILE_OK);
719 return;
720 }
721
722 // Start to process the source directory recursively.
723 // TODO(kinuko): This could be too expensive for same_file_system_==true
724 // and operation==MOVE case, probably we can just rename the root directory.
725 // http://crbug.com/172187
726 StartRecursiveOperation(src_root_, callback_);
727 }
728
729 void CopyOrMoveOperationDelegate::ProcessFile(
730 const FileSystemURL& src_url,
731 const StatusCallback& callback) {
732 if (!progress_callback_.is_null()) {
733 progress_callback_.Run(
734 FileSystemOperation::BEGIN_COPY_ENTRY, src_url, FileSystemURL(), 0);
735 }
736
737 FileSystemURL dest_url = CreateDestURL(src_url);
738 CopyOrMoveImpl* impl = NULL;
739 if (same_file_system_) {
740 impl = new CopyOrMoveOnSameFileSystemImpl(
741 operation_runner(), operation_type_, src_url, dest_url, option_,
742 base::Bind(&CopyOrMoveOperationDelegate::OnCopyFileProgress,
743 weak_factory_.GetWeakPtr(), src_url));
744 } else {
745 // Cross filesystem case.
746 base::File::Error error = base::File::FILE_ERROR_FAILED;
747 CopyOrMoveFileValidatorFactory* validator_factory =
748 file_system_context()->GetCopyOrMoveFileValidatorFactory(
749 dest_root_.type(), &error);
750 if (error != base::File::FILE_OK) {
751 callback.Run(error);
752 return;
753 }
754
755 if (!validator_factory) {
756 scoped_ptr<webkit_blob::FileStreamReader> reader =
757 file_system_context()->CreateFileStreamReader(
758 src_url, 0, base::Time());
759 scoped_ptr<FileStreamWriter> writer =
760 file_system_context()->CreateFileStreamWriter(dest_url, 0);
761 if (reader && writer) {
762 impl = new StreamCopyOrMoveImpl(
763 operation_runner(), operation_type_, src_url, dest_url, option_,
764 reader.Pass(), writer.Pass(),
765 base::Bind(&CopyOrMoveOperationDelegate::OnCopyFileProgress,
766 weak_factory_.GetWeakPtr(), src_url));
767 }
768 }
769
770 if (!impl) {
771 impl = new SnapshotCopyOrMoveImpl(
772 operation_runner(), operation_type_, src_url, dest_url, option_,
773 validator_factory,
774 base::Bind(&CopyOrMoveOperationDelegate::OnCopyFileProgress,
775 weak_factory_.GetWeakPtr(), src_url));
776 }
777 }
778
779 // Register the running task.
780 running_copy_set_.insert(impl);
781 impl->Run(base::Bind(
782 &CopyOrMoveOperationDelegate::DidCopyOrMoveFile,
783 weak_factory_.GetWeakPtr(), src_url, dest_url, callback, impl));
784 }
785
786 void CopyOrMoveOperationDelegate::ProcessDirectory(
787 const FileSystemURL& src_url,
788 const StatusCallback& callback) {
789 if (src_url == src_root_) {
790 // The src_root_ looks to be a directory.
791 // Try removing the dest_root_ to see if it exists and/or it is an
792 // empty directory.
793 // We do not invoke |progress_callback_| for source root, because it is
794 // already called in ProcessFile().
795 operation_runner()->RemoveDirectory(
796 dest_root_,
797 base::Bind(&CopyOrMoveOperationDelegate::DidTryRemoveDestRoot,
798 weak_factory_.GetWeakPtr(), callback));
799 return;
800 }
801
802 if (!progress_callback_.is_null()) {
803 progress_callback_.Run(
804 FileSystemOperation::BEGIN_COPY_ENTRY, src_url, FileSystemURL(), 0);
805 }
806
807 ProcessDirectoryInternal(src_url, CreateDestURL(src_url), callback);
808 }
809
810 void CopyOrMoveOperationDelegate::PostProcessDirectory(
811 const FileSystemURL& src_url,
812 const StatusCallback& callback) {
813 if (option_ == FileSystemOperation::OPTION_NONE) {
814 PostProcessDirectoryAfterTouchFile(
815 src_url, callback, base::File::FILE_OK);
816 return;
817 }
818
819 operation_runner()->GetMetadata(
820 src_url,
821 base::Bind(
822 &CopyOrMoveOperationDelegate::PostProcessDirectoryAfterGetMetadata,
823 weak_factory_.GetWeakPtr(), src_url, callback));
824 }
825
826 void CopyOrMoveOperationDelegate::OnCancel() {
827 // Request to cancel all running Copy/Move file.
828 for (std::set<CopyOrMoveImpl*>::iterator iter = running_copy_set_.begin();
829 iter != running_copy_set_.end(); ++iter)
830 (*iter)->Cancel();
831 }
832
833 void CopyOrMoveOperationDelegate::DidCopyOrMoveFile(
834 const FileSystemURL& src_url,
835 const FileSystemURL& dest_url,
836 const StatusCallback& callback,
837 CopyOrMoveImpl* impl,
838 base::File::Error error) {
839 running_copy_set_.erase(impl);
840 delete impl;
841
842 if (!progress_callback_.is_null() && error == base::File::FILE_OK) {
843 progress_callback_.Run(
844 FileSystemOperation::END_COPY_ENTRY, src_url, dest_url, 0);
845 }
846
847 callback.Run(error);
848 }
849
850 void CopyOrMoveOperationDelegate::DidTryRemoveDestRoot(
851 const StatusCallback& callback,
852 base::File::Error error) {
853 if (error == base::File::FILE_ERROR_NOT_A_DIRECTORY) {
854 callback_.Run(base::File::FILE_ERROR_INVALID_OPERATION);
855 return;
856 }
857 if (error != base::File::FILE_OK &&
858 error != base::File::FILE_ERROR_NOT_FOUND) {
859 callback_.Run(error);
860 return;
861 }
862
863 ProcessDirectoryInternal(src_root_, dest_root_, callback);
864 }
865
866 void CopyOrMoveOperationDelegate::ProcessDirectoryInternal(
867 const FileSystemURL& src_url,
868 const FileSystemURL& dest_url,
869 const StatusCallback& callback) {
870 // If operation_type == Move we may need to record directories and
871 // restore directory timestamps in the end, though it may have
872 // negative performance impact.
873 // See http://crbug.com/171284 for more details.
874 operation_runner()->CreateDirectory(
875 dest_url, false /* exclusive */, false /* recursive */,
876 base::Bind(&CopyOrMoveOperationDelegate::DidCreateDirectory,
877 weak_factory_.GetWeakPtr(), src_url, dest_url, callback));
878 }
879
880 void CopyOrMoveOperationDelegate::DidCreateDirectory(
881 const FileSystemURL& src_url,
882 const FileSystemURL& dest_url,
883 const StatusCallback& callback,
884 base::File::Error error) {
885 if (!progress_callback_.is_null() && error == base::File::FILE_OK) {
886 progress_callback_.Run(
887 FileSystemOperation::END_COPY_ENTRY, src_url, dest_url, 0);
888 }
889
890 callback.Run(error);
891 }
892
893 void CopyOrMoveOperationDelegate::PostProcessDirectoryAfterGetMetadata(
894 const FileSystemURL& src_url,
895 const StatusCallback& callback,
896 base::File::Error error,
897 const base::File::Info& file_info) {
898 if (error != base::File::FILE_OK) {
899 // Ignore the error, and run post process which should run after TouchFile.
900 PostProcessDirectoryAfterTouchFile(
901 src_url, callback, base::File::FILE_OK);
902 return;
903 }
904
905 operation_runner()->TouchFile(
906 CreateDestURL(src_url), base::Time::Now() /* last access */,
907 file_info.last_modified,
908 base::Bind(
909 &CopyOrMoveOperationDelegate::PostProcessDirectoryAfterTouchFile,
910 weak_factory_.GetWeakPtr(), src_url, callback));
911 }
912
913 void CopyOrMoveOperationDelegate::PostProcessDirectoryAfterTouchFile(
914 const FileSystemURL& src_url,
915 const StatusCallback& callback,
916 base::File::Error error) {
917 // Even if the TouchFile is failed, just ignore it.
918
919 if (operation_type_ == OPERATION_COPY) {
920 callback.Run(base::File::FILE_OK);
921 return;
922 }
923
924 DCHECK_EQ(OPERATION_MOVE, operation_type_);
925
926 // All files and subdirectories in the directory should be moved here,
927 // so remove the source directory for finalizing move operation.
928 operation_runner()->Remove(
929 src_url, false /* recursive */,
930 base::Bind(&CopyOrMoveOperationDelegate::DidRemoveSourceForMove,
931 weak_factory_.GetWeakPtr(), callback));
932 }
933
934 void CopyOrMoveOperationDelegate::DidRemoveSourceForMove(
935 const StatusCallback& callback,
936 base::File::Error error) {
937 if (error == base::File::FILE_ERROR_NOT_FOUND)
938 error = base::File::FILE_OK;
939 callback.Run(error);
940 }
941
942 void CopyOrMoveOperationDelegate::OnCopyFileProgress(
943 const FileSystemURL& src_url, int64 size) {
944 if (!progress_callback_.is_null()) {
945 progress_callback_.Run(
946 FileSystemOperation::PROGRESS, src_url, FileSystemURL(), size);
947 }
948 }
949
950 FileSystemURL CopyOrMoveOperationDelegate::CreateDestURL(
951 const FileSystemURL& src_url) const {
952 DCHECK_EQ(src_root_.type(), src_url.type());
953 DCHECK_EQ(src_root_.origin(), src_url.origin());
954
955 base::FilePath relative = dest_root_.virtual_path();
956 src_root_.virtual_path().AppendRelativePath(src_url.virtual_path(),
957 &relative);
958 return file_system_context()->CreateCrackedFileSystemURL(
959 dest_root_.origin(),
960 dest_root_.mount_type(),
961 relative);
962 }
963
964 } // namespace fileapi
OLDNEW
« no previous file with comments | « webkit/browser/fileapi/copy_or_move_operation_delegate.h ('k') | webkit/browser/fileapi/dragged_file_util.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698