72 lines
2.0 KiB
Diff
72 lines
2.0 KiB
Diff
|
From c841c4a7871d2f79f9b7e2d53734f55e8410e231 Mon Sep 17 00:00:00 2001
|
||
|
|
From: Alexander Batischev <eual.jp@gmail.com>
|
||
|
|
Date: Fri, 22 Jan 2016 22:55:48 +0300
|
||
|
|
Subject: [PATCH] Re-download, not delete, existing files. Fixes #169
|
||
|
|
|
||
|
|
---
|
||
|
|
include/poddlthread.h | 1 +
|
||
|
|
src/poddlthread.cpp | 19 +++++++++++++++++--
|
||
|
|
2 files changed, 18 insertions(+), 2 deletions(-)
|
||
|
|
|
||
|
|
diff --git a/include/poddlthread.h b/include/poddlthread.h
|
||
|
|
index cf0f1da..99299f3 100644
|
||
|
|
--- a/include/poddlthread.h
|
||
|
|
+++ b/include/poddlthread.h
|
||
|
|
@@ -23,6 +23,7 @@ class poddlthread {
|
||
|
|
protected:
|
||
|
|
double compute_kbps();
|
||
|
|
private:
|
||
|
|
+ void run();
|
||
|
|
void mkdir_p(const char * file);
|
||
|
|
download * dl;
|
||
|
|
std::shared_ptr<std::ofstream> f;
|
||
|
|
diff --git a/src/poddlthread.cpp b/src/poddlthread.cpp
|
||
|
|
index 3a1b390..afa6762 100644
|
||
|
|
--- a/src/poddlthread.cpp
|
||
|
|
+++ b/src/poddlthread.cpp
|
||
|
|
@@ -25,6 +25,13 @@ poddlthread::~poddlthread() {
|
||
|
|
}
|
||
|
|
|
||
|
|
void poddlthread::operator()() {
|
||
|
|
+ run();
|
||
|
|
+}
|
||
|
|
+
|
||
|
|
+void poddlthread::run() {
|
||
|
|
+ // are we resuming previous download?
|
||
|
|
+ bool resumed_download = false;
|
||
|
|
+
|
||
|
|
gettimeofday(&tv1, NULL);
|
||
|
|
++bytecount;
|
||
|
|
|
||
|
|
@@ -55,11 +62,13 @@ void poddlthread::operator()() {
|
||
|
|
mkdir_p(dl->filename());
|
||
|
|
f->open(dl->filename(), std::fstream::out);
|
||
|
|
dl->set_offset(0);
|
||
|
|
+ resumed_download = false;
|
||
|
|
} else {
|
||
|
|
LOG(LOG_INFO, "poddlthread::run: stat ok: starting download from %u", sb.st_size);
|
||
|
|
curl_easy_setopt(easyhandle, CURLOPT_RESUME_FROM, sb.st_size);
|
||
|
|
dl->set_offset(sb.st_size);
|
||
|
|
f->open(dl->filename(), std::fstream::out | std::fstream::app);
|
||
|
|
+ resumed_download = true;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (f->is_open()) {
|
||
|
|
@@ -75,8 +84,14 @@ void poddlthread::operator()() {
|
||
|
|
if (0 == success)
|
||
|
|
dl->set_status(DL_READY);
|
||
|
|
else if (dl->status() != DL_CANCELLED) {
|
||
|
|
- dl->set_status(DL_FAILED);
|
||
|
|
- ::unlink(dl->filename());
|
||
|
|
+ // attempt complete re-download
|
||
|
|
+ if (resumed_download) {
|
||
|
|
+ ::unlink(dl->filename());
|
||
|
|
+ this->run();
|
||
|
|
+ } else {
|
||
|
|
+ dl->set_status(DL_FAILED);
|
||
|
|
+ ::unlink(dl->filename());
|
||
|
|
+ }
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
dl->set_status(DL_FAILED);
|