fix: add offset for resuming to downloadPipedStream

lart2150 · Oct 5, 2024 · 041b177 · 041b177
1 parent 6c097da
commit 041b177
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/com/tivo/kmttg/main/http.java b/src/com/tivo/kmttg/main/http.java
@@ -282,7 +282,7 @@ public static Boolean downloadPiped(String urlString, String username, String pa
 	}
 
 	public static Boolean downloadPipedStream(String urlString, String username, String password, Boolean cookies,
-			jobData job) throws IOException, InterruptedException, Exception {
+			jobData job, String offset) throws IOException, InterruptedException, Exception {
 
 		BufferedInputStream in;
 		CloseableHttpResponse response;
@@ -292,6 +292,9 @@ public static Boolean downloadPipedStream(String urlString, String username, Str
 		CloseableHttpClient httpclient = createInsecureHttpClient(url.getHost(), url.getPort(), username, password);
 
 		final HttpGet httpget = new HttpGet(urlString);
+		if (offset != null) {
+			httpget.setHeader("Range", "bytes=" + offset + "-");
+		}
 
 		response = httpclient.execute(httpget);
 

diff --git a/src/com/tivo/kmttg/task/tdownload_decrypt.java b/src/com/tivo/kmttg/task/tdownload_decrypt.java
@@ -165,7 +165,7 @@ public Thread1(String urlString) {
             }
             public void run() {
                try {
-                  success = http.downloadPipedStream(urlString, "tivo", config.MAK, true, job);
+                  success = http.downloadPipedStream(urlString, "tivo", config.MAK, true, job, job.offset);
                   thread_running = false;
                } catch (Exception e) {
                   log.error("tdownload_decrypt"); log.error(Arrays.toString(e.getStackTrace()));