1// Copyright (c) 2012 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 "content/browser/download/drag_download_util.h" 6 7#include <string> 8 9#include "base/bind.h" 10#include "base/file_util.h" 11#include "base/files/file_path.h" 12#include "base/memory/scoped_ptr.h" 13#include "base/strings/string_number_conversions.h" 14#include "base/strings/string_util.h" 15#include "base/strings/utf_string_conversions.h" 16#include "base/threading/thread_restrictions.h" 17#include "content/public/browser/browser_thread.h" 18#include "net/base/file_stream.h" 19#include "net/base/net_errors.h" 20#include "url/gurl.h" 21 22using net::FileStream; 23 24namespace content { 25 26bool ParseDownloadMetadata(const string16& metadata, 27 string16* mime_type, 28 base::FilePath* file_name, 29 GURL* url) { 30 const char16 separator = L':'; 31 32 size_t mime_type_end_pos = metadata.find(separator); 33 if (mime_type_end_pos == string16::npos) 34 return false; 35 36 size_t file_name_end_pos = metadata.find(separator, mime_type_end_pos + 1); 37 if (file_name_end_pos == string16::npos) 38 return false; 39 40 GURL parsed_url = GURL(metadata.substr(file_name_end_pos + 1)); 41 if (!parsed_url.is_valid()) 42 return false; 43 44 if (mime_type) 45 *mime_type = metadata.substr(0, mime_type_end_pos); 46 if (file_name) { 47 string16 file_name_str = metadata.substr( 48 mime_type_end_pos + 1, file_name_end_pos - mime_type_end_pos - 1); 49#if defined(OS_WIN) 50 *file_name = base::FilePath(file_name_str); 51#else 52 *file_name = base::FilePath(UTF16ToUTF8(file_name_str)); 53#endif 54 } 55 if (url) 56 *url = parsed_url; 57 58 return true; 59} 60 61FileStream* CreateFileStreamForDrop(base::FilePath* file_path, 62 net::NetLog* net_log) { 63 DCHECK(file_path && !file_path->empty()); 64 65 scoped_ptr<FileStream> file_stream(new FileStream(net_log)); 66 const int kMaxSeq = 99; 67 for (int seq = 0; seq <= kMaxSeq; seq++) { 68 base::FilePath new_file_path; 69 if (seq == 0) { 70 new_file_path = *file_path; 71 } else { 72#if defined(OS_WIN) 73 string16 suffix = ASCIIToUTF16("-") + base::IntToString16(seq); 74#else 75 std::string suffix = std::string("-") + base::IntToString(seq); 76#endif 77 new_file_path = file_path->InsertBeforeExtension(suffix); 78 } 79 80 // http://crbug.com/110709 81 base::ThreadRestrictions::ScopedAllowIO allow_io; 82 83 // Explicitly (and redundantly check) for file -- despite the fact that our 84 // open won't overwrite -- just to avoid log spew. 85 if (!base::PathExists(new_file_path) && 86 file_stream->OpenSync(new_file_path, base::PLATFORM_FILE_CREATE | 87 base::PLATFORM_FILE_WRITE) == net::OK) { 88 *file_path = new_file_path; 89 return file_stream.release(); 90 } 91 } 92 93 return NULL; 94} 95 96PromiseFileFinalizer::PromiseFileFinalizer( 97 DragDownloadFile* drag_file_downloader) 98 : drag_file_downloader_(drag_file_downloader) { 99} 100 101void PromiseFileFinalizer::OnDownloadCompleted( 102 const base::FilePath& file_path) { 103 BrowserThread::PostTask( 104 BrowserThread::UI, FROM_HERE, 105 base::Bind(&PromiseFileFinalizer::Cleanup, this)); 106} 107 108void PromiseFileFinalizer::OnDownloadAborted() { 109 BrowserThread::PostTask( 110 BrowserThread::UI, FROM_HERE, 111 base::Bind(&PromiseFileFinalizer::Cleanup, this)); 112} 113 114PromiseFileFinalizer::~PromiseFileFinalizer() {} 115 116void PromiseFileFinalizer::Cleanup() { 117 if (drag_file_downloader_.get()) 118 drag_file_downloader_ = NULL; 119} 120 121} // namespace content 122