1110 lines · 24.7 KB · C
RawDownload
/* * Copyright (c) International Business Machines Corp., 2006 * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See * the GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA * * Author: Artem Bityutskiy (Битюцкий Артём) */ /* * This file includes implementation of UBI character device operations. * * There are two kinds of character devices in UBI: UBI character devices and * UBI volume character devices. UBI character devices allow users to * manipulate whole volumes: create, remove, and re-size them. Volume character * devices provide volume I/O capabilities. * * Major and minor numbers are assigned dynamically to both UBI and volume * character devices. * * Well, there is the third kind of character devices - the UBI control * character device, which allows to manipulate by UBI devices - create and * delete them. In other words, it is used for attaching and detaching MTD * devices. */ #include <linux/module.h>#include <linux/stat.h>#include <linux/slab.h>#include <linux/ioctl.h>#include <linux/capability.h>#include <linux/uaccess.h>#include <linux/compat.h>#include <linux/math64.h>#include <mtd/ubi-user.h>#include "ubi.h" /** * get_exclusive - get exclusive access to an UBI volume. * @desc: volume descriptor * * This function changes UBI volume open mode to "exclusive". Returns previous * mode value (positive integer) in case of success and a negative error code * in case of failure. */static int get_exclusive(struct ubi_volume_desc *desc){	int users, err;	struct ubi_volume *vol = desc->vol; 	spin_lock(&vol->ubi->volumes_lock);	users = vol->readers + vol->writers + vol->exclusive + vol->metaonly;	ubi_assert(users > 0);	if (users > 1) {		ubi_err(vol->ubi, "%d users for volume %d", users, vol->vol_id);		err = -EBUSY;	} else {		vol->readers = vol->writers = vol->metaonly = 0;		vol->exclusive = 1;		err = desc->mode;		desc->mode = UBI_EXCLUSIVE;	}	spin_unlock(&vol->ubi->volumes_lock); 	return err;} /** * revoke_exclusive - revoke exclusive mode. * @desc: volume descriptor * @mode: new mode to switch to */static void revoke_exclusive(struct ubi_volume_desc *desc, int mode){	struct ubi_volume *vol = desc->vol; 	spin_lock(&vol->ubi->volumes_lock);	ubi_assert(vol->readers == 0 && vol->writers == 0 && vol->metaonly == 0);	ubi_assert(vol->exclusive == 1 && desc->mode == UBI_EXCLUSIVE);	vol->exclusive = 0;	if (mode == UBI_READONLY)		vol->readers = 1;	else if (mode == UBI_READWRITE)		vol->writers = 1;	else if (mode == UBI_METAONLY)		vol->metaonly = 1;	else		vol->exclusive = 1;	spin_unlock(&vol->ubi->volumes_lock); 	desc->mode = mode;} static int vol_cdev_open(struct inode *inode, struct file *file){	struct ubi_volume_desc *desc;	int vol_id = iminor(inode) - 1, mode, ubi_num; 	ubi_num = ubi_major2num(imajor(inode));	if (ubi_num < 0)		return ubi_num; 	if (file->f_mode & FMODE_WRITE)		mode = UBI_READWRITE;	else		mode = UBI_READONLY; 	dbg_gen("open device %d, volume %d, mode %d",		ubi_num, vol_id, mode); 	desc = ubi_open_volume(ubi_num, vol_id, mode);	if (IS_ERR(desc))		return PTR_ERR(desc); 	file->private_data = desc;	return 0;} static int vol_cdev_release(struct inode *inode, struct file *file){	struct ubi_volume_desc *desc = file->private_data;	struct ubi_volume *vol = desc->vol; 	dbg_gen("release device %d, volume %d, mode %d",		vol->ubi->ubi_num, vol->vol_id, desc->mode); 	if (vol->updating) {		ubi_warn(vol->ubi, "update of volume %d not finished, volume is damaged",			 vol->vol_id);		ubi_assert(!vol->changing_leb);		vol->updating = 0;		vfree(vol->upd_buf);	} else if (vol->changing_leb) {		dbg_gen("only %lld of %lld bytes received for atomic LEB change for volume %d:%d, cancel",			vol->upd_received, vol->upd_bytes, vol->ubi->ubi_num,			vol->vol_id);		vol->changing_leb = 0;		vfree(vol->upd_buf);	} 	ubi_close_volume(desc);	return 0;} static loff_t vol_cdev_llseek(struct file *file, loff_t offset, int origin){	struct ubi_volume_desc *desc = file->private_data;	struct ubi_volume *vol = desc->vol; 	if (vol->updating) {		/* Update is in progress, seeking is prohibited */		ubi_err(vol->ubi, "updating");		return -EBUSY;	} 	return fixed_size_llseek(file, offset, origin, vol->used_bytes);} static int vol_cdev_fsync(struct file *file, loff_t start, loff_t end,			  int datasync){	struct ubi_volume_desc *desc = file->private_data;	struct ubi_device *ubi = desc->vol->ubi;	struct inode *inode = file_inode(file);	int err;	inode_lock(inode);	err = ubi_sync(ubi->ubi_num);	inode_unlock(inode);	return err;}  static ssize_t vol_cdev_read(struct file *file, __user char *buf, size_t count,			     loff_t *offp){	struct ubi_volume_desc *desc = file->private_data;	struct ubi_volume *vol = desc->vol;	struct ubi_device *ubi = vol->ubi;	int err, lnum, off, len,  tbuf_size;	size_t count_save = count;	void *tbuf; 	dbg_gen("read %zd bytes from offset %lld of volume %d",		count, *offp, vol->vol_id); 	if (vol->updating) {		ubi_err(vol->ubi, "updating");		return -EBUSY;	}	if (vol->upd_marker) {		ubi_err(vol->ubi, "damaged volume, update marker is set");		return -EBADF;	}	if (*offp == vol->used_bytes || count == 0)		return 0; 	if (vol->corrupted)		dbg_gen("read from corrupted volume %d", vol->vol_id); 	if (*offp + count > vol->used_bytes)		count_save = count = vol->used_bytes - *offp; 	tbuf_size = vol->usable_leb_size;	if (count < tbuf_size)		tbuf_size = ALIGN(count, ubi->min_io_size);	tbuf = vmalloc(tbuf_size);	if (!tbuf)		return -ENOMEM; 	len = count > tbuf_size ? tbuf_size : count;	lnum = div_u64_rem(*offp, vol->usable_leb_size, &off); 	do {		cond_resched(); 		if (off + len >= vol->usable_leb_size)			len = vol->usable_leb_size - off; 		err = ubi_eba_read_leb(ubi, vol, lnum, tbuf, off, len, 0);		if (err)			break; 		off += len;		if (off == vol->usable_leb_size) {			lnum += 1;			off -= vol->usable_leb_size;		} 		count -= len;		*offp += len; 		err = copy_to_user(buf, tbuf, len);		if (err) {			err = -EFAULT;			break;		} 		buf += len;		len = count > tbuf_size ? tbuf_size : count;	} while (count); 	vfree(tbuf);	return err ? err : count_save - count;} /* * This function allows to directly write to dynamic UBI volumes, without * issuing the volume update operation. */static ssize_t vol_cdev_direct_write(struct file *file, const char __user *buf,				     size_t count, loff_t *offp){	struct ubi_volume_desc *desc = file->private_data;	struct ubi_volume *vol = desc->vol;	struct ubi_device *ubi = vol->ubi;	int lnum, off, len, tbuf_size, err = 0;	size_t count_save = count;	char *tbuf; 	if (!vol->direct_writes)		return -EPERM; 	dbg_gen("requested: write %zd bytes to offset %lld of volume %u",		count, *offp, vol->vol_id); 	if (vol->vol_type == UBI_STATIC_VOLUME)		return -EROFS; 	lnum = div_u64_rem(*offp, vol->usable_leb_size, &off);	if (off & (ubi->min_io_size - 1)) {		ubi_err(ubi, "unaligned position");		return -EINVAL;	} 	if (*offp + count > vol->used_bytes)		count_save = count = vol->used_bytes - *offp; 	/* We can write only in fractions of the minimum I/O unit */	if (count & (ubi->min_io_size - 1)) {		ubi_err(ubi, "unaligned write length");		return -EINVAL;	} 	tbuf_size = vol->usable_leb_size;	if (count < tbuf_size)		tbuf_size = ALIGN(count, ubi->min_io_size);	tbuf = vmalloc(tbuf_size);	if (!tbuf)		return -ENOMEM; 	len = count > tbuf_size ? tbuf_size : count; 	while (count) {		cond_resched(); 		if (off + len >= vol->usable_leb_size)			len = vol->usable_leb_size - off; 		err = copy_from_user(tbuf, buf, len);		if (err) {			err = -EFAULT;			break;		} 		err = ubi_eba_write_leb(ubi, vol, lnum, tbuf, off, len);		if (err)			break; 		off += len;		if (off == vol->usable_leb_size) {			lnum += 1;			off -= vol->usable_leb_size;		} 		count -= len;		*offp += len;		buf += len;		len = count > tbuf_size ? tbuf_size : count;	} 	vfree(tbuf);	return err ? err : count_save - count;} static ssize_t vol_cdev_write(struct file *file, const char __user *buf,			      size_t count, loff_t *offp){	int err = 0;	struct ubi_volume_desc *desc = file->private_data;	struct ubi_volume *vol = desc->vol;	struct ubi_device *ubi = vol->ubi; 	if (!vol->updating && !vol->changing_leb)		return vol_cdev_direct_write(file, buf, count, offp); 	if (vol->updating)		err = ubi_more_update_data(ubi, vol, buf, count);	else		err = ubi_more_leb_change_data(ubi, vol, buf, count); 	if (err < 0) {		ubi_err(ubi, "cannot accept more %zd bytes of data, error %d",			count, err);		return err;	} 	if (err) {		/*		 * The operation is finished, @err contains number of actually		 * written bytes.		 */		count = err; 		if (vol->changing_leb) {			revoke_exclusive(desc, UBI_READWRITE);			return count;		} 		err = ubi_check_volume(ubi, vol->vol_id);		if (err < 0)			return err; 		if (err) {			ubi_warn(ubi, "volume %d on UBI device %d is corrupted",				 vol->vol_id, ubi->ubi_num);			vol->corrupted = 1;		}		vol->checked = 1;		ubi_volume_notify(ubi, vol, UBI_VOLUME_UPDATED);		revoke_exclusive(desc, UBI_READWRITE);	} 	return count;} static long vol_cdev_ioctl(struct file *file, unsigned int cmd,			   unsigned long arg){	int err = 0;	struct ubi_volume_desc *desc = file->private_data;	struct ubi_volume *vol = desc->vol;	struct ubi_device *ubi = vol->ubi;	void __user *argp = (void __user *)arg; 	switch (cmd) {	/* Volume update command */	case UBI_IOCVOLUP:	{		int64_t bytes, rsvd_bytes; 		if (!capable(CAP_SYS_RESOURCE)) {			err = -EPERM;			break;		} 		err = copy_from_user(&bytes, argp, sizeof(int64_t));		if (err) {			err = -EFAULT;			break;		} 		if (desc->mode == UBI_READONLY) {			err = -EROFS;			break;		} 		rsvd_bytes = (long long)vol->reserved_pebs *					vol->usable_leb_size;		if (bytes < 0 || bytes > rsvd_bytes) {			err = -EINVAL;			break;		} 		err = get_exclusive(desc);		if (err < 0)			break; 		err = ubi_start_update(ubi, vol, bytes);		if (bytes == 0) {			ubi_volume_notify(ubi, vol, UBI_VOLUME_UPDATED);			revoke_exclusive(desc, UBI_READWRITE);		}		break;	} 	/* Atomic logical eraseblock change command */	case UBI_IOCEBCH:	{		struct ubi_leb_change_req req; 		err = copy_from_user(&req, argp,				     sizeof(struct ubi_leb_change_req));		if (err) {			err = -EFAULT;			break;		} 		if (desc->mode == UBI_READONLY ||		    vol->vol_type == UBI_STATIC_VOLUME) {			err = -EROFS;			break;		} 		/* Validate the request */		err = -EINVAL;		if (!ubi_leb_valid(vol, req.lnum) ||		    req.bytes < 0 || req.bytes > vol->usable_leb_size)			break; 		err = get_exclusive(desc);		if (err < 0)			break; 		err = ubi_start_leb_change(ubi, vol, &req);		if (req.bytes == 0)			revoke_exclusive(desc, UBI_READWRITE);		break;	} 	/* Logical eraseblock erasure command */	case UBI_IOCEBER:	{		int32_t lnum; 		err = get_user(lnum, (__user int32_t *)argp);		if (err) {			err = -EFAULT;			break;		} 		if (desc->mode == UBI_READONLY ||		    vol->vol_type == UBI_STATIC_VOLUME) {			err = -EROFS;			break;		} 		if (!ubi_leb_valid(vol, lnum)) {			err = -EINVAL;			break;		} 		dbg_gen("erase LEB %d:%d", vol->vol_id, lnum);		err = ubi_eba_unmap_leb(ubi, vol, lnum);		if (err)			break; 		err = ubi_wl_flush(ubi, UBI_ALL, UBI_ALL);		break;	}

Lines 1–500 of 1110.