211 行
4.6 KiB
C
211 行
4.6 KiB
C
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <libgen.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
|
|
#include <curl/curl.h>
|
|
|
|
const char* sofname = "odl";
|
|
const char* version = "0.2.0";
|
|
const char* avalopt = "opv";
|
|
char* filename;
|
|
|
|
int opt;
|
|
int output_flag = 0;
|
|
int version_flag = 0;
|
|
int already_flag = 0;
|
|
int err_flag = 0;
|
|
|
|
char* get_filename(const char* url) {
|
|
char* fn_start = strrchr(url, '/');
|
|
if (fn_start == NULL) {
|
|
return NULL;
|
|
}
|
|
fn_start++;
|
|
|
|
char* query = strchr(fn_start, '?');
|
|
char* anchor = strchr(fn_start, '#');
|
|
char* fn_end = NULL;
|
|
|
|
if (query != NULL && anchor != NULL) {
|
|
fn_end = (query < anchor) ? query : anchor;
|
|
} else if (query != NULL) {
|
|
fn_end = query;
|
|
} else if (anchor != NULL) {
|
|
fn_end = anchor;
|
|
}
|
|
|
|
// URLでパラメートルがなければ、そのままファイル名をコピーして
|
|
if (fn_end == NULL) {
|
|
fn_end = strchr(fn_start, '\0');
|
|
}
|
|
|
|
size_t length = fn_end - fn_start;
|
|
|
|
char* extfn = malloc(length + 1);
|
|
if (extfn == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
strncpy(extfn, fn_start, length);
|
|
extfn[length] = '\0';
|
|
|
|
return extfn;
|
|
}
|
|
|
|
int progress_callback(void *cp, double dt, double dn, double ut, double un) {
|
|
(void)cp;
|
|
(void)ut;
|
|
(void)un;
|
|
|
|
double progress = (dn / dt) * 100.0;
|
|
char* status = "ダウンロード中";
|
|
if (progress == 100.0) status = "ダウンロード済み";
|
|
|
|
printf("\r[");
|
|
int barw = 50;
|
|
int pos = (int)(progress * barw / 100.0);
|
|
|
|
for (int i = 0; i < barw; ++i) {
|
|
if (i < pos) printf("=");
|
|
else if (i == pos) printf(">");
|
|
else printf(" ");
|
|
}
|
|
|
|
printf("] %.2f%% %s, %s", progress, filename, status);
|
|
fflush(stdout);
|
|
|
|
return 0;
|
|
}
|
|
|
|
void handle_o(int argc, char* argv[]) {
|
|
output_flag = 1;
|
|
if (optind < argc) {
|
|
if (filename != NULL) {
|
|
fprintf(
|
|
stderr,
|
|
"-oをご利用する場合、1つのファイルだけをダウンロード出来ます。\n"
|
|
);
|
|
}
|
|
filename = argv[optind];
|
|
} else {
|
|
fprintf(stderr, "-oの後でファイル名がありません。");
|
|
err_flag = 1;
|
|
}
|
|
}
|
|
|
|
void flags(int opt, int argc, char* argv[]) {
|
|
switch (opt) {
|
|
case 'o':
|
|
handle_o(argc, argv);
|
|
break;
|
|
case 'p':
|
|
already_flag = 1;
|
|
break;
|
|
case 'v':
|
|
version_flag = 1;
|
|
break;
|
|
default:
|
|
err_flag = 1;
|
|
printf("usage: %s-%s [-%s] [url ...]\n", sofname, version, avalopt);
|
|
break;
|
|
}
|
|
}
|
|
|
|
int downloader(CURL* curl, char* filename, const char* url) {
|
|
if (already_flag == 1 && access(filename, F_OK) != -1) {
|
|
printf("ファイルが既に存在しますので、ダウンロードしません。\n");
|
|
return 1;
|
|
}
|
|
curl_easy_setopt(curl, CURLOPT_PROGRESSFUNCTION, progress_callback);
|
|
curl_easy_setopt(curl, CURLOPT_NOPROGRESS, 0L);
|
|
|
|
FILE* file = fopen(filename, "wb");
|
|
if (!file) {
|
|
perror("ファイルを開けません。");
|
|
curl_easy_cleanup(curl);
|
|
return -1;
|
|
}
|
|
|
|
curl_easy_setopt(curl, CURLOPT_URL, url);
|
|
curl_easy_setopt(curl, CURLOPT_WRITEDATA, file);
|
|
CURLcode res = curl_easy_perform(curl);
|
|
fclose(file);
|
|
|
|
if (res != CURLE_OK) {
|
|
fprintf(stderr, "\nダウンロードに失敗: %s\n", curl_easy_strerror(res));
|
|
curl_easy_cleanup(curl);
|
|
return -1;
|
|
}
|
|
|
|
printf("\n");
|
|
|
|
return 0;
|
|
}
|
|
|
|
int main(int argc, char* argv[]) {
|
|
if (argc < 2) {
|
|
printf("usage: %s-%s [-%s] [url ...]\n", sofname, version, avalopt);
|
|
return 1;
|
|
}
|
|
|
|
while ((opt = getopt(argc, argv, avalopt)) != -1) {
|
|
flags(opt, argc, argv);
|
|
}
|
|
|
|
if (err_flag == 1) {
|
|
return 1;
|
|
}
|
|
|
|
if (version_flag == 1) {
|
|
printf("%s-%s\n", sofname, version);
|
|
return 0;
|
|
}
|
|
|
|
CURL* curl = curl_easy_init();
|
|
if (!curl) {
|
|
perror("curlを初期設置に失敗。");
|
|
return -1;
|
|
}
|
|
|
|
// 一つのファイルだけが可能
|
|
if (output_flag == 1) {
|
|
if (optind >= argc) {
|
|
fprintf(stderr, "-oの後でURLがありません。");
|
|
return 1;
|
|
}
|
|
|
|
if (filename == NULL) {
|
|
fprintf(stderr, "-oの後でファイル名がありません。");
|
|
return 1;
|
|
}
|
|
|
|
filename = argv[optind];
|
|
const char* url = argv[optind+1];
|
|
downloader(curl, filename, url);
|
|
|
|
curl_easy_cleanup(curl);
|
|
printf("\nダウンロードに完了しました。\n");
|
|
|
|
return 0;
|
|
}
|
|
|
|
// 複数ファイル名可能
|
|
for (int i = optind; i < argc; i++) {
|
|
const char* url = argv[i];
|
|
filename = get_filename(url);
|
|
if (filename == NULL) {
|
|
fprintf(stderr, "URLからファイル名を抽出出来ませんでした。\n");
|
|
continue;
|
|
}
|
|
|
|
downloader(curl, filename, url);
|
|
}
|
|
|
|
curl_easy_cleanup(curl);
|
|
printf("\nダウンロードに完了しました。\n");
|
|
|
|
return 0;
|
|
}
|