fileio: fileio_size() can now fail
Part of making the fileio API more robust. Signed-off-by: Øyvind Harboe <oyvind.harboe@zylin.com>
This commit is contained in:
@@ -720,14 +720,20 @@ COMMAND_HANDLER(mg_write_cmd)
|
||||
if (ret != ERROR_OK)
|
||||
return ret;
|
||||
|
||||
int filesize;
|
||||
buffer = malloc(MG_FILEIO_CHUNK);
|
||||
if (!buffer) {
|
||||
fileio_close(&fileio);
|
||||
return ERROR_FAIL;
|
||||
}
|
||||
int retval = fileio_size(&fileio, &filesize);
|
||||
if (retval != ERROR_OK) {
|
||||
fileio_close(&fileio);
|
||||
return retval;
|
||||
}
|
||||
|
||||
cnt = fileio_size(&fileio) / MG_FILEIO_CHUNK;
|
||||
res = fileio_size(&fileio) % MG_FILEIO_CHUNK;
|
||||
cnt = filesize / MG_FILEIO_CHUNK;
|
||||
res = filesize % MG_FILEIO_CHUNK;
|
||||
|
||||
struct duration bench;
|
||||
duration_start(&bench);
|
||||
@@ -752,8 +758,8 @@ COMMAND_HANDLER(mg_write_cmd)
|
||||
if (duration_measure(&bench) == ERROR_OK)
|
||||
{
|
||||
command_print(CMD_CTX, "wrote %ld bytes from file %s "
|
||||
"in %fs (%0.3f kB/s)", (long)fileio_size(&fileio), CMD_ARGV[1],
|
||||
duration_elapsed(&bench), duration_kbps(&bench, fileio_size(&fileio)));
|
||||
"in %fs (%0.3f kB/s)", (long)filesize, CMD_ARGV[1],
|
||||
duration_elapsed(&bench), duration_kbps(&bench, filesize));
|
||||
}
|
||||
|
||||
free(buffer);
|
||||
|
||||
Reference in New Issue
Block a user