91 lines
1.6 KiB
C
91 lines
1.6 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
/*
|
|
* Copyright (C) 2012-2014 Cyril Hrubis chrubis@suse.cz
|
|
* Copyright (c) 2021 Xie Ziyao <xieziyao@huawei.com>
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
|
|
#include "test.h"
|
|
#include "tst_process_state.h"
|
|
|
|
int tst_process_state_wait(const char *file, const int lineno,
|
|
void (*cleanup_fn)(void), pid_t pid,
|
|
const char state, unsigned int msec_timeout)
|
|
{
|
|
char proc_path[128], cur_state;
|
|
unsigned int msecs = 0;
|
|
|
|
snprintf(proc_path, sizeof(proc_path), "/proc/%i/stat", pid);
|
|
|
|
for (;;) {
|
|
safe_file_scanf(file, lineno, cleanup_fn, proc_path,
|
|
"%*i %*s %c", &cur_state);
|
|
|
|
if (state == cur_state)
|
|
break;
|
|
|
|
usleep(1000);
|
|
msecs += 1;
|
|
|
|
if (msec_timeout && msecs >= msec_timeout) {
|
|
errno = ETIMEDOUT;
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int tst_process_state_wait2(pid_t pid, const char state)
|
|
{
|
|
char proc_path[128], cur_state;
|
|
|
|
snprintf(proc_path, sizeof(proc_path), "/proc/%i/stat", pid);
|
|
|
|
for (;;) {
|
|
FILE *f = fopen(proc_path, "r");
|
|
|
|
if (!f) {
|
|
fprintf(stderr, "Failed to open '%s': %s\n",
|
|
proc_path, strerror(errno));
|
|
return 1;
|
|
}
|
|
|
|
if (fscanf(f, "%*i %*s %c", &cur_state) != 1) {
|
|
fclose(f);
|
|
fprintf(stderr, "Failed to read '%s': %s\n",
|
|
proc_path, strerror(errno));
|
|
return 1;
|
|
}
|
|
fclose(f);
|
|
|
|
if (state == cur_state)
|
|
return 0;
|
|
|
|
usleep(10000);
|
|
}
|
|
}
|
|
|
|
int tst_process_exit_wait(pid_t pid, unsigned int msec_timeout)
|
|
{
|
|
unsigned int msecs = 0;
|
|
|
|
for (;;) {
|
|
if (kill(pid, 0) && errno == ESRCH)
|
|
break;
|
|
|
|
usleep(1000);
|
|
msecs += 1;
|
|
|
|
if (msec_timeout && msecs >= msec_timeout) {
|
|
errno = ETIMEDOUT;
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
return 1;
|
|
}
|