busybox/miscutils/flash_lock_unlock.c

/* vi: set sw=4 ts=4: */
/* Ported to busybox from mtd-utils.
 *
 * Licensed under GPLv2, see file LICENSE in this tarball for details.
 */
#include "libbb.h"
#include <mtd/mtd-user.h>

int flash_lock_unlock_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int flash_lock_unlock_main(int argc UNUSED_PARAM, char **argv)
{
	/* note: fields in these structs are 32-bits.
	 * apparently we can't win anything by using off_t
	 * or long long's for offset and/or sectors vars. */
	struct mtd_info_user info;
	struct erase_info_user lock;
	unsigned long offset;
	long sectors;
	int fd;

#define do_lock (ENABLE_FLASH_LOCK && (!ENABLE_FLASH_UNLOCK || (applet_name[6] == 'l')))

	if (!argv[1])
		bb_show_usage();

	/* parse offset and number of sectors to lock */
	offset = 0;
	sectors = -1;
	if (do_lock) {
		if (!argv[2] || !argv[3])
			bb_show_usage();
		offset = xstrtoul(argv[2], 0);
		sectors = xstrtol(argv[3], 0);
	}

	fd = xopen(argv[1], O_RDWR);

	xioctl(fd, MEMGETINFO, &info);

	lock.start = 0;
	lock.length = info.size;
	if (do_lock) {
		unsigned long size = info.size - info.erasesize;
		if (offset > size) {
			bb_error_msg_and_die("%lx is beyond device size %lx\n",
					offset, size);
		}

		if (sectors == -1) {
			sectors = info.size / info.erasesize;
		} else {
// isn't this useless?
			unsigned long num = info.size / info.erasesize;
			if (sectors > num) {
				bb_error_msg_and_die("%ld are too many "
						"sectors, device only has "
						"%ld\n", sectors, num);
			}
		}

		lock.start = offset;
		lock.length = sectors * info.erasesize;
		xioctl(fd, MEMLOCK, &lock);
	} else {
		xioctl(fd, MEMUNLOCK, &lock);
	}

	return EXIT_SUCCESS;
}
1	/* vi: set sw=4 ts=4: */
2	/* Ported to busybox from mtd-utils.
3	*
4	* Licensed under GPLv2, see file LICENSE in this tarball for details.
5	*/
6	#include "libbb.h"
7	#include <mtd/mtd-user.h>
8
9	int flash_lock_unlock_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
10	int flash_lock_unlock_main(int argc UNUSED_PARAM, char **argv)
11	{
12	/* note: fields in these structs are 32-bits.
13	* apparently we can't win anything by using off_t
14	* or long long's for offset and/or sectors vars. */
15	struct mtd_info_user info;
16	struct erase_info_user lock;
17	unsigned long offset;
18	long sectors;
19	int fd;
20
21	#define do_lock (ENABLE_FLASH_LOCK && (!ENABLE_FLASH_UNLOCK \|\| (applet_name[6] == 'l')))
22
23	if (!argv[1])
24	bb_show_usage();
25
26	/* parse offset and number of sectors to lock */
27	offset = 0;
28	sectors = -1;
29	if (do_lock) {
30	if (!argv[2] \|\| !argv[3])
31	bb_show_usage();
32	offset = xstrtoul(argv[2], 0);
33	sectors = xstrtol(argv[3], 0);
34	}
35
36	fd = xopen(argv[1], O_RDWR);
37
38	xioctl(fd, MEMGETINFO, &info);
39
40	lock.start = 0;
41	lock.length = info.size;
42	if (do_lock) {
43	unsigned long size = info.size - info.erasesize;
44	if (offset > size) {
45	bb_error_msg_and_die("%lx is beyond device size %lx\n",
46	offset, size);
47	}
48
49	if (sectors == -1) {
50	sectors = info.size / info.erasesize;
51	} else {
52	// isn't this useless?
53	unsigned long num = info.size / info.erasesize;
54	if (sectors > num) {
55	bb_error_msg_and_die("%ld are too many "
56	"sectors, device only has "
57	"%ld\n", sectors, num);
58	}
59	}
60
61	lock.start = offset;
62	lock.length = sectors * info.erasesize;
63	xioctl(fd, MEMLOCK, &lock);
64	} else {
65	xioctl(fd, MEMUNLOCK, &lock);
66	}
67
68	return EXIT_SUCCESS;
69	}