xref: /freebsd/sys/kern/uipc_mqueue.c (revision 44c16975a297fa801cbe06a843e78feb7a422cbc)
1655291f2SDavid Xu /*-
2655291f2SDavid Xu  * Copyright (c) 2005 David Xu <davidxu@freebsd.org>
3655291f2SDavid Xu  * All rights reserved.
4655291f2SDavid Xu  *
5655291f2SDavid Xu  * Redistribution and use in source and binary forms, with or without
6655291f2SDavid Xu  * modification, are permitted provided that the following conditions
7655291f2SDavid Xu  * are met:
8655291f2SDavid Xu  * 1. Redistributions of source code must retain the above copyright
9655291f2SDavid Xu  *    notice, this list of conditions and the following disclaimer.
10655291f2SDavid Xu  * 2. Redistributions in binary form must reproduce the above copyright
11655291f2SDavid Xu  *    notice, this list of conditions and the following disclaimer in the
12655291f2SDavid Xu  *    documentation and/or other materials provided with the distribution.
13655291f2SDavid Xu  *
14655291f2SDavid Xu  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15655291f2SDavid Xu  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16655291f2SDavid Xu  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17655291f2SDavid Xu  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18655291f2SDavid Xu  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19655291f2SDavid Xu  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20655291f2SDavid Xu  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21655291f2SDavid Xu  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22655291f2SDavid Xu  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23655291f2SDavid Xu  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24655291f2SDavid Xu  * SUCH DAMAGE.
25655291f2SDavid Xu  *
26655291f2SDavid Xu  */
27655291f2SDavid Xu 
28655291f2SDavid Xu /*
29655291f2SDavid Xu  * POSIX message queue implementation.
30655291f2SDavid Xu  *
31655291f2SDavid Xu  * 1) A mqueue filesystem can be mounted, each message queue appears
32655291f2SDavid Xu  *    in mounted directory, user can change queue's permission and
33655291f2SDavid Xu  *    ownership, or remove a queue. Manually creating a file in the
34655291f2SDavid Xu  *    directory causes a message queue to be created in the kernel with
35655291f2SDavid Xu  *    default message queue attributes applied and same name used, this
36655291f2SDavid Xu  *    method is not advocated since mq_open syscall allows user to specify
37655291f2SDavid Xu  *    different attributes. Also the file system can be mounted multiple
38655291f2SDavid Xu  *    times at different mount points but shows same contents.
39655291f2SDavid Xu  *
40655291f2SDavid Xu  * 2) Standard POSIX message queue API. The syscalls do not use vfs layer,
41655291f2SDavid Xu  *    but directly operate on internal data structure, this allows user to
42655291f2SDavid Xu  *    use the IPC facility without having to mount mqueue file system.
43655291f2SDavid Xu  */
44655291f2SDavid Xu 
45655291f2SDavid Xu #include <sys/cdefs.h>
46655291f2SDavid Xu __FBSDID("$FreeBSD$");
47655291f2SDavid Xu 
482609222aSPawel Jakub Dawidek #include "opt_capsicum.h"
49afde2b65SKonstantin Belousov #include "opt_compat.h"
50afde2b65SKonstantin Belousov 
51655291f2SDavid Xu #include <sys/param.h>
52655291f2SDavid Xu #include <sys/kernel.h>
53655291f2SDavid Xu #include <sys/systm.h>
54655291f2SDavid Xu #include <sys/limits.h>
55adb023aeSJamie Gritton #include <sys/malloc.h>
56655291f2SDavid Xu #include <sys/buf.h>
574a144410SRobert Watson #include <sys/capsicum.h>
58655291f2SDavid Xu #include <sys/dirent.h>
59655291f2SDavid Xu #include <sys/event.h>
60b2f92ef9SDavid Xu #include <sys/eventhandler.h>
61655291f2SDavid Xu #include <sys/fcntl.h>
62655291f2SDavid Xu #include <sys/file.h>
635ee2d4acSDavid Xu #include <sys/filedesc.h>
64adb023aeSJamie Gritton #include <sys/jail.h>
65655291f2SDavid Xu #include <sys/lock.h>
66655291f2SDavid Xu #include <sys/module.h>
67655291f2SDavid Xu #include <sys/mount.h>
68655291f2SDavid Xu #include <sys/mqueue.h>
69655291f2SDavid Xu #include <sys/mutex.h>
70655291f2SDavid Xu #include <sys/namei.h>
716aeb05d7STom Rhodes #include <sys/posix4.h>
72655291f2SDavid Xu #include <sys/poll.h>
73acd3428bSRobert Watson #include <sys/priv.h>
74655291f2SDavid Xu #include <sys/proc.h>
75655291f2SDavid Xu #include <sys/queue.h>
76655291f2SDavid Xu #include <sys/sysproto.h>
77655291f2SDavid Xu #include <sys/stat.h>
78655291f2SDavid Xu #include <sys/syscall.h>
79655291f2SDavid Xu #include <sys/syscallsubr.h>
805ee2d4acSDavid Xu #include <sys/sysent.h>
81655291f2SDavid Xu #include <sys/sx.h>
82655291f2SDavid Xu #include <sys/sysctl.h>
83655291f2SDavid Xu #include <sys/taskqueue.h>
84655291f2SDavid Xu #include <sys/unistd.h>
859696feebSJohn Baldwin #include <sys/user.h>
86655291f2SDavid Xu #include <sys/vnode.h>
87655291f2SDavid Xu #include <machine/atomic.h>
88655291f2SDavid Xu 
89de5b1952SAlexander Leidinger FEATURE(p1003_1b_mqueue, "POSIX P1003.1B message queues support");
90de5b1952SAlexander Leidinger 
91655291f2SDavid Xu /*
92655291f2SDavid Xu  * Limits and constants
93655291f2SDavid Xu  */
94655291f2SDavid Xu #define	MQFS_NAMELEN		NAME_MAX
95655291f2SDavid Xu #define MQFS_DELEN		(8 + MQFS_NAMELEN)
96655291f2SDavid Xu 
97655291f2SDavid Xu /* node types */
98655291f2SDavid Xu typedef enum {
99655291f2SDavid Xu 	mqfstype_none = 0,
100655291f2SDavid Xu 	mqfstype_root,
101655291f2SDavid Xu 	mqfstype_dir,
102655291f2SDavid Xu 	mqfstype_this,
103655291f2SDavid Xu 	mqfstype_parent,
104655291f2SDavid Xu 	mqfstype_file,
105655291f2SDavid Xu 	mqfstype_symlink,
106655291f2SDavid Xu } mqfs_type_t;
107655291f2SDavid Xu 
108655291f2SDavid Xu struct mqfs_node;
109655291f2SDavid Xu 
110655291f2SDavid Xu /*
111655291f2SDavid Xu  * mqfs_info: describes a mqfs instance
112655291f2SDavid Xu  */
113655291f2SDavid Xu struct mqfs_info {
114655291f2SDavid Xu 	struct sx		mi_lock;
115655291f2SDavid Xu 	struct mqfs_node	*mi_root;
116655291f2SDavid Xu 	struct unrhdr		*mi_unrhdr;
117655291f2SDavid Xu };
118655291f2SDavid Xu 
119655291f2SDavid Xu struct mqfs_vdata {
120655291f2SDavid Xu 	LIST_ENTRY(mqfs_vdata)	mv_link;
121655291f2SDavid Xu 	struct mqfs_node	*mv_node;
122655291f2SDavid Xu 	struct vnode		*mv_vnode;
123655291f2SDavid Xu 	struct task		mv_task;
124655291f2SDavid Xu };
125655291f2SDavid Xu 
126655291f2SDavid Xu /*
127655291f2SDavid Xu  * mqfs_node: describes a node (file or directory) within a mqfs
128655291f2SDavid Xu  */
129655291f2SDavid Xu struct mqfs_node {
130655291f2SDavid Xu 	char			mn_name[MQFS_NAMELEN+1];
131655291f2SDavid Xu 	struct mqfs_info	*mn_info;
132655291f2SDavid Xu 	struct mqfs_node	*mn_parent;
133655291f2SDavid Xu 	LIST_HEAD(,mqfs_node)	mn_children;
134655291f2SDavid Xu 	LIST_ENTRY(mqfs_node)	mn_sibling;
135655291f2SDavid Xu 	LIST_HEAD(,mqfs_vdata)	mn_vnodes;
136adb023aeSJamie Gritton 	const void		*mn_pr_root;
137655291f2SDavid Xu 	int			mn_refcount;
138655291f2SDavid Xu 	mqfs_type_t		mn_type;
139655291f2SDavid Xu 	int			mn_deleted;
14060ae52f7SEd Schouten 	uint32_t		mn_fileno;
141655291f2SDavid Xu 	void			*mn_data;
142655291f2SDavid Xu 	struct timespec		mn_birth;
143655291f2SDavid Xu 	struct timespec		mn_ctime;
144655291f2SDavid Xu 	struct timespec		mn_atime;
145655291f2SDavid Xu 	struct timespec		mn_mtime;
146655291f2SDavid Xu 	uid_t			mn_uid;
147655291f2SDavid Xu 	gid_t			mn_gid;
148655291f2SDavid Xu 	int			mn_mode;
149655291f2SDavid Xu };
150655291f2SDavid Xu 
151655291f2SDavid Xu #define	VTON(vp)	(((struct mqfs_vdata *)((vp)->v_data))->mv_node)
152655291f2SDavid Xu #define VTOMQ(vp) 	((struct mqueue *)(VTON(vp)->mn_data))
153655291f2SDavid Xu #define	VFSTOMQFS(m)	((struct mqfs_info *)((m)->mnt_data))
154b2f92ef9SDavid Xu #define	FPTOMQ(fp)	((struct mqueue *)(((struct mqfs_node *) \
155b2f92ef9SDavid Xu 				(fp)->f_data)->mn_data))
156b2f92ef9SDavid Xu 
157adb023aeSJamie Gritton struct mqfs_osd {
158adb023aeSJamie Gritton 	struct task	mo_task;
159adb023aeSJamie Gritton 	const void	*mo_pr_root;
160adb023aeSJamie Gritton };
161adb023aeSJamie Gritton 
162655291f2SDavid Xu TAILQ_HEAD(msgq, mqueue_msg);
163655291f2SDavid Xu 
164655291f2SDavid Xu struct mqueue;
165b2f92ef9SDavid Xu 
166b2f92ef9SDavid Xu struct mqueue_notifier {
167b2f92ef9SDavid Xu 	LIST_ENTRY(mqueue_notifier)	nt_link;
168b2f92ef9SDavid Xu 	struct sigevent			nt_sigev;
169b2f92ef9SDavid Xu 	ksiginfo_t			nt_ksi;
170b2f92ef9SDavid Xu 	struct proc			*nt_proc;
171655291f2SDavid Xu };
172655291f2SDavid Xu 
173655291f2SDavid Xu struct mqueue {
174655291f2SDavid Xu 	struct mtx	mq_mutex;
175655291f2SDavid Xu 	int		mq_flags;
176655291f2SDavid Xu 	long		mq_maxmsg;
177655291f2SDavid Xu 	long		mq_msgsize;
178655291f2SDavid Xu 	long		mq_curmsgs;
179655291f2SDavid Xu 	long		mq_totalbytes;
180655291f2SDavid Xu 	struct msgq	mq_msgq;
181655291f2SDavid Xu 	int		mq_receivers;
182655291f2SDavid Xu 	int		mq_senders;
183655291f2SDavid Xu 	struct selinfo	mq_rsel;
184655291f2SDavid Xu 	struct selinfo	mq_wsel;
185b2f92ef9SDavid Xu 	struct mqueue_notifier	*mq_notifier;
186655291f2SDavid Xu };
187655291f2SDavid Xu 
188655291f2SDavid Xu #define	MQ_RSEL		0x01
189655291f2SDavid Xu #define	MQ_WSEL		0x02
190655291f2SDavid Xu 
191655291f2SDavid Xu struct mqueue_msg {
192655291f2SDavid Xu 	TAILQ_ENTRY(mqueue_msg)	msg_link;
193655291f2SDavid Xu 	unsigned int	msg_prio;
194655291f2SDavid Xu 	unsigned int	msg_size;
195655291f2SDavid Xu 	/* following real data... */
196655291f2SDavid Xu };
197655291f2SDavid Xu 
1986472ac3dSEd Schouten static SYSCTL_NODE(_kern, OID_AUTO, mqueue, CTLFLAG_RW, 0,
199655291f2SDavid Xu 	"POSIX real time message queue");
200655291f2SDavid Xu 
201655291f2SDavid Xu static int	default_maxmsg  = 10;
202655291f2SDavid Xu static int	default_msgsize = 1024;
203655291f2SDavid Xu 
2049947b459SDavid Xu static int	maxmsg = 100;
205655291f2SDavid Xu SYSCTL_INT(_kern_mqueue, OID_AUTO, maxmsg, CTLFLAG_RW,
206655291f2SDavid Xu     &maxmsg, 0, "Default maximum messages in queue");
207655291f2SDavid Xu static int	maxmsgsize = 16384;
208655291f2SDavid Xu SYSCTL_INT(_kern_mqueue, OID_AUTO, maxmsgsize, CTLFLAG_RW,
209655291f2SDavid Xu     &maxmsgsize, 0, "Default maximum message size");
210655291f2SDavid Xu static int	maxmq = 100;
211655291f2SDavid Xu SYSCTL_INT(_kern_mqueue, OID_AUTO, maxmq, CTLFLAG_RW,
212655291f2SDavid Xu     &maxmq, 0, "maximum message queues");
213655291f2SDavid Xu static int	curmq = 0;
214655291f2SDavid Xu SYSCTL_INT(_kern_mqueue, OID_AUTO, curmq, CTLFLAG_RW,
215655291f2SDavid Xu     &curmq, 0, "current message queue number");
216655291f2SDavid Xu static int	unloadable = 0;
217655291f2SDavid Xu static MALLOC_DEFINE(M_MQUEUEDATA, "mqdata", "mqueue data");
218655291f2SDavid Xu 
219b2f92ef9SDavid Xu static eventhandler_tag exit_tag;
220b2f92ef9SDavid Xu 
221655291f2SDavid Xu /* Only one instance per-system */
222655291f2SDavid Xu static struct mqfs_info		mqfs_data;
223655291f2SDavid Xu static uma_zone_t		mqnode_zone;
224655291f2SDavid Xu static uma_zone_t		mqueue_zone;
225655291f2SDavid Xu static uma_zone_t		mvdata_zone;
226b2f92ef9SDavid Xu static uma_zone_t		mqnoti_zone;
227655291f2SDavid Xu static struct vop_vector	mqfs_vnodeops;
228655291f2SDavid Xu static struct fileops		mqueueops;
229adb023aeSJamie Gritton static unsigned			mqfs_osd_jail_slot;
230655291f2SDavid Xu 
231655291f2SDavid Xu /*
232655291f2SDavid Xu  * Directory structure construction and manipulation
233655291f2SDavid Xu  */
234655291f2SDavid Xu #ifdef notyet
235655291f2SDavid Xu static struct mqfs_node	*mqfs_create_dir(struct mqfs_node *parent,
236ba0360b1SDavid Xu 	const char *name, int namelen, struct ucred *cred, int mode);
237102178d0SDavid Xu static struct mqfs_node	*mqfs_create_link(struct mqfs_node *parent,
238ba0360b1SDavid Xu 	const char *name, int namelen, struct ucred *cred, int mode);
239655291f2SDavid Xu #endif
240655291f2SDavid Xu 
241655291f2SDavid Xu static struct mqfs_node	*mqfs_create_file(struct mqfs_node *parent,
242ba0360b1SDavid Xu 	const char *name, int namelen, struct ucred *cred, int mode);
243655291f2SDavid Xu static int	mqfs_destroy(struct mqfs_node *mn);
244655291f2SDavid Xu static void	mqfs_fileno_alloc(struct mqfs_info *mi, struct mqfs_node *mn);
245655291f2SDavid Xu static void	mqfs_fileno_free(struct mqfs_info *mi, struct mqfs_node *mn);
246655291f2SDavid Xu static int	mqfs_allocv(struct mount *mp, struct vnode **vpp, struct mqfs_node *pn);
247adb023aeSJamie Gritton static int	mqfs_prison_create(void *obj, void *data);
248adb023aeSJamie Gritton static void	mqfs_prison_destructor(void *data);
249adb023aeSJamie Gritton static void	mqfs_prison_remove_task(void *context, int pending);
250655291f2SDavid Xu 
251655291f2SDavid Xu /*
252655291f2SDavid Xu  * Message queue construction and maniplation
253655291f2SDavid Xu  */
254655291f2SDavid Xu static struct mqueue	*mqueue_alloc(const struct mq_attr *attr);
255655291f2SDavid Xu static void	mqueue_free(struct mqueue *mq);
256655291f2SDavid Xu static int	mqueue_send(struct mqueue *mq, const char *msg_ptr,
257655291f2SDavid Xu 			size_t msg_len, unsigned msg_prio, int waitok,
258655291f2SDavid Xu 			const struct timespec *abs_timeout);
259655291f2SDavid Xu static int	mqueue_receive(struct mqueue *mq, char *msg_ptr,
260655291f2SDavid Xu 			size_t msg_len, unsigned *msg_prio, int waitok,
261655291f2SDavid Xu 			const struct timespec *abs_timeout);
262655291f2SDavid Xu static int	_mqueue_send(struct mqueue *mq, struct mqueue_msg *msg,
263655291f2SDavid Xu 			int timo);
264655291f2SDavid Xu static int	_mqueue_recv(struct mqueue *mq, struct mqueue_msg **msg,
265655291f2SDavid Xu 			int timo);
266655291f2SDavid Xu static void	mqueue_send_notification(struct mqueue *mq);
267b2f92ef9SDavid Xu static void	mqueue_fdclose(struct thread *td, int fd, struct file *fp);
268b2f92ef9SDavid Xu static void	mq_proc_exit(void *arg, struct proc *p);
269655291f2SDavid Xu 
270b2f92ef9SDavid Xu /*
271b2f92ef9SDavid Xu  * kqueue filters
272b2f92ef9SDavid Xu  */
273655291f2SDavid Xu static void	filt_mqdetach(struct knote *kn);
274655291f2SDavid Xu static int	filt_mqread(struct knote *kn, long hint);
275655291f2SDavid Xu static int	filt_mqwrite(struct knote *kn, long hint);
276655291f2SDavid Xu 
277e76d823bSRobert Watson struct filterops mq_rfiltops = {
278e76d823bSRobert Watson 	.f_isfd = 1,
279e76d823bSRobert Watson 	.f_detach = filt_mqdetach,
280e76d823bSRobert Watson 	.f_event = filt_mqread,
281e76d823bSRobert Watson };
282e76d823bSRobert Watson struct filterops mq_wfiltops = {
283e76d823bSRobert Watson 	.f_isfd = 1,
284e76d823bSRobert Watson 	.f_detach = filt_mqdetach,
285e76d823bSRobert Watson 	.f_event = filt_mqwrite,
286e76d823bSRobert Watson };
287655291f2SDavid Xu 
288655291f2SDavid Xu /*
289655291f2SDavid Xu  * Initialize fileno bitmap
290655291f2SDavid Xu  */
291655291f2SDavid Xu static void
292655291f2SDavid Xu mqfs_fileno_init(struct mqfs_info *mi)
293655291f2SDavid Xu {
294655291f2SDavid Xu 	struct unrhdr *up;
295655291f2SDavid Xu 
296655291f2SDavid Xu 	up = new_unrhdr(1, INT_MAX, NULL);
297655291f2SDavid Xu 	mi->mi_unrhdr = up;
298655291f2SDavid Xu }
299655291f2SDavid Xu 
300655291f2SDavid Xu /*
301655291f2SDavid Xu  * Tear down fileno bitmap
302655291f2SDavid Xu  */
303655291f2SDavid Xu static void
304655291f2SDavid Xu mqfs_fileno_uninit(struct mqfs_info *mi)
305655291f2SDavid Xu {
306655291f2SDavid Xu 	struct unrhdr *up;
307655291f2SDavid Xu 
308655291f2SDavid Xu 	up = mi->mi_unrhdr;
309655291f2SDavid Xu 	mi->mi_unrhdr = NULL;
310655291f2SDavid Xu 	delete_unrhdr(up);
311655291f2SDavid Xu }
312655291f2SDavid Xu 
313655291f2SDavid Xu /*
314655291f2SDavid Xu  * Allocate a file number
315655291f2SDavid Xu  */
316b2f92ef9SDavid Xu static void
317655291f2SDavid Xu mqfs_fileno_alloc(struct mqfs_info *mi, struct mqfs_node *mn)
318655291f2SDavid Xu {
319655291f2SDavid Xu 	/* make sure our parent has a file number */
320655291f2SDavid Xu 	if (mn->mn_parent && !mn->mn_parent->mn_fileno)
321655291f2SDavid Xu 		mqfs_fileno_alloc(mi, mn->mn_parent);
322655291f2SDavid Xu 
323655291f2SDavid Xu 	switch (mn->mn_type) {
324655291f2SDavid Xu 	case mqfstype_root:
325655291f2SDavid Xu 	case mqfstype_dir:
326655291f2SDavid Xu 	case mqfstype_file:
327655291f2SDavid Xu 	case mqfstype_symlink:
328655291f2SDavid Xu 		mn->mn_fileno = alloc_unr(mi->mi_unrhdr);
329655291f2SDavid Xu 		break;
330655291f2SDavid Xu 	case mqfstype_this:
331655291f2SDavid Xu 		KASSERT(mn->mn_parent != NULL,
332655291f2SDavid Xu 		    ("mqfstype_this node has no parent"));
333655291f2SDavid Xu 		mn->mn_fileno = mn->mn_parent->mn_fileno;
334655291f2SDavid Xu 		break;
335655291f2SDavid Xu 	case mqfstype_parent:
336655291f2SDavid Xu 		KASSERT(mn->mn_parent != NULL,
337655291f2SDavid Xu 		    ("mqfstype_parent node has no parent"));
338655291f2SDavid Xu 		if (mn->mn_parent == mi->mi_root) {
339655291f2SDavid Xu 			mn->mn_fileno = mn->mn_parent->mn_fileno;
340655291f2SDavid Xu 			break;
341655291f2SDavid Xu 		}
342655291f2SDavid Xu 		KASSERT(mn->mn_parent->mn_parent != NULL,
343655291f2SDavid Xu 		    ("mqfstype_parent node has no grandparent"));
344655291f2SDavid Xu 		mn->mn_fileno = mn->mn_parent->mn_parent->mn_fileno;
345655291f2SDavid Xu 		break;
346655291f2SDavid Xu 	default:
347655291f2SDavid Xu 		KASSERT(0,
348655291f2SDavid Xu 		    ("mqfs_fileno_alloc() called for unknown type node: %d",
349655291f2SDavid Xu 			mn->mn_type));
350655291f2SDavid Xu 		break;
351655291f2SDavid Xu 	}
352655291f2SDavid Xu }
353655291f2SDavid Xu 
354655291f2SDavid Xu /*
355655291f2SDavid Xu  * Release a file number
356655291f2SDavid Xu  */
357b2f92ef9SDavid Xu static void
358655291f2SDavid Xu mqfs_fileno_free(struct mqfs_info *mi, struct mqfs_node *mn)
359655291f2SDavid Xu {
360655291f2SDavid Xu 	switch (mn->mn_type) {
361655291f2SDavid Xu 	case mqfstype_root:
362655291f2SDavid Xu 	case mqfstype_dir:
363655291f2SDavid Xu 	case mqfstype_file:
364655291f2SDavid Xu 	case mqfstype_symlink:
365655291f2SDavid Xu 		free_unr(mi->mi_unrhdr, mn->mn_fileno);
366655291f2SDavid Xu 		break;
367655291f2SDavid Xu 	case mqfstype_this:
368655291f2SDavid Xu 	case mqfstype_parent:
369655291f2SDavid Xu 		/* ignore these, as they don't "own" their file number */
370655291f2SDavid Xu 		break;
371655291f2SDavid Xu 	default:
372655291f2SDavid Xu 		KASSERT(0,
373655291f2SDavid Xu 		    ("mqfs_fileno_free() called for unknown type node: %d",
374655291f2SDavid Xu 			mn->mn_type));
375655291f2SDavid Xu 		break;
376655291f2SDavid Xu 	}
377655291f2SDavid Xu }
378655291f2SDavid Xu 
379655291f2SDavid Xu static __inline struct mqfs_node *
380655291f2SDavid Xu mqnode_alloc(void)
381655291f2SDavid Xu {
382655291f2SDavid Xu 	return uma_zalloc(mqnode_zone, M_WAITOK | M_ZERO);
383655291f2SDavid Xu }
384655291f2SDavid Xu 
385655291f2SDavid Xu static __inline void
386655291f2SDavid Xu mqnode_free(struct mqfs_node *node)
387655291f2SDavid Xu {
388655291f2SDavid Xu 	uma_zfree(mqnode_zone, node);
389655291f2SDavid Xu }
390655291f2SDavid Xu 
391655291f2SDavid Xu static __inline void
392655291f2SDavid Xu mqnode_addref(struct mqfs_node *node)
393655291f2SDavid Xu {
394655291f2SDavid Xu 	atomic_fetchadd_int(&node->mn_refcount, 1);
395655291f2SDavid Xu }
396655291f2SDavid Xu 
397655291f2SDavid Xu static __inline void
398655291f2SDavid Xu mqnode_release(struct mqfs_node *node)
399655291f2SDavid Xu {
400fbc48e97SDavid Xu 	struct mqfs_info *mqfs;
401655291f2SDavid Xu 	int old, exp;
402655291f2SDavid Xu 
403fbc48e97SDavid Xu 	mqfs = node->mn_info;
404655291f2SDavid Xu 	old = atomic_fetchadd_int(&node->mn_refcount, -1);
405655291f2SDavid Xu 	if (node->mn_type == mqfstype_dir ||
406655291f2SDavid Xu 	    node->mn_type == mqfstype_root)
407655291f2SDavid Xu 		exp = 3; /* include . and .. */
408655291f2SDavid Xu 	else
409655291f2SDavid Xu 		exp = 1;
410fbc48e97SDavid Xu 	if (old == exp) {
411fbc48e97SDavid Xu 		int locked = sx_xlocked(&mqfs->mi_lock);
412fbc48e97SDavid Xu 		if (!locked)
413fbc48e97SDavid Xu 			sx_xlock(&mqfs->mi_lock);
414655291f2SDavid Xu 		mqfs_destroy(node);
415fbc48e97SDavid Xu 		if (!locked)
416fbc48e97SDavid Xu 			sx_xunlock(&mqfs->mi_lock);
417fbc48e97SDavid Xu 	}
418655291f2SDavid Xu }
419655291f2SDavid Xu 
420655291f2SDavid Xu /*
421655291f2SDavid Xu  * Add a node to a directory
422655291f2SDavid Xu  */
423655291f2SDavid Xu static int
424655291f2SDavid Xu mqfs_add_node(struct mqfs_node *parent, struct mqfs_node *node)
425655291f2SDavid Xu {
426655291f2SDavid Xu 	KASSERT(parent != NULL, ("%s(): parent is NULL", __func__));
427655291f2SDavid Xu 	KASSERT(parent->mn_info != NULL,
428655291f2SDavid Xu 	    ("%s(): parent has no mn_info", __func__));
429655291f2SDavid Xu 	KASSERT(parent->mn_type == mqfstype_dir ||
430655291f2SDavid Xu 	    parent->mn_type == mqfstype_root,
431655291f2SDavid Xu 	    ("%s(): parent is not a directory", __func__));
432655291f2SDavid Xu 
433655291f2SDavid Xu 	node->mn_info = parent->mn_info;
434655291f2SDavid Xu 	node->mn_parent = parent;
435655291f2SDavid Xu 	LIST_INIT(&node->mn_children);
436655291f2SDavid Xu 	LIST_INIT(&node->mn_vnodes);
437655291f2SDavid Xu 	LIST_INSERT_HEAD(&parent->mn_children, node, mn_sibling);
438655291f2SDavid Xu 	mqnode_addref(parent);
439655291f2SDavid Xu 	return (0);
440655291f2SDavid Xu }
441655291f2SDavid Xu 
442ba0360b1SDavid Xu static struct mqfs_node *
443ba0360b1SDavid Xu mqfs_create_node(const char *name, int namelen, struct ucred *cred, int mode,
444ba0360b1SDavid Xu 	int nodetype)
445ba0360b1SDavid Xu {
446ba0360b1SDavid Xu 	struct mqfs_node *node;
447ba0360b1SDavid Xu 
448ba0360b1SDavid Xu 	node = mqnode_alloc();
449ba0360b1SDavid Xu 	strncpy(node->mn_name, name, namelen);
450adb023aeSJamie Gritton 	node->mn_pr_root = cred->cr_prison->pr_root;
451ba0360b1SDavid Xu 	node->mn_type = nodetype;
452ba0360b1SDavid Xu 	node->mn_refcount = 1;
453fbc48e97SDavid Xu 	vfs_timestamp(&node->mn_birth);
454ba0360b1SDavid Xu 	node->mn_ctime = node->mn_atime = node->mn_mtime
455ba0360b1SDavid Xu 		= node->mn_birth;
456ba0360b1SDavid Xu 	node->mn_uid = cred->cr_uid;
457ba0360b1SDavid Xu 	node->mn_gid = cred->cr_gid;
458ba0360b1SDavid Xu 	node->mn_mode = mode;
459ba0360b1SDavid Xu 	return (node);
460ba0360b1SDavid Xu }
461ba0360b1SDavid Xu 
462ba0360b1SDavid Xu /*
463ba0360b1SDavid Xu  * Create a file
464ba0360b1SDavid Xu  */
465ba0360b1SDavid Xu static struct mqfs_node *
466ba0360b1SDavid Xu mqfs_create_file(struct mqfs_node *parent, const char *name, int namelen,
467ba0360b1SDavid Xu 	struct ucred *cred, int mode)
468ba0360b1SDavid Xu {
469ba0360b1SDavid Xu 	struct mqfs_node *node;
470ba0360b1SDavid Xu 
471ba0360b1SDavid Xu 	node = mqfs_create_node(name, namelen, cred, mode, mqfstype_file);
472ba0360b1SDavid Xu 	if (mqfs_add_node(parent, node) != 0) {
473ba0360b1SDavid Xu 		mqnode_free(node);
474ba0360b1SDavid Xu 		return (NULL);
475ba0360b1SDavid Xu 	}
476ba0360b1SDavid Xu 	return (node);
477ba0360b1SDavid Xu }
478ba0360b1SDavid Xu 
479655291f2SDavid Xu /*
480655291f2SDavid Xu  * Add . and .. to a directory
481655291f2SDavid Xu  */
482655291f2SDavid Xu static int
483655291f2SDavid Xu mqfs_fixup_dir(struct mqfs_node *parent)
484655291f2SDavid Xu {
485655291f2SDavid Xu 	struct mqfs_node *dir;
486655291f2SDavid Xu 
487655291f2SDavid Xu 	dir = mqnode_alloc();
488655291f2SDavid Xu 	dir->mn_name[0] = '.';
489655291f2SDavid Xu 	dir->mn_type = mqfstype_this;
490655291f2SDavid Xu 	dir->mn_refcount = 1;
491655291f2SDavid Xu 	if (mqfs_add_node(parent, dir) != 0) {
492655291f2SDavid Xu 		mqnode_free(dir);
493655291f2SDavid Xu 		return (-1);
494655291f2SDavid Xu 	}
495655291f2SDavid Xu 
496655291f2SDavid Xu 	dir = mqnode_alloc();
497655291f2SDavid Xu 	dir->mn_name[0] = dir->mn_name[1] = '.';
498655291f2SDavid Xu 	dir->mn_type = mqfstype_parent;
499655291f2SDavid Xu 	dir->mn_refcount = 1;
500655291f2SDavid Xu 
501655291f2SDavid Xu 	if (mqfs_add_node(parent, dir) != 0) {
502655291f2SDavid Xu 		mqnode_free(dir);
503655291f2SDavid Xu 		return (-1);
504655291f2SDavid Xu 	}
505655291f2SDavid Xu 
506655291f2SDavid Xu 	return (0);
507655291f2SDavid Xu }
508655291f2SDavid Xu 
509655291f2SDavid Xu #ifdef notyet
510655291f2SDavid Xu 
511655291f2SDavid Xu /*
512655291f2SDavid Xu  * Create a directory
513655291f2SDavid Xu  */
514b2f92ef9SDavid Xu static struct mqfs_node *
515ba0360b1SDavid Xu mqfs_create_dir(struct mqfs_node *parent, const char *name, int namelen,
516ba0360b1SDavid Xu 	struct ucred *cred, int mode)
517655291f2SDavid Xu {
518ba0360b1SDavid Xu 	struct mqfs_node *node;
519655291f2SDavid Xu 
520ba0360b1SDavid Xu 	node = mqfs_create_node(name, namelen, cred, mode, mqfstype_dir);
521ba0360b1SDavid Xu 	if (mqfs_add_node(parent, node) != 0) {
522ba0360b1SDavid Xu 		mqnode_free(node);
523655291f2SDavid Xu 		return (NULL);
524655291f2SDavid Xu 	}
525655291f2SDavid Xu 
526ba0360b1SDavid Xu 	if (mqfs_fixup_dir(node) != 0) {
527ba0360b1SDavid Xu 		mqfs_destroy(node);
528655291f2SDavid Xu 		return (NULL);
529655291f2SDavid Xu 	}
530ba0360b1SDavid Xu 	return (node);
531655291f2SDavid Xu }
532b2f92ef9SDavid Xu 
533b2f92ef9SDavid Xu /*
534b2f92ef9SDavid Xu  * Create a symlink
535b2f92ef9SDavid Xu  */
536b2f92ef9SDavid Xu static struct mqfs_node *
537ba0360b1SDavid Xu mqfs_create_link(struct mqfs_node *parent, const char *name, int namelen,
538ba0360b1SDavid Xu 	struct ucred *cred, int mode)
539b2f92ef9SDavid Xu {
540b2f92ef9SDavid Xu 	struct mqfs_node *node;
541b2f92ef9SDavid Xu 
542ba0360b1SDavid Xu 	node = mqfs_create_node(name, namelen, cred, mode, mqfstype_symlink);
543655291f2SDavid Xu 	if (mqfs_add_node(parent, node) != 0) {
544655291f2SDavid Xu 		mqnode_free(node);
545655291f2SDavid Xu 		return (NULL);
546655291f2SDavid Xu 	}
547655291f2SDavid Xu 	return (node);
548655291f2SDavid Xu }
549655291f2SDavid Xu 
550ba0360b1SDavid Xu #endif
551ba0360b1SDavid Xu 
552655291f2SDavid Xu /*
553655291f2SDavid Xu  * Destroy a node or a tree of nodes
554655291f2SDavid Xu  */
555b2f92ef9SDavid Xu static int
556655291f2SDavid Xu mqfs_destroy(struct mqfs_node *node)
557655291f2SDavid Xu {
558655291f2SDavid Xu 	struct mqfs_node *parent;
559655291f2SDavid Xu 
560655291f2SDavid Xu 	KASSERT(node != NULL,
561655291f2SDavid Xu 	    ("%s(): node is NULL", __func__));
562655291f2SDavid Xu 	KASSERT(node->mn_info != NULL,
563655291f2SDavid Xu 	    ("%s(): node has no mn_info", __func__));
564655291f2SDavid Xu 
565655291f2SDavid Xu 	/* destroy children */
566655291f2SDavid Xu 	if (node->mn_type == mqfstype_dir || node->mn_type == mqfstype_root)
567655291f2SDavid Xu 		while (! LIST_EMPTY(&node->mn_children))
568655291f2SDavid Xu 			mqfs_destroy(LIST_FIRST(&node->mn_children));
569655291f2SDavid Xu 
570655291f2SDavid Xu 	/* unlink from parent */
571655291f2SDavid Xu 	if ((parent = node->mn_parent) != NULL) {
572655291f2SDavid Xu 		KASSERT(parent->mn_info == node->mn_info,
573655291f2SDavid Xu 		    ("%s(): parent has different mn_info", __func__));
574655291f2SDavid Xu 		LIST_REMOVE(node, mn_sibling);
575655291f2SDavid Xu 	}
576655291f2SDavid Xu 
577655291f2SDavid Xu 	if (node->mn_fileno != 0)
578655291f2SDavid Xu 		mqfs_fileno_free(node->mn_info, node);
579655291f2SDavid Xu 	if (node->mn_data != NULL)
580655291f2SDavid Xu 		mqueue_free(node->mn_data);
581655291f2SDavid Xu 	mqnode_free(node);
582655291f2SDavid Xu 	return (0);
583655291f2SDavid Xu }
584655291f2SDavid Xu 
585655291f2SDavid Xu /*
586655291f2SDavid Xu  * Mount a mqfs instance
587655291f2SDavid Xu  */
588655291f2SDavid Xu static int
589dfd233edSAttilio Rao mqfs_mount(struct mount *mp)
590655291f2SDavid Xu {
591655291f2SDavid Xu 	struct statfs *sbp;
592655291f2SDavid Xu 
593655291f2SDavid Xu 	if (mp->mnt_flag & MNT_UPDATE)
594655291f2SDavid Xu 		return (EOPNOTSUPP);
595655291f2SDavid Xu 
596655291f2SDavid Xu 	mp->mnt_data = &mqfs_data;
5975da56ddbSTor Egge 	MNT_ILOCK(mp);
598655291f2SDavid Xu 	mp->mnt_flag |= MNT_LOCAL;
5995da56ddbSTor Egge 	MNT_IUNLOCK(mp);
600655291f2SDavid Xu 	vfs_getnewfsid(mp);
601655291f2SDavid Xu 
602655291f2SDavid Xu 	sbp = &mp->mnt_stat;
603655291f2SDavid Xu 	vfs_mountedfrom(mp, "mqueue");
604655291f2SDavid Xu 	sbp->f_bsize = PAGE_SIZE;
605655291f2SDavid Xu 	sbp->f_iosize = PAGE_SIZE;
606655291f2SDavid Xu 	sbp->f_blocks = 1;
607655291f2SDavid Xu 	sbp->f_bfree = 0;
608655291f2SDavid Xu 	sbp->f_bavail = 0;
609655291f2SDavid Xu 	sbp->f_files = 1;
610655291f2SDavid Xu 	sbp->f_ffree = 0;
611655291f2SDavid Xu 	return (0);
612655291f2SDavid Xu }
613655291f2SDavid Xu 
614655291f2SDavid Xu /*
615655291f2SDavid Xu  * Unmount a mqfs instance
616655291f2SDavid Xu  */
617655291f2SDavid Xu static int
618dfd233edSAttilio Rao mqfs_unmount(struct mount *mp, int mntflags)
619655291f2SDavid Xu {
620655291f2SDavid Xu 	int error;
621655291f2SDavid Xu 
622dfd233edSAttilio Rao 	error = vflush(mp, 0, (mntflags & MNT_FORCE) ?  FORCECLOSE : 0,
623dfd233edSAttilio Rao 	    curthread);
624655291f2SDavid Xu 	return (error);
625655291f2SDavid Xu }
626655291f2SDavid Xu 
627655291f2SDavid Xu /*
628655291f2SDavid Xu  * Return a root vnode
629655291f2SDavid Xu  */
630655291f2SDavid Xu static int
631dfd233edSAttilio Rao mqfs_root(struct mount *mp, int flags, struct vnode **vpp)
632655291f2SDavid Xu {
633655291f2SDavid Xu 	struct mqfs_info *mqfs;
634655291f2SDavid Xu 	int ret;
635655291f2SDavid Xu 
636655291f2SDavid Xu 	mqfs = VFSTOMQFS(mp);
637655291f2SDavid Xu 	ret = mqfs_allocv(mp, vpp, mqfs->mi_root);
638655291f2SDavid Xu 	return (ret);
639655291f2SDavid Xu }
640655291f2SDavid Xu 
641655291f2SDavid Xu /*
642655291f2SDavid Xu  * Return filesystem stats
643655291f2SDavid Xu  */
644655291f2SDavid Xu static int
645dfd233edSAttilio Rao mqfs_statfs(struct mount *mp, struct statfs *sbp)
646655291f2SDavid Xu {
647655291f2SDavid Xu 	/* XXX update statistics */
648655291f2SDavid Xu 	return (0);
649655291f2SDavid Xu }
650655291f2SDavid Xu 
651655291f2SDavid Xu /*
652655291f2SDavid Xu  * Initialize a mqfs instance
653655291f2SDavid Xu  */
654655291f2SDavid Xu static int
655655291f2SDavid Xu mqfs_init(struct vfsconf *vfc)
656655291f2SDavid Xu {
657655291f2SDavid Xu 	struct mqfs_node *root;
658655291f2SDavid Xu 	struct mqfs_info *mi;
659adb023aeSJamie Gritton 	struct prison *pr;
660adb023aeSJamie Gritton 	osd_method_t methods[PR_MAXMETHOD] = {
661adb023aeSJamie Gritton 	    [PR_METHOD_CREATE] = mqfs_prison_create,
662adb023aeSJamie Gritton 	};
663655291f2SDavid Xu 
664655291f2SDavid Xu 	mqnode_zone = uma_zcreate("mqnode", sizeof(struct mqfs_node),
665655291f2SDavid Xu 		NULL, NULL, NULL, NULL, UMA_ALIGN_PTR, 0);
666655291f2SDavid Xu 	mqueue_zone = uma_zcreate("mqueue", sizeof(struct mqueue),
667655291f2SDavid Xu 		NULL, NULL, NULL, NULL, UMA_ALIGN_PTR, 0);
668655291f2SDavid Xu 	mvdata_zone = uma_zcreate("mvdata",
669655291f2SDavid Xu 		sizeof(struct mqfs_vdata), NULL, NULL, NULL,
670655291f2SDavid Xu 		NULL, UMA_ALIGN_PTR, 0);
671b2f92ef9SDavid Xu 	mqnoti_zone = uma_zcreate("mqnotifier", sizeof(struct mqueue_notifier),
672655291f2SDavid Xu 		NULL, NULL, NULL, NULL, UMA_ALIGN_PTR, 0);
673655291f2SDavid Xu 	mi = &mqfs_data;
674655291f2SDavid Xu 	sx_init(&mi->mi_lock, "mqfs lock");
675655291f2SDavid Xu 	/* set up the root diretory */
676ba0360b1SDavid Xu 	root = mqfs_create_node("/", 1, curthread->td_ucred, 01777,
677ba0360b1SDavid Xu 		mqfstype_root);
678655291f2SDavid Xu 	root->mn_info = mi;
679655291f2SDavid Xu 	LIST_INIT(&root->mn_children);
680655291f2SDavid Xu 	LIST_INIT(&root->mn_vnodes);
681655291f2SDavid Xu 	mi->mi_root = root;
682655291f2SDavid Xu 	mqfs_fileno_init(mi);
683655291f2SDavid Xu 	mqfs_fileno_alloc(mi, root);
684655291f2SDavid Xu 	mqfs_fixup_dir(root);
685b2f92ef9SDavid Xu 	exit_tag = EVENTHANDLER_REGISTER(process_exit, mq_proc_exit, NULL,
686b2f92ef9SDavid Xu 	    EVENTHANDLER_PRI_ANY);
687b2f92ef9SDavid Xu 	mq_fdclose = mqueue_fdclose;
6885ee2d4acSDavid Xu 	p31b_setcfg(CTL_P1003_1B_MESSAGE_PASSING, _POSIX_MESSAGE_PASSING);
689*44c16975SJamie Gritton 
690*44c16975SJamie Gritton 	/* Note current jails. */
691adb023aeSJamie Gritton 	mqfs_osd_jail_slot = osd_jail_register(mqfs_prison_destructor, methods);
692adb023aeSJamie Gritton 	sx_slock(&allprison_lock);
693adb023aeSJamie Gritton 	TAILQ_FOREACH(pr, &allprison, pr_list)
694adb023aeSJamie Gritton 		(void)mqfs_prison_create(pr, NULL);
695adb023aeSJamie Gritton 	sx_sunlock(&allprison_lock);
696655291f2SDavid Xu 	return (0);
697655291f2SDavid Xu }
698655291f2SDavid Xu 
699655291f2SDavid Xu /*
700655291f2SDavid Xu  * Destroy a mqfs instance
701655291f2SDavid Xu  */
702655291f2SDavid Xu static int
703655291f2SDavid Xu mqfs_uninit(struct vfsconf *vfc)
704655291f2SDavid Xu {
705adb023aeSJamie Gritton 	unsigned slot;
706655291f2SDavid Xu 	struct mqfs_info *mi;
707655291f2SDavid Xu 
708655291f2SDavid Xu 	if (!unloadable)
709655291f2SDavid Xu 		return (EOPNOTSUPP);
710adb023aeSJamie Gritton 	slot = mqfs_osd_jail_slot;
711adb023aeSJamie Gritton 	mqfs_osd_jail_slot = 0;
712adb023aeSJamie Gritton 	osd_jail_deregister(slot);
713b2f92ef9SDavid Xu 	EVENTHANDLER_DEREGISTER(process_exit, exit_tag);
714655291f2SDavid Xu 	mi = &mqfs_data;
715655291f2SDavid Xu 	mqfs_destroy(mi->mi_root);
716655291f2SDavid Xu 	mi->mi_root = NULL;
717655291f2SDavid Xu 	mqfs_fileno_uninit(mi);
718655291f2SDavid Xu 	sx_destroy(&mi->mi_lock);
719b2f92ef9SDavid Xu 	uma_zdestroy(mqnode_zone);
720b2f92ef9SDavid Xu 	uma_zdestroy(mqueue_zone);
721b2f92ef9SDavid Xu 	uma_zdestroy(mvdata_zone);
722b2f92ef9SDavid Xu 	uma_zdestroy(mqnoti_zone);
723655291f2SDavid Xu 	return (0);
724655291f2SDavid Xu }
725655291f2SDavid Xu 
726655291f2SDavid Xu /*
727655291f2SDavid Xu  * task routine
728655291f2SDavid Xu  */
729655291f2SDavid Xu static void
730655291f2SDavid Xu do_recycle(void *context, int pending __unused)
731655291f2SDavid Xu {
732655291f2SDavid Xu 	struct vnode *vp = (struct vnode *)context;
733655291f2SDavid Xu 
734af6e6b87SEdward Tomasz Napierala 	vrecycle(vp);
735655291f2SDavid Xu 	vdrop(vp);
736655291f2SDavid Xu }
737655291f2SDavid Xu 
738655291f2SDavid Xu /*
739655291f2SDavid Xu  * Allocate a vnode
740655291f2SDavid Xu  */
741b2f92ef9SDavid Xu static int
742655291f2SDavid Xu mqfs_allocv(struct mount *mp, struct vnode **vpp, struct mqfs_node *pn)
743655291f2SDavid Xu {
744655291f2SDavid Xu 	struct mqfs_vdata *vd;
745fbc48e97SDavid Xu 	struct mqfs_info  *mqfs;
746fbc48e97SDavid Xu 	struct vnode *newvpp;
747655291f2SDavid Xu 	int error;
748655291f2SDavid Xu 
749fbc48e97SDavid Xu 	mqfs = pn->mn_info;
750fbc48e97SDavid Xu 	*vpp = NULL;
751fbc48e97SDavid Xu 	sx_xlock(&mqfs->mi_lock);
752655291f2SDavid Xu 	LIST_FOREACH(vd, &pn->mn_vnodes, mv_link) {
753fbc48e97SDavid Xu 		if (vd->mv_vnode->v_mount == mp) {
754fbc48e97SDavid Xu 			vhold(vd->mv_vnode);
755655291f2SDavid Xu 			break;
756655291f2SDavid Xu 		}
757fbc48e97SDavid Xu 	}
758655291f2SDavid Xu 
759655291f2SDavid Xu 	if (vd != NULL) {
760fbc48e97SDavid Xu found:
761655291f2SDavid Xu 		*vpp = vd->mv_vnode;
762fbc48e97SDavid Xu 		sx_xunlock(&mqfs->mi_lock);
763fbc48e97SDavid Xu 		error = vget(*vpp, LK_RETRY | LK_EXCLUSIVE, curthread);
764fbc48e97SDavid Xu 		vdrop(*vpp);
765fbc48e97SDavid Xu 		return (error);
766655291f2SDavid Xu 	}
767fbc48e97SDavid Xu 	sx_xunlock(&mqfs->mi_lock);
768655291f2SDavid Xu 
769fbc48e97SDavid Xu 	error = getnewvnode("mqueue", mp, &mqfs_vnodeops, &newvpp);
770655291f2SDavid Xu 	if (error)
771655291f2SDavid Xu 		return (error);
772fbc48e97SDavid Xu 	vn_lock(newvpp, LK_EXCLUSIVE | LK_RETRY);
773fbc48e97SDavid Xu 	error = insmntque(newvpp, mp);
774fbc48e97SDavid Xu 	if (error != 0)
77561b9d89fSTor Egge 		return (error);
776fbc48e97SDavid Xu 
777fbc48e97SDavid Xu 	sx_xlock(&mqfs->mi_lock);
778fbc48e97SDavid Xu 	/*
779fbc48e97SDavid Xu 	 * Check if it has already been allocated
780fbc48e97SDavid Xu 	 * while we were blocked.
781fbc48e97SDavid Xu 	 */
782fbc48e97SDavid Xu 	LIST_FOREACH(vd, &pn->mn_vnodes, mv_link) {
783fbc48e97SDavid Xu 		if (vd->mv_vnode->v_mount == mp) {
784fbc48e97SDavid Xu 			vhold(vd->mv_vnode);
785fbc48e97SDavid Xu 			sx_xunlock(&mqfs->mi_lock);
786fbc48e97SDavid Xu 
787fbc48e97SDavid Xu 			vgone(newvpp);
788fbc48e97SDavid Xu 			vput(newvpp);
789fbc48e97SDavid Xu 			goto found;
79061b9d89fSTor Egge 		}
791fbc48e97SDavid Xu 	}
792fbc48e97SDavid Xu 
793fbc48e97SDavid Xu 	*vpp = newvpp;
794fbc48e97SDavid Xu 
795655291f2SDavid Xu 	vd = uma_zalloc(mvdata_zone, M_WAITOK);
796655291f2SDavid Xu 	(*vpp)->v_data = vd;
797655291f2SDavid Xu 	vd->mv_vnode = *vpp;
798655291f2SDavid Xu 	vd->mv_node = pn;
799655291f2SDavid Xu 	TASK_INIT(&vd->mv_task, 0, do_recycle, *vpp);
800655291f2SDavid Xu 	LIST_INSERT_HEAD(&pn->mn_vnodes, vd, mv_link);
801655291f2SDavid Xu 	mqnode_addref(pn);
802655291f2SDavid Xu 	switch (pn->mn_type) {
803655291f2SDavid Xu 	case mqfstype_root:
804655291f2SDavid Xu 		(*vpp)->v_vflag = VV_ROOT;
805655291f2SDavid Xu 		/* fall through */
806655291f2SDavid Xu 	case mqfstype_dir:
807655291f2SDavid Xu 	case mqfstype_this:
808655291f2SDavid Xu 	case mqfstype_parent:
809655291f2SDavid Xu 		(*vpp)->v_type = VDIR;
810655291f2SDavid Xu 		break;
811655291f2SDavid Xu 	case mqfstype_file:
812655291f2SDavid Xu 		(*vpp)->v_type = VREG;
813655291f2SDavid Xu 		break;
814655291f2SDavid Xu 	case mqfstype_symlink:
815655291f2SDavid Xu 		(*vpp)->v_type = VLNK;
816655291f2SDavid Xu 		break;
817655291f2SDavid Xu 	case mqfstype_none:
818655291f2SDavid Xu 		KASSERT(0, ("mqfs_allocf called for null node\n"));
819655291f2SDavid Xu 	default:
820655291f2SDavid Xu 		panic("%s has unexpected type: %d", pn->mn_name, pn->mn_type);
821655291f2SDavid Xu 	}
822fbc48e97SDavid Xu 	sx_xunlock(&mqfs->mi_lock);
823655291f2SDavid Xu 	return (0);
824655291f2SDavid Xu }
825655291f2SDavid Xu 
826655291f2SDavid Xu /*
827655291f2SDavid Xu  * Search a directory entry
828655291f2SDavid Xu  */
829655291f2SDavid Xu static struct mqfs_node *
830adb023aeSJamie Gritton mqfs_search(struct mqfs_node *pd, const char *name, int len, struct ucred *cred)
831655291f2SDavid Xu {
832655291f2SDavid Xu 	struct mqfs_node *pn;
833adb023aeSJamie Gritton 	const void *pr_root;
834655291f2SDavid Xu 
835fbc48e97SDavid Xu 	sx_assert(&pd->mn_info->mi_lock, SX_LOCKED);
836adb023aeSJamie Gritton 	pr_root = cred->cr_prison->pr_root;
837655291f2SDavid Xu 	LIST_FOREACH(pn, &pd->mn_children, mn_sibling) {
838adb023aeSJamie Gritton 		/* Only match names within the same prison root directory */
839adb023aeSJamie Gritton 		if ((pn->mn_pr_root == NULL || pn->mn_pr_root == pr_root) &&
840adb023aeSJamie Gritton 		    strncmp(pn->mn_name, name, len) == 0 &&
8411cbae705SEd Schouten 		    pn->mn_name[len] == '\0')
842655291f2SDavid Xu 			return (pn);
843655291f2SDavid Xu 	}
844655291f2SDavid Xu 	return (NULL);
845655291f2SDavid Xu }
846655291f2SDavid Xu 
847655291f2SDavid Xu /*
848873fbcd7SRobert Watson  * Look up a file or directory.
849655291f2SDavid Xu  */
850655291f2SDavid Xu static int
851655291f2SDavid Xu mqfs_lookupx(struct vop_cachedlookup_args *ap)
852655291f2SDavid Xu {
853655291f2SDavid Xu 	struct componentname *cnp;
854655291f2SDavid Xu 	struct vnode *dvp, **vpp;
855655291f2SDavid Xu 	struct mqfs_node *pd;
856655291f2SDavid Xu 	struct mqfs_node *pn;
857fbc48e97SDavid Xu 	struct mqfs_info *mqfs;
858655291f2SDavid Xu 	int nameiop, flags, error, namelen;
859655291f2SDavid Xu 	char *pname;
860655291f2SDavid Xu 	struct thread *td;
861655291f2SDavid Xu 
862655291f2SDavid Xu 	cnp = ap->a_cnp;
863655291f2SDavid Xu 	vpp = ap->a_vpp;
864655291f2SDavid Xu 	dvp = ap->a_dvp;
865655291f2SDavid Xu 	pname = cnp->cn_nameptr;
866655291f2SDavid Xu 	namelen = cnp->cn_namelen;
867655291f2SDavid Xu 	td = cnp->cn_thread;
868655291f2SDavid Xu 	flags = cnp->cn_flags;
869655291f2SDavid Xu 	nameiop = cnp->cn_nameiop;
870655291f2SDavid Xu 	pd = VTON(dvp);
871655291f2SDavid Xu 	pn = NULL;
872fbc48e97SDavid Xu 	mqfs = pd->mn_info;
873655291f2SDavid Xu 	*vpp = NULLVP;
874655291f2SDavid Xu 
875655291f2SDavid Xu 	if (dvp->v_type != VDIR)
876655291f2SDavid Xu 		return (ENOTDIR);
877655291f2SDavid Xu 
878655291f2SDavid Xu 	error = VOP_ACCESS(dvp, VEXEC, cnp->cn_cred, cnp->cn_thread);
879655291f2SDavid Xu 	if (error)
880655291f2SDavid Xu 		return (error);
881655291f2SDavid Xu 
882655291f2SDavid Xu 	/* shortcut: check if the name is too long */
883655291f2SDavid Xu 	if (cnp->cn_namelen >= MQFS_NAMELEN)
884655291f2SDavid Xu 		return (ENOENT);
885655291f2SDavid Xu 
886655291f2SDavid Xu 	/* self */
887655291f2SDavid Xu 	if (namelen == 1 && pname[0] == '.') {
888655291f2SDavid Xu 		if ((flags & ISLASTCN) && nameiop != LOOKUP)
889655291f2SDavid Xu 			return (EINVAL);
890655291f2SDavid Xu 		pn = pd;
891655291f2SDavid Xu 		*vpp = dvp;
892655291f2SDavid Xu 		VREF(dvp);
893655291f2SDavid Xu 		return (0);
894655291f2SDavid Xu 	}
895655291f2SDavid Xu 
896655291f2SDavid Xu 	/* parent */
897655291f2SDavid Xu 	if (cnp->cn_flags & ISDOTDOT) {
898655291f2SDavid Xu 		if (dvp->v_vflag & VV_ROOT)
899655291f2SDavid Xu 			return (EIO);
900655291f2SDavid Xu 		if ((flags & ISLASTCN) && nameiop != LOOKUP)
901655291f2SDavid Xu 			return (EINVAL);
90222db15c0SAttilio Rao 		VOP_UNLOCK(dvp, 0);
903655291f2SDavid Xu 		KASSERT(pd->mn_parent, ("non-root directory has no parent"));
904655291f2SDavid Xu 		pn = pd->mn_parent;
905655291f2SDavid Xu 		error = mqfs_allocv(dvp->v_mount, vpp, pn);
906cb05b60aSAttilio Rao 		vn_lock(dvp, LK_EXCLUSIVE | LK_RETRY);
907655291f2SDavid Xu 		return (error);
908655291f2SDavid Xu 	}
909655291f2SDavid Xu 
910655291f2SDavid Xu 	/* named node */
911fbc48e97SDavid Xu 	sx_xlock(&mqfs->mi_lock);
912adb023aeSJamie Gritton 	pn = mqfs_search(pd, pname, namelen, cnp->cn_cred);
913fbc48e97SDavid Xu 	if (pn != NULL)
914fbc48e97SDavid Xu 		mqnode_addref(pn);
915fbc48e97SDavid Xu 	sx_xunlock(&mqfs->mi_lock);
916655291f2SDavid Xu 
917655291f2SDavid Xu 	/* found */
918655291f2SDavid Xu 	if (pn != NULL) {
919655291f2SDavid Xu 		/* DELETE */
920655291f2SDavid Xu 		if (nameiop == DELETE && (flags & ISLASTCN)) {
921655291f2SDavid Xu 			error = VOP_ACCESS(dvp, VWRITE, cnp->cn_cred, td);
922fbc48e97SDavid Xu 			if (error) {
923fbc48e97SDavid Xu 				mqnode_release(pn);
924655291f2SDavid Xu 				return (error);
925fbc48e97SDavid Xu 			}
926655291f2SDavid Xu 			if (*vpp == dvp) {
927655291f2SDavid Xu 				VREF(dvp);
928655291f2SDavid Xu 				*vpp = dvp;
929fbc48e97SDavid Xu 				mqnode_release(pn);
930655291f2SDavid Xu 				return (0);
931655291f2SDavid Xu 			}
932655291f2SDavid Xu 		}
933655291f2SDavid Xu 
934655291f2SDavid Xu 		/* allocate vnode */
935655291f2SDavid Xu 		error = mqfs_allocv(dvp->v_mount, vpp, pn);
936fbc48e97SDavid Xu 		mqnode_release(pn);
937655291f2SDavid Xu 		if (error == 0 && cnp->cn_flags & MAKEENTRY)
938655291f2SDavid Xu 			cache_enter(dvp, *vpp, cnp);
939655291f2SDavid Xu 		return (error);
940655291f2SDavid Xu 	}
941655291f2SDavid Xu 
942655291f2SDavid Xu 	/* not found */
943655291f2SDavid Xu 
944655291f2SDavid Xu 	/* will create a new entry in the directory ? */
945655291f2SDavid Xu 	if ((nameiop == CREATE || nameiop == RENAME) && (flags & LOCKPARENT)
946655291f2SDavid Xu 	    && (flags & ISLASTCN)) {
947655291f2SDavid Xu 		error = VOP_ACCESS(dvp, VWRITE, cnp->cn_cred, td);
948655291f2SDavid Xu 		if (error)
949655291f2SDavid Xu 			return (error);
950655291f2SDavid Xu 		cnp->cn_flags |= SAVENAME;
951655291f2SDavid Xu 		return (EJUSTRETURN);
952655291f2SDavid Xu 	}
953655291f2SDavid Xu 	return (ENOENT);
954655291f2SDavid Xu }
955655291f2SDavid Xu 
956655291f2SDavid Xu #if 0
957655291f2SDavid Xu struct vop_lookup_args {
958655291f2SDavid Xu 	struct vop_generic_args a_gen;
959655291f2SDavid Xu 	struct vnode *a_dvp;
960655291f2SDavid Xu 	struct vnode **a_vpp;
961655291f2SDavid Xu 	struct componentname *a_cnp;
962655291f2SDavid Xu };
963655291f2SDavid Xu #endif
964655291f2SDavid Xu 
965655291f2SDavid Xu /*
966655291f2SDavid Xu  * vnode lookup operation
967655291f2SDavid Xu  */
968655291f2SDavid Xu static int
969655291f2SDavid Xu mqfs_lookup(struct vop_cachedlookup_args *ap)
970655291f2SDavid Xu {
971655291f2SDavid Xu 	int rc;
972655291f2SDavid Xu 
973655291f2SDavid Xu 	rc = mqfs_lookupx(ap);
974655291f2SDavid Xu 	return (rc);
975655291f2SDavid Xu }
976655291f2SDavid Xu 
977655291f2SDavid Xu #if 0
978655291f2SDavid Xu struct vop_create_args {
979655291f2SDavid Xu 	struct vnode *a_dvp;
980655291f2SDavid Xu 	struct vnode **a_vpp;
981655291f2SDavid Xu 	struct componentname *a_cnp;
982655291f2SDavid Xu 	struct vattr *a_vap;
983655291f2SDavid Xu };
984655291f2SDavid Xu #endif
985655291f2SDavid Xu 
986655291f2SDavid Xu /*
987655291f2SDavid Xu  * vnode creation operation
988655291f2SDavid Xu  */
989655291f2SDavid Xu static int
990655291f2SDavid Xu mqfs_create(struct vop_create_args *ap)
991655291f2SDavid Xu {
992655291f2SDavid Xu 	struct mqfs_info *mqfs = VFSTOMQFS(ap->a_dvp->v_mount);
993655291f2SDavid Xu 	struct componentname *cnp = ap->a_cnp;
994655291f2SDavid Xu 	struct mqfs_node *pd;
995655291f2SDavid Xu 	struct mqfs_node *pn;
996655291f2SDavid Xu 	struct mqueue *mq;
997655291f2SDavid Xu 	int error;
998655291f2SDavid Xu 
999655291f2SDavid Xu 	pd = VTON(ap->a_dvp);
1000655291f2SDavid Xu 	if (pd->mn_type != mqfstype_root && pd->mn_type != mqfstype_dir)
1001655291f2SDavid Xu 		return (ENOTDIR);
1002655291f2SDavid Xu 	mq = mqueue_alloc(NULL);
1003655291f2SDavid Xu 	if (mq == NULL)
1004655291f2SDavid Xu 		return (EAGAIN);
1005655291f2SDavid Xu 	sx_xlock(&mqfs->mi_lock);
1006655291f2SDavid Xu 	if ((cnp->cn_flags & HASBUF) == 0)
1007655291f2SDavid Xu 		panic("%s: no name", __func__);
1008ba0360b1SDavid Xu 	pn = mqfs_create_file(pd, cnp->cn_nameptr, cnp->cn_namelen,
1009ba0360b1SDavid Xu 		cnp->cn_cred, ap->a_vap->va_mode);
1010fbc48e97SDavid Xu 	if (pn == NULL) {
1011fbc48e97SDavid Xu 		sx_xunlock(&mqfs->mi_lock);
1012ba0360b1SDavid Xu 		error = ENOSPC;
1013fbc48e97SDavid Xu 	} else {
1014fbc48e97SDavid Xu 		mqnode_addref(pn);
1015fbc48e97SDavid Xu 		sx_xunlock(&mqfs->mi_lock);
1016655291f2SDavid Xu 		error = mqfs_allocv(ap->a_dvp->v_mount, ap->a_vpp, pn);
1017fbc48e97SDavid Xu 		mqnode_release(pn);
1018ba0360b1SDavid Xu 		if (error)
1019ba0360b1SDavid Xu 			mqfs_destroy(pn);
1020ba0360b1SDavid Xu 		else
1021ba0360b1SDavid Xu 			pn->mn_data = mq;
1022ba0360b1SDavid Xu 	}
1023ba0360b1SDavid Xu 	if (error)
1024ba0360b1SDavid Xu 		mqueue_free(mq);
1025655291f2SDavid Xu 	return (error);
1026655291f2SDavid Xu }
1027655291f2SDavid Xu 
1028655291f2SDavid Xu /*
1029655291f2SDavid Xu  * Remove an entry
1030655291f2SDavid Xu  */
1031655291f2SDavid Xu static
1032655291f2SDavid Xu int do_unlink(struct mqfs_node *pn, struct ucred *ucred)
1033655291f2SDavid Xu {
1034655291f2SDavid Xu 	struct mqfs_node *parent;
1035655291f2SDavid Xu 	struct mqfs_vdata *vd;
1036655291f2SDavid Xu 	int error = 0;
1037655291f2SDavid Xu 
1038655291f2SDavid Xu 	sx_assert(&pn->mn_info->mi_lock, SX_LOCKED);
1039655291f2SDavid Xu 
1040655291f2SDavid Xu 	if (ucred->cr_uid != pn->mn_uid &&
104132f9753cSRobert Watson 	    (error = priv_check_cred(ucred, PRIV_MQ_ADMIN, 0)) != 0)
1042655291f2SDavid Xu 		error = EACCES;
1043655291f2SDavid Xu 	else if (!pn->mn_deleted) {
1044655291f2SDavid Xu 		parent = pn->mn_parent;
1045655291f2SDavid Xu 		pn->mn_parent = NULL;
1046655291f2SDavid Xu 		pn->mn_deleted = 1;
1047655291f2SDavid Xu 		LIST_REMOVE(pn, mn_sibling);
1048655291f2SDavid Xu 		LIST_FOREACH(vd, &pn->mn_vnodes, mv_link) {
1049655291f2SDavid Xu 			cache_purge(vd->mv_vnode);
1050655291f2SDavid Xu 			vhold(vd->mv_vnode);
1051655291f2SDavid Xu 			taskqueue_enqueue(taskqueue_thread, &vd->mv_task);
1052655291f2SDavid Xu 		}
1053655291f2SDavid Xu 		mqnode_release(pn);
1054655291f2SDavid Xu 		mqnode_release(parent);
1055655291f2SDavid Xu 	} else
1056655291f2SDavid Xu 		error = ENOENT;
1057655291f2SDavid Xu 	return (error);
1058655291f2SDavid Xu }
1059655291f2SDavid Xu 
1060655291f2SDavid Xu #if 0
1061655291f2SDavid Xu struct vop_remove_args {
1062655291f2SDavid Xu 	struct vnode *a_dvp;
1063655291f2SDavid Xu 	struct vnode *a_vp;
1064655291f2SDavid Xu 	struct componentname *a_cnp;
1065655291f2SDavid Xu };
1066655291f2SDavid Xu #endif
1067655291f2SDavid Xu 
1068655291f2SDavid Xu /*
1069655291f2SDavid Xu  * vnode removal operation
1070655291f2SDavid Xu  */
1071655291f2SDavid Xu static int
1072655291f2SDavid Xu mqfs_remove(struct vop_remove_args *ap)
1073655291f2SDavid Xu {
1074655291f2SDavid Xu 	struct mqfs_info *mqfs = VFSTOMQFS(ap->a_dvp->v_mount);
1075655291f2SDavid Xu 	struct mqfs_node *pn;
1076655291f2SDavid Xu 	int error;
1077655291f2SDavid Xu 
1078655291f2SDavid Xu 	if (ap->a_vp->v_type == VDIR)
1079655291f2SDavid Xu                 return (EPERM);
1080655291f2SDavid Xu 	pn = VTON(ap->a_vp);
1081655291f2SDavid Xu 	sx_xlock(&mqfs->mi_lock);
1082655291f2SDavid Xu 	error = do_unlink(pn, ap->a_cnp->cn_cred);
1083655291f2SDavid Xu 	sx_xunlock(&mqfs->mi_lock);
1084655291f2SDavid Xu 	return (error);
1085655291f2SDavid Xu }
1086655291f2SDavid Xu 
1087655291f2SDavid Xu #if 0
1088655291f2SDavid Xu struct vop_inactive_args {
1089655291f2SDavid Xu 	struct vnode *a_vp;
1090655291f2SDavid Xu 	struct thread *a_td;
1091655291f2SDavid Xu };
1092655291f2SDavid Xu #endif
1093655291f2SDavid Xu 
1094655291f2SDavid Xu static int
1095655291f2SDavid Xu mqfs_inactive(struct vop_inactive_args *ap)
1096655291f2SDavid Xu {
1097655291f2SDavid Xu 	struct mqfs_node *pn = VTON(ap->a_vp);
1098655291f2SDavid Xu 
1099655291f2SDavid Xu 	if (pn->mn_deleted)
1100af6e6b87SEdward Tomasz Napierala 		vrecycle(ap->a_vp);
1101655291f2SDavid Xu 	return (0);
1102655291f2SDavid Xu }
1103655291f2SDavid Xu 
1104655291f2SDavid Xu #if 0
1105655291f2SDavid Xu struct vop_reclaim_args {
1106655291f2SDavid Xu 	struct vop_generic_args a_gen;
1107655291f2SDavid Xu 	struct vnode *a_vp;
1108655291f2SDavid Xu 	struct thread *a_td;
1109655291f2SDavid Xu };
1110655291f2SDavid Xu #endif
1111655291f2SDavid Xu 
1112655291f2SDavid Xu static int
1113655291f2SDavid Xu mqfs_reclaim(struct vop_reclaim_args *ap)
1114655291f2SDavid Xu {
1115655291f2SDavid Xu 	struct mqfs_info *mqfs = VFSTOMQFS(ap->a_vp->v_mount);
1116655291f2SDavid Xu 	struct vnode *vp = ap->a_vp;
1117655291f2SDavid Xu 	struct mqfs_node *pn;
1118655291f2SDavid Xu 	struct mqfs_vdata *vd;
1119655291f2SDavid Xu 
1120655291f2SDavid Xu 	vd = vp->v_data;
1121655291f2SDavid Xu 	pn = vd->mv_node;
1122655291f2SDavid Xu 	sx_xlock(&mqfs->mi_lock);
1123655291f2SDavid Xu 	vp->v_data = NULL;
1124655291f2SDavid Xu 	LIST_REMOVE(vd, mv_link);
1125655291f2SDavid Xu 	uma_zfree(mvdata_zone, vd);
1126655291f2SDavid Xu 	mqnode_release(pn);
1127655291f2SDavid Xu 	sx_xunlock(&mqfs->mi_lock);
1128655291f2SDavid Xu 	return (0);
1129655291f2SDavid Xu }
1130655291f2SDavid Xu 
1131655291f2SDavid Xu #if 0
1132655291f2SDavid Xu struct vop_open_args {
1133655291f2SDavid Xu 	struct vop_generic_args a_gen;
1134655291f2SDavid Xu 	struct vnode *a_vp;
1135655291f2SDavid Xu 	int a_mode;
1136655291f2SDavid Xu 	struct ucred *a_cred;
1137655291f2SDavid Xu 	struct thread *a_td;
1138d6da6408SKonstantin Belousov 	struct file *a_fp;
1139655291f2SDavid Xu };
1140655291f2SDavid Xu #endif
1141655291f2SDavid Xu 
1142655291f2SDavid Xu static int
1143655291f2SDavid Xu mqfs_open(struct vop_open_args *ap)
1144655291f2SDavid Xu {
1145655291f2SDavid Xu 	return (0);
1146655291f2SDavid Xu }
1147655291f2SDavid Xu 
1148655291f2SDavid Xu #if 0
1149655291f2SDavid Xu struct vop_close_args {
1150655291f2SDavid Xu 	struct vop_generic_args a_gen;
1151655291f2SDavid Xu 	struct vnode *a_vp;
1152655291f2SDavid Xu 	int a_fflag;
1153655291f2SDavid Xu 	struct ucred *a_cred;
1154655291f2SDavid Xu 	struct thread *a_td;
1155655291f2SDavid Xu };
1156655291f2SDavid Xu #endif
1157655291f2SDavid Xu 
1158655291f2SDavid Xu static int
1159655291f2SDavid Xu mqfs_close(struct vop_close_args *ap)
1160655291f2SDavid Xu {
1161655291f2SDavid Xu 	return (0);
1162655291f2SDavid Xu }
1163655291f2SDavid Xu 
1164655291f2SDavid Xu #if 0
1165655291f2SDavid Xu struct vop_access_args {
1166655291f2SDavid Xu 	struct vop_generic_args a_gen;
1167655291f2SDavid Xu 	struct vnode *a_vp;
116815bc6b2bSEdward Tomasz Napierala 	accmode_t a_accmode;
1169655291f2SDavid Xu 	struct ucred *a_cred;
1170655291f2SDavid Xu 	struct thread *a_td;
1171655291f2SDavid Xu };
1172655291f2SDavid Xu #endif
1173655291f2SDavid Xu 
1174655291f2SDavid Xu /*
1175655291f2SDavid Xu  * Verify permissions
1176655291f2SDavid Xu  */
1177655291f2SDavid Xu static int
1178655291f2SDavid Xu mqfs_access(struct vop_access_args *ap)
1179655291f2SDavid Xu {
1180655291f2SDavid Xu 	struct vnode *vp = ap->a_vp;
1181655291f2SDavid Xu 	struct vattr vattr;
1182655291f2SDavid Xu 	int error;
1183655291f2SDavid Xu 
11840359a12eSAttilio Rao 	error = VOP_GETATTR(vp, &vattr, ap->a_cred);
1185655291f2SDavid Xu 	if (error)
1186655291f2SDavid Xu 		return (error);
1187655291f2SDavid Xu 	error = vaccess(vp->v_type, vattr.va_mode, vattr.va_uid,
118815bc6b2bSEdward Tomasz Napierala 	    vattr.va_gid, ap->a_accmode, ap->a_cred, NULL);
1189655291f2SDavid Xu 	return (error);
1190655291f2SDavid Xu }
1191655291f2SDavid Xu 
1192655291f2SDavid Xu #if 0
1193655291f2SDavid Xu struct vop_getattr_args {
1194655291f2SDavid Xu 	struct vop_generic_args a_gen;
1195655291f2SDavid Xu 	struct vnode *a_vp;
1196655291f2SDavid Xu 	struct vattr *a_vap;
1197655291f2SDavid Xu 	struct ucred *a_cred;
1198655291f2SDavid Xu };
1199655291f2SDavid Xu #endif
1200655291f2SDavid Xu 
1201655291f2SDavid Xu /*
1202655291f2SDavid Xu  * Get file attributes
1203655291f2SDavid Xu  */
1204655291f2SDavid Xu static int
1205655291f2SDavid Xu mqfs_getattr(struct vop_getattr_args *ap)
1206655291f2SDavid Xu {
1207655291f2SDavid Xu 	struct vnode *vp = ap->a_vp;
1208655291f2SDavid Xu 	struct mqfs_node *pn = VTON(vp);
1209655291f2SDavid Xu 	struct vattr *vap = ap->a_vap;
1210655291f2SDavid Xu 	int error = 0;
1211655291f2SDavid Xu 
1212655291f2SDavid Xu 	vap->va_type = vp->v_type;
1213655291f2SDavid Xu 	vap->va_mode = pn->mn_mode;
1214655291f2SDavid Xu 	vap->va_nlink = 1;
1215655291f2SDavid Xu 	vap->va_uid = pn->mn_uid;
1216655291f2SDavid Xu 	vap->va_gid = pn->mn_gid;
1217655291f2SDavid Xu 	vap->va_fsid = vp->v_mount->mnt_stat.f_fsid.val[0];
1218655291f2SDavid Xu 	vap->va_fileid = pn->mn_fileno;
1219655291f2SDavid Xu 	vap->va_size = 0;
1220655291f2SDavid Xu 	vap->va_blocksize = PAGE_SIZE;
1221655291f2SDavid Xu 	vap->va_bytes = vap->va_size = 0;
1222655291f2SDavid Xu 	vap->va_atime = pn->mn_atime;
1223655291f2SDavid Xu 	vap->va_mtime = pn->mn_mtime;
1224655291f2SDavid Xu 	vap->va_ctime = pn->mn_ctime;
1225655291f2SDavid Xu 	vap->va_birthtime = pn->mn_birth;
1226655291f2SDavid Xu 	vap->va_gen = 0;
1227655291f2SDavid Xu 	vap->va_flags = 0;
12284c5a20e3SKonstantin Belousov 	vap->va_rdev = NODEV;
1229655291f2SDavid Xu 	vap->va_bytes = 0;
1230655291f2SDavid Xu 	vap->va_filerev = 0;
1231655291f2SDavid Xu 	return (error);
1232655291f2SDavid Xu }
1233655291f2SDavid Xu 
1234655291f2SDavid Xu #if 0
1235655291f2SDavid Xu struct vop_setattr_args {
1236655291f2SDavid Xu 	struct vop_generic_args a_gen;
1237655291f2SDavid Xu 	struct vnode *a_vp;
1238655291f2SDavid Xu 	struct vattr *a_vap;
1239655291f2SDavid Xu 	struct ucred *a_cred;
1240655291f2SDavid Xu };
1241655291f2SDavid Xu #endif
1242655291f2SDavid Xu /*
1243655291f2SDavid Xu  * Set attributes
1244655291f2SDavid Xu  */
1245655291f2SDavid Xu static int
1246655291f2SDavid Xu mqfs_setattr(struct vop_setattr_args *ap)
1247655291f2SDavid Xu {
1248655291f2SDavid Xu 	struct mqfs_node *pn;
1249655291f2SDavid Xu 	struct vattr *vap;
1250655291f2SDavid Xu 	struct vnode *vp;
12510359a12eSAttilio Rao 	struct thread *td;
1252655291f2SDavid Xu 	int c, error;
1253655291f2SDavid Xu 	uid_t uid;
1254655291f2SDavid Xu 	gid_t gid;
1255655291f2SDavid Xu 
12560359a12eSAttilio Rao 	td = curthread;
1257655291f2SDavid Xu 	vap = ap->a_vap;
1258655291f2SDavid Xu 	vp = ap->a_vp;
1259655291f2SDavid Xu 	if ((vap->va_type != VNON) ||
1260655291f2SDavid Xu 	    (vap->va_nlink != VNOVAL) ||
1261655291f2SDavid Xu 	    (vap->va_fsid != VNOVAL) ||
1262655291f2SDavid Xu 	    (vap->va_fileid != VNOVAL) ||
1263655291f2SDavid Xu 	    (vap->va_blocksize != VNOVAL) ||
1264655291f2SDavid Xu 	    (vap->va_flags != VNOVAL && vap->va_flags != 0) ||
1265655291f2SDavid Xu 	    (vap->va_rdev != VNOVAL) ||
1266655291f2SDavid Xu 	    ((int)vap->va_bytes != VNOVAL) ||
1267655291f2SDavid Xu 	    (vap->va_gen != VNOVAL)) {
1268655291f2SDavid Xu 		return (EINVAL);
1269655291f2SDavid Xu 	}
1270655291f2SDavid Xu 
1271655291f2SDavid Xu 	pn = VTON(vp);
1272655291f2SDavid Xu 
1273655291f2SDavid Xu 	error = c = 0;
1274655291f2SDavid Xu 	if (vap->va_uid == (uid_t)VNOVAL)
1275655291f2SDavid Xu 		uid = pn->mn_uid;
1276655291f2SDavid Xu 	else
1277655291f2SDavid Xu 		uid = vap->va_uid;
1278655291f2SDavid Xu 	if (vap->va_gid == (gid_t)VNOVAL)
1279655291f2SDavid Xu 		gid = pn->mn_gid;
1280655291f2SDavid Xu 	else
1281655291f2SDavid Xu 		gid = vap->va_gid;
1282655291f2SDavid Xu 
1283655291f2SDavid Xu 	if (uid != pn->mn_uid || gid != pn->mn_gid) {
1284655291f2SDavid Xu 		/*
1285655291f2SDavid Xu 		 * To modify the ownership of a file, must possess VADMIN
1286655291f2SDavid Xu 		 * for that file.
1287655291f2SDavid Xu 		 */
12880359a12eSAttilio Rao 		if ((error = VOP_ACCESS(vp, VADMIN, ap->a_cred, td)))
1289655291f2SDavid Xu 			return (error);
1290acd3428bSRobert Watson 
1291acd3428bSRobert Watson 		/*
1292acd3428bSRobert Watson 		 * XXXRW: Why is there a privilege check here: shouldn't the
1293acd3428bSRobert Watson 		 * check in VOP_ACCESS() be enough?  Also, are the group bits
1294acd3428bSRobert Watson 		 * below definitely right?
1295acd3428bSRobert Watson 		 */
1296655291f2SDavid Xu 		if (((ap->a_cred->cr_uid != pn->mn_uid) || uid != pn->mn_uid ||
1297655291f2SDavid Xu 		    (gid != pn->mn_gid && !groupmember(gid, ap->a_cred))) &&
12980359a12eSAttilio Rao 		    (error = priv_check(td, PRIV_MQ_ADMIN)) != 0)
1299655291f2SDavid Xu 			return (error);
1300655291f2SDavid Xu 		pn->mn_uid = uid;
1301655291f2SDavid Xu 		pn->mn_gid = gid;
1302655291f2SDavid Xu 		c = 1;
1303655291f2SDavid Xu 	}
1304655291f2SDavid Xu 
1305655291f2SDavid Xu 	if (vap->va_mode != (mode_t)VNOVAL) {
1306655291f2SDavid Xu 		if ((ap->a_cred->cr_uid != pn->mn_uid) &&
13070359a12eSAttilio Rao 		    (error = priv_check(td, PRIV_MQ_ADMIN)))
1308655291f2SDavid Xu 			return (error);
1309655291f2SDavid Xu 		pn->mn_mode = vap->va_mode;
1310655291f2SDavid Xu 		c = 1;
1311655291f2SDavid Xu 	}
1312655291f2SDavid Xu 
1313655291f2SDavid Xu 	if (vap->va_atime.tv_sec != VNOVAL || vap->va_mtime.tv_sec != VNOVAL) {
1314655291f2SDavid Xu 		/* See the comment in ufs_vnops::ufs_setattr(). */
13150359a12eSAttilio Rao 		if ((error = VOP_ACCESS(vp, VADMIN, ap->a_cred, td)) &&
1316655291f2SDavid Xu 		    ((vap->va_vaflags & VA_UTIMES_NULL) == 0 ||
13170359a12eSAttilio Rao 		    (error = VOP_ACCESS(vp, VWRITE, ap->a_cred, td))))
1318655291f2SDavid Xu 			return (error);
1319655291f2SDavid Xu 		if (vap->va_atime.tv_sec != VNOVAL) {
1320655291f2SDavid Xu 			pn->mn_atime = vap->va_atime;
1321655291f2SDavid Xu 		}
1322655291f2SDavid Xu 		if (vap->va_mtime.tv_sec != VNOVAL) {
1323655291f2SDavid Xu 			pn->mn_mtime = vap->va_mtime;
1324655291f2SDavid Xu 		}
1325655291f2SDavid Xu 		c = 1;
1326655291f2SDavid Xu 	}
1327655291f2SDavid Xu 	if (c) {
1328655291f2SDavid Xu 		vfs_timestamp(&pn->mn_ctime);
1329655291f2SDavid Xu 	}
1330655291f2SDavid Xu 	return (0);
1331655291f2SDavid Xu }
1332655291f2SDavid Xu 
1333655291f2SDavid Xu #if 0
1334655291f2SDavid Xu struct vop_read_args {
1335655291f2SDavid Xu 	struct vop_generic_args a_gen;
1336655291f2SDavid Xu 	struct vnode *a_vp;
1337655291f2SDavid Xu 	struct uio *a_uio;
1338655291f2SDavid Xu 	int a_ioflag;
1339655291f2SDavid Xu 	struct ucred *a_cred;
1340655291f2SDavid Xu };
1341655291f2SDavid Xu #endif
1342655291f2SDavid Xu 
1343655291f2SDavid Xu /*
1344655291f2SDavid Xu  * Read from a file
1345655291f2SDavid Xu  */
1346655291f2SDavid Xu static int
1347655291f2SDavid Xu mqfs_read(struct vop_read_args *ap)
1348655291f2SDavid Xu {
1349655291f2SDavid Xu 	char buf[80];
1350655291f2SDavid Xu 	struct vnode *vp = ap->a_vp;
1351655291f2SDavid Xu 	struct uio *uio = ap->a_uio;
1352655291f2SDavid Xu 	struct mqfs_node *pn;
1353655291f2SDavid Xu 	struct mqueue *mq;
1354655291f2SDavid Xu 	int len, error;
1355655291f2SDavid Xu 
1356655291f2SDavid Xu 	if (vp->v_type != VREG)
1357655291f2SDavid Xu 		return (EINVAL);
1358655291f2SDavid Xu 
1359655291f2SDavid Xu 	pn = VTON(vp);
1360655291f2SDavid Xu 	mq = VTOMQ(vp);
1361655291f2SDavid Xu 	snprintf(buf, sizeof(buf),
1362655291f2SDavid Xu 		"QSIZE:%-10ld MAXMSG:%-10ld CURMSG:%-10ld MSGSIZE:%-10ld\n",
1363655291f2SDavid Xu 		mq->mq_totalbytes,
1364655291f2SDavid Xu 		mq->mq_maxmsg,
1365655291f2SDavid Xu 		mq->mq_curmsgs,
1366655291f2SDavid Xu 		mq->mq_msgsize);
1367655291f2SDavid Xu 	buf[sizeof(buf)-1] = '\0';
1368655291f2SDavid Xu 	len = strlen(buf);
1369655291f2SDavid Xu 	error = uiomove_frombuf(buf, len, uio);
1370655291f2SDavid Xu 	return (error);
1371655291f2SDavid Xu }
1372655291f2SDavid Xu 
1373655291f2SDavid Xu #if 0
1374655291f2SDavid Xu struct vop_readdir_args {
1375655291f2SDavid Xu 	struct vop_generic_args a_gen;
1376655291f2SDavid Xu 	struct vnode *a_vp;
1377655291f2SDavid Xu 	struct uio *a_uio;
1378655291f2SDavid Xu 	struct ucred *a_cred;
1379655291f2SDavid Xu 	int *a_eofflag;
1380655291f2SDavid Xu 	int *a_ncookies;
1381655291f2SDavid Xu 	u_long **a_cookies;
1382655291f2SDavid Xu };
1383655291f2SDavid Xu #endif
1384655291f2SDavid Xu 
1385655291f2SDavid Xu /*
1386655291f2SDavid Xu  * Return directory entries.
1387655291f2SDavid Xu  */
1388655291f2SDavid Xu static int
1389655291f2SDavid Xu mqfs_readdir(struct vop_readdir_args *ap)
1390655291f2SDavid Xu {
1391655291f2SDavid Xu 	struct vnode *vp;
1392655291f2SDavid Xu 	struct mqfs_info *mi;
1393655291f2SDavid Xu 	struct mqfs_node *pd;
1394655291f2SDavid Xu 	struct mqfs_node *pn;
1395655291f2SDavid Xu 	struct dirent entry;
1396655291f2SDavid Xu 	struct uio *uio;
1397adb023aeSJamie Gritton 	const void *pr_root;
1398655291f2SDavid Xu 	int *tmp_ncookies = NULL;
1399655291f2SDavid Xu 	off_t offset;
1400655291f2SDavid Xu 	int error, i;
1401655291f2SDavid Xu 
1402655291f2SDavid Xu 	vp = ap->a_vp;
1403655291f2SDavid Xu 	mi = VFSTOMQFS(vp->v_mount);
1404655291f2SDavid Xu 	pd = VTON(vp);
1405655291f2SDavid Xu 	uio = ap->a_uio;
1406655291f2SDavid Xu 
1407655291f2SDavid Xu 	if (vp->v_type != VDIR)
1408655291f2SDavid Xu 		return (ENOTDIR);
1409655291f2SDavid Xu 
1410655291f2SDavid Xu 	if (uio->uio_offset < 0)
1411655291f2SDavid Xu 		return (EINVAL);
1412655291f2SDavid Xu 
1413655291f2SDavid Xu 	if (ap->a_ncookies != NULL) {
1414655291f2SDavid Xu 		tmp_ncookies = ap->a_ncookies;
1415655291f2SDavid Xu 		*ap->a_ncookies = 0;
1416655291f2SDavid Xu 		ap->a_ncookies = NULL;
1417655291f2SDavid Xu         }
1418655291f2SDavid Xu 
1419655291f2SDavid Xu 	error = 0;
1420655291f2SDavid Xu 	offset = 0;
1421655291f2SDavid Xu 
1422adb023aeSJamie Gritton 	pr_root = ap->a_cred->cr_prison->pr_root;
1423655291f2SDavid Xu 	sx_xlock(&mi->mi_lock);
1424655291f2SDavid Xu 
1425655291f2SDavid Xu 	LIST_FOREACH(pn, &pd->mn_children, mn_sibling) {
1426655291f2SDavid Xu 		entry.d_reclen = sizeof(entry);
1427*44c16975SJamie Gritton 
1428adb023aeSJamie Gritton 		/*
1429adb023aeSJamie Gritton 		 * Only show names within the same prison root directory
1430adb023aeSJamie Gritton 		 * (or not associated with a prison, e.g. "." and "..").
1431adb023aeSJamie Gritton 		 */
1432adb023aeSJamie Gritton 		if (pn->mn_pr_root != NULL && pn->mn_pr_root != pr_root)
1433adb023aeSJamie Gritton 			continue;
1434655291f2SDavid Xu 		if (!pn->mn_fileno)
1435655291f2SDavid Xu 			mqfs_fileno_alloc(mi, pn);
1436655291f2SDavid Xu 		entry.d_fileno = pn->mn_fileno;
1437655291f2SDavid Xu 		for (i = 0; i < MQFS_NAMELEN - 1 && pn->mn_name[i] != '\0'; ++i)
1438655291f2SDavid Xu 			entry.d_name[i] = pn->mn_name[i];
1439655291f2SDavid Xu 		entry.d_name[i] = 0;
1440655291f2SDavid Xu 		entry.d_namlen = i;
1441655291f2SDavid Xu 		switch (pn->mn_type) {
1442655291f2SDavid Xu 		case mqfstype_root:
1443655291f2SDavid Xu 		case mqfstype_dir:
1444655291f2SDavid Xu 		case mqfstype_this:
1445655291f2SDavid Xu 		case mqfstype_parent:
1446655291f2SDavid Xu 			entry.d_type = DT_DIR;
1447655291f2SDavid Xu 			break;
1448655291f2SDavid Xu 		case mqfstype_file:
1449655291f2SDavid Xu 			entry.d_type = DT_REG;
1450655291f2SDavid Xu 			break;
1451655291f2SDavid Xu 		case mqfstype_symlink:
1452655291f2SDavid Xu 			entry.d_type = DT_LNK;
1453655291f2SDavid Xu 			break;
1454655291f2SDavid Xu 		default:
1455655291f2SDavid Xu 			panic("%s has unexpected node type: %d", pn->mn_name,
1456655291f2SDavid Xu 				pn->mn_type);
1457655291f2SDavid Xu 		}
1458655291f2SDavid Xu 		if (entry.d_reclen > uio->uio_resid)
1459655291f2SDavid Xu                         break;
1460655291f2SDavid Xu 		if (offset >= uio->uio_offset) {
1461655291f2SDavid Xu 			error = vfs_read_dirent(ap, &entry, offset);
1462655291f2SDavid Xu                         if (error)
1463655291f2SDavid Xu                                 break;
1464655291f2SDavid Xu                 }
1465655291f2SDavid Xu                 offset += entry.d_reclen;
1466655291f2SDavid Xu 	}
1467655291f2SDavid Xu 	sx_xunlock(&mi->mi_lock);
1468655291f2SDavid Xu 
1469655291f2SDavid Xu 	uio->uio_offset = offset;
1470655291f2SDavid Xu 
1471655291f2SDavid Xu 	if (tmp_ncookies != NULL)
1472655291f2SDavid Xu 		ap->a_ncookies = tmp_ncookies;
1473655291f2SDavid Xu 
1474655291f2SDavid Xu 	return (error);
1475655291f2SDavid Xu }
1476655291f2SDavid Xu 
1477655291f2SDavid Xu #ifdef notyet
1478655291f2SDavid Xu 
1479655291f2SDavid Xu #if 0
1480655291f2SDavid Xu struct vop_mkdir_args {
1481655291f2SDavid Xu 	struct vnode *a_dvp;
1482655291f2SDavid Xu 	struvt vnode **a_vpp;
1483655291f2SDavid Xu 	struvt componentname *a_cnp;
1484655291f2SDavid Xu 	struct vattr *a_vap;
1485655291f2SDavid Xu };
1486655291f2SDavid Xu #endif
1487655291f2SDavid Xu 
1488655291f2SDavid Xu /*
1489655291f2SDavid Xu  * Create a directory.
1490655291f2SDavid Xu  */
1491655291f2SDavid Xu static int
1492655291f2SDavid Xu mqfs_mkdir(struct vop_mkdir_args *ap)
1493655291f2SDavid Xu {
1494655291f2SDavid Xu 	struct mqfs_info *mqfs = VFSTOMQFS(ap->a_dvp->v_mount);
1495655291f2SDavid Xu 	struct componentname *cnp = ap->a_cnp;
1496655291f2SDavid Xu 	struct mqfs_node *pd = VTON(ap->a_dvp);
1497655291f2SDavid Xu 	struct mqfs_node *pn;
1498655291f2SDavid Xu 	int error;
1499655291f2SDavid Xu 
1500655291f2SDavid Xu 	if (pd->mn_type != mqfstype_root && pd->mn_type != mqfstype_dir)
1501655291f2SDavid Xu 		return (ENOTDIR);
1502655291f2SDavid Xu 	sx_xlock(&mqfs->mi_lock);
1503655291f2SDavid Xu 	if ((cnp->cn_flags & HASBUF) == 0)
1504655291f2SDavid Xu 		panic("%s: no name", __func__);
1505ba0360b1SDavid Xu 	pn = mqfs_create_dir(pd, cnp->cn_nameptr, cnp->cn_namelen,
1506ba0360b1SDavid Xu 		ap->a_vap->cn_cred, ap->a_vap->va_mode);
1507fbc48e97SDavid Xu 	if (pn != NULL)
1508fbc48e97SDavid Xu 		mqnode_addref(pn);
1509655291f2SDavid Xu 	sx_xunlock(&mqfs->mi_lock);
1510fbc48e97SDavid Xu 	if (pn == NULL) {
1511fbc48e97SDavid Xu 		error = ENOSPC;
1512fbc48e97SDavid Xu 	} else {
1513fbc48e97SDavid Xu 		error = mqfs_allocv(ap->a_dvp->v_mount, ap->a_vpp, pn);
1514fbc48e97SDavid Xu 		mqnode_release(pn);
1515fbc48e97SDavid Xu 	}
1516655291f2SDavid Xu 	return (error);
1517655291f2SDavid Xu }
1518655291f2SDavid Xu 
1519655291f2SDavid Xu #if 0
1520655291f2SDavid Xu struct vop_rmdir_args {
1521655291f2SDavid Xu 	struct vnode *a_dvp;
1522655291f2SDavid Xu 	struct vnode *a_vp;
1523655291f2SDavid Xu 	struct componentname *a_cnp;
1524655291f2SDavid Xu };
1525655291f2SDavid Xu #endif
1526655291f2SDavid Xu 
1527655291f2SDavid Xu /*
1528655291f2SDavid Xu  * Remove a directory.
1529655291f2SDavid Xu  */
1530655291f2SDavid Xu static int
1531655291f2SDavid Xu mqfs_rmdir(struct vop_rmdir_args *ap)
1532655291f2SDavid Xu {
1533655291f2SDavid Xu 	struct mqfs_info *mqfs = VFSTOMQFS(ap->a_dvp->v_mount);
1534655291f2SDavid Xu 	struct mqfs_node *pn = VTON(ap->a_vp);
1535655291f2SDavid Xu 	struct mqfs_node *pt;
1536655291f2SDavid Xu 
1537655291f2SDavid Xu 	if (pn->mn_type != mqfstype_dir)
1538655291f2SDavid Xu 		return (ENOTDIR);
1539655291f2SDavid Xu 
1540655291f2SDavid Xu 	sx_xlock(&mqfs->mi_lock);
1541655291f2SDavid Xu 	if (pn->mn_deleted) {
1542655291f2SDavid Xu 		sx_xunlock(&mqfs->mi_lock);
1543655291f2SDavid Xu 		return (ENOENT);
1544655291f2SDavid Xu 	}
1545655291f2SDavid Xu 
1546655291f2SDavid Xu 	pt = LIST_FIRST(&pn->mn_children);
1547655291f2SDavid Xu 	pt = LIST_NEXT(pt, mn_sibling);
1548655291f2SDavid Xu 	pt = LIST_NEXT(pt, mn_sibling);
1549655291f2SDavid Xu 	if (pt != NULL) {
1550655291f2SDavid Xu 		sx_xunlock(&mqfs->mi_lock);
1551655291f2SDavid Xu 		return (ENOTEMPTY);
1552655291f2SDavid Xu 	}
1553655291f2SDavid Xu 	pt = pn->mn_parent;
1554655291f2SDavid Xu 	pn->mn_parent = NULL;
1555655291f2SDavid Xu 	pn->mn_deleted = 1;
1556655291f2SDavid Xu 	LIST_REMOVE(pn, mn_sibling);
1557655291f2SDavid Xu 	mqnode_release(pn);
1558655291f2SDavid Xu 	mqnode_release(pt);
1559655291f2SDavid Xu 	sx_xunlock(&mqfs->mi_lock);
1560655291f2SDavid Xu 	cache_purge(ap->a_vp);
1561655291f2SDavid Xu 	return (0);
1562655291f2SDavid Xu }
1563655291f2SDavid Xu 
1564655291f2SDavid Xu #endif /* notyet */
1565655291f2SDavid Xu 
1566adb023aeSJamie Gritton 
1567adb023aeSJamie Gritton /*
1568adb023aeSJamie Gritton  * Set a destructor task with the prison's root
1569adb023aeSJamie Gritton  */
1570adb023aeSJamie Gritton static int
1571adb023aeSJamie Gritton mqfs_prison_create(void *obj, void *data __unused)
1572adb023aeSJamie Gritton {
1573adb023aeSJamie Gritton 	struct prison *pr = obj;
1574adb023aeSJamie Gritton 	struct mqfs_osd *mo;
1575adb023aeSJamie Gritton 	void *rsv;
1576adb023aeSJamie Gritton 
1577adb023aeSJamie Gritton 	if (pr->pr_root == pr->pr_parent->pr_root)
1578adb023aeSJamie Gritton 		return(0);
1579adb023aeSJamie Gritton 
1580adb023aeSJamie Gritton 	mo = malloc(sizeof(struct mqfs_osd), M_PRISON, M_WAITOK);
1581adb023aeSJamie Gritton 	rsv = osd_reserve(mqfs_osd_jail_slot);
1582adb023aeSJamie Gritton 	TASK_INIT(&mo->mo_task, 0, mqfs_prison_remove_task, mo);
1583adb023aeSJamie Gritton 	mtx_lock(&pr->pr_mtx);
1584adb023aeSJamie Gritton 	mo->mo_pr_root = pr->pr_root;
1585adb023aeSJamie Gritton 	(void)osd_jail_set_reserved(pr, mqfs_osd_jail_slot, rsv, mo);
1586adb023aeSJamie Gritton 	mtx_unlock(&pr->pr_mtx);
1587adb023aeSJamie Gritton 	return (0);
1588adb023aeSJamie Gritton }
1589adb023aeSJamie Gritton 
1590adb023aeSJamie Gritton /*
1591adb023aeSJamie Gritton  * Queue the task for after jail/OSD locks are released
1592adb023aeSJamie Gritton  */
1593adb023aeSJamie Gritton static void
1594adb023aeSJamie Gritton mqfs_prison_destructor(void *data)
1595adb023aeSJamie Gritton {
1596adb023aeSJamie Gritton 	struct mqfs_osd *mo = data;
1597adb023aeSJamie Gritton 
1598adb023aeSJamie Gritton 	if (mqfs_osd_jail_slot != 0)
1599adb023aeSJamie Gritton 		taskqueue_enqueue(taskqueue_thread, &mo->mo_task);
1600adb023aeSJamie Gritton 	else
1601adb023aeSJamie Gritton 		free(mo, M_PRISON);
1602adb023aeSJamie Gritton }
1603adb023aeSJamie Gritton 
1604adb023aeSJamie Gritton /*
1605adb023aeSJamie Gritton  * See if this prison root is obsolete, and clean up associated queues if it is
1606adb023aeSJamie Gritton  */
1607adb023aeSJamie Gritton static void
1608adb023aeSJamie Gritton mqfs_prison_remove_task(void *context, int pending)
1609adb023aeSJamie Gritton {
1610adb023aeSJamie Gritton 	struct mqfs_osd *mo = context;
1611adb023aeSJamie Gritton 	struct mqfs_node *pn, *tpn;
1612adb023aeSJamie Gritton 	const struct prison *pr;
1613adb023aeSJamie Gritton 	const void *pr_root;
1614adb023aeSJamie Gritton 	int found;
1615adb023aeSJamie Gritton 
1616adb023aeSJamie Gritton 	pr_root = mo->mo_pr_root;
1617adb023aeSJamie Gritton 	found = 0;
1618adb023aeSJamie Gritton 	sx_slock(&allprison_lock);
1619adb023aeSJamie Gritton 	TAILQ_FOREACH(pr, &allprison, pr_list) {
1620adb023aeSJamie Gritton 		if (pr->pr_root == pr_root)
1621adb023aeSJamie Gritton 			found = 1;
1622adb023aeSJamie Gritton 	}
1623adb023aeSJamie Gritton 	sx_sunlock(&allprison_lock);
1624adb023aeSJamie Gritton 	if (!found) {
1625adb023aeSJamie Gritton 		/*
1626adb023aeSJamie Gritton 		 * No jails are rooted in this directory anymore,
1627adb023aeSJamie Gritton 		 * so no queues should be either.
1628adb023aeSJamie Gritton 		 */
1629adb023aeSJamie Gritton 		sx_xlock(&mqfs_data.mi_lock);
1630adb023aeSJamie Gritton 		LIST_FOREACH_SAFE(pn, &mqfs_data.mi_root->mn_children,
1631adb023aeSJamie Gritton 		    mn_sibling, tpn) {
1632adb023aeSJamie Gritton 			if (pn->mn_pr_root == pr_root)
1633adb023aeSJamie Gritton 				(void)do_unlink(pn, curthread->td_ucred);
1634adb023aeSJamie Gritton 		}
1635adb023aeSJamie Gritton 		sx_xunlock(&mqfs_data.mi_lock);
1636adb023aeSJamie Gritton 	}
1637adb023aeSJamie Gritton 	free(mo, M_PRISON);
1638adb023aeSJamie Gritton }
1639adb023aeSJamie Gritton 
1640adb023aeSJamie Gritton 
1641655291f2SDavid Xu /*
1642655291f2SDavid Xu  * Allocate a message queue
1643655291f2SDavid Xu  */
1644655291f2SDavid Xu static struct mqueue *
1645655291f2SDavid Xu mqueue_alloc(const struct mq_attr *attr)
1646655291f2SDavid Xu {
1647655291f2SDavid Xu 	struct mqueue *mq;
1648655291f2SDavid Xu 
1649655291f2SDavid Xu 	if (curmq >= maxmq)
1650655291f2SDavid Xu 		return (NULL);
1651655291f2SDavid Xu 	mq = uma_zalloc(mqueue_zone, M_WAITOK | M_ZERO);
1652655291f2SDavid Xu 	TAILQ_INIT(&mq->mq_msgq);
1653655291f2SDavid Xu 	if (attr != NULL) {
1654655291f2SDavid Xu 		mq->mq_maxmsg = attr->mq_maxmsg;
1655655291f2SDavid Xu 		mq->mq_msgsize = attr->mq_msgsize;
1656655291f2SDavid Xu 	} else {
1657655291f2SDavid Xu 		mq->mq_maxmsg = default_maxmsg;
1658655291f2SDavid Xu 		mq->mq_msgsize = default_msgsize;
1659655291f2SDavid Xu 	}
1660b042e976SDavid Xu 	mtx_init(&mq->mq_mutex, "mqueue lock", NULL, MTX_DEF);
1661d8b0556cSKonstantin Belousov 	knlist_init_mtx(&mq->mq_rsel.si_note, &mq->mq_mutex);
1662d8b0556cSKonstantin Belousov 	knlist_init_mtx(&mq->mq_wsel.si_note, &mq->mq_mutex);
1663655291f2SDavid Xu 	atomic_add_int(&curmq, 1);
1664655291f2SDavid Xu 	return (mq);
1665655291f2SDavid Xu }
1666655291f2SDavid Xu 
1667655291f2SDavid Xu /*
1668655291f2SDavid Xu  * Destroy a message queue
1669655291f2SDavid Xu  */
1670655291f2SDavid Xu static void
1671655291f2SDavid Xu mqueue_free(struct mqueue *mq)
1672655291f2SDavid Xu {
1673655291f2SDavid Xu 	struct mqueue_msg *msg;
1674655291f2SDavid Xu 
1675655291f2SDavid Xu 	while ((msg = TAILQ_FIRST(&mq->mq_msgq)) != NULL) {
1676655291f2SDavid Xu 		TAILQ_REMOVE(&mq->mq_msgq, msg, msg_link);
16771ede983cSDag-Erling Smørgrav 		free(msg, M_MQUEUEDATA);
1678655291f2SDavid Xu 	}
1679655291f2SDavid Xu 
1680655291f2SDavid Xu 	mtx_destroy(&mq->mq_mutex);
16816aba400aSAttilio Rao 	seldrain(&mq->mq_rsel);
16826aba400aSAttilio Rao 	seldrain(&mq->mq_wsel);
1683655291f2SDavid Xu 	knlist_destroy(&mq->mq_rsel.si_note);
1684655291f2SDavid Xu 	knlist_destroy(&mq->mq_wsel.si_note);
1685655291f2SDavid Xu 	uma_zfree(mqueue_zone, mq);
1686655291f2SDavid Xu 	atomic_add_int(&curmq, -1);
1687655291f2SDavid Xu }
1688655291f2SDavid Xu 
1689655291f2SDavid Xu /*
1690655291f2SDavid Xu  * Load a message from user space
1691655291f2SDavid Xu  */
1692655291f2SDavid Xu static struct mqueue_msg *
1693655291f2SDavid Xu mqueue_loadmsg(const char *msg_ptr, size_t msg_size, int msg_prio)
1694655291f2SDavid Xu {
1695655291f2SDavid Xu 	struct mqueue_msg *msg;
1696655291f2SDavid Xu 	size_t len;
1697655291f2SDavid Xu 	int error;
1698655291f2SDavid Xu 
1699655291f2SDavid Xu 	len = sizeof(struct mqueue_msg) + msg_size;
17001ede983cSDag-Erling Smørgrav 	msg = malloc(len, M_MQUEUEDATA, M_WAITOK);
1701655291f2SDavid Xu 	error = copyin(msg_ptr, ((char *)msg) + sizeof(struct mqueue_msg),
1702655291f2SDavid Xu 	    msg_size);
1703655291f2SDavid Xu 	if (error) {
17041ede983cSDag-Erling Smørgrav 		free(msg, M_MQUEUEDATA);
1705655291f2SDavid Xu 		msg = NULL;
1706655291f2SDavid Xu 	} else {
1707655291f2SDavid Xu 		msg->msg_size = msg_size;
1708655291f2SDavid Xu 		msg->msg_prio = msg_prio;
1709655291f2SDavid Xu 	}
1710655291f2SDavid Xu 	return (msg);
1711655291f2SDavid Xu }
1712655291f2SDavid Xu 
1713655291f2SDavid Xu /*
1714655291f2SDavid Xu  * Save a message to user space
1715655291f2SDavid Xu  */
1716655291f2SDavid Xu static int
1717655291f2SDavid Xu mqueue_savemsg(struct mqueue_msg *msg, char *msg_ptr, int *msg_prio)
1718655291f2SDavid Xu {
1719655291f2SDavid Xu 	int error;
1720655291f2SDavid Xu 
1721655291f2SDavid Xu 	error = copyout(((char *)msg) + sizeof(*msg), msg_ptr,
1722655291f2SDavid Xu 		msg->msg_size);
1723b2f92ef9SDavid Xu 	if (error == 0 && msg_prio != NULL)
1724655291f2SDavid Xu 		error = copyout(&msg->msg_prio, msg_prio, sizeof(int));
1725655291f2SDavid Xu 	return (error);
1726655291f2SDavid Xu }
1727655291f2SDavid Xu 
1728655291f2SDavid Xu /*
1729655291f2SDavid Xu  * Free a message's memory
1730655291f2SDavid Xu  */
1731655291f2SDavid Xu static __inline void
1732655291f2SDavid Xu mqueue_freemsg(struct mqueue_msg *msg)
1733655291f2SDavid Xu {
17341ede983cSDag-Erling Smørgrav 	free(msg, M_MQUEUEDATA);
1735655291f2SDavid Xu }
1736655291f2SDavid Xu 
1737655291f2SDavid Xu /*
1738655291f2SDavid Xu  * Send a message. if waitok is false, thread will not be
1739655291f2SDavid Xu  * blocked if there is no data in queue, otherwise, absolute
1740655291f2SDavid Xu  * time will be checked.
1741655291f2SDavid Xu  */
1742655291f2SDavid Xu int
1743655291f2SDavid Xu mqueue_send(struct mqueue *mq, const char *msg_ptr,
1744655291f2SDavid Xu 	size_t msg_len, unsigned msg_prio, int waitok,
1745655291f2SDavid Xu 	const struct timespec *abs_timeout)
1746655291f2SDavid Xu {
1747655291f2SDavid Xu 	struct mqueue_msg *msg;
1748afde2b65SKonstantin Belousov 	struct timespec ts, ts2;
1749655291f2SDavid Xu 	struct timeval tv;
1750655291f2SDavid Xu 	int error;
1751655291f2SDavid Xu 
1752a6de716dSDavid Xu 	if (msg_prio >= MQ_PRIO_MAX)
1753a6de716dSDavid Xu 		return (EINVAL);
1754655291f2SDavid Xu 	if (msg_len > mq->mq_msgsize)
1755655291f2SDavid Xu 		return (EMSGSIZE);
1756655291f2SDavid Xu 	msg = mqueue_loadmsg(msg_ptr, msg_len, msg_prio);
1757655291f2SDavid Xu 	if (msg == NULL)
1758655291f2SDavid Xu 		return (EFAULT);
1759655291f2SDavid Xu 
1760655291f2SDavid Xu 	/* O_NONBLOCK case */
1761655291f2SDavid Xu 	if (!waitok) {
1762655291f2SDavid Xu 		error = _mqueue_send(mq, msg, -1);
1763655291f2SDavid Xu 		if (error)
1764655291f2SDavid Xu 			goto bad;
1765655291f2SDavid Xu 		return (0);
1766655291f2SDavid Xu 	}
1767655291f2SDavid Xu 
1768655291f2SDavid Xu 	/* we allow a null timeout (wait forever) */
1769655291f2SDavid Xu 	if (abs_timeout == NULL) {
1770655291f2SDavid Xu 		error = _mqueue_send(mq, msg, 0);
1771655291f2SDavid Xu 		if (error)
1772655291f2SDavid Xu 			goto bad;
1773655291f2SDavid Xu 		return (0);
1774655291f2SDavid Xu 	}
1775655291f2SDavid Xu 
1776655291f2SDavid Xu 	/* send it before checking time */
1777655291f2SDavid Xu 	error = _mqueue_send(mq, msg, -1);
1778655291f2SDavid Xu 	if (error == 0)
1779655291f2SDavid Xu 		return (0);
1780655291f2SDavid Xu 
1781655291f2SDavid Xu 	if (error != EAGAIN)
1782655291f2SDavid Xu 		goto bad;
1783655291f2SDavid Xu 
1784afde2b65SKonstantin Belousov 	if (abs_timeout->tv_nsec >= 1000000000 || abs_timeout->tv_nsec < 0) {
1785655291f2SDavid Xu 		error = EINVAL;
1786655291f2SDavid Xu 		goto bad;
1787655291f2SDavid Xu 	}
1788655291f2SDavid Xu 	for (;;) {
1789afde2b65SKonstantin Belousov 		ts2 = *abs_timeout;
1790a6de716dSDavid Xu 		getnanotime(&ts);
1791655291f2SDavid Xu 		timespecsub(&ts2, &ts);
1792655291f2SDavid Xu 		if (ts2.tv_sec < 0 || (ts2.tv_sec == 0 && ts2.tv_nsec <= 0)) {
1793655291f2SDavid Xu 			error = ETIMEDOUT;
1794655291f2SDavid Xu 			break;
1795655291f2SDavid Xu 		}
1796655291f2SDavid Xu 		TIMESPEC_TO_TIMEVAL(&tv, &ts2);
1797655291f2SDavid Xu 		error = _mqueue_send(mq, msg, tvtohz(&tv));
1798655291f2SDavid Xu 		if (error != ETIMEDOUT)
1799655291f2SDavid Xu 			break;
1800655291f2SDavid Xu 	}
1801655291f2SDavid Xu 	if (error == 0)
1802655291f2SDavid Xu 		return (0);
1803655291f2SDavid Xu bad:
1804655291f2SDavid Xu 	mqueue_freemsg(msg);
1805655291f2SDavid Xu 	return (error);
1806655291f2SDavid Xu }
1807655291f2SDavid Xu 
1808655291f2SDavid Xu /*
1809655291f2SDavid Xu  * Common routine to send a message
1810655291f2SDavid Xu  */
1811655291f2SDavid Xu static int
1812655291f2SDavid Xu _mqueue_send(struct mqueue *mq, struct mqueue_msg *msg, int timo)
1813655291f2SDavid Xu {
1814655291f2SDavid Xu 	struct mqueue_msg *msg2;
1815655291f2SDavid Xu 	int error = 0;
1816655291f2SDavid Xu 
1817655291f2SDavid Xu 	mtx_lock(&mq->mq_mutex);
1818655291f2SDavid Xu 	while (mq->mq_curmsgs >= mq->mq_maxmsg && error == 0) {
1819655291f2SDavid Xu 		if (timo < 0) {
1820655291f2SDavid Xu 			mtx_unlock(&mq->mq_mutex);
1821655291f2SDavid Xu 			return (EAGAIN);
1822655291f2SDavid Xu 		}
1823655291f2SDavid Xu 		mq->mq_senders++;
1824655291f2SDavid Xu 		error = msleep(&mq->mq_senders, &mq->mq_mutex,
18250f180a7cSJohn Baldwin 			    PCATCH, "mqsend", timo);
1826655291f2SDavid Xu 		mq->mq_senders--;
1827655291f2SDavid Xu 		if (error == EAGAIN)
1828655291f2SDavid Xu 			error = ETIMEDOUT;
1829655291f2SDavid Xu 	}
1830655291f2SDavid Xu 	if (mq->mq_curmsgs >= mq->mq_maxmsg) {
1831655291f2SDavid Xu 		mtx_unlock(&mq->mq_mutex);
1832655291f2SDavid Xu 		return (error);
1833655291f2SDavid Xu 	}
1834655291f2SDavid Xu 	error = 0;
1835655291f2SDavid Xu 	if (TAILQ_EMPTY(&mq->mq_msgq)) {
1836655291f2SDavid Xu 		TAILQ_INSERT_HEAD(&mq->mq_msgq, msg, msg_link);
1837655291f2SDavid Xu 	} else {
1838655291f2SDavid Xu 		if (msg->msg_prio <= TAILQ_LAST(&mq->mq_msgq, msgq)->msg_prio) {
1839655291f2SDavid Xu 			TAILQ_INSERT_TAIL(&mq->mq_msgq, msg, msg_link);
1840655291f2SDavid Xu 		} else {
1841655291f2SDavid Xu 			TAILQ_FOREACH(msg2, &mq->mq_msgq, msg_link) {
1842655291f2SDavid Xu 				if (msg2->msg_prio < msg->msg_prio)
1843655291f2SDavid Xu 					break;
1844655291f2SDavid Xu 			}
1845655291f2SDavid Xu 			TAILQ_INSERT_BEFORE(msg2, msg, msg_link);
1846655291f2SDavid Xu 		}
1847655291f2SDavid Xu 	}
1848655291f2SDavid Xu 	mq->mq_curmsgs++;
1849655291f2SDavid Xu 	mq->mq_totalbytes += msg->msg_size;
1850655291f2SDavid Xu 	if (mq->mq_receivers)
1851655291f2SDavid Xu 		wakeup_one(&mq->mq_receivers);
1852655291f2SDavid Xu 	else if (mq->mq_notifier != NULL)
1853655291f2SDavid Xu 		mqueue_send_notification(mq);
1854655291f2SDavid Xu 	if (mq->mq_flags & MQ_RSEL) {
1855655291f2SDavid Xu 		mq->mq_flags &= ~MQ_RSEL;
185603f70aecSDavid Xu 		selwakeup(&mq->mq_rsel);
1857655291f2SDavid Xu 	}
1858655291f2SDavid Xu 	KNOTE_LOCKED(&mq->mq_rsel.si_note, 0);
1859655291f2SDavid Xu 	mtx_unlock(&mq->mq_mutex);
1860655291f2SDavid Xu 	return (0);
1861655291f2SDavid Xu }
1862655291f2SDavid Xu 
1863655291f2SDavid Xu /*
1864655291f2SDavid Xu  * Send realtime a signal to process which registered itself
1865655291f2SDavid Xu  * successfully by mq_notify.
1866655291f2SDavid Xu  */
1867655291f2SDavid Xu static void
1868655291f2SDavid Xu mqueue_send_notification(struct mqueue *mq)
1869655291f2SDavid Xu {
1870b2f92ef9SDavid Xu 	struct mqueue_notifier *nt;
1871cf7d9a8cSDavid Xu 	struct thread *td;
1872b2f92ef9SDavid Xu 	struct proc *p;
1873cf7d9a8cSDavid Xu 	int error;
1874655291f2SDavid Xu 
1875655291f2SDavid Xu 	mtx_assert(&mq->mq_mutex, MA_OWNED);
1876b2f92ef9SDavid Xu 	nt = mq->mq_notifier;
1877052ea11cSDavid Xu 	if (nt->nt_sigev.sigev_notify != SIGEV_NONE) {
1878b2f92ef9SDavid Xu 		p = nt->nt_proc;
1879cf7d9a8cSDavid Xu 		error = sigev_findtd(p, &nt->nt_sigev, &td);
1880cf7d9a8cSDavid Xu 		if (error) {
1881cf7d9a8cSDavid Xu 			mq->mq_notifier = NULL;
1882cf7d9a8cSDavid Xu 			return;
1883cf7d9a8cSDavid Xu 		}
1884cf7d9a8cSDavid Xu 		if (!KSI_ONQ(&nt->nt_ksi)) {
1885cf7d9a8cSDavid Xu 			ksiginfo_set_sigev(&nt->nt_ksi, &nt->nt_sigev);
1886cf7d9a8cSDavid Xu 			tdsendsignal(p, td, nt->nt_ksi.ksi_signo, &nt->nt_ksi);
1887cf7d9a8cSDavid Xu 		}
1888b2f92ef9SDavid Xu 		PROC_UNLOCK(p);
1889052ea11cSDavid Xu 	}
1890655291f2SDavid Xu 	mq->mq_notifier = NULL;
1891655291f2SDavid Xu }
1892655291f2SDavid Xu 
1893655291f2SDavid Xu /*
1894655291f2SDavid Xu  * Get a message. if waitok is false, thread will not be
1895655291f2SDavid Xu  * blocked if there is no data in queue, otherwise, absolute
1896655291f2SDavid Xu  * time will be checked.
1897655291f2SDavid Xu  */
1898655291f2SDavid Xu int
1899655291f2SDavid Xu mqueue_receive(struct mqueue *mq, char *msg_ptr,
1900655291f2SDavid Xu 	size_t msg_len, unsigned *msg_prio, int waitok,
1901655291f2SDavid Xu 	const struct timespec *abs_timeout)
1902655291f2SDavid Xu {
1903655291f2SDavid Xu 	struct mqueue_msg *msg;
1904afde2b65SKonstantin Belousov 	struct timespec ts, ts2;
1905655291f2SDavid Xu 	struct timeval tv;
1906655291f2SDavid Xu 	int error;
1907655291f2SDavid Xu 
1908655291f2SDavid Xu 	if (msg_len < mq->mq_msgsize)
1909655291f2SDavid Xu 		return (EMSGSIZE);
1910655291f2SDavid Xu 
1911655291f2SDavid Xu 	/* O_NONBLOCK case */
1912655291f2SDavid Xu 	if (!waitok) {
1913655291f2SDavid Xu 		error = _mqueue_recv(mq, &msg, -1);
1914655291f2SDavid Xu 		if (error)
1915655291f2SDavid Xu 			return (error);
1916655291f2SDavid Xu 		goto received;
1917655291f2SDavid Xu 	}
1918655291f2SDavid Xu 
1919655291f2SDavid Xu 	/* we allow a null timeout (wait forever). */
1920655291f2SDavid Xu 	if (abs_timeout == NULL) {
1921655291f2SDavid Xu 		error = _mqueue_recv(mq, &msg, 0);
1922655291f2SDavid Xu 		if (error)
1923655291f2SDavid Xu 			return (error);
1924655291f2SDavid Xu 		goto received;
1925655291f2SDavid Xu 	}
1926655291f2SDavid Xu 
1927655291f2SDavid Xu 	/* try to get a message before checking time */
1928655291f2SDavid Xu 	error = _mqueue_recv(mq, &msg, -1);
1929655291f2SDavid Xu 	if (error == 0)
1930655291f2SDavid Xu 		goto received;
1931655291f2SDavid Xu 
1932655291f2SDavid Xu 	if (error != EAGAIN)
1933655291f2SDavid Xu 		return (error);
1934655291f2SDavid Xu 
1935afde2b65SKonstantin Belousov 	if (abs_timeout->tv_nsec >= 1000000000 || abs_timeout->tv_nsec < 0) {
1936655291f2SDavid Xu 		error = EINVAL;
1937655291f2SDavid Xu 		return (error);
1938655291f2SDavid Xu 	}
1939655291f2SDavid Xu 
1940655291f2SDavid Xu 	for (;;) {
1941afde2b65SKonstantin Belousov 		ts2 = *abs_timeout;
1942a6de716dSDavid Xu 		getnanotime(&ts);
1943655291f2SDavid Xu 		timespecsub(&ts2, &ts);
1944655291f2SDavid Xu 		if (ts2.tv_sec < 0 || (ts2.tv_sec == 0 && ts2.tv_nsec <= 0)) {
1945655291f2SDavid Xu 			error = ETIMEDOUT;
1946655291f2SDavid Xu 			return (error);
1947655291f2SDavid Xu 		}
1948655291f2SDavid Xu 		TIMESPEC_TO_TIMEVAL(&tv, &ts2);
1949655291f2SDavid Xu 		error = _mqueue_recv(mq, &msg, tvtohz(&tv));
1950655291f2SDavid Xu 		if (error == 0)
1951655291f2SDavid Xu 			break;
1952655291f2SDavid Xu 		if (error != ETIMEDOUT)
1953655291f2SDavid Xu 			return (error);
1954655291f2SDavid Xu 	}
1955655291f2SDavid Xu 
1956655291f2SDavid Xu received:
1957655291f2SDavid Xu 	error = mqueue_savemsg(msg, msg_ptr, msg_prio);
1958655291f2SDavid Xu 	if (error == 0) {
1959655291f2SDavid Xu 		curthread->td_retval[0] = msg->msg_size;
1960655291f2SDavid Xu 		curthread->td_retval[1] = 0;
1961655291f2SDavid Xu 	}
1962655291f2SDavid Xu 	mqueue_freemsg(msg);
1963655291f2SDavid Xu 	return (error);
1964655291f2SDavid Xu }
1965655291f2SDavid Xu 
1966655291f2SDavid Xu /*
1967655291f2SDavid Xu  * Common routine to receive a message
1968655291f2SDavid Xu  */
1969655291f2SDavid Xu static int
1970655291f2SDavid Xu _mqueue_recv(struct mqueue *mq, struct mqueue_msg **msg, int timo)
1971655291f2SDavid Xu {
1972655291f2SDavid Xu 	int error = 0;
1973655291f2SDavid Xu 
1974655291f2SDavid Xu 	mtx_lock(&mq->mq_mutex);
1975655291f2SDavid Xu 	while ((*msg = TAILQ_FIRST(&mq->mq_msgq)) == NULL && error == 0) {
1976655291f2SDavid Xu 		if (timo < 0) {
1977655291f2SDavid Xu 			mtx_unlock(&mq->mq_mutex);
1978655291f2SDavid Xu 			return (EAGAIN);
1979655291f2SDavid Xu 		}
1980655291f2SDavid Xu 		mq->mq_receivers++;
1981655291f2SDavid Xu 		error = msleep(&mq->mq_receivers, &mq->mq_mutex,
19820f180a7cSJohn Baldwin 			    PCATCH, "mqrecv", timo);
1983655291f2SDavid Xu 		mq->mq_receivers--;
1984655291f2SDavid Xu 		if (error == EAGAIN)
1985655291f2SDavid Xu 			error = ETIMEDOUT;
1986655291f2SDavid Xu 	}
1987655291f2SDavid Xu 	if (*msg != NULL) {
1988655291f2SDavid Xu 		error = 0;
1989655291f2SDavid Xu 		TAILQ_REMOVE(&mq->mq_msgq, *msg, msg_link);
1990655291f2SDavid Xu 		mq->mq_curmsgs--;
1991655291f2SDavid Xu 		mq->mq_totalbytes -= (*msg)->msg_size;
1992655291f2SDavid Xu 		if (mq->mq_senders)
1993655291f2SDavid Xu 			wakeup_one(&mq->mq_senders);
1994655291f2SDavid Xu 		if (mq->mq_flags & MQ_WSEL) {
1995655291f2SDavid Xu 			mq->mq_flags &= ~MQ_WSEL;
199603f70aecSDavid Xu 			selwakeup(&mq->mq_wsel);
1997655291f2SDavid Xu 		}
1998655291f2SDavid Xu 		KNOTE_LOCKED(&mq->mq_wsel.si_note, 0);
1999655291f2SDavid Xu 	}
2000655291f2SDavid Xu 	if (mq->mq_notifier != NULL && mq->mq_receivers == 0 &&
2001655291f2SDavid Xu 	    !TAILQ_EMPTY(&mq->mq_msgq)) {
2002655291f2SDavid Xu 		mqueue_send_notification(mq);
2003655291f2SDavid Xu 	}
2004655291f2SDavid Xu 	mtx_unlock(&mq->mq_mutex);
2005655291f2SDavid Xu 	return (error);
2006655291f2SDavid Xu }
2007655291f2SDavid Xu 
2008b2f92ef9SDavid Xu static __inline struct mqueue_notifier *
2009b2f92ef9SDavid Xu notifier_alloc(void)
2010655291f2SDavid Xu {
2011b2f92ef9SDavid Xu 	return (uma_zalloc(mqnoti_zone, M_WAITOK | M_ZERO));
2012655291f2SDavid Xu }
2013655291f2SDavid Xu 
2014655291f2SDavid Xu static __inline void
2015b2f92ef9SDavid Xu notifier_free(struct mqueue_notifier *p)
2016655291f2SDavid Xu {
2017b2f92ef9SDavid Xu 	uma_zfree(mqnoti_zone, p);
2018b2f92ef9SDavid Xu }
2019b2f92ef9SDavid Xu 
2020b2f92ef9SDavid Xu static struct mqueue_notifier *
2021b2f92ef9SDavid Xu notifier_search(struct proc *p, int fd)
2022b2f92ef9SDavid Xu {
2023b2f92ef9SDavid Xu 	struct mqueue_notifier *nt;
2024b2f92ef9SDavid Xu 
2025b2f92ef9SDavid Xu 	LIST_FOREACH(nt, &p->p_mqnotifier, nt_link) {
20269da8a32aSDavid Xu 		if (nt->nt_ksi.ksi_mqd == fd)
2027b2f92ef9SDavid Xu 			break;
2028b2f92ef9SDavid Xu 	}
2029b2f92ef9SDavid Xu 	return (nt);
2030b2f92ef9SDavid Xu }
2031b2f92ef9SDavid Xu 
2032102178d0SDavid Xu static __inline void
2033b2f92ef9SDavid Xu notifier_insert(struct proc *p, struct mqueue_notifier *nt)
2034b2f92ef9SDavid Xu {
2035b2f92ef9SDavid Xu 	LIST_INSERT_HEAD(&p->p_mqnotifier, nt, nt_link);
2036b2f92ef9SDavid Xu }
2037b2f92ef9SDavid Xu 
2038102178d0SDavid Xu static __inline void
2039b2f92ef9SDavid Xu notifier_delete(struct proc *p, struct mqueue_notifier *nt)
2040b2f92ef9SDavid Xu {
2041b2f92ef9SDavid Xu 	LIST_REMOVE(nt, nt_link);
2042b2f92ef9SDavid Xu 	notifier_free(nt);
2043b2f92ef9SDavid Xu }
2044b2f92ef9SDavid Xu 
2045b2f92ef9SDavid Xu static void
2046b2f92ef9SDavid Xu notifier_remove(struct proc *p, struct mqueue *mq, int fd)
2047b2f92ef9SDavid Xu {
2048b2f92ef9SDavid Xu 	struct mqueue_notifier *nt;
2049b2f92ef9SDavid Xu 
2050b2f92ef9SDavid Xu 	mtx_assert(&mq->mq_mutex, MA_OWNED);
2051b2f92ef9SDavid Xu 	PROC_LOCK(p);
2052b2f92ef9SDavid Xu 	nt = notifier_search(p, fd);
2053b2f92ef9SDavid Xu 	if (nt != NULL) {
2054b2f92ef9SDavid Xu 		if (mq->mq_notifier == nt)
2055b2f92ef9SDavid Xu 			mq->mq_notifier = NULL;
2056b2f92ef9SDavid Xu 		sigqueue_take(&nt->nt_ksi);
2057b2f92ef9SDavid Xu 		notifier_delete(p, nt);
2058b2f92ef9SDavid Xu 	}
2059b2f92ef9SDavid Xu 	PROC_UNLOCK(p);
2060655291f2SDavid Xu }
2061655291f2SDavid Xu 
2062afde2b65SKonstantin Belousov static int
2063afde2b65SKonstantin Belousov kern_kmq_open(struct thread *td, const char *upath, int flags, mode_t mode,
2064afde2b65SKonstantin Belousov     const struct mq_attr *attr)
2065655291f2SDavid Xu {
2066655291f2SDavid Xu 	char path[MQFS_NAMELEN + 1];
2067655291f2SDavid Xu 	struct mqfs_node *pn;
2068655291f2SDavid Xu 	struct filedesc *fdp;
2069655291f2SDavid Xu 	struct file *fp;
2070655291f2SDavid Xu 	struct mqueue *mq;
2071afde2b65SKonstantin Belousov 	int fd, error, len, cmode;
2072655291f2SDavid Xu 
2073655291f2SDavid Xu 	fdp = td->td_proc->p_fd;
2074afde2b65SKonstantin Belousov 	cmode = (((mode & ~fdp->fd_cmask) & ALLPERMS) & ~S_ISTXT);
2075f72b11a4SDavid Xu 	mq = NULL;
2076afde2b65SKonstantin Belousov 	if ((flags & O_CREAT) != 0 && attr != NULL) {
2077afde2b65SKonstantin Belousov 		if (attr->mq_maxmsg <= 0 || attr->mq_maxmsg > maxmsg)
2078655291f2SDavid Xu 			return (EINVAL);
2079afde2b65SKonstantin Belousov 		if (attr->mq_msgsize <= 0 || attr->mq_msgsize > maxmsgsize)
2080655291f2SDavid Xu 			return (EINVAL);
2081afde2b65SKonstantin Belousov 	}
2082655291f2SDavid Xu 
2083afde2b65SKonstantin Belousov 	error = copyinstr(upath, path, MQFS_NAMELEN + 1, NULL);
2084655291f2SDavid Xu         if (error)
2085655291f2SDavid Xu 		return (error);
2086655291f2SDavid Xu 
2087655291f2SDavid Xu 	/*
2088655291f2SDavid Xu 	 * The first character of name must be a slash  (/) character
2089655291f2SDavid Xu 	 * and the remaining characters of name cannot include any slash
2090655291f2SDavid Xu 	 * characters.
2091655291f2SDavid Xu 	 */
2092655291f2SDavid Xu 	len = strlen(path);
2093dc15eac0SEd Schouten 	if (len < 2 || path[0] != '/' || strchr(path + 1, '/') != NULL)
2094655291f2SDavid Xu 		return (EINVAL);
2095655291f2SDavid Xu 
2096b68cf25fSJilles Tjoelker 	error = falloc(td, &fp, &fd, O_CLOEXEC);
2097655291f2SDavid Xu 	if (error)
2098655291f2SDavid Xu 		return (error);
2099655291f2SDavid Xu 
2100655291f2SDavid Xu 	sx_xlock(&mqfs_data.mi_lock);
2101adb023aeSJamie Gritton 	pn = mqfs_search(mqfs_data.mi_root, path + 1, len - 1, td->td_ucred);
2102655291f2SDavid Xu 	if (pn == NULL) {
2103655291f2SDavid Xu 		if (!(flags & O_CREAT)) {
2104655291f2SDavid Xu 			error = ENOENT;
2105655291f2SDavid Xu 		} else {
2106afde2b65SKonstantin Belousov 			mq = mqueue_alloc(attr);
2107b2f92ef9SDavid Xu 			if (mq == NULL) {
2108b2f92ef9SDavid Xu 				error = ENFILE;
2109b2f92ef9SDavid Xu 			} else {
2110655291f2SDavid Xu 				pn = mqfs_create_file(mqfs_data.mi_root,
2111ba0360b1SDavid Xu 				         path + 1, len - 1, td->td_ucred,
2112ba0360b1SDavid Xu 					 cmode);
2113655291f2SDavid Xu 				if (pn == NULL) {
2114655291f2SDavid Xu 					error = ENOSPC;
2115b2f92ef9SDavid Xu 					mqueue_free(mq);
2116b2f92ef9SDavid Xu 				}
2117655291f2SDavid Xu 			}
2118655291f2SDavid Xu 		}
2119655291f2SDavid Xu 
2120655291f2SDavid Xu 		if (error == 0) {
2121655291f2SDavid Xu 			pn->mn_data = mq;
2122655291f2SDavid Xu 		}
2123655291f2SDavid Xu 	} else {
2124655291f2SDavid Xu 		if ((flags & (O_CREAT | O_EXCL)) == (O_CREAT | O_EXCL)) {
2125655291f2SDavid Xu 			error = EEXIST;
2126655291f2SDavid Xu 		} else {
212715bc6b2bSEdward Tomasz Napierala 			accmode_t accmode = 0;
2128655291f2SDavid Xu 
2129655291f2SDavid Xu 			if (flags & FREAD)
213015bc6b2bSEdward Tomasz Napierala 				accmode |= VREAD;
2131655291f2SDavid Xu 			if (flags & FWRITE)
213215bc6b2bSEdward Tomasz Napierala 				accmode |= VWRITE;
2133655291f2SDavid Xu 			error = vaccess(VREG, pn->mn_mode, pn->mn_uid,
213415bc6b2bSEdward Tomasz Napierala 				    pn->mn_gid, accmode, td->td_ucred, NULL);
2135655291f2SDavid Xu 		}
2136655291f2SDavid Xu 	}
2137655291f2SDavid Xu 
2138655291f2SDavid Xu 	if (error) {
2139655291f2SDavid Xu 		sx_xunlock(&mqfs_data.mi_lock);
214090f54cbfSMateusz Guzik 		fdclose(td, fp, fd);
2141655291f2SDavid Xu 		fdrop(fp, td);
2142655291f2SDavid Xu 		return (error);
2143655291f2SDavid Xu 	}
2144655291f2SDavid Xu 
2145655291f2SDavid Xu 	mqnode_addref(pn);
2146655291f2SDavid Xu 	sx_xunlock(&mqfs_data.mi_lock);
2147655291f2SDavid Xu 
2148397c19d1SJeff Roberson 	finit(fp, flags & (FREAD | FWRITE | O_NONBLOCK), DTYPE_MQUEUE, pn,
2149397c19d1SJeff Roberson 	    &mqueueops);
2150655291f2SDavid Xu 
2151655291f2SDavid Xu 	td->td_retval[0] = fd;
2152655291f2SDavid Xu 	fdrop(fp, td);
2153655291f2SDavid Xu 	return (0);
2154655291f2SDavid Xu }
2155655291f2SDavid Xu 
2156655291f2SDavid Xu /*
2157afde2b65SKonstantin Belousov  * Syscall to open a message queue.
2158afde2b65SKonstantin Belousov  */
2159afde2b65SKonstantin Belousov int
21608451d0ddSKip Macy sys_kmq_open(struct thread *td, struct kmq_open_args *uap)
2161afde2b65SKonstantin Belousov {
2162afde2b65SKonstantin Belousov 	struct mq_attr attr;
2163afde2b65SKonstantin Belousov 	int flags, error;
2164afde2b65SKonstantin Belousov 
21656fbdb9f4SJilles Tjoelker 	if ((uap->flags & O_ACCMODE) == O_ACCMODE || uap->flags & O_EXEC)
2166afde2b65SKonstantin Belousov 		return (EINVAL);
2167afde2b65SKonstantin Belousov 	flags = FFLAGS(uap->flags);
2168afde2b65SKonstantin Belousov 	if ((flags & O_CREAT) != 0 && uap->attr != NULL) {
2169afde2b65SKonstantin Belousov 		error = copyin(uap->attr, &attr, sizeof(attr));
2170afde2b65SKonstantin Belousov 		if (error)
2171afde2b65SKonstantin Belousov 			return (error);
2172afde2b65SKonstantin Belousov 	}
2173afde2b65SKonstantin Belousov 	return (kern_kmq_open(td, uap->path, flags, uap->mode,
2174afde2b65SKonstantin Belousov 	    uap->attr != NULL ? &attr : NULL));
2175afde2b65SKonstantin Belousov }
2176afde2b65SKonstantin Belousov 
2177afde2b65SKonstantin Belousov /*
2178873fbcd7SRobert Watson  * Syscall to unlink a message queue.
2179655291f2SDavid Xu  */
2180655291f2SDavid Xu int
21818451d0ddSKip Macy sys_kmq_unlink(struct thread *td, struct kmq_unlink_args *uap)
2182655291f2SDavid Xu {
2183655291f2SDavid Xu 	char path[MQFS_NAMELEN+1];
2184655291f2SDavid Xu 	struct mqfs_node *pn;
2185655291f2SDavid Xu 	int error, len;
2186655291f2SDavid Xu 
2187655291f2SDavid Xu 	error = copyinstr(uap->path, path, MQFS_NAMELEN + 1, NULL);
2188655291f2SDavid Xu         if (error)
2189655291f2SDavid Xu 		return (error);
2190655291f2SDavid Xu 
2191655291f2SDavid Xu 	len = strlen(path);
2192dc15eac0SEd Schouten 	if (len < 2 || path[0] != '/' || strchr(path + 1, '/') != NULL)
2193655291f2SDavid Xu 		return (EINVAL);
2194655291f2SDavid Xu 
2195655291f2SDavid Xu 	sx_xlock(&mqfs_data.mi_lock);
2196adb023aeSJamie Gritton 	pn = mqfs_search(mqfs_data.mi_root, path + 1, len - 1, td->td_ucred);
2197655291f2SDavid Xu 	if (pn != NULL)
2198655291f2SDavid Xu 		error = do_unlink(pn, td->td_ucred);
2199655291f2SDavid Xu 	else
2200655291f2SDavid Xu 		error = ENOENT;
2201655291f2SDavid Xu 	sx_xunlock(&mqfs_data.mi_lock);
2202655291f2SDavid Xu 	return (error);
2203655291f2SDavid Xu }
2204655291f2SDavid Xu 
22057008be5bSPawel Jakub Dawidek typedef int (*_fgetf)(struct thread *, int, cap_rights_t *, struct file **);
2206655291f2SDavid Xu 
2207655291f2SDavid Xu /*
2208655291f2SDavid Xu  * Get message queue by giving file slot
2209655291f2SDavid Xu  */
2210655291f2SDavid Xu static int
22117008be5bSPawel Jakub Dawidek _getmq(struct thread *td, int fd, cap_rights_t *rightsp, _fgetf func,
2212655291f2SDavid Xu        struct file **fpp, struct mqfs_node **ppn, struct mqueue **pmq)
2213655291f2SDavid Xu {
2214655291f2SDavid Xu 	struct mqfs_node *pn;
2215655291f2SDavid Xu 	int error;
2216655291f2SDavid Xu 
22177008be5bSPawel Jakub Dawidek 	error = func(td, fd, rightsp, fpp);
2218655291f2SDavid Xu 	if (error)
2219655291f2SDavid Xu 		return (error);
2220655291f2SDavid Xu 	if (&mqueueops != (*fpp)->f_ops) {
2221655291f2SDavid Xu 		fdrop(*fpp, td);
2222655291f2SDavid Xu 		return (EBADF);
2223655291f2SDavid Xu 	}
2224b2f92ef9SDavid Xu 	pn = (*fpp)->f_data;
2225655291f2SDavid Xu 	if (ppn)
2226655291f2SDavid Xu 		*ppn = pn;
2227655291f2SDavid Xu 	if (pmq)
2228655291f2SDavid Xu 		*pmq = pn->mn_data;
2229655291f2SDavid Xu 	return (0);
2230655291f2SDavid Xu }
2231655291f2SDavid Xu 
2232655291f2SDavid Xu static __inline int
2233655291f2SDavid Xu getmq(struct thread *td, int fd, struct file **fpp, struct mqfs_node **ppn,
2234655291f2SDavid Xu 	struct mqueue **pmq)
2235655291f2SDavid Xu {
22367008be5bSPawel Jakub Dawidek 	cap_rights_t rights;
22377008be5bSPawel Jakub Dawidek 
2238ed5848c8SPawel Jakub Dawidek 	return _getmq(td, fd, cap_rights_init(&rights, CAP_EVENT), fget,
22397008be5bSPawel Jakub Dawidek 	    fpp, ppn, pmq);
2240655291f2SDavid Xu }
2241655291f2SDavid Xu 
2242655291f2SDavid Xu static __inline int
2243655291f2SDavid Xu getmq_read(struct thread *td, int fd, struct file **fpp,
2244655291f2SDavid Xu 	 struct mqfs_node **ppn, struct mqueue **pmq)
2245655291f2SDavid Xu {
22467008be5bSPawel Jakub Dawidek 	cap_rights_t rights;
22477008be5bSPawel Jakub Dawidek 
22487008be5bSPawel Jakub Dawidek 	return _getmq(td, fd, cap_rights_init(&rights, CAP_READ), fget_read,
22497008be5bSPawel Jakub Dawidek 	    fpp, ppn, pmq);
2250655291f2SDavid Xu }
2251655291f2SDavid Xu 
2252655291f2SDavid Xu static __inline int
2253655291f2SDavid Xu getmq_write(struct thread *td, int fd, struct file **fpp,
2254655291f2SDavid Xu 	struct mqfs_node **ppn, struct mqueue **pmq)
2255655291f2SDavid Xu {
22567008be5bSPawel Jakub Dawidek 	cap_rights_t rights;
22577008be5bSPawel Jakub Dawidek 
22587008be5bSPawel Jakub Dawidek 	return _getmq(td, fd, cap_rights_init(&rights, CAP_WRITE), fget_write,
22597008be5bSPawel Jakub Dawidek 	    fpp, ppn, pmq);
2260655291f2SDavid Xu }
2261655291f2SDavid Xu 
2262afde2b65SKonstantin Belousov static int
2263afde2b65SKonstantin Belousov kern_kmq_setattr(struct thread *td, int mqd, const struct mq_attr *attr,
2264afde2b65SKonstantin Belousov     struct mq_attr *oattr)
2265655291f2SDavid Xu {
2266655291f2SDavid Xu 	struct mqueue *mq;
2267655291f2SDavid Xu 	struct file *fp;
2268397c19d1SJeff Roberson 	u_int oflag, flag;
2269655291f2SDavid Xu 	int error;
2270655291f2SDavid Xu 
2271afde2b65SKonstantin Belousov 	if (attr != NULL && (attr->mq_flags & ~O_NONBLOCK) != 0)
2272655291f2SDavid Xu 		return (EINVAL);
2273afde2b65SKonstantin Belousov 	error = getmq(td, mqd, &fp, NULL, &mq);
2274655291f2SDavid Xu 	if (error)
2275655291f2SDavid Xu 		return (error);
2276afde2b65SKonstantin Belousov 	oattr->mq_maxmsg  = mq->mq_maxmsg;
2277afde2b65SKonstantin Belousov 	oattr->mq_msgsize = mq->mq_msgsize;
2278afde2b65SKonstantin Belousov 	oattr->mq_curmsgs = mq->mq_curmsgs;
2279afde2b65SKonstantin Belousov 	if (attr != NULL) {
2280397c19d1SJeff Roberson 		do {
2281397c19d1SJeff Roberson 			oflag = flag = fp->f_flag;
2282397c19d1SJeff Roberson 			flag &= ~O_NONBLOCK;
2283afde2b65SKonstantin Belousov 			flag |= (attr->mq_flags & O_NONBLOCK);
2284397c19d1SJeff Roberson 		} while (atomic_cmpset_int(&fp->f_flag, oflag, flag) == 0);
2285397c19d1SJeff Roberson 	} else
2286397c19d1SJeff Roberson 		oflag = fp->f_flag;
2287afde2b65SKonstantin Belousov 	oattr->mq_flags = (O_NONBLOCK & oflag);
2288655291f2SDavid Xu 	fdrop(fp, td);
2289afde2b65SKonstantin Belousov 	return (error);
2290afde2b65SKonstantin Belousov }
2291afde2b65SKonstantin Belousov 
2292afde2b65SKonstantin Belousov int
22938451d0ddSKip Macy sys_kmq_setattr(struct thread *td, struct kmq_setattr_args *uap)
2294afde2b65SKonstantin Belousov {
2295afde2b65SKonstantin Belousov 	struct mq_attr attr, oattr;
2296afde2b65SKonstantin Belousov 	int error;
2297afde2b65SKonstantin Belousov 
2298afde2b65SKonstantin Belousov 	if (uap->attr != NULL) {
2299afde2b65SKonstantin Belousov 		error = copyin(uap->attr, &attr, sizeof(attr));
2300afde2b65SKonstantin Belousov 		if (error != 0)
2301afde2b65SKonstantin Belousov 			return (error);
2302afde2b65SKonstantin Belousov 	}
2303afde2b65SKonstantin Belousov 	error = kern_kmq_setattr(td, uap->mqd, uap->attr != NULL ? &attr : NULL,
2304afde2b65SKonstantin Belousov 	    &oattr);
2305afde2b65SKonstantin Belousov 	if (error != 0)
2306afde2b65SKonstantin Belousov 		return (error);
2307afde2b65SKonstantin Belousov 	if (uap->oattr != NULL)
2308655291f2SDavid Xu 		error = copyout(&oattr, uap->oattr, sizeof(oattr));
2309655291f2SDavid Xu 	return (error);
2310655291f2SDavid Xu }
2311655291f2SDavid Xu 
2312655291f2SDavid Xu int
23138451d0ddSKip Macy sys_kmq_timedreceive(struct thread *td, struct kmq_timedreceive_args *uap)
2314655291f2SDavid Xu {
2315655291f2SDavid Xu 	struct mqueue *mq;
2316655291f2SDavid Xu 	struct file *fp;
2317afde2b65SKonstantin Belousov 	struct timespec *abs_timeout, ets;
2318655291f2SDavid Xu 	int error;
2319655291f2SDavid Xu 	int waitok;
2320655291f2SDavid Xu 
2321655291f2SDavid Xu 	error = getmq_read(td, uap->mqd, &fp, NULL, &mq);
2322655291f2SDavid Xu 	if (error)
2323655291f2SDavid Xu 		return (error);
2324afde2b65SKonstantin Belousov 	if (uap->abs_timeout != NULL) {
2325afde2b65SKonstantin Belousov 		error = copyin(uap->abs_timeout, &ets, sizeof(ets));
2326afde2b65SKonstantin Belousov 		if (error != 0)
2327afde2b65SKonstantin Belousov 			return (error);
2328afde2b65SKonstantin Belousov 		abs_timeout = &ets;
2329afde2b65SKonstantin Belousov 	} else
2330afde2b65SKonstantin Belousov 		abs_timeout = NULL;
2331655291f2SDavid Xu 	waitok = !(fp->f_flag & O_NONBLOCK);
2332655291f2SDavid Xu 	error = mqueue_receive(mq, uap->msg_ptr, uap->msg_len,
2333afde2b65SKonstantin Belousov 		uap->msg_prio, waitok, abs_timeout);
2334655291f2SDavid Xu 	fdrop(fp, td);
2335655291f2SDavid Xu 	return (error);
2336655291f2SDavid Xu }
2337655291f2SDavid Xu 
2338655291f2SDavid Xu int
23398451d0ddSKip Macy sys_kmq_timedsend(struct thread *td, struct kmq_timedsend_args *uap)
2340655291f2SDavid Xu {
2341655291f2SDavid Xu 	struct mqueue *mq;
2342655291f2SDavid Xu 	struct file *fp;
2343afde2b65SKonstantin Belousov 	struct timespec *abs_timeout, ets;
2344655291f2SDavid Xu 	int error, waitok;
2345655291f2SDavid Xu 
2346655291f2SDavid Xu 	error = getmq_write(td, uap->mqd, &fp, NULL, &mq);
2347655291f2SDavid Xu 	if (error)
2348655291f2SDavid Xu 		return (error);
2349afde2b65SKonstantin Belousov 	if (uap->abs_timeout != NULL) {
2350afde2b65SKonstantin Belousov 		error = copyin(uap->abs_timeout, &ets, sizeof(ets));
2351afde2b65SKonstantin Belousov 		if (error != 0)
2352afde2b65SKonstantin Belousov 			return (error);
2353afde2b65SKonstantin Belousov 		abs_timeout = &ets;
2354afde2b65SKonstantin Belousov 	} else
2355afde2b65SKonstantin Belousov 		abs_timeout = NULL;
2356655291f2SDavid Xu 	waitok = !(fp->f_flag & O_NONBLOCK);
2357655291f2SDavid Xu 	error = mqueue_send(mq, uap->msg_ptr, uap->msg_len,
2358afde2b65SKonstantin Belousov 		uap->msg_prio, waitok, abs_timeout);
2359655291f2SDavid Xu 	fdrop(fp, td);
2360655291f2SDavid Xu 	return (error);
2361655291f2SDavid Xu }
2362655291f2SDavid Xu 
2363058262c6SKonstantin Belousov static int
2364058262c6SKonstantin Belousov kern_kmq_notify(struct thread *td, int mqd, struct sigevent *sigev)
2365655291f2SDavid Xu {
23662af0c790SGleb Smirnoff #ifdef CAPABILITIES
23677008be5bSPawel Jakub Dawidek 	cap_rights_t rights;
23682af0c790SGleb Smirnoff #endif
2369b2f92ef9SDavid Xu 	struct filedesc *fdp;
2370b2f92ef9SDavid Xu 	struct proc *p;
2371655291f2SDavid Xu 	struct mqueue *mq;
2372a9d2f8d8SRobert Watson 	struct file *fp, *fp2;
2373b2f92ef9SDavid Xu 	struct mqueue_notifier *nt, *newnt = NULL;
2374655291f2SDavid Xu 	int error;
2375655291f2SDavid Xu 
2376058262c6SKonstantin Belousov 	if (sigev != NULL) {
2377058262c6SKonstantin Belousov 		if (sigev->sigev_notify != SIGEV_SIGNAL &&
2378058262c6SKonstantin Belousov 		    sigev->sigev_notify != SIGEV_THREAD_ID &&
2379058262c6SKonstantin Belousov 		    sigev->sigev_notify != SIGEV_NONE)
2380655291f2SDavid Xu 			return (EINVAL);
2381058262c6SKonstantin Belousov 		if ((sigev->sigev_notify == SIGEV_SIGNAL ||
2382058262c6SKonstantin Belousov 		    sigev->sigev_notify == SIGEV_THREAD_ID) &&
2383058262c6SKonstantin Belousov 		    !_SIG_VALID(sigev->sigev_signo))
2384655291f2SDavid Xu 			return (EINVAL);
2385655291f2SDavid Xu 	}
2386058262c6SKonstantin Belousov 	p = td->td_proc;
2387058262c6SKonstantin Belousov 	fdp = td->td_proc->p_fd;
2388058262c6SKonstantin Belousov 	error = getmq(td, mqd, &fp, NULL, &mq);
2389655291f2SDavid Xu 	if (error)
2390655291f2SDavid Xu 		return (error);
2391b2f92ef9SDavid Xu again:
23925e3f7694SRobert Watson 	FILEDESC_SLOCK(fdp);
2393058262c6SKonstantin Belousov 	fp2 = fget_locked(fdp, mqd);
2394a9d2f8d8SRobert Watson 	if (fp2 == NULL) {
2395a9d2f8d8SRobert Watson 		FILEDESC_SUNLOCK(fdp);
2396a9d2f8d8SRobert Watson 		error = EBADF;
2397a9d2f8d8SRobert Watson 		goto out;
2398a9d2f8d8SRobert Watson 	}
23992609222aSPawel Jakub Dawidek #ifdef CAPABILITIES
24007008be5bSPawel Jakub Dawidek 	error = cap_check(cap_rights(fdp, mqd),
2401ed5848c8SPawel Jakub Dawidek 	    cap_rights_init(&rights, CAP_EVENT));
2402a9d2f8d8SRobert Watson 	if (error) {
2403a9d2f8d8SRobert Watson 		FILEDESC_SUNLOCK(fdp);
2404a9d2f8d8SRobert Watson 		goto out;
2405a9d2f8d8SRobert Watson 	}
24062609222aSPawel Jakub Dawidek #endif
2407a9d2f8d8SRobert Watson 	if (fp2 != fp) {
24085e3f7694SRobert Watson 		FILEDESC_SUNLOCK(fdp);
2409b2f92ef9SDavid Xu 		error = EBADF;
2410b2f92ef9SDavid Xu 		goto out;
2411b2f92ef9SDavid Xu 	}
2412655291f2SDavid Xu 	mtx_lock(&mq->mq_mutex);
24135e3f7694SRobert Watson 	FILEDESC_SUNLOCK(fdp);
2414058262c6SKonstantin Belousov 	if (sigev != NULL) {
2415655291f2SDavid Xu 		if (mq->mq_notifier != NULL) {
2416655291f2SDavid Xu 			error = EBUSY;
2417052ea11cSDavid Xu 		} else {
2418b2f92ef9SDavid Xu 			PROC_LOCK(p);
2419058262c6SKonstantin Belousov 			nt = notifier_search(p, mqd);
2420b2f92ef9SDavid Xu 			if (nt == NULL) {
2421b2f92ef9SDavid Xu 				if (newnt == NULL) {
2422b2f92ef9SDavid Xu 					PROC_UNLOCK(p);
2423b2f92ef9SDavid Xu 					mtx_unlock(&mq->mq_mutex);
2424b2f92ef9SDavid Xu 					newnt = notifier_alloc();
2425b2f92ef9SDavid Xu 					goto again;
2426b2f92ef9SDavid Xu 				}
2427b2f92ef9SDavid Xu 			}
2428b2f92ef9SDavid Xu 
2429b2f92ef9SDavid Xu 			if (nt != NULL) {
2430b2f92ef9SDavid Xu 				sigqueue_take(&nt->nt_ksi);
2431b2f92ef9SDavid Xu 				if (newnt != NULL) {
2432b2f92ef9SDavid Xu 					notifier_free(newnt);
2433b2f92ef9SDavid Xu 					newnt = NULL;
2434b2f92ef9SDavid Xu 				}
2435b2f92ef9SDavid Xu 			} else {
2436b2f92ef9SDavid Xu 				nt = newnt;
2437b2f92ef9SDavid Xu 				newnt = NULL;
2438b2f92ef9SDavid Xu 				ksiginfo_init(&nt->nt_ksi);
2439b2f92ef9SDavid Xu 				nt->nt_ksi.ksi_flags |= KSI_INS | KSI_EXT;
2440b2f92ef9SDavid Xu 				nt->nt_ksi.ksi_code = SI_MESGQ;
2441b2f92ef9SDavid Xu 				nt->nt_proc = p;
2442058262c6SKonstantin Belousov 				nt->nt_ksi.ksi_mqd = mqd;
2443b2f92ef9SDavid Xu 				notifier_insert(p, nt);
2444b2f92ef9SDavid Xu 			}
2445058262c6SKonstantin Belousov 			nt->nt_sigev = *sigev;
2446b2f92ef9SDavid Xu 			mq->mq_notifier = nt;
2447b2f92ef9SDavid Xu 			PROC_UNLOCK(p);
2448655291f2SDavid Xu 			/*
2449b2f92ef9SDavid Xu 			 * if there is no receivers and message queue
2450b2f92ef9SDavid Xu 			 * is not empty, we should send notification
2451b2f92ef9SDavid Xu 			 * as soon as possible.
2452655291f2SDavid Xu 			 */
2453655291f2SDavid Xu 			if (mq->mq_receivers == 0 &&
2454655291f2SDavid Xu 			    !TAILQ_EMPTY(&mq->mq_msgq))
2455655291f2SDavid Xu 				mqueue_send_notification(mq);
2456655291f2SDavid Xu 		}
2457655291f2SDavid Xu 	} else {
2458058262c6SKonstantin Belousov 		notifier_remove(p, mq, mqd);
2459655291f2SDavid Xu 	}
2460655291f2SDavid Xu 	mtx_unlock(&mq->mq_mutex);
2461b2f92ef9SDavid Xu 
2462b2f92ef9SDavid Xu out:
2463655291f2SDavid Xu 	fdrop(fp, td);
2464b2f92ef9SDavid Xu 	if (newnt != NULL)
2465b2f92ef9SDavid Xu 		notifier_free(newnt);
2466655291f2SDavid Xu 	return (error);
2467655291f2SDavid Xu }
2468655291f2SDavid Xu 
2469058262c6SKonstantin Belousov int
2470058262c6SKonstantin Belousov sys_kmq_notify(struct thread *td, struct kmq_notify_args *uap)
2471058262c6SKonstantin Belousov {
2472058262c6SKonstantin Belousov 	struct sigevent ev, *evp;
2473058262c6SKonstantin Belousov 	int error;
2474058262c6SKonstantin Belousov 
2475058262c6SKonstantin Belousov 	if (uap->sigev == NULL) {
2476058262c6SKonstantin Belousov 		evp = NULL;
2477058262c6SKonstantin Belousov 	} else {
2478058262c6SKonstantin Belousov 		error = copyin(uap->sigev, &ev, sizeof(ev));
2479058262c6SKonstantin Belousov 		if (error != 0)
2480058262c6SKonstantin Belousov 			return (error);
2481058262c6SKonstantin Belousov 		evp = &ev;
2482058262c6SKonstantin Belousov 	}
2483058262c6SKonstantin Belousov 	return (kern_kmq_notify(td, uap->mqd, evp));
2484058262c6SKonstantin Belousov }
2485058262c6SKonstantin Belousov 
2486b2f92ef9SDavid Xu static void
2487b2f92ef9SDavid Xu mqueue_fdclose(struct thread *td, int fd, struct file *fp)
2488b2f92ef9SDavid Xu {
2489b2f92ef9SDavid Xu 	struct filedesc *fdp;
2490b2f92ef9SDavid Xu 	struct mqueue *mq;
2491b2f92ef9SDavid Xu 
2492b2f92ef9SDavid Xu 	fdp = td->td_proc->p_fd;
24935e3f7694SRobert Watson 	FILEDESC_LOCK_ASSERT(fdp);
24945e3f7694SRobert Watson 
2495b2f92ef9SDavid Xu 	if (fp->f_ops == &mqueueops) {
2496b2f92ef9SDavid Xu 		mq = FPTOMQ(fp);
2497b2f92ef9SDavid Xu 		mtx_lock(&mq->mq_mutex);
2498b2f92ef9SDavid Xu 		notifier_remove(td->td_proc, mq, fd);
2499b2f92ef9SDavid Xu 
2500b2f92ef9SDavid Xu 		/* have to wakeup thread in same process */
2501b2f92ef9SDavid Xu 		if (mq->mq_flags & MQ_RSEL) {
2502b2f92ef9SDavid Xu 			mq->mq_flags &= ~MQ_RSEL;
250303f70aecSDavid Xu 			selwakeup(&mq->mq_rsel);
2504b2f92ef9SDavid Xu 		}
2505b2f92ef9SDavid Xu 		if (mq->mq_flags & MQ_WSEL) {
2506b2f92ef9SDavid Xu 			mq->mq_flags &= ~MQ_WSEL;
250703f70aecSDavid Xu 			selwakeup(&mq->mq_wsel);
2508b2f92ef9SDavid Xu 		}
2509b2f92ef9SDavid Xu 		mtx_unlock(&mq->mq_mutex);
2510b2f92ef9SDavid Xu 	}
2511b2f92ef9SDavid Xu }
2512b2f92ef9SDavid Xu 
2513b2f92ef9SDavid Xu static void
2514b2f92ef9SDavid Xu mq_proc_exit(void *arg __unused, struct proc *p)
2515b2f92ef9SDavid Xu {
2516b2f92ef9SDavid Xu 	struct filedesc *fdp;
2517b2f92ef9SDavid Xu 	struct file *fp;
2518b2f92ef9SDavid Xu 	struct mqueue *mq;
2519b2f92ef9SDavid Xu 	int i;
2520b2f92ef9SDavid Xu 
2521b2f92ef9SDavid Xu 	fdp = p->p_fd;
25225e3f7694SRobert Watson 	FILEDESC_SLOCK(fdp);
2523b2f92ef9SDavid Xu 	for (i = 0; i < fdp->fd_nfiles; ++i) {
2524b2f92ef9SDavid Xu 		fp = fget_locked(fdp, i);
2525b2f92ef9SDavid Xu 		if (fp != NULL && fp->f_ops == &mqueueops) {
2526b2f92ef9SDavid Xu 			mq = FPTOMQ(fp);
2527b2f92ef9SDavid Xu 			mtx_lock(&mq->mq_mutex);
2528b2f92ef9SDavid Xu 			notifier_remove(p, FPTOMQ(fp), i);
2529b2f92ef9SDavid Xu 			mtx_unlock(&mq->mq_mutex);
2530b2f92ef9SDavid Xu 		}
2531b2f92ef9SDavid Xu 	}
25325e3f7694SRobert Watson 	FILEDESC_SUNLOCK(fdp);
2533b2f92ef9SDavid Xu 	KASSERT(LIST_EMPTY(&p->p_mqnotifier), ("mq notifiers left"));
2534b2f92ef9SDavid Xu }
2535b2f92ef9SDavid Xu 
2536655291f2SDavid Xu static int
2537655291f2SDavid Xu mqf_poll(struct file *fp, int events, struct ucred *active_cred,
2538655291f2SDavid Xu 	struct thread *td)
2539655291f2SDavid Xu {
2540655291f2SDavid Xu 	struct mqueue *mq = FPTOMQ(fp);
2541655291f2SDavid Xu 	int revents = 0;
2542655291f2SDavid Xu 
2543655291f2SDavid Xu 	mtx_lock(&mq->mq_mutex);
2544655291f2SDavid Xu 	if (events & (POLLIN | POLLRDNORM)) {
2545655291f2SDavid Xu 		if (mq->mq_curmsgs) {
2546655291f2SDavid Xu 			revents |= events & (POLLIN | POLLRDNORM);
2547655291f2SDavid Xu 		} else {
2548655291f2SDavid Xu 			mq->mq_flags |= MQ_RSEL;
2549655291f2SDavid Xu 			selrecord(td, &mq->mq_rsel);
2550655291f2SDavid Xu  		}
2551655291f2SDavid Xu 	}
2552655291f2SDavid Xu 	if (events & POLLOUT) {
2553655291f2SDavid Xu 		if (mq->mq_curmsgs < mq->mq_maxmsg)
2554655291f2SDavid Xu 			revents |= POLLOUT;
2555655291f2SDavid Xu 		else {
2556655291f2SDavid Xu 			mq->mq_flags |= MQ_WSEL;
2557655291f2SDavid Xu 			selrecord(td, &mq->mq_wsel);
2558655291f2SDavid Xu 		}
2559655291f2SDavid Xu 	}
2560655291f2SDavid Xu 	mtx_unlock(&mq->mq_mutex);
2561655291f2SDavid Xu 	return (revents);
2562655291f2SDavid Xu }
2563655291f2SDavid Xu 
2564655291f2SDavid Xu static int
2565655291f2SDavid Xu mqf_close(struct file *fp, struct thread *td)
2566655291f2SDavid Xu {
2567655291f2SDavid Xu 	struct mqfs_node *pn;
2568655291f2SDavid Xu 
2569655291f2SDavid Xu 	fp->f_ops = &badfileops;
2570b2f92ef9SDavid Xu 	pn = fp->f_data;
2571655291f2SDavid Xu 	fp->f_data = NULL;
2572655291f2SDavid Xu 	sx_xlock(&mqfs_data.mi_lock);
2573655291f2SDavid Xu 	mqnode_release(pn);
2574655291f2SDavid Xu 	sx_xunlock(&mqfs_data.mi_lock);
2575655291f2SDavid Xu 	return (0);
2576655291f2SDavid Xu }
2577655291f2SDavid Xu 
2578655291f2SDavid Xu static int
2579655291f2SDavid Xu mqf_stat(struct file *fp, struct stat *st, struct ucred *active_cred,
2580655291f2SDavid Xu 	struct thread *td)
2581655291f2SDavid Xu {
2582b2f92ef9SDavid Xu 	struct mqfs_node *pn = fp->f_data;
2583655291f2SDavid Xu 
2584655291f2SDavid Xu 	bzero(st, sizeof *st);
25859c00bb91SKonstantin Belousov 	sx_xlock(&mqfs_data.mi_lock);
2586510ea843SEd Schouten 	st->st_atim = pn->mn_atime;
2587510ea843SEd Schouten 	st->st_mtim = pn->mn_mtime;
2588510ea843SEd Schouten 	st->st_ctim = pn->mn_ctime;
2589510ea843SEd Schouten 	st->st_birthtim = pn->mn_birth;
2590655291f2SDavid Xu 	st->st_uid = pn->mn_uid;
2591655291f2SDavid Xu 	st->st_gid = pn->mn_gid;
2592655291f2SDavid Xu 	st->st_mode = S_IFIFO | pn->mn_mode;
25939c00bb91SKonstantin Belousov 	sx_xunlock(&mqfs_data.mi_lock);
2594655291f2SDavid Xu 	return (0);
2595655291f2SDavid Xu }
2596655291f2SDavid Xu 
2597655291f2SDavid Xu static int
25989c00bb91SKonstantin Belousov mqf_chmod(struct file *fp, mode_t mode, struct ucred *active_cred,
25999c00bb91SKonstantin Belousov     struct thread *td)
26009c00bb91SKonstantin Belousov {
26019c00bb91SKonstantin Belousov 	struct mqfs_node *pn;
26029c00bb91SKonstantin Belousov 	int error;
26039c00bb91SKonstantin Belousov 
26049c00bb91SKonstantin Belousov 	error = 0;
26059c00bb91SKonstantin Belousov 	pn = fp->f_data;
26069c00bb91SKonstantin Belousov 	sx_xlock(&mqfs_data.mi_lock);
26079c00bb91SKonstantin Belousov 	error = vaccess(VREG, pn->mn_mode, pn->mn_uid, pn->mn_gid, VADMIN,
26089c00bb91SKonstantin Belousov 	    active_cred, NULL);
26099c00bb91SKonstantin Belousov 	if (error != 0)
26109c00bb91SKonstantin Belousov 		goto out;
26119c00bb91SKonstantin Belousov 	pn->mn_mode = mode & ACCESSPERMS;
26129c00bb91SKonstantin Belousov out:
26139c00bb91SKonstantin Belousov 	sx_xunlock(&mqfs_data.mi_lock);
26149c00bb91SKonstantin Belousov 	return (error);
26159c00bb91SKonstantin Belousov }
26169c00bb91SKonstantin Belousov 
26179c00bb91SKonstantin Belousov static int
26189c00bb91SKonstantin Belousov mqf_chown(struct file *fp, uid_t uid, gid_t gid, struct ucred *active_cred,
26199c00bb91SKonstantin Belousov     struct thread *td)
26209c00bb91SKonstantin Belousov {
26219c00bb91SKonstantin Belousov 	struct mqfs_node *pn;
26229c00bb91SKonstantin Belousov 	int error;
26239c00bb91SKonstantin Belousov 
26249c00bb91SKonstantin Belousov 	error = 0;
26259c00bb91SKonstantin Belousov 	pn = fp->f_data;
26269c00bb91SKonstantin Belousov 	sx_xlock(&mqfs_data.mi_lock);
26279c00bb91SKonstantin Belousov 	if (uid == (uid_t)-1)
26289c00bb91SKonstantin Belousov 		uid = pn->mn_uid;
26299c00bb91SKonstantin Belousov 	if (gid == (gid_t)-1)
26309c00bb91SKonstantin Belousov 		gid = pn->mn_gid;
26319c00bb91SKonstantin Belousov 	if (((uid != pn->mn_uid && uid != active_cred->cr_uid) ||
26329c00bb91SKonstantin Belousov 	    (gid != pn->mn_gid && !groupmember(gid, active_cred))) &&
26339c00bb91SKonstantin Belousov 	    (error = priv_check_cred(active_cred, PRIV_VFS_CHOWN, 0)))
26349c00bb91SKonstantin Belousov 		goto out;
26359c00bb91SKonstantin Belousov 	pn->mn_uid = uid;
26369c00bb91SKonstantin Belousov 	pn->mn_gid = gid;
26379c00bb91SKonstantin Belousov out:
26389c00bb91SKonstantin Belousov 	sx_xunlock(&mqfs_data.mi_lock);
26399c00bb91SKonstantin Belousov 	return (error);
26409c00bb91SKonstantin Belousov }
26419c00bb91SKonstantin Belousov 
26429c00bb91SKonstantin Belousov static int
2643655291f2SDavid Xu mqf_kqfilter(struct file *fp, struct knote *kn)
2644655291f2SDavid Xu {
2645655291f2SDavid Xu 	struct mqueue *mq = FPTOMQ(fp);
2646655291f2SDavid Xu 	int error = 0;
2647655291f2SDavid Xu 
2648655291f2SDavid Xu 	if (kn->kn_filter == EVFILT_READ) {
2649655291f2SDavid Xu 		kn->kn_fop = &mq_rfiltops;
2650655291f2SDavid Xu 		knlist_add(&mq->mq_rsel.si_note, kn, 0);
2651655291f2SDavid Xu 	} else if (kn->kn_filter == EVFILT_WRITE) {
2652655291f2SDavid Xu 		kn->kn_fop = &mq_wfiltops;
2653655291f2SDavid Xu 		knlist_add(&mq->mq_wsel.si_note, kn, 0);
2654655291f2SDavid Xu 	} else
2655655291f2SDavid Xu 		error = EINVAL;
2656655291f2SDavid Xu 	return (error);
2657655291f2SDavid Xu }
2658655291f2SDavid Xu 
2659655291f2SDavid Xu static void
2660655291f2SDavid Xu filt_mqdetach(struct knote *kn)
2661655291f2SDavid Xu {
2662655291f2SDavid Xu 	struct mqueue *mq = FPTOMQ(kn->kn_fp);
2663655291f2SDavid Xu 
2664655291f2SDavid Xu 	if (kn->kn_filter == EVFILT_READ)
2665655291f2SDavid Xu 		knlist_remove(&mq->mq_rsel.si_note, kn, 0);
2666655291f2SDavid Xu 	else if (kn->kn_filter == EVFILT_WRITE)
2667655291f2SDavid Xu 		knlist_remove(&mq->mq_wsel.si_note, kn, 0);
2668655291f2SDavid Xu 	else
2669655291f2SDavid Xu 		panic("filt_mqdetach");
2670655291f2SDavid Xu }
2671655291f2SDavid Xu 
2672655291f2SDavid Xu static int
2673655291f2SDavid Xu filt_mqread(struct knote *kn, long hint)
2674655291f2SDavid Xu {
2675655291f2SDavid Xu 	struct mqueue *mq = FPTOMQ(kn->kn_fp);
2676655291f2SDavid Xu 
2677655291f2SDavid Xu 	mtx_assert(&mq->mq_mutex, MA_OWNED);
2678655291f2SDavid Xu 	return (mq->mq_curmsgs != 0);
2679655291f2SDavid Xu }
2680655291f2SDavid Xu 
2681655291f2SDavid Xu static int
2682655291f2SDavid Xu filt_mqwrite(struct knote *kn, long hint)
2683655291f2SDavid Xu {
2684655291f2SDavid Xu 	struct mqueue *mq = FPTOMQ(kn->kn_fp);
2685655291f2SDavid Xu 
2686655291f2SDavid Xu 	mtx_assert(&mq->mq_mutex, MA_OWNED);
2687655291f2SDavid Xu 	return (mq->mq_curmsgs < mq->mq_maxmsg);
2688655291f2SDavid Xu }
2689655291f2SDavid Xu 
26909696feebSJohn Baldwin static int
26919696feebSJohn Baldwin mqf_fill_kinfo(struct file *fp, struct kinfo_file *kif, struct filedesc *fdp)
26929696feebSJohn Baldwin {
26939696feebSJohn Baldwin 
26949696feebSJohn Baldwin 	kif->kf_type = KF_TYPE_MQUEUE;
26959696feebSJohn Baldwin 	return (0);
26969696feebSJohn Baldwin }
26979696feebSJohn Baldwin 
2698655291f2SDavid Xu static struct fileops mqueueops = {
26992d69d0dcSJohn Baldwin 	.fo_read		= invfo_rdwr,
27002d69d0dcSJohn Baldwin 	.fo_write		= invfo_rdwr,
27012d69d0dcSJohn Baldwin 	.fo_truncate		= invfo_truncate,
27022d69d0dcSJohn Baldwin 	.fo_ioctl		= invfo_ioctl,
2703655291f2SDavid Xu 	.fo_poll		= mqf_poll,
2704655291f2SDavid Xu 	.fo_kqfilter		= mqf_kqfilter,
2705655291f2SDavid Xu 	.fo_stat		= mqf_stat,
27062d69d0dcSJohn Baldwin 	.fo_close		= mqf_close,
27079c00bb91SKonstantin Belousov 	.fo_chmod		= mqf_chmod,
27089c00bb91SKonstantin Belousov 	.fo_chown		= mqf_chown,
2709ca04d21dSGleb Smirnoff 	.fo_sendfile		= invfo_sendfile,
27109696feebSJohn Baldwin 	.fo_fill_kinfo		= mqf_fill_kinfo,
2711655291f2SDavid Xu };
2712655291f2SDavid Xu 
2713655291f2SDavid Xu static struct vop_vector mqfs_vnodeops = {
2714655291f2SDavid Xu 	.vop_default 		= &default_vnodeops,
2715655291f2SDavid Xu 	.vop_access		= mqfs_access,
2716655291f2SDavid Xu 	.vop_cachedlookup	= mqfs_lookup,
2717655291f2SDavid Xu 	.vop_lookup		= vfs_cache_lookup,
2718655291f2SDavid Xu 	.vop_reclaim		= mqfs_reclaim,
2719655291f2SDavid Xu 	.vop_create		= mqfs_create,
2720655291f2SDavid Xu 	.vop_remove		= mqfs_remove,
2721655291f2SDavid Xu 	.vop_inactive		= mqfs_inactive,
2722655291f2SDavid Xu 	.vop_open		= mqfs_open,
2723655291f2SDavid Xu 	.vop_close		= mqfs_close,
2724655291f2SDavid Xu 	.vop_getattr		= mqfs_getattr,
2725655291f2SDavid Xu 	.vop_setattr		= mqfs_setattr,
2726655291f2SDavid Xu 	.vop_read		= mqfs_read,
2727655291f2SDavid Xu 	.vop_write		= VOP_EOPNOTSUPP,
2728655291f2SDavid Xu 	.vop_readdir		= mqfs_readdir,
2729655291f2SDavid Xu 	.vop_mkdir		= VOP_EOPNOTSUPP,
2730655291f2SDavid Xu 	.vop_rmdir		= VOP_EOPNOTSUPP
2731655291f2SDavid Xu };
2732655291f2SDavid Xu 
2733655291f2SDavid Xu static struct vfsops mqfs_vfsops = {
2734655291f2SDavid Xu 	.vfs_init 		= mqfs_init,
2735655291f2SDavid Xu 	.vfs_uninit		= mqfs_uninit,
2736655291f2SDavid Xu 	.vfs_mount		= mqfs_mount,
2737655291f2SDavid Xu 	.vfs_unmount		= mqfs_unmount,
2738655291f2SDavid Xu 	.vfs_root		= mqfs_root,
2739655291f2SDavid Xu 	.vfs_statfs		= mqfs_statfs,
2740655291f2SDavid Xu };
2741655291f2SDavid Xu 
2742afde2b65SKonstantin Belousov static struct vfsconf mqueuefs_vfsconf = {
2743afde2b65SKonstantin Belousov 	.vfc_version = VFS_VERSION,
2744afde2b65SKonstantin Belousov 	.vfc_name = "mqueuefs",
2745afde2b65SKonstantin Belousov 	.vfc_vfsops = &mqfs_vfsops,
2746afde2b65SKonstantin Belousov 	.vfc_typenum = -1,
2747afde2b65SKonstantin Belousov 	.vfc_flags = VFCF_SYNTHETIC
2748afde2b65SKonstantin Belousov };
2749655291f2SDavid Xu 
2750afde2b65SKonstantin Belousov static struct syscall_helper_data mq_syscalls[] = {
2751afde2b65SKonstantin Belousov 	SYSCALL_INIT_HELPER(kmq_open),
2752afde2b65SKonstantin Belousov 	SYSCALL_INIT_HELPER(kmq_setattr),
2753afde2b65SKonstantin Belousov 	SYSCALL_INIT_HELPER(kmq_timedsend),
2754afde2b65SKonstantin Belousov 	SYSCALL_INIT_HELPER(kmq_timedreceive),
2755afde2b65SKonstantin Belousov 	SYSCALL_INIT_HELPER(kmq_notify),
2756afde2b65SKonstantin Belousov 	SYSCALL_INIT_HELPER(kmq_unlink),
2757afde2b65SKonstantin Belousov 	SYSCALL_INIT_LAST
2758afde2b65SKonstantin Belousov };
2759afde2b65SKonstantin Belousov 
2760afde2b65SKonstantin Belousov #ifdef COMPAT_FREEBSD32
2761afde2b65SKonstantin Belousov #include <compat/freebsd32/freebsd32.h>
2762afde2b65SKonstantin Belousov #include <compat/freebsd32/freebsd32_proto.h>
2763058262c6SKonstantin Belousov #include <compat/freebsd32/freebsd32_signal.h>
2764afde2b65SKonstantin Belousov #include <compat/freebsd32/freebsd32_syscall.h>
2765afde2b65SKonstantin Belousov #include <compat/freebsd32/freebsd32_util.h>
2766afde2b65SKonstantin Belousov 
2767afde2b65SKonstantin Belousov static void
2768afde2b65SKonstantin Belousov mq_attr_from32(const struct mq_attr32 *from, struct mq_attr *to)
2769afde2b65SKonstantin Belousov {
2770afde2b65SKonstantin Belousov 
2771afde2b65SKonstantin Belousov 	to->mq_flags = from->mq_flags;
2772afde2b65SKonstantin Belousov 	to->mq_maxmsg = from->mq_maxmsg;
2773afde2b65SKonstantin Belousov 	to->mq_msgsize = from->mq_msgsize;
2774afde2b65SKonstantin Belousov 	to->mq_curmsgs = from->mq_curmsgs;
2775afde2b65SKonstantin Belousov }
2776afde2b65SKonstantin Belousov 
2777afde2b65SKonstantin Belousov static void
2778afde2b65SKonstantin Belousov mq_attr_to32(const struct mq_attr *from, struct mq_attr32 *to)
2779afde2b65SKonstantin Belousov {
2780afde2b65SKonstantin Belousov 
2781afde2b65SKonstantin Belousov 	to->mq_flags = from->mq_flags;
2782afde2b65SKonstantin Belousov 	to->mq_maxmsg = from->mq_maxmsg;
2783afde2b65SKonstantin Belousov 	to->mq_msgsize = from->mq_msgsize;
2784afde2b65SKonstantin Belousov 	to->mq_curmsgs = from->mq_curmsgs;
2785afde2b65SKonstantin Belousov }
2786afde2b65SKonstantin Belousov 
2787afde2b65SKonstantin Belousov int
2788afde2b65SKonstantin Belousov freebsd32_kmq_open(struct thread *td, struct freebsd32_kmq_open_args *uap)
2789afde2b65SKonstantin Belousov {
2790afde2b65SKonstantin Belousov 	struct mq_attr attr;
2791afde2b65SKonstantin Belousov 	struct mq_attr32 attr32;
2792afde2b65SKonstantin Belousov 	int flags, error;
2793afde2b65SKonstantin Belousov 
27946fbdb9f4SJilles Tjoelker 	if ((uap->flags & O_ACCMODE) == O_ACCMODE || uap->flags & O_EXEC)
2795afde2b65SKonstantin Belousov 		return (EINVAL);
2796afde2b65SKonstantin Belousov 	flags = FFLAGS(uap->flags);
2797afde2b65SKonstantin Belousov 	if ((flags & O_CREAT) != 0 && uap->attr != NULL) {
2798afde2b65SKonstantin Belousov 		error = copyin(uap->attr, &attr32, sizeof(attr32));
2799afde2b65SKonstantin Belousov 		if (error)
2800afde2b65SKonstantin Belousov 			return (error);
2801afde2b65SKonstantin Belousov 		mq_attr_from32(&attr32, &attr);
2802afde2b65SKonstantin Belousov 	}
2803afde2b65SKonstantin Belousov 	return (kern_kmq_open(td, uap->path, flags, uap->mode,
2804afde2b65SKonstantin Belousov 	    uap->attr != NULL ? &attr : NULL));
2805afde2b65SKonstantin Belousov }
2806afde2b65SKonstantin Belousov 
2807afde2b65SKonstantin Belousov int
2808afde2b65SKonstantin Belousov freebsd32_kmq_setattr(struct thread *td, struct freebsd32_kmq_setattr_args *uap)
2809afde2b65SKonstantin Belousov {
2810afde2b65SKonstantin Belousov 	struct mq_attr attr, oattr;
2811afde2b65SKonstantin Belousov 	struct mq_attr32 attr32, oattr32;
2812afde2b65SKonstantin Belousov 	int error;
2813afde2b65SKonstantin Belousov 
2814afde2b65SKonstantin Belousov 	if (uap->attr != NULL) {
2815afde2b65SKonstantin Belousov 		error = copyin(uap->attr, &attr32, sizeof(attr32));
2816afde2b65SKonstantin Belousov 		if (error != 0)
2817afde2b65SKonstantin Belousov 			return (error);
2818afde2b65SKonstantin Belousov 		mq_attr_from32(&attr32, &attr);
2819afde2b65SKonstantin Belousov 	}
2820afde2b65SKonstantin Belousov 	error = kern_kmq_setattr(td, uap->mqd, uap->attr != NULL ? &attr : NULL,
2821afde2b65SKonstantin Belousov 	    &oattr);
2822afde2b65SKonstantin Belousov 	if (error != 0)
2823afde2b65SKonstantin Belousov 		return (error);
2824afde2b65SKonstantin Belousov 	if (uap->oattr != NULL) {
2825afde2b65SKonstantin Belousov 		mq_attr_to32(&oattr, &oattr32);
2826afde2b65SKonstantin Belousov 		error = copyout(&oattr32, uap->oattr, sizeof(oattr32));
2827afde2b65SKonstantin Belousov 	}
2828afde2b65SKonstantin Belousov 	return (error);
2829afde2b65SKonstantin Belousov }
2830afde2b65SKonstantin Belousov 
2831afde2b65SKonstantin Belousov int
2832afde2b65SKonstantin Belousov freebsd32_kmq_timedsend(struct thread *td,
2833afde2b65SKonstantin Belousov     struct freebsd32_kmq_timedsend_args *uap)
2834afde2b65SKonstantin Belousov {
2835afde2b65SKonstantin Belousov 	struct mqueue *mq;
2836afde2b65SKonstantin Belousov 	struct file *fp;
2837afde2b65SKonstantin Belousov 	struct timespec32 ets32;
2838afde2b65SKonstantin Belousov 	struct timespec *abs_timeout, ets;
2839afde2b65SKonstantin Belousov 	int error;
2840afde2b65SKonstantin Belousov 	int waitok;
2841afde2b65SKonstantin Belousov 
284228f865b0SPawel Jakub Dawidek 	error = getmq_write(td, uap->mqd, &fp, NULL, &mq);
2843afde2b65SKonstantin Belousov 	if (error)
2844afde2b65SKonstantin Belousov 		return (error);
2845afde2b65SKonstantin Belousov 	if (uap->abs_timeout != NULL) {
2846afde2b65SKonstantin Belousov 		error = copyin(uap->abs_timeout, &ets32, sizeof(ets32));
2847afde2b65SKonstantin Belousov 		if (error != 0)
2848afde2b65SKonstantin Belousov 			return (error);
2849afde2b65SKonstantin Belousov 		CP(ets32, ets, tv_sec);
2850afde2b65SKonstantin Belousov 		CP(ets32, ets, tv_nsec);
2851afde2b65SKonstantin Belousov 		abs_timeout = &ets;
2852afde2b65SKonstantin Belousov 	} else
2853afde2b65SKonstantin Belousov 		abs_timeout = NULL;
2854afde2b65SKonstantin Belousov 	waitok = !(fp->f_flag & O_NONBLOCK);
2855afde2b65SKonstantin Belousov 	error = mqueue_send(mq, uap->msg_ptr, uap->msg_len,
2856afde2b65SKonstantin Belousov 		uap->msg_prio, waitok, abs_timeout);
2857afde2b65SKonstantin Belousov 	fdrop(fp, td);
2858afde2b65SKonstantin Belousov 	return (error);
2859afde2b65SKonstantin Belousov }
2860afde2b65SKonstantin Belousov 
2861afde2b65SKonstantin Belousov int
2862afde2b65SKonstantin Belousov freebsd32_kmq_timedreceive(struct thread *td,
2863afde2b65SKonstantin Belousov     struct freebsd32_kmq_timedreceive_args *uap)
2864afde2b65SKonstantin Belousov {
2865afde2b65SKonstantin Belousov 	struct mqueue *mq;
2866afde2b65SKonstantin Belousov 	struct file *fp;
2867afde2b65SKonstantin Belousov 	struct timespec32 ets32;
2868afde2b65SKonstantin Belousov 	struct timespec *abs_timeout, ets;
2869afde2b65SKonstantin Belousov 	int error, waitok;
2870afde2b65SKonstantin Belousov 
287128f865b0SPawel Jakub Dawidek 	error = getmq_read(td, uap->mqd, &fp, NULL, &mq);
2872afde2b65SKonstantin Belousov 	if (error)
2873afde2b65SKonstantin Belousov 		return (error);
2874afde2b65SKonstantin Belousov 	if (uap->abs_timeout != NULL) {
2875afde2b65SKonstantin Belousov 		error = copyin(uap->abs_timeout, &ets32, sizeof(ets32));
2876afde2b65SKonstantin Belousov 		if (error != 0)
2877afde2b65SKonstantin Belousov 			return (error);
2878afde2b65SKonstantin Belousov 		CP(ets32, ets, tv_sec);
2879afde2b65SKonstantin Belousov 		CP(ets32, ets, tv_nsec);
2880afde2b65SKonstantin Belousov 		abs_timeout = &ets;
2881afde2b65SKonstantin Belousov 	} else
2882afde2b65SKonstantin Belousov 		abs_timeout = NULL;
2883afde2b65SKonstantin Belousov 	waitok = !(fp->f_flag & O_NONBLOCK);
2884afde2b65SKonstantin Belousov 	error = mqueue_receive(mq, uap->msg_ptr, uap->msg_len,
2885afde2b65SKonstantin Belousov 		uap->msg_prio, waitok, abs_timeout);
2886afde2b65SKonstantin Belousov 	fdrop(fp, td);
2887afde2b65SKonstantin Belousov 	return (error);
2888afde2b65SKonstantin Belousov }
2889afde2b65SKonstantin Belousov 
2890058262c6SKonstantin Belousov int
2891058262c6SKonstantin Belousov freebsd32_kmq_notify(struct thread *td, struct freebsd32_kmq_notify_args *uap)
2892058262c6SKonstantin Belousov {
2893058262c6SKonstantin Belousov 	struct sigevent ev, *evp;
2894058262c6SKonstantin Belousov 	struct sigevent32 ev32;
2895058262c6SKonstantin Belousov 	int error;
2896058262c6SKonstantin Belousov 
2897058262c6SKonstantin Belousov 	if (uap->sigev == NULL) {
2898058262c6SKonstantin Belousov 		evp = NULL;
2899058262c6SKonstantin Belousov 	} else {
2900058262c6SKonstantin Belousov 		error = copyin(uap->sigev, &ev32, sizeof(ev32));
2901058262c6SKonstantin Belousov 		if (error != 0)
2902058262c6SKonstantin Belousov 			return (error);
2903058262c6SKonstantin Belousov 		error = convert_sigevent32(&ev32, &ev);
2904058262c6SKonstantin Belousov 		if (error != 0)
2905058262c6SKonstantin Belousov 			return (error);
2906058262c6SKonstantin Belousov 		evp = &ev;
2907058262c6SKonstantin Belousov 	}
2908058262c6SKonstantin Belousov 	return (kern_kmq_notify(td, uap->mqd, evp));
2909058262c6SKonstantin Belousov }
2910058262c6SKonstantin Belousov 
2911afde2b65SKonstantin Belousov static struct syscall_helper_data mq32_syscalls[] = {
2912afde2b65SKonstantin Belousov 	SYSCALL32_INIT_HELPER(freebsd32_kmq_open),
2913afde2b65SKonstantin Belousov 	SYSCALL32_INIT_HELPER(freebsd32_kmq_setattr),
2914afde2b65SKonstantin Belousov 	SYSCALL32_INIT_HELPER(freebsd32_kmq_timedsend),
2915afde2b65SKonstantin Belousov 	SYSCALL32_INIT_HELPER(freebsd32_kmq_timedreceive),
2916058262c6SKonstantin Belousov 	SYSCALL32_INIT_HELPER(freebsd32_kmq_notify),
29178451d0ddSKip Macy 	SYSCALL32_INIT_HELPER_COMPAT(kmq_unlink),
2918afde2b65SKonstantin Belousov 	SYSCALL_INIT_LAST
2919afde2b65SKonstantin Belousov };
2920afde2b65SKonstantin Belousov #endif
2921afde2b65SKonstantin Belousov 
2922afde2b65SKonstantin Belousov static int
2923afde2b65SKonstantin Belousov mqinit(void)
2924afde2b65SKonstantin Belousov {
2925afde2b65SKonstantin Belousov 	int error;
2926afde2b65SKonstantin Belousov 
2927e015b1abSMateusz Guzik 	error = syscall_helper_register(mq_syscalls, SY_THR_STATIC_KLD);
2928afde2b65SKonstantin Belousov 	if (error != 0)
2929afde2b65SKonstantin Belousov 		return (error);
2930afde2b65SKonstantin Belousov #ifdef COMPAT_FREEBSD32
2931e015b1abSMateusz Guzik 	error = syscall32_helper_register(mq32_syscalls, SY_THR_STATIC_KLD);
2932afde2b65SKonstantin Belousov 	if (error != 0)
2933afde2b65SKonstantin Belousov 		return (error);
2934afde2b65SKonstantin Belousov #endif
2935afde2b65SKonstantin Belousov 	return (0);
2936afde2b65SKonstantin Belousov }
2937afde2b65SKonstantin Belousov 
2938afde2b65SKonstantin Belousov static int
2939afde2b65SKonstantin Belousov mqunload(void)
2940afde2b65SKonstantin Belousov {
2941afde2b65SKonstantin Belousov 
2942afde2b65SKonstantin Belousov #ifdef COMPAT_FREEBSD32
2943afde2b65SKonstantin Belousov 	syscall32_helper_unregister(mq32_syscalls);
2944afde2b65SKonstantin Belousov #endif
2945afde2b65SKonstantin Belousov 	syscall_helper_unregister(mq_syscalls);
2946afde2b65SKonstantin Belousov 	return (0);
2947afde2b65SKonstantin Belousov }
2948afde2b65SKonstantin Belousov 
2949afde2b65SKonstantin Belousov static int
2950afde2b65SKonstantin Belousov mq_modload(struct module *module, int cmd, void *arg)
2951afde2b65SKonstantin Belousov {
2952afde2b65SKonstantin Belousov 	int error = 0;
2953afde2b65SKonstantin Belousov 
2954afde2b65SKonstantin Belousov 	error = vfs_modevent(module, cmd, arg);
2955afde2b65SKonstantin Belousov 	if (error != 0)
2956afde2b65SKonstantin Belousov 		return (error);
2957afde2b65SKonstantin Belousov 
2958afde2b65SKonstantin Belousov 	switch (cmd) {
2959afde2b65SKonstantin Belousov 	case MOD_LOAD:
2960afde2b65SKonstantin Belousov 		error = mqinit();
2961afde2b65SKonstantin Belousov 		if (error != 0)
2962afde2b65SKonstantin Belousov 			mqunload();
2963afde2b65SKonstantin Belousov 		break;
2964afde2b65SKonstantin Belousov 	case MOD_UNLOAD:
2965afde2b65SKonstantin Belousov 		error = mqunload();
2966afde2b65SKonstantin Belousov 		break;
2967afde2b65SKonstantin Belousov 	default:
2968afde2b65SKonstantin Belousov 		break;
2969afde2b65SKonstantin Belousov 	}
2970afde2b65SKonstantin Belousov 	return (error);
2971afde2b65SKonstantin Belousov }
2972afde2b65SKonstantin Belousov 
2973afde2b65SKonstantin Belousov static moduledata_t mqueuefs_mod = {
2974afde2b65SKonstantin Belousov 	"mqueuefs",
2975afde2b65SKonstantin Belousov 	mq_modload,
2976afde2b65SKonstantin Belousov 	&mqueuefs_vfsconf
2977afde2b65SKonstantin Belousov };
2978afde2b65SKonstantin Belousov DECLARE_MODULE(mqueuefs, mqueuefs_mod, SI_SUB_VFS, SI_ORDER_MIDDLE);
297947bf2cf9SDavid Xu MODULE_VERSION(mqueuefs, 1);
2980