Ported some cURL and HTTP-related changes from Snowglobe.

1 files changed, 26 insertions, 8 deletions

diff --git a/linden/indra/llmessage/llcurl.cpp b/linden/indra/llmessage/llcurl.cpp
index 811d4af..12321fe 100644
--- a/linden/indra/llmessage/llcurl.cpp
+++ b/linden/indra/llmessage/llcurl.cpp
@@ -220,7 +220,7 @@ public:
         U32 report(CURLcode);
         void getTransferInfo(LLCurl::TransferInfo* info);
 
-        void prepRequest(const std::string& url, ResponderPtr, bool post = false);
+        void prepRequest(const std::string& url, const std::vector<std::string>& headers, ResponderPtr, bool post = false);
         
         const char* getErrorBuffer();
 
@@ -432,7 +432,9 @@ size_t curlHeaderCallback(void* data, size_t size, size_t nmemb, void* user_data
         return n;
 }
 
-void LLCurl::Easy::prepRequest(const std::string& url, ResponderPtr responder, bool post)
+void LLCurl::Easy::prepRequest(const std::string& url,
+                                                           const std::vector<std::string>& headers,
+                                                           ResponderPtr responder, bool post)
 {
         resetState();
         
@@ -465,8 +467,13 @@ void LLCurl::Easy::prepRequest(const std::string& url, ResponderPtr responder, b
         {
                 slist_append("Connection: keep-alive");
                 slist_append("Keep-alive: 300");
+                // Accept and other headers
+                for (std::vector<std::string>::const_iterator iter = headers.begin();
+                         iter != headers.end(); ++iter)
+                {
+                        slist_append((*iter).c_str());
+                }
         }
-        // *FIX: should have ACCEPT headers
 }
 
 ////////////////////////////////////////////////////////////////////////////
@@ -676,15 +683,18 @@ LLCurlRequest::LLCurlRequest() :
         mActiveMulti(NULL),
         mActiveRequestCount(0)
 {
+        mThreadID = LLThread::currentID();
 }
 
 LLCurlRequest::~LLCurlRequest()
 {
+        llassert_always(mThreadID == LLThread::currentID());
         for_each(mMultiSet.begin(), mMultiSet.end(), DeletePointer());
 }
 
 void LLCurlRequest::addMulti()
 {
+        llassert_always(mThreadID == LLThread::currentID());
         LLCurl::Multi* multi = new LLCurl::Multi();
         mMultiSet.insert(multi);
         mActiveMulti = multi;
@@ -714,17 +724,20 @@ bool LLCurlRequest::addEasy(LLCurl::Easy* easy)
 
 void LLCurlRequest::get(const std::string& url, LLCurl::ResponderPtr responder)
 {
-        getByteRange(url, 0, -1, responder);
+        getByteRange(url, headers_t(), 0, -1, responder);
 }
         
-bool LLCurlRequest::getByteRange(const std::string& url, S32 offset, S32 length, LLCurl::ResponderPtr responder)
+bool LLCurlRequest::getByteRange(const std::string& url,
+                                                                 const headers_t& headers,
+                                                                 S32 offset, S32 length,
+                                                                 LLCurl::ResponderPtr responder)
 {
         LLCurl::Easy* easy = allocEasy();
         if (!easy)
         {
                 return false;
         }
-        easy->prepRequest(url, responder);
+        easy->prepRequest(url, headers, responder);
         easy->setopt(CURLOPT_HTTPGET, 1);
         if (length > 0)
         {
@@ -736,14 +749,17 @@ bool LLCurlRequest::getByteRange(const std::string& url, S32 offset, S32 length,
         return res;
 }
 
-bool LLCurlRequest::post(const std::string& url, const LLSD& data, LLCurl::ResponderPtr responder)
+bool LLCurlRequest::post(const std::string& url,
+                                                 const headers_t& headers,
+                                                 const LLSD& data,
+                                                 LLCurl::ResponderPtr responder)
 {
         LLCurl::Easy* easy = allocEasy();
         if (!easy)
         {
                 return false;
         }
-        easy->prepRequest(url, responder);
+        easy->prepRequest(url, headers, responder);
 
         LLSDSerialize::toXML(data, easy->getInput());
         S32 bytes = easy->getInput().str().length();
@@ -763,6 +779,7 @@ bool LLCurlRequest::post(const std::string& url, const LLSD& data, LLCurl::Respo
 // Note: call once per frame
 S32 LLCurlRequest::process()
 {
+        llassert_always(mThreadID == LLThread::currentID());
         S32 res = 0;
         for (curlmulti_set_t::iterator iter = mMultiSet.begin();
                  iter != mMultiSet.end(); )
@@ -782,6 +799,7 @@ S32 LLCurlRequest::process()
 
 S32 LLCurlRequest::getQueued()
 {
+        llassert_always(mThreadID == LLThread::currentID());
         S32 queued = 0;
         for (curlmulti_set_t::iterator iter = mMultiSet.begin();
                  iter != mMultiSet.end(); )