2 files changed, 31 insertions, 0 deletions

diff --git a/include/opkele/curl.h b/include/opkele/curl.h
index 1029b34..298f095 100644
--- a/include/opkele/curl.h
+++ b/include/opkele/curl.h
@@ -33,6 +33,12 @@ namespace opkele {
 
                 virtual size_t write(void *p,size_t s,size_t nm) { return 0; }
                 CURLcode set_write();
+
+                virtual int progress(double dlt,double dln,double ult,double uln) { return 0; }
+                CURLcode set_progress();
+
+                virtual size_t header(void *p,size_t s,size_t nm) { return s*nm; }
+                CURLcode set_header();
         };
 
     }
diff --git a/lib/curl.cc b/lib/curl.cc
index 3e69b47..6172828 100644
--- a/lib/curl.cc
+++ b/lib/curl.cc
@@ -49,6 +49,31 @@ namespace opkele {
             return r;
         }
 
+        static int _progress(void *cp,double dlt,double dln,double ult,double uln) {
+            return ((curl_t*)cp)->progress(dlt,dln,ult,uln);
+        }
+
+        CURLcode curl_t::set_progress() {
+            assert(_c);
+            CURLcode r;
+            (r = easy_setopt(CURLOPT_PROGRESSDATA,this))
+            || (r = easy_setopt(CURLOPT_PROGRESSFUNCTION,_progress))
+            || (r = easy_setopt(CURLOPT_NOPROGRESS,0));
+            return r;
+        }
+
+        static size_t _header(void *p,size_t s,size_t nm,void *stream) {
+            return ((curl_t*)stream)->header(p,s,nm);
+        }
+
+        CURLcode curl_t::set_header() {
+            assert(_c);
+            CURLcode r;
+            (r = easy_setopt(CURLOPT_HEADERDATA,this))
+            || (r=easy_setopt(CURLOPT_HEADERFUNCTION,_header));
+            return r;
+        }
+
     }
 
 }