diff options
Diffstat (limited to 'runtime/staprun/stapio.c')
-rw-r--r-- | runtime/staprun/stapio.c | 69 |
1 files changed, 69 insertions, 0 deletions
diff --git a/runtime/staprun/stapio.c b/runtime/staprun/stapio.c new file mode 100644 index 00000000..696167af --- /dev/null +++ b/runtime/staprun/stapio.c @@ -0,0 +1,69 @@ +/* -*- linux-c -*- + * + * stapio.c - SystemTap module io handler. + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. + * + * Copyright (C) 2005-2007 Red Hat, Inc. + * + */ + +#include "staprun.h" +#include <pwd.h> +char *__name__ = "stapio"; + +int main(int argc, char **argv) +{ + setup_signals(); + + parse_args(argc, argv); + + if (buffer_size) + dbug(1, "Using a buffer of %u bytes.\n", buffer_size); + + if (optind < argc) { + parse_modpath(argv[optind++]); + dbug(2, "modpath=\"%s\", modname=\"%s\"\n", modpath, modname); + } + + if (optind < argc) { + if (attach_mod) { + err("ERROR: Cannot have module options with attach (-A).\n"); + usage(argv[0]); + } else { + unsigned start_idx = 3; /* reserve three slots in modoptions[] */ + while (optind < argc && start_idx+1 < MAXMODOPTIONS) + modoptions[start_idx++] = argv[optind++]; + modoptions[start_idx] = NULL; + } + } + + if (modpath == NULL || *modpath == '\0') { + err("ERROR: Need a module name or path to load.\n"); + usage(argv[0]); + } + + if (init_stapio()) + exit(1); + + initialized = 1; + + if (stp_main_loop()) { + err("ERROR: Couldn't enter main loop. Exiting.\n"); + exit(1); + } + + return 0; +} |