150 lines
2.8 KiB
C
150 lines
2.8 KiB
C
/* url.c - Alpine Package Keeper (APK)
|
|
*
|
|
* Copyright (C) 2005-2008 Natanael Copa <n@tanael.org>
|
|
* Copyright (C) 2008-2011 Timo Teräs <timo.teras@iki.fi>
|
|
* All rights reserved.
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
* under the terms of the GNU General Public License version 2 as published
|
|
* by the Free Software Foundation. See http://www.gnu.org/ for details.
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <fcntl.h>
|
|
#include <errno.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <sys/wait.h>
|
|
|
|
#include "apk_io.h"
|
|
|
|
const char *apk_url_local_file(const char *url)
|
|
{
|
|
if (strncmp(url, "file:", 5) == 0)
|
|
return &url[5];
|
|
|
|
if (strncmp(url, "http:", 5) != 0 &&
|
|
strncmp(url, "https:", 6) != 0 &&
|
|
strncmp(url, "ftp:", 4) != 0)
|
|
return url;
|
|
|
|
return NULL;
|
|
}
|
|
|
|
static int translate_wget(int status)
|
|
{
|
|
if (!WIFEXITED(status))
|
|
return -EFAULT;
|
|
|
|
switch (WEXITSTATUS(status)) {
|
|
case 0:
|
|
return 0;
|
|
case 3:
|
|
return -EIO;
|
|
case 4:
|
|
return -ECONNABORTED;
|
|
case 8:
|
|
return -ENOENT;
|
|
default:
|
|
return -EFAULT;
|
|
}
|
|
}
|
|
|
|
static int fork_wget(const char *url, pid_t *ppid)
|
|
{
|
|
pid_t pid;
|
|
int fds[2];
|
|
|
|
if (pipe(fds) < 0)
|
|
return -1;
|
|
|
|
pid = fork();
|
|
if (pid == -1) {
|
|
close(fds[0]);
|
|
close(fds[1]);
|
|
return -1;
|
|
}
|
|
|
|
if (pid == 0) {
|
|
setsid();
|
|
close(fds[0]);
|
|
dup2(open("/dev/null", O_RDONLY), STDIN_FILENO);
|
|
dup2(fds[1], STDOUT_FILENO);
|
|
execlp("wget", "wget", "-q", "-O", "-", url, NULL);
|
|
/* fall back to busybox wget
|
|
* See http://redmine.alpinelinux.org/issues/347
|
|
*/
|
|
execlp("busybox", "wget", "-q", "-O", "-", url, NULL);
|
|
exit(0);
|
|
}
|
|
|
|
close(fds[1]);
|
|
|
|
if (ppid != NULL)
|
|
*ppid = pid;
|
|
|
|
return fds[0];
|
|
}
|
|
|
|
struct apk_istream *apk_istream_from_url(const char *url)
|
|
{
|
|
pid_t pid;
|
|
int fd;
|
|
|
|
if (apk_url_local_file(url) != NULL)
|
|
return apk_istream_from_file(AT_FDCWD, apk_url_local_file(url));
|
|
|
|
fd = fork_wget(url, &pid);
|
|
return apk_istream_from_fd_pid(fd, pid, translate_wget);
|
|
}
|
|
|
|
struct apk_istream *apk_istream_from_url_gz(const char *file)
|
|
{
|
|
return apk_bstream_gunzip(apk_bstream_from_url(file));
|
|
}
|
|
|
|
struct apk_bstream *apk_bstream_from_url(const char *url)
|
|
{
|
|
pid_t pid;
|
|
int fd;
|
|
|
|
if (apk_url_local_file(url))
|
|
return apk_bstream_from_file(AT_FDCWD, url);
|
|
|
|
fd = fork_wget(url, &pid);
|
|
return apk_bstream_from_fd_pid(fd, pid, translate_wget);
|
|
}
|
|
|
|
int apk_url_download(const char *url, int atfd, const char *file)
|
|
{
|
|
pid_t pid;
|
|
int status, fd;
|
|
|
|
fd = openat(atfd, file, O_CREAT|O_RDWR|O_TRUNC, 0644);
|
|
if (fd < 0)
|
|
return -errno;
|
|
|
|
pid = fork();
|
|
if (pid == -1)
|
|
return -1;
|
|
|
|
if (pid == 0) {
|
|
setsid();
|
|
dup2(open("/dev/null", O_RDONLY), STDIN_FILENO);
|
|
dup2(fd, STDOUT_FILENO);
|
|
execlp("wget", "wget", "-q", "-O", "-", url, NULL);
|
|
exit(0);
|
|
}
|
|
|
|
close(fd);
|
|
waitpid(pid, &status, 0);
|
|
status = translate_wget(status);
|
|
if (status != 0) {
|
|
unlinkat(atfd, file, 0);
|
|
return status;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|