Read in the source image completely before starting to write the output
image. This makes it possible to pass the same file as input and output.
Signed-off-by: Sascha Hauer <s.hauer@xxxxxxxxxxxxxx>
---
scripts/imx/imx-image.c | 46 +++++++++++++++++++++++-----------------------
1 file changed, 23 insertions(+), 23 deletions(-)
diff --git a/scripts/imx/imx-image.c b/scripts/imx/imx-image.c
index a588882..c13e059 100644
--- a/scripts/imx/imx-image.c
+++ b/scripts/imx/imx-image.c
@@ -666,11 +666,11 @@ int main(int argc, char *argv[])
char *imagename = NULL;
char *outfile = NULL;
void *buf;
- size_t image_size = 0, load_size;
+ size_t image_size = 0, load_size, insize;
+ void *infile;
struct stat s;
int infd, outfd;
int dcd_only = 0;
- int now = 0;
while ((opt = getopt(argc, argv, "c:hf:o:bdp")) != -1) {
switch (opt) {
@@ -779,6 +779,24 @@ int main(int argc, char *argv[])
exit(1);
}
+ infd = open(imagename, O_RDONLY);
+ if (infd < 0) {
+ perror("open");
+ exit(1);
+ }
+
+ ret = fstat(infd, &s);
+ if (ret)
+ return ret;
+
+ insize = s.st_size;
+ infile = malloc(insize);
+ if (!infile)
+ exit(1);
+
+ xread(infd, infile, insize);
+ close(infd);
+
outfd = open(outfile, O_WRONLY | O_CREAT | O_TRUNC, S_IRUSR | S_IWUSR);
if (outfd < 0) {
perror("open");
@@ -799,30 +817,12 @@ int main(int argc, char *argv[])
}
}
- infd = open(imagename, O_RDONLY);
- if (infd < 0) {
- perror("open");
+ ret = xwrite(outfd, infile, insize);
+ if (ret) {
+ perror("write");
exit(1);
}
- while (image_size) {
- now = image_size < 4096 ? image_size : 4096;
-
- ret = xread(infd, buf, now);
- if (ret) {
- perror("read");
- exit(1);
- }
-
- ret = xwrite(outfd, buf, now);
- if (ret) {
- perror("write");
- exit(1);
- }
-
- image_size -= now;
- }
-
ret = close(outfd);
if (ret) {
perror("close");
--
2.1.4
_______________________________________________
barebox mailing list
barebox@xxxxxxxxxxxxxxxxxxx
http://lists.infradead.org/mailman/listinfo/barebox
