1 /*
2 * Copyright (C) 2012-2014 Cyril Hrubis chrubis@suse.cz
3 *
4 * This program is free software; you can redistribute it and/or modify it
5 * under the terms of version 2 of the GNU General Public License as
6 * published by the Free Software Foundation.
7 *
8 * This program is distributed in the hope that it would be useful, but
9 * WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
11 *
12 * Further, this software is distributed without any warranty that it is
13 * free of the rightful claim of any third person regarding infringement
14 * or the like. Any license provided herein, whether implied or
15 * otherwise, applies only to this software file. Patent licenses, if
16 * any, provided herein do not apply to combinations of this program with
17 * other software, or any other product whatsoever.
18 *
19 * You should have received a copy of the GNU General Public License along
20 * with this program; if not, write the Free Software Foundation, Inc.,
21 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
22 */
23
24 #include <stdio.h>
25 #include <string.h>
26 #include <errno.h>
27
28 #include "test.h"
29 #include "tst_process_state.h"
30
tst_process_state_wait(const char * file,const int lineno,void (* cleanup_fn)(void),pid_t pid,const char state,unsigned int msec_timeout)31 int tst_process_state_wait(const char *file, const int lineno,
32 void (*cleanup_fn)(void), pid_t pid,
33 const char state, unsigned int msec_timeout)
34 {
35 char proc_path[128], cur_state;
36 unsigned int msecs = 0;
37
38 snprintf(proc_path, sizeof(proc_path), "/proc/%i/stat", pid);
39
40 for (;;) {
41 safe_file_scanf(file, lineno, cleanup_fn, proc_path,
42 "%*i %*s %c", &cur_state);
43
44 if (state == cur_state)
45 break;
46
47 usleep(1000);
48 msecs += 1;
49
50 if (msec_timeout && msecs >= msec_timeout) {
51 errno = ETIMEDOUT;
52 return -1;
53 }
54 }
55
56 return 0;
57 }
58
tst_process_state_wait2(pid_t pid,const char state)59 int tst_process_state_wait2(pid_t pid, const char state)
60 {
61 char proc_path[128], cur_state;
62
63 snprintf(proc_path, sizeof(proc_path), "/proc/%i/stat", pid);
64
65 for (;;) {
66 FILE *f = fopen(proc_path, "r");
67 if (!f) {
68 fprintf(stderr, "Failed to open '%s': %s\n",
69 proc_path, strerror(errno));
70 return 1;
71 }
72
73 if (fscanf(f, "%*i %*s %c", &cur_state) != 1) {
74 fclose(f);
75 fprintf(stderr, "Failed to read '%s': %s\n",
76 proc_path, strerror(errno));
77 return 1;
78 }
79 fclose(f);
80
81 if (state == cur_state)
82 return 0;
83
84 usleep(10000);
85 }
86 }
87