bc: use only G_interrupt as interrupt flag
[oweals/busybox.git] / miscutils / ubirename.c
1 /* ubirename - port of the ubirename from the mtd-utils package
2  *
3  * A utility to rename one UBI volume.
4  *
5  * 2016-03-01 Sven Eisenberg <sven.eisenberg@novero.com>
6  *
7  * Licensed under GPLv2, see file LICENSE in this source tree.
8  */
9 //config:config UBIRENAME
10 //config:       bool "ubirename (2.2 kb)"
11 //config:       default y
12 //config:       select PLATFORM_LINUX
13 //config:       help
14 //config:       Utility to rename UBI volumes
15
16 //applet:IF_UBIRENAME(APPLET(ubirename, BB_DIR_USR_SBIN, BB_SUID_DROP))
17 /* not NOEXEC: if flash operation stalls, use less memory in "hung" process */
18
19 //kbuild:lib-$(CONFIG_UBIRENAME) += ubirename.o
20
21 //usage:#define ubirename_trivial_usage
22 //usage:        "UBI_DEVICE OLD_VOLNAME NEW_VOLNAME [OLD2 NEW2]..."
23 //usage:#define ubirename_full_usage "\n\n"
24 //usage:        "Rename UBI volumes on UBI_DEVICE"
25
26 #include "libbb.h"
27 #include <mtd/mtd-user.h>
28
29 #ifndef __packed
30 # define __packed __attribute__((packed))
31 #endif
32
33 // from ubi-media.h
34 #define UBI_MAX_VOLUME_NAME 127
35 #define UBI_MAX_VOLUMES     128
36 // end ubi-media.h
37
38 // from ubi-user.h
39 /* ioctl commands of UBI character devices */
40 #define UBI_IOC_MAGIC 'o'
41
42 /* Re-name volumes */
43 #define UBI_IOCRNVOL _IOW(UBI_IOC_MAGIC, 3, struct ubi_rnvol_req)
44
45 /* Maximum amount of UBI volumes that can be re-named at one go */
46 #define UBI_MAX_RNVOL 32
47
48 struct ubi_rnvol_req {
49         int32_t count;
50         int8_t  padding1[12];
51         struct {
52                 int32_t vol_id;
53                 int16_t name_len;
54                 int8_t  padding2[2];
55                 char    name[UBI_MAX_VOLUME_NAME + 1];
56         } ents[UBI_MAX_RNVOL];
57 } __packed;
58 // end ubi-user.h
59
60 int ubirename_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
61 int ubirename_main(int argc, char **argv)
62 {
63         struct ubi_rnvol_req *rnvol;
64         const char *ubi_devname;
65         unsigned ubi_devnum;
66         unsigned n;
67
68         /* argc can be 4, 6, 8, ... */
69         if ((argc & 1) || (argc >>= 1) < 2)
70                 bb_show_usage();
71
72         rnvol = xzalloc(sizeof(*rnvol));
73         rnvol->count = --argc;
74         if (argc > ARRAY_SIZE(rnvol->ents))
75                 bb_error_msg_and_die("too many renames requested");
76
77         ubi_devname = argv[1];
78         ubi_devnum = ubi_devnum_from_devname(ubi_devname);
79
80         n = 0;
81         argv += 2;
82         while (argv[0]) {
83                 rnvol->ents[n].vol_id = ubi_get_volid_by_name(ubi_devnum, argv[0]);
84
85                 /* strnlen avoids overflow of 16-bit field (paranoia) */
86                 rnvol->ents[n].name_len = strnlen(argv[1], sizeof(rnvol->ents[n].name));
87                 if (rnvol->ents[n].name_len >= sizeof(rnvol->ents[n].name))
88                         bb_error_msg_and_die("new name '%s' is too long", argv[1]);
89
90                 strcpy(rnvol->ents[n].name, argv[1]);
91                 n++;
92                 argv += 2;
93         }
94
95         xioctl(xopen(ubi_devname, O_RDONLY), UBI_IOCRNVOL, rnvol);
96
97         return 0;
98 }