sergiotarxz/nuttx-apps
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

examples/wget: Saving a file downloaded with wget

Browse Source 
wget [-o <local-path>] <url>
This commit is contained in:
Fadil R. Berisha 2021-12-15 10:37:40 -05:00 committed by Xiang Xiao
parent 2db90a8be5
commit 2aadb22301
1 changed files with 27 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
examples/wget/wget_main.c
View File

@ -26,6 +26,8 @@
#include <stdint.h> #include <stdint.h>
#include <unistd.h> #include <unistd.h>
#include <fcntl.h>
#include <sys/stat.h>
#include <netinet/in.h> #include <netinet/in.h>
#include <arpa/inet.h> #include <arpa/inet.h>
@ -70,6 +72,7 @@
****************************************************************************/ ****************************************************************************/
static char g_iobuffer[512]; static char g_iobuffer[512];
static int g_fd;
/**************************************************************************** /****************************************************************************
* Private Functions * Private Functions
@ -82,7 +85,7 @@ static char g_iobuffer[512];
static int callback(FAR char **buffer, int offset, int datend, static int callback(FAR char **buffer, int offset, int datend,
FAR int *buflen, FAR void *arg) FAR int *buflen, FAR void *arg)
{ {
ssize_t written = write(1, &((*buffer)[offset]), datend - offset); ssize_t written = write(g_fd, &((*buffer)[offset]), datend - offset);
if (written == -1) if (written == -1)
{ {
return -errno; return -errno;
@ -152,19 +155,38 @@ int main(int argc, FAR char *argv[])
ctx.buflen = 512; ctx.buflen = 512;
ctx.sink_callback = callback; ctx.sink_callback = callback;
ctx.sink_callback_arg = NULL; ctx.sink_callback_arg = NULL;
if (argc > 1)
if (argc >= 3)
{
ctx.url = argv[argc - 1];
g_fd = open(argv[argc - 2],
O_WRONLY | O_CREAT | O_TRUNC,
S_IRWXU | S_IRWXG | S_IRWXO);
if (g_fd < 0)
{
printf("cannot create file %s \n", argv[argc - 2]);
}
}
else if (argc == 2)
{ {
ctx.url = argv[1]; ctx.url = argv[1];
g_fd = 1;
} }
else else
{ {
ctx.url = CONFIG_EXAMPLES_WGET_URL; ctx.url = CONFIG_EXAMPLES_WGET_URL;
g_fd = 1;
} }
int ret = webclient_perform(&ctx); if (g_fd >= 0)
if (ret != 0)
{ {
printf("webclient_perform failed with %d\n", ret); int ret = webclient_perform(&ctx);
if (ret != 0)
{
printf("webclient_perform failed with %d\n", ret);
}
close(g_fd);
} }
return 0; return 0;