Home | History | Annotate | Download | only in download 
      1 // Copyright (c) 2011 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 // Download utility implementation
      6 
      7 #include "chrome/browser/download/download_util.h"
      8 
      9 #if defined(OS_WIN)
     10 #include <shobjidl.h>
     11 #endif
     12 #include <string>
     13 
     14 #include "base/file_util.h"
     15 #include "base/i18n/rtl.h"
     16 #include "base/i18n/time_formatting.h"
     17 #include "base/lazy_instance.h"
     18 #include "base/metrics/histogram.h"
     19 #include "base/path_service.h"
     20 #include "base/string16.h"
     21 #include "base/string_number_conversions.h"
     22 #include "base/stringprintf.h"
     23 #include "base/sys_string_conversions.h"
     24 #include "base/threading/thread_restrictions.h"
     25 #include "base/utf_string_conversions.h"
     26 #include "base/value_conversions.h"
     27 #include "base/values.h"
     28 #include "base/win/windows_version.h"
     29 #include "chrome/browser/download/download_extensions.h"
     30 #include "chrome/browser/download/download_item.h"
     31 #include "chrome/browser/download/download_item_model.h"
     32 #include "chrome/browser/download/download_manager.h"
     33 #include "chrome/browser/download/download_types.h"
     34 #include "chrome/browser/extensions/crx_installer.h"
     35 #include "chrome/browser/extensions/extension_install_ui.h"
     36 #include "chrome/browser/extensions/extension_service.h"
     37 #include "chrome/browser/history/download_create_info.h"
     38 #include "chrome/browser/net/chrome_url_request_context.h"
     39 #include "chrome/browser/profiles/profile.h"
     40 #include "chrome/browser/ui/browser.h"
     41 #include "chrome/common/chrome_paths.h"
     42 #include "chrome/common/time_format.h"
     43 #include "content/browser/browser_thread.h"
     44 #include "content/browser/renderer_host/render_view_host.h"
     45 #include "content/browser/renderer_host/resource_dispatcher_host.h"
     46 #include "content/browser/tab_contents/tab_contents.h"
     47 #include "content/common/notification_service.h"
     48 #include "grit/generated_resources.h"
     49 #include "grit/locale_settings.h"
     50 #include "grit/theme_resources.h"
     51 #include "net/base/mime_util.h"
     52 #include "net/base/net_util.h"
     53 #include "skia/ext/image_operations.h"
     54 #include "third_party/skia/include/core/SkPath.h"
     55 #include "third_party/skia/include/core/SkShader.h"
     56 #include "ui/base/l10n/l10n_util.h"
     57 #include "ui/base/resource/resource_bundle.h"
     58 #include "ui/gfx/canvas_skia.h"
     59 #include "ui/gfx/image.h"
     60 #include "ui/gfx/rect.h"
     61 
     62 #if defined(TOOLKIT_VIEWS)
     63 #include "ui/base/dragdrop/os_exchange_data.h"
     64 #include "views/drag_utils.h"
     65 #endif
     66 
     67 #if defined(TOOLKIT_USES_GTK)
     68 #if defined(TOOLKIT_VIEWS)
     69 #include "ui/base/dragdrop/drag_drop_types.h"
     70 #include "views/widget/widget_gtk.h"
     71 #elif defined(TOOLKIT_GTK)
     72 #include "chrome/browser/ui/gtk/custom_drag.h"
     73 #endif  // defined(TOOLKIT_GTK)
     74 #endif  // defined(TOOLKIT_USES_GTK)
     75 
     76 #if defined(OS_WIN)
     77 #include "base/win/scoped_comptr.h"
     78 #include "chrome/browser/ui/browser_list.h"
     79 #include "chrome/browser/ui/views/frame/browser_view.h"
     80 #include "ui/base/dragdrop/drag_source.h"
     81 #include "ui/base/dragdrop/os_exchange_data_provider_win.h"
     82 #endif
     83 
     84 // TODO(phajdan.jr): Find some standard location for this, maintaining
     85 // the same value on all platforms.
     86 static const double PI = 3.141592653589793;
     87 
     88 namespace download_util {
     89 
     90 // How many times to cycle the complete animation. This should be an odd number
     91 // so that the animation ends faded out.
     92 static const int kCompleteAnimationCycles = 5;
     93 
     94 // The maximum number of 'uniquified' files we will try to create.
     95 // This is used when the filename we're trying to download is already in use,
     96 // so we create a new unique filename by appending " (nnn)" before the
     97 // extension, where 1 <= nnn <= kMaxUniqueFiles.
     98 // Also used by code that cleans up said files.
     99 static const int kMaxUniqueFiles = 100;
    100 
    101 namespace {
    102 
    103 #if defined(OS_WIN)
    104 // Returns whether the specified extension is automatically integrated into the
    105 // windows shell.
    106 bool IsShellIntegratedExtension(const string16& extension) {
    107   string16 extension_lower = StringToLowerASCII(extension);
    108 
    109   static const wchar_t* const integrated_extensions[] = {
    110     // See <http://msdn.microsoft.com/en-us/library/ms811694.aspx>.
    111     L"local",
    112     // Right-clicking on shortcuts can be magical.
    113     L"lnk",
    114   };
    115 
    116   for (int i = 0; i < arraysize(integrated_extensions); ++i) {
    117     if (extension_lower == integrated_extensions[i])
    118       return true;
    119   }
    120 
    121   // See <http://www.juniper.net/security/auto/vulnerabilities/vuln2612.html>.
    122   // That vulnerability report is not exactly on point, but files become magical
    123   // if their end in a CLSID.  Here we block extensions that look like CLSIDs.
    124   if (!extension_lower.empty() && extension_lower[0] == L'{' &&
    125       extension_lower[extension_lower.length() - 1] == L'}')
    126     return true;
    127 
    128   return false;
    129 }
    130 
    131 // Returns whether the specified file name is a reserved name on windows.
    132 // This includes names like "com2.zip" (which correspond to devices) and
    133 // desktop.ini and thumbs.db which have special meaning to the windows shell.
    134 bool IsReservedName(const string16& filename) {
    135   // This list is taken from the MSDN article "Naming a file"
    136   // http://msdn2.microsoft.com/en-us/library/aa365247(VS.85).aspx
    137   // I also added clock$ because GetSaveFileName seems to consider it as a
    138   // reserved name too.
    139   static const wchar_t* const known_devices[] = {
    140     L"con", L"prn", L"aux", L"nul", L"com1", L"com2", L"com3", L"com4", L"com5",
    141     L"com6", L"com7", L"com8", L"com9", L"lpt1", L"lpt2", L"lpt3", L"lpt4",
    142     L"lpt5", L"lpt6", L"lpt7", L"lpt8", L"lpt9", L"clock$"
    143   };
    144   string16 filename_lower = StringToLowerASCII(filename);
    145 
    146   for (int i = 0; i < arraysize(known_devices); ++i) {
    147     // Exact match.
    148     if (filename_lower == known_devices[i])
    149       return true;
    150     // Starts with "DEVICE.".
    151     if (filename_lower.find(string16(known_devices[i]) + L".") == 0)
    152       return true;
    153   }
    154 
    155   static const wchar_t* const magic_names[] = {
    156     // These file names are used by the "Customize folder" feature of the shell.
    157     L"desktop.ini",
    158     L"thumbs.db",
    159   };
    160 
    161   for (int i = 0; i < arraysize(magic_names); ++i) {
    162     if (filename_lower == magic_names[i])
    163       return true;
    164   }
    165 
    166   return false;
    167 }
    168 #endif  // OS_WIN
    169 
    170 }  // namespace
    171 
    172 // Download temporary file creation --------------------------------------------
    173 
    174 class DefaultDownloadDirectory {
    175  public:
    176   const FilePath& path() const { return path_; }
    177  private:
    178   DefaultDownloadDirectory() {
    179     if (!PathService::Get(chrome::DIR_DEFAULT_DOWNLOADS, &path_)) {
    180       NOTREACHED();
    181     }
    182     if (DownloadPathIsDangerous(path_)) {
    183       if (!PathService::Get(chrome::DIR_DEFAULT_DOWNLOADS_SAFE, &path_)) {
    184         NOTREACHED();
    185       }
    186     }
    187   }
    188   friend struct base::DefaultLazyInstanceTraits<DefaultDownloadDirectory>;
    189   FilePath path_;
    190 };
    191 
    192 static base::LazyInstance<DefaultDownloadDirectory>
    193     g_default_download_directory(base::LINKER_INITIALIZED);
    194 
    195 const FilePath& GetDefaultDownloadDirectory() {
    196   return g_default_download_directory.Get().path();
    197 }
    198 
    199 bool CreateTemporaryFileForDownload(FilePath* temp_file) {
    200   if (file_util::CreateTemporaryFileInDir(GetDefaultDownloadDirectory(),
    201                                           temp_file))
    202     return true;
    203   return file_util::CreateTemporaryFile(temp_file);
    204 }
    205 
    206 bool DownloadPathIsDangerous(const FilePath& download_path) {
    207   FilePath desktop_dir;
    208   if (!PathService::Get(chrome::DIR_USER_DESKTOP, &desktop_dir)) {
    209     NOTREACHED();
    210     return false;
    211   }
    212   return (download_path == desktop_dir);
    213 }
    214 
    215 void GenerateExtension(const FilePath& file_name,
    216                        const std::string& mime_type,
    217                        FilePath::StringType* generated_extension) {
    218   // We're worried about two things here:
    219   //
    220   // 1) Usability.  If the site fails to provide a file extension, we want to
    221   //    guess a reasonable file extension based on the content type.
    222   //
    223   // 2) Shell integration.  Some file extensions automatically integrate with
    224   //    the shell.  We block these extensions to prevent a malicious web site
    225   //    from integrating with the user's shell.
    226 
    227   // See if our file name already contains an extension.
    228   FilePath::StringType extension = file_name.Extension();
    229   if (!extension.empty())
    230     extension.erase(extension.begin());  // Erase preceding '.'.
    231 
    232 #if defined(OS_WIN)
    233   static const FilePath::CharType default_extension[] =
    234       FILE_PATH_LITERAL("download");
    235 
    236   // Rename shell-integrated extensions.
    237   if (IsShellIntegratedExtension(extension))
    238     extension.assign(default_extension);
    239 #endif
    240 
    241   if (extension.empty()) {
    242     // The GetPreferredExtensionForMimeType call will end up going to disk.  Do
    243     // this on another thread to avoid slowing the IO thread.
    244     // http://crbug.com/61827
    245     base::ThreadRestrictions::ScopedAllowIO allow_io;
    246     net::GetPreferredExtensionForMimeType(mime_type, &extension);
    247   }
    248 
    249   generated_extension->swap(extension);
    250 }
    251 
    252 void GenerateFileNameFromInfo(DownloadCreateInfo* info,
    253                               FilePath* generated_name) {
    254   GenerateFileName(GURL(info->url()),
    255                    info->content_disposition,
    256                    info->referrer_charset,
    257                    info->mime_type,
    258                    generated_name);
    259 }
    260 
    261 void GenerateFileName(const GURL& url,
    262                       const std::string& content_disposition,
    263                       const std::string& referrer_charset,
    264                       const std::string& mime_type,
    265                       FilePath* generated_name) {
    266   string16 default_file_name(
    267       l10n_util::GetStringUTF16(IDS_DEFAULT_DOWNLOAD_FILENAME));
    268 
    269   string16 new_name = net::GetSuggestedFilename(GURL(url),
    270                                                 content_disposition,
    271                                                 referrer_charset,
    272                                                 default_file_name);
    273 
    274   // TODO(evan): this code is totally wrong -- we should just generate
    275   // Unicode filenames and do all this encoding switching at the end.
    276   // However, I'm just shuffling wrong code around, at least not adding
    277   // to it.
    278 #if defined(OS_WIN)
    279   *generated_name = FilePath(new_name);
    280 #else
    281   *generated_name = FilePath(
    282       base::SysWideToNativeMB(UTF16ToWide(new_name)));
    283 #endif
    284 
    285   DCHECK(!generated_name->empty());
    286 
    287   GenerateSafeFileName(mime_type, generated_name);
    288 }
    289 
    290 void GenerateSafeFileName(const std::string& mime_type, FilePath* file_name) {
    291   // Make sure we get the right file extension
    292   FilePath::StringType extension;
    293   GenerateExtension(*file_name, mime_type, &extension);
    294   *file_name = file_name->ReplaceExtension(extension);
    295 
    296 #if defined(OS_WIN)
    297   // Prepend "_" to the file name if it's a reserved name
    298   FilePath::StringType leaf_name = file_name->BaseName().value();
    299   DCHECK(!leaf_name.empty());
    300   if (IsReservedName(leaf_name)) {
    301     leaf_name = FilePath::StringType(FILE_PATH_LITERAL("_")) + leaf_name;
    302     *file_name = file_name->DirName();
    303     if (file_name->value() == FilePath::kCurrentDirectory) {
    304       *file_name = FilePath(leaf_name);
    305     } else {
    306       *file_name = file_name->Append(leaf_name);
    307     }
    308   }
    309 #endif
    310 }
    311 
    312 void OpenChromeExtension(Profile* profile,
    313                          DownloadManager* download_manager,
    314                          const DownloadItem& download_item) {
    315   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
    316   DCHECK(download_item.is_extension_install());
    317 
    318   ExtensionService* service = profile->GetExtensionService();
    319   CHECK(service);
    320   NotificationService* nservice = NotificationService::current();
    321   GURL nonconst_download_url = download_item.url();
    322   nservice->Notify(NotificationType::EXTENSION_READY_FOR_INSTALL,
    323                    Source<DownloadManager>(download_manager),
    324                    Details<GURL>(&nonconst_download_url));
    325 
    326   scoped_refptr<CrxInstaller> installer(
    327       new CrxInstaller(service, new ExtensionInstallUI(profile)));
    328   installer->set_delete_source(true);
    329 
    330   if (UserScript::IsURLUserScript(download_item.url(),
    331                                   download_item.mime_type())) {
    332     installer->InstallUserScript(download_item.full_path(),
    333                                  download_item.url());
    334     return;
    335   }
    336 
    337   bool is_gallery_download = service->IsDownloadFromGallery(
    338       download_item.url(), download_item.referrer_url());
    339   installer->set_original_mime_type(download_item.original_mime_type());
    340   installer->set_apps_require_extension_mime_type(true);
    341   installer->set_original_url(download_item.url());
    342   installer->set_is_gallery_install(is_gallery_download);
    343   installer->InstallCrx(download_item.full_path());
    344   installer->set_allow_silent_install(is_gallery_download);
    345 }
    346 
    347 void RecordDownloadCount(DownloadCountTypes type) {
    348   UMA_HISTOGRAM_ENUMERATION(
    349       "Download.Counts", type, DOWNLOAD_COUNT_TYPES_LAST_ENTRY);
    350 }
    351 
    352 // Download progress painting --------------------------------------------------
    353 
    354 // Common bitmaps used for download progress animations. We load them once the
    355 // first time we do a progress paint, then reuse them as they are always the
    356 // same.
    357 SkBitmap* g_foreground_16 = NULL;
    358 SkBitmap* g_background_16 = NULL;
    359 SkBitmap* g_foreground_32 = NULL;
    360 SkBitmap* g_background_32 = NULL;
    361 
    362 void PaintDownloadProgress(gfx::Canvas* canvas,
    363 #if defined(TOOLKIT_VIEWS)
    364                            views::View* containing_view,
    365 #endif
    366                            int origin_x,
    367                            int origin_y,
    368                            int start_angle,
    369                            int percent_done,
    370                            PaintDownloadProgressSize size) {
    371   // Load up our common bitmaps
    372   if (!g_background_16) {
    373     ResourceBundle& rb = ResourceBundle::GetSharedInstance();
    374     g_foreground_16 = rb.GetBitmapNamed(IDR_DOWNLOAD_PROGRESS_FOREGROUND_16);
    375     g_background_16 = rb.GetBitmapNamed(IDR_DOWNLOAD_PROGRESS_BACKGROUND_16);
    376     g_foreground_32 = rb.GetBitmapNamed(IDR_DOWNLOAD_PROGRESS_FOREGROUND_32);
    377     g_background_32 = rb.GetBitmapNamed(IDR_DOWNLOAD_PROGRESS_BACKGROUND_32);
    378   }
    379 
    380   SkBitmap* background = (size == BIG) ? g_background_32 : g_background_16;
    381   SkBitmap* foreground = (size == BIG) ? g_foreground_32 : g_foreground_16;
    382 
    383   const int kProgressIconSize = (size == BIG) ? kBigProgressIconSize :
    384                                                 kSmallProgressIconSize;
    385 
    386   // We start by storing the bounds of the background and foreground bitmaps
    387   // so that it is easy to mirror the bounds if the UI layout is RTL.
    388   gfx::Rect background_bounds(origin_x, origin_y,
    389                               background->width(), background->height());
    390   gfx::Rect foreground_bounds(origin_x, origin_y,
    391                               foreground->width(), foreground->height());
    392 
    393 #if defined(TOOLKIT_VIEWS)
    394   // Mirror the positions if necessary.
    395   int mirrored_x = containing_view->GetMirroredXForRect(background_bounds);
    396   background_bounds.set_x(mirrored_x);
    397   mirrored_x = containing_view->GetMirroredXForRect(foreground_bounds);
    398   foreground_bounds.set_x(mirrored_x);
    399 #endif
    400 
    401   // Draw the background progress image.
    402   SkPaint background_paint;
    403   canvas->DrawBitmapInt(*background,
    404                         background_bounds.x(),
    405                         background_bounds.y(),
    406                         background_paint);
    407 
    408   // Layer the foreground progress image in an arc proportional to the download
    409   // progress. The arc grows clockwise, starting in the midnight position, as
    410   // the download progresses. However, if the download does not have known total
    411   // size (the server didn't give us one), then we just spin an arc around until
    412   // we're done.
    413   float sweep_angle = 0.0;
    414   float start_pos = static_cast<float>(kStartAngleDegrees);
    415   if (percent_done < 0) {
    416     sweep_angle = kUnknownAngleDegrees;
    417     start_pos = static_cast<float>(start_angle);
    418   } else if (percent_done > 0) {
    419     sweep_angle = static_cast<float>(kMaxDegrees / 100.0 * percent_done);
    420   }
    421 
    422   // Set up an arc clipping region for the foreground image. Don't bother using
    423   // a clipping region if it would round to 360 (really 0) degrees, since that
    424   // would eliminate the foreground completely and be quite confusing (it would
    425   // look like 0% complete when it should be almost 100%).
    426   SkPaint foreground_paint;
    427   if (sweep_angle < static_cast<float>(kMaxDegrees - 1)) {
    428     SkRect oval;
    429     oval.set(SkIntToScalar(foreground_bounds.x()),
    430              SkIntToScalar(foreground_bounds.y()),
    431              SkIntToScalar(foreground_bounds.x() + kProgressIconSize),
    432              SkIntToScalar(foreground_bounds.y() + kProgressIconSize));
    433     SkPath path;
    434     path.arcTo(oval,
    435                SkFloatToScalar(start_pos),
    436                SkFloatToScalar(sweep_angle), false);
    437     path.lineTo(SkIntToScalar(foreground_bounds.x() + kProgressIconSize / 2),
    438                 SkIntToScalar(foreground_bounds.y() + kProgressIconSize / 2));
    439 
    440     SkShader* shader =
    441         SkShader::CreateBitmapShader(*foreground,
    442                                      SkShader::kClamp_TileMode,
    443                                      SkShader::kClamp_TileMode);
    444     SkMatrix shader_scale;
    445     shader_scale.setTranslate(SkIntToScalar(foreground_bounds.x()),
    446                               SkIntToScalar(foreground_bounds.y()));
    447     shader->setLocalMatrix(shader_scale);
    448     foreground_paint.setShader(shader);
    449     foreground_paint.setAntiAlias(true);
    450     shader->unref();
    451     canvas->AsCanvasSkia()->drawPath(path, foreground_paint);
    452     return;
    453   }
    454 
    455   canvas->DrawBitmapInt(*foreground,
    456                         foreground_bounds.x(),
    457                         foreground_bounds.y(),
    458                         foreground_paint);
    459 }
    460 
    461 void PaintDownloadComplete(gfx::Canvas* canvas,
    462 #if defined(TOOLKIT_VIEWS)
    463                            views::View* containing_view,
    464 #endif
    465                            int origin_x,
    466                            int origin_y,
    467                            double animation_progress,
    468                            PaintDownloadProgressSize size) {
    469   // Load up our common bitmaps.
    470   if (!g_foreground_16) {
    471     ResourceBundle& rb = ResourceBundle::GetSharedInstance();
    472     g_foreground_16 = rb.GetBitmapNamed(IDR_DOWNLOAD_PROGRESS_FOREGROUND_16);
    473     g_foreground_32 = rb.GetBitmapNamed(IDR_DOWNLOAD_PROGRESS_FOREGROUND_32);
    474   }
    475 
    476   SkBitmap* complete = (size == BIG) ? g_foreground_32 : g_foreground_16;
    477 
    478   gfx::Rect complete_bounds(origin_x, origin_y,
    479                             complete->width(), complete->height());
    480 #if defined(TOOLKIT_VIEWS)
    481   // Mirror the positions if necessary.
    482   complete_bounds.set_x(containing_view->GetMirroredXForRect(complete_bounds));
    483 #endif
    484 
    485   // Start at full opacity, then loop back and forth five times before ending
    486   // at zero opacity.
    487   double opacity = sin(animation_progress * PI * kCompleteAnimationCycles +
    488                    PI/2) / 2 + 0.5;
    489 
    490   canvas->SaveLayerAlpha(static_cast<int>(255.0 * opacity), complete_bounds);
    491   canvas->AsCanvasSkia()->drawARGB(0, 255, 255, 255, SkXfermode::kClear_Mode);
    492   canvas->DrawBitmapInt(*complete, complete_bounds.x(), complete_bounds.y());
    493   canvas->Restore();
    494 }
    495 
    496 void PaintDownloadInterrupted(gfx::Canvas* canvas,
    497 #if defined(TOOLKIT_VIEWS)
    498                               views::View* containing_view,
    499 #endif
    500                               int origin_x,
    501                               int origin_y,
    502                               double animation_progress,
    503                               PaintDownloadProgressSize size) {
    504   // Load up our common bitmaps.
    505   if (!g_foreground_16) {
    506     ResourceBundle& rb = ResourceBundle::GetSharedInstance();
    507     g_foreground_16 = rb.GetBitmapNamed(IDR_DOWNLOAD_PROGRESS_FOREGROUND_16);
    508     g_foreground_32 = rb.GetBitmapNamed(IDR_DOWNLOAD_PROGRESS_FOREGROUND_32);
    509   }
    510 
    511   SkBitmap* complete = (size == BIG) ? g_foreground_32 : g_foreground_16;
    512 
    513   gfx::Rect complete_bounds(origin_x, origin_y,
    514                             complete->width(), complete->height());
    515 #if defined(TOOLKIT_VIEWS)
    516   // Mirror the positions if necessary.
    517   complete_bounds.set_x(containing_view->GetMirroredXForRect(complete_bounds));
    518 #endif
    519 
    520   // Start at zero opacity, then loop back and forth five times before ending
    521   // at full opacity.
    522   double opacity = sin(
    523       (1.0 - animation_progress) * PI * kCompleteAnimationCycles + PI/2) / 2 +
    524           0.5;
    525 
    526   canvas->SaveLayerAlpha(static_cast<int>(255.0 * opacity), complete_bounds);
    527   canvas->AsCanvasSkia()->drawARGB(0, 255, 255, 255, SkXfermode::kClear_Mode);
    528   canvas->DrawBitmapInt(*complete, complete_bounds.x(), complete_bounds.y());
    529   canvas->Restore();
    530 }
    531 
    532 // Load a language dependent height so that the dangerous download confirmation
    533 // message doesn't overlap with the download link label.
    534 int GetBigProgressIconSize() {
    535   static int big_progress_icon_size = 0;
    536   if (big_progress_icon_size == 0) {
    537     string16 locale_size_str =
    538         l10n_util::GetStringUTF16(IDS_DOWNLOAD_BIG_PROGRESS_SIZE);
    539     bool rc = base::StringToInt(locale_size_str, &big_progress_icon_size);
    540     if (!rc || big_progress_icon_size < kBigProgressIconSize) {
    541       NOTREACHED();
    542       big_progress_icon_size = kBigProgressIconSize;
    543     }
    544   }
    545 
    546   return big_progress_icon_size;
    547 }
    548 
    549 int GetBigProgressIconOffset() {
    550   return (GetBigProgressIconSize() - kBigIconSize) / 2;
    551 }
    552 
    553 #if defined(TOOLKIT_VIEWS)
    554 // Download dragging
    555 void DragDownload(const DownloadItem* download,
    556                   gfx::Image* icon,
    557                   gfx::NativeView view) {
    558   DCHECK(download);
    559 
    560   // Set up our OLE machinery
    561   ui::OSExchangeData data;
    562 
    563   if (icon) {
    564     drag_utils::CreateDragImageForFile(
    565         download->GetFileNameToReportUser(), *icon, &data);
    566   }
    567 
    568   const FilePath full_path = download->full_path();
    569   data.SetFilename(full_path);
    570 
    571   std::string mime_type = download->mime_type();
    572   if (mime_type.empty())
    573     net::GetMimeTypeFromFile(full_path, &mime_type);
    574 
    575   // Add URL so that we can load supported files when dragged to TabContents.
    576   if (net::IsSupportedMimeType(mime_type)) {
    577     data.SetURL(net::FilePathToFileURL(full_path),
    578                 download->GetFileNameToReportUser().LossyDisplayName());
    579   }
    580 
    581 #if defined(OS_WIN)
    582   scoped_refptr<ui::DragSource> drag_source(new ui::DragSource);
    583 
    584   // Run the drag and drop loop
    585   DWORD effects;
    586   DoDragDrop(ui::OSExchangeDataProviderWin::GetIDataObject(data),
    587              drag_source.get(), DROPEFFECT_COPY | DROPEFFECT_LINK, &effects);
    588 #elif defined(TOOLKIT_USES_GTK)
    589   GtkWidget* root = gtk_widget_get_toplevel(view);
    590   if (!root)
    591     return;
    592 
    593   views::WidgetGtk* widget = static_cast<views::WidgetGtk*>(
    594       views::NativeWidget::GetNativeWidgetForNativeView(root));
    595   if (!widget)
    596     return;
    597 
    598   widget->DoDrag(data,
    599                  ui::DragDropTypes::DRAG_COPY | ui::DragDropTypes::DRAG_LINK);
    600 #endif  // OS_WIN
    601 }
    602 #elif defined(USE_X11)
    603 void DragDownload(const DownloadItem* download,
    604                   gfx::Image* icon,
    605                   gfx::NativeView view) {
    606   DownloadItemDrag::BeginDrag(download, icon);
    607 }
    608 #endif  // USE_X11
    609 
    610 DictionaryValue* CreateDownloadItemValue(DownloadItem* download, int id) {
    611   DictionaryValue* file_value = new DictionaryValue();
    612 
    613   file_value->SetInteger("started",
    614       static_cast<int>(download->start_time().ToTimeT()));
    615   file_value->SetString("since_string",
    616       TimeFormat::RelativeDate(download->start_time(), NULL));
    617   file_value->SetString("date_string",
    618       base::TimeFormatShortDate(download->start_time()));
    619   file_value->SetInteger("id", id);
    620   file_value->Set("file_path",
    621                   base::CreateFilePathValue(download->GetTargetFilePath()));
    622   // Keep file names as LTR.
    623   string16 file_name = download->GetFileNameToReportUser().LossyDisplayName();
    624   file_name = base::i18n::GetDisplayStringInLTRDirectionality(file_name);
    625   file_value->SetString("file_name", file_name);
    626   file_value->SetString("url", download->url().spec());
    627   file_value->SetBoolean("otr", download->is_otr());
    628 
    629   if (download->IsInProgress()) {
    630     if (download->safety_state() == DownloadItem::DANGEROUS) {
    631       file_value->SetString("state", "DANGEROUS");
    632       DCHECK(download->danger_type() == DownloadItem::DANGEROUS_FILE ||
    633              download->danger_type() == DownloadItem::DANGEROUS_URL);
    634       const char* danger_type_value =
    635           download->danger_type() == DownloadItem::DANGEROUS_FILE ?
    636           "DANGEROUS_FILE" : "DANGEROUS_URL";
    637       file_value->SetString("danger_type", danger_type_value);
    638     } else if (download->is_paused()) {
    639       file_value->SetString("state", "PAUSED");
    640     } else {
    641       file_value->SetString("state", "IN_PROGRESS");
    642     }
    643 
    644     file_value->SetString("progress_status_text",
    645         GetProgressStatusText(download));
    646 
    647     file_value->SetInteger("percent",
    648         static_cast<int>(download->PercentComplete()));
    649     file_value->SetInteger("received",
    650         static_cast<int>(download->received_bytes()));
    651   } else if (download->IsInterrupted()) {
    652     file_value->SetString("state", "INTERRUPTED");
    653 
    654     file_value->SetString("progress_status_text",
    655         GetProgressStatusText(download));
    656 
    657     file_value->SetInteger("percent",
    658         static_cast<int>(download->PercentComplete()));
    659     file_value->SetInteger("received",
    660         static_cast<int>(download->received_bytes()));
    661   } else if (download->IsCancelled()) {
    662     file_value->SetString("state", "CANCELLED");
    663   } else if (download->IsComplete()) {
    664     if (download->safety_state() == DownloadItem::DANGEROUS) {
    665       file_value->SetString("state", "DANGEROUS");
    666     } else {
    667       file_value->SetString("state", "COMPLETE");
    668     }
    669   }
    670 
    671   file_value->SetInteger("total",
    672       static_cast<int>(download->total_bytes()));
    673 
    674   return file_value;
    675 }
    676 
    677 string16 GetProgressStatusText(DownloadItem* download) {
    678   int64 total = download->total_bytes();
    679   int64 size = download->received_bytes();
    680   DataUnits amount_units = GetByteDisplayUnits(size);
    681   string16 received_size = FormatBytes(size, amount_units, true);
    682   string16 amount = received_size;
    683 
    684   // Adjust both strings for the locale direction since we don't yet know which
    685   // string we'll end up using for constructing the final progress string.
    686   base::i18n::AdjustStringForLocaleDirection(&amount);
    687 
    688   if (total) {
    689     amount_units = GetByteDisplayUnits(total);
    690     string16 total_text = FormatBytes(total, amount_units, true);
    691     base::i18n::AdjustStringForLocaleDirection(&total_text);
    692 
    693     base::i18n::AdjustStringForLocaleDirection(&received_size);
    694     amount = l10n_util::GetStringFUTF16(IDS_DOWNLOAD_TAB_PROGRESS_SIZE,
    695                                         received_size,
    696                                         total_text);
    697   } else {
    698     amount.assign(received_size);
    699   }
    700   int64 current_speed = download->CurrentSpeed();
    701   amount_units = GetByteDisplayUnits(current_speed);
    702   string16 speed_text = FormatSpeed(current_speed, amount_units, true);
    703   base::i18n::AdjustStringForLocaleDirection(&speed_text);
    704 
    705   base::TimeDelta remaining;
    706   string16 time_remaining;
    707   if (download->is_paused())
    708     time_remaining = l10n_util::GetStringUTF16(IDS_DOWNLOAD_PROGRESS_PAUSED);
    709   else if (download->TimeRemaining(&remaining))
    710     time_remaining = TimeFormat::TimeRemaining(remaining);
    711 
    712   if (time_remaining.empty()) {
    713     base::i18n::AdjustStringForLocaleDirection(&amount);
    714     return l10n_util::GetStringFUTF16(
    715         IDS_DOWNLOAD_TAB_PROGRESS_STATUS_TIME_UNKNOWN, speed_text, amount);
    716   }
    717   return l10n_util::GetStringFUTF16(IDS_DOWNLOAD_TAB_PROGRESS_STATUS,
    718                                     speed_text, amount, time_remaining);
    719 }
    720 
    721 #if !defined(OS_MACOSX)
    722 void UpdateAppIconDownloadProgress(int download_count,
    723                                    bool progress_known,
    724                                    float progress) {
    725 #if defined(OS_WIN)
    726   // Taskbar progress bar is only supported on Win7.
    727   if (base::win::GetVersion() < base::win::VERSION_WIN7)
    728     return;
    729 
    730   base::win::ScopedComPtr<ITaskbarList3> taskbar;
    731   HRESULT result = taskbar.CreateInstance(CLSID_TaskbarList, NULL,
    732                                           CLSCTX_INPROC_SERVER);
    733   if (FAILED(result)) {
    734     VLOG(1) << "Failed creating a TaskbarList object: " << result;
    735     return;
    736   }
    737 
    738   result = taskbar->HrInit();
    739   if (FAILED(result)) {
    740     LOG(ERROR) << "Failed initializing an ITaskbarList3 interface.";
    741     return;
    742   }
    743 
    744   // Iterate through all the browser windows, and draw the progress bar.
    745   for (BrowserList::const_iterator browser_iterator = BrowserList::begin();
    746       browser_iterator != BrowserList::end(); browser_iterator++) {
    747     Browser* browser = *browser_iterator;
    748     BrowserWindow* window = browser->window();
    749     if (!window)
    750       continue;
    751     HWND frame = window->GetNativeHandle();
    752     if (download_count == 0 || progress == 1.0f)
    753       taskbar->SetProgressState(frame, TBPF_NOPROGRESS);
    754     else if (!progress_known)
    755       taskbar->SetProgressState(frame, TBPF_INDETERMINATE);
    756     else
    757       taskbar->SetProgressValue(frame, static_cast<int>(progress * 100), 100);
    758   }
    759 #endif
    760 }
    761 #endif
    762 
    763 // Appends the passed the number between parenthesis the path before the
    764 // extension.
    765 void AppendNumberToPath(FilePath* path, int number) {
    766   *path = path->InsertBeforeExtensionASCII(StringPrintf(" (%d)", number));
    767 }
    768 
    769 // Attempts to find a number that can be appended to that path to make it
    770 // unique. If |path| does not exist, 0 is returned.  If it fails to find such
    771 // a number, -1 is returned.
    772 int GetUniquePathNumber(const FilePath& path) {
    773   if (!file_util::PathExists(path))
    774     return 0;
    775 
    776   FilePath new_path;
    777   for (int count = 1; count <= kMaxUniqueFiles; ++count) {
    778     new_path = FilePath(path);
    779     AppendNumberToPath(&new_path, count);
    780 
    781     if (!file_util::PathExists(new_path))
    782       return count;
    783   }
    784 
    785   return -1;
    786 }
    787 
    788 void DownloadUrl(
    789     const GURL& url,
    790     const GURL& referrer,
    791     const std::string& referrer_charset,
    792     const DownloadSaveInfo& save_info,
    793     ResourceDispatcherHost* rdh,
    794     int render_process_host_id,
    795     int render_view_id,
    796     net::URLRequestContextGetter* request_context_getter) {
    797   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
    798 
    799   net::URLRequestContext* context =
    800       request_context_getter->GetURLRequestContext();
    801   context->set_referrer_charset(referrer_charset);
    802 
    803   rdh->BeginDownload(url,
    804                      referrer,
    805                      save_info,
    806                      true,  // Show "Save as" UI.
    807                      render_process_host_id,
    808                      render_view_id,
    809                      context);
    810 }
    811 
    812 void CancelDownloadRequest(ResourceDispatcherHost* rdh,
    813                            int render_process_id,
    814                            int request_id) {
    815   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
    816   // |rdh| may be NULL in unit tests.
    817   if (rdh)
    818     rdh->CancelRequest(render_process_id, request_id, false);
    819 }
    820 
    821 void NotifyDownloadInitiated(int render_process_id, int render_view_id) {
    822   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
    823   RenderViewHost* rvh = RenderViewHost::FromID(render_process_id,
    824                                                render_view_id);
    825   if (!rvh)
    826     return;
    827 
    828   NotificationService::current()->Notify(NotificationType::DOWNLOAD_INITIATED,
    829                                          Source<RenderViewHost>(rvh),
    830                                          NotificationService::NoDetails());
    831 }
    832 
    833 int GetUniquePathNumberWithCrDownload(const FilePath& path) {
    834   if (!file_util::PathExists(path) &&
    835       !file_util::PathExists(GetCrDownloadPath(path)))
    836     return 0;
    837 
    838   FilePath new_path;
    839   for (int count = 1; count <= kMaxUniqueFiles; ++count) {
    840     new_path = FilePath(path);
    841     AppendNumberToPath(&new_path, count);
    842 
    843     if (!file_util::PathExists(new_path) &&
    844         !file_util::PathExists(GetCrDownloadPath(new_path)))
    845       return count;
    846   }
    847 
    848   return -1;
    849 }
    850 
    851 namespace {
    852 
    853 // NOTE: If index is 0, deletes files that do not have the " (nnn)" appended.
    854 void DeleteUniqueDownloadFile(const FilePath& path, int index) {
    855   FilePath new_path(path);
    856   if (index > 0)
    857     AppendNumberToPath(&new_path, index);
    858   file_util::Delete(new_path, false);
    859 }
    860 
    861 }  // namespace
    862 
    863 void EraseUniqueDownloadFiles(const FilePath& path) {
    864   FilePath cr_path = GetCrDownloadPath(path);
    865 
    866   for (int index = 0; index <= kMaxUniqueFiles; ++index) {
    867     DeleteUniqueDownloadFile(path, index);
    868     DeleteUniqueDownloadFile(cr_path, index);
    869   }
    870 }
    871 
    872 FilePath GetCrDownloadPath(const FilePath& suggested_path) {
    873   FilePath::StringType file_name;
    874   base::SStringPrintf(
    875       &file_name,
    876       PRFilePathLiteral FILE_PATH_LITERAL(".crdownload"),
    877       suggested_path.value().c_str());
    878   return FilePath(file_name);
    879 }
    880 
    881 // TODO(erikkay,phajdan.jr): This is apparently not being exercised in tests.
    882 bool IsDangerous(DownloadCreateInfo* info, Profile* profile, bool auto_open) {
    883   DownloadDangerLevel danger_level = GetFileDangerLevel(
    884       info->suggested_path.BaseName());
    885   if (danger_level == Dangerous)
    886     return !(auto_open && info->has_user_gesture);
    887   if (danger_level == AllowOnUserGesture && !info->has_user_gesture)
    888     return true;
    889   if (info->is_extension_install) {
    890     // Extensions that are not from the gallery are considered dangerous.
    891     ExtensionService* service = profile->GetExtensionService();
    892     if (!service ||
    893         !service->IsDownloadFromGallery(info->url(), info->referrer_url))
    894       return true;
    895   }
    896   return false;
    897 }
    898 
    899 }  // namespace download_util
    900