File: delay.c

package info (click to toggle)
strace 6.13%2Bds-1
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 71,488 kB
  • sloc: ansic: 176,497; sh: 9,675; makefile: 4,133; cpp: 885; awk: 353; perl: 267; exp: 62; sed: 9
file content (126 lines) | stat: -rw-r--r-- 2,896 bytes parent folder | download | duplicates (6)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
/*
 * Check delay injection.
 *
 * Copyright (c) 2018-2023 The strace developers.
 * All rights reserved.
 *
 * SPDX-License-Identifier: GPL-2.0-or-later
 */

#include "tests.h"
#include <inttypes.h>
#include <limits.h>
#include <stdio.h>
#include <stdint.h>
#include <stdlib.h>
#include <unistd.h>
#include <time.h>
#include <sys/time.h>
#include <sys/wait.h>
#include "scno.h"
#include "kernel_timeval.h"

static int64_t
usecs_from_tv(const kernel_old_timeval_t *const tv)
{
    return (int64_t) tv->tv_sec * 1000000 + tv->tv_usec;
}

static int64_t
usecs_from_ts(const struct timespec *const ts)
{
    return (int64_t) ts->tv_sec * 1000000 + ts->tv_nsec / 1000;
}

static void
check_(const int64_t got, const bool ge, const int64_t orig, const int nproc,
       const int exitcode)
{
	const int64_t thresh = (orig * (ge ? nproc - 1 : nproc + 1)) / nproc;

	if (ge ? got >= thresh : got <= thresh)
		return;

	fprintf(stderr, "Got delay of %" PRId64 ", %s than threshold value of "
			"%" PRId64 " (expected nominal delay value is %" PRId64
			")\n", got, ge ? "less" : "more", thresh, orig);

	_exit(exitcode);
}

static void
check_delay(const kernel_old_timeval_t *const tv0,
	    const struct timespec *const ts,
	    const kernel_old_timeval_t *const tv1,
	    const int nproc,
	    const int64_t delay_enter,
	    const int64_t delay_exit)
{
	const int64_t us0 = usecs_from_tv(tv0);
	const int64_t us  = usecs_from_ts(ts);
	const int64_t us1 = usecs_from_tv(tv1);

	check_(us - us0, true,  delay_exit,  nproc, 1);
	check_(us - us0, false, delay_exit,  nproc, 2);
	check_(us1 - us, true,  delay_enter, nproc, 3);
	check_(us1 - us, false, delay_enter, nproc, 4);
}

static void
run(const int nproc, const int delay_enter, const int delay_exit)
{
	kernel_old_timeval_t prev = { 0, 0 }, now;

	for (int i = 0; i < nproc; ++i, prev = now) {
		struct timespec ts;

		if (i && clock_gettime(CLOCK_REALTIME, &ts))
			perror_msg_and_fail("clock_gettime");

		if (syscall(__NR_gettimeofday, &now, NULL))
			perror_msg_and_fail("gettimeofday");

		if (!i)
			continue;

		check_delay(&prev, &ts, &now, nproc, delay_enter, delay_exit);
	}

	_exit(0);
}

int
main(int ac, char *av[])
{
	if (ac != 4)
		error_msg_and_fail("usage: delay <nproc> <delay_enter> <delay_exit>");

	const int nproc = atoi(av[1]);
	if (nproc <= 1)
		perror_msg_and_fail("invalid nproc: %s", av[1]);

	const int delay_enter = atoi(av[2]);
	if (delay_enter <= 0)
		perror_msg_and_fail("invalid delay_enter: %s", av[2]);

	const int delay_exit = atoi(av[3]);
	if (delay_exit <= 0)
		perror_msg_and_fail("invalid delay_exit: %s", av[3]);

	for (int i = 0; i < nproc; ++i) {
		pid_t pid = fork();

		if (pid)
			usleep(MAX(delay_enter, delay_exit) / nproc);
		else
			run(nproc, delay_enter, delay_exit);
	}

	int status;
	while (wait(&status) > 0) {
		if (status)
			perror_msg_and_fail("wait status %d", status);
	}

	return 0;
}