1a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)// Copyright 2014 The Chromium Authors. All rights reserved.
2a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be
3a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)// found in the LICENSE file.
4a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)
5a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)#include "content/browser/loader/temporary_file_stream.h"
6a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)
7a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)#include "base/bind.h"
8a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)#include "base/callback.h"
95c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu#include "base/files/file_proxy.h"
10a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)#include "base/memory/ref_counted.h"
11a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)#include "content/public/browser/browser_thread.h"
12a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)#include "net/base/file_stream.h"
131320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "storage/common/blob/shareable_file_reference.h"
14a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)
1503b57e008b61dfcb1fbad3aea950ae0e001748b0Torne (Richard Coles)using storage::ShareableFileReference;
16a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)
17a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)namespace content {
18a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)
19a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)namespace {
20a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)
21a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)void DidCreateTemporaryFile(
22a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)    const CreateTemporaryFileStreamCallback& callback,
235c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu    scoped_ptr<base::FileProxy> file_proxy,
24a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)    base::File::Error error_code,
25a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)    const base::FilePath& file_path) {
26a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
27a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)
285c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu  if (!file_proxy->IsValid()) {
29a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)    callback.Run(error_code, scoped_ptr<net::FileStream>(), NULL);
30a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)    return;
31a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)  }
32a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)
335c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu  scoped_refptr<base::TaskRunner> task_runner =
345c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu      BrowserThread::GetMessageLoopProxyForThread(BrowserThread::FILE);
355c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu
36a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)  // Cancelled or not, create the deletable_file so the temporary is cleaned up.
37a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)  scoped_refptr<ShareableFileReference> deletable_file =
38a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)      ShareableFileReference::GetOrCreate(
39a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)          file_path,
40a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)          ShareableFileReference::DELETE_ON_FINAL_RELEASE,
415c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu          task_runner.get());
42a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)
435c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu  scoped_ptr<net::FileStream> file_stream(
445c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu      new net::FileStream(file_proxy->TakeFile(), task_runner));
45a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)
461320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci  callback.Run(error_code, file_stream.Pass(), deletable_file.get());
47a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)}
48a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)
49a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)}  // namespace
50a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)
51a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)void CreateTemporaryFileStream(
52a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)    const CreateTemporaryFileStreamCallback& callback) {
53a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
54a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)
555c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu  scoped_ptr<base::FileProxy> file_proxy(new base::FileProxy(
565c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu      BrowserThread::GetMessageLoopProxyForThread(BrowserThread::FILE).get()));
575c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu  base::FileProxy* proxy = file_proxy.get();
585c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu  proxy->CreateTemporary(
595c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu      base::File::FLAG_ASYNC,
605c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu      base::Bind(&DidCreateTemporaryFile, callback, Passed(&file_proxy)));
61a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)}
62a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)
63a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)}  // namespace content
64