diff options
author | Harvey Wu <harveywu95@gmail.com> | 2021-06-01 15:34:53 +0800 |
---|---|---|
committer | David Oberhollenzer <david.oberhollenzer@sigma-star.at> | 2021-06-12 17:17:15 +0200 |
commit | a456ebc21bc64b5ef273e186fcff191563799a83 (patch) | |
tree | 175e249212f6a9d634cfc8e8a7b11eb666221b86 | |
parent | 51d6820f49ca4a203cf30dd99bc83f25d569eb4b (diff) |
misc-utils: flashcp: Add new function that copy only different blocks
- The original flashcp process is erase, write and verify all blocks
in one time from file to device.
This patch will add a function that only copy different block data
from file to device. The function will compare blocks by block
between file and device, then erase and write block data from file
to device if found different block.
Signed-off-by: Harvey Wu <harveywu95@gmail.com>
Signed-off-by: David Oberhollenzer <david.oberhollenzer@sigma-star.at>
-rw-r--r-- | misc-utils/flashcp.c | 93 |
1 files changed, 92 insertions, 1 deletions
diff --git a/misc-utils/flashcp.c b/misc-utils/flashcp.c index d7b0a59..341c210 100644 --- a/misc-utils/flashcp.c +++ b/misc-utils/flashcp.c @@ -67,6 +67,7 @@ #define FLAG_FILENAME 0x04 #define FLAG_DEVICE 0x08 #define FLAG_ERASE_ALL 0x10 +#define FLAG_PARTITION 0x20 /* error levels */ #define LOG_NORMAL 1 @@ -96,6 +97,7 @@ static NORETURN void showusage(bool error) "\n" " -h | --help Show this help message\n" " -v | --verbose Show progress reports\n" + " -p | --partition Only copy different block from file to device\n" " -A | --erase-all Erases the whole device regardless of the image size\n" " -V | --version Show version information and exit\n" " <filename> File which you want to copy to flash\n" @@ -181,10 +183,11 @@ int main (int argc,char *argv[]) for (;;) { int option_index = 0; - static const char *short_options = "hvAV"; + static const char *short_options = "hvpAV"; static const struct option long_options[] = { {"help", no_argument, 0, 'h'}, {"verbose", no_argument, 0, 'v'}, + {"partition", no_argument, 0, 'p'}, {"erase-all", no_argument, 0, 'A'}, {"version", no_argument, 0, 'V'}, {0, 0, 0, 0}, @@ -205,6 +208,10 @@ int main (int argc,char *argv[]) flags |= FLAG_VERBOSE; DEBUG("Got FLAG_VERBOSE\n"); break; + case 'p': + flags |= FLAG_PARTITION; + DEBUG("Got FLAG_PARTITION"); + break; case 'A': flags |= FLAG_ERASE_ALL; DEBUG("Got FLAG_ERASE_ALL\n"); @@ -257,6 +264,12 @@ int main (int argc,char *argv[]) exit (EXIT_FAILURE); } + /* diff block flashcp */ + if (flags & FLAG_PARTITION) + { + goto DIFF_BLOCKS; + } + /***************************************************** * erase enough blocks so that we can write the file * *****************************************************/ @@ -403,4 +416,82 @@ int main (int argc,char *argv[]) DEBUG("Verified %d / %lluk bytes\n",written,(unsigned long long)filestat.st_size); exit (EXIT_SUCCESS); + + /********************************************* + * Copy different blocks from file to device * + ********************************************/ +DIFF_BLOCKS: + safe_rewind (fil_fd,filename); + safe_rewind (dev_fd,device); + size = filestat.st_size; + i = mtd.erasesize; + erase.start = 0; + erase.length = (filestat.st_size + mtd.erasesize - 1) / mtd.erasesize; + erase.length *= mtd.erasesize; + written = 0; + unsigned long current_dev_block = 0; + int diffBlock = 0; + int blocks = erase.length / mtd.erasesize; + erase.length = mtd.erasesize; + + if (flags & FLAG_VERBOSE) + log_printf (LOG_NORMAL, + "\rProcessing blocks: 0/%d (%d%%)", blocks, PERCENTAGE (0,blocks)); + for (int s = 1; s <= blocks; s++) + { + if (size < mtd.erasesize) i = size; + if (flags & FLAG_VERBOSE) + log_printf (LOG_NORMAL, + "\rProcessing blocks: %d/%d (%d%%)", s, blocks, PERCENTAGE (s,blocks)); + + /* read from filename */ + safe_read (fil_fd,filename,src,i,flags & FLAG_VERBOSE); + + /* read from device */ + current_dev_block = lseek(dev_fd, 0, SEEK_CUR); + safe_read (dev_fd,device,dest,i,flags & FLAG_VERBOSE); + + /* compare buffers, if not the same, erase and write the block */ + if (memcmp (src,dest,i)) + { + diffBlock++; + /* erase block */ + lseek(dev_fd, current_dev_block, SEEK_SET); + if (ioctl (dev_fd,MEMERASE,&erase) < 0) + { + log_printf (LOG_NORMAL,"\n"); + log_printf (LOG_ERROR, + "While erasing blocks 0x%.8x-0x%.8x on %s: %m\n", + (unsigned int) erase.start,(unsigned int) (erase.start + erase.length),device); + exit (EXIT_FAILURE); + } + + /* write to device */ + lseek(dev_fd, current_dev_block, SEEK_SET); + result = write (dev_fd,src,i); + if (i != result) + { + if (flags & FLAG_VERBOSE) log_printf (LOG_NORMAL,"\n"); + if (result < 0) + { + log_printf (LOG_ERROR, + "While writing data to 0x%.8lx-0x%.8lx on %s: %m\n", + written,written + i,device); + exit (EXIT_FAILURE); + } + log_printf (LOG_ERROR, + "Short write count returned while writing to x%.8zx-0x%.8zx on %s: %zu/%llu bytes written to flash\n", + written,written + i,device,written + result,(unsigned long long)filestat.st_size); + exit (EXIT_FAILURE); + } + } + + erase.start += i; + written += i; + size -= i; + } + + if (flags & FLAG_VERBOSE) log_printf (LOG_NORMAL, "\ndiff blocks: %d\n", diffBlock); + + exit (EXIT_SUCCESS); } |