OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 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 | 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 "base/test/test_file_util.h" | 5 #include "base/test/test_file_util.h" |
6 | 6 |
7 #include <errno.h> | 7 #include <errno.h> |
8 #include <fcntl.h> | 8 #include <fcntl.h> |
9 #include <sys/stat.h> | 9 #include <sys/stat.h> |
10 #include <sys/types.h> | 10 #include <sys/types.h> |
(...skipping 14 matching lines...) Expand all Loading... |
25 bool DenyFilePermission(const FilePath& path, mode_t permission) { | 25 bool DenyFilePermission(const FilePath& path, mode_t permission) { |
26 struct stat stat_buf; | 26 struct stat stat_buf; |
27 if (stat(path.value().c_str(), &stat_buf) != 0) | 27 if (stat(path.value().c_str(), &stat_buf) != 0) |
28 return false; | 28 return false; |
29 stat_buf.st_mode &= ~permission; | 29 stat_buf.st_mode &= ~permission; |
30 | 30 |
31 int rv = HANDLE_EINTR(chmod(path.value().c_str(), stat_buf.st_mode)); | 31 int rv = HANDLE_EINTR(chmod(path.value().c_str(), stat_buf.st_mode)); |
32 return rv == 0; | 32 return rv == 0; |
33 } | 33 } |
34 | 34 |
| 35 // Gets a blob indicating the permission information for |path|. |
| 36 // |length| is the length of the blob. Zero on failure. |
| 37 // Returns the blob pointer, or NULL on failure. |
| 38 void* GetPermissionInfo(const FilePath& path, size_t* length) { |
| 39 DCHECK(length); |
| 40 *length = 0; |
| 41 |
| 42 struct stat stat_buf; |
| 43 if (stat(path.value().c_str(), &stat_buf) != 0) |
| 44 return NULL; |
| 45 |
| 46 *length = sizeof(mode_t); |
| 47 mode_t* mode = new mode_t; |
| 48 *mode = stat_buf.st_mode & ~S_IFMT; // Filter out file/path kind. |
| 49 |
| 50 return mode; |
| 51 } |
| 52 |
| 53 // Restores the permission information for |path|, given the blob retrieved |
| 54 // using |GetPermissionInfo()|. |
| 55 // |info| is the pointer to the blob. |
| 56 // |length| is the length of the blob. |
| 57 // Either |info| or |length| may be NULL/0, in which case nothing happens. |
| 58 bool RestorePermissionInfo(const FilePath& path, void* info, size_t length) { |
| 59 if (!info || (length == 0)) |
| 60 return false; |
| 61 |
| 62 DCHECK_EQ(sizeof(mode_t), length); |
| 63 mode_t* mode = reinterpret_cast<mode_t*>(info); |
| 64 |
| 65 int rv = HANDLE_EINTR(chmod(path.value().c_str(), *mode)); |
| 66 |
| 67 delete mode; |
| 68 |
| 69 return rv == 0; |
| 70 } |
| 71 |
35 } // namespace | 72 } // namespace |
36 | 73 |
37 bool DieFileDie(const FilePath& file, bool recurse) { | 74 bool DieFileDie(const FilePath& file, bool recurse) { |
38 // There is no need to workaround Windows problems on POSIX. | 75 // There is no need to workaround Windows problems on POSIX. |
39 // Just pass-through. | 76 // Just pass-through. |
40 return file_util::Delete(file, recurse); | 77 return file_util::Delete(file, recurse); |
41 } | 78 } |
42 | 79 |
43 // Mostly a verbatim copy of CopyDirectory | 80 // Mostly a verbatim copy of CopyDirectory |
44 bool CopyRecursiveDirNoCache(const FilePath& source_dir, | 81 bool CopyRecursiveDirNoCache(const FilePath& source_dir, |
(...skipping 88 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
133 } | 170 } |
134 | 171 |
135 bool MakeFileUnreadable(const FilePath& path) { | 172 bool MakeFileUnreadable(const FilePath& path) { |
136 return DenyFilePermission(path, S_IRUSR | S_IRGRP | S_IROTH); | 173 return DenyFilePermission(path, S_IRUSR | S_IRGRP | S_IROTH); |
137 } | 174 } |
138 | 175 |
139 bool MakeFileUnwritable(const FilePath& path) { | 176 bool MakeFileUnwritable(const FilePath& path) { |
140 return DenyFilePermission(path, S_IWUSR | S_IWGRP | S_IWOTH); | 177 return DenyFilePermission(path, S_IWUSR | S_IWGRP | S_IWOTH); |
141 } | 178 } |
142 | 179 |
| 180 PermissionRestorer::PermissionRestorer(const FilePath& path) |
| 181 : path_(path), info_(NULL), length_(0) { |
| 182 info_ = GetPermissionInfo(path_, &length_); |
| 183 DCHECK(info_ != NULL); |
| 184 DCHECK_NE(0u, length_); |
| 185 } |
| 186 |
| 187 PermissionRestorer::~PermissionRestorer() { |
| 188 if (!RestorePermissionInfo(path_, info_, length_)) |
| 189 NOTREACHED(); |
| 190 } |
| 191 |
143 } // namespace file_util | 192 } // namespace file_util |
OLD | NEW |