xref: /kernel/linux/linux-5.10/drivers/md/dm-ioctl.c (revision 8c2ecf20)
1/*
2 * Copyright (C) 2001, 2002 Sistina Software (UK) Limited.
3 * Copyright (C) 2004 - 2006 Red Hat, Inc. All rights reserved.
4 *
5 * This file is released under the GPL.
6 */
7
8#include "dm-core.h"
9
10#include <linux/module.h>
11#include <linux/vmalloc.h>
12#include <linux/miscdevice.h>
13#include <linux/sched/mm.h>
14#include <linux/init.h>
15#include <linux/wait.h>
16#include <linux/slab.h>
17#include <linux/dm-ioctl.h>
18#include <linux/hdreg.h>
19#include <linux/compat.h>
20#include <linux/nospec.h>
21
22#include <linux/uaccess.h>
23
24#define DM_MSG_PREFIX "ioctl"
25#define DM_DRIVER_EMAIL "dm-devel@redhat.com"
26
27struct dm_file {
28	/*
29	 * poll will wait until the global event number is greater than
30	 * this value.
31	 */
32	volatile unsigned global_event_nr;
33};
34
35/*-----------------------------------------------------------------
36 * The ioctl interface needs to be able to look up devices by
37 * name or uuid.
38 *---------------------------------------------------------------*/
39struct hash_cell {
40	struct list_head name_list;
41	struct list_head uuid_list;
42
43	char *name;
44	char *uuid;
45	struct mapped_device *md;
46	struct dm_table *new_map;
47};
48
49struct vers_iter {
50    size_t param_size;
51    struct dm_target_versions *vers, *old_vers;
52    char *end;
53    uint32_t flags;
54};
55
56
57#define NUM_BUCKETS 64
58#define MASK_BUCKETS (NUM_BUCKETS - 1)
59static struct list_head _name_buckets[NUM_BUCKETS];
60static struct list_head _uuid_buckets[NUM_BUCKETS];
61
62static void dm_hash_remove_all(bool keep_open_devices, bool mark_deferred, bool only_deferred);
63
64/*
65 * Guards access to both hash tables.
66 */
67static DECLARE_RWSEM(_hash_lock);
68
69/*
70 * Protects use of mdptr to obtain hash cell name and uuid from mapped device.
71 */
72static DEFINE_MUTEX(dm_hash_cells_mutex);
73
74static void init_buckets(struct list_head *buckets)
75{
76	unsigned int i;
77
78	for (i = 0; i < NUM_BUCKETS; i++)
79		INIT_LIST_HEAD(buckets + i);
80}
81
82static int dm_hash_init(void)
83{
84	init_buckets(_name_buckets);
85	init_buckets(_uuid_buckets);
86	return 0;
87}
88
89static void dm_hash_exit(void)
90{
91	dm_hash_remove_all(false, false, false);
92}
93
94/*-----------------------------------------------------------------
95 * Hash function:
96 * We're not really concerned with the str hash function being
97 * fast since it's only used by the ioctl interface.
98 *---------------------------------------------------------------*/
99static unsigned int hash_str(const char *str)
100{
101	const unsigned int hash_mult = 2654435387U;
102	unsigned int h = 0;
103
104	while (*str)
105		h = (h + (unsigned int) *str++) * hash_mult;
106
107	return h & MASK_BUCKETS;
108}
109
110/*-----------------------------------------------------------------
111 * Code for looking up a device by name
112 *---------------------------------------------------------------*/
113static struct hash_cell *__get_name_cell(const char *str)
114{
115	struct hash_cell *hc;
116	unsigned int h = hash_str(str);
117
118	list_for_each_entry (hc, _name_buckets + h, name_list)
119		if (!strcmp(hc->name, str)) {
120			dm_get(hc->md);
121			return hc;
122		}
123
124	return NULL;
125}
126
127static struct hash_cell *__get_uuid_cell(const char *str)
128{
129	struct hash_cell *hc;
130	unsigned int h = hash_str(str);
131
132	list_for_each_entry (hc, _uuid_buckets + h, uuid_list)
133		if (!strcmp(hc->uuid, str)) {
134			dm_get(hc->md);
135			return hc;
136		}
137
138	return NULL;
139}
140
141static struct hash_cell *__get_dev_cell(uint64_t dev)
142{
143	struct mapped_device *md;
144	struct hash_cell *hc;
145
146	md = dm_get_md(huge_decode_dev(dev));
147	if (!md)
148		return NULL;
149
150	hc = dm_get_mdptr(md);
151	if (!hc) {
152		dm_put(md);
153		return NULL;
154	}
155
156	return hc;
157}
158
159/*-----------------------------------------------------------------
160 * Inserting, removing and renaming a device.
161 *---------------------------------------------------------------*/
162static struct hash_cell *alloc_cell(const char *name, const char *uuid,
163				    struct mapped_device *md)
164{
165	struct hash_cell *hc;
166
167	hc = kmalloc(sizeof(*hc), GFP_KERNEL);
168	if (!hc)
169		return NULL;
170
171	hc->name = kstrdup(name, GFP_KERNEL);
172	if (!hc->name) {
173		kfree(hc);
174		return NULL;
175	}
176
177	if (!uuid)
178		hc->uuid = NULL;
179
180	else {
181		hc->uuid = kstrdup(uuid, GFP_KERNEL);
182		if (!hc->uuid) {
183			kfree(hc->name);
184			kfree(hc);
185			return NULL;
186		}
187	}
188
189	INIT_LIST_HEAD(&hc->name_list);
190	INIT_LIST_HEAD(&hc->uuid_list);
191	hc->md = md;
192	hc->new_map = NULL;
193	return hc;
194}
195
196static void free_cell(struct hash_cell *hc)
197{
198	if (hc) {
199		kfree(hc->name);
200		kfree(hc->uuid);
201		kfree(hc);
202	}
203}
204
205/*
206 * The kdev_t and uuid of a device can never change once it is
207 * initially inserted.
208 */
209static int dm_hash_insert(const char *name, const char *uuid, struct mapped_device *md)
210{
211	struct hash_cell *cell, *hc;
212
213	/*
214	 * Allocate the new cells.
215	 */
216	cell = alloc_cell(name, uuid, md);
217	if (!cell)
218		return -ENOMEM;
219
220	/*
221	 * Insert the cell into both hash tables.
222	 */
223	down_write(&_hash_lock);
224	hc = __get_name_cell(name);
225	if (hc) {
226		dm_put(hc->md);
227		goto bad;
228	}
229
230	list_add(&cell->name_list, _name_buckets + hash_str(name));
231
232	if (uuid) {
233		hc = __get_uuid_cell(uuid);
234		if (hc) {
235			list_del(&cell->name_list);
236			dm_put(hc->md);
237			goto bad;
238		}
239		list_add(&cell->uuid_list, _uuid_buckets + hash_str(uuid));
240	}
241	dm_get(md);
242	mutex_lock(&dm_hash_cells_mutex);
243	dm_set_mdptr(md, cell);
244	mutex_unlock(&dm_hash_cells_mutex);
245	up_write(&_hash_lock);
246
247	return 0;
248
249 bad:
250	up_write(&_hash_lock);
251	free_cell(cell);
252	return -EBUSY;
253}
254
255static struct dm_table *__hash_remove(struct hash_cell *hc)
256{
257	struct dm_table *table;
258	int srcu_idx;
259
260	/* remove from the dev hash */
261	list_del(&hc->uuid_list);
262	list_del(&hc->name_list);
263	mutex_lock(&dm_hash_cells_mutex);
264	dm_set_mdptr(hc->md, NULL);
265	mutex_unlock(&dm_hash_cells_mutex);
266
267	table = dm_get_live_table(hc->md, &srcu_idx);
268	if (table)
269		dm_table_event(table);
270	dm_put_live_table(hc->md, srcu_idx);
271
272	table = NULL;
273	if (hc->new_map)
274		table = hc->new_map;
275	dm_put(hc->md);
276	free_cell(hc);
277
278	return table;
279}
280
281static void dm_hash_remove_all(bool keep_open_devices, bool mark_deferred, bool only_deferred)
282{
283	int i, dev_skipped;
284	struct hash_cell *hc;
285	struct mapped_device *md;
286	struct dm_table *t;
287
288retry:
289	dev_skipped = 0;
290
291	down_write(&_hash_lock);
292
293	for (i = 0; i < NUM_BUCKETS; i++) {
294		list_for_each_entry(hc, _name_buckets + i, name_list) {
295			md = hc->md;
296			dm_get(md);
297
298			if (keep_open_devices &&
299			    dm_lock_for_deletion(md, mark_deferred, only_deferred)) {
300				dm_put(md);
301				dev_skipped++;
302				continue;
303			}
304
305			t = __hash_remove(hc);
306
307			up_write(&_hash_lock);
308
309			if (t) {
310				dm_sync_table(md);
311				dm_table_destroy(t);
312			}
313			dm_put(md);
314			if (likely(keep_open_devices))
315				dm_destroy(md);
316			else
317				dm_destroy_immediate(md);
318
319			/*
320			 * Some mapped devices may be using other mapped
321			 * devices, so repeat until we make no further
322			 * progress.  If a new mapped device is created
323			 * here it will also get removed.
324			 */
325			goto retry;
326		}
327	}
328
329	up_write(&_hash_lock);
330
331	if (dev_skipped)
332		DMWARN("remove_all left %d open device(s)", dev_skipped);
333}
334
335/*
336 * Set the uuid of a hash_cell that isn't already set.
337 */
338static void __set_cell_uuid(struct hash_cell *hc, char *new_uuid)
339{
340	mutex_lock(&dm_hash_cells_mutex);
341	hc->uuid = new_uuid;
342	mutex_unlock(&dm_hash_cells_mutex);
343
344	list_add(&hc->uuid_list, _uuid_buckets + hash_str(new_uuid));
345}
346
347/*
348 * Changes the name of a hash_cell and returns the old name for
349 * the caller to free.
350 */
351static char *__change_cell_name(struct hash_cell *hc, char *new_name)
352{
353	char *old_name;
354
355	/*
356	 * Rename and move the name cell.
357	 */
358	list_del(&hc->name_list);
359	old_name = hc->name;
360
361	mutex_lock(&dm_hash_cells_mutex);
362	hc->name = new_name;
363	mutex_unlock(&dm_hash_cells_mutex);
364
365	list_add(&hc->name_list, _name_buckets + hash_str(new_name));
366
367	return old_name;
368}
369
370static struct mapped_device *dm_hash_rename(struct dm_ioctl *param,
371					    const char *new)
372{
373	char *new_data, *old_name = NULL;
374	struct hash_cell *hc;
375	struct dm_table *table;
376	struct mapped_device *md;
377	unsigned change_uuid = (param->flags & DM_UUID_FLAG) ? 1 : 0;
378	int srcu_idx;
379
380	/*
381	 * duplicate new.
382	 */
383	new_data = kstrdup(new, GFP_KERNEL);
384	if (!new_data)
385		return ERR_PTR(-ENOMEM);
386
387	down_write(&_hash_lock);
388
389	/*
390	 * Is new free ?
391	 */
392	if (change_uuid)
393		hc = __get_uuid_cell(new);
394	else
395		hc = __get_name_cell(new);
396
397	if (hc) {
398		DMWARN("Unable to change %s on mapped device %s to one that "
399		       "already exists: %s",
400		       change_uuid ? "uuid" : "name",
401		       param->name, new);
402		dm_put(hc->md);
403		up_write(&_hash_lock);
404		kfree(new_data);
405		return ERR_PTR(-EBUSY);
406	}
407
408	/*
409	 * Is there such a device as 'old' ?
410	 */
411	hc = __get_name_cell(param->name);
412	if (!hc) {
413		DMWARN("Unable to rename non-existent device, %s to %s%s",
414		       param->name, change_uuid ? "uuid " : "", new);
415		up_write(&_hash_lock);
416		kfree(new_data);
417		return ERR_PTR(-ENXIO);
418	}
419
420	/*
421	 * Does this device already have a uuid?
422	 */
423	if (change_uuid && hc->uuid) {
424		DMWARN("Unable to change uuid of mapped device %s to %s "
425		       "because uuid is already set to %s",
426		       param->name, new, hc->uuid);
427		dm_put(hc->md);
428		up_write(&_hash_lock);
429		kfree(new_data);
430		return ERR_PTR(-EINVAL);
431	}
432
433	if (change_uuid)
434		__set_cell_uuid(hc, new_data);
435	else
436		old_name = __change_cell_name(hc, new_data);
437
438	/*
439	 * Wake up any dm event waiters.
440	 */
441	table = dm_get_live_table(hc->md, &srcu_idx);
442	if (table)
443		dm_table_event(table);
444	dm_put_live_table(hc->md, srcu_idx);
445
446	if (!dm_kobject_uevent(hc->md, KOBJ_CHANGE, param->event_nr))
447		param->flags |= DM_UEVENT_GENERATED_FLAG;
448
449	md = hc->md;
450	up_write(&_hash_lock);
451	kfree(old_name);
452
453	return md;
454}
455
456void dm_deferred_remove(void)
457{
458	dm_hash_remove_all(true, false, true);
459}
460
461/*-----------------------------------------------------------------
462 * Implementation of the ioctl commands
463 *---------------------------------------------------------------*/
464/*
465 * All the ioctl commands get dispatched to functions with this
466 * prototype.
467 */
468typedef int (*ioctl_fn)(struct file *filp, struct dm_ioctl *param, size_t param_size);
469
470static int remove_all(struct file *filp, struct dm_ioctl *param, size_t param_size)
471{
472	dm_hash_remove_all(true, !!(param->flags & DM_DEFERRED_REMOVE), false);
473	param->data_size = 0;
474	return 0;
475}
476
477/*
478 * Round up the ptr to an 8-byte boundary.
479 */
480#define ALIGN_MASK 7
481static inline size_t align_val(size_t val)
482{
483	return (val + ALIGN_MASK) & ~ALIGN_MASK;
484}
485static inline void *align_ptr(void *ptr)
486{
487	return (void *)align_val((size_t)ptr);
488}
489
490/*
491 * Retrieves the data payload buffer from an already allocated
492 * struct dm_ioctl.
493 */
494static void *get_result_buffer(struct dm_ioctl *param, size_t param_size,
495			       size_t *len)
496{
497	param->data_start = align_ptr(param + 1) - (void *) param;
498
499	if (param->data_start < param_size)
500		*len = param_size - param->data_start;
501	else
502		*len = 0;
503
504	return ((void *) param) + param->data_start;
505}
506
507static int list_devices(struct file *filp, struct dm_ioctl *param, size_t param_size)
508{
509	unsigned int i;
510	struct hash_cell *hc;
511	size_t len, needed = 0;
512	struct gendisk *disk;
513	struct dm_name_list *orig_nl, *nl, *old_nl = NULL;
514	uint32_t *event_nr;
515
516	down_write(&_hash_lock);
517
518	/*
519	 * Loop through all the devices working out how much
520	 * space we need.
521	 */
522	for (i = 0; i < NUM_BUCKETS; i++) {
523		list_for_each_entry (hc, _name_buckets + i, name_list) {
524			needed += align_val(offsetof(struct dm_name_list, name) + strlen(hc->name) + 1);
525			needed += align_val(sizeof(uint32_t));
526		}
527	}
528
529	/*
530	 * Grab our output buffer.
531	 */
532	nl = orig_nl = get_result_buffer(param, param_size, &len);
533	if (len < needed || len < sizeof(nl->dev)) {
534		param->flags |= DM_BUFFER_FULL_FLAG;
535		goto out;
536	}
537	param->data_size = param->data_start + needed;
538
539	nl->dev = 0;	/* Flags no data */
540
541	/*
542	 * Now loop through filling out the names.
543	 */
544	for (i = 0; i < NUM_BUCKETS; i++) {
545		list_for_each_entry (hc, _name_buckets + i, name_list) {
546			if (old_nl)
547				old_nl->next = (uint32_t) ((void *) nl -
548							   (void *) old_nl);
549			disk = dm_disk(hc->md);
550			nl->dev = huge_encode_dev(disk_devt(disk));
551			nl->next = 0;
552			strcpy(nl->name, hc->name);
553
554			old_nl = nl;
555			event_nr = align_ptr(nl->name + strlen(hc->name) + 1);
556			*event_nr = dm_get_event_nr(hc->md);
557			nl = align_ptr(event_nr + 1);
558		}
559	}
560	/*
561	 * If mismatch happens, security may be compromised due to buffer
562	 * overflow, so it's better to crash.
563	 */
564	BUG_ON((char *)nl - (char *)orig_nl != needed);
565
566 out:
567	up_write(&_hash_lock);
568	return 0;
569}
570
571static void list_version_get_needed(struct target_type *tt, void *needed_param)
572{
573    size_t *needed = needed_param;
574
575    *needed += sizeof(struct dm_target_versions);
576    *needed += strlen(tt->name) + 1;
577    *needed += ALIGN_MASK;
578}
579
580static void list_version_get_info(struct target_type *tt, void *param)
581{
582    struct vers_iter *info = param;
583
584    /* Check space - it might have changed since the first iteration */
585    if ((char *)info->vers + sizeof(tt->version) + strlen(tt->name) + 1 >
586	info->end) {
587
588	info->flags = DM_BUFFER_FULL_FLAG;
589	return;
590    }
591
592    if (info->old_vers)
593	info->old_vers->next = (uint32_t) ((void *)info->vers -
594					   (void *)info->old_vers);
595    info->vers->version[0] = tt->version[0];
596    info->vers->version[1] = tt->version[1];
597    info->vers->version[2] = tt->version[2];
598    info->vers->next = 0;
599    strcpy(info->vers->name, tt->name);
600
601    info->old_vers = info->vers;
602    info->vers = align_ptr(((void *) ++info->vers) + strlen(tt->name) + 1);
603}
604
605static int __list_versions(struct dm_ioctl *param, size_t param_size, const char *name)
606{
607	size_t len, needed = 0;
608	struct dm_target_versions *vers;
609	struct vers_iter iter_info;
610	struct target_type *tt = NULL;
611
612	if (name) {
613		tt = dm_get_target_type(name);
614		if (!tt)
615			return -EINVAL;
616	}
617
618	/*
619	 * Loop through all the devices working out how much
620	 * space we need.
621	 */
622	if (!tt)
623		dm_target_iterate(list_version_get_needed, &needed);
624	else
625		list_version_get_needed(tt, &needed);
626
627	/*
628	 * Grab our output buffer.
629	 */
630	vers = get_result_buffer(param, param_size, &len);
631	if (len < needed) {
632		param->flags |= DM_BUFFER_FULL_FLAG;
633		goto out;
634	}
635	param->data_size = param->data_start + needed;
636
637	iter_info.param_size = param_size;
638	iter_info.old_vers = NULL;
639	iter_info.vers = vers;
640	iter_info.flags = 0;
641	iter_info.end = (char *)vers + needed;
642
643	/*
644	 * Now loop through filling out the names & versions.
645	 */
646	if (!tt)
647		dm_target_iterate(list_version_get_info, &iter_info);
648	else
649		list_version_get_info(tt, &iter_info);
650	param->flags |= iter_info.flags;
651
652 out:
653	if (tt)
654		dm_put_target_type(tt);
655	return 0;
656}
657
658static int list_versions(struct file *filp, struct dm_ioctl *param, size_t param_size)
659{
660	return __list_versions(param, param_size, NULL);
661}
662
663static int get_target_version(struct file *filp, struct dm_ioctl *param, size_t param_size)
664{
665	return __list_versions(param, param_size, param->name);
666}
667
668static int check_name(const char *name)
669{
670	if (strchr(name, '/')) {
671		DMWARN("invalid device name");
672		return -EINVAL;
673	}
674
675	return 0;
676}
677
678/*
679 * On successful return, the caller must not attempt to acquire
680 * _hash_lock without first calling dm_put_live_table, because dm_table_destroy
681 * waits for this dm_put_live_table and could be called under this lock.
682 */
683static struct dm_table *dm_get_inactive_table(struct mapped_device *md, int *srcu_idx)
684{
685	struct hash_cell *hc;
686	struct dm_table *table = NULL;
687
688	/* increment rcu count, we don't care about the table pointer */
689	dm_get_live_table(md, srcu_idx);
690
691	down_read(&_hash_lock);
692	hc = dm_get_mdptr(md);
693	if (!hc || hc->md != md) {
694		DMWARN("device has been removed from the dev hash table.");
695		goto out;
696	}
697
698	table = hc->new_map;
699
700out:
701	up_read(&_hash_lock);
702
703	return table;
704}
705
706static struct dm_table *dm_get_live_or_inactive_table(struct mapped_device *md,
707						      struct dm_ioctl *param,
708						      int *srcu_idx)
709{
710	return (param->flags & DM_QUERY_INACTIVE_TABLE_FLAG) ?
711		dm_get_inactive_table(md, srcu_idx) : dm_get_live_table(md, srcu_idx);
712}
713
714/*
715 * Fills in a dm_ioctl structure, ready for sending back to
716 * userland.
717 */
718static void __dev_status(struct mapped_device *md, struct dm_ioctl *param)
719{
720	struct gendisk *disk = dm_disk(md);
721	struct dm_table *table;
722	int srcu_idx;
723
724	param->flags &= ~(DM_SUSPEND_FLAG | DM_READONLY_FLAG |
725			  DM_ACTIVE_PRESENT_FLAG | DM_INTERNAL_SUSPEND_FLAG);
726
727	if (dm_suspended_md(md))
728		param->flags |= DM_SUSPEND_FLAG;
729
730	if (dm_suspended_internally_md(md))
731		param->flags |= DM_INTERNAL_SUSPEND_FLAG;
732
733	if (dm_test_deferred_remove_flag(md))
734		param->flags |= DM_DEFERRED_REMOVE;
735
736	param->dev = huge_encode_dev(disk_devt(disk));
737
738	/*
739	 * Yes, this will be out of date by the time it gets back
740	 * to userland, but it is still very useful for
741	 * debugging.
742	 */
743	param->open_count = dm_open_count(md);
744
745	param->event_nr = dm_get_event_nr(md);
746	param->target_count = 0;
747
748	table = dm_get_live_table(md, &srcu_idx);
749	if (table) {
750		if (!(param->flags & DM_QUERY_INACTIVE_TABLE_FLAG)) {
751			if (get_disk_ro(disk))
752				param->flags |= DM_READONLY_FLAG;
753			param->target_count = dm_table_get_num_targets(table);
754		}
755
756		param->flags |= DM_ACTIVE_PRESENT_FLAG;
757	}
758	dm_put_live_table(md, srcu_idx);
759
760	if (param->flags & DM_QUERY_INACTIVE_TABLE_FLAG) {
761		int srcu_idx;
762		table = dm_get_inactive_table(md, &srcu_idx);
763		if (table) {
764			if (!(dm_table_get_mode(table) & FMODE_WRITE))
765				param->flags |= DM_READONLY_FLAG;
766			param->target_count = dm_table_get_num_targets(table);
767		}
768		dm_put_live_table(md, srcu_idx);
769	}
770}
771
772static int dev_create(struct file *filp, struct dm_ioctl *param, size_t param_size)
773{
774	int r, m = DM_ANY_MINOR;
775	struct mapped_device *md;
776
777	r = check_name(param->name);
778	if (r)
779		return r;
780
781	if (param->flags & DM_PERSISTENT_DEV_FLAG)
782		m = MINOR(huge_decode_dev(param->dev));
783
784	r = dm_create(m, &md);
785	if (r)
786		return r;
787
788	r = dm_hash_insert(param->name, *param->uuid ? param->uuid : NULL, md);
789	if (r) {
790		dm_put(md);
791		dm_destroy(md);
792		return r;
793	}
794
795	param->flags &= ~DM_INACTIVE_PRESENT_FLAG;
796
797	__dev_status(md, param);
798
799	dm_put(md);
800
801	return 0;
802}
803
804/*
805 * Always use UUID for lookups if it's present, otherwise use name or dev.
806 */
807static struct hash_cell *__find_device_hash_cell(struct dm_ioctl *param)
808{
809	struct hash_cell *hc = NULL;
810
811	if (*param->uuid) {
812		if (*param->name || param->dev)
813			return NULL;
814
815		hc = __get_uuid_cell(param->uuid);
816		if (!hc)
817			return NULL;
818	} else if (*param->name) {
819		if (param->dev)
820			return NULL;
821
822		hc = __get_name_cell(param->name);
823		if (!hc)
824			return NULL;
825	} else if (param->dev) {
826		hc = __get_dev_cell(param->dev);
827		if (!hc)
828			return NULL;
829	} else
830		return NULL;
831
832	/*
833	 * Sneakily write in both the name and the uuid
834	 * while we have the cell.
835	 */
836	strlcpy(param->name, hc->name, sizeof(param->name));
837	if (hc->uuid)
838		strlcpy(param->uuid, hc->uuid, sizeof(param->uuid));
839	else
840		param->uuid[0] = '\0';
841
842	if (hc->new_map)
843		param->flags |= DM_INACTIVE_PRESENT_FLAG;
844	else
845		param->flags &= ~DM_INACTIVE_PRESENT_FLAG;
846
847	return hc;
848}
849
850static struct mapped_device *find_device(struct dm_ioctl *param)
851{
852	struct hash_cell *hc;
853	struct mapped_device *md = NULL;
854
855	down_read(&_hash_lock);
856	hc = __find_device_hash_cell(param);
857	if (hc)
858		md = hc->md;
859	up_read(&_hash_lock);
860
861	return md;
862}
863
864static int dev_remove(struct file *filp, struct dm_ioctl *param, size_t param_size)
865{
866	struct hash_cell *hc;
867	struct mapped_device *md;
868	int r;
869	struct dm_table *t;
870
871	down_write(&_hash_lock);
872	hc = __find_device_hash_cell(param);
873
874	if (!hc) {
875		DMDEBUG_LIMIT("device doesn't appear to be in the dev hash table.");
876		up_write(&_hash_lock);
877		return -ENXIO;
878	}
879
880	md = hc->md;
881
882	/*
883	 * Ensure the device is not open and nothing further can open it.
884	 */
885	r = dm_lock_for_deletion(md, !!(param->flags & DM_DEFERRED_REMOVE), false);
886	if (r) {
887		if (r == -EBUSY && param->flags & DM_DEFERRED_REMOVE) {
888			up_write(&_hash_lock);
889			dm_put(md);
890			return 0;
891		}
892		DMDEBUG_LIMIT("unable to remove open device %s", hc->name);
893		up_write(&_hash_lock);
894		dm_put(md);
895		return r;
896	}
897
898	t = __hash_remove(hc);
899	up_write(&_hash_lock);
900
901	if (t) {
902		dm_sync_table(md);
903		dm_table_destroy(t);
904	}
905
906	param->flags &= ~DM_DEFERRED_REMOVE;
907
908	if (!dm_kobject_uevent(md, KOBJ_REMOVE, param->event_nr))
909		param->flags |= DM_UEVENT_GENERATED_FLAG;
910
911	dm_put(md);
912	dm_destroy(md);
913	return 0;
914}
915
916/*
917 * Check a string doesn't overrun the chunk of
918 * memory we copied from userland.
919 */
920static int invalid_str(char *str, void *end)
921{
922	while ((void *) str < end)
923		if (!*str++)
924			return 0;
925
926	return -EINVAL;
927}
928
929static int dev_rename(struct file *filp, struct dm_ioctl *param, size_t param_size)
930{
931	int r;
932	char *new_data = (char *) param + param->data_start;
933	struct mapped_device *md;
934	unsigned change_uuid = (param->flags & DM_UUID_FLAG) ? 1 : 0;
935
936	if (new_data < param->data ||
937	    invalid_str(new_data, (void *) param + param_size) || !*new_data ||
938	    strlen(new_data) > (change_uuid ? DM_UUID_LEN - 1 : DM_NAME_LEN - 1)) {
939		DMWARN("Invalid new mapped device name or uuid string supplied.");
940		return -EINVAL;
941	}
942
943	if (!change_uuid) {
944		r = check_name(new_data);
945		if (r)
946			return r;
947	}
948
949	md = dm_hash_rename(param, new_data);
950	if (IS_ERR(md))
951		return PTR_ERR(md);
952
953	__dev_status(md, param);
954	dm_put(md);
955
956	return 0;
957}
958
959static int dev_set_geometry(struct file *filp, struct dm_ioctl *param, size_t param_size)
960{
961	int r = -EINVAL, x;
962	struct mapped_device *md;
963	struct hd_geometry geometry;
964	unsigned long indata[4];
965	char *geostr = (char *) param + param->data_start;
966	char dummy;
967
968	md = find_device(param);
969	if (!md)
970		return -ENXIO;
971
972	if (geostr < param->data ||
973	    invalid_str(geostr, (void *) param + param_size)) {
974		DMWARN("Invalid geometry supplied.");
975		goto out;
976	}
977
978	x = sscanf(geostr, "%lu %lu %lu %lu%c", indata,
979		   indata + 1, indata + 2, indata + 3, &dummy);
980
981	if (x != 4) {
982		DMWARN("Unable to interpret geometry settings.");
983		goto out;
984	}
985
986	if (indata[0] > 65535 || indata[1] > 255 ||
987	    indata[2] > 255 || indata[3] > ULONG_MAX) {
988		DMWARN("Geometry exceeds range limits.");
989		goto out;
990	}
991
992	geometry.cylinders = indata[0];
993	geometry.heads = indata[1];
994	geometry.sectors = indata[2];
995	geometry.start = indata[3];
996
997	r = dm_set_geometry(md, &geometry);
998
999	param->data_size = 0;
1000
1001out:
1002	dm_put(md);
1003	return r;
1004}
1005
1006static int do_suspend(struct dm_ioctl *param)
1007{
1008	int r = 0;
1009	unsigned suspend_flags = DM_SUSPEND_LOCKFS_FLAG;
1010	struct mapped_device *md;
1011
1012	md = find_device(param);
1013	if (!md)
1014		return -ENXIO;
1015
1016	if (param->flags & DM_SKIP_LOCKFS_FLAG)
1017		suspend_flags &= ~DM_SUSPEND_LOCKFS_FLAG;
1018	if (param->flags & DM_NOFLUSH_FLAG)
1019		suspend_flags |= DM_SUSPEND_NOFLUSH_FLAG;
1020
1021	if (!dm_suspended_md(md)) {
1022		r = dm_suspend(md, suspend_flags);
1023		if (r)
1024			goto out;
1025	}
1026
1027	__dev_status(md, param);
1028
1029out:
1030	dm_put(md);
1031
1032	return r;
1033}
1034
1035static int do_resume(struct dm_ioctl *param)
1036{
1037	int r = 0;
1038	unsigned suspend_flags = DM_SUSPEND_LOCKFS_FLAG;
1039	struct hash_cell *hc;
1040	struct mapped_device *md;
1041	struct dm_table *new_map, *old_map = NULL;
1042
1043	down_write(&_hash_lock);
1044
1045	hc = __find_device_hash_cell(param);
1046	if (!hc) {
1047		DMDEBUG_LIMIT("device doesn't appear to be in the dev hash table.");
1048		up_write(&_hash_lock);
1049		return -ENXIO;
1050	}
1051
1052	md = hc->md;
1053
1054	new_map = hc->new_map;
1055	hc->new_map = NULL;
1056	param->flags &= ~DM_INACTIVE_PRESENT_FLAG;
1057
1058	up_write(&_hash_lock);
1059
1060	/* Do we need to load a new map ? */
1061	if (new_map) {
1062		/* Suspend if it isn't already suspended */
1063		if (param->flags & DM_SKIP_LOCKFS_FLAG)
1064			suspend_flags &= ~DM_SUSPEND_LOCKFS_FLAG;
1065		if (param->flags & DM_NOFLUSH_FLAG)
1066			suspend_flags |= DM_SUSPEND_NOFLUSH_FLAG;
1067		if (!dm_suspended_md(md))
1068			dm_suspend(md, suspend_flags);
1069
1070		old_map = dm_swap_table(md, new_map);
1071		if (IS_ERR(old_map)) {
1072			dm_sync_table(md);
1073			dm_table_destroy(new_map);
1074			dm_put(md);
1075			return PTR_ERR(old_map);
1076		}
1077
1078		if (dm_table_get_mode(new_map) & FMODE_WRITE)
1079			set_disk_ro(dm_disk(md), 0);
1080		else
1081			set_disk_ro(dm_disk(md), 1);
1082	}
1083
1084	if (dm_suspended_md(md)) {
1085		r = dm_resume(md);
1086		if (!r && !dm_kobject_uevent(md, KOBJ_CHANGE, param->event_nr))
1087			param->flags |= DM_UEVENT_GENERATED_FLAG;
1088	}
1089
1090	/*
1091	 * Since dm_swap_table synchronizes RCU, nobody should be in
1092	 * read-side critical section already.
1093	 */
1094	if (old_map)
1095		dm_table_destroy(old_map);
1096
1097	if (!r)
1098		__dev_status(md, param);
1099
1100	dm_put(md);
1101	return r;
1102}
1103
1104/*
1105 * Set or unset the suspension state of a device.
1106 * If the device already is in the requested state we just return its status.
1107 */
1108static int dev_suspend(struct file *filp, struct dm_ioctl *param, size_t param_size)
1109{
1110	if (param->flags & DM_SUSPEND_FLAG)
1111		return do_suspend(param);
1112
1113	return do_resume(param);
1114}
1115
1116/*
1117 * Copies device info back to user space, used by
1118 * the create and info ioctls.
1119 */
1120static int dev_status(struct file *filp, struct dm_ioctl *param, size_t param_size)
1121{
1122	struct mapped_device *md;
1123
1124	md = find_device(param);
1125	if (!md)
1126		return -ENXIO;
1127
1128	__dev_status(md, param);
1129	dm_put(md);
1130
1131	return 0;
1132}
1133
1134/*
1135 * Build up the status struct for each target
1136 */
1137static void retrieve_status(struct dm_table *table,
1138			    struct dm_ioctl *param, size_t param_size)
1139{
1140	unsigned int i, num_targets;
1141	struct dm_target_spec *spec;
1142	char *outbuf, *outptr;
1143	status_type_t type;
1144	size_t remaining, len, used = 0;
1145	unsigned status_flags = 0;
1146
1147	outptr = outbuf = get_result_buffer(param, param_size, &len);
1148
1149	if (param->flags & DM_STATUS_TABLE_FLAG)
1150		type = STATUSTYPE_TABLE;
1151	else
1152		type = STATUSTYPE_INFO;
1153
1154	/* Get all the target info */
1155	num_targets = dm_table_get_num_targets(table);
1156	for (i = 0; i < num_targets; i++) {
1157		struct dm_target *ti = dm_table_get_target(table, i);
1158		size_t l;
1159
1160		remaining = len - (outptr - outbuf);
1161		if (remaining <= sizeof(struct dm_target_spec)) {
1162			param->flags |= DM_BUFFER_FULL_FLAG;
1163			break;
1164		}
1165
1166		spec = (struct dm_target_spec *) outptr;
1167
1168		spec->status = 0;
1169		spec->sector_start = ti->begin;
1170		spec->length = ti->len;
1171		strncpy(spec->target_type, ti->type->name,
1172			sizeof(spec->target_type) - 1);
1173
1174		outptr += sizeof(struct dm_target_spec);
1175		remaining = len - (outptr - outbuf);
1176		if (remaining <= 0) {
1177			param->flags |= DM_BUFFER_FULL_FLAG;
1178			break;
1179		}
1180
1181		/* Get the status/table string from the target driver */
1182		if (ti->type->status) {
1183			if (param->flags & DM_NOFLUSH_FLAG)
1184				status_flags |= DM_STATUS_NOFLUSH_FLAG;
1185			ti->type->status(ti, type, status_flags, outptr, remaining);
1186		} else
1187			outptr[0] = '\0';
1188
1189		l = strlen(outptr) + 1;
1190		if (l == remaining) {
1191			param->flags |= DM_BUFFER_FULL_FLAG;
1192			break;
1193		}
1194
1195		outptr += l;
1196		used = param->data_start + (outptr - outbuf);
1197
1198		outptr = align_ptr(outptr);
1199		spec->next = outptr - outbuf;
1200	}
1201
1202	if (used)
1203		param->data_size = used;
1204
1205	param->target_count = num_targets;
1206}
1207
1208/*
1209 * Wait for a device to report an event
1210 */
1211static int dev_wait(struct file *filp, struct dm_ioctl *param, size_t param_size)
1212{
1213	int r = 0;
1214	struct mapped_device *md;
1215	struct dm_table *table;
1216	int srcu_idx;
1217
1218	md = find_device(param);
1219	if (!md)
1220		return -ENXIO;
1221
1222	/*
1223	 * Wait for a notification event
1224	 */
1225	if (dm_wait_event(md, param->event_nr)) {
1226		r = -ERESTARTSYS;
1227		goto out;
1228	}
1229
1230	/*
1231	 * The userland program is going to want to know what
1232	 * changed to trigger the event, so we may as well tell
1233	 * him and save an ioctl.
1234	 */
1235	__dev_status(md, param);
1236
1237	table = dm_get_live_or_inactive_table(md, param, &srcu_idx);
1238	if (table)
1239		retrieve_status(table, param, param_size);
1240	dm_put_live_table(md, srcu_idx);
1241
1242out:
1243	dm_put(md);
1244
1245	return r;
1246}
1247
1248/*
1249 * Remember the global event number and make it possible to poll
1250 * for further events.
1251 */
1252static int dev_arm_poll(struct file *filp, struct dm_ioctl *param, size_t param_size)
1253{
1254	struct dm_file *priv = filp->private_data;
1255
1256	priv->global_event_nr = atomic_read(&dm_global_event_nr);
1257
1258	return 0;
1259}
1260
1261static inline fmode_t get_mode(struct dm_ioctl *param)
1262{
1263	fmode_t mode = FMODE_READ | FMODE_WRITE;
1264
1265	if (param->flags & DM_READONLY_FLAG)
1266		mode = FMODE_READ;
1267
1268	return mode;
1269}
1270
1271static int next_target(struct dm_target_spec *last, uint32_t next, void *end,
1272		       struct dm_target_spec **spec, char **target_params)
1273{
1274	*spec = (struct dm_target_spec *) ((unsigned char *) last + next);
1275	*target_params = (char *) (*spec + 1);
1276
1277	if (*spec < (last + 1))
1278		return -EINVAL;
1279
1280	return invalid_str(*target_params, end);
1281}
1282
1283static int populate_table(struct dm_table *table,
1284			  struct dm_ioctl *param, size_t param_size)
1285{
1286	int r;
1287	unsigned int i = 0;
1288	struct dm_target_spec *spec = (struct dm_target_spec *) param;
1289	uint32_t next = param->data_start;
1290	void *end = (void *) param + param_size;
1291	char *target_params;
1292
1293	if (!param->target_count) {
1294		DMWARN("populate_table: no targets specified");
1295		return -EINVAL;
1296	}
1297
1298	for (i = 0; i < param->target_count; i++) {
1299
1300		r = next_target(spec, next, end, &spec, &target_params);
1301		if (r) {
1302			DMWARN("unable to find target");
1303			return r;
1304		}
1305
1306		r = dm_table_add_target(table, spec->target_type,
1307					(sector_t) spec->sector_start,
1308					(sector_t) spec->length,
1309					target_params);
1310		if (r) {
1311			DMWARN("error adding target to table");
1312			return r;
1313		}
1314
1315		next = spec->next;
1316	}
1317
1318	return dm_table_complete(table);
1319}
1320
1321static bool is_valid_type(enum dm_queue_mode cur, enum dm_queue_mode new)
1322{
1323	if (cur == new ||
1324	    (cur == DM_TYPE_BIO_BASED && new == DM_TYPE_DAX_BIO_BASED))
1325		return true;
1326
1327	return false;
1328}
1329
1330static int table_load(struct file *filp, struct dm_ioctl *param, size_t param_size)
1331{
1332	int r;
1333	struct hash_cell *hc;
1334	struct dm_table *t, *old_map = NULL;
1335	struct mapped_device *md;
1336	struct target_type *immutable_target_type;
1337
1338	md = find_device(param);
1339	if (!md)
1340		return -ENXIO;
1341
1342	r = dm_table_create(&t, get_mode(param), param->target_count, md);
1343	if (r)
1344		goto err;
1345
1346	/* Protect md->type and md->queue against concurrent table loads. */
1347	dm_lock_md_type(md);
1348	r = populate_table(t, param, param_size);
1349	if (r)
1350		goto err_unlock_md_type;
1351
1352	immutable_target_type = dm_get_immutable_target_type(md);
1353	if (immutable_target_type &&
1354	    (immutable_target_type != dm_table_get_immutable_target_type(t)) &&
1355	    !dm_table_get_wildcard_target(t)) {
1356		DMWARN("can't replace immutable target type %s",
1357		       immutable_target_type->name);
1358		r = -EINVAL;
1359		goto err_unlock_md_type;
1360	}
1361
1362	if (dm_get_md_type(md) == DM_TYPE_NONE) {
1363		/* Initial table load: acquire type of table. */
1364		dm_set_md_type(md, dm_table_get_type(t));
1365
1366		/* setup md->queue to reflect md's type (may block) */
1367		r = dm_setup_md_queue(md, t);
1368		if (r) {
1369			DMWARN("unable to set up device queue for new table.");
1370			goto err_unlock_md_type;
1371		}
1372	} else if (!is_valid_type(dm_get_md_type(md), dm_table_get_type(t))) {
1373		DMWARN("can't change device type (old=%u vs new=%u) after initial table load.",
1374		       dm_get_md_type(md), dm_table_get_type(t));
1375		r = -EINVAL;
1376		goto err_unlock_md_type;
1377	}
1378
1379	dm_unlock_md_type(md);
1380
1381	/* stage inactive table */
1382	down_write(&_hash_lock);
1383	hc = dm_get_mdptr(md);
1384	if (!hc || hc->md != md) {
1385		DMWARN("device has been removed from the dev hash table.");
1386		up_write(&_hash_lock);
1387		r = -ENXIO;
1388		goto err_destroy_table;
1389	}
1390
1391	if (hc->new_map)
1392		old_map = hc->new_map;
1393	hc->new_map = t;
1394	up_write(&_hash_lock);
1395
1396	param->flags |= DM_INACTIVE_PRESENT_FLAG;
1397	__dev_status(md, param);
1398
1399	if (old_map) {
1400		dm_sync_table(md);
1401		dm_table_destroy(old_map);
1402	}
1403
1404	dm_put(md);
1405
1406	return 0;
1407
1408err_unlock_md_type:
1409	dm_unlock_md_type(md);
1410err_destroy_table:
1411	dm_table_destroy(t);
1412err:
1413	dm_put(md);
1414
1415	return r;
1416}
1417
1418static int table_clear(struct file *filp, struct dm_ioctl *param, size_t param_size)
1419{
1420	struct hash_cell *hc;
1421	struct mapped_device *md;
1422	struct dm_table *old_map = NULL;
1423
1424	down_write(&_hash_lock);
1425
1426	hc = __find_device_hash_cell(param);
1427	if (!hc) {
1428		DMDEBUG_LIMIT("device doesn't appear to be in the dev hash table.");
1429		up_write(&_hash_lock);
1430		return -ENXIO;
1431	}
1432
1433	if (hc->new_map) {
1434		old_map = hc->new_map;
1435		hc->new_map = NULL;
1436	}
1437
1438	md = hc->md;
1439	up_write(&_hash_lock);
1440
1441	param->flags &= ~DM_INACTIVE_PRESENT_FLAG;
1442	__dev_status(md, param);
1443
1444	if (old_map) {
1445		dm_sync_table(md);
1446		dm_table_destroy(old_map);
1447	}
1448	dm_put(md);
1449
1450	return 0;
1451}
1452
1453/*
1454 * Retrieves a list of devices used by a particular dm device.
1455 */
1456static void retrieve_deps(struct dm_table *table,
1457			  struct dm_ioctl *param, size_t param_size)
1458{
1459	unsigned int count = 0;
1460	struct list_head *tmp;
1461	size_t len, needed;
1462	struct dm_dev_internal *dd;
1463	struct dm_target_deps *deps;
1464
1465	deps = get_result_buffer(param, param_size, &len);
1466
1467	/*
1468	 * Count the devices.
1469	 */
1470	list_for_each (tmp, dm_table_get_devices(table))
1471		count++;
1472
1473	/*
1474	 * Check we have enough space.
1475	 */
1476	needed = struct_size(deps, dev, count);
1477	if (len < needed) {
1478		param->flags |= DM_BUFFER_FULL_FLAG;
1479		return;
1480	}
1481
1482	/*
1483	 * Fill in the devices.
1484	 */
1485	deps->count = count;
1486	count = 0;
1487	list_for_each_entry (dd, dm_table_get_devices(table), list)
1488		deps->dev[count++] = huge_encode_dev(dd->dm_dev->bdev->bd_dev);
1489
1490	param->data_size = param->data_start + needed;
1491}
1492
1493static int table_deps(struct file *filp, struct dm_ioctl *param, size_t param_size)
1494{
1495	struct mapped_device *md;
1496	struct dm_table *table;
1497	int srcu_idx;
1498
1499	md = find_device(param);
1500	if (!md)
1501		return -ENXIO;
1502
1503	__dev_status(md, param);
1504
1505	table = dm_get_live_or_inactive_table(md, param, &srcu_idx);
1506	if (table)
1507		retrieve_deps(table, param, param_size);
1508	dm_put_live_table(md, srcu_idx);
1509
1510	dm_put(md);
1511
1512	return 0;
1513}
1514
1515/*
1516 * Return the status of a device as a text string for each
1517 * target.
1518 */
1519static int table_status(struct file *filp, struct dm_ioctl *param, size_t param_size)
1520{
1521	struct mapped_device *md;
1522	struct dm_table *table;
1523	int srcu_idx;
1524
1525	md = find_device(param);
1526	if (!md)
1527		return -ENXIO;
1528
1529	__dev_status(md, param);
1530
1531	table = dm_get_live_or_inactive_table(md, param, &srcu_idx);
1532	if (table)
1533		retrieve_status(table, param, param_size);
1534	dm_put_live_table(md, srcu_idx);
1535
1536	dm_put(md);
1537
1538	return 0;
1539}
1540
1541/*
1542 * Process device-mapper dependent messages.  Messages prefixed with '@'
1543 * are processed by the DM core.  All others are delivered to the target.
1544 * Returns a number <= 1 if message was processed by device mapper.
1545 * Returns 2 if message should be delivered to the target.
1546 */
1547static int message_for_md(struct mapped_device *md, unsigned argc, char **argv,
1548			  char *result, unsigned maxlen)
1549{
1550	int r;
1551
1552	if (**argv != '@')
1553		return 2; /* no '@' prefix, deliver to target */
1554
1555	if (!strcasecmp(argv[0], "@cancel_deferred_remove")) {
1556		if (argc != 1) {
1557			DMERR("Invalid arguments for @cancel_deferred_remove");
1558			return -EINVAL;
1559		}
1560		return dm_cancel_deferred_remove(md);
1561	}
1562
1563	r = dm_stats_message(md, argc, argv, result, maxlen);
1564	if (r < 2)
1565		return r;
1566
1567	DMERR("Unsupported message sent to DM core: %s", argv[0]);
1568	return -EINVAL;
1569}
1570
1571/*
1572 * Pass a message to the target that's at the supplied device offset.
1573 */
1574static int target_message(struct file *filp, struct dm_ioctl *param, size_t param_size)
1575{
1576	int r, argc;
1577	char **argv;
1578	struct mapped_device *md;
1579	struct dm_table *table;
1580	struct dm_target *ti;
1581	struct dm_target_msg *tmsg = (void *) param + param->data_start;
1582	size_t maxlen;
1583	char *result = get_result_buffer(param, param_size, &maxlen);
1584	int srcu_idx;
1585
1586	md = find_device(param);
1587	if (!md)
1588		return -ENXIO;
1589
1590	if (tmsg < (struct dm_target_msg *) param->data ||
1591	    invalid_str(tmsg->message, (void *) param + param_size)) {
1592		DMWARN("Invalid target message parameters.");
1593		r = -EINVAL;
1594		goto out;
1595	}
1596
1597	r = dm_split_args(&argc, &argv, tmsg->message);
1598	if (r) {
1599		DMWARN("Failed to split target message parameters");
1600		goto out;
1601	}
1602
1603	if (!argc) {
1604		DMWARN("Empty message received.");
1605		r = -EINVAL;
1606		goto out_argv;
1607	}
1608
1609	r = message_for_md(md, argc, argv, result, maxlen);
1610	if (r <= 1)
1611		goto out_argv;
1612
1613	table = dm_get_live_table(md, &srcu_idx);
1614	if (!table)
1615		goto out_table;
1616
1617	if (dm_deleting_md(md)) {
1618		r = -ENXIO;
1619		goto out_table;
1620	}
1621
1622	ti = dm_table_find_target(table, tmsg->sector);
1623	if (!ti) {
1624		DMWARN("Target message sector outside device.");
1625		r = -EINVAL;
1626	} else if (ti->type->message)
1627		r = ti->type->message(ti, argc, argv, result, maxlen);
1628	else {
1629		DMWARN("Target type does not support messages");
1630		r = -EINVAL;
1631	}
1632
1633 out_table:
1634	dm_put_live_table(md, srcu_idx);
1635 out_argv:
1636	kfree(argv);
1637 out:
1638	if (r >= 0)
1639		__dev_status(md, param);
1640
1641	if (r == 1) {
1642		param->flags |= DM_DATA_OUT_FLAG;
1643		if (dm_message_test_buffer_overflow(result, maxlen))
1644			param->flags |= DM_BUFFER_FULL_FLAG;
1645		else
1646			param->data_size = param->data_start + strlen(result) + 1;
1647		r = 0;
1648	}
1649
1650	dm_put(md);
1651	return r;
1652}
1653
1654/*
1655 * The ioctl parameter block consists of two parts, a dm_ioctl struct
1656 * followed by a data buffer.  This flag is set if the second part,
1657 * which has a variable size, is not used by the function processing
1658 * the ioctl.
1659 */
1660#define IOCTL_FLAGS_NO_PARAMS		1
1661#define IOCTL_FLAGS_ISSUE_GLOBAL_EVENT	2
1662
1663/*-----------------------------------------------------------------
1664 * Implementation of open/close/ioctl on the special char
1665 * device.
1666 *---------------------------------------------------------------*/
1667static ioctl_fn lookup_ioctl(unsigned int cmd, int *ioctl_flags)
1668{
1669	static const struct {
1670		int cmd;
1671		int flags;
1672		ioctl_fn fn;
1673	} _ioctls[] = {
1674		{DM_VERSION_CMD, 0, NULL}, /* version is dealt with elsewhere */
1675		{DM_REMOVE_ALL_CMD, IOCTL_FLAGS_NO_PARAMS | IOCTL_FLAGS_ISSUE_GLOBAL_EVENT, remove_all},
1676		{DM_LIST_DEVICES_CMD, 0, list_devices},
1677
1678		{DM_DEV_CREATE_CMD, IOCTL_FLAGS_NO_PARAMS | IOCTL_FLAGS_ISSUE_GLOBAL_EVENT, dev_create},
1679		{DM_DEV_REMOVE_CMD, IOCTL_FLAGS_NO_PARAMS | IOCTL_FLAGS_ISSUE_GLOBAL_EVENT, dev_remove},
1680		{DM_DEV_RENAME_CMD, IOCTL_FLAGS_ISSUE_GLOBAL_EVENT, dev_rename},
1681		{DM_DEV_SUSPEND_CMD, IOCTL_FLAGS_NO_PARAMS, dev_suspend},
1682		{DM_DEV_STATUS_CMD, IOCTL_FLAGS_NO_PARAMS, dev_status},
1683		{DM_DEV_WAIT_CMD, 0, dev_wait},
1684
1685		{DM_TABLE_LOAD_CMD, 0, table_load},
1686		{DM_TABLE_CLEAR_CMD, IOCTL_FLAGS_NO_PARAMS, table_clear},
1687		{DM_TABLE_DEPS_CMD, 0, table_deps},
1688		{DM_TABLE_STATUS_CMD, 0, table_status},
1689
1690		{DM_LIST_VERSIONS_CMD, 0, list_versions},
1691
1692		{DM_TARGET_MSG_CMD, 0, target_message},
1693		{DM_DEV_SET_GEOMETRY_CMD, 0, dev_set_geometry},
1694		{DM_DEV_ARM_POLL, IOCTL_FLAGS_NO_PARAMS, dev_arm_poll},
1695		{DM_GET_TARGET_VERSION, 0, get_target_version},
1696	};
1697
1698	if (unlikely(cmd >= ARRAY_SIZE(_ioctls)))
1699		return NULL;
1700
1701	cmd = array_index_nospec(cmd, ARRAY_SIZE(_ioctls));
1702	*ioctl_flags = _ioctls[cmd].flags;
1703	return _ioctls[cmd].fn;
1704}
1705
1706/*
1707 * As well as checking the version compatibility this always
1708 * copies the kernel interface version out.
1709 */
1710static int check_version(unsigned int cmd, struct dm_ioctl __user *user)
1711{
1712	uint32_t version[3];
1713	int r = 0;
1714
1715	if (copy_from_user(version, user->version, sizeof(version)))
1716		return -EFAULT;
1717
1718	if ((DM_VERSION_MAJOR != version[0]) ||
1719	    (DM_VERSION_MINOR < version[1])) {
1720		DMWARN("ioctl interface mismatch: "
1721		       "kernel(%u.%u.%u), user(%u.%u.%u), cmd(%d)",
1722		       DM_VERSION_MAJOR, DM_VERSION_MINOR,
1723		       DM_VERSION_PATCHLEVEL,
1724		       version[0], version[1], version[2], cmd);
1725		r = -EINVAL;
1726	}
1727
1728	/*
1729	 * Fill in the kernel version.
1730	 */
1731	version[0] = DM_VERSION_MAJOR;
1732	version[1] = DM_VERSION_MINOR;
1733	version[2] = DM_VERSION_PATCHLEVEL;
1734	if (copy_to_user(user->version, version, sizeof(version)))
1735		return -EFAULT;
1736
1737	return r;
1738}
1739
1740#define DM_PARAMS_MALLOC	0x0001	/* Params allocated with kvmalloc() */
1741#define DM_WIPE_BUFFER		0x0010	/* Wipe input buffer before returning from ioctl */
1742
1743static void free_params(struct dm_ioctl *param, size_t param_size, int param_flags)
1744{
1745	if (param_flags & DM_WIPE_BUFFER)
1746		memset(param, 0, param_size);
1747
1748	if (param_flags & DM_PARAMS_MALLOC)
1749		kvfree(param);
1750}
1751
1752static int copy_params(struct dm_ioctl __user *user, struct dm_ioctl *param_kernel,
1753		       int ioctl_flags, struct dm_ioctl **param, int *param_flags)
1754{
1755	struct dm_ioctl *dmi;
1756	int secure_data;
1757	const size_t minimum_data_size = offsetof(struct dm_ioctl, data);
1758	unsigned noio_flag;
1759
1760	if (copy_from_user(param_kernel, user, minimum_data_size))
1761		return -EFAULT;
1762
1763	if (unlikely(param_kernel->data_size < minimum_data_size) ||
1764	    unlikely(param_kernel->data_size > DM_MAX_TARGETS * DM_MAX_TARGET_PARAMS))
1765		return -EINVAL;
1766
1767	secure_data = param_kernel->flags & DM_SECURE_DATA_FLAG;
1768
1769	*param_flags = secure_data ? DM_WIPE_BUFFER : 0;
1770
1771	if (ioctl_flags & IOCTL_FLAGS_NO_PARAMS) {
1772		dmi = param_kernel;
1773		dmi->data_size = minimum_data_size;
1774		goto data_copied;
1775	}
1776
1777	/*
1778	 * Use __GFP_HIGH to avoid low memory issues when a device is
1779	 * suspended and the ioctl is needed to resume it.
1780	 * Use kmalloc() rather than vmalloc() when we can.
1781	 */
1782	dmi = NULL;
1783	noio_flag = memalloc_noio_save();
1784	dmi = kvmalloc(param_kernel->data_size, GFP_KERNEL | __GFP_HIGH);
1785	memalloc_noio_restore(noio_flag);
1786
1787	if (!dmi) {
1788		if (secure_data && clear_user(user, param_kernel->data_size))
1789			return -EFAULT;
1790		return -ENOMEM;
1791	}
1792
1793	*param_flags |= DM_PARAMS_MALLOC;
1794
1795	/* Copy from param_kernel (which was already copied from user) */
1796	memcpy(dmi, param_kernel, minimum_data_size);
1797
1798	if (copy_from_user(&dmi->data, (char __user *)user + minimum_data_size,
1799			   param_kernel->data_size - minimum_data_size))
1800		goto bad;
1801data_copied:
1802	/* Wipe the user buffer so we do not return it to userspace */
1803	if (secure_data && clear_user(user, param_kernel->data_size))
1804		goto bad;
1805
1806	*param = dmi;
1807	return 0;
1808
1809bad:
1810	free_params(dmi, param_kernel->data_size, *param_flags);
1811
1812	return -EFAULT;
1813}
1814
1815static int validate_params(uint cmd, struct dm_ioctl *param)
1816{
1817	/* Always clear this flag */
1818	param->flags &= ~DM_BUFFER_FULL_FLAG;
1819	param->flags &= ~DM_UEVENT_GENERATED_FLAG;
1820	param->flags &= ~DM_SECURE_DATA_FLAG;
1821	param->flags &= ~DM_DATA_OUT_FLAG;
1822
1823	/* Ignores parameters */
1824	if (cmd == DM_REMOVE_ALL_CMD ||
1825	    cmd == DM_LIST_DEVICES_CMD ||
1826	    cmd == DM_LIST_VERSIONS_CMD)
1827		return 0;
1828
1829	if (cmd == DM_DEV_CREATE_CMD) {
1830		if (!*param->name) {
1831			DMWARN("name not supplied when creating device");
1832			return -EINVAL;
1833		}
1834	} else if (*param->uuid && *param->name) {
1835		DMWARN("only supply one of name or uuid, cmd(%u)", cmd);
1836		return -EINVAL;
1837	}
1838
1839	/* Ensure strings are terminated */
1840	param->name[DM_NAME_LEN - 1] = '\0';
1841	param->uuid[DM_UUID_LEN - 1] = '\0';
1842
1843	return 0;
1844}
1845
1846static int ctl_ioctl(struct file *file, uint command, struct dm_ioctl __user *user)
1847{
1848	int r = 0;
1849	int ioctl_flags;
1850	int param_flags;
1851	unsigned int cmd;
1852	struct dm_ioctl *param;
1853	ioctl_fn fn = NULL;
1854	size_t input_param_size;
1855	struct dm_ioctl param_kernel;
1856
1857	/* only root can play with this */
1858	if (!capable(CAP_SYS_ADMIN))
1859		return -EACCES;
1860
1861	if (_IOC_TYPE(command) != DM_IOCTL)
1862		return -ENOTTY;
1863
1864	cmd = _IOC_NR(command);
1865
1866	/*
1867	 * Check the interface version passed in.  This also
1868	 * writes out the kernel's interface version.
1869	 */
1870	r = check_version(cmd, user);
1871	if (r)
1872		return r;
1873
1874	/*
1875	 * Nothing more to do for the version command.
1876	 */
1877	if (cmd == DM_VERSION_CMD)
1878		return 0;
1879
1880	fn = lookup_ioctl(cmd, &ioctl_flags);
1881	if (!fn) {
1882		DMWARN("dm_ctl_ioctl: unknown command 0x%x", command);
1883		return -ENOTTY;
1884	}
1885
1886	/*
1887	 * Copy the parameters into kernel space.
1888	 */
1889	r = copy_params(user, &param_kernel, ioctl_flags, &param, &param_flags);
1890
1891	if (r)
1892		return r;
1893
1894	input_param_size = param->data_size;
1895	r = validate_params(cmd, param);
1896	if (r)
1897		goto out;
1898
1899	param->data_size = offsetof(struct dm_ioctl, data);
1900	r = fn(file, param, input_param_size);
1901
1902	if (unlikely(param->flags & DM_BUFFER_FULL_FLAG) &&
1903	    unlikely(ioctl_flags & IOCTL_FLAGS_NO_PARAMS))
1904		DMERR("ioctl %d tried to output some data but has IOCTL_FLAGS_NO_PARAMS set", cmd);
1905
1906	if (!r && ioctl_flags & IOCTL_FLAGS_ISSUE_GLOBAL_EVENT)
1907		dm_issue_global_event();
1908
1909	/*
1910	 * Copy the results back to userland.
1911	 */
1912	if (!r && copy_to_user(user, param, param->data_size))
1913		r = -EFAULT;
1914
1915out:
1916	free_params(param, input_param_size, param_flags);
1917	return r;
1918}
1919
1920static long dm_ctl_ioctl(struct file *file, uint command, ulong u)
1921{
1922	return (long)ctl_ioctl(file, command, (struct dm_ioctl __user *)u);
1923}
1924
1925#ifdef CONFIG_COMPAT
1926static long dm_compat_ctl_ioctl(struct file *file, uint command, ulong u)
1927{
1928	return (long)dm_ctl_ioctl(file, command, (ulong) compat_ptr(u));
1929}
1930#else
1931#define dm_compat_ctl_ioctl NULL
1932#endif
1933
1934static int dm_open(struct inode *inode, struct file *filp)
1935{
1936	int r;
1937	struct dm_file *priv;
1938
1939	r = nonseekable_open(inode, filp);
1940	if (unlikely(r))
1941		return r;
1942
1943	priv = filp->private_data = kmalloc(sizeof(struct dm_file), GFP_KERNEL);
1944	if (!priv)
1945		return -ENOMEM;
1946
1947	priv->global_event_nr = atomic_read(&dm_global_event_nr);
1948
1949	return 0;
1950}
1951
1952static int dm_release(struct inode *inode, struct file *filp)
1953{
1954	kfree(filp->private_data);
1955	return 0;
1956}
1957
1958static __poll_t dm_poll(struct file *filp, poll_table *wait)
1959{
1960	struct dm_file *priv = filp->private_data;
1961	__poll_t mask = 0;
1962
1963	poll_wait(filp, &dm_global_eventq, wait);
1964
1965	if ((int)(atomic_read(&dm_global_event_nr) - priv->global_event_nr) > 0)
1966		mask |= EPOLLIN;
1967
1968	return mask;
1969}
1970
1971static const struct file_operations _ctl_fops = {
1972	.open    = dm_open,
1973	.release = dm_release,
1974	.poll    = dm_poll,
1975	.unlocked_ioctl	 = dm_ctl_ioctl,
1976	.compat_ioctl = dm_compat_ctl_ioctl,
1977	.owner	 = THIS_MODULE,
1978	.llseek  = noop_llseek,
1979};
1980
1981static struct miscdevice _dm_misc = {
1982	.minor		= MAPPER_CTRL_MINOR,
1983	.name  		= DM_NAME,
1984	.nodename	= DM_DIR "/" DM_CONTROL_NODE,
1985	.fops  		= &_ctl_fops
1986};
1987
1988MODULE_ALIAS_MISCDEV(MAPPER_CTRL_MINOR);
1989MODULE_ALIAS("devname:" DM_DIR "/" DM_CONTROL_NODE);
1990
1991/*
1992 * Create misc character device and link to DM_DIR/control.
1993 */
1994int __init dm_interface_init(void)
1995{
1996	int r;
1997
1998	r = dm_hash_init();
1999	if (r)
2000		return r;
2001
2002	r = misc_register(&_dm_misc);
2003	if (r) {
2004		DMERR("misc_register failed for control device");
2005		dm_hash_exit();
2006		return r;
2007	}
2008
2009	DMINFO("%d.%d.%d%s initialised: %s", DM_VERSION_MAJOR,
2010	       DM_VERSION_MINOR, DM_VERSION_PATCHLEVEL, DM_VERSION_EXTRA,
2011	       DM_DRIVER_EMAIL);
2012	return 0;
2013}
2014
2015void dm_interface_exit(void)
2016{
2017	misc_deregister(&_dm_misc);
2018	dm_hash_exit();
2019}
2020
2021/**
2022 * dm_copy_name_and_uuid - Copy mapped device name & uuid into supplied buffers
2023 * @md: Pointer to mapped_device
2024 * @name: Buffer (size DM_NAME_LEN) for name
2025 * @uuid: Buffer (size DM_UUID_LEN) for uuid or empty string if uuid not defined
2026 */
2027int dm_copy_name_and_uuid(struct mapped_device *md, char *name, char *uuid)
2028{
2029	int r = 0;
2030	struct hash_cell *hc;
2031
2032	if (!md)
2033		return -ENXIO;
2034
2035	mutex_lock(&dm_hash_cells_mutex);
2036	hc = dm_get_mdptr(md);
2037	if (!hc || hc->md != md) {
2038		r = -ENXIO;
2039		goto out;
2040	}
2041
2042	if (name)
2043		strcpy(name, hc->name);
2044	if (uuid)
2045		strcpy(uuid, hc->uuid ? : "");
2046
2047out:
2048	mutex_unlock(&dm_hash_cells_mutex);
2049
2050	return r;
2051}
2052EXPORT_SYMBOL_GPL(dm_copy_name_and_uuid);
2053
2054/**
2055 * dm_early_create - create a mapped device in early boot.
2056 *
2057 * @dmi: Contains main information of the device mapping to be created.
2058 * @spec_array: array of pointers to struct dm_target_spec. Describes the
2059 * mapping table of the device.
2060 * @target_params_array: array of strings with the parameters to a specific
2061 * target.
2062 *
2063 * Instead of having the struct dm_target_spec and the parameters for every
2064 * target embedded at the end of struct dm_ioctl (as performed in a normal
2065 * ioctl), pass them as arguments, so the caller doesn't need to serialize them.
2066 * The size of the spec_array and target_params_array is given by
2067 * @dmi->target_count.
2068 * This function is supposed to be called in early boot, so locking mechanisms
2069 * to protect against concurrent loads are not required.
2070 */
2071int __init dm_early_create(struct dm_ioctl *dmi,
2072			   struct dm_target_spec **spec_array,
2073			   char **target_params_array)
2074{
2075	int r, m = DM_ANY_MINOR;
2076	struct dm_table *t, *old_map;
2077	struct mapped_device *md;
2078	unsigned int i;
2079
2080	if (!dmi->target_count)
2081		return -EINVAL;
2082
2083	r = check_name(dmi->name);
2084	if (r)
2085		return r;
2086
2087	if (dmi->flags & DM_PERSISTENT_DEV_FLAG)
2088		m = MINOR(huge_decode_dev(dmi->dev));
2089
2090	/* alloc dm device */
2091	r = dm_create(m, &md);
2092	if (r)
2093		return r;
2094
2095	/* hash insert */
2096	r = dm_hash_insert(dmi->name, *dmi->uuid ? dmi->uuid : NULL, md);
2097	if (r)
2098		goto err_destroy_dm;
2099
2100	/* alloc table */
2101	r = dm_table_create(&t, get_mode(dmi), dmi->target_count, md);
2102	if (r)
2103		goto err_hash_remove;
2104
2105	/* add targets */
2106	for (i = 0; i < dmi->target_count; i++) {
2107		r = dm_table_add_target(t, spec_array[i]->target_type,
2108					(sector_t) spec_array[i]->sector_start,
2109					(sector_t) spec_array[i]->length,
2110					target_params_array[i]);
2111		if (r) {
2112			DMWARN("error adding target to table");
2113			goto err_destroy_table;
2114		}
2115	}
2116
2117	/* finish table */
2118	r = dm_table_complete(t);
2119	if (r)
2120		goto err_destroy_table;
2121
2122	md->type = dm_table_get_type(t);
2123	/* setup md->queue to reflect md's type (may block) */
2124	r = dm_setup_md_queue(md, t);
2125	if (r) {
2126		DMWARN("unable to set up device queue for new table.");
2127		goto err_destroy_table;
2128	}
2129
2130	/* Set new map */
2131	dm_suspend(md, 0);
2132	old_map = dm_swap_table(md, t);
2133	if (IS_ERR(old_map)) {
2134		r = PTR_ERR(old_map);
2135		goto err_destroy_table;
2136	}
2137	set_disk_ro(dm_disk(md), !!(dmi->flags & DM_READONLY_FLAG));
2138
2139	/* resume device */
2140	r = dm_resume(md);
2141	if (r)
2142		goto err_destroy_table;
2143
2144	DMINFO("%s (%s) is ready", md->disk->disk_name, dmi->name);
2145	dm_put(md);
2146	return 0;
2147
2148err_destroy_table:
2149	dm_table_destroy(t);
2150err_hash_remove:
2151	(void) __hash_remove(__get_name_cell(dmi->name));
2152	/* release reference from __get_name_cell */
2153	dm_put(md);
2154err_destroy_dm:
2155	dm_put(md);
2156	dm_destroy(md);
2157	return r;
2158}
2159