OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/shell_integration_linux.h" | 5 #include "chrome/browser/shell_integration_linux.h" |
6 | 6 |
7 #include <fcntl.h> | 7 #include <fcntl.h> |
8 | 8 |
9 #if defined(USE_GLIB) | 9 #if defined(USE_GLIB) |
10 #include <glib.h> | 10 #include <glib.h> |
(...skipping 145 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
156 | 156 |
157 int fd = openat(desktop_fd, shortcut_filename.value().c_str(), | 157 int fd = openat(desktop_fd, shortcut_filename.value().c_str(), |
158 O_CREAT | O_EXCL | O_WRONLY, | 158 O_CREAT | O_EXCL | O_WRONLY, |
159 S_IRWXU | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH); | 159 S_IRWXU | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH); |
160 if (fd < 0) { | 160 if (fd < 0) { |
161 if (IGNORE_EINTR(close(desktop_fd)) < 0) | 161 if (IGNORE_EINTR(close(desktop_fd)) < 0) |
162 PLOG(ERROR) << "close"; | 162 PLOG(ERROR) << "close"; |
163 return false; | 163 return false; |
164 } | 164 } |
165 | 165 |
166 ssize_t bytes_written = base::WriteFileDescriptor(fd, contents.data(), | 166 if (!base::WriteFileDescriptor(fd, contents.c_str(), contents.size())) { |
167 contents.length()); | |
168 if (IGNORE_EINTR(close(fd)) < 0) | |
169 PLOG(ERROR) << "close"; | |
170 | |
171 if (bytes_written != static_cast<ssize_t>(contents.length())) { | |
172 // Delete the file. No shortuct is better than corrupted one. Use unlinkat | 167 // Delete the file. No shortuct is better than corrupted one. Use unlinkat |
173 // to make sure we're deleting the file in the directory we think we are. | 168 // to make sure we're deleting the file in the directory we think we are. |
174 // Even if an attacker manager to put something other at | 169 // Even if an attacker manager to put something other at |
175 // |shortcut_filename| we'll just undo his action. | 170 // |shortcut_filename| we'll just undo his action. |
176 unlinkat(desktop_fd, shortcut_filename.value().c_str(), 0); | 171 unlinkat(desktop_fd, shortcut_filename.value().c_str(), 0); |
177 } | 172 } |
178 | 173 |
| 174 if (IGNORE_EINTR(close(fd)) < 0) |
| 175 PLOG(ERROR) << "close"; |
| 176 |
179 if (IGNORE_EINTR(close(desktop_fd)) < 0) | 177 if (IGNORE_EINTR(close(desktop_fd)) < 0) |
180 PLOG(ERROR) << "close"; | 178 PLOG(ERROR) << "close"; |
181 | 179 |
182 return true; | 180 return true; |
183 } | 181 } |
184 | 182 |
185 void DeleteShortcutOnDesktop(const base::FilePath& shortcut_filename) { | 183 void DeleteShortcutOnDesktop(const base::FilePath& shortcut_filename) { |
186 base::FilePath desktop_path; | 184 base::FilePath desktop_path; |
187 if (PathService::Get(base::DIR_USER_DESKTOP, &desktop_path)) | 185 if (PathService::Get(base::DIR_USER_DESKTOP, &desktop_path)) |
188 base::DeleteFile(desktop_path.Append(shortcut_filename), false); | 186 base::DeleteFile(desktop_path.Append(shortcut_filename), false); |
(...skipping 885 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1074 for (std::vector<base::FilePath>::const_iterator it = | 1072 for (std::vector<base::FilePath>::const_iterator it = |
1075 shortcut_filenames_app_menu.begin(); | 1073 shortcut_filenames_app_menu.begin(); |
1076 it != shortcut_filenames_app_menu.end(); ++it) { | 1074 it != shortcut_filenames_app_menu.end(); ++it) { |
1077 DeleteShortcutInApplicationsMenu(*it, | 1075 DeleteShortcutInApplicationsMenu(*it, |
1078 base::FilePath(kDirectoryFilename)); | 1076 base::FilePath(kDirectoryFilename)); |
1079 } | 1077 } |
1080 } | 1078 } |
1081 } | 1079 } |
1082 | 1080 |
1083 } // namespace shell_integration_linux | 1081 } // namespace shell_integration_linux |
OLD | NEW |