X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=lib%2Fcommand-line.c;h=cb73a25c90c46500a60a0ab79a7fda7baffc6b9b;hb=HEAD;hp=7800c0b6ee6fe8da2e21682cee2cafad2fe29ad7;hpb=85606e05b691be7c2f2d4bcf0e91170b71ec8fbb;p=sliver-openvswitch.git diff --git a/lib/command-line.c b/lib/command-line.c index 7800c0b6e..cb73a25c9 100644 --- a/lib/command-line.c +++ b/lib/command-line.c @@ -1,5 +1,5 @@ /* - * Copyright (c) 2008, 2009, 2010, 2011, 2013 Nicira, Inc. + * Copyright (c) 2008, 2009, 2010, 2011, 2013, 2014 Nicira, Inc. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. @@ -93,10 +93,17 @@ run_command(int argc, char *argv[], const struct command commands[]) /* Process title. */ -#ifdef LINUX_DATAPATH -static char *argv_start; /* Start of command-line arguments in memory. */ -static size_t argv_size; /* Number of bytes of command-line arguments. */ -static char *saved_proctitle; /* Saved command-line arguments. */ +#ifdef __linux__ +static struct ovs_mutex proctitle_mutex = OVS_MUTEX_INITIALIZER; + +/* Start of command-line arguments in memory. */ +static char *argv_start OVS_GUARDED_BY(proctitle_mutex); + +/* Number of bytes of command-line arguments. */ +static size_t argv_size OVS_GUARDED_BY(proctitle_mutex); + +/* Saved command-line arguments. */ +static char *saved_proctitle OVS_GUARDED_BY(proctitle_mutex); /* Prepares the process so that proctitle_set() can later succeed. * @@ -117,6 +124,7 @@ proctitle_init(int argc, char **argv) return; } + ovs_mutex_lock(&proctitle_mutex); /* Specialized version of first loop iteration below. */ argv_start = argv[0]; argv_size = strlen(argv[0]) + 1; @@ -140,6 +148,7 @@ proctitle_init(int argc, char **argv) /* Copy out the old argument so we can reuse the space. */ argv[i] = xstrdup(argv[i]); } + ovs_mutex_unlock(&proctitle_mutex); } /* Changes the name of the process, as shown by "ps", to the program name @@ -150,8 +159,9 @@ proctitle_set(const char *format, ...) va_list args; int n; + ovs_mutex_lock(&proctitle_mutex); if (!argv_start || argv_size < 8) { - return; + goto out; } if (!saved_proctitle) { @@ -172,19 +182,24 @@ proctitle_set(const char *format, ...) memset(&argv_start[n], '\0', argv_size - n); } va_end(args); + +out: + ovs_mutex_unlock(&proctitle_mutex); } /* Restores the process's original command line, as seen by "ps". */ void proctitle_restore(void) { + ovs_mutex_lock(&proctitle_mutex); if (saved_proctitle) { memcpy(argv_start, saved_proctitle, argv_size); free(saved_proctitle); saved_proctitle = NULL; } + ovs_mutex_unlock(&proctitle_mutex); } -#else /* !LINUX_DATAPATH*/ +#else /* !__linux__ */ /* Stubs that don't do anything on non-Linux systems. */ void @@ -204,4 +219,4 @@ void proctitle_restore(void) { } -#endif /* !LINUX_DATAPATH */ +#endif /* !__linux__ */