toolkit/components/jsdownloads/src/DownloadPlatform.cpp@129ffea94266
toolkit/components/jsdownloads/src/DownloadPlatform.cpp
Sat, 03 Jan 2015 20:18:00 +0100
- author
- Michael Schloh von Bennewitz <michael@schloh.com>
- date
- Sat, 03 Jan 2015 20:18:00 +0100
- branch
- TOR_BUG_3246
- changeset 7
- 129ffea94266
- permissions
- -rw-r--r--
Conditionally enable double key logic according to:
private browsing mode or privacy.thirdparty.isolate preference and
implement in GetCookieStringCommon and FindCookie where it counts...
With some reservations of how to convince FindCookie users to test
condition and pass a nullptr when disabling double key logic.
1 /* This Source Code Form is subject to the terms of the Mozilla Public
2 * License, v. 2.0. If a copy of the MPL was not distributed with this
3 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
5 #include "DownloadPlatform.h"
6 #include "nsAutoPtr.h"
7 #include "nsString.h"
8 #include "nsIURI.h"
9 #include "nsIFile.h"
10 #include "nsDirectoryServiceDefs.h"
12 #include "mozilla/Preferences.h"
14 #define PREF_BDM_ADDTORECENTDOCS "browser.download.manager.addToRecentDocs"
16 #ifdef XP_WIN
17 #include <shlobj.h>
18 #include <urlmon.h>
19 #include "nsILocalFileWin.h"
20 #endif
22 #ifdef XP_MACOSX
23 #include <CoreFoundation/CoreFoundation.h>
24 #endif
26 #ifdef MOZ_WIDGET_ANDROID
27 #include "AndroidBridge.h"
28 #endif
30 #ifdef MOZ_WIDGET_GTK
31 #include <gtk/gtk.h>
32 #endif
34 using namespace mozilla;
36 DownloadPlatform *DownloadPlatform::gDownloadPlatformService = nullptr;
38 NS_IMPL_ISUPPORTS(DownloadPlatform, mozIDownloadPlatform);
40 DownloadPlatform* DownloadPlatform::GetDownloadPlatform()
41 {
42 if (!gDownloadPlatformService) {
43 gDownloadPlatformService = new DownloadPlatform();
44 }
46 NS_ADDREF(gDownloadPlatformService);
48 #if defined(MOZ_WIDGET_GTK)
49 g_type_init();
50 #endif
52 return gDownloadPlatformService;
53 }
55 #ifdef MOZ_ENABLE_GIO
56 static void gio_set_metadata_done(GObject *source_obj, GAsyncResult *res, gpointer user_data)
57 {
58 GError *err = nullptr;
59 g_file_set_attributes_finish(G_FILE(source_obj), res, nullptr, &err);
60 if (err) {
61 #ifdef DEBUG
62 NS_DebugBreak(NS_DEBUG_WARNING, "Set file metadata failed: ", err->message, __FILE__, __LINE__);
63 #endif
64 g_error_free(err);
65 }
66 }
67 #endif
69 nsresult DownloadPlatform::DownloadDone(nsIURI* aSource, nsIFile* aTarget,
70 const nsACString& aContentType, bool aIsPrivate)
71 {
72 #if defined(XP_WIN) || defined(XP_MACOSX) || defined(MOZ_WIDGET_ANDROID) || defined(MOZ_WIDGET_GTK)
73 nsAutoString path;
74 if (aTarget && NS_SUCCEEDED(aTarget->GetPath(path))) {
75 #if defined(XP_WIN) || defined(MOZ_WIDGET_GTK)
76 // On Windows and Gtk, add the download to the system's "recent documents"
77 // list, with a pref to disable.
78 {
79 bool addToRecentDocs = Preferences::GetBool(PREF_BDM_ADDTORECENTDOCS);
80 if (addToRecentDocs && !aIsPrivate) {
81 #ifdef XP_WIN
82 ::SHAddToRecentDocs(SHARD_PATHW, path.get());
83 #elif defined(MOZ_WIDGET_GTK)
84 GtkRecentManager* manager = gtk_recent_manager_get_default();
86 gchar* uri = g_filename_to_uri(NS_ConvertUTF16toUTF8(path).get(),
87 nullptr, nullptr);
88 if (uri) {
89 gtk_recent_manager_add_item(manager, uri);
90 g_free(uri);
91 }
92 #endif
93 }
94 #ifdef MOZ_ENABLE_GIO
95 // Use GIO to store the source URI for later display in the file manager.
96 GFile* gio_file = g_file_new_for_path(NS_ConvertUTF16toUTF8(path).get());
97 nsCString source_uri;
98 aSource->GetSpec(source_uri);
99 GFileInfo *file_info = g_file_info_new();
100 g_file_info_set_attribute_string(file_info, "metadata::download-uri", source_uri.get());
101 g_file_set_attributes_async(gio_file,
102 file_info,
103 G_FILE_QUERY_INFO_NONE,
104 G_PRIORITY_DEFAULT,
105 nullptr, gio_set_metadata_done, nullptr);
106 g_object_unref(file_info);
107 g_object_unref(gio_file);
108 #endif
109 }
110 #endif
111 #ifdef XP_MACOSX
112 // On OS X, make the downloads stack bounce.
113 CFStringRef observedObject = ::CFStringCreateWithCString(kCFAllocatorDefault,
114 NS_ConvertUTF16toUTF8(path).get(),
115 kCFStringEncodingUTF8);
116 CFNotificationCenterRef center = ::CFNotificationCenterGetDistributedCenter();
117 ::CFNotificationCenterPostNotification(center, CFSTR("com.apple.DownloadFileFinished"),
118 observedObject, nullptr, TRUE);
119 ::CFRelease(observedObject);
120 #endif
121 #ifdef MOZ_WIDGET_ANDROID
122 if (!aContentType.IsEmpty()) {
123 mozilla::widget::android::GeckoAppShell::ScanMedia(path, NS_ConvertUTF8toUTF16(aContentType));
124 }
125 #endif
126 }
128 #ifdef XP_WIN
129 // Adjust file attributes so that by default, new files are indexed by
130 // desktop search services. Skip off those that land in the temp folder.
131 nsCOMPtr<nsIFile> tempDir, fileDir;
132 nsresult rv = NS_GetSpecialDirectory(NS_OS_TEMP_DIR, getter_AddRefs(tempDir));
133 NS_ENSURE_SUCCESS(rv, rv);
134 aTarget->GetParent(getter_AddRefs(fileDir));
136 bool isTemp = false;
137 if (fileDir) {
138 fileDir->Equals(tempDir, &isTemp);
139 }
141 nsCOMPtr<nsILocalFileWin> localFileWin(do_QueryInterface(aTarget));
142 if (!isTemp && localFileWin) {
143 localFileWin->SetFileAttributesWin(nsILocalFileWin::WFA_SEARCH_INDEXED);
144 }
145 #endif
147 #endif
149 return NS_OK;
150 }
152 nsresult DownloadPlatform::MapUrlToZone(const nsAString& aURL,
153 uint32_t* aZone)
154 {
155 #ifdef XP_WIN
156 nsRefPtr<IInternetSecurityManager> inetSecMgr;
157 if (FAILED(CoCreateInstance(CLSID_InternetSecurityManager, NULL,
158 CLSCTX_ALL, IID_IInternetSecurityManager,
159 getter_AddRefs(inetSecMgr)))) {
160 return NS_ERROR_UNEXPECTED;
161 }
163 DWORD zone;
164 if (inetSecMgr->MapUrlToZone(PromiseFlatString(aURL).get(),
165 &zone, 0) != S_OK) {
166 return NS_ERROR_UNEXPECTED;
167 } else {
168 *aZone = zone;
169 }
171 return NS_OK;
172 #else
173 return NS_ERROR_NOT_IMPLEMENTED;
174 #endif
175 }
