123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159 |
- /*-
- * SPDX-License-Identifier: BSD-2-Clause
- *
- * Copyright (c) 2007 Robert N. M. Watson
- * All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions
- * are met:
- * 1. Redistributions of source code must retain the above copyright
- * notice, this list of conditions and the following disclaimer.
- * 2. Redistributions in binary form must reproduce the above copyright
- * notice, this list of conditions and the following disclaimer in the
- * documentation and/or other materials provided with the distribution.
- *
- * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
- * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
- * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
- * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
- * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
- * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
- * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
- * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
- * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
- * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
- * SUCH DAMAGE.
- */
- #include <sys/types.h>
- #include <sys/sysctl.h>
- #include <err.h>
- #include <errno.h>
- #include <stdio.h>
- #include <stdlib.h>
- #include <string.h>
- #include <sysexits.h>
- #include "ddb.h"
- /*
- * These commands manage DDB(4) scripts from user space. For better or worse,
- * the setting and unsetting of scripts is only poorly represented using
- * sysctl(8), and this interface provides a more user-friendly way to
- * accomplish this management, wrapped around lower-level sysctls. For
- * completeness, listing of scripts is also included.
- */
- #define SYSCTL_SCRIPT "debug.ddb.scripting.script"
- #define SYSCTL_SCRIPTS "debug.ddb.scripting.scripts"
- #define SYSCTL_UNSCRIPT "debug.ddb.scripting.unscript"
- /*
- * Print all scripts (scriptname==NULL) or a specific script.
- */
- static void
- ddb_list_scripts(const char *scriptname)
- {
- char *buffer, *line, *nextline;
- char *line_script, *line_scriptname;
- size_t buflen, len;
- int ret;
- repeat:
- if (sysctlbyname(SYSCTL_SCRIPTS, NULL, &buflen, NULL, 0) < 0)
- err(EX_OSERR, "sysctl: %s", SYSCTL_SCRIPTS);
- if (buflen == 0)
- return;
- buffer = malloc(buflen);
- if (buffer == NULL)
- err(EX_OSERR, "malloc");
- bzero(buffer, buflen);
- len = buflen;
- ret = sysctlbyname(SYSCTL_SCRIPTS, buffer, &len, NULL, 0);
- if (ret < 0 && errno != ENOMEM)
- err(EX_OSERR, "sysctl: %s", SYSCTL_SCRIPTS);
- if (ret < 0) {
- free(buffer);
- goto repeat;
- }
- /*
- * We nul'd the buffer before calling sysctl(), so at worst empty.
- *
- * If a specific script hasn't been requested, print it all.
- */
- if (scriptname == NULL) {
- printf("%s", buffer);
- free(buffer);
- return;
- }
- /*
- * If a specific script has been requested, we have to parse the
- * string to find it.
- */
- nextline = buffer;
- while ((line = strsep(&nextline, "\n")) != NULL) {
- line_script = line;
- line_scriptname = strsep(&line_script, "=");
- if (line_script == NULL)
- continue;
- if (strcmp(scriptname, line_scriptname) != 0)
- continue;
- printf("%s\n", line_script);
- break;
- }
- if (line == NULL) {
- errno = ENOENT;
- err(EX_DATAERR, "%s", scriptname);
- }
- free(buffer);
- }
- /*
- * "ddb script" can be used to either print or set a script.
- */
- void
- ddb_script(int argc, char *argv[])
- {
- if (argc != 2)
- usage();
- argv++;
- argc--;
- if (strchr(argv[0], '=') != 0) {
- if (sysctlbyname(SYSCTL_SCRIPT, NULL, NULL, argv[0],
- strlen(argv[0]) + 1) < 0)
- err(EX_OSERR, "sysctl: %s", SYSCTL_SCRIPTS);
- } else
- ddb_list_scripts(argv[0]);
- }
- void
- ddb_scripts(int argc, char *argv[] __unused)
- {
- if (argc != 1)
- usage();
- ddb_list_scripts(NULL);
- }
- void
- ddb_unscript(int argc, char *argv[])
- {
- int ret;
- if (argc != 2)
- usage();
- argv++;
- argc--;
- ret = sysctlbyname(SYSCTL_UNSCRIPT, NULL, NULL, argv[0],
- strlen(argv[0]) + 1);
- if (ret < 0 && errno == EINVAL) {
- errno = ENOENT;
- err(EX_DATAERR, "sysctl: %s", argv[0]);
- } else if (ret < 0)
- err(EX_OSERR, "sysctl: %s", SYSCTL_UNSCRIPT);
- }
|