Commit 749fb263 authored by Eugeniu Rosca's avatar Eugeniu Rosca Committed by Shuah Khan

selftests: watchdog: use getopt_long()

Switch from manual argv[] parsing to getopt_long() argument processing.
This creates more readable code and allows easier feature addition.
This also fixes some segmentation faults introduced by
commit 1dbdcc81 ("selftests: watchdog: accept multiple params on
command line"), when options -t or -p are not given the required value:

./watchdog-test -p 1 -t
./watchdog-test -t 1 -p

No changes are intended in the way watchdog-test interacts with the
kernel. The only noticible changes, tightly related to the addition
of getopt (and done for easier maintenance),  are:
- help message has been reworked and migrated to a dedicated function.
- all short/long options and the help message are sorted alphabetically.
- all case statements inside the getopt loop are sorted alphabetically.

Fixes: 1dbdcc81 ("selftests: watchdog: accept multiple params on command line")
Signed-off-by: default avatarEugeniu Rosca <erosca@de.adit-jv.com>
Signed-off-by: default avatarShuah Khan <shuahkh@osg.samsung.com>
parent 0c528da8
...@@ -9,12 +9,22 @@ ...@@ -9,12 +9,22 @@
#include <unistd.h> #include <unistd.h>
#include <fcntl.h> #include <fcntl.h>
#include <signal.h> #include <signal.h>
#include <getopt.h>
#include <sys/ioctl.h> #include <sys/ioctl.h>
#include <linux/types.h> #include <linux/types.h>
#include <linux/watchdog.h> #include <linux/watchdog.h>
int fd; int fd;
const char v = 'V'; const char v = 'V';
static const char sopts[] = "dehp:t:";
static const struct option lopts[] = {
{"disable", no_argument, NULL, 'd'},
{"enable", no_argument, NULL, 'e'},
{"help", no_argument, NULL, 'h'},
{"pingrate", required_argument, NULL, 'p'},
{"timeout", required_argument, NULL, 't'},
{NULL, no_argument, NULL, 0x0}
};
/* /*
* This function simply sends an IOCTL to the driver, which in turn ticks * This function simply sends an IOCTL to the driver, which in turn ticks
...@@ -48,12 +58,25 @@ static void term(int sig) ...@@ -48,12 +58,25 @@ static void term(int sig)
exit(0); exit(0);
} }
static void usage(char *progname)
{
printf("Usage: %s [options]\n", progname);
printf(" -d, --disable Turn off the watchdog timer\n");
printf(" -e, --enable Turn on the watchdog timer\n");
printf(" -h, --help Print the help message\n");
printf(" -p, --pingrate=P Set ping rate to P seconds\n");
printf(" -t, --timeout=T Set timeout to T seconds\n");
printf("\n");
printf("Parameters are parsed left-to-right in real-time.\n");
printf("Example: %s -d -t 10 -p 5 -e\n", progname);
}
int main(int argc, char *argv[]) int main(int argc, char *argv[])
{ {
int flags; int flags;
unsigned int ping_rate = 1; unsigned int ping_rate = 1;
int ret; int ret;
int i; int c;
setbuf(stdout, NULL); setbuf(stdout, NULL);
...@@ -64,33 +87,32 @@ int main(int argc, char *argv[]) ...@@ -64,33 +87,32 @@ int main(int argc, char *argv[])
exit(-1); exit(-1);
} }
for (i = 1; i < argc; i++) { while ((c = getopt_long(argc, argv, sopts, lopts, NULL)) != -1) {
if (!strncasecmp(argv[i], "-d", 2)) { switch (c) {
case 'd':
flags = WDIOS_DISABLECARD; flags = WDIOS_DISABLECARD;
ret = ioctl(fd, WDIOC_SETOPTIONS, &flags); ret = ioctl(fd, WDIOC_SETOPTIONS, &flags);
if (!ret) if (!ret)
printf("Watchdog card disabled.\n"); printf("Watchdog card disabled.\n");
} else if (!strncasecmp(argv[i], "-e", 2)) { break;
case 'e':
flags = WDIOS_ENABLECARD; flags = WDIOS_ENABLECARD;
ret = ioctl(fd, WDIOC_SETOPTIONS, &flags); ret = ioctl(fd, WDIOC_SETOPTIONS, &flags);
if (!ret) if (!ret)
printf("Watchdog card enabled.\n"); printf("Watchdog card enabled.\n");
} else if (!strncasecmp(argv[i], "-t", 2) && argv[2]) { break;
flags = atoi(argv[i + 1]); case 'p':
ping_rate = strtoul(optarg, NULL, 0);
printf("Watchdog ping rate set to %u seconds.\n", ping_rate);
break;
case 't':
flags = atoi(optarg);
ret = ioctl(fd, WDIOC_SETTIMEOUT, &flags); ret = ioctl(fd, WDIOC_SETTIMEOUT, &flags);
if (!ret) if (!ret)
printf("Watchdog timeout set to %u seconds.\n", flags); printf("Watchdog timeout set to %u seconds.\n", flags);
i++; break;
} else if (!strncasecmp(argv[i], "-p", 2) && argv[2]) { default:
ping_rate = strtoul(argv[i + 1], NULL, 0); usage(argv[0]);
printf("Watchdog ping rate set to %u seconds.\n", ping_rate);
i++;
} else {
printf("-d to disable, -e to enable, -t <n> to set "
"the timeout,\n-p <n> to set the ping rate, and ");
printf("run by itself to tick the card.\n");
printf("Parameters are parsed left-to-right in real-time.\n");
printf("Example: %s -d -t 10 -p 5 -e\n", argv[0]);
goto end; goto end;
} }
} }
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment