You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

89 lines
1.5 KiB

// SPDX-License-Identifier: GPL-2.0-or-later
/*
* Copyright (c) 2016 Linux Test Project
*/
#include <errno.h>
#include <limits.h>
#include <stdio.h>
#include <stdlib.h>
#define TST_NO_DEFAULT_MAIN
#include "tst_test.h"
static void print_help(void)
{
printf("Usage: tst_checkpoint wait TIMEOUT ID\n");
printf(" or: tst_checkpoint wake TIMEOUT ID NR_WAKE\n");
printf(" TIMEOUT - timeout in ms\n");
printf(" ID - checkpoint id\n");
printf(" NR_WAKE - number of processes to wake up\n");
}
static int get_val(const char *name, const char *arg, unsigned int *val)
{
unsigned long temp;
char *e;
errno = 0;
temp = strtoul(arg, &e, 10);
if (errno || (e == arg) || (temp > UINT_MAX)) {
fprintf(stderr, "ERROR: Invalid %s '%s'\n",
name, arg);
return -1;
}
*val = temp;
return 0;
}
int main(int argc, char *argv[])
{
unsigned int id, timeout, nr_wake;
int ret;
int type;
if (argc < 3)
goto help;
if (!strcmp(argv[1], "wait")) {
type = 0;
if (argc != 4)
goto help;
} else if (!strcmp(argv[1], "wake")) {
type = 1;
if (argc != 5)
goto help;
if (get_val("NR_WAKE", argv[4], &nr_wake))
goto help;
} else {
fprintf(stderr, "ERROR: Invalid COMMAND '%s'\n",
argv[1]);
goto help;
}
if (get_val("TIMEOUT", argv[2], &timeout)
|| get_val("ID", argv[3], &id)) {
goto help;
}
tst_reinit();
if (type)
ret = tst_checkpoint_wake(id, nr_wake, timeout);
else
ret = tst_checkpoint_wait(id, timeout);
if (ret)
return 1;
else
return 0;
help:
print_help();
return 1;
}