120 lines
3.5 KiB
C++
120 lines
3.5 KiB
C++
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#include <ctype.h>
|
|
#include "bitvector.h"
|
|
#include "parse_options.h"
|
|
#include "hash_table.h"
|
|
|
|
const char *root = "";
|
|
bool lump_kernel;
|
|
bool lump_libraries;
|
|
Bitvector pid_include_vector(32768);
|
|
Bitvector pid_exclude_vector(32768);
|
|
bool include_some_pids;
|
|
bool exclude_some_pids;
|
|
|
|
HashTable<int> excluded_procedures(2000);
|
|
HashTable<int> included_procedures(2000);
|
|
bool exclude_some_procedures;
|
|
bool include_some_procedures;
|
|
|
|
bool exclude_kernel_syms;
|
|
bool exclude_library_syms;
|
|
bool include_kernel_syms;
|
|
bool include_library_syms;
|
|
bool demangle = true;
|
|
|
|
static const char *OptionsUsageStr =
|
|
" -e :kernel exclude all kernel symbols\n"
|
|
" -e :libs exclude all library symbols\n"
|
|
" -e <func> exclude function <func>\n"
|
|
" -e <pid> exclude process <pid>\n"
|
|
" -i :kernel include all kernel symbols\n"
|
|
" -i :libs include all library symbols\n"
|
|
" -i <func> include function <func>\n"
|
|
" -i <pid> include process <pid>\n"
|
|
" -l :kernel lump all the kernel symbols together\n"
|
|
" -l :libs lump all the library symbols together\n"
|
|
" -m do not demangle C++ symbols (m for 'mangle')\n"
|
|
" -r <root> use <root> as the path for finding ELF executables\n"
|
|
;
|
|
|
|
void OptionsUsage()
|
|
{
|
|
fprintf(stderr, "%s", OptionsUsageStr);
|
|
}
|
|
|
|
void ParseOptions(int argc, char **argv)
|
|
{
|
|
bool err = false;
|
|
while (!err) {
|
|
int opt = getopt(argc, argv, "+e:i:l:mr:");
|
|
if (opt == -1)
|
|
break;
|
|
switch (opt) {
|
|
case 'e':
|
|
if (*optarg == ':') {
|
|
if (strcmp(optarg, ":kernel") == 0)
|
|
exclude_kernel_syms = true;
|
|
else if (strcmp(optarg, ":libs") == 0)
|
|
exclude_library_syms = true;
|
|
else
|
|
err = true;
|
|
excluded_procedures.Update(optarg, 1);
|
|
exclude_some_procedures = true;
|
|
} else if (isdigit(*optarg)) {
|
|
int bitnum = atoi(optarg);
|
|
pid_exclude_vector.SetBit(bitnum);
|
|
exclude_some_pids = true;
|
|
} else {
|
|
excluded_procedures.Update(optarg, 1);
|
|
exclude_some_procedures = true;
|
|
}
|
|
break;
|
|
case 'i':
|
|
if (*optarg == ':') {
|
|
if (strcmp(optarg, ":kernel") == 0)
|
|
include_kernel_syms = true;
|
|
else if (strcmp(optarg, ":libs") == 0)
|
|
include_library_syms = true;
|
|
else
|
|
err = true;
|
|
included_procedures.Update(optarg, 1);
|
|
include_some_procedures = true;
|
|
} else if (isdigit(*optarg)) {
|
|
int bitnum = atoi(optarg);
|
|
pid_include_vector.SetBit(bitnum);
|
|
include_some_pids = true;
|
|
} else {
|
|
included_procedures.Update(optarg, 1);
|
|
include_some_procedures = true;
|
|
}
|
|
break;
|
|
case 'l':
|
|
if (strcmp(optarg, ":kernel") == 0)
|
|
lump_kernel = true;
|
|
else if (strcmp(optarg, ":libs") == 0)
|
|
lump_libraries = true;
|
|
else
|
|
err = true;
|
|
break;
|
|
case 'm':
|
|
demangle = false;
|
|
break;
|
|
case 'r':
|
|
root = optarg;
|
|
break;
|
|
default:
|
|
err = true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (err) {
|
|
Usage(argv[0]);
|
|
exit(1);
|
|
}
|
|
}
|