sysmo-bts
/
barebox
9
0
Fork 0
Code Releases Activity

commands/cp: add verbose mode which displays progress bar 

Signed-off-by: Hubert Feurstein <h.feurstein@gmail.com>
Signed-off-by: Sascha Hauer <s.hauer@pengutronix.de>
This commit is contained in:
Hubert Feurstein 2011-11-22 11:24:44 +01:00 committed by Sascha Hauer
parent a59471d133
commit 7bb009c744
4 changed files with 42 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
commands/cp.c
View File

@ -32,6 +32,7 @@
#include <fs.h>
#include <malloc.h>
#include <libgen.h>
#include <getopt.h>
/**
* @param[in] cmdtp FIXME
@ -44,8 +45,21 @@ static int do_cp(struct command *cmdtp, int argc, char *argv[])
struct stat statbuf;
int last_is_dir = 0;
int i;
int opt;
int verbose = 0;
int argc_min;
if (argc < 3)
while ((opt = getopt(argc, argv, "v")) > 0) {
switch (opt) {
case 'v':
verbose = 1;
break;
}
}
argc_min = optind + 2;
if (argc < argc_min)
return COMMAND_ERROR_USAGE;
if (!stat(argv[argc - 1], &statbuf)) {
@ -53,21 +67,21 @@ static int do_cp(struct command *cmdtp, int argc, char *argv[])
last_is_dir = 1;
}
if (argc > 3 && !last_is_dir) {
if (argc > argc_min && !last_is_dir) {
printf("cp: target `%s' is not a directory\n", argv[argc - 1]);
return 1;
}
for (i = 1; i < argc - 1; i++) {
for (i = optind; i < argc - 1; i++) {
if (last_is_dir) {
char *dst;
dst = concat_path_file(argv[argc - 1], basename(argv[i]));
ret = copy_file(argv[i], dst);
ret = copy_file(argv[i], dst, verbose);
if (ret)
goto out;
free(dst);
} else {
ret = copy_file(argv[i], argv[argc - 1]);
ret = copy_file(argv[i], argv[argc - 1], verbose);
if (ret)
goto out;
}
@ -79,7 +93,7 @@ out:
}
BAREBOX_CMD_HELP_START(cp)
BAREBOX_CMD_HELP_USAGE("cp <source> <destination>\n")
BAREBOX_CMD_HELP_USAGE("cp [-v] <source> <destination>\n")
BAREBOX_CMD_HELP_SHORT("copy file from <source> to <destination>.\n")
BAREBOX_CMD_HELP_END

2
drivers/usb/gadget/dfu.c
View File

@ -259,7 +259,7 @@ static int handle_dnload(struct usb_function *f, const struct usb_ctrlrequest *c
ret = -EINVAL;
goto err_out;
}
ret = copy_file(DFU_TEMPFILE, dfu_devs[dfualt].dev);
ret = copy_file(DFU_TEMPFILE, dfu_devs[dfualt].dev, 0);
if (ret) {
printf("copy file failed\n");
ret = -EINVAL;

2
include/libbb.h
View File

@ -26,7 +26,7 @@ int recursive_action(const char *fileName, unsigned flags,
char * safe_strncpy(char *dst, const char *src, size_t size);
int copy_file(const char *src, const char *dst);
int copy_file(const char *src, const char *dst, int verbose);
int process_escape_sequence(const char *source, char *dest, int destlen);

21
lib/copy_file.c
View File

@ -4,20 +4,23 @@
#include <errno.h>
#include <malloc.h>
#include <libbb.h>
#include <progress.h>
#define RW_BUF_SIZE (ulong)4096
/**
* @param[in] src FIXME
* @param[out] dst FIXME
* @param[in] verbose FIXME
*/
int copy_file(const char *src, const char *dst)
int copy_file(const char *src, const char *dst, int verbose)
{
char *rw_buf = NULL;
int srcfd = 0, dstfd = 0;
int r, w;
int ret = 1;
void *buf;
int total = 0;
rw_buf = xmalloc(RW_BUF_SIZE);
@ -33,6 +36,15 @@ int copy_file(const char *src, const char *dst)
goto out;
}
if (verbose) {
struct stat statbuf;
if (stat(src, &statbuf) < 0)
statbuf.st_size = 0;
init_progression_bar(statbuf.st_size);
}
while(1) {
r = read(srcfd, rw_buf, RW_BUF_SIZE);
if (r < 0) {
@ -51,11 +63,18 @@ int copy_file(const char *src, const char *dst)
}
buf += w;
r -= w;
total += w;
}
if (verbose)
show_progress(total);
}
ret = 0;
out:
if (verbose)
putchar('\n');
free(rw_buf);
if (srcfd > 0)
close(srcfd);