summaryrefslogtreecommitdiff
path: root/miscutils/flash_lock_unlock.c
diff options
context:
space:
mode:
Diffstat (limited to 'miscutils/flash_lock_unlock.c')
-rw-r--r--miscutils/flash_lock_unlock.c69
1 files changed, 69 insertions, 0 deletions
diff --git a/miscutils/flash_lock_unlock.c b/miscutils/flash_lock_unlock.c
new file mode 100644
index 0000000..f4e2f73
--- /dev/null
+++ b/miscutils/flash_lock_unlock.c
@@ -0,0 +1,69 @@
+/* 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;
+}