diff options
Diffstat (limited to '')
-rw-r--r-- | lib/util/util_process.c | 101 |
1 files changed, 101 insertions, 0 deletions
diff --git a/lib/util/util_process.c b/lib/util/util_process.c new file mode 100644 index 0000000..a10b85c --- /dev/null +++ b/lib/util/util_process.c @@ -0,0 +1,101 @@ +/* + * Unix SMB/CIFS implementation. + * + * Process utils. + * + * Copyright (c) 2013 Andreas Schneider <asn@samba.org> + * + * 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 3 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, see <http://www.gnu.org/licenses/>. + */ + +#include "util_process.h" +#include "replace.h" + +#ifdef HAVE_SYS_PRCTL_H +#include <sys/prctl.h> +#endif + +/* + * These variables are static so that we can access them + * with process_get_short_title() and process_get_long_title(). The + * purpose of this is to allow smb_panic_log() to print them. + */ +static char short_comment[16] = {0,}; +static char long_comment[256] = {0,}; +static char binary_name[256]; + +void process_set_title(const char *short_format, const char *long_format, ...) +{ +#if defined(HAVE_PRCTL) && defined(PR_SET_NAME) + if (short_format != NULL) { + va_list ap; + + va_start(ap, long_format); + vsnprintf(short_comment, sizeof(short_comment), short_format, ap); + va_end(ap); + + prctl(PR_SET_NAME, (unsigned long) short_comment, 0, 0, 0); + } +#endif + + if (long_format != NULL) { + va_list ap; + + va_start(ap, long_format); + vsnprintf(long_comment, sizeof(long_comment), long_format, ap); + va_end(ap); + + setproctitle("%s", long_comment); + } +} + +const char *process_get_short_title(void) +{ + return short_comment; +} + +const char *process_get_long_title(void) +{ + return long_comment; +} + +/* + * This is just for debugging in a panic, so we don't want to do + * anything more than return a fixed pointer, so we save a copy to a + * static variable. + */ +void process_save_binary_name(const char *progname) +{ + strlcpy(binary_name, progname, sizeof(binary_name)); +} + +/* Samba binaries will set this during popt handling */ +const char *process_get_saved_binary_name(void) +{ + return binary_name; +} + + +int prctl_set_comment(const char *comment_format, ...) +{ + char comment[16]; + va_list ap; + + va_start(ap, comment_format); + vsnprintf(comment, sizeof(comment), comment_format, ap); + va_end(ap); + + process_set_title("%s", "%s", comment); + return 0; +} |