| OLD | NEW |
| (Empty) |
| 1 // Copyright 2014 The Crashpad Authors. All rights reserved. | |
| 2 // | |
| 3 // Licensed under the Apache License, Version 2.0 (the "License"); | |
| 4 // you may not use this file except in compliance with the License. | |
| 5 // You may obtain a copy of the License at | |
| 6 // | |
| 7 // http://www.apache.org/licenses/LICENSE-2.0 | |
| 8 // | |
| 9 // Unless required by applicable law or agreed to in writing, software | |
| 10 // distributed under the License is distributed on an "AS IS" BASIS, | |
| 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
| 12 // See the License for the specific language governing permissions and | |
| 13 // limitations under the License. | |
| 14 | |
| 15 #include "util/posix/process_util.h" | |
| 16 | |
| 17 #include <string.h> | |
| 18 #include <sys/sysctl.h> | |
| 19 #include <sys/types.h> | |
| 20 | |
| 21 #include "base/basictypes.h" | |
| 22 | |
| 23 namespace crashpad { | |
| 24 | |
| 25 bool ProcessArgumentsForPID(pid_t pid, std::vector<std::string>* argv) { | |
| 26 // The format of KERN_PROCARGS2 is explained in 10.9.2 adv_cmds-153/ps/print.c | |
| 27 // getproclline(). It is an int (argc) followed by the executable’s string | |
| 28 // area. The string area consists of NUL-terminated strings, beginning with | |
| 29 // the executable path, and then starting on an aligned boundary, all of the | |
| 30 // elements of argv, envp, and applev. | |
| 31 | |
| 32 // It is possible for a process to exec() in between the two sysctl() calls | |
| 33 // below. If that happens, and the string area of the new program is larger | |
| 34 // than that of the old one, args_size_estimate will be too small. To detect | |
| 35 // this situation, the second sysctl() attempts to fetch args_size_estimate + | |
| 36 // 1 bytes, expecting to only receive args_size_estimate. If it gets the extra | |
| 37 // byte, it indicates that the string area has grown, and the sysctl() pair | |
| 38 // will be retried a limited number of times. | |
| 39 | |
| 40 size_t args_size_estimate; | |
| 41 size_t args_size; | |
| 42 std::string args; | |
| 43 int tries = 3; | |
| 44 do { | |
| 45 int mib[] = {CTL_KERN, KERN_PROCARGS2, pid}; | |
| 46 int rv = | |
| 47 sysctl(mib, arraysize(mib), nullptr, &args_size_estimate, nullptr, 0); | |
| 48 if (rv != 0) { | |
| 49 return false; | |
| 50 } | |
| 51 | |
| 52 args_size = args_size_estimate + 1; | |
| 53 args.resize(args_size); | |
| 54 rv = sysctl(mib, arraysize(mib), &args[0], &args_size, nullptr, 0); | |
| 55 if (rv != 0) { | |
| 56 return false; | |
| 57 } | |
| 58 } while (args_size == args_size_estimate + 1 && tries--); | |
| 59 | |
| 60 if (args_size == args_size_estimate + 1) { | |
| 61 return false; | |
| 62 } | |
| 63 | |
| 64 // KERN_PROCARGS2 needs to at least contain argc. | |
| 65 if (args_size < sizeof(int)) { | |
| 66 return false; | |
| 67 } | |
| 68 args.resize(args_size); | |
| 69 | |
| 70 // Get argc. | |
| 71 int argc; | |
| 72 memcpy(&argc, &args[0], sizeof(argc)); | |
| 73 | |
| 74 // Find the end of the executable path. | |
| 75 size_t start_pos = sizeof(argc); | |
| 76 size_t nul_pos = args.find('\0', start_pos); | |
| 77 if (nul_pos == std::string::npos) { | |
| 78 return false; | |
| 79 } | |
| 80 | |
| 81 // Find the beginning of the string area. | |
| 82 start_pos = args.find_first_not_of('\0', nul_pos); | |
| 83 if (start_pos == std::string::npos) { | |
| 84 return false; | |
| 85 } | |
| 86 | |
| 87 std::vector<std::string> local_argv; | |
| 88 while (argc-- && nul_pos != std::string::npos) { | |
| 89 nul_pos = args.find('\0', start_pos); | |
| 90 local_argv.push_back(args.substr(start_pos, nul_pos - start_pos)); | |
| 91 start_pos = nul_pos + 1; | |
| 92 } | |
| 93 | |
| 94 if (argc >= 0) { | |
| 95 // Not every argument was recovered. | |
| 96 return false; | |
| 97 } | |
| 98 | |
| 99 argv->swap(local_argv); | |
| 100 return true; | |
| 101 } | |
| 102 | |
| 103 } // namespace crashpad | |
| OLD | NEW |