Fix headers and add function to set zchunk context in download context

Signed-off-by: Jonathan Dieter <jdieter@gmail.com>

diff --git a/include/zck.h b/include/zck.h
index 0de2c76197657754e8107ff7479150502c8ce670..fe3239ad8259a132d9408003ae50baab3f7a3e04 100644 (file)
--- a/include/zck.h
+++ b/include/zck.h
@@ -288,6 +288,9 @@ void zck_dl_free(zckDL **dl);
 /* Get zchunk context from download context */
 zckCtx *zck_dl_get_zck(zckDL *dl)
     __attribute__ ((warn_unused_result));
+/* Set zchunk context in download context */
+int zck_dl_set_zck(zckDL *dl, zckCtx *zck)
+    __attribute__ ((warn_unused_result));
 /* Clear regex used for extracting download ranges from multipart download */
 void zck_dl_clear_regex(zckDL *dl);
 /* Download and process the header from url */
@@ -303,7 +306,7 @@ size_t zck_dl_get_bytes_uploaded(zckDL *dl)
 int zck_dl_set_range(zckDL *dl, zckRange *range)
     __attribute__ ((warn_unused_result));
 /* Get download ranges from zchunk download context */
-int zck_dl_set_range(zckDL *dl, zckRange *range)
+zckRange *zck_dl_get_range(zckDL *dl)
     __attribute__ ((warn_unused_result));
 
 /* Set header callback function */

diff --git a/src/lib/dl/dl.c b/src/lib/dl/dl.c
index 316c1d5079556ec27c20028886e2908ea441bc12..6139fb220143f731d98cbbed6e53694a1442140f 100644 (file)
--- a/src/lib/dl/dl.c
+++ b/src/lib/dl/dl.c
@@ -332,6 +332,12 @@ zckCtx PUBLIC *zck_dl_get_zck(zckDL *dl) {
     return dl->zck;
 }
 
+int PUBLIC zck_dl_set_zck(zckDL *dl, zckCtx *zck) {
+    if(dl == NULL)
+        return False;
+    dl->zck = zck;
+    return True;
+}
 int PUBLIC zck_dl_set_range(zckDL *dl, zckRange *range) {
     if(dl == NULL)
         return False;