M7350/filesystems/mtd-utils/flash_otp_write.c

123 lines
2.4 KiB
C
Raw Permalink Normal View History

2024-09-09 08:57:42 +00:00
/*
* flash_otp_write.c -- write One-Time-Program data
*/
#define PROGRAM_NAME "flash_otp_write"
#include <stdio.h>
#include <unistd.h>
#include <fcntl.h>
#include <string.h>
#include <stdlib.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/ioctl.h>
#include <common.h>
#include <mtd/mtd-user.h>
ssize_t xread(int fd, void *buf, size_t count)
{
ssize_t ret, done = 0;
retry:
ret = read(fd, buf + done, count - done);
if (ret < 0)
return ret;
done += ret;
if (ret == 0 /* EOF */ || done == count)
return done;
else
goto retry;
}
int main(int argc,char *argv[])
{
int fd, val, ret, size, wrote, len;
mtd_info_t mtdInfo;
off_t offset;
char *p, buf[2048];
if (argc != 4 || strcmp(argv[1], "-u")) {
fprintf(stderr, "Usage: %s -u <device> <offset>\n", PROGRAM_NAME);
fprintf(stderr, "the raw data to write should be provided on stdin\n");
fprintf(stderr, "CAUTION! ONCE SET TO 0, OTP DATA BITS CAN'T BE ERASED!\n");
return EINVAL;
}
fd = open(argv[2], O_WRONLY);
if (fd < 0) {
perror(argv[2]);
return errno;
}
val = MTD_OTP_USER;
ret = ioctl(fd, OTPSELECT, &val);
if (ret < 0) {
perror("OTPSELECT");
return errno;
}
if (ioctl(fd, MEMGETINFO, &mtdInfo)) {
perror("MEMGETINFO");
return errno;
}
offset = (off_t)strtoull(argv[3], &p, 0);
if (argv[3][0] == 0 || *p != 0) {
fprintf(stderr, "%s: bad offset value\n", PROGRAM_NAME);
return ERANGE;
}
if (lseek(fd, offset, SEEK_SET) == (off_t)-1) {
perror("lseek()");
return errno;
}
printf("Writing OTP user data on %s at offset 0x%"PRIxoff_t"\n", argv[2], offset);
if (mtd_type_is_nand_user(&mtdInfo))
len = mtdInfo.writesize;
else
len = 256;
if (len > sizeof(buf)) {
printf("huh, writesize (%d) bigger than buffer (%zu)\n",
len, sizeof(buf));
return ENOMEM;
}
wrote = 0;
while ((size = xread(0, buf, len))) {
if (size < 0) {
perror("read()");
return errno;
}
p = buf;
while (size > 0) {
if (mtd_type_is_nand_user(&mtdInfo)) {
/* Fill remain buffers with 0xff */
memset(buf + size, 0xff, mtdInfo.writesize - size);
size = mtdInfo.writesize;
}
ret = write(fd, p, size);
if (ret < 0) {
perror("write()");
return errno;
}
if (ret == 0) {
printf("write() returned 0 after writing %d bytes\n", wrote);
return 0;
}
p += ret;
wrote += ret;
size -= ret;
}
}
printf("Wrote %d bytes of OTP user data\n", wrote);
return 0;
}