| The Android Open Source Project | 5c11852 | 2008-10-21 07:00:00 -0700 | [diff] [blame^] | 1 | #include <stdio.h> |
| 2 | #include <stdlib.h> |
| 3 | #include <string.h> |
| 4 | #include <unistd.h> |
| 5 | #include <ctype.h> |
| 6 | #include "bitvector.h" |
| 7 | #include "parse_options.h" |
| 8 | #include "hash_table.h" |
| 9 | |
| 10 | const char *root = ""; |
| 11 | bool lump_kernel; |
| 12 | bool lump_libraries; |
| 13 | Bitvector pid_include_vector(32768); |
| 14 | Bitvector pid_exclude_vector(32768); |
| 15 | bool include_some_pids; |
| 16 | bool exclude_some_pids; |
| 17 | |
| 18 | HashTable<int> excluded_procedures(2000); |
| 19 | HashTable<int> included_procedures(2000); |
| 20 | bool exclude_some_procedures; |
| 21 | bool include_some_procedures; |
| 22 | |
| 23 | bool exclude_kernel_syms; |
| 24 | bool exclude_library_syms; |
| 25 | bool include_kernel_syms; |
| 26 | bool include_library_syms; |
| 27 | bool demangle = true; |
| 28 | |
| 29 | static const char *OptionsUsageStr = |
| 30 | " -e :kernel exclude all kernel symbols\n" |
| 31 | " -e :libs exclude all library symbols\n" |
| 32 | " -e <func> exclude function <func>\n" |
| 33 | " -e <pid> exclude process <pid>\n" |
| 34 | " -i :kernel include all kernel symbols\n" |
| 35 | " -i :libs include all library symbols\n" |
| 36 | " -i <func> include function <func>\n" |
| 37 | " -i <pid> include process <pid>\n" |
| 38 | " -l :kernel lump all the kernel symbols together\n" |
| 39 | " -l :libs lump all the library symbols together\n" |
| 40 | " -m do not demangle C++ symbols (m for 'mangle')\n" |
| 41 | " -r <root> use <root> as the path for finding ELF executables\n" |
| 42 | ; |
| 43 | |
| 44 | void OptionsUsage() |
| 45 | { |
| 46 | fprintf(stderr, "%s", OptionsUsageStr); |
| 47 | } |
| 48 | |
| 49 | void ParseOptions(int argc, char **argv) |
| 50 | { |
| 51 | bool err = false; |
| 52 | while (!err) { |
| 53 | int opt = getopt(argc, argv, "+e:i:l:mr:"); |
| 54 | if (opt == -1) |
| 55 | break; |
| 56 | switch (opt) { |
| 57 | case 'e': |
| 58 | if (*optarg == ':') { |
| 59 | if (strcmp(optarg, ":kernel") == 0) |
| 60 | exclude_kernel_syms = true; |
| 61 | else if (strcmp(optarg, ":libs") == 0) |
| 62 | exclude_library_syms = true; |
| 63 | else |
| 64 | err = true; |
| 65 | excluded_procedures.Update(optarg, 1); |
| 66 | exclude_some_procedures = true; |
| 67 | } else if (isdigit(*optarg)) { |
| 68 | int bitnum = atoi(optarg); |
| 69 | pid_exclude_vector.SetBit(bitnum); |
| 70 | exclude_some_pids = true; |
| 71 | } else { |
| 72 | excluded_procedures.Update(optarg, 1); |
| 73 | exclude_some_procedures = true; |
| 74 | } |
| 75 | break; |
| 76 | case 'i': |
| 77 | if (*optarg == ':') { |
| 78 | if (strcmp(optarg, ":kernel") == 0) |
| 79 | include_kernel_syms = true; |
| 80 | else if (strcmp(optarg, ":libs") == 0) |
| 81 | include_library_syms = true; |
| 82 | else |
| 83 | err = true; |
| 84 | included_procedures.Update(optarg, 1); |
| 85 | include_some_procedures = true; |
| 86 | } else if (isdigit(*optarg)) { |
| 87 | int bitnum = atoi(optarg); |
| 88 | pid_include_vector.SetBit(bitnum); |
| 89 | include_some_pids = true; |
| 90 | } else { |
| 91 | included_procedures.Update(optarg, 1); |
| 92 | include_some_procedures = true; |
| 93 | } |
| 94 | break; |
| 95 | case 'l': |
| 96 | if (strcmp(optarg, ":kernel") == 0) |
| 97 | lump_kernel = true; |
| 98 | else if (strcmp(optarg, ":libs") == 0) |
| 99 | lump_libraries = true; |
| 100 | else |
| 101 | err = true; |
| 102 | break; |
| 103 | case 'm': |
| 104 | demangle = false; |
| 105 | break; |
| 106 | case 'r': |
| 107 | root = optarg; |
| 108 | break; |
| 109 | default: |
| 110 | err = true; |
| 111 | break; |
| 112 | } |
| 113 | } |
| 114 | |
| 115 | if (err) { |
| 116 | Usage(argv[0]); |
| 117 | exit(1); |
| 118 | } |
| 119 | } |