static int do_memset(int argc, char *argv[])
{
loff_t s, c, n;
int fd;
char *buf;
int mode = O_RWSIZE_1;
int ret = 1;
char *file = "/dev/mem";
if (mem_parse_options(argc, argv, "bwlqd:", &mode, NULL, &file,
NULL) < 0)
return 1;
if (optind + 3 > argc)
return COMMAND_ERROR_USAGE;
s = strtoull_suffix(argv[optind], NULL, 0);
c = strtoull_suffix(argv[optind + 1], NULL, 0);
n = strtoull_suffix(argv[optind + 2], NULL, 0);
fd = open_and_lseek(file, mode | O_WRONLY, s);
if (fd < 0)
return 1;
buf = xmalloc(RW_BUF_SIZE);
memset(buf, c, RW_BUF_SIZE);
while (n > 0) {
int now;
now = min((loff_t)RW_BUF_SIZE, n);
ret = write(fd, buf, now);
if (ret < 0) {
perror("write");
ret = 1;
goto out;
}
n -= now;
}
ret = 0;
out:
close(fd);
free(buf);
return ret;
}
static int do_ubimkvol(int argc, char *argv[])
{
struct ubi_mkvol_req req;
int fd, ret;
uint64_t size;
if (argc != 4)
return COMMAND_ERROR_USAGE;
size = strtoull_suffix(argv[3], NULL, 0);
req.name_len = min_t(int, strlen(argv[2]), UBI_VOL_NAME_MAX);
strncpy(req.name, argv[2], req.name_len);
req.name[req.name_len] = 0;
req.vol_type = UBI_DYNAMIC_VOLUME;
req.bytes = size;
req.vol_id = UBI_VOL_NUM_AUTO;
req.alignment = 1;
fd = open(argv[1], O_WRONLY);
if (fd < 0) {
perror("open");
return 1;
}
ret = ioctl(fd, UBI_IOCMKVOL, &req);
if (ret)
printf("failed to create: %s\n", strerror(-ret));
close(fd);
return ret ? 1 : 0;
}
void parseopt_llu_suffix(const char *options, const char *opt,
unsigned long long *val)
{
const char *start;
size_t optlen = strlen(opt);
unsigned long long v;
char *endp;
again:
start = strstr(options, opt);
if (!start)
return;
if (start > options && start[-1] != ',') {
options = start;
goto again;
}
if (start[optlen] != '=') {
options = start;
goto again;
}
v = strtoull_suffix(start + optlen + 1, &endp, 0);
if (*endp == ',' || *endp == '\0')
*val = v;
}
BAREBOX_CMD_END
static int do_ubimkvol(int argc, char *argv[])
{
int opt;
struct ubi_mkvol_req req;
int fd, ret;
uint64_t size;
uint32_t ubinum;
req.vol_type = UBI_DYNAMIC_VOLUME;
while ((opt = getopt(argc, argv, "t:")) > 0) {
switch (opt) {
case 't':
if (!strcmp(optarg, "dynamic"))
req.vol_type = UBI_DYNAMIC_VOLUME;
else if (!strcmp(optarg, "static"))
req.vol_type = UBI_STATIC_VOLUME;
else
return COMMAND_ERROR_USAGE;
break;
default:
return COMMAND_ERROR_USAGE;
}
}
if (argc - optind != 3)
return COMMAND_ERROR_USAGE;
size = strtoull_suffix(argv[optind+2], NULL, 0);
req.name_len = min_t(int, strlen(argv[optind+1]), UBI_VOL_NAME_MAX);
strncpy(req.name, argv[optind+1], req.name_len);
req.name[req.name_len] = 0;
req.bytes = size;
req.vol_id = UBI_VOL_NUM_AUTO;
req.alignment = 1;
fd = open(argv[optind], O_WRONLY);
if (fd < 0) {
perror("open");
return 1;
}
ret = ioctl(fd, UBI_IOCGETUBINUM, &ubinum);
if (ret) {
printf("failed to get ubinum: %s\n", strerror(-ret));
goto err;
}
ret = ubi_api_create_volume(ubinum, &req);
err:
close(fd);
return ret ? 1 : 0;
}
static int parse_opt(int argc, char *argv[])
{
srand(get_time_ns());
memset(&args, 0, sizeof(args));
args.ubi_ver = 1;
args.image_seq = rand();
while (1) {
int key;
unsigned long int image_seq;
key = getopt(argc, argv, "nyyqve:x:s:O:f:S:");
if (key == -1)
break;
switch (key) {
case 's':
args.subpage_size = strtoull_suffix(optarg, NULL, 0);
if (args.subpage_size <= 0)
return errmsg("bad sub-page size: \"%s\"", optarg);
if (!is_power_of_2(args.subpage_size))
return errmsg("sub-page size should be power of 2");
break;
case 'O':
args.vid_hdr_offs = simple_strtoul(optarg, NULL, 0);
if (args.vid_hdr_offs <= 0)
return errmsg("bad VID header offset: \"%s\"", optarg);
break;
case 'e':
args.ec = simple_strtoull(optarg, NULL, 0);
if (args.ec < 0)
return errmsg("bad erase counter value: \"%s\"", optarg);
if (args.ec >= EC_MAX)
return errmsg("too high erase %llu, counter, max is %u", args.ec, EC_MAX);
args.override_ec = 1;
break;
case 'f':
args.image = optarg;
break;
case 'n':
args.novtbl = 1;
break;
case 'y':
args.yes = 1;
break;
case 'q':
args.quiet = 1;
break;
case 'x':
args.ubi_ver = simple_strtoul(optarg, NULL, 0);
if (args.ubi_ver < 0)
return errmsg("bad UBI version: \"%s\"", optarg);
break;
case 'Q':
image_seq = simple_strtoul(optarg, NULL, 0);
if (image_seq > 0xFFFFFFFF)
return errmsg("bad UBI image sequence number: \"%s\"", optarg);
args.image_seq = image_seq;
break;
case 'v':
args.verbose = 1;
break;
default:
return COMMAND_ERROR_USAGE;
}
}
if (args.quiet && args.verbose)
return errmsg("using \"-q\" and \"-v\" at the same time does not make sense");
if (optind == argc)
return errmsg("MTD device name was not specified");
else if (optind != argc - 1)
return errmsg("more then one MTD device specified");
if (args.image && args.novtbl)
return errmsg("-n cannot be used together with -f");
args.node = argv[optind];
return 0;
}
static int mtd_part_do_parse_one(char *devname, const char *partstr,
char **endp, loff_t *offset,
loff_t devsize, size_t *retsize,
unsigned int pflags)
{
loff_t size;
char *end;
char buf[PATH_MAX] = {};
unsigned long flags = 0;
int ret = 0;
struct cdev *cdev;
memset(buf, 0, PATH_MAX);
if (*partstr == '-') {
size = SIZE_REMAINING;
end = (char *)partstr + 1;
} else {
size = strtoull_suffix(partstr, &end, 0);
}
if (*end == '@')
*offset = strtoull_suffix(end+1, &end, 0);
if (size == SIZE_REMAINING)
size = devsize - *offset;
partstr = end;
if (*partstr == '(') {
partstr++;
end = strchr((char *) partstr, ')');
if (!end) {
printf("could not find matching ')'\n");
return -EINVAL;
}
if (pflags & PART_ADD_DEVNAME)
sprintf(buf, "%s.", devname);
memcpy(buf + strlen(buf), partstr, end - partstr);
end++;
}
if (size + *offset > devsize) {
printf("%s: partition end is beyond device\n", buf);
return -EINVAL;
}
partstr = end;
if (*partstr == 'r' && *(partstr + 1) == 'o') {
flags |= DEVFS_PARTITION_READONLY;
end = (char *)(partstr + 2);
}
if (endp)
*endp = end;
*retsize = size;
cdev = devfs_add_partition(devname, *offset, size, flags, buf);
if (IS_ERR(cdev)) {
ret = PTR_ERR(cdev);
printf("cannot create %s: %s\n", buf, strerror(-ret));
}
return ret;
}
static int do_mem_mm(int argc, char *argv[])
{
int ret = 0;
int fd;
char *filename = "/dev/mem";
int mode = O_RWSIZE_4;
loff_t adr;
int swab = 0;
u8 val8;
u16 val16;
u32 val32;
u64 val64;
u64 value, mask;
if (mem_parse_options(argc, argv, "bwlqd:", &mode, NULL, &filename,
&swab) < 0)
return 1;
if (optind + 2 >= argc)
return COMMAND_ERROR_USAGE;
adr = strtoull_suffix(argv[optind++], NULL, 0);
value = simple_strtoull(argv[optind++], NULL, 0);
mask = simple_strtoull(argv[optind++], NULL, 0);
fd = open_and_lseek(filename, mode | O_RDWR, adr);
if (fd < 0)
return 1;
switch (mode) {
case O_RWSIZE_1:
ret = pread(fd, &val8, 1, adr);
if (ret < 0)
goto out_read;
val8 &= ~mask;
val8 |= (value & mask);
ret = pwrite(fd, &val8, 1, adr);
if (ret < 0)
goto out_write;
break;
case O_RWSIZE_2:
ret = pread(fd, &val16, 2, adr);
if (ret < 0)
goto out_read;
val16 &= ~mask;
val16 |= (value & mask);
ret = pwrite(fd, &val16, 2, adr);
if (ret < 0)
goto out_write;
break;
case O_RWSIZE_4:
if (ret < 0)
goto out_read;
ret = pread(fd, &val32, 4, adr);
val32 &= ~mask;
val32 |= (value & mask);
ret = pwrite(fd, &val32, 4, adr);
if (ret < 0)
goto out_write;
break;
case O_RWSIZE_8:
if (ret < 0)
goto out_read;
ret = pread(fd, &val64, 8, adr);
val64 &= ~mask;
val64 |= (value & mask);
ret = pwrite(fd, &val64, 8, adr);
if (ret < 0)
goto out_write;
break;
}
close(fd);
return 0;
out_read:
perror("read");
close(fd);
return 1;
out_write:
perror("write");
close(fd);
return 1;
}
