328 lines · 8.1 KB · C
RawDownload
// SPDX-License-Identifier: GPL-2.0-only /* * Copyright (C) 2017 The Chromium OS Authors <chromium-os-dev@chromium.org> * * This file is released under the GPLv2. */ #include <linux/ctype.h>#include <linux/delay.h>#include <linux/device.h>#include <linux/device-mapper.h>#include <linux/init.h>#include <linux/list.h>#include <linux/moduleparam.h> #define DM_MSG_PREFIX "init"#define DM_MAX_DEVICES 256#define DM_MAX_TARGETS 256#define DM_MAX_STR_SIZE 4096#define DM_MAX_WAITFOR 256 static char *create; static char *waitfor[DM_MAX_WAITFOR]; /* * Format: dm-mod.create=<name>,<uuid>,<minor>,<flags>,<table>[,<table>+][;<name>,<uuid>,<minor>,<flags>,<table>[,<table>+]+] * Table format: <start_sector> <num_sectors> <target_type> <target_args> * Block devices to wait for to become available before setting up tables: * dm-mod.waitfor=<device1>[,..,<deviceN>] * * See Documentation/admin-guide/device-mapper/dm-init.rst for dm-mod.create="..." format * details. */ struct dm_device {	struct dm_ioctl dmi;	struct dm_target_spec *table[DM_MAX_TARGETS];	char *target_args_array[DM_MAX_TARGETS];	struct list_head list;}; static const char * const dm_allowed_targets[] __initconst = {	"crypt",	"delay",	"linear",	"snapshot-origin",	"striped",	"verity",}; static int __init dm_verify_target_type(const char *target){	unsigned int i; 	for (i = 0; i < ARRAY_SIZE(dm_allowed_targets); i++) {		if (!strcmp(dm_allowed_targets[i], target))			return 0;	}	return -EINVAL;} static void __init dm_setup_cleanup(struct list_head *devices){	struct dm_device *dev, *tmp;	unsigned int i; 	list_for_each_entry_safe(dev, tmp, devices, list) {		list_del(&dev->list);		for (i = 0; i < dev->dmi.target_count; i++) {			kfree(dev->table[i]);			kfree(dev->target_args_array[i]);		}		kfree(dev);	}} /** * str_field_delimit - delimit a string based on a separator char. * @str: the pointer to the string to delimit. * @separator: char that delimits the field * * Find a @separator and replace it by '\0'. * Remove leading and trailing spaces. * Return the remainder string after the @separator. */static char __init *str_field_delimit(char **str, char separator){	char *s; 	/* TODO: add support for escaped characters */	*str = skip_spaces(*str);	s = strchr(*str, separator);	/* Delimit the field and remove trailing spaces */	if (s)		*s = '\0';	*str = strim(*str);	return s ? ++s : NULL;} /** * dm_parse_table_entry - parse a table entry * @dev: device to store the parsed information. * @str: the pointer to a string with the format: *	<start_sector> <num_sectors> <target_type> <target_args>[, ...] * * Return the remainder string after the table entry, i.e, after the comma which * delimits the entry or NULL if reached the end of the string. */static char __init *dm_parse_table_entry(struct dm_device *dev, char *str){	const unsigned int n = dev->dmi.target_count - 1;	struct dm_target_spec *sp;	unsigned int i;	/* fields:  */	char *field[4];	char *next; 	field[0] = str;	/* Delimit first 3 fields that are separated by space */	for (i = 0; i < ARRAY_SIZE(field) - 1; i++) {		field[i + 1] = str_field_delimit(&field[i], ' ');		if (!field[i + 1])			return ERR_PTR(-EINVAL);	}	/* Delimit last field that can be terminated by comma */	next = str_field_delimit(&field[i], ','); 	sp = kzalloc_obj(*sp);	if (!sp)		return ERR_PTR(-ENOMEM);	dev->table[n] = sp; 	/* start_sector */	if (kstrtoull(field[0], 0, &sp->sector_start))		return ERR_PTR(-EINVAL);	/* num_sector */	if (kstrtoull(field[1], 0, &sp->length))		return ERR_PTR(-EINVAL);	/* target_type */	strscpy(sp->target_type, field[2], sizeof(sp->target_type));	if (dm_verify_target_type(sp->target_type)) {		DMERR("invalid type \"%s\"", sp->target_type);		return ERR_PTR(-EINVAL);	}	/* target_args */	dev->target_args_array[n] = kstrndup(field[3], DM_MAX_STR_SIZE,					     GFP_KERNEL);	if (!dev->target_args_array[n])		return ERR_PTR(-ENOMEM); 	return next;} /** * dm_parse_table - parse "dm-mod.create=" table field * @dev: device to store the parsed information. * @str: the pointer to a string with the format: *	<table>[,<table>+] */static int __init dm_parse_table(struct dm_device *dev, char *str){	char *table_entry = str; 	while (table_entry) {		DMDEBUG("parsing table \"%s\"", str);		if (++dev->dmi.target_count > DM_MAX_TARGETS) {			DMERR("too many targets %u > %d",			      dev->dmi.target_count, DM_MAX_TARGETS);			return -EINVAL;		}		table_entry = dm_parse_table_entry(dev, table_entry);		if (IS_ERR(table_entry)) {			DMERR("couldn't parse table");			return PTR_ERR(table_entry);		}	} 	return 0;} /** * dm_parse_device_entry - parse a device entry * @dev: device to store the parsed information. * @str: the pointer to a string with the format: *	name,uuid,minor,flags,table[; ...] * * Return the remainder string after the table entry, i.e, after the semi-colon * which delimits the entry or NULL if reached the end of the string. */static char __init *dm_parse_device_entry(struct dm_device *dev, char *str){	/* There are 5 fields: name,uuid,minor,flags,table; */	char *field[5];	unsigned int i;	char *next; 	field[0] = str;	/* Delimit first 4 fields that are separated by comma */	for (i = 0; i < ARRAY_SIZE(field) - 1; i++) {		field[i+1] = str_field_delimit(&field[i], ',');		if (!field[i+1])			return ERR_PTR(-EINVAL);	}	/* Delimit last field that can be delimited by semi-colon */	next = str_field_delimit(&field[i], ';'); 	/* name */	strscpy(dev->dmi.name, field[0], sizeof(dev->dmi.name));	/* uuid */	strscpy(dev->dmi.uuid, field[1], sizeof(dev->dmi.uuid));	/* minor */	if (strlen(field[2])) {		if (kstrtoull(field[2], 0, &dev->dmi.dev) ||		    dev->dmi.dev >= (1 << MINORBITS))			return ERR_PTR(-EINVAL);		dev->dmi.dev = huge_encode_dev((dev_t)dev->dmi.dev);		dev->dmi.flags |= DM_PERSISTENT_DEV_FLAG;	}	/* flags */	if (!strcmp(field[3], "ro"))		dev->dmi.flags |= DM_READONLY_FLAG;	else if (strcmp(field[3], "rw"))		return ERR_PTR(-EINVAL);	/* table */	if (dm_parse_table(dev, field[4]))		return ERR_PTR(-EINVAL); 	return next;} /** * dm_parse_devices - parse "dm-mod.create=" argument * @devices: list of struct dm_device to store the parsed information. * @str: the pointer to a string with the format: *	<device>[;<device>+] */static int __init dm_parse_devices(struct list_head *devices, char *str){	unsigned long ndev = 0;	struct dm_device *dev;	char *device = str; 	DMDEBUG("parsing \"%s\"", str);	while (device) {		dev = kzalloc_obj(*dev);		if (!dev)			return -ENOMEM;		list_add_tail(&dev->list, devices); 		if (++ndev > DM_MAX_DEVICES) {			DMERR("too many devices %lu > %d",			      ndev, DM_MAX_DEVICES);			return -EINVAL;		} 		device = dm_parse_device_entry(dev, device);		if (IS_ERR(device)) {			DMERR("couldn't parse device");			return PTR_ERR(device);		}	} 	return 0;} /** * dm_init_init - parse "dm-mod.create=" argument and configure drivers */static int __init dm_init_init(void){	struct dm_device *dev;	LIST_HEAD(devices);	char *str;	int i, r; 	if (!create)		return 0; 	if (strlen(create) >= DM_MAX_STR_SIZE) {		DMERR("Argument is too big. Limit is %d", DM_MAX_STR_SIZE);		return -EINVAL;	}	str = kstrndup(create, DM_MAX_STR_SIZE, GFP_KERNEL);	if (!str)		return -ENOMEM; 	r = dm_parse_devices(&devices, str);	if (r)		goto out; 	DMINFO("waiting for all devices to be available before creating mapped devices");	wait_for_device_probe(); 	for (i = 0; i < ARRAY_SIZE(waitfor); i++) {		if (waitfor[i]) {			dev_t dev; 			DMINFO("waiting for device %s ...", waitfor[i]);			while (early_lookup_bdev(waitfor[i], &dev))				fsleep(5000);		}	} 	if (waitfor[0]) {		wait_for_device_probe();		DMINFO("all devices available");	} 	list_for_each_entry(dev, &devices, list) {		if (dm_early_create(&dev->dmi, dev->table,				    dev->target_args_array))			break;	}out:	kfree(str);	dm_setup_cleanup(&devices);	return r;} late_initcall(dm_init_init); module_param(create, charp, 0);MODULE_PARM_DESC(create, "Create a mapped device in early boot"); module_param_array(waitfor, charp, NULL, 0);MODULE_PARM_DESC(waitfor, "Devices to wait for before setting up tables");