dnotify.c 10.9 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5
/*
 * Directory notifications for Linux.
 *
 * Copyright (C) 2000,2001,2002 Stephen Rothwell
 *
6 7 8
 * Copyright (C) 2009 Eric Paris <Red Hat Inc>
 * dnotify was largly rewritten to use the new fsnotify infrastructure
 *
L
Linus Torvalds 已提交
9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the
 * Free Software Foundation; either version 2, or (at your option) any
 * later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 */
#include <linux/fs.h>
#include <linux/module.h>
#include <linux/sched.h>
#include <linux/dnotify.h>
#include <linux/init.h>
#include <linux/spinlock.h>
#include <linux/slab.h>
A
Al Viro 已提交
26
#include <linux/fdtable.h>
27
#include <linux/fsnotify_backend.h>
L
Linus Torvalds 已提交
28

29
int dir_notify_enable __read_mostly = 1;
L
Linus Torvalds 已提交
30

31
static struct kmem_cache *dnotify_struct_cache __read_mostly;
E
Eric Paris 已提交
32
static struct kmem_cache *dnotify_mark_cache __read_mostly;
33 34 35
static struct fsnotify_group *dnotify_group __read_mostly;

/*
36
 * dnotify will attach one of these to each inode (i_fsnotify_marks) which
37 38 39
 * is being watched by dnotify.  If multiple userspace applications are watching
 * the same directory with dnotify their information is chained in dn
 */
E
Eric Paris 已提交
40 41
struct dnotify_mark {
	struct fsnotify_mark fsn_mark;
42 43
	struct dnotify_struct *dn;
};
L
Linus Torvalds 已提交
44

45 46 47 48 49 50 51 52
/*
 * When a process starts or stops watching an inode the set of events which
 * dnotify cares about for that inode may change.  This function runs the
 * list of everything receiving dnotify events about this directory and calculates
 * the set of all those events.  After it updates what dnotify is interested in
 * it calls the fsnotify function so it can update the set of all events relevant
 * to this inode.
 */
E
Eric Paris 已提交
53
static void dnotify_recalc_inode_mask(struct fsnotify_mark *fsn_mark)
L
Linus Torvalds 已提交
54
{
55
	__u32 new_mask, old_mask;
L
Linus Torvalds 已提交
56
	struct dnotify_struct *dn;
E
Eric Paris 已提交
57 58 59
	struct dnotify_mark *dn_mark  = container_of(fsn_mark,
						     struct dnotify_mark,
						     fsn_mark);
60

E
Eric Paris 已提交
61
	assert_spin_locked(&fsn_mark->lock);
L
Linus Torvalds 已提交
62

E
Eric Paris 已提交
63
	old_mask = fsn_mark->mask;
L
Linus Torvalds 已提交
64
	new_mask = 0;
E
Eric Paris 已提交
65
	for (dn = dn_mark->dn; dn != NULL; dn = dn->dn_next)
66
		new_mask |= (dn->dn_mask & ~FS_DN_MULTISHOT);
67
	fsnotify_set_mark_mask_locked(fsn_mark, new_mask);
68 69 70 71

	if (old_mask == new_mask)
		return;

72
	fsnotify_recalc_mask(fsn_mark->connector);
L
Linus Torvalds 已提交
73 74
}

75 76 77 78 79 80 81 82 83
/*
 * Mains fsnotify call where events are delivered to dnotify.
 * Find the dnotify mark on the relevant inode, run the list of dnotify structs
 * on that mark and determine which of them has expressed interest in receiving
 * events of this type.  When found send the correct process and signal and
 * destroy the dnotify struct if it was not registered to receive multiple
 * events.
 */
static int dnotify_handle_event(struct fsnotify_group *group,
84
				struct inode *inode,
85 86
				struct fsnotify_mark *inode_mark,
				struct fsnotify_mark *vfsmount_mark,
87
				u32 mask, const void *data, int data_type,
88 89
				const unsigned char *file_name, u32 cookie,
				struct fsnotify_iter_info *iter_info)
90
{
E
Eric Paris 已提交
91
	struct dnotify_mark *dn_mark;
92 93 94
	struct dnotify_struct *dn;
	struct dnotify_struct **prev;
	struct fown_struct *fown;
95
	__u32 test_mask = mask & ~FS_EVENT_ON_CHILD;
96

97 98 99 100
	/* not a dir, dnotify doesn't care */
	if (!S_ISDIR(inode->i_mode))
		return 0;

101 102 103
	BUG_ON(vfsmount_mark);

	dn_mark = container_of(inode_mark, struct dnotify_mark, fsn_mark);
104

105
	spin_lock(&inode_mark->lock);
E
Eric Paris 已提交
106
	prev = &dn_mark->dn;
107
	while ((dn = *prev) != NULL) {
108
		if ((dn->dn_mask & test_mask) == 0) {
109 110 111 112 113 114 115 116 117 118
			prev = &dn->dn_next;
			continue;
		}
		fown = &dn->dn_filp->f_owner;
		send_sigio(fown, dn->dn_fd, POLL_MSG);
		if (dn->dn_mask & FS_DN_MULTISHOT)
			prev = &dn->dn_next;
		else {
			*prev = dn->dn_next;
			kmem_cache_free(dnotify_struct_cache, dn);
119
			dnotify_recalc_inode_mask(inode_mark);
120 121 122
		}
	}

123
	spin_unlock(&inode_mark->lock);
124 125 126 127

	return 0;
}

E
Eric Paris 已提交
128
static void dnotify_free_mark(struct fsnotify_mark *fsn_mark)
129
{
E
Eric Paris 已提交
130 131 132
	struct dnotify_mark *dn_mark = container_of(fsn_mark,
						    struct dnotify_mark,
						    fsn_mark);
133

E
Eric Paris 已提交
134
	BUG_ON(dn_mark->dn);
135

E
Eric Paris 已提交
136
	kmem_cache_free(dnotify_mark_cache, dn_mark);
137 138 139 140 141 142 143 144
}

static struct fsnotify_ops dnotify_fsnotify_ops = {
	.handle_event = dnotify_handle_event,
};

/*
 * Called every time a file is closed.  Looks first for a dnotify mark on the
145
 * inode.  If one is found run all of the ->dn structures attached to that
146 147
 * mark for one relevant to this process closing the file and remove that
 * dnotify_struct.  If that was the last dnotify_struct also remove the
148
 * fsnotify_mark.
149
 */
L
Linus Torvalds 已提交
150 151
void dnotify_flush(struct file *filp, fl_owner_t id)
{
E
Eric Paris 已提交
152 153
	struct fsnotify_mark *fsn_mark;
	struct dnotify_mark *dn_mark;
L
Linus Torvalds 已提交
154 155 156
	struct dnotify_struct *dn;
	struct dnotify_struct **prev;
	struct inode *inode;
157
	bool free = false;
L
Linus Torvalds 已提交
158

A
Al Viro 已提交
159
	inode = file_inode(filp);
L
Linus Torvalds 已提交
160 161
	if (!S_ISDIR(inode->i_mode))
		return;
162

163
	fsn_mark = fsnotify_find_inode_mark(dnotify_group, inode);
E
Eric Paris 已提交
164
	if (!fsn_mark)
165
		return;
E
Eric Paris 已提交
166
	dn_mark = container_of(fsn_mark, struct dnotify_mark, fsn_mark);
167

168
	mutex_lock(&dnotify_group->mark_mutex);
169

E
Eric Paris 已提交
170 171
	spin_lock(&fsn_mark->lock);
	prev = &dn_mark->dn;
L
Linus Torvalds 已提交
172 173 174
	while ((dn = *prev) != NULL) {
		if ((dn->dn_owner == id) && (dn->dn_filp == filp)) {
			*prev = dn->dn_next;
175
			kmem_cache_free(dnotify_struct_cache, dn);
E
Eric Paris 已提交
176
			dnotify_recalc_inode_mask(fsn_mark);
L
Linus Torvalds 已提交
177 178 179 180
			break;
		}
		prev = &dn->dn_next;
	}
181

E
Eric Paris 已提交
182
	spin_unlock(&fsn_mark->lock);
183

184 185
	/* nothing else could have found us thanks to the dnotify_groups
	   mark_mutex */
186 187 188 189
	if (dn_mark->dn == NULL) {
		fsnotify_detach_mark(fsn_mark);
		free = true;
	}
190

191
	mutex_unlock(&dnotify_group->mark_mutex);
192

193 194
	if (free)
		fsnotify_free_mark(fsn_mark);
E
Eric Paris 已提交
195
	fsnotify_put_mark(fsn_mark);
196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218
}

/* this conversion is done only at watch creation */
static __u32 convert_arg(unsigned long arg)
{
	__u32 new_mask = FS_EVENT_ON_CHILD;

	if (arg & DN_MULTISHOT)
		new_mask |= FS_DN_MULTISHOT;
	if (arg & DN_DELETE)
		new_mask |= (FS_DELETE | FS_MOVED_FROM);
	if (arg & DN_MODIFY)
		new_mask |= FS_MODIFY;
	if (arg & DN_ACCESS)
		new_mask |= FS_ACCESS;
	if (arg & DN_ATTRIB)
		new_mask |= FS_ATTRIB;
	if (arg & DN_RENAME)
		new_mask |= FS_DN_RENAME;
	if (arg & DN_CREATE)
		new_mask |= (FS_CREATE | FS_MOVED_TO);

	return new_mask;
L
Linus Torvalds 已提交
219 220
}

221 222
/*
 * If multiple processes watch the same inode with dnotify there is only one
223
 * dnotify mark in inode->i_fsnotify_marks but we chain a dnotify_struct
224 225 226
 * onto that mark.  This function either attaches the new dnotify_struct onto
 * that list, or it |= the mask onto an existing dnofiy_struct.
 */
E
Eric Paris 已提交
227
static int attach_dn(struct dnotify_struct *dn, struct dnotify_mark *dn_mark,
228 229 230 231
		     fl_owner_t id, int fd, struct file *filp, __u32 mask)
{
	struct dnotify_struct *odn;

E
Eric Paris 已提交
232
	odn = dn_mark->dn;
233 234 235 236 237 238 239 240 241 242 243 244 245 246
	while (odn != NULL) {
		/* adding more events to existing dnofiy_struct? */
		if ((odn->dn_owner == id) && (odn->dn_filp == filp)) {
			odn->dn_fd = fd;
			odn->dn_mask |= mask;
			return -EEXIST;
		}
		odn = odn->dn_next;
	}

	dn->dn_mask = mask;
	dn->dn_fd = fd;
	dn->dn_filp = filp;
	dn->dn_owner = id;
E
Eric Paris 已提交
247 248
	dn->dn_next = dn_mark->dn;
	dn_mark->dn = dn;
249 250 251 252 253 254 255 256 257

	return 0;
}

/*
 * When a process calls fcntl to attach a dnotify watch to a directory it ends
 * up here.  Allocate both a mark for fsnotify to add and a dnotify_struct to be
 * attached to the fsnotify_mark.
 */
L
Linus Torvalds 已提交
258 259
int fcntl_dirnotify(int fd, struct file *filp, unsigned long arg)
{
E
Eric Paris 已提交
260 261
	struct dnotify_mark *new_dn_mark, *dn_mark;
	struct fsnotify_mark *new_fsn_mark, *fsn_mark;
L
Linus Torvalds 已提交
262 263 264
	struct dnotify_struct *dn;
	struct inode *inode;
	fl_owner_t id = current->files;
A
Al Viro 已提交
265
	struct file *f;
266 267 268 269
	int destroy = 0, error = 0;
	__u32 mask;

	/* we use these to tell if we need to kfree */
E
Eric Paris 已提交
270
	new_fsn_mark = NULL;
271
	dn = NULL;
L
Linus Torvalds 已提交
272

273 274 275 276 277 278
	if (!dir_notify_enable) {
		error = -EINVAL;
		goto out_err;
	}

	/* a 0 mask means we are explicitly removing the watch */
L
Linus Torvalds 已提交
279 280
	if ((arg & ~DN_MULTISHOT) == 0) {
		dnotify_flush(filp, id);
281 282
		error = 0;
		goto out_err;
L
Linus Torvalds 已提交
283
	}
284 285

	/* dnotify only works on directories */
A
Al Viro 已提交
286
	inode = file_inode(filp);
287 288 289
	if (!S_ISDIR(inode->i_mode)) {
		error = -ENOTDIR;
		goto out_err;
L
Linus Torvalds 已提交
290 291
	}

292 293 294 295 296 297
	/* expect most fcntl to add new rather than augment old */
	dn = kmem_cache_alloc(dnotify_struct_cache, GFP_KERNEL);
	if (!dn) {
		error = -ENOMEM;
		goto out_err;
	}
A
Al Viro 已提交
298

299
	/* new fsnotify mark, we expect most fcntl calls to add a new mark */
E
Eric Paris 已提交
300 301
	new_dn_mark = kmem_cache_alloc(dnotify_mark_cache, GFP_KERNEL);
	if (!new_dn_mark) {
302 303 304
		error = -ENOMEM;
		goto out_err;
	}
L
Linus Torvalds 已提交
305

306 307
	/* convert the userspace DN_* "arg" to the internal FS_* defines in fsnotify */
	mask = convert_arg(arg);
L
Linus Torvalds 已提交
308

E
Eric Paris 已提交
309 310 311 312 313
	/* set up the new_fsn_mark and new_dn_mark */
	new_fsn_mark = &new_dn_mark->fsn_mark;
	fsnotify_init_mark(new_fsn_mark, dnotify_free_mark);
	new_fsn_mark->mask = mask;
	new_dn_mark->dn = NULL;
L
Linus Torvalds 已提交
314

315
	/* this is needed to prevent the fcntl/close race described below */
316
	mutex_lock(&dnotify_group->mark_mutex);
L
Linus Torvalds 已提交
317

E
Eric Paris 已提交
318
	/* add the new_fsn_mark or find an old one. */
319
	fsn_mark = fsnotify_find_inode_mark(dnotify_group, inode);
E
Eric Paris 已提交
320 321 322
	if (fsn_mark) {
		dn_mark = container_of(fsn_mark, struct dnotify_mark, fsn_mark);
		spin_lock(&fsn_mark->lock);
323
	} else {
324 325
		fsnotify_add_mark_locked(new_fsn_mark, dnotify_group, inode,
					 NULL, 0);
E
Eric Paris 已提交
326 327 328 329 330
		spin_lock(&new_fsn_mark->lock);
		fsn_mark = new_fsn_mark;
		dn_mark = new_dn_mark;
		/* we used new_fsn_mark, so don't free it */
		new_fsn_mark = NULL;
331
	}
L
Linus Torvalds 已提交
332

333 334 335
	rcu_read_lock();
	f = fcheck(fd);
	rcu_read_unlock();
L
Linus Torvalds 已提交
336

337 338
	/* if (f != filp) means that we lost a race and another task/thread
	 * actually closed the fd we are still playing with before we grabbed
339 340 341
	 * the dnotify_groups mark_mutex and fsn_mark->lock.  Since closing the
	 * fd is the only time we clean up the marks we need to get our mark
	 * off the list. */
342 343
	if (f != filp) {
		/* if we added ourselves, shoot ourselves, it's possible that
E
Eric Paris 已提交
344
		 * the flush actually did shoot this fsn_mark.  That's fine too
345
		 * since multiple calls to destroy_mark is perfectly safe, if
E
Eric Paris 已提交
346
		 * we found a dn_mark already attached to the inode, just sod
347 348
		 * off silently as the flush at close time dealt with it.
		 */
E
Eric Paris 已提交
349
		if (dn_mark == new_dn_mark)
350 351 352
			destroy = 1;
		goto out;
	}
L
Linus Torvalds 已提交
353

354
	__f_setown(filp, task_pid(current), PIDTYPE_PID, 0);
355

E
Eric Paris 已提交
356 357
	error = attach_dn(dn, dn_mark, id, fd, filp, mask);
	/* !error means that we attached the dn to the dn_mark, so don't free it */
358 359 360 361 362 363 364
	if (!error)
		dn = NULL;
	/* -EEXIST means that we didn't add this new dn and used an old one.
	 * that isn't an error (and the unused dn should be freed) */
	else if (error == -EEXIST)
		error = 0;

E
Eric Paris 已提交
365
	dnotify_recalc_inode_mask(fsn_mark);
366
out:
E
Eric Paris 已提交
367
	spin_unlock(&fsn_mark->lock);
368 369

	if (destroy)
370
		fsnotify_detach_mark(fsn_mark);
371
	mutex_unlock(&dnotify_group->mark_mutex);
372 373
	if (destroy)
		fsnotify_free_mark(fsn_mark);
E
Eric Paris 已提交
374
	fsnotify_put_mark(fsn_mark);
375
out_err:
E
Eric Paris 已提交
376 377
	if (new_fsn_mark)
		fsnotify_put_mark(new_fsn_mark);
378 379 380
	if (dn)
		kmem_cache_free(dnotify_struct_cache, dn);
	return error;
L
Linus Torvalds 已提交
381 382 383 384
}

static int __init dnotify_init(void)
{
385
	dnotify_struct_cache = KMEM_CACHE(dnotify_struct, SLAB_PANIC);
E
Eric Paris 已提交
386
	dnotify_mark_cache = KMEM_CACHE(dnotify_mark, SLAB_PANIC);
387

388
	dnotify_group = fsnotify_alloc_group(&dnotify_fsnotify_ops);
389 390
	if (IS_ERR(dnotify_group))
		panic("unable to allocate fsnotify group for dnotify\n");
L
Linus Torvalds 已提交
391 392 393 394
	return 0;
}

module_init(dnotify_init)