70 lines
1.6 KiB
C
70 lines
1.6 KiB
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;
|
||
|
|
}
|