levin@chromium.org | 5c52868 | 2011-03-28 10:54:15 +0900 | [diff] [blame] | 1 | // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
sgk@google.com | e1f4a24 | 2009-04-22 02:20:10 +0900 | [diff] [blame] | 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
thestig@chromium.org | a5a2b6e | 2009-07-17 14:55:51 +0900 | [diff] [blame] | 5 | #include "base/linux_util.h" |
sgk@google.com | e1f4a24 | 2009-04-22 02:20:10 +0900 | [diff] [blame] | 6 | |
thestig@chromium.org | 2feb1c8 | 2009-10-29 13:02:55 +0900 | [diff] [blame] | 7 | #include <dirent.h> |
| 8 | #include <errno.h> |
thestig@chromium.org | 44836d4 | 2010-07-17 04:28:17 +0900 | [diff] [blame] | 9 | #include <fcntl.h> |
thestig@chromium.org | eea8306 | 2009-12-02 17:45:01 +0900 | [diff] [blame] | 10 | #include <glib.h> |
sgk@google.com | e1f4a24 | 2009-04-22 02:20:10 +0900 | [diff] [blame] | 11 | #include <stdlib.h> |
thestig@chromium.org | 2feb1c8 | 2009-10-29 13:02:55 +0900 | [diff] [blame] | 12 | #include <sys/stat.h> |
thestig@chromium.org | 44836d4 | 2010-07-17 04:28:17 +0900 | [diff] [blame] | 13 | #include <sys/types.h> |
thestig@chromium.org | 2feb1c8 | 2009-10-29 13:02:55 +0900 | [diff] [blame] | 14 | #include <unistd.h> |
sgk@google.com | e1f4a24 | 2009-04-22 02:20:10 +0900 | [diff] [blame] | 15 | |
thestig@chromium.org | a5a2b6e | 2009-07-17 14:55:51 +0900 | [diff] [blame] | 16 | #include <vector> |
| 17 | |
| 18 | #include "base/command_line.h" |
thestig@chromium.org | 44836d4 | 2010-07-17 04:28:17 +0900 | [diff] [blame] | 19 | #include "base/file_util.h" |
levin@chromium.org | 5c52868 | 2011-03-28 10:54:15 +0900 | [diff] [blame] | 20 | #include "base/memory/scoped_ptr.h" |
| 21 | #include "base/memory/singleton.h" |
thestig@chromium.org | eea8306 | 2009-12-02 17:45:01 +0900 | [diff] [blame] | 22 | #include "base/path_service.h" |
thestig@chromium.org | a5a2b6e | 2009-07-17 14:55:51 +0900 | [diff] [blame] | 23 | #include "base/process_util.h" |
mattm@chromium.org | 625865e | 2009-07-22 09:22:49 +0900 | [diff] [blame] | 24 | #include "base/string_util.h" |
brettw@chromium.org | abe477a | 2011-01-21 13:55:52 +0900 | [diff] [blame] | 25 | #include "base/synchronization/lock.h" |
mattm@chromium.org | 625865e | 2009-07-22 09:22:49 +0900 | [diff] [blame] | 26 | |
| 27 | namespace { |
| 28 | |
thestig@chromium.org | 9d0e95c | 2009-10-22 04:04:17 +0900 | [diff] [blame] | 29 | // Not needed for OS_CHROMEOS. |
| 30 | #if defined(OS_LINUX) |
thestig@chromium.org | 741196d | 2009-10-13 11:51:07 +0900 | [diff] [blame] | 31 | enum LinuxDistroState { |
| 32 | STATE_DID_NOT_CHECK = 0, |
| 33 | STATE_CHECK_STARTED = 1, |
| 34 | STATE_CHECK_FINISHED = 2, |
| 35 | }; |
| 36 | |
| 37 | // Helper class for GetLinuxDistro(). |
| 38 | class LinuxDistroHelper { |
| 39 | public: |
| 40 | // Retrieves the Singleton. |
satish@chromium.org | ce3f487 | 2010-12-14 01:52:35 +0900 | [diff] [blame] | 41 | static LinuxDistroHelper* GetInstance() { |
thestig@chromium.org | 741196d | 2009-10-13 11:51:07 +0900 | [diff] [blame] | 42 | return Singleton<LinuxDistroHelper>::get(); |
| 43 | } |
| 44 | |
| 45 | // The simple state machine goes from: |
| 46 | // STATE_DID_NOT_CHECK -> STATE_CHECK_STARTED -> STATE_CHECK_FINISHED. |
| 47 | LinuxDistroHelper() : state_(STATE_DID_NOT_CHECK) {} |
| 48 | ~LinuxDistroHelper() {} |
| 49 | |
| 50 | // Retrieve the current state, if we're in STATE_DID_NOT_CHECK, |
| 51 | // we automatically move to STATE_CHECK_STARTED so nobody else will |
| 52 | // do the check. |
| 53 | LinuxDistroState State() { |
brettw@chromium.org | abe477a | 2011-01-21 13:55:52 +0900 | [diff] [blame] | 54 | base::AutoLock scoped_lock(lock_); |
thestig@chromium.org | 741196d | 2009-10-13 11:51:07 +0900 | [diff] [blame] | 55 | if (STATE_DID_NOT_CHECK == state_) { |
| 56 | state_ = STATE_CHECK_STARTED; |
| 57 | return STATE_DID_NOT_CHECK; |
| 58 | } |
| 59 | return state_; |
| 60 | } |
| 61 | |
| 62 | // Indicate the check finished, move to STATE_CHECK_FINISHED. |
| 63 | void CheckFinished() { |
brettw@chromium.org | abe477a | 2011-01-21 13:55:52 +0900 | [diff] [blame] | 64 | base::AutoLock scoped_lock(lock_); |
david.mike.futcher@gmail.com | 9eb2aa5 | 2011-04-19 05:07:08 +0900 | [diff] [blame] | 65 | DCHECK_EQ(STATE_CHECK_STARTED, state_); |
thestig@chromium.org | 741196d | 2009-10-13 11:51:07 +0900 | [diff] [blame] | 66 | state_ = STATE_CHECK_FINISHED; |
| 67 | } |
| 68 | |
| 69 | private: |
brettw@chromium.org | abe477a | 2011-01-21 13:55:52 +0900 | [diff] [blame] | 70 | base::Lock lock_; |
thestig@chromium.org | 741196d | 2009-10-13 11:51:07 +0900 | [diff] [blame] | 71 | LinuxDistroState state_; |
| 72 | }; |
thestig@chromium.org | 9d0e95c | 2009-10-22 04:04:17 +0900 | [diff] [blame] | 73 | #endif // if defined(OS_LINUX) |
thestig@chromium.org | 741196d | 2009-10-13 11:51:07 +0900 | [diff] [blame] | 74 | |
thestig@chromium.org | 2feb1c8 | 2009-10-29 13:02:55 +0900 | [diff] [blame] | 75 | // expected prefix of the target of the /proc/self/fd/%d link for a socket |
jhawkins@chromium.org | 88fbaf6 | 2012-01-28 09:34:40 +0900 | [diff] [blame] | 76 | const char kSocketLinkPrefix[] = "socket:["; |
thestig@chromium.org | 2feb1c8 | 2009-10-29 13:02:55 +0900 | [diff] [blame] | 77 | |
| 78 | // Parse a symlink in /proc/pid/fd/$x and return the inode number of the |
| 79 | // socket. |
| 80 | // inode_out: (output) set to the inode number on success |
| 81 | // path: e.g. /proc/1234/fd/5 (must be a UNIX domain socket descriptor) |
| 82 | // log: if true, log messages about failure details |
| 83 | bool ProcPathGetInode(ino_t* inode_out, const char* path, bool log = false) { |
| 84 | DCHECK(inode_out); |
| 85 | DCHECK(path); |
| 86 | |
| 87 | char buf[256]; |
| 88 | const ssize_t n = readlink(path, buf, sizeof(buf) - 1); |
| 89 | if (n == -1) { |
| 90 | if (log) { |
brettw@chromium.org | 5faed3c | 2011-10-27 06:48:00 +0900 | [diff] [blame] | 91 | DLOG(WARNING) << "Failed to read the inode number for a socket from /proc" |
thestig@chromium.org | 2feb1c8 | 2009-10-29 13:02:55 +0900 | [diff] [blame] | 92 | "(" << errno << ")"; |
| 93 | } |
| 94 | return false; |
| 95 | } |
| 96 | buf[n] = 0; |
| 97 | |
| 98 | if (memcmp(kSocketLinkPrefix, buf, sizeof(kSocketLinkPrefix) - 1)) { |
| 99 | if (log) { |
brettw@chromium.org | 5faed3c | 2011-10-27 06:48:00 +0900 | [diff] [blame] | 100 | DLOG(WARNING) << "The descriptor passed from the crashing process wasn't " |
| 101 | " a UNIX domain socket."; |
thestig@chromium.org | 2feb1c8 | 2009-10-29 13:02:55 +0900 | [diff] [blame] | 102 | } |
| 103 | return false; |
| 104 | } |
| 105 | |
| 106 | char *endptr; |
| 107 | const unsigned long long int inode_ul = |
| 108 | strtoull(buf + sizeof(kSocketLinkPrefix) - 1, &endptr, 10); |
| 109 | if (*endptr != ']') |
| 110 | return false; |
| 111 | |
| 112 | if (inode_ul == ULLONG_MAX) { |
| 113 | if (log) { |
brettw@chromium.org | 5faed3c | 2011-10-27 06:48:00 +0900 | [diff] [blame] | 114 | DLOG(WARNING) << "Failed to parse a socket's inode number: the number " |
| 115 | "was too large. Please report this bug: " << buf; |
thestig@chromium.org | 2feb1c8 | 2009-10-29 13:02:55 +0900 | [diff] [blame] | 116 | } |
| 117 | return false; |
| 118 | } |
| 119 | |
| 120 | *inode_out = inode_ul; |
| 121 | return true; |
| 122 | } |
| 123 | |
thestig@chromium.org | b6ba943 | 2010-04-03 10:05:39 +0900 | [diff] [blame] | 124 | } // namespace |
thestig@chromium.org | a5a2b6e | 2009-07-17 14:55:51 +0900 | [diff] [blame] | 125 | |
sgk@google.com | e1f4a24 | 2009-04-22 02:20:10 +0900 | [diff] [blame] | 126 | namespace base { |
| 127 | |
mnissler@chromium.org | 8584b6d | 2010-08-26 17:55:22 +0900 | [diff] [blame] | 128 | // Account for the terminating null character. |
| 129 | static const int kDistroSize = 128 + 1; |
| 130 | |
thestig@chromium.org | a5a2b6e | 2009-07-17 14:55:51 +0900 | [diff] [blame] | 131 | // We use this static string to hold the Linux distro info. If we |
| 132 | // crash, the crash handler code will send this in the crash dump. |
mnissler@chromium.org | 8584b6d | 2010-08-26 17:55:22 +0900 | [diff] [blame] | 133 | char g_linux_distro[kDistroSize] = |
saintlou@chromium.org | 6716159d | 2011-11-23 11:07:45 +0900 | [diff] [blame] | 134 | #if defined(OS_CHROMEOS) && defined(USE_AURA) |
| 135 | "CrOS Aura"; |
rbyers@chromium.org | 9f23568 | 2011-11-11 01:04:02 +0900 | [diff] [blame] | 136 | #elif defined(OS_CHROMEOS) |
thestig@chromium.org | 9d0e95c | 2009-10-22 04:04:17 +0900 | [diff] [blame] | 137 | "CrOS"; |
| 138 | #else // if defined(OS_LINUX) |
| 139 | "Unknown"; |
| 140 | #endif |
thestig@chromium.org | a5a2b6e | 2009-07-17 14:55:51 +0900 | [diff] [blame] | 141 | |
| 142 | std::string GetLinuxDistro() { |
thestig@chromium.org | 9d0e95c | 2009-10-22 04:04:17 +0900 | [diff] [blame] | 143 | #if defined(OS_CHROMEOS) |
mnissler@chromium.org | 8584b6d | 2010-08-26 17:55:22 +0900 | [diff] [blame] | 144 | return g_linux_distro; |
pvalchev@google.com | 1d919db | 2010-03-10 16:46:43 +0900 | [diff] [blame] | 145 | #elif defined(OS_LINUX) |
satish@chromium.org | ce3f487 | 2010-12-14 01:52:35 +0900 | [diff] [blame] | 146 | LinuxDistroHelper* distro_state_singleton = LinuxDistroHelper::GetInstance(); |
thestig@chromium.org | 741196d | 2009-10-13 11:51:07 +0900 | [diff] [blame] | 147 | LinuxDistroState state = distro_state_singleton->State(); |
| 148 | if (STATE_DID_NOT_CHECK == state) { |
| 149 | // We do this check only once per process. If it fails, there's |
| 150 | // little reason to believe it will work if we attempt to run |
| 151 | // lsb_release again. |
thestig@chromium.org | a5a2b6e | 2009-07-17 14:55:51 +0900 | [diff] [blame] | 152 | std::vector<std::string> argv; |
| 153 | argv.push_back("lsb_release"); |
| 154 | argv.push_back("-d"); |
| 155 | std::string output; |
| 156 | base::GetAppOutput(CommandLine(argv), &output); |
| 157 | if (output.length() > 0) { |
| 158 | // lsb_release -d should return: Description:<tab>Distro Info |
thakis@chromium.org | 96473bc | 2011-11-10 06:55:08 +0900 | [diff] [blame] | 159 | const char field[] = "Description:\t"; |
| 160 | if (output.compare(0, strlen(field), field) == 0) { |
| 161 | SetLinuxDistro(output.substr(strlen(field))); |
thestig@chromium.org | b5299f1 | 2009-10-22 07:09:33 +0900 | [diff] [blame] | 162 | } |
thestig@chromium.org | a5a2b6e | 2009-07-17 14:55:51 +0900 | [diff] [blame] | 163 | } |
thestig@chromium.org | 741196d | 2009-10-13 11:51:07 +0900 | [diff] [blame] | 164 | distro_state_singleton->CheckFinished(); |
mnissler@chromium.org | 8584b6d | 2010-08-26 17:55:22 +0900 | [diff] [blame] | 165 | return g_linux_distro; |
thestig@chromium.org | 741196d | 2009-10-13 11:51:07 +0900 | [diff] [blame] | 166 | } else if (STATE_CHECK_STARTED == state) { |
| 167 | // If the distro check above is in progress in some other thread, we're |
| 168 | // not going to wait for the results. |
| 169 | return "Unknown"; |
| 170 | } else { |
| 171 | // In STATE_CHECK_FINISHED, no more writing to |linux_distro|. |
mnissler@chromium.org | 8584b6d | 2010-08-26 17:55:22 +0900 | [diff] [blame] | 172 | return g_linux_distro; |
thestig@chromium.org | a5a2b6e | 2009-07-17 14:55:51 +0900 | [diff] [blame] | 173 | } |
pvalchev@google.com | 1d919db | 2010-03-10 16:46:43 +0900 | [diff] [blame] | 174 | #else |
| 175 | NOTIMPLEMENTED(); |
robert.nagy@gmail.com | ea54e46 | 2011-10-25 07:05:27 +0900 | [diff] [blame] | 176 | return "Unknown"; |
thestig@chromium.org | 9d0e95c | 2009-10-22 04:04:17 +0900 | [diff] [blame] | 177 | #endif |
thestig@chromium.org | a5a2b6e | 2009-07-17 14:55:51 +0900 | [diff] [blame] | 178 | } |
| 179 | |
mnissler@chromium.org | 8584b6d | 2010-08-26 17:55:22 +0900 | [diff] [blame] | 180 | void SetLinuxDistro(const std::string& distro) { |
| 181 | std::string trimmed_distro; |
| 182 | TrimWhitespaceASCII(distro, TRIM_ALL, &trimmed_distro); |
| 183 | base::strlcpy(g_linux_distro, trimmed_distro.c_str(), kDistroSize); |
| 184 | } |
| 185 | |
thestig@chromium.org | 2feb1c8 | 2009-10-29 13:02:55 +0900 | [diff] [blame] | 186 | bool FileDescriptorGetInode(ino_t* inode_out, int fd) { |
| 187 | DCHECK(inode_out); |
| 188 | |
| 189 | struct stat buf; |
| 190 | if (fstat(fd, &buf) < 0) |
| 191 | return false; |
| 192 | |
| 193 | if (!S_ISSOCK(buf.st_mode)) |
| 194 | return false; |
| 195 | |
| 196 | *inode_out = buf.st_ino; |
| 197 | return true; |
| 198 | } |
| 199 | |
| 200 | bool FindProcessHoldingSocket(pid_t* pid_out, ino_t socket_inode) { |
| 201 | DCHECK(pid_out); |
| 202 | bool already_found = false; |
| 203 | |
| 204 | DIR* proc = opendir("/proc"); |
| 205 | if (!proc) { |
brettw@chromium.org | 5faed3c | 2011-10-27 06:48:00 +0900 | [diff] [blame] | 206 | DLOG(WARNING) << "Cannot open /proc"; |
thestig@chromium.org | 2feb1c8 | 2009-10-29 13:02:55 +0900 | [diff] [blame] | 207 | return false; |
| 208 | } |
| 209 | |
| 210 | std::vector<pid_t> pids; |
| 211 | |
| 212 | struct dirent* dent; |
| 213 | while ((dent = readdir(proc))) { |
| 214 | char *endptr; |
| 215 | const unsigned long int pid_ul = strtoul(dent->d_name, &endptr, 10); |
| 216 | if (pid_ul == ULONG_MAX || *endptr) |
| 217 | continue; |
| 218 | pids.push_back(pid_ul); |
| 219 | } |
| 220 | closedir(proc); |
| 221 | |
| 222 | for (std::vector<pid_t>::const_iterator |
| 223 | i = pids.begin(); i != pids.end(); ++i) { |
| 224 | const pid_t current_pid = *i; |
| 225 | char buf[256]; |
| 226 | snprintf(buf, sizeof(buf), "/proc/%d/fd", current_pid); |
| 227 | DIR* fd = opendir(buf); |
| 228 | if (!fd) |
| 229 | continue; |
| 230 | |
| 231 | while ((dent = readdir(fd))) { |
| 232 | if (snprintf(buf, sizeof(buf), "/proc/%d/fd/%s", current_pid, |
| 233 | dent->d_name) >= static_cast<int>(sizeof(buf))) { |
| 234 | continue; |
| 235 | } |
| 236 | |
| 237 | ino_t fd_inode; |
| 238 | if (ProcPathGetInode(&fd_inode, buf)) { |
| 239 | if (fd_inode == socket_inode) { |
| 240 | if (already_found) { |
| 241 | closedir(fd); |
| 242 | return false; |
| 243 | } |
| 244 | |
| 245 | already_found = true; |
| 246 | *pid_out = current_pid; |
| 247 | break; |
| 248 | } |
| 249 | } |
| 250 | } |
| 251 | |
| 252 | closedir(fd); |
| 253 | } |
| 254 | |
| 255 | return already_found; |
| 256 | } |
| 257 | |
kmixter@chromium.org | e406ed7 | 2011-06-21 13:21:06 +0900 | [diff] [blame] | 258 | pid_t FindThreadIDWithSyscall(pid_t pid, const std::string& expected_data, |
| 259 | bool* syscall_supported) { |
thestig@chromium.org | 44836d4 | 2010-07-17 04:28:17 +0900 | [diff] [blame] | 260 | char buf[256]; |
| 261 | snprintf(buf, sizeof(buf), "/proc/%d/task", pid); |
kmixter@chromium.org | e406ed7 | 2011-06-21 13:21:06 +0900 | [diff] [blame] | 262 | |
| 263 | if (syscall_supported != NULL) |
| 264 | *syscall_supported = false; |
| 265 | |
thestig@chromium.org | 44836d4 | 2010-07-17 04:28:17 +0900 | [diff] [blame] | 266 | DIR* task = opendir(buf); |
| 267 | if (!task) { |
brettw@chromium.org | 5faed3c | 2011-10-27 06:48:00 +0900 | [diff] [blame] | 268 | DLOG(WARNING) << "Cannot open " << buf; |
thestig@chromium.org | 44836d4 | 2010-07-17 04:28:17 +0900 | [diff] [blame] | 269 | return -1; |
| 270 | } |
| 271 | |
| 272 | std::vector<pid_t> tids; |
| 273 | struct dirent* dent; |
| 274 | while ((dent = readdir(task))) { |
| 275 | char *endptr; |
| 276 | const unsigned long int tid_ul = strtoul(dent->d_name, &endptr, 10); |
| 277 | if (tid_ul == ULONG_MAX || *endptr) |
| 278 | continue; |
| 279 | tids.push_back(tid_ul); |
| 280 | } |
| 281 | closedir(task); |
| 282 | |
| 283 | scoped_array<char> syscall_data(new char[expected_data.length()]); |
| 284 | for (std::vector<pid_t>::const_iterator |
| 285 | i = tids.begin(); i != tids.end(); ++i) { |
| 286 | const pid_t current_tid = *i; |
| 287 | snprintf(buf, sizeof(buf), "/proc/%d/task/%d/syscall", pid, current_tid); |
| 288 | int fd = open(buf, O_RDONLY); |
| 289 | if (fd < 0) |
| 290 | continue; |
kmixter@chromium.org | e406ed7 | 2011-06-21 13:21:06 +0900 | [diff] [blame] | 291 | if (syscall_supported != NULL) |
| 292 | *syscall_supported = true; |
thestig@chromium.org | 44836d4 | 2010-07-17 04:28:17 +0900 | [diff] [blame] | 293 | bool read_ret = |
| 294 | file_util::ReadFromFD(fd, syscall_data.get(), expected_data.length()); |
| 295 | close(fd); |
| 296 | if (!read_ret) |
| 297 | continue; |
| 298 | |
| 299 | if (0 == strncmp(expected_data.c_str(), syscall_data.get(), |
| 300 | expected_data.length())) { |
| 301 | return current_tid; |
| 302 | } |
| 303 | } |
| 304 | return -1; |
| 305 | } |
| 306 | |
sgk@google.com | e1f4a24 | 2009-04-22 02:20:10 +0900 | [diff] [blame] | 307 | } // namespace base |