summaryrefslogtreecommitdiffstats
path: root/tests/common/unlinkat_failure.c
blob: 5b8bf9589123294bd91ee9f8914db842cba64578 (plain)
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
/* 
 * gcc unlinkat_failure.c -o unlinkat_failure.so -shared -ldl 
 * LD_PRELOAD=./unlinkat_failure.so FAILURE_PATH=/etc/shadow ./test /etc/shadow
 */

#define _GNU_SOURCE
#include <dlfcn.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <stdarg.h>
#include <fcntl.h>
#include <string.h>
#include <errno.h>
#include <assert.h>

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>


typedef int (*unlinkat_type) (int dirfd, const char *pathname, int flags);
static unlinkat_type next_unlinkat;

static const char *failure_path = NULL;
static dev_t failure_dev = -1;
static ino_t failure_ino = -1;

int unlinkat (int dirfd, const char *pathname, int flags)
{
	if (NULL == next_unlinkat)
	{
		next_unlinkat = dlsym (RTLD_NEXT, "unlinkat");
		assert (NULL != next_unlinkat);
	}
	if (NULL == failure_path) {
		struct stat sb;
		failure_path = getenv ("FAILURE_PATH");
		if (NULL == failure_path) {
			fputs ("No FAILURE_PATH defined\n", stderr);
		}
		if (lstat (failure_path, &sb) != 0) {
			fputs ("Can't lstat FAILURE_PATH\n", stderr);
		}
		failure_dev = sb.st_dev;
		failure_ino = sb.st_ino;
	}

	if (   (NULL != pathname)
	    && (NULL != failure_path)) {
		struct stat sb;
		if (   (fstatat (dirfd, pathname, &sb, flags) == 0)
		    && (sb.st_dev == failure_dev)
		    && (sb.st_ino == failure_ino)) {
			fprintf (stderr, "unlinkat FAILURE %s\n", failure_path);
			errno = EBUSY;
			return -1;
		}
	}

	return next_unlinkat (dirfd, pathname, flags);
}