encapsulated write functionality into curl_t

Signed-off-by: Michael Krelin <hacker@klever.net>

1 files changed, 12 insertions, 0 deletions

diff --git a/lib/curl.cc b/lib/curl.cc
index 418aa79..3e69b47 100644
--- a/lib/curl.cc
+++ b/lib/curl.cc
@@ -24,19 +24,31 @@ namespace opkele {
             (r=easy_setopt(CURLOPT_FOLLOWLOCATION,1))
             || (r=easy_setopt(CURLOPT_MAXREDIRS,5))
             || (r=easy_setopt(CURLOPT_DNS_CACHE_TIMEOUT,120))
             || (r=easy_setopt(CURLOPT_DNS_USE_GLOBAL_CACHE,1))
             || (r=easy_setopt(CURLOPT_USERAGENT,PACKAGE_NAME"/"PACKAGE_SRC_VERSION))
             || (r=easy_setopt(CURLOPT_TIMEOUT,20))
         #ifdefDISABLE_CURL_SSL_VERIFYHOST
             || (r=easy_setopt(CURLOPT_SSL_VERIFYHOST,0))
 #endif
         #ifdefDISABLE_CURL_SSL_VERIFYPEER
             || (r=easy_setopt(CURLOPT_SSL_VERIFYPEER,0))
 #endif
             ;
             return r;
         }
 
+        static size_t _write(void *p,size_t s,size_t nm,void *stream) {
+            return ((curl_t*)stream)->write(p,s,nm);
+        }
+
+        CURLcode curl_t::set_write() {
+            assert(_c);
+            CURLcode r;
+            (r = easy_setopt(CURLOPT_WRITEDATA,this))
+            || (r = easy_setopt(CURLOPT_WRITEFUNCTION,_write));
+            return r;
+        }
+
     }
 
 }