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