Baruch Siach | 3324c96 | 2010-06-16 12:22:58 +0200 | [diff] [blame] | 1 | /* Ported to busybox from mtd-utils. |
| 2 | * |
Denys Vlasenko | 0ef64bd | 2010-08-16 20:14:46 +0200 | [diff] [blame] | 3 | * Licensed under GPLv2, see file LICENSE in this source tree. |
Baruch Siach | 3324c96 | 2010-06-16 12:22:58 +0200 | [diff] [blame] | 4 | */ |
| 5 | |
Baruch Siach | 3324c96 | 2010-06-16 12:22:58 +0200 | [diff] [blame] | 6 | //config:config UBIATTACH |
| 7 | //config: bool "ubiattach" |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 8 | //config: default y |
Denys Vlasenko | e3b1a1f | 2011-02-26 22:24:08 +0100 | [diff] [blame] | 9 | //config: select PLATFORM_LINUX |
Baruch Siach | 3324c96 | 2010-06-16 12:22:58 +0200 | [diff] [blame] | 10 | //config: help |
| 11 | //config: Attach MTD device to an UBI device. |
| 12 | //config: |
| 13 | //config:config UBIDETACH |
| 14 | //config: bool "ubidetach" |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 15 | //config: default y |
Denys Vlasenko | e3b1a1f | 2011-02-26 22:24:08 +0100 | [diff] [blame] | 16 | //config: select PLATFORM_LINUX |
Baruch Siach | 3324c96 | 2010-06-16 12:22:58 +0200 | [diff] [blame] | 17 | //config: help |
| 18 | //config: Detach MTD device from an UBI device. |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 19 | //config: |
| 20 | //config:config UBIMKVOL |
| 21 | //config: bool "ubimkvol" |
| 22 | //config: default y |
Denys Vlasenko | e3b1a1f | 2011-02-26 22:24:08 +0100 | [diff] [blame] | 23 | //config: select PLATFORM_LINUX |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 24 | //config: help |
| 25 | //config: Create a UBI volume. |
| 26 | //config: |
| 27 | //config:config UBIRMVOL |
| 28 | //config: bool "ubirmvol" |
| 29 | //config: default y |
Denys Vlasenko | e3b1a1f | 2011-02-26 22:24:08 +0100 | [diff] [blame] | 30 | //config: select PLATFORM_LINUX |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 31 | //config: help |
| 32 | //config: Delete a UBI volume. |
| 33 | //config: |
| 34 | //config:config UBIRSVOL |
| 35 | //config: bool "ubirsvol" |
| 36 | //config: default y |
Denys Vlasenko | e3b1a1f | 2011-02-26 22:24:08 +0100 | [diff] [blame] | 37 | //config: select PLATFORM_LINUX |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 38 | //config: help |
| 39 | //config: Resize a UBI volume. |
Reuben Dowle | a610842 | 2011-04-26 04:27:48 +0200 | [diff] [blame] | 40 | //config: |
| 41 | //config:config UBIUPDATEVOL |
| 42 | //config: bool "ubiupdatevol" |
| 43 | //config: default y |
| 44 | //config: select PLATFORM_LINUX |
| 45 | //config: help |
| 46 | //config: Update a UBI volume. |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 47 | |
Denys Vlasenko | 205d48e | 2017-01-29 14:57:33 +0100 | [diff] [blame] | 48 | // APPLET_ODDNAME:name main location suid_type help |
| 49 | //applet:IF_UBIATTACH( APPLET_ODDNAME(ubiattach, ubi_tools, BB_DIR_USR_SBIN, BB_SUID_DROP, ubiattach)) |
| 50 | //applet:IF_UBIDETACH( APPLET_ODDNAME(ubidetach, ubi_tools, BB_DIR_USR_SBIN, BB_SUID_DROP, ubidetach)) |
| 51 | //applet:IF_UBIMKVOL( APPLET_ODDNAME(ubimkvol, ubi_tools, BB_DIR_USR_SBIN, BB_SUID_DROP, ubimkvol)) |
| 52 | //applet:IF_UBIRMVOL( APPLET_ODDNAME(ubirmvol, ubi_tools, BB_DIR_USR_SBIN, BB_SUID_DROP, ubirmvol)) |
| 53 | //applet:IF_UBIRSVOL( APPLET_ODDNAME(ubirsvol, ubi_tools, BB_DIR_USR_SBIN, BB_SUID_DROP, ubirsvol)) |
Reuben Dowle | a610842 | 2011-04-26 04:27:48 +0200 | [diff] [blame] | 54 | //applet:IF_UBIUPDATEVOL(APPLET_ODDNAME(ubiupdatevol, ubi_tools, BB_DIR_USR_SBIN, BB_SUID_DROP, ubiupdatevol)) |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 55 | |
Denys Vlasenko | d1993f1 | 2011-04-26 04:32:05 +0200 | [diff] [blame] | 56 | //kbuild:lib-$(CONFIG_UBIATTACH) += ubi_tools.o |
| 57 | //kbuild:lib-$(CONFIG_UBIDETACH) += ubi_tools.o |
| 58 | //kbuild:lib-$(CONFIG_UBIMKVOL) += ubi_tools.o |
| 59 | //kbuild:lib-$(CONFIG_UBIRMVOL) += ubi_tools.o |
| 60 | //kbuild:lib-$(CONFIG_UBIRSVOL) += ubi_tools.o |
| 61 | //kbuild:lib-$(CONFIG_UBIUPDATEVOL) += ubi_tools.o |
Baruch Siach | 3324c96 | 2010-06-16 12:22:58 +0200 | [diff] [blame] | 62 | |
| 63 | #include "libbb.h" |
Denys Vlasenko | 7fe1e3f | 2012-03-11 18:04:14 +0100 | [diff] [blame] | 64 | /* Some versions of kernel have broken headers, need this hack */ |
| 65 | #ifndef __packed |
| 66 | # define __packed __attribute__((packed)) |
| 67 | #endif |
Baruch Siach | 3324c96 | 2010-06-16 12:22:58 +0200 | [diff] [blame] | 68 | #include <mtd/ubi-user.h> |
| 69 | |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 70 | #define do_attach (ENABLE_UBIATTACH && applet_name[3] == 'a') |
| 71 | #define do_detach (ENABLE_UBIDETACH && applet_name[3] == 'd') |
| 72 | #define do_mkvol (ENABLE_UBIMKVOL && applet_name[3] == 'm') |
| 73 | #define do_rmvol (ENABLE_UBIRMVOL && applet_name[4] == 'm') |
| 74 | #define do_rsvol (ENABLE_UBIRSVOL && applet_name[4] == 's') |
Reuben Dowle | a610842 | 2011-04-26 04:27:48 +0200 | [diff] [blame] | 75 | #define do_update (ENABLE_UBIUPDATEVOL && applet_name[3] == 'u') |
Baruch Siach | 3324c96 | 2010-06-16 12:22:58 +0200 | [diff] [blame] | 76 | |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 77 | static unsigned get_num_from_file(const char *path, unsigned max, const char *errmsg) |
| 78 | { |
| 79 | char buf[sizeof(long long)*3]; |
| 80 | unsigned long long num; |
Denys Vlasenko | 0bfb9c2 | 2011-04-26 04:31:03 +0200 | [diff] [blame] | 81 | |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 82 | if (open_read_close(path, buf, sizeof(buf)) < 0) |
| 83 | bb_perror_msg_and_die(errmsg, path); |
| 84 | /* It can be \n terminated, xatoull won't work well */ |
| 85 | if (sscanf(buf, "%llu", &num) != 1 || num > max) |
| 86 | bb_error_msg_and_die(errmsg, path); |
| 87 | return num; |
| 88 | } |
| 89 | |
| 90 | /* To prevent malloc(1G) accidents */ |
| 91 | #define MAX_SANE_ERASEBLOCK (16*1024*1024) |
Baruch Siach | 3324c96 | 2010-06-16 12:22:58 +0200 | [diff] [blame] | 92 | |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 93 | int ubi_tools_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; |
| 94 | int ubi_tools_main(int argc UNUSED_PARAM, char **argv) |
Baruch Siach | 3324c96 | 2010-06-16 12:22:58 +0200 | [diff] [blame] | 95 | { |
Denys Vlasenko | e9b3fcc | 2013-06-30 02:47:45 +0200 | [diff] [blame] | 96 | static const struct suffix_mult size_suffixes[] = { |
| 97 | { "KiB", 1024 }, |
| 98 | { "MiB", 1024*1024 }, |
| 99 | { "GiB", 1024*1024*1024 }, |
| 100 | { "", 0 } |
| 101 | }; |
| 102 | |
Baruch Siach | 3324c96 | 2010-06-16 12:22:58 +0200 | [diff] [blame] | 103 | unsigned opts; |
| 104 | char *ubi_ctrl; |
Baruch Siach | 3324c96 | 2010-06-16 12:22:58 +0200 | [diff] [blame] | 105 | int fd; |
| 106 | int mtd_num; |
| 107 | int dev_num = UBI_DEV_NUM_AUTO; |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 108 | int vol_id = UBI_VOL_NUM_AUTO; |
Micke Prag | 07fa09a | 2014-11-02 11:08:24 +0100 | [diff] [blame] | 109 | int vid_hdr_offset = 0; |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 110 | char *vol_name; |
| 111 | unsigned long long size_bytes = size_bytes; /* for compiler */ |
| 112 | char *size_bytes_str; |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 113 | int alignment = 1; |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 114 | char *type; |
| 115 | union { |
| 116 | struct ubi_attach_req attach_req; |
| 117 | struct ubi_mkvol_req mkvol_req; |
| 118 | struct ubi_rsvol_req rsvol_req; |
| 119 | } req_structs; |
| 120 | #define attach_req req_structs.attach_req |
| 121 | #define mkvol_req req_structs.mkvol_req |
| 122 | #define rsvol_req req_structs.rsvol_req |
| 123 | char path[sizeof("/sys/class/ubi/ubi%d_%d/usable_eb_size") |
| 124 | + 2 * sizeof(int)*3 + /*just in case:*/ 16]; |
| 125 | #define path_sys_class_ubi_ubi (path + sizeof("/sys/class/ubi/ubi")-1) |
| 126 | |
Paul B. Henson | 985345d | 2013-08-04 21:07:20 +0200 | [diff] [blame] | 127 | strcpy(path, "/sys/class/ubi/ubi"); |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 128 | memset(&req_structs, 0, sizeof(req_structs)); |
Baruch Siach | 3324c96 | 2010-06-16 12:22:58 +0200 | [diff] [blame] | 129 | |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 130 | #define OPTION_m (1 << 0) |
| 131 | #define OPTION_d (1 << 1) |
| 132 | #define OPTION_n (1 << 2) |
| 133 | #define OPTION_N (1 << 3) |
| 134 | #define OPTION_s (1 << 4) |
| 135 | #define OPTION_a (1 << 5) |
| 136 | #define OPTION_t (1 << 6) |
Denys Vlasenko | 186b98a | 2014-09-25 22:10:32 +0200 | [diff] [blame] | 137 | if (do_mkvol) { |
Denys Vlasenko | 237bedd | 2016-07-06 21:58:02 +0200 | [diff] [blame] | 138 | opt_complementary = "-1"; |
| 139 | opts = getopt32(argv, "md:+n:+N:s:a:+t:O:+", |
Denys Vlasenko | 186b98a | 2014-09-25 22:10:32 +0200 | [diff] [blame] | 140 | &dev_num, &vol_id, |
Micke Prag | 07fa09a | 2014-11-02 11:08:24 +0100 | [diff] [blame] | 141 | &vol_name, &size_bytes_str, &alignment, &type, |
| 142 | &vid_hdr_offset |
Denys Vlasenko | 186b98a | 2014-09-25 22:10:32 +0200 | [diff] [blame] | 143 | ); |
| 144 | } else |
| 145 | if (do_update) { |
| 146 | opt_complementary = "-1"; |
| 147 | opts = getopt32(argv, "s:at", &size_bytes_str); |
| 148 | opts *= OPTION_s; |
| 149 | } else { |
Denys Vlasenko | 237bedd | 2016-07-06 21:58:02 +0200 | [diff] [blame] | 150 | opt_complementary = "-1"; |
| 151 | opts = getopt32(argv, "m:+d:+n:+N:s:a:+t:", |
Denys Vlasenko | 186b98a | 2014-09-25 22:10:32 +0200 | [diff] [blame] | 152 | &mtd_num, &dev_num, &vol_id, |
| 153 | &vol_name, &size_bytes_str, &alignment, &type |
| 154 | ); |
| 155 | } |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 156 | |
| 157 | if (opts & OPTION_s) |
Denys Vlasenko | e9b3fcc | 2013-06-30 02:47:45 +0200 | [diff] [blame] | 158 | size_bytes = xatoull_sfx(size_bytes_str, size_suffixes); |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 159 | argv += optind; |
Denys Vlasenko | f798ba9 | 2013-06-30 02:46:44 +0200 | [diff] [blame] | 160 | ubi_ctrl = *argv++; |
Baruch Siach | 3324c96 | 2010-06-16 12:22:58 +0200 | [diff] [blame] | 161 | |
| 162 | fd = xopen(ubi_ctrl, O_RDWR); |
Denys Vlasenko | 8d3e225 | 2010-08-31 12:42:06 +0200 | [diff] [blame] | 163 | //xfstat(fd, &st, ubi_ctrl); |
Baruch Siach | 3324c96 | 2010-06-16 12:22:58 +0200 | [diff] [blame] | 164 | //if (!S_ISCHR(st.st_mode)) |
Denys Vlasenko | 8d3e225 | 2010-08-31 12:42:06 +0200 | [diff] [blame] | 165 | // bb_error_msg_and_die("%s: not a char device", ubi_ctrl); |
Baruch Siach | 3324c96 | 2010-06-16 12:22:58 +0200 | [diff] [blame] | 166 | |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 167 | //usage:#define ubiattach_trivial_usage |
Micke Prag | 07fa09a | 2014-11-02 11:08:24 +0100 | [diff] [blame] | 168 | //usage: "-m MTD_NUM [-d UBI_NUM] [-O VID_HDR_OFF] UBI_CTRL_DEV" |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 169 | //usage:#define ubiattach_full_usage "\n\n" |
| 170 | //usage: "Attach MTD device to UBI\n" |
| 171 | //usage: "\n -m MTD_NUM MTD device number to attach" |
| 172 | //usage: "\n -d UBI_NUM UBI device number to assign" |
Micke Prag | 07fa09a | 2014-11-02 11:08:24 +0100 | [diff] [blame] | 173 | //usage: "\n -O VID_HDR_OFF VID header offset" |
Baruch Siach | 3324c96 | 2010-06-16 12:22:58 +0200 | [diff] [blame] | 174 | if (do_attach) { |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 175 | if (!(opts & OPTION_m)) |
Baruch Siach | 3324c96 | 2010-06-16 12:22:58 +0200 | [diff] [blame] | 176 | bb_error_msg_and_die("%s device not specified", "MTD"); |
| 177 | |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 178 | attach_req.mtd_num = mtd_num; |
| 179 | attach_req.ubi_num = dev_num; |
Micke Prag | 07fa09a | 2014-11-02 11:08:24 +0100 | [diff] [blame] | 180 | attach_req.vid_hdr_offset = vid_hdr_offset; |
Baruch Siach | 3324c96 | 2010-06-16 12:22:58 +0200 | [diff] [blame] | 181 | |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 182 | xioctl(fd, UBI_IOCATT, &attach_req); |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 183 | } else |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 184 | |
| 185 | //usage:#define ubidetach_trivial_usage |
| 186 | //usage: "-d UBI_NUM UBI_CTRL_DEV" |
| 187 | //usage:#define ubidetach_full_usage "\n\n" |
| 188 | //usage: "Detach MTD device from UBI\n" |
| 189 | //usage: "\n -d UBI_NUM UBI device number" |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 190 | if (do_detach) { |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 191 | if (!(opts & OPTION_d)) |
Baruch Siach | 3324c96 | 2010-06-16 12:22:58 +0200 | [diff] [blame] | 192 | bb_error_msg_and_die("%s device not specified", "UBI"); |
| 193 | |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 194 | /* FIXME? kernel expects int32_t* here: */ |
Baruch Siach | 3324c96 | 2010-06-16 12:22:58 +0200 | [diff] [blame] | 195 | xioctl(fd, UBI_IOCDET, &dev_num); |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 196 | } else |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 197 | |
| 198 | //usage:#define ubimkvol_trivial_usage |
Sven Eisenberg | b068cf2 | 2016-04-03 21:53:12 +0200 | [diff] [blame] | 199 | //usage: "-N NAME [-s SIZE | -m] UBI_DEVICE" |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 200 | //usage:#define ubimkvol_full_usage "\n\n" |
| 201 | //usage: "Create UBI volume\n" |
| 202 | //usage: "\n -a ALIGNMENT Volume alignment (default 1)" |
| 203 | //usage: "\n -m Set volume size to maximum available" |
| 204 | //usage: "\n -n VOLID Volume ID. If not specified," |
| 205 | //usage: "\n assigned automatically" |
| 206 | //usage: "\n -N NAME Volume name" |
| 207 | //usage: "\n -s SIZE Size in bytes" |
| 208 | //usage: "\n -t TYPE Volume type (static|dynamic)" |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 209 | if (do_mkvol) { |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 210 | if (opts & OPTION_m) { |
Paul B. Henson | f86404f | 2013-06-02 17:48:05 -0700 | [diff] [blame] | 211 | unsigned leb_avail; |
| 212 | unsigned leb_size; |
| 213 | unsigned num; |
Paul B. Henson | f86404f | 2013-06-02 17:48:05 -0700 | [diff] [blame] | 214 | char *p; |
Paul B. Henson | f86404f | 2013-06-02 17:48:05 -0700 | [diff] [blame] | 215 | |
Sven Eisenberg | b068cf2 | 2016-04-03 21:53:12 +0200 | [diff] [blame] | 216 | num = ubi_devnum_from_devname(ubi_ctrl); |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 217 | p = path_sys_class_ubi_ubi + sprintf(path_sys_class_ubi_ubi, "%u/", num); |
Paul B. Henson | f86404f | 2013-06-02 17:48:05 -0700 | [diff] [blame] | 218 | |
| 219 | strcpy(p, "avail_eraseblocks"); |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 220 | leb_avail = get_num_from_file(path, UINT_MAX, "Can't get available eraseblocks from '%s'"); |
Paul B. Henson | f86404f | 2013-06-02 17:48:05 -0700 | [diff] [blame] | 221 | |
| 222 | strcpy(p, "eraseblock_size"); |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 223 | leb_size = get_num_from_file(path, MAX_SANE_ERASEBLOCK, "Can't get eraseblock size from '%s'"); |
Paul B. Henson | f86404f | 2013-06-02 17:48:05 -0700 | [diff] [blame] | 224 | |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 225 | size_bytes = leb_avail * (unsigned long long)leb_size; |
| 226 | //if (size_bytes <= 0) |
| 227 | // bb_error_msg_and_die("%s invalid maximum size calculated", "UBI"); |
Paul B. Henson | f86404f | 2013-06-02 17:48:05 -0700 | [diff] [blame] | 228 | } else |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 229 | if (!(opts & OPTION_s)) |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 230 | bb_error_msg_and_die("size not specified"); |
| 231 | |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 232 | if (!(opts & OPTION_N)) |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 233 | bb_error_msg_and_die("name not specified"); |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 234 | |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 235 | mkvol_req.vol_id = vol_id; |
| 236 | mkvol_req.vol_type = UBI_DYNAMIC_VOLUME; |
| 237 | if ((opts & OPTION_t) && type[0] == 's') |
| 238 | mkvol_req.vol_type = UBI_STATIC_VOLUME; |
| 239 | mkvol_req.alignment = alignment; |
| 240 | mkvol_req.bytes = size_bytes; /* signed int64_t */ |
| 241 | strncpy(mkvol_req.name, vol_name, UBI_MAX_VOLUME_NAME); |
| 242 | mkvol_req.name_len = strlen(vol_name); |
| 243 | if (mkvol_req.name_len > UBI_MAX_VOLUME_NAME) |
| 244 | bb_error_msg_and_die("volume name too long: '%s'", vol_name); |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 245 | |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 246 | xioctl(fd, UBI_IOCMKVOL, &mkvol_req); |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 247 | } else |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 248 | |
| 249 | //usage:#define ubirmvol_trivial_usage |
Sven Eisenberg | b068cf2 | 2016-04-03 21:53:12 +0200 | [diff] [blame] | 250 | //usage: "-n VOLID / -N VOLNAME UBI_DEVICE" |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 251 | //usage:#define ubirmvol_full_usage "\n\n" |
| 252 | //usage: "Remove UBI volume\n" |
| 253 | //usage: "\n -n VOLID Volume ID" |
Sven Eisenberg | b068cf2 | 2016-04-03 21:53:12 +0200 | [diff] [blame] | 254 | //usage: "\n -N VOLNAME Volume name" |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 255 | if (do_rmvol) { |
Sven Eisenberg | b068cf2 | 2016-04-03 21:53:12 +0200 | [diff] [blame] | 256 | if (!(opts & (OPTION_n|OPTION_N))) |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 257 | bb_error_msg_and_die("volume id not specified"); |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 258 | |
Sven Eisenberg | b068cf2 | 2016-04-03 21:53:12 +0200 | [diff] [blame] | 259 | if (opts & OPTION_N) { |
| 260 | unsigned num = ubi_devnum_from_devname(ubi_ctrl); |
Denys Vlasenko | 6aab992 | 2016-04-03 22:24:51 +0200 | [diff] [blame] | 261 | vol_id = ubi_get_volid_by_name(num, vol_name); |
Sven Eisenberg | b068cf2 | 2016-04-03 21:53:12 +0200 | [diff] [blame] | 262 | } |
| 263 | |
| 264 | if (sizeof(vol_id) != 4) { |
| 265 | /* kernel expects int32_t* in this ioctl */ |
| 266 | int32_t t = vol_id; |
| 267 | xioctl(fd, UBI_IOCRMVOL, &t); |
| 268 | } else { |
| 269 | xioctl(fd, UBI_IOCRMVOL, &vol_id); |
| 270 | } |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 271 | } else |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 272 | |
| 273 | //usage:#define ubirsvol_trivial_usage |
Sven Eisenberg | b068cf2 | 2016-04-03 21:53:12 +0200 | [diff] [blame] | 274 | //usage: "-n VOLID -s SIZE UBI_DEVICE" |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 275 | //usage:#define ubirsvol_full_usage "\n\n" |
| 276 | //usage: "Resize UBI volume\n" |
| 277 | //usage: "\n -n VOLID Volume ID" |
| 278 | //usage: "\n -s SIZE Size in bytes" |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 279 | if (do_rsvol) { |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 280 | if (!(opts & OPTION_s)) |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 281 | bb_error_msg_and_die("size not specified"); |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 282 | if (!(opts & OPTION_n)) |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 283 | bb_error_msg_and_die("volume id not specified"); |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 284 | |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 285 | rsvol_req.bytes = size_bytes; /* signed int64_t */ |
| 286 | rsvol_req.vol_id = vol_id; |
Reuben Dowle | 3e2834f | 2011-02-05 03:18:08 +0100 | [diff] [blame] | 287 | |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 288 | xioctl(fd, UBI_IOCRSVOL, &rsvol_req); |
Reuben Dowle | a610842 | 2011-04-26 04:27:48 +0200 | [diff] [blame] | 289 | } else |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 290 | |
| 291 | //usage:#define ubiupdatevol_trivial_usage |
Sven Eisenberg | b068cf2 | 2016-04-03 21:53:12 +0200 | [diff] [blame] | 292 | //usage: "[-t | [-s SIZE] IMG_FILE] UBI_DEVICE" |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 293 | //usage:#define ubiupdatevol_full_usage "\n\n" |
| 294 | //usage: "Update UBI volume\n" |
| 295 | //usage: "\n -t Truncate to zero size" |
| 296 | //usage: "\n -s SIZE Size in bytes to resize to" |
Reuben Dowle | a610842 | 2011-04-26 04:27:48 +0200 | [diff] [blame] | 297 | if (do_update) { |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 298 | int64_t bytes64; |
Reuben Dowle | a610842 | 2011-04-26 04:27:48 +0200 | [diff] [blame] | 299 | |
| 300 | if (opts & OPTION_t) { |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 301 | /* truncate the volume by starting an update for size 0 */ |
| 302 | bytes64 = 0; |
| 303 | /* this ioctl expects int64_t* parameter */ |
| 304 | xioctl(fd, UBI_IOCVOLUP, &bytes64); |
Reuben Dowle | a610842 | 2011-04-26 04:27:48 +0200 | [diff] [blame] | 305 | } |
| 306 | else { |
| 307 | struct stat st; |
Reuben Dowle | a610842 | 2011-04-26 04:27:48 +0200 | [diff] [blame] | 308 | unsigned ubinum, volnum; |
| 309 | unsigned leb_size; |
| 310 | ssize_t len; |
| 311 | char *input_data; |
| 312 | |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 313 | /* Assume that device is in normal format. */ |
| 314 | /* Removes need for scanning sysfs tree as full libubi does. */ |
Reuben Dowle | a610842 | 2011-04-26 04:27:48 +0200 | [diff] [blame] | 315 | if (sscanf(ubi_ctrl, "/dev/ubi%u_%u", &ubinum, &volnum) != 2) |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 316 | bb_error_msg_and_die("wrong format of UBI device name"); |
Reuben Dowle | a610842 | 2011-04-26 04:27:48 +0200 | [diff] [blame] | 317 | |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 318 | sprintf(path_sys_class_ubi_ubi, "%u_%u/usable_eb_size", ubinum, volnum); |
| 319 | leb_size = get_num_from_file(path, MAX_SANE_ERASEBLOCK, "Can't get usable eraseblock size from '%s'"); |
Reuben Dowle | a610842 | 2011-04-26 04:27:48 +0200 | [diff] [blame] | 320 | |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 321 | if (!(opts & OPTION_s)) { |
| 322 | if (!*argv) |
Reuben Dowle | a610842 | 2011-04-26 04:27:48 +0200 | [diff] [blame] | 323 | bb_show_usage(); |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 324 | xmove_fd(xopen(*argv, O_RDONLY), STDIN_FILENO); |
Denys Vlasenko | 186b98a | 2014-09-25 22:10:32 +0200 | [diff] [blame] | 325 | xfstat(STDIN_FILENO, &st, *argv); |
| 326 | size_bytes = st.st_size; |
Reuben Dowle | a610842 | 2011-04-26 04:27:48 +0200 | [diff] [blame] | 327 | } |
| 328 | |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 329 | bytes64 = size_bytes; |
| 330 | /* this ioctl expects signed int64_t* parameter */ |
| 331 | xioctl(fd, UBI_IOCVOLUP, &bytes64); |
Reuben Dowle | a610842 | 2011-04-26 04:27:48 +0200 | [diff] [blame] | 332 | |
| 333 | input_data = xmalloc(leb_size); |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 334 | while ((len = full_read(STDIN_FILENO, input_data, leb_size)) > 0) { |
Reuben Dowle | a610842 | 2011-04-26 04:27:48 +0200 | [diff] [blame] | 335 | xwrite(fd, input_data, len); |
| 336 | } |
| 337 | if (len < 0) |
Denys Vlasenko | 65bdf0b | 2013-06-30 02:38:18 +0200 | [diff] [blame] | 338 | bb_perror_msg_and_die("UBI volume update failed"); |
Reuben Dowle | a610842 | 2011-04-26 04:27:48 +0200 | [diff] [blame] | 339 | } |
Baruch Siach | 3324c96 | 2010-06-16 12:22:58 +0200 | [diff] [blame] | 340 | } |
| 341 | |
| 342 | if (ENABLE_FEATURE_CLEAN_UP) |
| 343 | close(fd); |
| 344 | |
| 345 | return EXIT_SUCCESS; |
| 346 | } |