xref: /freebsd/sys/kern/uipc_mqueue.c (revision 1fe80828e72af1c4f17f946eb8fe0d4070ae22c7)
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 
48afde2b65SKonstantin Belousov #include "opt_compat.h"
49afde2b65SKonstantin Belousov 
50655291f2SDavid Xu #include <sys/param.h>
51655291f2SDavid Xu #include <sys/kernel.h>
52655291f2SDavid Xu #include <sys/systm.h>
53655291f2SDavid Xu #include <sys/limits.h>
54655291f2SDavid Xu #include <sys/buf.h>
55655291f2SDavid Xu #include <sys/dirent.h>
56655291f2SDavid Xu #include <sys/event.h>
57b2f92ef9SDavid Xu #include <sys/eventhandler.h>
58655291f2SDavid Xu #include <sys/fcntl.h>
59655291f2SDavid Xu #include <sys/file.h>
605ee2d4acSDavid Xu #include <sys/filedesc.h>
61655291f2SDavid Xu #include <sys/lock.h>
62655291f2SDavid Xu #include <sys/malloc.h>
63655291f2SDavid Xu #include <sys/module.h>
64655291f2SDavid Xu #include <sys/mount.h>
65655291f2SDavid Xu #include <sys/mqueue.h>
66655291f2SDavid Xu #include <sys/mutex.h>
67655291f2SDavid Xu #include <sys/namei.h>
686aeb05d7STom Rhodes #include <sys/posix4.h>
69655291f2SDavid Xu #include <sys/poll.h>
70acd3428bSRobert Watson #include <sys/priv.h>
71655291f2SDavid Xu #include <sys/proc.h>
72655291f2SDavid Xu #include <sys/queue.h>
73655291f2SDavid Xu #include <sys/sysproto.h>
74655291f2SDavid Xu #include <sys/stat.h>
75655291f2SDavid Xu #include <sys/syscall.h>
76655291f2SDavid Xu #include <sys/syscallsubr.h>
775ee2d4acSDavid Xu #include <sys/sysent.h>
78655291f2SDavid Xu #include <sys/sx.h>
79655291f2SDavid Xu #include <sys/sysctl.h>
80655291f2SDavid Xu #include <sys/taskqueue.h>
81655291f2SDavid Xu #include <sys/unistd.h>
82655291f2SDavid Xu #include <sys/vnode.h>
83655291f2SDavid Xu #include <machine/atomic.h>
84655291f2SDavid Xu 
85de5b1952SAlexander Leidinger FEATURE(p1003_1b_mqueue, "POSIX P1003.1B message queues support");
86de5b1952SAlexander Leidinger 
87655291f2SDavid Xu /*
88655291f2SDavid Xu  * Limits and constants
89655291f2SDavid Xu  */
90655291f2SDavid Xu #define	MQFS_NAMELEN		NAME_MAX
91655291f2SDavid Xu #define MQFS_DELEN		(8 + MQFS_NAMELEN)
92655291f2SDavid Xu 
93655291f2SDavid Xu /* node types */
94655291f2SDavid Xu typedef enum {
95655291f2SDavid Xu 	mqfstype_none = 0,
96655291f2SDavid Xu 	mqfstype_root,
97655291f2SDavid Xu 	mqfstype_dir,
98655291f2SDavid Xu 	mqfstype_this,
99655291f2SDavid Xu 	mqfstype_parent,
100655291f2SDavid Xu 	mqfstype_file,
101655291f2SDavid Xu 	mqfstype_symlink,
102655291f2SDavid Xu } mqfs_type_t;
103655291f2SDavid Xu 
104655291f2SDavid Xu struct mqfs_node;
105655291f2SDavid Xu 
106655291f2SDavid Xu /*
107655291f2SDavid Xu  * mqfs_info: describes a mqfs instance
108655291f2SDavid Xu  */
109655291f2SDavid Xu struct mqfs_info {
110655291f2SDavid Xu 	struct sx		mi_lock;
111655291f2SDavid Xu 	struct mqfs_node	*mi_root;
112655291f2SDavid Xu 	struct unrhdr		*mi_unrhdr;
113655291f2SDavid Xu };
114655291f2SDavid Xu 
115655291f2SDavid Xu struct mqfs_vdata {
116655291f2SDavid Xu 	LIST_ENTRY(mqfs_vdata)	mv_link;
117655291f2SDavid Xu 	struct mqfs_node	*mv_node;
118655291f2SDavid Xu 	struct vnode		*mv_vnode;
119655291f2SDavid Xu 	struct task		mv_task;
120655291f2SDavid Xu };
121655291f2SDavid Xu 
122655291f2SDavid Xu /*
123655291f2SDavid Xu  * mqfs_node: describes a node (file or directory) within a mqfs
124655291f2SDavid Xu  */
125655291f2SDavid Xu struct mqfs_node {
126655291f2SDavid Xu 	char			mn_name[MQFS_NAMELEN+1];
127655291f2SDavid Xu 	struct mqfs_info	*mn_info;
128655291f2SDavid Xu 	struct mqfs_node	*mn_parent;
129655291f2SDavid Xu 	LIST_HEAD(,mqfs_node)	mn_children;
130655291f2SDavid Xu 	LIST_ENTRY(mqfs_node)	mn_sibling;
131655291f2SDavid Xu 	LIST_HEAD(,mqfs_vdata)	mn_vnodes;
132655291f2SDavid Xu 	int			mn_refcount;
133655291f2SDavid Xu 	mqfs_type_t		mn_type;
134655291f2SDavid Xu 	int			mn_deleted;
13560ae52f7SEd Schouten 	uint32_t		mn_fileno;
136655291f2SDavid Xu 	void			*mn_data;
137655291f2SDavid Xu 	struct timespec		mn_birth;
138655291f2SDavid Xu 	struct timespec		mn_ctime;
139655291f2SDavid Xu 	struct timespec		mn_atime;
140655291f2SDavid Xu 	struct timespec		mn_mtime;
141655291f2SDavid Xu 	uid_t			mn_uid;
142655291f2SDavid Xu 	gid_t			mn_gid;
143655291f2SDavid Xu 	int			mn_mode;
144655291f2SDavid Xu };
145655291f2SDavid Xu 
146655291f2SDavid Xu #define	VTON(vp)	(((struct mqfs_vdata *)((vp)->v_data))->mv_node)
147655291f2SDavid Xu #define VTOMQ(vp) 	((struct mqueue *)(VTON(vp)->mn_data))
148655291f2SDavid Xu #define	VFSTOMQFS(m)	((struct mqfs_info *)((m)->mnt_data))
149b2f92ef9SDavid Xu #define	FPTOMQ(fp)	((struct mqueue *)(((struct mqfs_node *) \
150b2f92ef9SDavid Xu 				(fp)->f_data)->mn_data))
151b2f92ef9SDavid Xu 
152655291f2SDavid Xu TAILQ_HEAD(msgq, mqueue_msg);
153655291f2SDavid Xu 
154655291f2SDavid Xu struct mqueue;
155b2f92ef9SDavid Xu 
156b2f92ef9SDavid Xu struct mqueue_notifier {
157b2f92ef9SDavid Xu 	LIST_ENTRY(mqueue_notifier)	nt_link;
158b2f92ef9SDavid Xu 	struct sigevent			nt_sigev;
159b2f92ef9SDavid Xu 	ksiginfo_t			nt_ksi;
160b2f92ef9SDavid Xu 	struct proc			*nt_proc;
161655291f2SDavid Xu };
162655291f2SDavid Xu 
163655291f2SDavid Xu struct mqueue {
164655291f2SDavid Xu 	struct mtx	mq_mutex;
165655291f2SDavid Xu 	int		mq_flags;
166655291f2SDavid Xu 	long		mq_maxmsg;
167655291f2SDavid Xu 	long		mq_msgsize;
168655291f2SDavid Xu 	long		mq_curmsgs;
169655291f2SDavid Xu 	long		mq_totalbytes;
170655291f2SDavid Xu 	struct msgq	mq_msgq;
171655291f2SDavid Xu 	int		mq_receivers;
172655291f2SDavid Xu 	int		mq_senders;
173655291f2SDavid Xu 	struct selinfo	mq_rsel;
174655291f2SDavid Xu 	struct selinfo	mq_wsel;
175b2f92ef9SDavid Xu 	struct mqueue_notifier	*mq_notifier;
176655291f2SDavid Xu };
177655291f2SDavid Xu 
178655291f2SDavid Xu #define	MQ_RSEL		0x01
179655291f2SDavid Xu #define	MQ_WSEL		0x02
180655291f2SDavid Xu 
181655291f2SDavid Xu struct mqueue_msg {
182655291f2SDavid Xu 	TAILQ_ENTRY(mqueue_msg)	msg_link;
183655291f2SDavid Xu 	unsigned int	msg_prio;
184655291f2SDavid Xu 	unsigned int	msg_size;
185655291f2SDavid Xu 	/* following real data... */
186655291f2SDavid Xu };
187655291f2SDavid Xu 
188655291f2SDavid Xu SYSCTL_NODE(_kern, OID_AUTO, mqueue, CTLFLAG_RW, 0,
189655291f2SDavid Xu 	"POSIX real time message queue");
190655291f2SDavid Xu 
191655291f2SDavid Xu static int	default_maxmsg  = 10;
192655291f2SDavid Xu static int	default_msgsize = 1024;
193655291f2SDavid Xu 
1949947b459SDavid Xu static int	maxmsg = 100;
195655291f2SDavid Xu SYSCTL_INT(_kern_mqueue, OID_AUTO, maxmsg, CTLFLAG_RW,
196655291f2SDavid Xu     &maxmsg, 0, "Default maximum messages in queue");
197655291f2SDavid Xu static int	maxmsgsize = 16384;
198655291f2SDavid Xu SYSCTL_INT(_kern_mqueue, OID_AUTO, maxmsgsize, CTLFLAG_RW,
199655291f2SDavid Xu     &maxmsgsize, 0, "Default maximum message size");
200655291f2SDavid Xu static int	maxmq = 100;
201655291f2SDavid Xu SYSCTL_INT(_kern_mqueue, OID_AUTO, maxmq, CTLFLAG_RW,
202655291f2SDavid Xu     &maxmq, 0, "maximum message queues");
203655291f2SDavid Xu static int	curmq = 0;
204655291f2SDavid Xu SYSCTL_INT(_kern_mqueue, OID_AUTO, curmq, CTLFLAG_RW,
205655291f2SDavid Xu     &curmq, 0, "current message queue number");
206655291f2SDavid Xu static int	unloadable = 0;
207655291f2SDavid Xu static MALLOC_DEFINE(M_MQUEUEDATA, "mqdata", "mqueue data");
208655291f2SDavid Xu 
209b2f92ef9SDavid Xu static eventhandler_tag exit_tag;
210b2f92ef9SDavid Xu 
211655291f2SDavid Xu /* Only one instance per-system */
212655291f2SDavid Xu static struct mqfs_info		mqfs_data;
213655291f2SDavid Xu static uma_zone_t		mqnode_zone;
214655291f2SDavid Xu static uma_zone_t		mqueue_zone;
215655291f2SDavid Xu static uma_zone_t		mvdata_zone;
216b2f92ef9SDavid Xu static uma_zone_t		mqnoti_zone;
217655291f2SDavid Xu static struct vop_vector	mqfs_vnodeops;
218655291f2SDavid Xu static struct fileops		mqueueops;
219655291f2SDavid Xu 
220655291f2SDavid Xu /*
221655291f2SDavid Xu  * Directory structure construction and manipulation
222655291f2SDavid Xu  */
223655291f2SDavid Xu #ifdef notyet
224655291f2SDavid Xu static struct mqfs_node	*mqfs_create_dir(struct mqfs_node *parent,
225ba0360b1SDavid Xu 	const char *name, int namelen, struct ucred *cred, int mode);
226102178d0SDavid Xu static struct mqfs_node	*mqfs_create_link(struct mqfs_node *parent,
227ba0360b1SDavid Xu 	const char *name, int namelen, struct ucred *cred, int mode);
228655291f2SDavid Xu #endif
229655291f2SDavid Xu 
230655291f2SDavid Xu static struct mqfs_node	*mqfs_create_file(struct mqfs_node *parent,
231ba0360b1SDavid Xu 	const char *name, int namelen, struct ucred *cred, int mode);
232655291f2SDavid Xu static int	mqfs_destroy(struct mqfs_node *mn);
233655291f2SDavid Xu static void	mqfs_fileno_alloc(struct mqfs_info *mi, struct mqfs_node *mn);
234655291f2SDavid Xu static void	mqfs_fileno_free(struct mqfs_info *mi, struct mqfs_node *mn);
235655291f2SDavid Xu static int	mqfs_allocv(struct mount *mp, struct vnode **vpp, struct mqfs_node *pn);
236655291f2SDavid Xu 
237655291f2SDavid Xu /*
238655291f2SDavid Xu  * Message queue construction and maniplation
239655291f2SDavid Xu  */
240655291f2SDavid Xu static struct mqueue	*mqueue_alloc(const struct mq_attr *attr);
241655291f2SDavid Xu static void	mqueue_free(struct mqueue *mq);
242655291f2SDavid Xu static int	mqueue_send(struct mqueue *mq, const char *msg_ptr,
243655291f2SDavid Xu 			size_t msg_len, unsigned msg_prio, int waitok,
244655291f2SDavid Xu 			const struct timespec *abs_timeout);
245655291f2SDavid Xu static int	mqueue_receive(struct mqueue *mq, char *msg_ptr,
246655291f2SDavid Xu 			size_t msg_len, unsigned *msg_prio, int waitok,
247655291f2SDavid Xu 			const struct timespec *abs_timeout);
248655291f2SDavid Xu static int	_mqueue_send(struct mqueue *mq, struct mqueue_msg *msg,
249655291f2SDavid Xu 			int timo);
250655291f2SDavid Xu static int	_mqueue_recv(struct mqueue *mq, struct mqueue_msg **msg,
251655291f2SDavid Xu 			int timo);
252655291f2SDavid Xu static void	mqueue_send_notification(struct mqueue *mq);
253b2f92ef9SDavid Xu static void	mqueue_fdclose(struct thread *td, int fd, struct file *fp);
254b2f92ef9SDavid Xu static void	mq_proc_exit(void *arg, struct proc *p);
255655291f2SDavid Xu 
256b2f92ef9SDavid Xu /*
257b2f92ef9SDavid Xu  * kqueue filters
258b2f92ef9SDavid Xu  */
259655291f2SDavid Xu static void	filt_mqdetach(struct knote *kn);
260655291f2SDavid Xu static int	filt_mqread(struct knote *kn, long hint);
261655291f2SDavid Xu static int	filt_mqwrite(struct knote *kn, long hint);
262655291f2SDavid Xu 
263e76d823bSRobert Watson struct filterops mq_rfiltops = {
264e76d823bSRobert Watson 	.f_isfd = 1,
265e76d823bSRobert Watson 	.f_detach = filt_mqdetach,
266e76d823bSRobert Watson 	.f_event = filt_mqread,
267e76d823bSRobert Watson };
268e76d823bSRobert Watson struct filterops mq_wfiltops = {
269e76d823bSRobert Watson 	.f_isfd = 1,
270e76d823bSRobert Watson 	.f_detach = filt_mqdetach,
271e76d823bSRobert Watson 	.f_event = filt_mqwrite,
272e76d823bSRobert Watson };
273655291f2SDavid Xu 
274655291f2SDavid Xu /*
275655291f2SDavid Xu  * Initialize fileno bitmap
276655291f2SDavid Xu  */
277655291f2SDavid Xu static void
278655291f2SDavid Xu mqfs_fileno_init(struct mqfs_info *mi)
279655291f2SDavid Xu {
280655291f2SDavid Xu 	struct unrhdr *up;
281655291f2SDavid Xu 
282655291f2SDavid Xu 	up = new_unrhdr(1, INT_MAX, NULL);
283655291f2SDavid Xu 	mi->mi_unrhdr = up;
284655291f2SDavid Xu }
285655291f2SDavid Xu 
286655291f2SDavid Xu /*
287655291f2SDavid Xu  * Tear down fileno bitmap
288655291f2SDavid Xu  */
289655291f2SDavid Xu static void
290655291f2SDavid Xu mqfs_fileno_uninit(struct mqfs_info *mi)
291655291f2SDavid Xu {
292655291f2SDavid Xu 	struct unrhdr *up;
293655291f2SDavid Xu 
294655291f2SDavid Xu 	up = mi->mi_unrhdr;
295655291f2SDavid Xu 	mi->mi_unrhdr = NULL;
296655291f2SDavid Xu 	delete_unrhdr(up);
297655291f2SDavid Xu }
298655291f2SDavid Xu 
299655291f2SDavid Xu /*
300655291f2SDavid Xu  * Allocate a file number
301655291f2SDavid Xu  */
302b2f92ef9SDavid Xu static void
303655291f2SDavid Xu mqfs_fileno_alloc(struct mqfs_info *mi, struct mqfs_node *mn)
304655291f2SDavid Xu {
305655291f2SDavid Xu 	/* make sure our parent has a file number */
306655291f2SDavid Xu 	if (mn->mn_parent && !mn->mn_parent->mn_fileno)
307655291f2SDavid Xu 		mqfs_fileno_alloc(mi, mn->mn_parent);
308655291f2SDavid Xu 
309655291f2SDavid Xu 	switch (mn->mn_type) {
310655291f2SDavid Xu 	case mqfstype_root:
311655291f2SDavid Xu 	case mqfstype_dir:
312655291f2SDavid Xu 	case mqfstype_file:
313655291f2SDavid Xu 	case mqfstype_symlink:
314655291f2SDavid Xu 		mn->mn_fileno = alloc_unr(mi->mi_unrhdr);
315655291f2SDavid Xu 		break;
316655291f2SDavid Xu 	case mqfstype_this:
317655291f2SDavid Xu 		KASSERT(mn->mn_parent != NULL,
318655291f2SDavid Xu 		    ("mqfstype_this node has no parent"));
319655291f2SDavid Xu 		mn->mn_fileno = mn->mn_parent->mn_fileno;
320655291f2SDavid Xu 		break;
321655291f2SDavid Xu 	case mqfstype_parent:
322655291f2SDavid Xu 		KASSERT(mn->mn_parent != NULL,
323655291f2SDavid Xu 		    ("mqfstype_parent node has no parent"));
324655291f2SDavid Xu 		if (mn->mn_parent == mi->mi_root) {
325655291f2SDavid Xu 			mn->mn_fileno = mn->mn_parent->mn_fileno;
326655291f2SDavid Xu 			break;
327655291f2SDavid Xu 		}
328655291f2SDavid Xu 		KASSERT(mn->mn_parent->mn_parent != NULL,
329655291f2SDavid Xu 		    ("mqfstype_parent node has no grandparent"));
330655291f2SDavid Xu 		mn->mn_fileno = mn->mn_parent->mn_parent->mn_fileno;
331655291f2SDavid Xu 		break;
332655291f2SDavid Xu 	default:
333655291f2SDavid Xu 		KASSERT(0,
334655291f2SDavid Xu 		    ("mqfs_fileno_alloc() called for unknown type node: %d",
335655291f2SDavid Xu 			mn->mn_type));
336655291f2SDavid Xu 		break;
337655291f2SDavid Xu 	}
338655291f2SDavid Xu }
339655291f2SDavid Xu 
340655291f2SDavid Xu /*
341655291f2SDavid Xu  * Release a file number
342655291f2SDavid Xu  */
343b2f92ef9SDavid Xu static void
344655291f2SDavid Xu mqfs_fileno_free(struct mqfs_info *mi, struct mqfs_node *mn)
345655291f2SDavid Xu {
346655291f2SDavid Xu 	switch (mn->mn_type) {
347655291f2SDavid Xu 	case mqfstype_root:
348655291f2SDavid Xu 	case mqfstype_dir:
349655291f2SDavid Xu 	case mqfstype_file:
350655291f2SDavid Xu 	case mqfstype_symlink:
351655291f2SDavid Xu 		free_unr(mi->mi_unrhdr, mn->mn_fileno);
352655291f2SDavid Xu 		break;
353655291f2SDavid Xu 	case mqfstype_this:
354655291f2SDavid Xu 	case mqfstype_parent:
355655291f2SDavid Xu 		/* ignore these, as they don't "own" their file number */
356655291f2SDavid Xu 		break;
357655291f2SDavid Xu 	default:
358655291f2SDavid Xu 		KASSERT(0,
359655291f2SDavid Xu 		    ("mqfs_fileno_free() called for unknown type node: %d",
360655291f2SDavid Xu 			mn->mn_type));
361655291f2SDavid Xu 		break;
362655291f2SDavid Xu 	}
363655291f2SDavid Xu }
364655291f2SDavid Xu 
365655291f2SDavid Xu static __inline struct mqfs_node *
366655291f2SDavid Xu mqnode_alloc(void)
367655291f2SDavid Xu {
368655291f2SDavid Xu 	return uma_zalloc(mqnode_zone, M_WAITOK | M_ZERO);
369655291f2SDavid Xu }
370655291f2SDavid Xu 
371655291f2SDavid Xu static __inline void
372655291f2SDavid Xu mqnode_free(struct mqfs_node *node)
373655291f2SDavid Xu {
374655291f2SDavid Xu 	uma_zfree(mqnode_zone, node);
375655291f2SDavid Xu }
376655291f2SDavid Xu 
377655291f2SDavid Xu static __inline void
378655291f2SDavid Xu mqnode_addref(struct mqfs_node *node)
379655291f2SDavid Xu {
380655291f2SDavid Xu 	atomic_fetchadd_int(&node->mn_refcount, 1);
381655291f2SDavid Xu }
382655291f2SDavid Xu 
383655291f2SDavid Xu static __inline void
384655291f2SDavid Xu mqnode_release(struct mqfs_node *node)
385655291f2SDavid Xu {
386fbc48e97SDavid Xu 	struct mqfs_info *mqfs;
387655291f2SDavid Xu 	int old, exp;
388655291f2SDavid Xu 
389fbc48e97SDavid Xu 	mqfs = node->mn_info;
390655291f2SDavid Xu 	old = atomic_fetchadd_int(&node->mn_refcount, -1);
391655291f2SDavid Xu 	if (node->mn_type == mqfstype_dir ||
392655291f2SDavid Xu 	    node->mn_type == mqfstype_root)
393655291f2SDavid Xu 		exp = 3; /* include . and .. */
394655291f2SDavid Xu 	else
395655291f2SDavid Xu 		exp = 1;
396fbc48e97SDavid Xu 	if (old == exp) {
397fbc48e97SDavid Xu 		int locked = sx_xlocked(&mqfs->mi_lock);
398fbc48e97SDavid Xu 		if (!locked)
399fbc48e97SDavid Xu 			sx_xlock(&mqfs->mi_lock);
400655291f2SDavid Xu 		mqfs_destroy(node);
401fbc48e97SDavid Xu 		if (!locked)
402fbc48e97SDavid Xu 			sx_xunlock(&mqfs->mi_lock);
403fbc48e97SDavid Xu 	}
404655291f2SDavid Xu }
405655291f2SDavid Xu 
406655291f2SDavid Xu /*
407655291f2SDavid Xu  * Add a node to a directory
408655291f2SDavid Xu  */
409655291f2SDavid Xu static int
410655291f2SDavid Xu mqfs_add_node(struct mqfs_node *parent, struct mqfs_node *node)
411655291f2SDavid Xu {
412655291f2SDavid Xu 	KASSERT(parent != NULL, ("%s(): parent is NULL", __func__));
413655291f2SDavid Xu 	KASSERT(parent->mn_info != NULL,
414655291f2SDavid Xu 	    ("%s(): parent has no mn_info", __func__));
415655291f2SDavid Xu 	KASSERT(parent->mn_type == mqfstype_dir ||
416655291f2SDavid Xu 	    parent->mn_type == mqfstype_root,
417655291f2SDavid Xu 	    ("%s(): parent is not a directory", __func__));
418655291f2SDavid Xu 
419655291f2SDavid Xu 	node->mn_info = parent->mn_info;
420655291f2SDavid Xu 	node->mn_parent = parent;
421655291f2SDavid Xu 	LIST_INIT(&node->mn_children);
422655291f2SDavid Xu 	LIST_INIT(&node->mn_vnodes);
423655291f2SDavid Xu 	LIST_INSERT_HEAD(&parent->mn_children, node, mn_sibling);
424655291f2SDavid Xu 	mqnode_addref(parent);
425655291f2SDavid Xu 	return (0);
426655291f2SDavid Xu }
427655291f2SDavid Xu 
428ba0360b1SDavid Xu static struct mqfs_node *
429ba0360b1SDavid Xu mqfs_create_node(const char *name, int namelen, struct ucred *cred, int mode,
430ba0360b1SDavid Xu 	int nodetype)
431ba0360b1SDavid Xu {
432ba0360b1SDavid Xu 	struct mqfs_node *node;
433ba0360b1SDavid Xu 
434ba0360b1SDavid Xu 	node = mqnode_alloc();
435ba0360b1SDavid Xu 	strncpy(node->mn_name, name, namelen);
436ba0360b1SDavid Xu 	node->mn_type = nodetype;
437ba0360b1SDavid Xu 	node->mn_refcount = 1;
438fbc48e97SDavid Xu 	vfs_timestamp(&node->mn_birth);
439ba0360b1SDavid Xu 	node->mn_ctime = node->mn_atime = node->mn_mtime
440ba0360b1SDavid Xu 		= node->mn_birth;
441ba0360b1SDavid Xu 	node->mn_uid = cred->cr_uid;
442ba0360b1SDavid Xu 	node->mn_gid = cred->cr_gid;
443ba0360b1SDavid Xu 	node->mn_mode = mode;
444ba0360b1SDavid Xu 	return (node);
445ba0360b1SDavid Xu }
446ba0360b1SDavid Xu 
447ba0360b1SDavid Xu /*
448ba0360b1SDavid Xu  * Create a file
449ba0360b1SDavid Xu  */
450ba0360b1SDavid Xu static struct mqfs_node *
451ba0360b1SDavid Xu mqfs_create_file(struct mqfs_node *parent, const char *name, int namelen,
452ba0360b1SDavid Xu 	struct ucred *cred, int mode)
453ba0360b1SDavid Xu {
454ba0360b1SDavid Xu 	struct mqfs_node *node;
455ba0360b1SDavid Xu 
456ba0360b1SDavid Xu 	node = mqfs_create_node(name, namelen, cred, mode, mqfstype_file);
457ba0360b1SDavid Xu 	if (mqfs_add_node(parent, node) != 0) {
458ba0360b1SDavid Xu 		mqnode_free(node);
459ba0360b1SDavid Xu 		return (NULL);
460ba0360b1SDavid Xu 	}
461ba0360b1SDavid Xu 	return (node);
462ba0360b1SDavid Xu }
463ba0360b1SDavid Xu 
464655291f2SDavid Xu /*
465655291f2SDavid Xu  * Add . and .. to a directory
466655291f2SDavid Xu  */
467655291f2SDavid Xu static int
468655291f2SDavid Xu mqfs_fixup_dir(struct mqfs_node *parent)
469655291f2SDavid Xu {
470655291f2SDavid Xu 	struct mqfs_node *dir;
471655291f2SDavid Xu 
472655291f2SDavid Xu 	dir = mqnode_alloc();
473655291f2SDavid Xu 	dir->mn_name[0] = '.';
474655291f2SDavid Xu 	dir->mn_type = mqfstype_this;
475655291f2SDavid Xu 	dir->mn_refcount = 1;
476655291f2SDavid Xu 	if (mqfs_add_node(parent, dir) != 0) {
477655291f2SDavid Xu 		mqnode_free(dir);
478655291f2SDavid Xu 		return (-1);
479655291f2SDavid Xu 	}
480655291f2SDavid Xu 
481655291f2SDavid Xu 	dir = mqnode_alloc();
482655291f2SDavid Xu 	dir->mn_name[0] = dir->mn_name[1] = '.';
483655291f2SDavid Xu 	dir->mn_type = mqfstype_parent;
484655291f2SDavid Xu 	dir->mn_refcount = 1;
485655291f2SDavid Xu 
486655291f2SDavid Xu 	if (mqfs_add_node(parent, dir) != 0) {
487655291f2SDavid Xu 		mqnode_free(dir);
488655291f2SDavid Xu 		return (-1);
489655291f2SDavid Xu 	}
490655291f2SDavid Xu 
491655291f2SDavid Xu 	return (0);
492655291f2SDavid Xu }
493655291f2SDavid Xu 
494655291f2SDavid Xu #ifdef notyet
495655291f2SDavid Xu 
496655291f2SDavid Xu /*
497655291f2SDavid Xu  * Create a directory
498655291f2SDavid Xu  */
499b2f92ef9SDavid Xu static struct mqfs_node *
500ba0360b1SDavid Xu mqfs_create_dir(struct mqfs_node *parent, const char *name, int namelen,
501ba0360b1SDavid Xu 	struct ucred *cred, int mode)
502655291f2SDavid Xu {
503ba0360b1SDavid Xu 	struct mqfs_node *node;
504655291f2SDavid Xu 
505ba0360b1SDavid Xu 	node = mqfs_create_node(name, namelen, cred, mode, mqfstype_dir);
506ba0360b1SDavid Xu 	if (mqfs_add_node(parent, node) != 0) {
507ba0360b1SDavid Xu 		mqnode_free(node);
508655291f2SDavid Xu 		return (NULL);
509655291f2SDavid Xu 	}
510655291f2SDavid Xu 
511ba0360b1SDavid Xu 	if (mqfs_fixup_dir(node) != 0) {
512ba0360b1SDavid Xu 		mqfs_destroy(node);
513655291f2SDavid Xu 		return (NULL);
514655291f2SDavid Xu 	}
515ba0360b1SDavid Xu 	return (node);
516655291f2SDavid Xu }
517b2f92ef9SDavid Xu 
518b2f92ef9SDavid Xu /*
519b2f92ef9SDavid Xu  * Create a symlink
520b2f92ef9SDavid Xu  */
521b2f92ef9SDavid Xu static struct mqfs_node *
522ba0360b1SDavid Xu mqfs_create_link(struct mqfs_node *parent, const char *name, int namelen,
523ba0360b1SDavid Xu 	struct ucred *cred, int mode)
524b2f92ef9SDavid Xu {
525b2f92ef9SDavid Xu 	struct mqfs_node *node;
526b2f92ef9SDavid Xu 
527ba0360b1SDavid Xu 	node = mqfs_create_node(name, namelen, cred, mode, mqfstype_symlink);
528655291f2SDavid Xu 	if (mqfs_add_node(parent, node) != 0) {
529655291f2SDavid Xu 		mqnode_free(node);
530655291f2SDavid Xu 		return (NULL);
531655291f2SDavid Xu 	}
532655291f2SDavid Xu 	return (node);
533655291f2SDavid Xu }
534655291f2SDavid Xu 
535ba0360b1SDavid Xu #endif
536ba0360b1SDavid Xu 
537655291f2SDavid Xu /*
538655291f2SDavid Xu  * Destroy a node or a tree of nodes
539655291f2SDavid Xu  */
540b2f92ef9SDavid Xu static int
541655291f2SDavid Xu mqfs_destroy(struct mqfs_node *node)
542655291f2SDavid Xu {
543655291f2SDavid Xu 	struct mqfs_node *parent;
544655291f2SDavid Xu 
545655291f2SDavid Xu 	KASSERT(node != NULL,
546655291f2SDavid Xu 	    ("%s(): node is NULL", __func__));
547655291f2SDavid Xu 	KASSERT(node->mn_info != NULL,
548655291f2SDavid Xu 	    ("%s(): node has no mn_info", __func__));
549655291f2SDavid Xu 
550655291f2SDavid Xu 	/* destroy children */
551655291f2SDavid Xu 	if (node->mn_type == mqfstype_dir || node->mn_type == mqfstype_root)
552655291f2SDavid Xu 		while (! LIST_EMPTY(&node->mn_children))
553655291f2SDavid Xu 			mqfs_destroy(LIST_FIRST(&node->mn_children));
554655291f2SDavid Xu 
555655291f2SDavid Xu 	/* unlink from parent */
556655291f2SDavid Xu 	if ((parent = node->mn_parent) != NULL) {
557655291f2SDavid Xu 		KASSERT(parent->mn_info == node->mn_info,
558655291f2SDavid Xu 		    ("%s(): parent has different mn_info", __func__));
559655291f2SDavid Xu 		LIST_REMOVE(node, mn_sibling);
560655291f2SDavid Xu 	}
561655291f2SDavid Xu 
562655291f2SDavid Xu 	if (node->mn_fileno != 0)
563655291f2SDavid Xu 		mqfs_fileno_free(node->mn_info, node);
564655291f2SDavid Xu 	if (node->mn_data != NULL)
565655291f2SDavid Xu 		mqueue_free(node->mn_data);
566655291f2SDavid Xu 	mqnode_free(node);
567655291f2SDavid Xu 	return (0);
568655291f2SDavid Xu }
569655291f2SDavid Xu 
570655291f2SDavid Xu /*
571655291f2SDavid Xu  * Mount a mqfs instance
572655291f2SDavid Xu  */
573655291f2SDavid Xu static int
574dfd233edSAttilio Rao mqfs_mount(struct mount *mp)
575655291f2SDavid Xu {
576655291f2SDavid Xu 	struct statfs *sbp;
577655291f2SDavid Xu 
578655291f2SDavid Xu 	if (mp->mnt_flag & MNT_UPDATE)
579655291f2SDavid Xu 		return (EOPNOTSUPP);
580655291f2SDavid Xu 
581655291f2SDavid Xu 	mp->mnt_data = &mqfs_data;
5825da56ddbSTor Egge 	MNT_ILOCK(mp);
583655291f2SDavid Xu 	mp->mnt_flag |= MNT_LOCAL;
5849da8a32aSDavid Xu 	mp->mnt_kern_flag |= MNTK_MPSAFE;
5855da56ddbSTor Egge 	MNT_IUNLOCK(mp);
586655291f2SDavid Xu 	vfs_getnewfsid(mp);
587655291f2SDavid Xu 
588655291f2SDavid Xu 	sbp = &mp->mnt_stat;
589655291f2SDavid Xu 	vfs_mountedfrom(mp, "mqueue");
590655291f2SDavid Xu 	sbp->f_bsize = PAGE_SIZE;
591655291f2SDavid Xu 	sbp->f_iosize = PAGE_SIZE;
592655291f2SDavid Xu 	sbp->f_blocks = 1;
593655291f2SDavid Xu 	sbp->f_bfree = 0;
594655291f2SDavid Xu 	sbp->f_bavail = 0;
595655291f2SDavid Xu 	sbp->f_files = 1;
596655291f2SDavid Xu 	sbp->f_ffree = 0;
597655291f2SDavid Xu 	return (0);
598655291f2SDavid Xu }
599655291f2SDavid Xu 
600655291f2SDavid Xu /*
601655291f2SDavid Xu  * Unmount a mqfs instance
602655291f2SDavid Xu  */
603655291f2SDavid Xu static int
604dfd233edSAttilio Rao mqfs_unmount(struct mount *mp, int mntflags)
605655291f2SDavid Xu {
606655291f2SDavid Xu 	int error;
607655291f2SDavid Xu 
608dfd233edSAttilio Rao 	error = vflush(mp, 0, (mntflags & MNT_FORCE) ?  FORCECLOSE : 0,
609dfd233edSAttilio Rao 	    curthread);
610655291f2SDavid Xu 	return (error);
611655291f2SDavid Xu }
612655291f2SDavid Xu 
613655291f2SDavid Xu /*
614655291f2SDavid Xu  * Return a root vnode
615655291f2SDavid Xu  */
616655291f2SDavid Xu static int
617dfd233edSAttilio Rao mqfs_root(struct mount *mp, int flags, struct vnode **vpp)
618655291f2SDavid Xu {
619655291f2SDavid Xu 	struct mqfs_info *mqfs;
620655291f2SDavid Xu 	int ret;
621655291f2SDavid Xu 
622655291f2SDavid Xu 	mqfs = VFSTOMQFS(mp);
623655291f2SDavid Xu 	ret = mqfs_allocv(mp, vpp, mqfs->mi_root);
624655291f2SDavid Xu 	return (ret);
625655291f2SDavid Xu }
626655291f2SDavid Xu 
627655291f2SDavid Xu /*
628655291f2SDavid Xu  * Return filesystem stats
629655291f2SDavid Xu  */
630655291f2SDavid Xu static int
631dfd233edSAttilio Rao mqfs_statfs(struct mount *mp, struct statfs *sbp)
632655291f2SDavid Xu {
633655291f2SDavid Xu 	/* XXX update statistics */
634655291f2SDavid Xu 	return (0);
635655291f2SDavid Xu }
636655291f2SDavid Xu 
637655291f2SDavid Xu /*
638655291f2SDavid Xu  * Initialize a mqfs instance
639655291f2SDavid Xu  */
640655291f2SDavid Xu static int
641655291f2SDavid Xu mqfs_init(struct vfsconf *vfc)
642655291f2SDavid Xu {
643655291f2SDavid Xu 	struct mqfs_node *root;
644655291f2SDavid Xu 	struct mqfs_info *mi;
645655291f2SDavid Xu 
646655291f2SDavid Xu 	mqnode_zone = uma_zcreate("mqnode", sizeof(struct mqfs_node),
647655291f2SDavid Xu 		NULL, NULL, NULL, NULL, UMA_ALIGN_PTR, 0);
648655291f2SDavid Xu 	mqueue_zone = uma_zcreate("mqueue", sizeof(struct mqueue),
649655291f2SDavid Xu 		NULL, NULL, NULL, NULL, UMA_ALIGN_PTR, 0);
650655291f2SDavid Xu 	mvdata_zone = uma_zcreate("mvdata",
651655291f2SDavid Xu 		sizeof(struct mqfs_vdata), NULL, NULL, NULL,
652655291f2SDavid Xu 		NULL, UMA_ALIGN_PTR, 0);
653b2f92ef9SDavid Xu 	mqnoti_zone = uma_zcreate("mqnotifier", sizeof(struct mqueue_notifier),
654655291f2SDavid Xu 		NULL, NULL, NULL, NULL, UMA_ALIGN_PTR, 0);
655655291f2SDavid Xu 	mi = &mqfs_data;
656655291f2SDavid Xu 	sx_init(&mi->mi_lock, "mqfs lock");
657655291f2SDavid Xu 	/* set up the root diretory */
658ba0360b1SDavid Xu 	root = mqfs_create_node("/", 1, curthread->td_ucred, 01777,
659ba0360b1SDavid Xu 		mqfstype_root);
660655291f2SDavid Xu 	root->mn_info = mi;
661655291f2SDavid Xu 	LIST_INIT(&root->mn_children);
662655291f2SDavid Xu 	LIST_INIT(&root->mn_vnodes);
663655291f2SDavid Xu 	mi->mi_root = root;
664655291f2SDavid Xu 	mqfs_fileno_init(mi);
665655291f2SDavid Xu 	mqfs_fileno_alloc(mi, root);
666655291f2SDavid Xu 	mqfs_fixup_dir(root);
667b2f92ef9SDavid Xu 	exit_tag = EVENTHANDLER_REGISTER(process_exit, mq_proc_exit, NULL,
668b2f92ef9SDavid Xu 	    EVENTHANDLER_PRI_ANY);
669b2f92ef9SDavid Xu 	mq_fdclose = mqueue_fdclose;
6705ee2d4acSDavid Xu 	p31b_setcfg(CTL_P1003_1B_MESSAGE_PASSING, _POSIX_MESSAGE_PASSING);
671655291f2SDavid Xu 	return (0);
672655291f2SDavid Xu }
673655291f2SDavid Xu 
674655291f2SDavid Xu /*
675655291f2SDavid Xu  * Destroy a mqfs instance
676655291f2SDavid Xu  */
677655291f2SDavid Xu static int
678655291f2SDavid Xu mqfs_uninit(struct vfsconf *vfc)
679655291f2SDavid Xu {
680655291f2SDavid Xu 	struct mqfs_info *mi;
681655291f2SDavid Xu 
682655291f2SDavid Xu 	if (!unloadable)
683655291f2SDavid Xu 		return (EOPNOTSUPP);
684b2f92ef9SDavid Xu 	EVENTHANDLER_DEREGISTER(process_exit, exit_tag);
685655291f2SDavid Xu 	mi = &mqfs_data;
686655291f2SDavid Xu 	mqfs_destroy(mi->mi_root);
687655291f2SDavid Xu 	mi->mi_root = NULL;
688655291f2SDavid Xu 	mqfs_fileno_uninit(mi);
689655291f2SDavid Xu 	sx_destroy(&mi->mi_lock);
690b2f92ef9SDavid Xu 	uma_zdestroy(mqnode_zone);
691b2f92ef9SDavid Xu 	uma_zdestroy(mqueue_zone);
692b2f92ef9SDavid Xu 	uma_zdestroy(mvdata_zone);
693b2f92ef9SDavid Xu 	uma_zdestroy(mqnoti_zone);
694655291f2SDavid Xu 	return (0);
695655291f2SDavid Xu }
696655291f2SDavid Xu 
697655291f2SDavid Xu /*
698655291f2SDavid Xu  * task routine
699655291f2SDavid Xu  */
700655291f2SDavid Xu static void
701655291f2SDavid Xu do_recycle(void *context, int pending __unused)
702655291f2SDavid Xu {
703655291f2SDavid Xu 	struct vnode *vp = (struct vnode *)context;
704655291f2SDavid Xu 
705655291f2SDavid Xu 	vrecycle(vp, curthread);
706655291f2SDavid Xu 	vdrop(vp);
707655291f2SDavid Xu }
708655291f2SDavid Xu 
709655291f2SDavid Xu /*
710655291f2SDavid Xu  * Allocate a vnode
711655291f2SDavid Xu  */
712b2f92ef9SDavid Xu static int
713655291f2SDavid Xu mqfs_allocv(struct mount *mp, struct vnode **vpp, struct mqfs_node *pn)
714655291f2SDavid Xu {
715655291f2SDavid Xu 	struct mqfs_vdata *vd;
716fbc48e97SDavid Xu 	struct mqfs_info  *mqfs;
717fbc48e97SDavid Xu 	struct vnode *newvpp;
718655291f2SDavid Xu 	int error;
719655291f2SDavid Xu 
720fbc48e97SDavid Xu 	mqfs = pn->mn_info;
721fbc48e97SDavid Xu 	*vpp = NULL;
722fbc48e97SDavid Xu 	sx_xlock(&mqfs->mi_lock);
723655291f2SDavid Xu 	LIST_FOREACH(vd, &pn->mn_vnodes, mv_link) {
724fbc48e97SDavid Xu 		if (vd->mv_vnode->v_mount == mp) {
725fbc48e97SDavid Xu 			vhold(vd->mv_vnode);
726655291f2SDavid Xu 			break;
727655291f2SDavid Xu 		}
728fbc48e97SDavid Xu 	}
729655291f2SDavid Xu 
730655291f2SDavid Xu 	if (vd != NULL) {
731fbc48e97SDavid Xu found:
732655291f2SDavid Xu 		*vpp = vd->mv_vnode;
733fbc48e97SDavid Xu 		sx_xunlock(&mqfs->mi_lock);
734fbc48e97SDavid Xu 		error = vget(*vpp, LK_RETRY | LK_EXCLUSIVE, curthread);
735fbc48e97SDavid Xu 		vdrop(*vpp);
736fbc48e97SDavid Xu 		return (error);
737655291f2SDavid Xu 	}
738fbc48e97SDavid Xu 	sx_xunlock(&mqfs->mi_lock);
739655291f2SDavid Xu 
740fbc48e97SDavid Xu 	error = getnewvnode("mqueue", mp, &mqfs_vnodeops, &newvpp);
741655291f2SDavid Xu 	if (error)
742655291f2SDavid Xu 		return (error);
743fbc48e97SDavid Xu 	vn_lock(newvpp, LK_EXCLUSIVE | LK_RETRY);
744fbc48e97SDavid Xu 	error = insmntque(newvpp, mp);
745fbc48e97SDavid Xu 	if (error != 0)
74661b9d89fSTor Egge 		return (error);
747fbc48e97SDavid Xu 
748fbc48e97SDavid Xu 	sx_xlock(&mqfs->mi_lock);
749fbc48e97SDavid Xu 	/*
750fbc48e97SDavid Xu 	 * Check if it has already been allocated
751fbc48e97SDavid Xu 	 * while we were blocked.
752fbc48e97SDavid Xu 	 */
753fbc48e97SDavid Xu 	LIST_FOREACH(vd, &pn->mn_vnodes, mv_link) {
754fbc48e97SDavid Xu 		if (vd->mv_vnode->v_mount == mp) {
755fbc48e97SDavid Xu 			vhold(vd->mv_vnode);
756fbc48e97SDavid Xu 			sx_xunlock(&mqfs->mi_lock);
757fbc48e97SDavid Xu 
758fbc48e97SDavid Xu 			vgone(newvpp);
759fbc48e97SDavid Xu 			vput(newvpp);
760fbc48e97SDavid Xu 			goto found;
76161b9d89fSTor Egge 		}
762fbc48e97SDavid Xu 	}
763fbc48e97SDavid Xu 
764fbc48e97SDavid Xu 	*vpp = newvpp;
765fbc48e97SDavid Xu 
766655291f2SDavid Xu 	vd = uma_zalloc(mvdata_zone, M_WAITOK);
767655291f2SDavid Xu 	(*vpp)->v_data = vd;
768655291f2SDavid Xu 	vd->mv_vnode = *vpp;
769655291f2SDavid Xu 	vd->mv_node = pn;
770655291f2SDavid Xu 	TASK_INIT(&vd->mv_task, 0, do_recycle, *vpp);
771655291f2SDavid Xu 	LIST_INSERT_HEAD(&pn->mn_vnodes, vd, mv_link);
772655291f2SDavid Xu 	mqnode_addref(pn);
773655291f2SDavid Xu 	switch (pn->mn_type) {
774655291f2SDavid Xu 	case mqfstype_root:
775655291f2SDavid Xu 		(*vpp)->v_vflag = VV_ROOT;
776655291f2SDavid Xu 		/* fall through */
777655291f2SDavid Xu 	case mqfstype_dir:
778655291f2SDavid Xu 	case mqfstype_this:
779655291f2SDavid Xu 	case mqfstype_parent:
780655291f2SDavid Xu 		(*vpp)->v_type = VDIR;
781655291f2SDavid Xu 		break;
782655291f2SDavid Xu 	case mqfstype_file:
783655291f2SDavid Xu 		(*vpp)->v_type = VREG;
784655291f2SDavid Xu 		break;
785655291f2SDavid Xu 	case mqfstype_symlink:
786655291f2SDavid Xu 		(*vpp)->v_type = VLNK;
787655291f2SDavid Xu 		break;
788655291f2SDavid Xu 	case mqfstype_none:
789655291f2SDavid Xu 		KASSERT(0, ("mqfs_allocf called for null node\n"));
790655291f2SDavid Xu 	default:
791655291f2SDavid Xu 		panic("%s has unexpected type: %d", pn->mn_name, pn->mn_type);
792655291f2SDavid Xu 	}
793fbc48e97SDavid Xu 	sx_xunlock(&mqfs->mi_lock);
794655291f2SDavid Xu 	return (0);
795655291f2SDavid Xu }
796655291f2SDavid Xu 
797655291f2SDavid Xu /*
798655291f2SDavid Xu  * Search a directory entry
799655291f2SDavid Xu  */
800655291f2SDavid Xu static struct mqfs_node *
801655291f2SDavid Xu mqfs_search(struct mqfs_node *pd, const char *name, int len)
802655291f2SDavid Xu {
803655291f2SDavid Xu 	struct mqfs_node *pn;
804655291f2SDavid Xu 
805fbc48e97SDavid Xu 	sx_assert(&pd->mn_info->mi_lock, SX_LOCKED);
806655291f2SDavid Xu 	LIST_FOREACH(pn, &pd->mn_children, mn_sibling) {
8071cbae705SEd Schouten 		if (strncmp(pn->mn_name, name, len) == 0 &&
8081cbae705SEd Schouten 		    pn->mn_name[len] == '\0')
809655291f2SDavid Xu 			return (pn);
810655291f2SDavid Xu 	}
811655291f2SDavid Xu 	return (NULL);
812655291f2SDavid Xu }
813655291f2SDavid Xu 
814655291f2SDavid Xu /*
815873fbcd7SRobert Watson  * Look up a file or directory.
816655291f2SDavid Xu  */
817655291f2SDavid Xu static int
818655291f2SDavid Xu mqfs_lookupx(struct vop_cachedlookup_args *ap)
819655291f2SDavid Xu {
820655291f2SDavid Xu 	struct componentname *cnp;
821655291f2SDavid Xu 	struct vnode *dvp, **vpp;
822655291f2SDavid Xu 	struct mqfs_node *pd;
823655291f2SDavid Xu 	struct mqfs_node *pn;
824fbc48e97SDavid Xu 	struct mqfs_info *mqfs;
825655291f2SDavid Xu 	int nameiop, flags, error, namelen;
826655291f2SDavid Xu 	char *pname;
827655291f2SDavid Xu 	struct thread *td;
828655291f2SDavid Xu 
829655291f2SDavid Xu 	cnp = ap->a_cnp;
830655291f2SDavid Xu 	vpp = ap->a_vpp;
831655291f2SDavid Xu 	dvp = ap->a_dvp;
832655291f2SDavid Xu 	pname = cnp->cn_nameptr;
833655291f2SDavid Xu 	namelen = cnp->cn_namelen;
834655291f2SDavid Xu 	td = cnp->cn_thread;
835655291f2SDavid Xu 	flags = cnp->cn_flags;
836655291f2SDavid Xu 	nameiop = cnp->cn_nameiop;
837655291f2SDavid Xu 	pd = VTON(dvp);
838655291f2SDavid Xu 	pn = NULL;
839fbc48e97SDavid Xu 	mqfs = pd->mn_info;
840655291f2SDavid Xu 	*vpp = NULLVP;
841655291f2SDavid Xu 
842655291f2SDavid Xu 	if (dvp->v_type != VDIR)
843655291f2SDavid Xu 		return (ENOTDIR);
844655291f2SDavid Xu 
845655291f2SDavid Xu 	error = VOP_ACCESS(dvp, VEXEC, cnp->cn_cred, cnp->cn_thread);
846655291f2SDavid Xu 	if (error)
847655291f2SDavid Xu 		return (error);
848655291f2SDavid Xu 
849655291f2SDavid Xu 	/* shortcut: check if the name is too long */
850655291f2SDavid Xu 	if (cnp->cn_namelen >= MQFS_NAMELEN)
851655291f2SDavid Xu 		return (ENOENT);
852655291f2SDavid Xu 
853655291f2SDavid Xu 	/* self */
854655291f2SDavid Xu 	if (namelen == 1 && pname[0] == '.') {
855655291f2SDavid Xu 		if ((flags & ISLASTCN) && nameiop != LOOKUP)
856655291f2SDavid Xu 			return (EINVAL);
857655291f2SDavid Xu 		pn = pd;
858655291f2SDavid Xu 		*vpp = dvp;
859655291f2SDavid Xu 		VREF(dvp);
860655291f2SDavid Xu 		return (0);
861655291f2SDavid Xu 	}
862655291f2SDavid Xu 
863655291f2SDavid Xu 	/* parent */
864655291f2SDavid Xu 	if (cnp->cn_flags & ISDOTDOT) {
865655291f2SDavid Xu 		if (dvp->v_vflag & VV_ROOT)
866655291f2SDavid Xu 			return (EIO);
867655291f2SDavid Xu 		if ((flags & ISLASTCN) && nameiop != LOOKUP)
868655291f2SDavid Xu 			return (EINVAL);
86922db15c0SAttilio Rao 		VOP_UNLOCK(dvp, 0);
870655291f2SDavid Xu 		KASSERT(pd->mn_parent, ("non-root directory has no parent"));
871655291f2SDavid Xu 		pn = pd->mn_parent;
872655291f2SDavid Xu 		error = mqfs_allocv(dvp->v_mount, vpp, pn);
873cb05b60aSAttilio Rao 		vn_lock(dvp, LK_EXCLUSIVE | LK_RETRY);
874655291f2SDavid Xu 		return (error);
875655291f2SDavid Xu 	}
876655291f2SDavid Xu 
877655291f2SDavid Xu 	/* named node */
878fbc48e97SDavid Xu 	sx_xlock(&mqfs->mi_lock);
879655291f2SDavid Xu 	pn = mqfs_search(pd, pname, namelen);
880fbc48e97SDavid Xu 	if (pn != NULL)
881fbc48e97SDavid Xu 		mqnode_addref(pn);
882fbc48e97SDavid Xu 	sx_xunlock(&mqfs->mi_lock);
883655291f2SDavid Xu 
884655291f2SDavid Xu 	/* found */
885655291f2SDavid Xu 	if (pn != NULL) {
886655291f2SDavid Xu 		/* DELETE */
887655291f2SDavid Xu 		if (nameiop == DELETE && (flags & ISLASTCN)) {
888655291f2SDavid Xu 			error = VOP_ACCESS(dvp, VWRITE, cnp->cn_cred, td);
889fbc48e97SDavid Xu 			if (error) {
890fbc48e97SDavid Xu 				mqnode_release(pn);
891655291f2SDavid Xu 				return (error);
892fbc48e97SDavid Xu 			}
893655291f2SDavid Xu 			if (*vpp == dvp) {
894655291f2SDavid Xu 				VREF(dvp);
895655291f2SDavid Xu 				*vpp = dvp;
896fbc48e97SDavid Xu 				mqnode_release(pn);
897655291f2SDavid Xu 				return (0);
898655291f2SDavid Xu 			}
899655291f2SDavid Xu 		}
900655291f2SDavid Xu 
901655291f2SDavid Xu 		/* allocate vnode */
902655291f2SDavid Xu 		error = mqfs_allocv(dvp->v_mount, vpp, pn);
903fbc48e97SDavid Xu 		mqnode_release(pn);
904655291f2SDavid Xu 		if (error == 0 && cnp->cn_flags & MAKEENTRY)
905655291f2SDavid Xu 			cache_enter(dvp, *vpp, cnp);
906655291f2SDavid Xu 		return (error);
907655291f2SDavid Xu 	}
908655291f2SDavid Xu 
909655291f2SDavid Xu 	/* not found */
910655291f2SDavid Xu 
911655291f2SDavid Xu 	/* will create a new entry in the directory ? */
912655291f2SDavid Xu 	if ((nameiop == CREATE || nameiop == RENAME) && (flags & LOCKPARENT)
913655291f2SDavid Xu 	    && (flags & ISLASTCN)) {
914655291f2SDavid Xu 		error = VOP_ACCESS(dvp, VWRITE, cnp->cn_cred, td);
915655291f2SDavid Xu 		if (error)
916655291f2SDavid Xu 			return (error);
917655291f2SDavid Xu 		cnp->cn_flags |= SAVENAME;
918655291f2SDavid Xu 		return (EJUSTRETURN);
919655291f2SDavid Xu 	}
920655291f2SDavid Xu 	return (ENOENT);
921655291f2SDavid Xu }
922655291f2SDavid Xu 
923655291f2SDavid Xu #if 0
924655291f2SDavid Xu struct vop_lookup_args {
925655291f2SDavid Xu 	struct vop_generic_args a_gen;
926655291f2SDavid Xu 	struct vnode *a_dvp;
927655291f2SDavid Xu 	struct vnode **a_vpp;
928655291f2SDavid Xu 	struct componentname *a_cnp;
929655291f2SDavid Xu };
930655291f2SDavid Xu #endif
931655291f2SDavid Xu 
932655291f2SDavid Xu /*
933655291f2SDavid Xu  * vnode lookup operation
934655291f2SDavid Xu  */
935655291f2SDavid Xu static int
936655291f2SDavid Xu mqfs_lookup(struct vop_cachedlookup_args *ap)
937655291f2SDavid Xu {
938655291f2SDavid Xu 	int rc;
939655291f2SDavid Xu 
940655291f2SDavid Xu 	rc = mqfs_lookupx(ap);
941655291f2SDavid Xu 	return (rc);
942655291f2SDavid Xu }
943655291f2SDavid Xu 
944655291f2SDavid Xu #if 0
945655291f2SDavid Xu struct vop_create_args {
946655291f2SDavid Xu 	struct vnode *a_dvp;
947655291f2SDavid Xu 	struct vnode **a_vpp;
948655291f2SDavid Xu 	struct componentname *a_cnp;
949655291f2SDavid Xu 	struct vattr *a_vap;
950655291f2SDavid Xu };
951655291f2SDavid Xu #endif
952655291f2SDavid Xu 
953655291f2SDavid Xu /*
954655291f2SDavid Xu  * vnode creation operation
955655291f2SDavid Xu  */
956655291f2SDavid Xu static int
957655291f2SDavid Xu mqfs_create(struct vop_create_args *ap)
958655291f2SDavid Xu {
959655291f2SDavid Xu 	struct mqfs_info *mqfs = VFSTOMQFS(ap->a_dvp->v_mount);
960655291f2SDavid Xu 	struct componentname *cnp = ap->a_cnp;
961655291f2SDavid Xu 	struct mqfs_node *pd;
962655291f2SDavid Xu 	struct mqfs_node *pn;
963655291f2SDavid Xu 	struct mqueue *mq;
964655291f2SDavid Xu 	int error;
965655291f2SDavid Xu 
966655291f2SDavid Xu 	pd = VTON(ap->a_dvp);
967655291f2SDavid Xu 	if (pd->mn_type != mqfstype_root && pd->mn_type != mqfstype_dir)
968655291f2SDavid Xu 		return (ENOTDIR);
969655291f2SDavid Xu 	mq = mqueue_alloc(NULL);
970655291f2SDavid Xu 	if (mq == NULL)
971655291f2SDavid Xu 		return (EAGAIN);
972655291f2SDavid Xu 	sx_xlock(&mqfs->mi_lock);
973655291f2SDavid Xu 	if ((cnp->cn_flags & HASBUF) == 0)
974655291f2SDavid Xu 		panic("%s: no name", __func__);
975ba0360b1SDavid Xu 	pn = mqfs_create_file(pd, cnp->cn_nameptr, cnp->cn_namelen,
976ba0360b1SDavid Xu 		cnp->cn_cred, ap->a_vap->va_mode);
977fbc48e97SDavid Xu 	if (pn == NULL) {
978fbc48e97SDavid Xu 		sx_xunlock(&mqfs->mi_lock);
979ba0360b1SDavid Xu 		error = ENOSPC;
980fbc48e97SDavid Xu 	} else {
981fbc48e97SDavid Xu 		mqnode_addref(pn);
982fbc48e97SDavid Xu 		sx_xunlock(&mqfs->mi_lock);
983655291f2SDavid Xu 		error = mqfs_allocv(ap->a_dvp->v_mount, ap->a_vpp, pn);
984fbc48e97SDavid Xu 		mqnode_release(pn);
985ba0360b1SDavid Xu 		if (error)
986ba0360b1SDavid Xu 			mqfs_destroy(pn);
987ba0360b1SDavid Xu 		else
988ba0360b1SDavid Xu 			pn->mn_data = mq;
989ba0360b1SDavid Xu 	}
990ba0360b1SDavid Xu 	if (error)
991ba0360b1SDavid Xu 		mqueue_free(mq);
992655291f2SDavid Xu 	return (error);
993655291f2SDavid Xu }
994655291f2SDavid Xu 
995655291f2SDavid Xu /*
996655291f2SDavid Xu  * Remove an entry
997655291f2SDavid Xu  */
998655291f2SDavid Xu static
999655291f2SDavid Xu int do_unlink(struct mqfs_node *pn, struct ucred *ucred)
1000655291f2SDavid Xu {
1001655291f2SDavid Xu 	struct mqfs_node *parent;
1002655291f2SDavid Xu 	struct mqfs_vdata *vd;
1003655291f2SDavid Xu 	int error = 0;
1004655291f2SDavid Xu 
1005655291f2SDavid Xu 	sx_assert(&pn->mn_info->mi_lock, SX_LOCKED);
1006655291f2SDavid Xu 
1007655291f2SDavid Xu 	if (ucred->cr_uid != pn->mn_uid &&
100832f9753cSRobert Watson 	    (error = priv_check_cred(ucred, PRIV_MQ_ADMIN, 0)) != 0)
1009655291f2SDavid Xu 		error = EACCES;
1010655291f2SDavid Xu 	else if (!pn->mn_deleted) {
1011655291f2SDavid Xu 		parent = pn->mn_parent;
1012655291f2SDavid Xu 		pn->mn_parent = NULL;
1013655291f2SDavid Xu 		pn->mn_deleted = 1;
1014655291f2SDavid Xu 		LIST_REMOVE(pn, mn_sibling);
1015655291f2SDavid Xu 		LIST_FOREACH(vd, &pn->mn_vnodes, mv_link) {
1016655291f2SDavid Xu 			cache_purge(vd->mv_vnode);
1017655291f2SDavid Xu 			vhold(vd->mv_vnode);
1018655291f2SDavid Xu 			taskqueue_enqueue(taskqueue_thread, &vd->mv_task);
1019655291f2SDavid Xu 		}
1020655291f2SDavid Xu 		mqnode_release(pn);
1021655291f2SDavid Xu 		mqnode_release(parent);
1022655291f2SDavid Xu 	} else
1023655291f2SDavid Xu 		error = ENOENT;
1024655291f2SDavid Xu 	return (error);
1025655291f2SDavid Xu }
1026655291f2SDavid Xu 
1027655291f2SDavid Xu #if 0
1028655291f2SDavid Xu struct vop_remove_args {
1029655291f2SDavid Xu 	struct vnode *a_dvp;
1030655291f2SDavid Xu 	struct vnode *a_vp;
1031655291f2SDavid Xu 	struct componentname *a_cnp;
1032655291f2SDavid Xu };
1033655291f2SDavid Xu #endif
1034655291f2SDavid Xu 
1035655291f2SDavid Xu /*
1036655291f2SDavid Xu  * vnode removal operation
1037655291f2SDavid Xu  */
1038655291f2SDavid Xu static int
1039655291f2SDavid Xu mqfs_remove(struct vop_remove_args *ap)
1040655291f2SDavid Xu {
1041655291f2SDavid Xu 	struct mqfs_info *mqfs = VFSTOMQFS(ap->a_dvp->v_mount);
1042655291f2SDavid Xu 	struct mqfs_node *pn;
1043655291f2SDavid Xu 	int error;
1044655291f2SDavid Xu 
1045655291f2SDavid Xu 	if (ap->a_vp->v_type == VDIR)
1046655291f2SDavid Xu                 return (EPERM);
1047655291f2SDavid Xu 	pn = VTON(ap->a_vp);
1048655291f2SDavid Xu 	sx_xlock(&mqfs->mi_lock);
1049655291f2SDavid Xu 	error = do_unlink(pn, ap->a_cnp->cn_cred);
1050655291f2SDavid Xu 	sx_xunlock(&mqfs->mi_lock);
1051655291f2SDavid Xu 	return (error);
1052655291f2SDavid Xu }
1053655291f2SDavid Xu 
1054655291f2SDavid Xu #if 0
1055655291f2SDavid Xu struct vop_inactive_args {
1056655291f2SDavid Xu 	struct vnode *a_vp;
1057655291f2SDavid Xu 	struct thread *a_td;
1058655291f2SDavid Xu };
1059655291f2SDavid Xu #endif
1060655291f2SDavid Xu 
1061655291f2SDavid Xu static int
1062655291f2SDavid Xu mqfs_inactive(struct vop_inactive_args *ap)
1063655291f2SDavid Xu {
1064655291f2SDavid Xu 	struct mqfs_node *pn = VTON(ap->a_vp);
1065655291f2SDavid Xu 
1066655291f2SDavid Xu 	if (pn->mn_deleted)
1067655291f2SDavid Xu 		vrecycle(ap->a_vp, ap->a_td);
1068655291f2SDavid Xu 	return (0);
1069655291f2SDavid Xu }
1070655291f2SDavid Xu 
1071655291f2SDavid Xu #if 0
1072655291f2SDavid Xu struct vop_reclaim_args {
1073655291f2SDavid Xu 	struct vop_generic_args a_gen;
1074655291f2SDavid Xu 	struct vnode *a_vp;
1075655291f2SDavid Xu 	struct thread *a_td;
1076655291f2SDavid Xu };
1077655291f2SDavid Xu #endif
1078655291f2SDavid Xu 
1079655291f2SDavid Xu static int
1080655291f2SDavid Xu mqfs_reclaim(struct vop_reclaim_args *ap)
1081655291f2SDavid Xu {
1082655291f2SDavid Xu 	struct mqfs_info *mqfs = VFSTOMQFS(ap->a_vp->v_mount);
1083655291f2SDavid Xu 	struct vnode *vp = ap->a_vp;
1084655291f2SDavid Xu 	struct mqfs_node *pn;
1085655291f2SDavid Xu 	struct mqfs_vdata *vd;
1086655291f2SDavid Xu 
1087655291f2SDavid Xu 	vd = vp->v_data;
1088655291f2SDavid Xu 	pn = vd->mv_node;
1089655291f2SDavid Xu 	sx_xlock(&mqfs->mi_lock);
1090655291f2SDavid Xu 	vp->v_data = NULL;
1091655291f2SDavid Xu 	LIST_REMOVE(vd, mv_link);
1092655291f2SDavid Xu 	uma_zfree(mvdata_zone, vd);
1093655291f2SDavid Xu 	mqnode_release(pn);
1094655291f2SDavid Xu 	sx_xunlock(&mqfs->mi_lock);
1095655291f2SDavid Xu 	return (0);
1096655291f2SDavid Xu }
1097655291f2SDavid Xu 
1098655291f2SDavid Xu #if 0
1099655291f2SDavid Xu struct vop_open_args {
1100655291f2SDavid Xu 	struct vop_generic_args a_gen;
1101655291f2SDavid Xu 	struct vnode *a_vp;
1102655291f2SDavid Xu 	int a_mode;
1103655291f2SDavid Xu 	struct ucred *a_cred;
1104655291f2SDavid Xu 	struct thread *a_td;
1105d6da6408SKonstantin Belousov 	struct file *a_fp;
1106655291f2SDavid Xu };
1107655291f2SDavid Xu #endif
1108655291f2SDavid Xu 
1109655291f2SDavid Xu static int
1110655291f2SDavid Xu mqfs_open(struct vop_open_args *ap)
1111655291f2SDavid Xu {
1112655291f2SDavid Xu 	return (0);
1113655291f2SDavid Xu }
1114655291f2SDavid Xu 
1115655291f2SDavid Xu #if 0
1116655291f2SDavid Xu struct vop_close_args {
1117655291f2SDavid Xu 	struct vop_generic_args a_gen;
1118655291f2SDavid Xu 	struct vnode *a_vp;
1119655291f2SDavid Xu 	int a_fflag;
1120655291f2SDavid Xu 	struct ucred *a_cred;
1121655291f2SDavid Xu 	struct thread *a_td;
1122655291f2SDavid Xu };
1123655291f2SDavid Xu #endif
1124655291f2SDavid Xu 
1125655291f2SDavid Xu static int
1126655291f2SDavid Xu mqfs_close(struct vop_close_args *ap)
1127655291f2SDavid Xu {
1128655291f2SDavid Xu 	return (0);
1129655291f2SDavid Xu }
1130655291f2SDavid Xu 
1131655291f2SDavid Xu #if 0
1132655291f2SDavid Xu struct vop_access_args {
1133655291f2SDavid Xu 	struct vop_generic_args a_gen;
1134655291f2SDavid Xu 	struct vnode *a_vp;
113515bc6b2bSEdward Tomasz Napierala 	accmode_t a_accmode;
1136655291f2SDavid Xu 	struct ucred *a_cred;
1137655291f2SDavid Xu 	struct thread *a_td;
1138655291f2SDavid Xu };
1139655291f2SDavid Xu #endif
1140655291f2SDavid Xu 
1141655291f2SDavid Xu /*
1142655291f2SDavid Xu  * Verify permissions
1143655291f2SDavid Xu  */
1144655291f2SDavid Xu static int
1145655291f2SDavid Xu mqfs_access(struct vop_access_args *ap)
1146655291f2SDavid Xu {
1147655291f2SDavid Xu 	struct vnode *vp = ap->a_vp;
1148655291f2SDavid Xu 	struct vattr vattr;
1149655291f2SDavid Xu 	int error;
1150655291f2SDavid Xu 
11510359a12eSAttilio Rao 	error = VOP_GETATTR(vp, &vattr, ap->a_cred);
1152655291f2SDavid Xu 	if (error)
1153655291f2SDavid Xu 		return (error);
1154655291f2SDavid Xu 	error = vaccess(vp->v_type, vattr.va_mode, vattr.va_uid,
115515bc6b2bSEdward Tomasz Napierala 	    vattr.va_gid, ap->a_accmode, ap->a_cred, NULL);
1156655291f2SDavid Xu 	return (error);
1157655291f2SDavid Xu }
1158655291f2SDavid Xu 
1159655291f2SDavid Xu #if 0
1160655291f2SDavid Xu struct vop_getattr_args {
1161655291f2SDavid Xu 	struct vop_generic_args a_gen;
1162655291f2SDavid Xu 	struct vnode *a_vp;
1163655291f2SDavid Xu 	struct vattr *a_vap;
1164655291f2SDavid Xu 	struct ucred *a_cred;
1165655291f2SDavid Xu };
1166655291f2SDavid Xu #endif
1167655291f2SDavid Xu 
1168655291f2SDavid Xu /*
1169655291f2SDavid Xu  * Get file attributes
1170655291f2SDavid Xu  */
1171655291f2SDavid Xu static int
1172655291f2SDavid Xu mqfs_getattr(struct vop_getattr_args *ap)
1173655291f2SDavid Xu {
1174655291f2SDavid Xu 	struct vnode *vp = ap->a_vp;
1175655291f2SDavid Xu 	struct mqfs_node *pn = VTON(vp);
1176655291f2SDavid Xu 	struct vattr *vap = ap->a_vap;
1177655291f2SDavid Xu 	int error = 0;
1178655291f2SDavid Xu 
1179655291f2SDavid Xu 	vap->va_type = vp->v_type;
1180655291f2SDavid Xu 	vap->va_mode = pn->mn_mode;
1181655291f2SDavid Xu 	vap->va_nlink = 1;
1182655291f2SDavid Xu 	vap->va_uid = pn->mn_uid;
1183655291f2SDavid Xu 	vap->va_gid = pn->mn_gid;
1184655291f2SDavid Xu 	vap->va_fsid = vp->v_mount->mnt_stat.f_fsid.val[0];
1185655291f2SDavid Xu 	vap->va_fileid = pn->mn_fileno;
1186655291f2SDavid Xu 	vap->va_size = 0;
1187655291f2SDavid Xu 	vap->va_blocksize = PAGE_SIZE;
1188655291f2SDavid Xu 	vap->va_bytes = vap->va_size = 0;
1189655291f2SDavid Xu 	vap->va_atime = pn->mn_atime;
1190655291f2SDavid Xu 	vap->va_mtime = pn->mn_mtime;
1191655291f2SDavid Xu 	vap->va_ctime = pn->mn_ctime;
1192655291f2SDavid Xu 	vap->va_birthtime = pn->mn_birth;
1193655291f2SDavid Xu 	vap->va_gen = 0;
1194655291f2SDavid Xu 	vap->va_flags = 0;
11954c5a20e3SKonstantin Belousov 	vap->va_rdev = NODEV;
1196655291f2SDavid Xu 	vap->va_bytes = 0;
1197655291f2SDavid Xu 	vap->va_filerev = 0;
1198655291f2SDavid Xu 	return (error);
1199655291f2SDavid Xu }
1200655291f2SDavid Xu 
1201655291f2SDavid Xu #if 0
1202655291f2SDavid Xu struct vop_setattr_args {
1203655291f2SDavid Xu 	struct vop_generic_args a_gen;
1204655291f2SDavid Xu 	struct vnode *a_vp;
1205655291f2SDavid Xu 	struct vattr *a_vap;
1206655291f2SDavid Xu 	struct ucred *a_cred;
1207655291f2SDavid Xu };
1208655291f2SDavid Xu #endif
1209655291f2SDavid Xu /*
1210655291f2SDavid Xu  * Set attributes
1211655291f2SDavid Xu  */
1212655291f2SDavid Xu static int
1213655291f2SDavid Xu mqfs_setattr(struct vop_setattr_args *ap)
1214655291f2SDavid Xu {
1215655291f2SDavid Xu 	struct mqfs_node *pn;
1216655291f2SDavid Xu 	struct vattr *vap;
1217655291f2SDavid Xu 	struct vnode *vp;
12180359a12eSAttilio Rao 	struct thread *td;
1219655291f2SDavid Xu 	int c, error;
1220655291f2SDavid Xu 	uid_t uid;
1221655291f2SDavid Xu 	gid_t gid;
1222655291f2SDavid Xu 
12230359a12eSAttilio Rao 	td = curthread;
1224655291f2SDavid Xu 	vap = ap->a_vap;
1225655291f2SDavid Xu 	vp = ap->a_vp;
1226655291f2SDavid Xu 	if ((vap->va_type != VNON) ||
1227655291f2SDavid Xu 	    (vap->va_nlink != VNOVAL) ||
1228655291f2SDavid Xu 	    (vap->va_fsid != VNOVAL) ||
1229655291f2SDavid Xu 	    (vap->va_fileid != VNOVAL) ||
1230655291f2SDavid Xu 	    (vap->va_blocksize != VNOVAL) ||
1231655291f2SDavid Xu 	    (vap->va_flags != VNOVAL && vap->va_flags != 0) ||
1232655291f2SDavid Xu 	    (vap->va_rdev != VNOVAL) ||
1233655291f2SDavid Xu 	    ((int)vap->va_bytes != VNOVAL) ||
1234655291f2SDavid Xu 	    (vap->va_gen != VNOVAL)) {
1235655291f2SDavid Xu 		return (EINVAL);
1236655291f2SDavid Xu 	}
1237655291f2SDavid Xu 
1238655291f2SDavid Xu 	pn = VTON(vp);
1239655291f2SDavid Xu 
1240655291f2SDavid Xu 	error = c = 0;
1241655291f2SDavid Xu 	if (vap->va_uid == (uid_t)VNOVAL)
1242655291f2SDavid Xu 		uid = pn->mn_uid;
1243655291f2SDavid Xu 	else
1244655291f2SDavid Xu 		uid = vap->va_uid;
1245655291f2SDavid Xu 	if (vap->va_gid == (gid_t)VNOVAL)
1246655291f2SDavid Xu 		gid = pn->mn_gid;
1247655291f2SDavid Xu 	else
1248655291f2SDavid Xu 		gid = vap->va_gid;
1249655291f2SDavid Xu 
1250655291f2SDavid Xu 	if (uid != pn->mn_uid || gid != pn->mn_gid) {
1251655291f2SDavid Xu 		/*
1252655291f2SDavid Xu 		 * To modify the ownership of a file, must possess VADMIN
1253655291f2SDavid Xu 		 * for that file.
1254655291f2SDavid Xu 		 */
12550359a12eSAttilio Rao 		if ((error = VOP_ACCESS(vp, VADMIN, ap->a_cred, td)))
1256655291f2SDavid Xu 			return (error);
1257acd3428bSRobert Watson 
1258acd3428bSRobert Watson 		/*
1259acd3428bSRobert Watson 		 * XXXRW: Why is there a privilege check here: shouldn't the
1260acd3428bSRobert Watson 		 * check in VOP_ACCESS() be enough?  Also, are the group bits
1261acd3428bSRobert Watson 		 * below definitely right?
1262acd3428bSRobert Watson 		 */
1263655291f2SDavid Xu 		if (((ap->a_cred->cr_uid != pn->mn_uid) || uid != pn->mn_uid ||
1264655291f2SDavid Xu 		    (gid != pn->mn_gid && !groupmember(gid, ap->a_cred))) &&
12650359a12eSAttilio Rao 		    (error = priv_check(td, PRIV_MQ_ADMIN)) != 0)
1266655291f2SDavid Xu 			return (error);
1267655291f2SDavid Xu 		pn->mn_uid = uid;
1268655291f2SDavid Xu 		pn->mn_gid = gid;
1269655291f2SDavid Xu 		c = 1;
1270655291f2SDavid Xu 	}
1271655291f2SDavid Xu 
1272655291f2SDavid Xu 	if (vap->va_mode != (mode_t)VNOVAL) {
1273655291f2SDavid Xu 		if ((ap->a_cred->cr_uid != pn->mn_uid) &&
12740359a12eSAttilio Rao 		    (error = priv_check(td, PRIV_MQ_ADMIN)))
1275655291f2SDavid Xu 			return (error);
1276655291f2SDavid Xu 		pn->mn_mode = vap->va_mode;
1277655291f2SDavid Xu 		c = 1;
1278655291f2SDavid Xu 	}
1279655291f2SDavid Xu 
1280655291f2SDavid Xu 	if (vap->va_atime.tv_sec != VNOVAL || vap->va_mtime.tv_sec != VNOVAL) {
1281655291f2SDavid Xu 		/* See the comment in ufs_vnops::ufs_setattr(). */
12820359a12eSAttilio Rao 		if ((error = VOP_ACCESS(vp, VADMIN, ap->a_cred, td)) &&
1283655291f2SDavid Xu 		    ((vap->va_vaflags & VA_UTIMES_NULL) == 0 ||
12840359a12eSAttilio Rao 		    (error = VOP_ACCESS(vp, VWRITE, ap->a_cred, td))))
1285655291f2SDavid Xu 			return (error);
1286655291f2SDavid Xu 		if (vap->va_atime.tv_sec != VNOVAL) {
1287655291f2SDavid Xu 			pn->mn_atime = vap->va_atime;
1288655291f2SDavid Xu 		}
1289655291f2SDavid Xu 		if (vap->va_mtime.tv_sec != VNOVAL) {
1290655291f2SDavid Xu 			pn->mn_mtime = vap->va_mtime;
1291655291f2SDavid Xu 		}
1292655291f2SDavid Xu 		c = 1;
1293655291f2SDavid Xu 	}
1294655291f2SDavid Xu 	if (c) {
1295655291f2SDavid Xu 		vfs_timestamp(&pn->mn_ctime);
1296655291f2SDavid Xu 	}
1297655291f2SDavid Xu 	return (0);
1298655291f2SDavid Xu }
1299655291f2SDavid Xu 
1300655291f2SDavid Xu #if 0
1301655291f2SDavid Xu struct vop_read_args {
1302655291f2SDavid Xu 	struct vop_generic_args a_gen;
1303655291f2SDavid Xu 	struct vnode *a_vp;
1304655291f2SDavid Xu 	struct uio *a_uio;
1305655291f2SDavid Xu 	int a_ioflag;
1306655291f2SDavid Xu 	struct ucred *a_cred;
1307655291f2SDavid Xu };
1308655291f2SDavid Xu #endif
1309655291f2SDavid Xu 
1310655291f2SDavid Xu /*
1311655291f2SDavid Xu  * Read from a file
1312655291f2SDavid Xu  */
1313655291f2SDavid Xu static int
1314655291f2SDavid Xu mqfs_read(struct vop_read_args *ap)
1315655291f2SDavid Xu {
1316655291f2SDavid Xu 	char buf[80];
1317655291f2SDavid Xu 	struct vnode *vp = ap->a_vp;
1318655291f2SDavid Xu 	struct uio *uio = ap->a_uio;
1319655291f2SDavid Xu 	struct mqfs_node *pn;
1320655291f2SDavid Xu 	struct mqueue *mq;
1321655291f2SDavid Xu 	int len, error;
1322655291f2SDavid Xu 
1323655291f2SDavid Xu 	if (vp->v_type != VREG)
1324655291f2SDavid Xu 		return (EINVAL);
1325655291f2SDavid Xu 
1326655291f2SDavid Xu 	pn = VTON(vp);
1327655291f2SDavid Xu 	mq = VTOMQ(vp);
1328655291f2SDavid Xu 	snprintf(buf, sizeof(buf),
1329655291f2SDavid Xu 		"QSIZE:%-10ld MAXMSG:%-10ld CURMSG:%-10ld MSGSIZE:%-10ld\n",
1330655291f2SDavid Xu 		mq->mq_totalbytes,
1331655291f2SDavid Xu 		mq->mq_maxmsg,
1332655291f2SDavid Xu 		mq->mq_curmsgs,
1333655291f2SDavid Xu 		mq->mq_msgsize);
1334655291f2SDavid Xu 	buf[sizeof(buf)-1] = '\0';
1335655291f2SDavid Xu 	len = strlen(buf);
1336655291f2SDavid Xu 	error = uiomove_frombuf(buf, len, uio);
1337655291f2SDavid Xu 	return (error);
1338655291f2SDavid Xu }
1339655291f2SDavid Xu 
1340655291f2SDavid Xu #if 0
1341655291f2SDavid Xu struct vop_readdir_args {
1342655291f2SDavid Xu 	struct vop_generic_args a_gen;
1343655291f2SDavid Xu 	struct vnode *a_vp;
1344655291f2SDavid Xu 	struct uio *a_uio;
1345655291f2SDavid Xu 	struct ucred *a_cred;
1346655291f2SDavid Xu 	int *a_eofflag;
1347655291f2SDavid Xu 	int *a_ncookies;
1348655291f2SDavid Xu 	u_long **a_cookies;
1349655291f2SDavid Xu };
1350655291f2SDavid Xu #endif
1351655291f2SDavid Xu 
1352655291f2SDavid Xu /*
1353655291f2SDavid Xu  * Return directory entries.
1354655291f2SDavid Xu  */
1355655291f2SDavid Xu static int
1356655291f2SDavid Xu mqfs_readdir(struct vop_readdir_args *ap)
1357655291f2SDavid Xu {
1358655291f2SDavid Xu 	struct vnode *vp;
1359655291f2SDavid Xu 	struct mqfs_info *mi;
1360655291f2SDavid Xu 	struct mqfs_node *pd;
1361655291f2SDavid Xu 	struct mqfs_node *pn;
1362655291f2SDavid Xu 	struct dirent entry;
1363655291f2SDavid Xu 	struct uio *uio;
1364655291f2SDavid Xu 	int *tmp_ncookies = NULL;
1365655291f2SDavid Xu 	off_t offset;
1366655291f2SDavid Xu 	int error, i;
1367655291f2SDavid Xu 
1368655291f2SDavid Xu 	vp = ap->a_vp;
1369655291f2SDavid Xu 	mi = VFSTOMQFS(vp->v_mount);
1370655291f2SDavid Xu 	pd = VTON(vp);
1371655291f2SDavid Xu 	uio = ap->a_uio;
1372655291f2SDavid Xu 
1373655291f2SDavid Xu 	if (vp->v_type != VDIR)
1374655291f2SDavid Xu 		return (ENOTDIR);
1375655291f2SDavid Xu 
1376655291f2SDavid Xu 	if (uio->uio_offset < 0)
1377655291f2SDavid Xu 		return (EINVAL);
1378655291f2SDavid Xu 
1379655291f2SDavid Xu 	if (ap->a_ncookies != NULL) {
1380655291f2SDavid Xu 		tmp_ncookies = ap->a_ncookies;
1381655291f2SDavid Xu 		*ap->a_ncookies = 0;
1382655291f2SDavid Xu 		ap->a_ncookies = NULL;
1383655291f2SDavid Xu         }
1384655291f2SDavid Xu 
1385655291f2SDavid Xu 	error = 0;
1386655291f2SDavid Xu 	offset = 0;
1387655291f2SDavid Xu 
1388655291f2SDavid Xu 	sx_xlock(&mi->mi_lock);
1389655291f2SDavid Xu 
1390655291f2SDavid Xu 	LIST_FOREACH(pn, &pd->mn_children, mn_sibling) {
1391655291f2SDavid Xu 		entry.d_reclen = sizeof(entry);
1392655291f2SDavid Xu 		if (!pn->mn_fileno)
1393655291f2SDavid Xu 			mqfs_fileno_alloc(mi, pn);
1394655291f2SDavid Xu 		entry.d_fileno = pn->mn_fileno;
1395655291f2SDavid Xu 		for (i = 0; i < MQFS_NAMELEN - 1 && pn->mn_name[i] != '\0'; ++i)
1396655291f2SDavid Xu 			entry.d_name[i] = pn->mn_name[i];
1397655291f2SDavid Xu 		entry.d_name[i] = 0;
1398655291f2SDavid Xu 		entry.d_namlen = i;
1399655291f2SDavid Xu 		switch (pn->mn_type) {
1400655291f2SDavid Xu 		case mqfstype_root:
1401655291f2SDavid Xu 		case mqfstype_dir:
1402655291f2SDavid Xu 		case mqfstype_this:
1403655291f2SDavid Xu 		case mqfstype_parent:
1404655291f2SDavid Xu 			entry.d_type = DT_DIR;
1405655291f2SDavid Xu 			break;
1406655291f2SDavid Xu 		case mqfstype_file:
1407655291f2SDavid Xu 			entry.d_type = DT_REG;
1408655291f2SDavid Xu 			break;
1409655291f2SDavid Xu 		case mqfstype_symlink:
1410655291f2SDavid Xu 			entry.d_type = DT_LNK;
1411655291f2SDavid Xu 			break;
1412655291f2SDavid Xu 		default:
1413655291f2SDavid Xu 			panic("%s has unexpected node type: %d", pn->mn_name,
1414655291f2SDavid Xu 				pn->mn_type);
1415655291f2SDavid Xu 		}
1416655291f2SDavid Xu 		if (entry.d_reclen > uio->uio_resid)
1417655291f2SDavid Xu                         break;
1418655291f2SDavid Xu 		if (offset >= uio->uio_offset) {
1419655291f2SDavid Xu 			error = vfs_read_dirent(ap, &entry, offset);
1420655291f2SDavid Xu                         if (error)
1421655291f2SDavid Xu                                 break;
1422655291f2SDavid Xu                 }
1423655291f2SDavid Xu                 offset += entry.d_reclen;
1424655291f2SDavid Xu 	}
1425655291f2SDavid Xu 	sx_xunlock(&mi->mi_lock);
1426655291f2SDavid Xu 
1427655291f2SDavid Xu 	uio->uio_offset = offset;
1428655291f2SDavid Xu 
1429655291f2SDavid Xu 	if (tmp_ncookies != NULL)
1430655291f2SDavid Xu 		ap->a_ncookies = tmp_ncookies;
1431655291f2SDavid Xu 
1432655291f2SDavid Xu 	return (error);
1433655291f2SDavid Xu }
1434655291f2SDavid Xu 
1435655291f2SDavid Xu #ifdef notyet
1436655291f2SDavid Xu 
1437655291f2SDavid Xu #if 0
1438655291f2SDavid Xu struct vop_mkdir_args {
1439655291f2SDavid Xu 	struct vnode *a_dvp;
1440655291f2SDavid Xu 	struvt vnode **a_vpp;
1441655291f2SDavid Xu 	struvt componentname *a_cnp;
1442655291f2SDavid Xu 	struct vattr *a_vap;
1443655291f2SDavid Xu };
1444655291f2SDavid Xu #endif
1445655291f2SDavid Xu 
1446655291f2SDavid Xu /*
1447655291f2SDavid Xu  * Create a directory.
1448655291f2SDavid Xu  */
1449655291f2SDavid Xu static int
1450655291f2SDavid Xu mqfs_mkdir(struct vop_mkdir_args *ap)
1451655291f2SDavid Xu {
1452655291f2SDavid Xu 	struct mqfs_info *mqfs = VFSTOMQFS(ap->a_dvp->v_mount);
1453655291f2SDavid Xu 	struct componentname *cnp = ap->a_cnp;
1454655291f2SDavid Xu 	struct mqfs_node *pd = VTON(ap->a_dvp);
1455655291f2SDavid Xu 	struct mqfs_node *pn;
1456655291f2SDavid Xu 	int error;
1457655291f2SDavid Xu 
1458655291f2SDavid Xu 	if (pd->mn_type != mqfstype_root && pd->mn_type != mqfstype_dir)
1459655291f2SDavid Xu 		return (ENOTDIR);
1460655291f2SDavid Xu 	sx_xlock(&mqfs->mi_lock);
1461655291f2SDavid Xu 	if ((cnp->cn_flags & HASBUF) == 0)
1462655291f2SDavid Xu 		panic("%s: no name", __func__);
1463ba0360b1SDavid Xu 	pn = mqfs_create_dir(pd, cnp->cn_nameptr, cnp->cn_namelen,
1464ba0360b1SDavid Xu 		ap->a_vap->cn_cred, ap->a_vap->va_mode);
1465fbc48e97SDavid Xu 	if (pn != NULL)
1466fbc48e97SDavid Xu 		mqnode_addref(pn);
1467655291f2SDavid Xu 	sx_xunlock(&mqfs->mi_lock);
1468fbc48e97SDavid Xu 	if (pn == NULL) {
1469fbc48e97SDavid Xu 		error = ENOSPC;
1470fbc48e97SDavid Xu 	} else {
1471fbc48e97SDavid Xu 		error = mqfs_allocv(ap->a_dvp->v_mount, ap->a_vpp, pn);
1472fbc48e97SDavid Xu 		mqnode_release(pn);
1473fbc48e97SDavid Xu 	}
1474655291f2SDavid Xu 	return (error);
1475655291f2SDavid Xu }
1476655291f2SDavid Xu 
1477655291f2SDavid Xu #if 0
1478655291f2SDavid Xu struct vop_rmdir_args {
1479655291f2SDavid Xu 	struct vnode *a_dvp;
1480655291f2SDavid Xu 	struct vnode *a_vp;
1481655291f2SDavid Xu 	struct componentname *a_cnp;
1482655291f2SDavid Xu };
1483655291f2SDavid Xu #endif
1484655291f2SDavid Xu 
1485655291f2SDavid Xu /*
1486655291f2SDavid Xu  * Remove a directory.
1487655291f2SDavid Xu  */
1488655291f2SDavid Xu static int
1489655291f2SDavid Xu mqfs_rmdir(struct vop_rmdir_args *ap)
1490655291f2SDavid Xu {
1491655291f2SDavid Xu 	struct mqfs_info *mqfs = VFSTOMQFS(ap->a_dvp->v_mount);
1492655291f2SDavid Xu 	struct mqfs_node *pn = VTON(ap->a_vp);
1493655291f2SDavid Xu 	struct mqfs_node *pt;
1494655291f2SDavid Xu 
1495655291f2SDavid Xu 	if (pn->mn_type != mqfstype_dir)
1496655291f2SDavid Xu 		return (ENOTDIR);
1497655291f2SDavid Xu 
1498655291f2SDavid Xu 	sx_xlock(&mqfs->mi_lock);
1499655291f2SDavid Xu 	if (pn->mn_deleted) {
1500655291f2SDavid Xu 		sx_xunlock(&mqfs->mi_lock);
1501655291f2SDavid Xu 		return (ENOENT);
1502655291f2SDavid Xu 	}
1503655291f2SDavid Xu 
1504655291f2SDavid Xu 	pt = LIST_FIRST(&pn->mn_children);
1505655291f2SDavid Xu 	pt = LIST_NEXT(pt, mn_sibling);
1506655291f2SDavid Xu 	pt = LIST_NEXT(pt, mn_sibling);
1507655291f2SDavid Xu 	if (pt != NULL) {
1508655291f2SDavid Xu 		sx_xunlock(&mqfs->mi_lock);
1509655291f2SDavid Xu 		return (ENOTEMPTY);
1510655291f2SDavid Xu 	}
1511655291f2SDavid Xu 	pt = pn->mn_parent;
1512655291f2SDavid Xu 	pn->mn_parent = NULL;
1513655291f2SDavid Xu 	pn->mn_deleted = 1;
1514655291f2SDavid Xu 	LIST_REMOVE(pn, mn_sibling);
1515655291f2SDavid Xu 	mqnode_release(pn);
1516655291f2SDavid Xu 	mqnode_release(pt);
1517655291f2SDavid Xu 	sx_xunlock(&mqfs->mi_lock);
1518655291f2SDavid Xu 	cache_purge(ap->a_vp);
1519655291f2SDavid Xu 	return (0);
1520655291f2SDavid Xu }
1521655291f2SDavid Xu 
1522655291f2SDavid Xu #endif /* notyet */
1523655291f2SDavid Xu 
1524655291f2SDavid Xu /*
1525655291f2SDavid Xu  * Allocate a message queue
1526655291f2SDavid Xu  */
1527655291f2SDavid Xu static struct mqueue *
1528655291f2SDavid Xu mqueue_alloc(const struct mq_attr *attr)
1529655291f2SDavid Xu {
1530655291f2SDavid Xu 	struct mqueue *mq;
1531655291f2SDavid Xu 
1532655291f2SDavid Xu 	if (curmq >= maxmq)
1533655291f2SDavid Xu 		return (NULL);
1534655291f2SDavid Xu 	mq = uma_zalloc(mqueue_zone, M_WAITOK | M_ZERO);
1535655291f2SDavid Xu 	TAILQ_INIT(&mq->mq_msgq);
1536655291f2SDavid Xu 	if (attr != NULL) {
1537655291f2SDavid Xu 		mq->mq_maxmsg = attr->mq_maxmsg;
1538655291f2SDavid Xu 		mq->mq_msgsize = attr->mq_msgsize;
1539655291f2SDavid Xu 	} else {
1540655291f2SDavid Xu 		mq->mq_maxmsg = default_maxmsg;
1541655291f2SDavid Xu 		mq->mq_msgsize = default_msgsize;
1542655291f2SDavid Xu 	}
1543b042e976SDavid Xu 	mtx_init(&mq->mq_mutex, "mqueue lock", NULL, MTX_DEF);
1544d8b0556cSKonstantin Belousov 	knlist_init_mtx(&mq->mq_rsel.si_note, &mq->mq_mutex);
1545d8b0556cSKonstantin Belousov 	knlist_init_mtx(&mq->mq_wsel.si_note, &mq->mq_mutex);
1546655291f2SDavid Xu 	atomic_add_int(&curmq, 1);
1547655291f2SDavid Xu 	return (mq);
1548655291f2SDavid Xu }
1549655291f2SDavid Xu 
1550655291f2SDavid Xu /*
1551655291f2SDavid Xu  * Destroy a message queue
1552655291f2SDavid Xu  */
1553655291f2SDavid Xu static void
1554655291f2SDavid Xu mqueue_free(struct mqueue *mq)
1555655291f2SDavid Xu {
1556655291f2SDavid Xu 	struct mqueue_msg *msg;
1557655291f2SDavid Xu 
1558655291f2SDavid Xu 	while ((msg = TAILQ_FIRST(&mq->mq_msgq)) != NULL) {
1559655291f2SDavid Xu 		TAILQ_REMOVE(&mq->mq_msgq, msg, msg_link);
15601ede983cSDag-Erling Smørgrav 		free(msg, M_MQUEUEDATA);
1561655291f2SDavid Xu 	}
1562655291f2SDavid Xu 
1563655291f2SDavid Xu 	mtx_destroy(&mq->mq_mutex);
1564655291f2SDavid Xu 	knlist_destroy(&mq->mq_rsel.si_note);
1565655291f2SDavid Xu 	knlist_destroy(&mq->mq_wsel.si_note);
1566655291f2SDavid Xu 	uma_zfree(mqueue_zone, mq);
1567655291f2SDavid Xu 	atomic_add_int(&curmq, -1);
1568655291f2SDavid Xu }
1569655291f2SDavid Xu 
1570655291f2SDavid Xu /*
1571655291f2SDavid Xu  * Load a message from user space
1572655291f2SDavid Xu  */
1573655291f2SDavid Xu static struct mqueue_msg *
1574655291f2SDavid Xu mqueue_loadmsg(const char *msg_ptr, size_t msg_size, int msg_prio)
1575655291f2SDavid Xu {
1576655291f2SDavid Xu 	struct mqueue_msg *msg;
1577655291f2SDavid Xu 	size_t len;
1578655291f2SDavid Xu 	int error;
1579655291f2SDavid Xu 
1580655291f2SDavid Xu 	len = sizeof(struct mqueue_msg) + msg_size;
15811ede983cSDag-Erling Smørgrav 	msg = malloc(len, M_MQUEUEDATA, M_WAITOK);
1582655291f2SDavid Xu 	error = copyin(msg_ptr, ((char *)msg) + sizeof(struct mqueue_msg),
1583655291f2SDavid Xu 	    msg_size);
1584655291f2SDavid Xu 	if (error) {
15851ede983cSDag-Erling Smørgrav 		free(msg, M_MQUEUEDATA);
1586655291f2SDavid Xu 		msg = NULL;
1587655291f2SDavid Xu 	} else {
1588655291f2SDavid Xu 		msg->msg_size = msg_size;
1589655291f2SDavid Xu 		msg->msg_prio = msg_prio;
1590655291f2SDavid Xu 	}
1591655291f2SDavid Xu 	return (msg);
1592655291f2SDavid Xu }
1593655291f2SDavid Xu 
1594655291f2SDavid Xu /*
1595655291f2SDavid Xu  * Save a message to user space
1596655291f2SDavid Xu  */
1597655291f2SDavid Xu static int
1598655291f2SDavid Xu mqueue_savemsg(struct mqueue_msg *msg, char *msg_ptr, int *msg_prio)
1599655291f2SDavid Xu {
1600655291f2SDavid Xu 	int error;
1601655291f2SDavid Xu 
1602655291f2SDavid Xu 	error = copyout(((char *)msg) + sizeof(*msg), msg_ptr,
1603655291f2SDavid Xu 		msg->msg_size);
1604b2f92ef9SDavid Xu 	if (error == 0 && msg_prio != NULL)
1605655291f2SDavid Xu 		error = copyout(&msg->msg_prio, msg_prio, sizeof(int));
1606655291f2SDavid Xu 	return (error);
1607655291f2SDavid Xu }
1608655291f2SDavid Xu 
1609655291f2SDavid Xu /*
1610655291f2SDavid Xu  * Free a message's memory
1611655291f2SDavid Xu  */
1612655291f2SDavid Xu static __inline void
1613655291f2SDavid Xu mqueue_freemsg(struct mqueue_msg *msg)
1614655291f2SDavid Xu {
16151ede983cSDag-Erling Smørgrav 	free(msg, M_MQUEUEDATA);
1616655291f2SDavid Xu }
1617655291f2SDavid Xu 
1618655291f2SDavid Xu /*
1619655291f2SDavid Xu  * Send a message. if waitok is false, thread will not be
1620655291f2SDavid Xu  * blocked if there is no data in queue, otherwise, absolute
1621655291f2SDavid Xu  * time will be checked.
1622655291f2SDavid Xu  */
1623655291f2SDavid Xu int
1624655291f2SDavid Xu mqueue_send(struct mqueue *mq, const char *msg_ptr,
1625655291f2SDavid Xu 	size_t msg_len, unsigned msg_prio, int waitok,
1626655291f2SDavid Xu 	const struct timespec *abs_timeout)
1627655291f2SDavid Xu {
1628655291f2SDavid Xu 	struct mqueue_msg *msg;
1629afde2b65SKonstantin Belousov 	struct timespec ts, ts2;
1630655291f2SDavid Xu 	struct timeval tv;
1631655291f2SDavid Xu 	int error;
1632655291f2SDavid Xu 
1633a6de716dSDavid Xu 	if (msg_prio >= MQ_PRIO_MAX)
1634a6de716dSDavid Xu 		return (EINVAL);
1635655291f2SDavid Xu 	if (msg_len > mq->mq_msgsize)
1636655291f2SDavid Xu 		return (EMSGSIZE);
1637655291f2SDavid Xu 	msg = mqueue_loadmsg(msg_ptr, msg_len, msg_prio);
1638655291f2SDavid Xu 	if (msg == NULL)
1639655291f2SDavid Xu 		return (EFAULT);
1640655291f2SDavid Xu 
1641655291f2SDavid Xu 	/* O_NONBLOCK case */
1642655291f2SDavid Xu 	if (!waitok) {
1643655291f2SDavid Xu 		error = _mqueue_send(mq, msg, -1);
1644655291f2SDavid Xu 		if (error)
1645655291f2SDavid Xu 			goto bad;
1646655291f2SDavid Xu 		return (0);
1647655291f2SDavid Xu 	}
1648655291f2SDavid Xu 
1649655291f2SDavid Xu 	/* we allow a null timeout (wait forever) */
1650655291f2SDavid Xu 	if (abs_timeout == NULL) {
1651655291f2SDavid Xu 		error = _mqueue_send(mq, msg, 0);
1652655291f2SDavid Xu 		if (error)
1653655291f2SDavid Xu 			goto bad;
1654655291f2SDavid Xu 		return (0);
1655655291f2SDavid Xu 	}
1656655291f2SDavid Xu 
1657655291f2SDavid Xu 	/* send it before checking time */
1658655291f2SDavid Xu 	error = _mqueue_send(mq, msg, -1);
1659655291f2SDavid Xu 	if (error == 0)
1660655291f2SDavid Xu 		return (0);
1661655291f2SDavid Xu 
1662655291f2SDavid Xu 	if (error != EAGAIN)
1663655291f2SDavid Xu 		goto bad;
1664655291f2SDavid Xu 
1665afde2b65SKonstantin Belousov 	if (abs_timeout->tv_nsec >= 1000000000 || abs_timeout->tv_nsec < 0) {
1666655291f2SDavid Xu 		error = EINVAL;
1667655291f2SDavid Xu 		goto bad;
1668655291f2SDavid Xu 	}
1669655291f2SDavid Xu 	for (;;) {
1670afde2b65SKonstantin Belousov 		ts2 = *abs_timeout;
1671a6de716dSDavid Xu 		getnanotime(&ts);
1672655291f2SDavid Xu 		timespecsub(&ts2, &ts);
1673655291f2SDavid Xu 		if (ts2.tv_sec < 0 || (ts2.tv_sec == 0 && ts2.tv_nsec <= 0)) {
1674655291f2SDavid Xu 			error = ETIMEDOUT;
1675655291f2SDavid Xu 			break;
1676655291f2SDavid Xu 		}
1677655291f2SDavid Xu 		TIMESPEC_TO_TIMEVAL(&tv, &ts2);
1678655291f2SDavid Xu 		error = _mqueue_send(mq, msg, tvtohz(&tv));
1679655291f2SDavid Xu 		if (error != ETIMEDOUT)
1680655291f2SDavid Xu 			break;
1681655291f2SDavid Xu 	}
1682655291f2SDavid Xu 	if (error == 0)
1683655291f2SDavid Xu 		return (0);
1684655291f2SDavid Xu bad:
1685655291f2SDavid Xu 	mqueue_freemsg(msg);
1686655291f2SDavid Xu 	return (error);
1687655291f2SDavid Xu }
1688655291f2SDavid Xu 
1689655291f2SDavid Xu /*
1690655291f2SDavid Xu  * Common routine to send a message
1691655291f2SDavid Xu  */
1692655291f2SDavid Xu static int
1693655291f2SDavid Xu _mqueue_send(struct mqueue *mq, struct mqueue_msg *msg, int timo)
1694655291f2SDavid Xu {
1695655291f2SDavid Xu 	struct mqueue_msg *msg2;
1696655291f2SDavid Xu 	int error = 0;
1697655291f2SDavid Xu 
1698655291f2SDavid Xu 	mtx_lock(&mq->mq_mutex);
1699655291f2SDavid Xu 	while (mq->mq_curmsgs >= mq->mq_maxmsg && error == 0) {
1700655291f2SDavid Xu 		if (timo < 0) {
1701655291f2SDavid Xu 			mtx_unlock(&mq->mq_mutex);
1702655291f2SDavid Xu 			return (EAGAIN);
1703655291f2SDavid Xu 		}
1704655291f2SDavid Xu 		mq->mq_senders++;
1705655291f2SDavid Xu 		error = msleep(&mq->mq_senders, &mq->mq_mutex,
17060f180a7cSJohn Baldwin 			    PCATCH, "mqsend", timo);
1707655291f2SDavid Xu 		mq->mq_senders--;
1708655291f2SDavid Xu 		if (error == EAGAIN)
1709655291f2SDavid Xu 			error = ETIMEDOUT;
1710655291f2SDavid Xu 	}
1711655291f2SDavid Xu 	if (mq->mq_curmsgs >= mq->mq_maxmsg) {
1712655291f2SDavid Xu 		mtx_unlock(&mq->mq_mutex);
1713655291f2SDavid Xu 		return (error);
1714655291f2SDavid Xu 	}
1715655291f2SDavid Xu 	error = 0;
1716655291f2SDavid Xu 	if (TAILQ_EMPTY(&mq->mq_msgq)) {
1717655291f2SDavid Xu 		TAILQ_INSERT_HEAD(&mq->mq_msgq, msg, msg_link);
1718655291f2SDavid Xu 	} else {
1719655291f2SDavid Xu 		if (msg->msg_prio <= TAILQ_LAST(&mq->mq_msgq, msgq)->msg_prio) {
1720655291f2SDavid Xu 			TAILQ_INSERT_TAIL(&mq->mq_msgq, msg, msg_link);
1721655291f2SDavid Xu 		} else {
1722655291f2SDavid Xu 			TAILQ_FOREACH(msg2, &mq->mq_msgq, msg_link) {
1723655291f2SDavid Xu 				if (msg2->msg_prio < msg->msg_prio)
1724655291f2SDavid Xu 					break;
1725655291f2SDavid Xu 			}
1726655291f2SDavid Xu 			TAILQ_INSERT_BEFORE(msg2, msg, msg_link);
1727655291f2SDavid Xu 		}
1728655291f2SDavid Xu 	}
1729655291f2SDavid Xu 	mq->mq_curmsgs++;
1730655291f2SDavid Xu 	mq->mq_totalbytes += msg->msg_size;
1731655291f2SDavid Xu 	if (mq->mq_receivers)
1732655291f2SDavid Xu 		wakeup_one(&mq->mq_receivers);
1733655291f2SDavid Xu 	else if (mq->mq_notifier != NULL)
1734655291f2SDavid Xu 		mqueue_send_notification(mq);
1735655291f2SDavid Xu 	if (mq->mq_flags & MQ_RSEL) {
1736655291f2SDavid Xu 		mq->mq_flags &= ~MQ_RSEL;
173703f70aecSDavid Xu 		selwakeup(&mq->mq_rsel);
1738655291f2SDavid Xu 	}
1739655291f2SDavid Xu 	KNOTE_LOCKED(&mq->mq_rsel.si_note, 0);
1740655291f2SDavid Xu 	mtx_unlock(&mq->mq_mutex);
1741655291f2SDavid Xu 	return (0);
1742655291f2SDavid Xu }
1743655291f2SDavid Xu 
1744655291f2SDavid Xu /*
1745655291f2SDavid Xu  * Send realtime a signal to process which registered itself
1746655291f2SDavid Xu  * successfully by mq_notify.
1747655291f2SDavid Xu  */
1748655291f2SDavid Xu static void
1749655291f2SDavid Xu mqueue_send_notification(struct mqueue *mq)
1750655291f2SDavid Xu {
1751b2f92ef9SDavid Xu 	struct mqueue_notifier *nt;
1752cf7d9a8cSDavid Xu 	struct thread *td;
1753b2f92ef9SDavid Xu 	struct proc *p;
1754cf7d9a8cSDavid Xu 	int error;
1755655291f2SDavid Xu 
1756655291f2SDavid Xu 	mtx_assert(&mq->mq_mutex, MA_OWNED);
1757b2f92ef9SDavid Xu 	nt = mq->mq_notifier;
1758052ea11cSDavid Xu 	if (nt->nt_sigev.sigev_notify != SIGEV_NONE) {
1759b2f92ef9SDavid Xu 		p = nt->nt_proc;
1760cf7d9a8cSDavid Xu 		error = sigev_findtd(p, &nt->nt_sigev, &td);
1761cf7d9a8cSDavid Xu 		if (error) {
1762cf7d9a8cSDavid Xu 			mq->mq_notifier = NULL;
1763cf7d9a8cSDavid Xu 			return;
1764cf7d9a8cSDavid Xu 		}
1765cf7d9a8cSDavid Xu 		if (!KSI_ONQ(&nt->nt_ksi)) {
1766cf7d9a8cSDavid Xu 			ksiginfo_set_sigev(&nt->nt_ksi, &nt->nt_sigev);
1767cf7d9a8cSDavid Xu 			tdsendsignal(p, td, nt->nt_ksi.ksi_signo, &nt->nt_ksi);
1768cf7d9a8cSDavid Xu 		}
1769b2f92ef9SDavid Xu 		PROC_UNLOCK(p);
1770052ea11cSDavid Xu 	}
1771655291f2SDavid Xu 	mq->mq_notifier = NULL;
1772655291f2SDavid Xu }
1773655291f2SDavid Xu 
1774655291f2SDavid Xu /*
1775655291f2SDavid Xu  * Get a message. if waitok is false, thread will not be
1776655291f2SDavid Xu  * blocked if there is no data in queue, otherwise, absolute
1777655291f2SDavid Xu  * time will be checked.
1778655291f2SDavid Xu  */
1779655291f2SDavid Xu int
1780655291f2SDavid Xu mqueue_receive(struct mqueue *mq, char *msg_ptr,
1781655291f2SDavid Xu 	size_t msg_len, unsigned *msg_prio, int waitok,
1782655291f2SDavid Xu 	const struct timespec *abs_timeout)
1783655291f2SDavid Xu {
1784655291f2SDavid Xu 	struct mqueue_msg *msg;
1785afde2b65SKonstantin Belousov 	struct timespec ts, ts2;
1786655291f2SDavid Xu 	struct timeval tv;
1787655291f2SDavid Xu 	int error;
1788655291f2SDavid Xu 
1789655291f2SDavid Xu 	if (msg_len < mq->mq_msgsize)
1790655291f2SDavid Xu 		return (EMSGSIZE);
1791655291f2SDavid Xu 
1792655291f2SDavid Xu 	/* O_NONBLOCK case */
1793655291f2SDavid Xu 	if (!waitok) {
1794655291f2SDavid Xu 		error = _mqueue_recv(mq, &msg, -1);
1795655291f2SDavid Xu 		if (error)
1796655291f2SDavid Xu 			return (error);
1797655291f2SDavid Xu 		goto received;
1798655291f2SDavid Xu 	}
1799655291f2SDavid Xu 
1800655291f2SDavid Xu 	/* we allow a null timeout (wait forever). */
1801655291f2SDavid Xu 	if (abs_timeout == NULL) {
1802655291f2SDavid Xu 		error = _mqueue_recv(mq, &msg, 0);
1803655291f2SDavid Xu 		if (error)
1804655291f2SDavid Xu 			return (error);
1805655291f2SDavid Xu 		goto received;
1806655291f2SDavid Xu 	}
1807655291f2SDavid Xu 
1808655291f2SDavid Xu 	/* try to get a message before checking time */
1809655291f2SDavid Xu 	error = _mqueue_recv(mq, &msg, -1);
1810655291f2SDavid Xu 	if (error == 0)
1811655291f2SDavid Xu 		goto received;
1812655291f2SDavid Xu 
1813655291f2SDavid Xu 	if (error != EAGAIN)
1814655291f2SDavid Xu 		return (error);
1815655291f2SDavid Xu 
1816afde2b65SKonstantin Belousov 	if (abs_timeout->tv_nsec >= 1000000000 || abs_timeout->tv_nsec < 0) {
1817655291f2SDavid Xu 		error = EINVAL;
1818655291f2SDavid Xu 		return (error);
1819655291f2SDavid Xu 	}
1820655291f2SDavid Xu 
1821655291f2SDavid Xu 	for (;;) {
1822afde2b65SKonstantin Belousov 		ts2 = *abs_timeout;
1823a6de716dSDavid Xu 		getnanotime(&ts);
1824655291f2SDavid Xu 		timespecsub(&ts2, &ts);
1825655291f2SDavid Xu 		if (ts2.tv_sec < 0 || (ts2.tv_sec == 0 && ts2.tv_nsec <= 0)) {
1826655291f2SDavid Xu 			error = ETIMEDOUT;
1827655291f2SDavid Xu 			return (error);
1828655291f2SDavid Xu 		}
1829655291f2SDavid Xu 		TIMESPEC_TO_TIMEVAL(&tv, &ts2);
1830655291f2SDavid Xu 		error = _mqueue_recv(mq, &msg, tvtohz(&tv));
1831655291f2SDavid Xu 		if (error == 0)
1832655291f2SDavid Xu 			break;
1833655291f2SDavid Xu 		if (error != ETIMEDOUT)
1834655291f2SDavid Xu 			return (error);
1835655291f2SDavid Xu 	}
1836655291f2SDavid Xu 
1837655291f2SDavid Xu received:
1838655291f2SDavid Xu 	error = mqueue_savemsg(msg, msg_ptr, msg_prio);
1839655291f2SDavid Xu 	if (error == 0) {
1840655291f2SDavid Xu 		curthread->td_retval[0] = msg->msg_size;
1841655291f2SDavid Xu 		curthread->td_retval[1] = 0;
1842655291f2SDavid Xu 	}
1843655291f2SDavid Xu 	mqueue_freemsg(msg);
1844655291f2SDavid Xu 	return (error);
1845655291f2SDavid Xu }
1846655291f2SDavid Xu 
1847655291f2SDavid Xu /*
1848655291f2SDavid Xu  * Common routine to receive a message
1849655291f2SDavid Xu  */
1850655291f2SDavid Xu static int
1851655291f2SDavid Xu _mqueue_recv(struct mqueue *mq, struct mqueue_msg **msg, int timo)
1852655291f2SDavid Xu {
1853655291f2SDavid Xu 	int error = 0;
1854655291f2SDavid Xu 
1855655291f2SDavid Xu 	mtx_lock(&mq->mq_mutex);
1856655291f2SDavid Xu 	while ((*msg = TAILQ_FIRST(&mq->mq_msgq)) == NULL && error == 0) {
1857655291f2SDavid Xu 		if (timo < 0) {
1858655291f2SDavid Xu 			mtx_unlock(&mq->mq_mutex);
1859655291f2SDavid Xu 			return (EAGAIN);
1860655291f2SDavid Xu 		}
1861655291f2SDavid Xu 		mq->mq_receivers++;
1862655291f2SDavid Xu 		error = msleep(&mq->mq_receivers, &mq->mq_mutex,
18630f180a7cSJohn Baldwin 			    PCATCH, "mqrecv", timo);
1864655291f2SDavid Xu 		mq->mq_receivers--;
1865655291f2SDavid Xu 		if (error == EAGAIN)
1866655291f2SDavid Xu 			error = ETIMEDOUT;
1867655291f2SDavid Xu 	}
1868655291f2SDavid Xu 	if (*msg != NULL) {
1869655291f2SDavid Xu 		error = 0;
1870655291f2SDavid Xu 		TAILQ_REMOVE(&mq->mq_msgq, *msg, msg_link);
1871655291f2SDavid Xu 		mq->mq_curmsgs--;
1872655291f2SDavid Xu 		mq->mq_totalbytes -= (*msg)->msg_size;
1873655291f2SDavid Xu 		if (mq->mq_senders)
1874655291f2SDavid Xu 			wakeup_one(&mq->mq_senders);
1875655291f2SDavid Xu 		if (mq->mq_flags & MQ_WSEL) {
1876655291f2SDavid Xu 			mq->mq_flags &= ~MQ_WSEL;
187703f70aecSDavid Xu 			selwakeup(&mq->mq_wsel);
1878655291f2SDavid Xu 		}
1879655291f2SDavid Xu 		KNOTE_LOCKED(&mq->mq_wsel.si_note, 0);
1880655291f2SDavid Xu 	}
1881655291f2SDavid Xu 	if (mq->mq_notifier != NULL && mq->mq_receivers == 0 &&
1882655291f2SDavid Xu 	    !TAILQ_EMPTY(&mq->mq_msgq)) {
1883655291f2SDavid Xu 		mqueue_send_notification(mq);
1884655291f2SDavid Xu 	}
1885655291f2SDavid Xu 	mtx_unlock(&mq->mq_mutex);
1886655291f2SDavid Xu 	return (error);
1887655291f2SDavid Xu }
1888655291f2SDavid Xu 
1889b2f92ef9SDavid Xu static __inline struct mqueue_notifier *
1890b2f92ef9SDavid Xu notifier_alloc(void)
1891655291f2SDavid Xu {
1892b2f92ef9SDavid Xu 	return (uma_zalloc(mqnoti_zone, M_WAITOK | M_ZERO));
1893655291f2SDavid Xu }
1894655291f2SDavid Xu 
1895655291f2SDavid Xu static __inline void
1896b2f92ef9SDavid Xu notifier_free(struct mqueue_notifier *p)
1897655291f2SDavid Xu {
1898b2f92ef9SDavid Xu 	uma_zfree(mqnoti_zone, p);
1899b2f92ef9SDavid Xu }
1900b2f92ef9SDavid Xu 
1901b2f92ef9SDavid Xu static struct mqueue_notifier *
1902b2f92ef9SDavid Xu notifier_search(struct proc *p, int fd)
1903b2f92ef9SDavid Xu {
1904b2f92ef9SDavid Xu 	struct mqueue_notifier *nt;
1905b2f92ef9SDavid Xu 
1906b2f92ef9SDavid Xu 	LIST_FOREACH(nt, &p->p_mqnotifier, nt_link) {
19079da8a32aSDavid Xu 		if (nt->nt_ksi.ksi_mqd == fd)
1908b2f92ef9SDavid Xu 			break;
1909b2f92ef9SDavid Xu 	}
1910b2f92ef9SDavid Xu 	return (nt);
1911b2f92ef9SDavid Xu }
1912b2f92ef9SDavid Xu 
1913102178d0SDavid Xu static __inline void
1914b2f92ef9SDavid Xu notifier_insert(struct proc *p, struct mqueue_notifier *nt)
1915b2f92ef9SDavid Xu {
1916b2f92ef9SDavid Xu 	LIST_INSERT_HEAD(&p->p_mqnotifier, nt, nt_link);
1917b2f92ef9SDavid Xu }
1918b2f92ef9SDavid Xu 
1919102178d0SDavid Xu static __inline void
1920b2f92ef9SDavid Xu notifier_delete(struct proc *p, struct mqueue_notifier *nt)
1921b2f92ef9SDavid Xu {
1922b2f92ef9SDavid Xu 	LIST_REMOVE(nt, nt_link);
1923b2f92ef9SDavid Xu 	notifier_free(nt);
1924b2f92ef9SDavid Xu }
1925b2f92ef9SDavid Xu 
1926b2f92ef9SDavid Xu static void
1927b2f92ef9SDavid Xu notifier_remove(struct proc *p, struct mqueue *mq, int fd)
1928b2f92ef9SDavid Xu {
1929b2f92ef9SDavid Xu 	struct mqueue_notifier *nt;
1930b2f92ef9SDavid Xu 
1931b2f92ef9SDavid Xu 	mtx_assert(&mq->mq_mutex, MA_OWNED);
1932b2f92ef9SDavid Xu 	PROC_LOCK(p);
1933b2f92ef9SDavid Xu 	nt = notifier_search(p, fd);
1934b2f92ef9SDavid Xu 	if (nt != NULL) {
1935b2f92ef9SDavid Xu 		if (mq->mq_notifier == nt)
1936b2f92ef9SDavid Xu 			mq->mq_notifier = NULL;
1937b2f92ef9SDavid Xu 		sigqueue_take(&nt->nt_ksi);
1938b2f92ef9SDavid Xu 		notifier_delete(p, nt);
1939b2f92ef9SDavid Xu 	}
1940b2f92ef9SDavid Xu 	PROC_UNLOCK(p);
1941655291f2SDavid Xu }
1942655291f2SDavid Xu 
1943afde2b65SKonstantin Belousov static int
1944afde2b65SKonstantin Belousov kern_kmq_open(struct thread *td, const char *upath, int flags, mode_t mode,
1945afde2b65SKonstantin Belousov     const struct mq_attr *attr)
1946655291f2SDavid Xu {
1947655291f2SDavid Xu 	char path[MQFS_NAMELEN + 1];
1948655291f2SDavid Xu 	struct mqfs_node *pn;
1949655291f2SDavid Xu 	struct filedesc *fdp;
1950655291f2SDavid Xu 	struct file *fp;
1951655291f2SDavid Xu 	struct mqueue *mq;
1952afde2b65SKonstantin Belousov 	int fd, error, len, cmode;
1953655291f2SDavid Xu 
1954655291f2SDavid Xu 	fdp = td->td_proc->p_fd;
1955afde2b65SKonstantin Belousov 	cmode = (((mode & ~fdp->fd_cmask) & ALLPERMS) & ~S_ISTXT);
1956f72b11a4SDavid Xu 	mq = NULL;
1957afde2b65SKonstantin Belousov 	if ((flags & O_CREAT) != 0 && attr != NULL) {
1958afde2b65SKonstantin Belousov 		if (attr->mq_maxmsg <= 0 || attr->mq_maxmsg > maxmsg)
1959655291f2SDavid Xu 			return (EINVAL);
1960afde2b65SKonstantin Belousov 		if (attr->mq_msgsize <= 0 || attr->mq_msgsize > maxmsgsize)
1961655291f2SDavid Xu 			return (EINVAL);
1962afde2b65SKonstantin Belousov 	}
1963655291f2SDavid Xu 
1964afde2b65SKonstantin Belousov 	error = copyinstr(upath, path, MQFS_NAMELEN + 1, NULL);
1965655291f2SDavid Xu         if (error)
1966655291f2SDavid Xu 		return (error);
1967655291f2SDavid Xu 
1968655291f2SDavid Xu 	/*
1969655291f2SDavid Xu 	 * The first character of name must be a slash  (/) character
1970655291f2SDavid Xu 	 * and the remaining characters of name cannot include any slash
1971655291f2SDavid Xu 	 * characters.
1972655291f2SDavid Xu 	 */
1973655291f2SDavid Xu 	len = strlen(path);
1974655291f2SDavid Xu 	if (len < 2  || path[0] != '/' || index(path + 1, '/') != NULL)
1975655291f2SDavid Xu 		return (EINVAL);
1976655291f2SDavid Xu 
1977*1fe80828SKonstantin Belousov 	error = falloc(td, &fp, &fd, 0);
1978655291f2SDavid Xu 	if (error)
1979655291f2SDavid Xu 		return (error);
1980655291f2SDavid Xu 
1981655291f2SDavid Xu 	sx_xlock(&mqfs_data.mi_lock);
1982655291f2SDavid Xu 	pn = mqfs_search(mqfs_data.mi_root, path + 1, len - 1);
1983655291f2SDavid Xu 	if (pn == NULL) {
1984655291f2SDavid Xu 		if (!(flags & O_CREAT)) {
1985655291f2SDavid Xu 			error = ENOENT;
1986655291f2SDavid Xu 		} else {
1987afde2b65SKonstantin Belousov 			mq = mqueue_alloc(attr);
1988b2f92ef9SDavid Xu 			if (mq == NULL) {
1989b2f92ef9SDavid Xu 				error = ENFILE;
1990b2f92ef9SDavid Xu 			} else {
1991655291f2SDavid Xu 				pn = mqfs_create_file(mqfs_data.mi_root,
1992ba0360b1SDavid Xu 				         path + 1, len - 1, td->td_ucred,
1993ba0360b1SDavid Xu 					 cmode);
1994655291f2SDavid Xu 				if (pn == NULL) {
1995655291f2SDavid Xu 					error = ENOSPC;
1996b2f92ef9SDavid Xu 					mqueue_free(mq);
1997b2f92ef9SDavid Xu 				}
1998655291f2SDavid Xu 			}
1999655291f2SDavid Xu 		}
2000655291f2SDavid Xu 
2001655291f2SDavid Xu 		if (error == 0) {
2002655291f2SDavid Xu 			pn->mn_data = mq;
2003655291f2SDavid Xu 		}
2004655291f2SDavid Xu 	} else {
2005655291f2SDavid Xu 		if ((flags & (O_CREAT | O_EXCL)) == (O_CREAT | O_EXCL)) {
2006655291f2SDavid Xu 			error = EEXIST;
2007655291f2SDavid Xu 		} else {
200815bc6b2bSEdward Tomasz Napierala 			accmode_t accmode = 0;
2009655291f2SDavid Xu 
2010655291f2SDavid Xu 			if (flags & FREAD)
201115bc6b2bSEdward Tomasz Napierala 				accmode |= VREAD;
2012655291f2SDavid Xu 			if (flags & FWRITE)
201315bc6b2bSEdward Tomasz Napierala 				accmode |= VWRITE;
2014655291f2SDavid Xu 			error = vaccess(VREG, pn->mn_mode, pn->mn_uid,
201515bc6b2bSEdward Tomasz Napierala 				    pn->mn_gid, accmode, td->td_ucred, NULL);
2016655291f2SDavid Xu 		}
2017655291f2SDavid Xu 	}
2018655291f2SDavid Xu 
2019655291f2SDavid Xu 	if (error) {
2020655291f2SDavid Xu 		sx_xunlock(&mqfs_data.mi_lock);
2021655291f2SDavid Xu 		fdclose(fdp, fp, fd, td);
2022655291f2SDavid Xu 		fdrop(fp, td);
2023655291f2SDavid Xu 		return (error);
2024655291f2SDavid Xu 	}
2025655291f2SDavid Xu 
2026655291f2SDavid Xu 	mqnode_addref(pn);
2027655291f2SDavid Xu 	sx_xunlock(&mqfs_data.mi_lock);
2028655291f2SDavid Xu 
2029397c19d1SJeff Roberson 	finit(fp, flags & (FREAD | FWRITE | O_NONBLOCK), DTYPE_MQUEUE, pn,
2030397c19d1SJeff Roberson 	    &mqueueops);
2031655291f2SDavid Xu 
20325e3f7694SRobert Watson 	FILEDESC_XLOCK(fdp);
2033655291f2SDavid Xu 	if (fdp->fd_ofiles[fd] == fp)
2034655291f2SDavid Xu 		fdp->fd_ofileflags[fd] |= UF_EXCLOSE;
20355e3f7694SRobert Watson 	FILEDESC_XUNLOCK(fdp);
2036655291f2SDavid Xu 	td->td_retval[0] = fd;
2037655291f2SDavid Xu 	fdrop(fp, td);
2038655291f2SDavid Xu 	return (0);
2039655291f2SDavid Xu }
2040655291f2SDavid Xu 
2041655291f2SDavid Xu /*
2042afde2b65SKonstantin Belousov  * Syscall to open a message queue.
2043afde2b65SKonstantin Belousov  */
2044afde2b65SKonstantin Belousov int
2045afde2b65SKonstantin Belousov kmq_open(struct thread *td, struct kmq_open_args *uap)
2046afde2b65SKonstantin Belousov {
2047afde2b65SKonstantin Belousov 	struct mq_attr attr;
2048afde2b65SKonstantin Belousov 	int flags, error;
2049afde2b65SKonstantin Belousov 
2050afde2b65SKonstantin Belousov 	if ((uap->flags & O_ACCMODE) == O_ACCMODE)
2051afde2b65SKonstantin Belousov 		return (EINVAL);
2052afde2b65SKonstantin Belousov 	flags = FFLAGS(uap->flags);
2053afde2b65SKonstantin Belousov 	if ((flags & O_CREAT) != 0 && uap->attr != NULL) {
2054afde2b65SKonstantin Belousov 		error = copyin(uap->attr, &attr, sizeof(attr));
2055afde2b65SKonstantin Belousov 		if (error)
2056afde2b65SKonstantin Belousov 			return (error);
2057afde2b65SKonstantin Belousov 	}
2058afde2b65SKonstantin Belousov 	return (kern_kmq_open(td, uap->path, flags, uap->mode,
2059afde2b65SKonstantin Belousov 	    uap->attr != NULL ? &attr : NULL));
2060afde2b65SKonstantin Belousov }
2061afde2b65SKonstantin Belousov 
2062afde2b65SKonstantin Belousov /*
2063873fbcd7SRobert Watson  * Syscall to unlink a message queue.
2064655291f2SDavid Xu  */
2065655291f2SDavid Xu int
206661d3a4efSDavid Xu kmq_unlink(struct thread *td, struct kmq_unlink_args *uap)
2067655291f2SDavid Xu {
2068655291f2SDavid Xu 	char path[MQFS_NAMELEN+1];
2069655291f2SDavid Xu 	struct mqfs_node *pn;
2070655291f2SDavid Xu 	int error, len;
2071655291f2SDavid Xu 
2072655291f2SDavid Xu 	error = copyinstr(uap->path, path, MQFS_NAMELEN + 1, NULL);
2073655291f2SDavid Xu         if (error)
2074655291f2SDavid Xu 		return (error);
2075655291f2SDavid Xu 
2076655291f2SDavid Xu 	len = strlen(path);
2077655291f2SDavid Xu 	if (len < 2  || path[0] != '/' || index(path + 1, '/') != NULL)
2078655291f2SDavid Xu 		return (EINVAL);
2079655291f2SDavid Xu 
2080655291f2SDavid Xu 	sx_xlock(&mqfs_data.mi_lock);
2081655291f2SDavid Xu 	pn = mqfs_search(mqfs_data.mi_root, path + 1, len - 1);
2082655291f2SDavid Xu 	if (pn != NULL)
2083655291f2SDavid Xu 		error = do_unlink(pn, td->td_ucred);
2084655291f2SDavid Xu 	else
2085655291f2SDavid Xu 		error = ENOENT;
2086655291f2SDavid Xu 	sx_xunlock(&mqfs_data.mi_lock);
2087655291f2SDavid Xu 	return (error);
2088655291f2SDavid Xu }
2089655291f2SDavid Xu 
2090655291f2SDavid Xu typedef int (*_fgetf)(struct thread *, int, struct file **);
2091655291f2SDavid Xu 
2092655291f2SDavid Xu /*
2093655291f2SDavid Xu  * Get message queue by giving file slot
2094655291f2SDavid Xu  */
2095655291f2SDavid Xu static int
2096655291f2SDavid Xu _getmq(struct thread *td, int fd, _fgetf func,
2097655291f2SDavid Xu        struct file **fpp, struct mqfs_node **ppn, struct mqueue **pmq)
2098655291f2SDavid Xu {
2099655291f2SDavid Xu 	struct mqfs_node *pn;
2100655291f2SDavid Xu 	int error;
2101655291f2SDavid Xu 
2102655291f2SDavid Xu 	error = func(td, fd, fpp);
2103655291f2SDavid Xu 	if (error)
2104655291f2SDavid Xu 		return (error);
2105655291f2SDavid Xu 	if (&mqueueops != (*fpp)->f_ops) {
2106655291f2SDavid Xu 		fdrop(*fpp, td);
2107655291f2SDavid Xu 		return (EBADF);
2108655291f2SDavid Xu 	}
2109b2f92ef9SDavid Xu 	pn = (*fpp)->f_data;
2110655291f2SDavid Xu 	if (ppn)
2111655291f2SDavid Xu 		*ppn = pn;
2112655291f2SDavid Xu 	if (pmq)
2113655291f2SDavid Xu 		*pmq = pn->mn_data;
2114655291f2SDavid Xu 	return (0);
2115655291f2SDavid Xu }
2116655291f2SDavid Xu 
2117655291f2SDavid Xu static __inline int
2118655291f2SDavid Xu getmq(struct thread *td, int fd, struct file **fpp, struct mqfs_node **ppn,
2119655291f2SDavid Xu 	struct mqueue **pmq)
2120655291f2SDavid Xu {
2121655291f2SDavid Xu 	return _getmq(td, fd, fget, fpp, ppn, pmq);
2122655291f2SDavid Xu }
2123655291f2SDavid Xu 
2124655291f2SDavid Xu static __inline int
2125655291f2SDavid Xu getmq_read(struct thread *td, int fd, struct file **fpp,
2126655291f2SDavid Xu 	 struct mqfs_node **ppn, struct mqueue **pmq)
2127655291f2SDavid Xu {
2128655291f2SDavid Xu 	return _getmq(td, fd, fget_read, fpp, ppn, pmq);
2129655291f2SDavid Xu }
2130655291f2SDavid Xu 
2131655291f2SDavid Xu static __inline int
2132655291f2SDavid Xu getmq_write(struct thread *td, int fd, struct file **fpp,
2133655291f2SDavid Xu 	struct mqfs_node **ppn, struct mqueue **pmq)
2134655291f2SDavid Xu {
2135655291f2SDavid Xu 	return _getmq(td, fd, fget_write, fpp, ppn, pmq);
2136655291f2SDavid Xu }
2137655291f2SDavid Xu 
2138afde2b65SKonstantin Belousov static int
2139afde2b65SKonstantin Belousov kern_kmq_setattr(struct thread *td, int mqd, const struct mq_attr *attr,
2140afde2b65SKonstantin Belousov     struct mq_attr *oattr)
2141655291f2SDavid Xu {
2142655291f2SDavid Xu 	struct mqueue *mq;
2143655291f2SDavid Xu 	struct file *fp;
2144397c19d1SJeff Roberson 	u_int oflag, flag;
2145655291f2SDavid Xu 	int error;
2146655291f2SDavid Xu 
2147afde2b65SKonstantin Belousov 	if (attr != NULL && (attr->mq_flags & ~O_NONBLOCK) != 0)
2148655291f2SDavid Xu 		return (EINVAL);
2149afde2b65SKonstantin Belousov 	error = getmq(td, mqd, &fp, NULL, &mq);
2150655291f2SDavid Xu 	if (error)
2151655291f2SDavid Xu 		return (error);
2152afde2b65SKonstantin Belousov 	oattr->mq_maxmsg  = mq->mq_maxmsg;
2153afde2b65SKonstantin Belousov 	oattr->mq_msgsize = mq->mq_msgsize;
2154afde2b65SKonstantin Belousov 	oattr->mq_curmsgs = mq->mq_curmsgs;
2155afde2b65SKonstantin Belousov 	if (attr != NULL) {
2156397c19d1SJeff Roberson 		do {
2157397c19d1SJeff Roberson 			oflag = flag = fp->f_flag;
2158397c19d1SJeff Roberson 			flag &= ~O_NONBLOCK;
2159afde2b65SKonstantin Belousov 			flag |= (attr->mq_flags & O_NONBLOCK);
2160397c19d1SJeff Roberson 		} while (atomic_cmpset_int(&fp->f_flag, oflag, flag) == 0);
2161397c19d1SJeff Roberson 	} else
2162397c19d1SJeff Roberson 		oflag = fp->f_flag;
2163afde2b65SKonstantin Belousov 	oattr->mq_flags = (O_NONBLOCK & oflag);
2164655291f2SDavid Xu 	fdrop(fp, td);
2165afde2b65SKonstantin Belousov 	return (error);
2166afde2b65SKonstantin Belousov }
2167afde2b65SKonstantin Belousov 
2168afde2b65SKonstantin Belousov int
2169afde2b65SKonstantin Belousov kmq_setattr(struct thread *td, struct kmq_setattr_args *uap)
2170afde2b65SKonstantin Belousov {
2171afde2b65SKonstantin Belousov 	struct mq_attr attr, oattr;
2172afde2b65SKonstantin Belousov 	int error;
2173afde2b65SKonstantin Belousov 
2174afde2b65SKonstantin Belousov 	if (uap->attr != NULL) {
2175afde2b65SKonstantin Belousov 		error = copyin(uap->attr, &attr, sizeof(attr));
2176afde2b65SKonstantin Belousov 		if (error != 0)
2177afde2b65SKonstantin Belousov 			return (error);
2178afde2b65SKonstantin Belousov 	}
2179afde2b65SKonstantin Belousov 	error = kern_kmq_setattr(td, uap->mqd, uap->attr != NULL ? &attr : NULL,
2180afde2b65SKonstantin Belousov 	    &oattr);
2181afde2b65SKonstantin Belousov 	if (error != 0)
2182afde2b65SKonstantin Belousov 		return (error);
2183afde2b65SKonstantin Belousov 	if (uap->oattr != NULL)
2184655291f2SDavid Xu 		error = copyout(&oattr, uap->oattr, sizeof(oattr));
2185655291f2SDavid Xu 	return (error);
2186655291f2SDavid Xu }
2187655291f2SDavid Xu 
2188655291f2SDavid Xu int
218961d3a4efSDavid Xu kmq_timedreceive(struct thread *td, struct kmq_timedreceive_args *uap)
2190655291f2SDavid Xu {
2191655291f2SDavid Xu 	struct mqueue *mq;
2192655291f2SDavid Xu 	struct file *fp;
2193afde2b65SKonstantin Belousov 	struct timespec *abs_timeout, ets;
2194655291f2SDavid Xu 	int error;
2195655291f2SDavid Xu 	int waitok;
2196655291f2SDavid Xu 
2197655291f2SDavid Xu 	error = getmq_read(td, uap->mqd, &fp, NULL, &mq);
2198655291f2SDavid Xu 	if (error)
2199655291f2SDavid Xu 		return (error);
2200afde2b65SKonstantin Belousov 	if (uap->abs_timeout != NULL) {
2201afde2b65SKonstantin Belousov 		error = copyin(uap->abs_timeout, &ets, sizeof(ets));
2202afde2b65SKonstantin Belousov 		if (error != 0)
2203afde2b65SKonstantin Belousov 			return (error);
2204afde2b65SKonstantin Belousov 		abs_timeout = &ets;
2205afde2b65SKonstantin Belousov 	} else
2206afde2b65SKonstantin Belousov 		abs_timeout = NULL;
2207655291f2SDavid Xu 	waitok = !(fp->f_flag & O_NONBLOCK);
2208655291f2SDavid Xu 	error = mqueue_receive(mq, uap->msg_ptr, uap->msg_len,
2209afde2b65SKonstantin Belousov 		uap->msg_prio, waitok, abs_timeout);
2210655291f2SDavid Xu 	fdrop(fp, td);
2211655291f2SDavid Xu 	return (error);
2212655291f2SDavid Xu }
2213655291f2SDavid Xu 
2214655291f2SDavid Xu int
221561d3a4efSDavid Xu kmq_timedsend(struct thread *td, struct kmq_timedsend_args *uap)
2216655291f2SDavid Xu {
2217655291f2SDavid Xu 	struct mqueue *mq;
2218655291f2SDavid Xu 	struct file *fp;
2219afde2b65SKonstantin Belousov 	struct timespec *abs_timeout, ets;
2220655291f2SDavid Xu 	int error, waitok;
2221655291f2SDavid Xu 
2222655291f2SDavid Xu 	error = getmq_write(td, uap->mqd, &fp, NULL, &mq);
2223655291f2SDavid Xu 	if (error)
2224655291f2SDavid Xu 		return (error);
2225afde2b65SKonstantin Belousov 	if (uap->abs_timeout != NULL) {
2226afde2b65SKonstantin Belousov 		error = copyin(uap->abs_timeout, &ets, sizeof(ets));
2227afde2b65SKonstantin Belousov 		if (error != 0)
2228afde2b65SKonstantin Belousov 			return (error);
2229afde2b65SKonstantin Belousov 		abs_timeout = &ets;
2230afde2b65SKonstantin Belousov 	} else
2231afde2b65SKonstantin Belousov 		abs_timeout = NULL;
2232655291f2SDavid Xu 	waitok = !(fp->f_flag & O_NONBLOCK);
2233655291f2SDavid Xu 	error = mqueue_send(mq, uap->msg_ptr, uap->msg_len,
2234afde2b65SKonstantin Belousov 		uap->msg_prio, waitok, abs_timeout);
2235655291f2SDavid Xu 	fdrop(fp, td);
2236655291f2SDavid Xu 	return (error);
2237655291f2SDavid Xu }
2238655291f2SDavid Xu 
2239655291f2SDavid Xu int
224061d3a4efSDavid Xu kmq_notify(struct thread *td, struct kmq_notify_args *uap)
2241655291f2SDavid Xu {
2242655291f2SDavid Xu 	struct sigevent ev;
2243b2f92ef9SDavid Xu 	struct filedesc *fdp;
2244b2f92ef9SDavid Xu 	struct proc *p;
2245655291f2SDavid Xu 	struct mqueue *mq;
2246655291f2SDavid Xu 	struct file *fp;
2247b2f92ef9SDavid Xu 	struct mqueue_notifier *nt, *newnt = NULL;
2248655291f2SDavid Xu 	int error;
2249655291f2SDavid Xu 
2250b2f92ef9SDavid Xu 	p = td->td_proc;
2251b2f92ef9SDavid Xu 	fdp = td->td_proc->p_fd;
2252655291f2SDavid Xu 	if (uap->sigev) {
2253655291f2SDavid Xu 		error = copyin(uap->sigev, &ev, sizeof(ev));
2254655291f2SDavid Xu 		if (error)
2255655291f2SDavid Xu 			return (error);
2256655291f2SDavid Xu 		if (ev.sigev_notify != SIGEV_SIGNAL &&
22579947b459SDavid Xu 		    ev.sigev_notify != SIGEV_THREAD_ID &&
22589947b459SDavid Xu 		    ev.sigev_notify != SIGEV_NONE)
2259655291f2SDavid Xu 			return (EINVAL);
2260655291f2SDavid Xu 		if ((ev.sigev_notify == SIGEV_SIGNAL ||
2261655291f2SDavid Xu 		     ev.sigev_notify == SIGEV_THREAD_ID) &&
2262655291f2SDavid Xu 			!_SIG_VALID(ev.sigev_signo))
2263655291f2SDavid Xu 			return (EINVAL);
2264655291f2SDavid Xu 	}
2265655291f2SDavid Xu 	error = getmq(td, uap->mqd, &fp, NULL, &mq);
2266655291f2SDavid Xu 	if (error)
2267655291f2SDavid Xu 		return (error);
2268b2f92ef9SDavid Xu again:
22695e3f7694SRobert Watson 	FILEDESC_SLOCK(fdp);
2270b2f92ef9SDavid Xu 	if (fget_locked(fdp, uap->mqd) != fp) {
22715e3f7694SRobert Watson 		FILEDESC_SUNLOCK(fdp);
2272b2f92ef9SDavid Xu 		error = EBADF;
2273b2f92ef9SDavid Xu 		goto out;
2274b2f92ef9SDavid Xu 	}
2275655291f2SDavid Xu 	mtx_lock(&mq->mq_mutex);
22765e3f7694SRobert Watson 	FILEDESC_SUNLOCK(fdp);
2277655291f2SDavid Xu 	if (uap->sigev != NULL) {
2278655291f2SDavid Xu 		if (mq->mq_notifier != NULL) {
2279655291f2SDavid Xu 			error = EBUSY;
2280052ea11cSDavid Xu 		} else {
2281b2f92ef9SDavid Xu 			PROC_LOCK(p);
2282b2f92ef9SDavid Xu 			nt = notifier_search(p, uap->mqd);
2283b2f92ef9SDavid Xu 			if (nt == NULL) {
2284b2f92ef9SDavid Xu 				if (newnt == NULL) {
2285b2f92ef9SDavid Xu 					PROC_UNLOCK(p);
2286b2f92ef9SDavid Xu 					mtx_unlock(&mq->mq_mutex);
2287b2f92ef9SDavid Xu 					newnt = notifier_alloc();
2288b2f92ef9SDavid Xu 					goto again;
2289b2f92ef9SDavid Xu 				}
2290b2f92ef9SDavid Xu 			}
2291b2f92ef9SDavid Xu 
2292b2f92ef9SDavid Xu 			if (nt != NULL) {
2293b2f92ef9SDavid Xu 				sigqueue_take(&nt->nt_ksi);
2294b2f92ef9SDavid Xu 				if (newnt != NULL) {
2295b2f92ef9SDavid Xu 					notifier_free(newnt);
2296b2f92ef9SDavid Xu 					newnt = NULL;
2297b2f92ef9SDavid Xu 				}
2298b2f92ef9SDavid Xu 			} else {
2299b2f92ef9SDavid Xu 				nt = newnt;
2300b2f92ef9SDavid Xu 				newnt = NULL;
2301b2f92ef9SDavid Xu 				ksiginfo_init(&nt->nt_ksi);
2302b2f92ef9SDavid Xu 				nt->nt_ksi.ksi_flags |= KSI_INS | KSI_EXT;
2303b2f92ef9SDavid Xu 				nt->nt_ksi.ksi_code = SI_MESGQ;
2304b2f92ef9SDavid Xu 				nt->nt_proc = p;
23059da8a32aSDavid Xu 				nt->nt_ksi.ksi_mqd = uap->mqd;
2306b2f92ef9SDavid Xu 				notifier_insert(p, nt);
2307b2f92ef9SDavid Xu 			}
2308b2f92ef9SDavid Xu 			nt->nt_sigev = ev;
2309b2f92ef9SDavid Xu 			mq->mq_notifier = nt;
2310b2f92ef9SDavid Xu 			PROC_UNLOCK(p);
2311655291f2SDavid Xu 			/*
2312b2f92ef9SDavid Xu 			 * if there is no receivers and message queue
2313b2f92ef9SDavid Xu 			 * is not empty, we should send notification
2314b2f92ef9SDavid Xu 			 * as soon as possible.
2315655291f2SDavid Xu 			 */
2316655291f2SDavid Xu 			if (mq->mq_receivers == 0 &&
2317655291f2SDavid Xu 			    !TAILQ_EMPTY(&mq->mq_msgq))
2318655291f2SDavid Xu 				mqueue_send_notification(mq);
2319655291f2SDavid Xu 		}
2320655291f2SDavid Xu 	} else {
2321b2f92ef9SDavid Xu 		notifier_remove(p, mq, uap->mqd);
2322655291f2SDavid Xu 	}
2323655291f2SDavid Xu 	mtx_unlock(&mq->mq_mutex);
2324b2f92ef9SDavid Xu 
2325b2f92ef9SDavid Xu out:
2326655291f2SDavid Xu 	fdrop(fp, td);
2327b2f92ef9SDavid Xu 	if (newnt != NULL)
2328b2f92ef9SDavid Xu 		notifier_free(newnt);
2329655291f2SDavid Xu 	return (error);
2330655291f2SDavid Xu }
2331655291f2SDavid Xu 
2332b2f92ef9SDavid Xu static void
2333b2f92ef9SDavid Xu mqueue_fdclose(struct thread *td, int fd, struct file *fp)
2334b2f92ef9SDavid Xu {
2335b2f92ef9SDavid Xu 	struct filedesc *fdp;
2336b2f92ef9SDavid Xu 	struct mqueue *mq;
2337b2f92ef9SDavid Xu 
2338b2f92ef9SDavid Xu 	fdp = td->td_proc->p_fd;
23395e3f7694SRobert Watson 	FILEDESC_LOCK_ASSERT(fdp);
23405e3f7694SRobert Watson 
2341b2f92ef9SDavid Xu 	if (fp->f_ops == &mqueueops) {
2342b2f92ef9SDavid Xu 		mq = FPTOMQ(fp);
2343b2f92ef9SDavid Xu 		mtx_lock(&mq->mq_mutex);
2344b2f92ef9SDavid Xu 		notifier_remove(td->td_proc, mq, fd);
2345b2f92ef9SDavid Xu 
2346b2f92ef9SDavid Xu 		/* have to wakeup thread in same process */
2347b2f92ef9SDavid Xu 		if (mq->mq_flags & MQ_RSEL) {
2348b2f92ef9SDavid Xu 			mq->mq_flags &= ~MQ_RSEL;
234903f70aecSDavid Xu 			selwakeup(&mq->mq_rsel);
2350b2f92ef9SDavid Xu 		}
2351b2f92ef9SDavid Xu 		if (mq->mq_flags & MQ_WSEL) {
2352b2f92ef9SDavid Xu 			mq->mq_flags &= ~MQ_WSEL;
235303f70aecSDavid Xu 			selwakeup(&mq->mq_wsel);
2354b2f92ef9SDavid Xu 		}
2355b2f92ef9SDavid Xu 		mtx_unlock(&mq->mq_mutex);
2356b2f92ef9SDavid Xu 	}
2357b2f92ef9SDavid Xu }
2358b2f92ef9SDavid Xu 
2359b2f92ef9SDavid Xu static void
2360b2f92ef9SDavid Xu mq_proc_exit(void *arg __unused, struct proc *p)
2361b2f92ef9SDavid Xu {
2362b2f92ef9SDavid Xu 	struct filedesc *fdp;
2363b2f92ef9SDavid Xu 	struct file *fp;
2364b2f92ef9SDavid Xu 	struct mqueue *mq;
2365b2f92ef9SDavid Xu 	int i;
2366b2f92ef9SDavid Xu 
2367b2f92ef9SDavid Xu 	fdp = p->p_fd;
23685e3f7694SRobert Watson 	FILEDESC_SLOCK(fdp);
2369b2f92ef9SDavid Xu 	for (i = 0; i < fdp->fd_nfiles; ++i) {
2370b2f92ef9SDavid Xu 		fp = fget_locked(fdp, i);
2371b2f92ef9SDavid Xu 		if (fp != NULL && fp->f_ops == &mqueueops) {
2372b2f92ef9SDavid Xu 			mq = FPTOMQ(fp);
2373b2f92ef9SDavid Xu 			mtx_lock(&mq->mq_mutex);
2374b2f92ef9SDavid Xu 			notifier_remove(p, FPTOMQ(fp), i);
2375b2f92ef9SDavid Xu 			mtx_unlock(&mq->mq_mutex);
2376b2f92ef9SDavid Xu 		}
2377b2f92ef9SDavid Xu 	}
23785e3f7694SRobert Watson 	FILEDESC_SUNLOCK(fdp);
2379b2f92ef9SDavid Xu 	KASSERT(LIST_EMPTY(&p->p_mqnotifier), ("mq notifiers left"));
2380b2f92ef9SDavid Xu }
2381b2f92ef9SDavid Xu 
2382655291f2SDavid Xu static int
2383655291f2SDavid Xu mqf_read(struct file *fp, struct uio *uio, struct ucred *active_cred,
2384655291f2SDavid Xu 	int flags, struct thread *td)
2385655291f2SDavid Xu {
2386655291f2SDavid Xu 	return (EOPNOTSUPP);
2387655291f2SDavid Xu }
2388655291f2SDavid Xu 
2389655291f2SDavid Xu static int
2390655291f2SDavid Xu mqf_write(struct file *fp, struct uio *uio, struct ucred *active_cred,
2391655291f2SDavid Xu 	int flags, struct thread *td)
2392655291f2SDavid Xu {
2393655291f2SDavid Xu 	return (EOPNOTSUPP);
2394655291f2SDavid Xu }
2395655291f2SDavid Xu 
2396655291f2SDavid Xu static int
2397e4650294SJohn Baldwin mqf_truncate(struct file *fp, off_t length, struct ucred *active_cred,
2398e4650294SJohn Baldwin     struct thread *td)
2399e4650294SJohn Baldwin {
2400e4650294SJohn Baldwin 
2401e4650294SJohn Baldwin 	return (EINVAL);
2402e4650294SJohn Baldwin }
2403e4650294SJohn Baldwin 
2404e4650294SJohn Baldwin static int
2405655291f2SDavid Xu mqf_ioctl(struct file *fp, u_long cmd, void *data,
2406655291f2SDavid Xu 	struct ucred *active_cred, struct thread *td)
2407655291f2SDavid Xu {
2408655291f2SDavid Xu 	return (ENOTTY);
2409655291f2SDavid Xu }
2410655291f2SDavid Xu 
2411655291f2SDavid Xu static int
2412655291f2SDavid Xu mqf_poll(struct file *fp, int events, struct ucred *active_cred,
2413655291f2SDavid Xu 	struct thread *td)
2414655291f2SDavid Xu {
2415655291f2SDavid Xu 	struct mqueue *mq = FPTOMQ(fp);
2416655291f2SDavid Xu 	int revents = 0;
2417655291f2SDavid Xu 
2418655291f2SDavid Xu 	mtx_lock(&mq->mq_mutex);
2419655291f2SDavid Xu 	if (events & (POLLIN | POLLRDNORM)) {
2420655291f2SDavid Xu 		if (mq->mq_curmsgs) {
2421655291f2SDavid Xu 			revents |= events & (POLLIN | POLLRDNORM);
2422655291f2SDavid Xu 		} else {
2423655291f2SDavid Xu 			mq->mq_flags |= MQ_RSEL;
2424655291f2SDavid Xu 			selrecord(td, &mq->mq_rsel);
2425655291f2SDavid Xu  		}
2426655291f2SDavid Xu 	}
2427655291f2SDavid Xu 	if (events & POLLOUT) {
2428655291f2SDavid Xu 		if (mq->mq_curmsgs < mq->mq_maxmsg)
2429655291f2SDavid Xu 			revents |= POLLOUT;
2430655291f2SDavid Xu 		else {
2431655291f2SDavid Xu 			mq->mq_flags |= MQ_WSEL;
2432655291f2SDavid Xu 			selrecord(td, &mq->mq_wsel);
2433655291f2SDavid Xu 		}
2434655291f2SDavid Xu 	}
2435655291f2SDavid Xu 	mtx_unlock(&mq->mq_mutex);
2436655291f2SDavid Xu 	return (revents);
2437655291f2SDavid Xu }
2438655291f2SDavid Xu 
2439655291f2SDavid Xu static int
2440655291f2SDavid Xu mqf_close(struct file *fp, struct thread *td)
2441655291f2SDavid Xu {
2442655291f2SDavid Xu 	struct mqfs_node *pn;
2443655291f2SDavid Xu 
2444655291f2SDavid Xu 	fp->f_ops = &badfileops;
2445b2f92ef9SDavid Xu 	pn = fp->f_data;
2446655291f2SDavid Xu 	fp->f_data = NULL;
2447655291f2SDavid Xu 	sx_xlock(&mqfs_data.mi_lock);
2448655291f2SDavid Xu 	mqnode_release(pn);
2449655291f2SDavid Xu 	sx_xunlock(&mqfs_data.mi_lock);
2450655291f2SDavid Xu 	return (0);
2451655291f2SDavid Xu }
2452655291f2SDavid Xu 
2453655291f2SDavid Xu static int
2454655291f2SDavid Xu mqf_stat(struct file *fp, struct stat *st, struct ucred *active_cred,
2455655291f2SDavid Xu 	struct thread *td)
2456655291f2SDavid Xu {
2457b2f92ef9SDavid Xu 	struct mqfs_node *pn = fp->f_data;
2458655291f2SDavid Xu 
2459655291f2SDavid Xu 	bzero(st, sizeof *st);
2460510ea843SEd Schouten 	st->st_atim = pn->mn_atime;
2461510ea843SEd Schouten 	st->st_mtim = pn->mn_mtime;
2462510ea843SEd Schouten 	st->st_ctim = pn->mn_ctime;
2463510ea843SEd Schouten 	st->st_birthtim = pn->mn_birth;
2464655291f2SDavid Xu 	st->st_uid = pn->mn_uid;
2465655291f2SDavid Xu 	st->st_gid = pn->mn_gid;
2466655291f2SDavid Xu 	st->st_mode = S_IFIFO | pn->mn_mode;
2467655291f2SDavid Xu 	return (0);
2468655291f2SDavid Xu }
2469655291f2SDavid Xu 
2470655291f2SDavid Xu static int
2471655291f2SDavid Xu mqf_kqfilter(struct file *fp, struct knote *kn)
2472655291f2SDavid Xu {
2473655291f2SDavid Xu 	struct mqueue *mq = FPTOMQ(fp);
2474655291f2SDavid Xu 	int error = 0;
2475655291f2SDavid Xu 
2476655291f2SDavid Xu 	if (kn->kn_filter == EVFILT_READ) {
2477655291f2SDavid Xu 		kn->kn_fop = &mq_rfiltops;
2478655291f2SDavid Xu 		knlist_add(&mq->mq_rsel.si_note, kn, 0);
2479655291f2SDavid Xu 	} else if (kn->kn_filter == EVFILT_WRITE) {
2480655291f2SDavid Xu 		kn->kn_fop = &mq_wfiltops;
2481655291f2SDavid Xu 		knlist_add(&mq->mq_wsel.si_note, kn, 0);
2482655291f2SDavid Xu 	} else
2483655291f2SDavid Xu 		error = EINVAL;
2484655291f2SDavid Xu 	return (error);
2485655291f2SDavid Xu }
2486655291f2SDavid Xu 
2487655291f2SDavid Xu static void
2488655291f2SDavid Xu filt_mqdetach(struct knote *kn)
2489655291f2SDavid Xu {
2490655291f2SDavid Xu 	struct mqueue *mq = FPTOMQ(kn->kn_fp);
2491655291f2SDavid Xu 
2492655291f2SDavid Xu 	if (kn->kn_filter == EVFILT_READ)
2493655291f2SDavid Xu 		knlist_remove(&mq->mq_rsel.si_note, kn, 0);
2494655291f2SDavid Xu 	else if (kn->kn_filter == EVFILT_WRITE)
2495655291f2SDavid Xu 		knlist_remove(&mq->mq_wsel.si_note, kn, 0);
2496655291f2SDavid Xu 	else
2497655291f2SDavid Xu 		panic("filt_mqdetach");
2498655291f2SDavid Xu }
2499655291f2SDavid Xu 
2500655291f2SDavid Xu static int
2501655291f2SDavid Xu filt_mqread(struct knote *kn, long hint)
2502655291f2SDavid Xu {
2503655291f2SDavid Xu 	struct mqueue *mq = FPTOMQ(kn->kn_fp);
2504655291f2SDavid Xu 
2505655291f2SDavid Xu 	mtx_assert(&mq->mq_mutex, MA_OWNED);
2506655291f2SDavid Xu 	return (mq->mq_curmsgs != 0);
2507655291f2SDavid Xu }
2508655291f2SDavid Xu 
2509655291f2SDavid Xu static int
2510655291f2SDavid Xu filt_mqwrite(struct knote *kn, long hint)
2511655291f2SDavid Xu {
2512655291f2SDavid Xu 	struct mqueue *mq = FPTOMQ(kn->kn_fp);
2513655291f2SDavid Xu 
2514655291f2SDavid Xu 	mtx_assert(&mq->mq_mutex, MA_OWNED);
2515655291f2SDavid Xu 	return (mq->mq_curmsgs < mq->mq_maxmsg);
2516655291f2SDavid Xu }
2517655291f2SDavid Xu 
2518655291f2SDavid Xu static struct fileops mqueueops = {
2519655291f2SDavid Xu 	.fo_read		= mqf_read,
2520655291f2SDavid Xu 	.fo_write		= mqf_write,
2521e4650294SJohn Baldwin 	.fo_truncate		= mqf_truncate,
2522655291f2SDavid Xu 	.fo_ioctl		= mqf_ioctl,
2523655291f2SDavid Xu 	.fo_poll		= mqf_poll,
2524655291f2SDavid Xu 	.fo_kqfilter		= mqf_kqfilter,
2525655291f2SDavid Xu 	.fo_stat		= mqf_stat,
2526655291f2SDavid Xu 	.fo_close		= mqf_close
2527655291f2SDavid Xu };
2528655291f2SDavid Xu 
2529655291f2SDavid Xu static struct vop_vector mqfs_vnodeops = {
2530655291f2SDavid Xu 	.vop_default 		= &default_vnodeops,
2531655291f2SDavid Xu 	.vop_access		= mqfs_access,
2532655291f2SDavid Xu 	.vop_cachedlookup	= mqfs_lookup,
2533655291f2SDavid Xu 	.vop_lookup		= vfs_cache_lookup,
2534655291f2SDavid Xu 	.vop_reclaim		= mqfs_reclaim,
2535655291f2SDavid Xu 	.vop_create		= mqfs_create,
2536655291f2SDavid Xu 	.vop_remove		= mqfs_remove,
2537655291f2SDavid Xu 	.vop_inactive		= mqfs_inactive,
2538655291f2SDavid Xu 	.vop_open		= mqfs_open,
2539655291f2SDavid Xu 	.vop_close		= mqfs_close,
2540655291f2SDavid Xu 	.vop_getattr		= mqfs_getattr,
2541655291f2SDavid Xu 	.vop_setattr		= mqfs_setattr,
2542655291f2SDavid Xu 	.vop_read		= mqfs_read,
2543655291f2SDavid Xu 	.vop_write		= VOP_EOPNOTSUPP,
2544655291f2SDavid Xu 	.vop_readdir		= mqfs_readdir,
2545655291f2SDavid Xu 	.vop_mkdir		= VOP_EOPNOTSUPP,
2546655291f2SDavid Xu 	.vop_rmdir		= VOP_EOPNOTSUPP
2547655291f2SDavid Xu };
2548655291f2SDavid Xu 
2549655291f2SDavid Xu static struct vfsops mqfs_vfsops = {
2550655291f2SDavid Xu 	.vfs_init 		= mqfs_init,
2551655291f2SDavid Xu 	.vfs_uninit		= mqfs_uninit,
2552655291f2SDavid Xu 	.vfs_mount		= mqfs_mount,
2553655291f2SDavid Xu 	.vfs_unmount		= mqfs_unmount,
2554655291f2SDavid Xu 	.vfs_root		= mqfs_root,
2555655291f2SDavid Xu 	.vfs_statfs		= mqfs_statfs,
2556655291f2SDavid Xu };
2557655291f2SDavid Xu 
2558afde2b65SKonstantin Belousov static struct vfsconf mqueuefs_vfsconf = {
2559afde2b65SKonstantin Belousov 	.vfc_version = VFS_VERSION,
2560afde2b65SKonstantin Belousov 	.vfc_name = "mqueuefs",
2561afde2b65SKonstantin Belousov 	.vfc_vfsops = &mqfs_vfsops,
2562afde2b65SKonstantin Belousov 	.vfc_typenum = -1,
2563afde2b65SKonstantin Belousov 	.vfc_flags = VFCF_SYNTHETIC
2564afde2b65SKonstantin Belousov };
2565655291f2SDavid Xu 
2566afde2b65SKonstantin Belousov static struct syscall_helper_data mq_syscalls[] = {
2567afde2b65SKonstantin Belousov 	SYSCALL_INIT_HELPER(kmq_open),
2568afde2b65SKonstantin Belousov 	SYSCALL_INIT_HELPER(kmq_setattr),
2569afde2b65SKonstantin Belousov 	SYSCALL_INIT_HELPER(kmq_timedsend),
2570afde2b65SKonstantin Belousov 	SYSCALL_INIT_HELPER(kmq_timedreceive),
2571afde2b65SKonstantin Belousov 	SYSCALL_INIT_HELPER(kmq_notify),
2572afde2b65SKonstantin Belousov 	SYSCALL_INIT_HELPER(kmq_unlink),
2573afde2b65SKonstantin Belousov 	SYSCALL_INIT_LAST
2574afde2b65SKonstantin Belousov };
2575afde2b65SKonstantin Belousov 
2576afde2b65SKonstantin Belousov #ifdef COMPAT_FREEBSD32
2577afde2b65SKonstantin Belousov #include <compat/freebsd32/freebsd32.h>
2578afde2b65SKonstantin Belousov #include <compat/freebsd32/freebsd32_proto.h>
2579afde2b65SKonstantin Belousov #include <compat/freebsd32/freebsd32_syscall.h>
2580afde2b65SKonstantin Belousov #include <compat/freebsd32/freebsd32_util.h>
2581afde2b65SKonstantin Belousov 
2582afde2b65SKonstantin Belousov static void
2583afde2b65SKonstantin Belousov mq_attr_from32(const struct mq_attr32 *from, struct mq_attr *to)
2584afde2b65SKonstantin Belousov {
2585afde2b65SKonstantin Belousov 
2586afde2b65SKonstantin Belousov 	to->mq_flags = from->mq_flags;
2587afde2b65SKonstantin Belousov 	to->mq_maxmsg = from->mq_maxmsg;
2588afde2b65SKonstantin Belousov 	to->mq_msgsize = from->mq_msgsize;
2589afde2b65SKonstantin Belousov 	to->mq_curmsgs = from->mq_curmsgs;
2590afde2b65SKonstantin Belousov }
2591afde2b65SKonstantin Belousov 
2592afde2b65SKonstantin Belousov static void
2593afde2b65SKonstantin Belousov mq_attr_to32(const struct mq_attr *from, struct mq_attr32 *to)
2594afde2b65SKonstantin Belousov {
2595afde2b65SKonstantin Belousov 
2596afde2b65SKonstantin Belousov 	to->mq_flags = from->mq_flags;
2597afde2b65SKonstantin Belousov 	to->mq_maxmsg = from->mq_maxmsg;
2598afde2b65SKonstantin Belousov 	to->mq_msgsize = from->mq_msgsize;
2599afde2b65SKonstantin Belousov 	to->mq_curmsgs = from->mq_curmsgs;
2600afde2b65SKonstantin Belousov }
2601afde2b65SKonstantin Belousov 
2602afde2b65SKonstantin Belousov int
2603afde2b65SKonstantin Belousov freebsd32_kmq_open(struct thread *td, struct freebsd32_kmq_open_args *uap)
2604afde2b65SKonstantin Belousov {
2605afde2b65SKonstantin Belousov 	struct mq_attr attr;
2606afde2b65SKonstantin Belousov 	struct mq_attr32 attr32;
2607afde2b65SKonstantin Belousov 	int flags, error;
2608afde2b65SKonstantin Belousov 
2609afde2b65SKonstantin Belousov 	if ((uap->flags & O_ACCMODE) == O_ACCMODE)
2610afde2b65SKonstantin Belousov 		return (EINVAL);
2611afde2b65SKonstantin Belousov 	flags = FFLAGS(uap->flags);
2612afde2b65SKonstantin Belousov 	if ((flags & O_CREAT) != 0 && uap->attr != NULL) {
2613afde2b65SKonstantin Belousov 		error = copyin(uap->attr, &attr32, sizeof(attr32));
2614afde2b65SKonstantin Belousov 		if (error)
2615afde2b65SKonstantin Belousov 			return (error);
2616afde2b65SKonstantin Belousov 		mq_attr_from32(&attr32, &attr);
2617afde2b65SKonstantin Belousov 	}
2618afde2b65SKonstantin Belousov 	return (kern_kmq_open(td, uap->path, flags, uap->mode,
2619afde2b65SKonstantin Belousov 	    uap->attr != NULL ? &attr : NULL));
2620afde2b65SKonstantin Belousov }
2621afde2b65SKonstantin Belousov 
2622afde2b65SKonstantin Belousov int
2623afde2b65SKonstantin Belousov freebsd32_kmq_setattr(struct thread *td, struct freebsd32_kmq_setattr_args *uap)
2624afde2b65SKonstantin Belousov {
2625afde2b65SKonstantin Belousov 	struct mq_attr attr, oattr;
2626afde2b65SKonstantin Belousov 	struct mq_attr32 attr32, oattr32;
2627afde2b65SKonstantin Belousov 	int error;
2628afde2b65SKonstantin Belousov 
2629afde2b65SKonstantin Belousov 	if (uap->attr != NULL) {
2630afde2b65SKonstantin Belousov 		error = copyin(uap->attr, &attr32, sizeof(attr32));
2631afde2b65SKonstantin Belousov 		if (error != 0)
2632afde2b65SKonstantin Belousov 			return (error);
2633afde2b65SKonstantin Belousov 		mq_attr_from32(&attr32, &attr);
2634afde2b65SKonstantin Belousov 	}
2635afde2b65SKonstantin Belousov 	error = kern_kmq_setattr(td, uap->mqd, uap->attr != NULL ? &attr : NULL,
2636afde2b65SKonstantin Belousov 	    &oattr);
2637afde2b65SKonstantin Belousov 	if (error != 0)
2638afde2b65SKonstantin Belousov 		return (error);
2639afde2b65SKonstantin Belousov 	if (uap->oattr != NULL) {
2640afde2b65SKonstantin Belousov 		mq_attr_to32(&oattr, &oattr32);
2641afde2b65SKonstantin Belousov 		error = copyout(&oattr32, uap->oattr, sizeof(oattr32));
2642afde2b65SKonstantin Belousov 	}
2643afde2b65SKonstantin Belousov 	return (error);
2644afde2b65SKonstantin Belousov }
2645afde2b65SKonstantin Belousov 
2646afde2b65SKonstantin Belousov int
2647afde2b65SKonstantin Belousov freebsd32_kmq_timedsend(struct thread *td,
2648afde2b65SKonstantin Belousov     struct freebsd32_kmq_timedsend_args *uap)
2649afde2b65SKonstantin Belousov {
2650afde2b65SKonstantin Belousov 	struct mqueue *mq;
2651afde2b65SKonstantin Belousov 	struct file *fp;
2652afde2b65SKonstantin Belousov 	struct timespec32 ets32;
2653afde2b65SKonstantin Belousov 	struct timespec *abs_timeout, ets;
2654afde2b65SKonstantin Belousov 	int error;
2655afde2b65SKonstantin Belousov 	int waitok;
2656afde2b65SKonstantin Belousov 
2657afde2b65SKonstantin Belousov 	error = getmq_read(td, uap->mqd, &fp, NULL, &mq);
2658afde2b65SKonstantin Belousov 	if (error)
2659afde2b65SKonstantin Belousov 		return (error);
2660afde2b65SKonstantin Belousov 	if (uap->abs_timeout != NULL) {
2661afde2b65SKonstantin Belousov 		error = copyin(uap->abs_timeout, &ets32, sizeof(ets32));
2662afde2b65SKonstantin Belousov 		if (error != 0)
2663afde2b65SKonstantin Belousov 			return (error);
2664afde2b65SKonstantin Belousov 		CP(ets32, ets, tv_sec);
2665afde2b65SKonstantin Belousov 		CP(ets32, ets, tv_nsec);
2666afde2b65SKonstantin Belousov 		abs_timeout = &ets;
2667afde2b65SKonstantin Belousov 	} else
2668afde2b65SKonstantin Belousov 		abs_timeout = NULL;
2669afde2b65SKonstantin Belousov 	waitok = !(fp->f_flag & O_NONBLOCK);
2670afde2b65SKonstantin Belousov 	error = mqueue_send(mq, uap->msg_ptr, uap->msg_len,
2671afde2b65SKonstantin Belousov 		uap->msg_prio, waitok, abs_timeout);
2672afde2b65SKonstantin Belousov 	fdrop(fp, td);
2673afde2b65SKonstantin Belousov 	return (error);
2674afde2b65SKonstantin Belousov }
2675afde2b65SKonstantin Belousov 
2676afde2b65SKonstantin Belousov int
2677afde2b65SKonstantin Belousov freebsd32_kmq_timedreceive(struct thread *td,
2678afde2b65SKonstantin Belousov     struct freebsd32_kmq_timedreceive_args *uap)
2679afde2b65SKonstantin Belousov {
2680afde2b65SKonstantin Belousov 	struct mqueue *mq;
2681afde2b65SKonstantin Belousov 	struct file *fp;
2682afde2b65SKonstantin Belousov 	struct timespec32 ets32;
2683afde2b65SKonstantin Belousov 	struct timespec *abs_timeout, ets;
2684afde2b65SKonstantin Belousov 	int error, waitok;
2685afde2b65SKonstantin Belousov 
2686afde2b65SKonstantin Belousov 	error = getmq_write(td, uap->mqd, &fp, NULL, &mq);
2687afde2b65SKonstantin Belousov 	if (error)
2688afde2b65SKonstantin Belousov 		return (error);
2689afde2b65SKonstantin Belousov 	if (uap->abs_timeout != NULL) {
2690afde2b65SKonstantin Belousov 		error = copyin(uap->abs_timeout, &ets32, sizeof(ets32));
2691afde2b65SKonstantin Belousov 		if (error != 0)
2692afde2b65SKonstantin Belousov 			return (error);
2693afde2b65SKonstantin Belousov 		CP(ets32, ets, tv_sec);
2694afde2b65SKonstantin Belousov 		CP(ets32, ets, tv_nsec);
2695afde2b65SKonstantin Belousov 		abs_timeout = &ets;
2696afde2b65SKonstantin Belousov 	} else
2697afde2b65SKonstantin Belousov 		abs_timeout = NULL;
2698afde2b65SKonstantin Belousov 	waitok = !(fp->f_flag & O_NONBLOCK);
2699afde2b65SKonstantin Belousov 	error = mqueue_receive(mq, uap->msg_ptr, uap->msg_len,
2700afde2b65SKonstantin Belousov 		uap->msg_prio, waitok, abs_timeout);
2701afde2b65SKonstantin Belousov 	fdrop(fp, td);
2702afde2b65SKonstantin Belousov 	return (error);
2703afde2b65SKonstantin Belousov }
2704afde2b65SKonstantin Belousov 
2705afde2b65SKonstantin Belousov static struct syscall_helper_data mq32_syscalls[] = {
2706afde2b65SKonstantin Belousov 	SYSCALL32_INIT_HELPER(freebsd32_kmq_open),
2707afde2b65SKonstantin Belousov 	SYSCALL32_INIT_HELPER(freebsd32_kmq_setattr),
2708afde2b65SKonstantin Belousov 	SYSCALL32_INIT_HELPER(freebsd32_kmq_timedsend),
2709afde2b65SKonstantin Belousov 	SYSCALL32_INIT_HELPER(freebsd32_kmq_timedreceive),
2710afde2b65SKonstantin Belousov 	SYSCALL32_INIT_HELPER(kmq_notify),
2711afde2b65SKonstantin Belousov 	SYSCALL32_INIT_HELPER(kmq_unlink),
2712afde2b65SKonstantin Belousov 	SYSCALL_INIT_LAST
2713afde2b65SKonstantin Belousov };
2714afde2b65SKonstantin Belousov #endif
2715afde2b65SKonstantin Belousov 
2716afde2b65SKonstantin Belousov static int
2717afde2b65SKonstantin Belousov mqinit(void)
2718afde2b65SKonstantin Belousov {
2719afde2b65SKonstantin Belousov 	int error;
2720afde2b65SKonstantin Belousov 
2721afde2b65SKonstantin Belousov 	error = syscall_helper_register(mq_syscalls);
2722afde2b65SKonstantin Belousov 	if (error != 0)
2723afde2b65SKonstantin Belousov 		return (error);
2724afde2b65SKonstantin Belousov #ifdef COMPAT_FREEBSD32
2725afde2b65SKonstantin Belousov 	error = syscall32_helper_register(mq32_syscalls);
2726afde2b65SKonstantin Belousov 	if (error != 0)
2727afde2b65SKonstantin Belousov 		return (error);
2728afde2b65SKonstantin Belousov #endif
2729afde2b65SKonstantin Belousov 	return (0);
2730afde2b65SKonstantin Belousov }
2731afde2b65SKonstantin Belousov 
2732afde2b65SKonstantin Belousov static int
2733afde2b65SKonstantin Belousov mqunload(void)
2734afde2b65SKonstantin Belousov {
2735afde2b65SKonstantin Belousov 
2736afde2b65SKonstantin Belousov #ifdef COMPAT_FREEBSD32
2737afde2b65SKonstantin Belousov 	syscall32_helper_unregister(mq32_syscalls);
2738afde2b65SKonstantin Belousov #endif
2739afde2b65SKonstantin Belousov 	syscall_helper_unregister(mq_syscalls);
2740afde2b65SKonstantin Belousov 	return (0);
2741afde2b65SKonstantin Belousov }
2742afde2b65SKonstantin Belousov 
2743afde2b65SKonstantin Belousov static int
2744afde2b65SKonstantin Belousov mq_modload(struct module *module, int cmd, void *arg)
2745afde2b65SKonstantin Belousov {
2746afde2b65SKonstantin Belousov 	int error = 0;
2747afde2b65SKonstantin Belousov 
2748afde2b65SKonstantin Belousov 	error = vfs_modevent(module, cmd, arg);
2749afde2b65SKonstantin Belousov 	if (error != 0)
2750afde2b65SKonstantin Belousov 		return (error);
2751afde2b65SKonstantin Belousov 
2752afde2b65SKonstantin Belousov 	switch (cmd) {
2753afde2b65SKonstantin Belousov 	case MOD_LOAD:
2754afde2b65SKonstantin Belousov 		error = mqinit();
2755afde2b65SKonstantin Belousov 		if (error != 0)
2756afde2b65SKonstantin Belousov 			mqunload();
2757afde2b65SKonstantin Belousov 		break;
2758afde2b65SKonstantin Belousov 	case MOD_UNLOAD:
2759afde2b65SKonstantin Belousov 		error = mqunload();
2760afde2b65SKonstantin Belousov 		break;
2761afde2b65SKonstantin Belousov 	default:
2762afde2b65SKonstantin Belousov 		break;
2763afde2b65SKonstantin Belousov 	}
2764afde2b65SKonstantin Belousov 	return (error);
2765afde2b65SKonstantin Belousov }
2766afde2b65SKonstantin Belousov 
2767afde2b65SKonstantin Belousov static moduledata_t mqueuefs_mod = {
2768afde2b65SKonstantin Belousov 	"mqueuefs",
2769afde2b65SKonstantin Belousov 	mq_modload,
2770afde2b65SKonstantin Belousov 	&mqueuefs_vfsconf
2771afde2b65SKonstantin Belousov };
2772afde2b65SKonstantin Belousov DECLARE_MODULE(mqueuefs, mqueuefs_mod, SI_SUB_VFS, SI_ORDER_MIDDLE);
277347bf2cf9SDavid Xu MODULE_VERSION(mqueuefs, 1);
2774