Second Life viewer sources 1.21.3-RC

11 files changed, 96 insertions, 38 deletions

diff --git a/linden/indra/llmessage/llares.cpp b/linden/indra/llmessage/llares.cpp
index 5a310b3..7f573d2 100644
--- a/linden/indra/llmessage/llares.cpp
+++ b/linden/indra/llmessage/llares.cpp
@@ -100,9 +100,16 @@ void LLAres::QueryResponder::queryError(int code)
                         << LLAres::strerror(code) << llendl;
 }
 
-LLAres::LLAres()
+LLAres::LLAres() :
+chan_(NULL), mInitSuccess(false)
 {
-        ares_init(&chan_);
+        if (ares_init(&chan_) != ARES_SUCCESS)
+        {
+                llwarns << "Could not succesfully initialize ares!" << llendl;
+                return;
+        }
+
+        mInitSuccess = true;
 }
 
 LLAres::~LLAres()
diff --git a/linden/indra/llmessage/llares.h b/linden/indra/llmessage/llares.h
index 5d72170..17f6085 100644
--- a/linden/indra/llmessage/llares.h
+++ b/linden/indra/llmessage/llares.h
@@ -437,9 +437,11 @@ public:
          */
         static const char *strerror(int code);
 
+        bool isInitialized(void) { return mInitSuccess; }
+
 protected:
         ares_channel chan_;
-
+        bool mInitSuccess;
 };
         
 /**
diff --git a/linden/indra/llmessage/llhttpclient.cpp b/linden/indra/llmessage/llhttpclient.cpp
index caeaee1..fc2612f 100644
--- a/linden/indra/llmessage/llhttpclient.cpp
+++ b/linden/indra/llmessage/llhttpclient.cpp
@@ -56,7 +56,7 @@ namespace
         {
         public:
                 LLHTTPClientURLAdaptor(LLCurl::ResponderPtr responder)
-                        : mResponder(responder), mStatus(499),
+                        : LLURLRequestComplete(), mResponder(responder), mStatus(499),
                           mReason("LLURLRequest complete w/no status")
                 {
                 }
@@ -67,6 +67,8 @@ namespace
 
                 virtual void httpStatus(U32 status, const std::string& reason)
                 {
+                        LLURLRequestComplete::httpStatus(status,reason);
+
                         mStatus = status;
                         mReason = reason;
                 }
diff --git a/linden/indra/llmessage/llmessagetemplate.h b/linden/indra/llmessage/llmessagetemplate.h
index f92c4b7..716c618 100644
--- a/linden/indra/llmessage/llmessagetemplate.h
+++ b/linden/indra/llmessage/llmessagetemplate.h
@@ -265,6 +265,7 @@ enum EMsgDeprecation
 {
         MD_NOTDEPRECATED,
         MD_UDPDEPRECATED,
+        MD_UDPBLACKLISTED,
         MD_DEPRECATED
 };
 
@@ -299,7 +300,7 @@ public:
         ~LLMessageTemplate()
         {
                 for_each(mMemberBlocks.begin(), mMemberBlocks.end(), DeletePointer());
-}
+        }
 
         void addBlock(LLMessageBlock *blockp)
         {
@@ -380,6 +381,11 @@ public:
                 return trustedSource ? mBanFromTrusted : mBanFromUntrusted;
         }
 
+        bool isUdpBanned() const
+        {
+                return mDeprecation == MD_UDPBLACKLISTED;
+        }
+
         friend std::ostream&     operator<<(std::ostream& s, LLMessageTemplate &msg);
 
         const LLMessageBlock* getBlock(char* name) const
diff --git a/linden/indra/llmessage/llmessagetemplateparser.cpp b/linden/indra/llmessage/llmessagetemplateparser.cpp
index d6adba8..9f6eeca 100644
--- a/linden/indra/llmessage/llmessagetemplateparser.cpp
+++ b/linden/indra/llmessage/llmessagetemplateparser.cpp
@@ -525,6 +525,10 @@ LLMessageTemplate * LLTemplateParser::parseMessage(LLTemplateTokenizer & tokens)
         {
                 templatep->setDeprecation(MD_UDPDEPRECATED);
         }
+        else if (tokens.want("UDPBlackListed"))
+        {
+                templatep->setDeprecation(MD_UDPBLACKLISTED);
+        }
         else if (tokens.want("NotDeprecated"))
         {
                 // this is the default value, but it can't hurt to set it twice
diff --git a/linden/indra/llmessage/lltemplatemessagebuilder.cpp b/linden/indra/llmessage/lltemplatemessagebuilder.cpp
index cd06c07..6d3ff8f 100644
--- a/linden/indra/llmessage/lltemplatemessagebuilder.cpp
+++ b/linden/indra/llmessage/lltemplatemessagebuilder.cpp
@@ -60,7 +60,6 @@ LLTemplateMessageBuilder::~LLTemplateMessageBuilder()
         mCurrentSMessageData = NULL;
 }
 
-
 // virtual
 void LLTemplateMessageBuilder::newMessage(const char *name)
 {
diff --git a/linden/indra/llmessage/lltemplatemessagereader.cpp b/linden/indra/llmessage/lltemplatemessagereader.cpp
index f5da14b..2c457e8 100644
--- a/linden/indra/llmessage/lltemplatemessagereader.cpp
+++ b/linden/indra/llmessage/lltemplatemessagereader.cpp
@@ -791,12 +791,16 @@ bool LLTemplateMessageReader::isTrusted() const
         return mCurrentRMessageTemplate->getTrust() == MT_TRUST;
 }
 
-//virtual 
 bool LLTemplateMessageReader::isBanned(bool trustedSource) const
 {
         return mCurrentRMessageTemplate->isBanned(trustedSource);
 }
 
+bool LLTemplateMessageReader::isUdpBanned() const
+{
+        return mCurrentRMessageTemplate->isUdpBanned();
+}
+
 //virtual 
 void LLTemplateMessageReader::copyToBuilder(LLMessageBuilder& builder) const
 {
diff --git a/linden/indra/llmessage/lltemplatemessagereader.h b/linden/indra/llmessage/lltemplatemessagereader.h
index 98f41ae..ff124f3 100644
--- a/linden/indra/llmessage/lltemplatemessagereader.h
+++ b/linden/indra/llmessage/lltemplatemessagereader.h
@@ -109,7 +109,8 @@ public:
 
         bool isTrusted() const;
         bool isBanned(bool trusted_source) const;
-
+        bool isUdpBanned() const;
+        
 private:
 
         void getData(const char *blockname, const char *varname, void *datap, 
diff --git a/linden/indra/llmessage/llurlrequest.cpp b/linden/indra/llmessage/llurlrequest.cpp
index 42f3f04..fbd002b 100644
--- a/linden/indra/llmessage/llurlrequest.cpp
+++ b/linden/indra/llmessage/llurlrequest.cpp
@@ -461,58 +461,76 @@ size_t LLURLRequest::upCallback(
 
 static size_t headerCallback(void* data, size_t size, size_t nmemb, void* user)
 {
-        const char* headerLine = (const char*)data;
-        size_t headerLen = size * nmemb;
+        const char* header_line = (const char*)data;
+        size_t header_len = size * nmemb;
         LLURLRequestComplete* complete = (LLURLRequestComplete*)user;
 
+        if (!complete || !header_line)
+        {
+                return header_len;
+        }
+
         // *TODO: This should be a utility in llstring.h: isascii()
-        for (size_t i = 0; i < headerLen; ++i)
+        for (size_t i = 0; i < header_len; ++i)
         {
-                if (headerLine[i] < 0)
+                if (header_line[i] < 0)
                 {
-                        return headerLen;
+                        return header_len;
                 }
         }
 
-        size_t sep;
-        for (sep = 0; sep < headerLen  &&  headerLine[sep] != ':'; ++sep) { }
-
-        if (sep < headerLen && complete)
-        {
-                std::string key(headerLine, sep);
-                std::string value(headerLine + sep + 1, headerLen - sep - 1);
-
-                key = utf8str_tolower(utf8str_trim(key));
-                value = utf8str_trim(value);
+        std::string header(header_line, header_len);
 
-                complete->header(key, value);
-        }
-        else
+        // Per HTTP spec the first header line must be the status line.
+        if (!complete->haveHTTPStatus())
         {
-                std::string s(headerLine, headerLen);
-
-                std::string::iterator end = s.end();
-                std::string::iterator pos1 = std::find(s.begin(), end, ' ');
+                std::string::iterator end = header.end();
+                std::string::iterator pos1 = std::find(header.begin(), end, ' ');
                 if (pos1 != end) ++pos1;
                 std::string::iterator pos2 = std::find(pos1, end, ' ');
                 if (pos2 != end) ++pos2;
                 std::string::iterator pos3 = std::find(pos2, end, '\r');
 
-                std::string version(s.begin(), pos1);
+                std::string version(header.begin(), pos1);
                 std::string status(pos1, pos2);
                 std::string reason(pos2, pos3);
 
                 int statusCode = atoi(status.c_str());
                 if (statusCode > 0)
                 {
-                        if (complete)
-                        {
-                                complete->httpStatus((U32)statusCode, reason);
-                        }
+                        complete->httpStatus((U32)statusCode, reason);
+                }
+                else
+                {
+                        llwarns << "Unable to parse http response status line: "
+                                        << header << llendl;
+                        complete->httpStatus(499,"Unable to parse status line.");
+                }
+                return header_len;
+        }
+
+        std::string::iterator sep = std::find(header.begin(),header.end(),':');
+
+        if (sep != header.end())
+        {
+                std::string key(header.begin(), sep);
+                std::string value(sep + 1, header.end());
+
+                key = utf8str_tolower(utf8str_trim(key));
+                value = utf8str_trim(value);
+
+                complete->header(key, value);
+        }
+        else
+        {
+                LLStringUtil::trim(header);
+                if (!header.empty())
+                {
+                        llwarns << "Unable to parse header: " << header << llendl;
                 }
         }
 
-        return headerLen;
+        return header_len;
 }
 
 /**
@@ -553,7 +571,8 @@ LLIOPipe::EStatus LLContextURLExtractor::process_impl(
  * LLURLRequestComplete
  */
 LLURLRequestComplete::LLURLRequestComplete() :
-        mRequestStatus(LLIOPipe::STATUS_ERROR)
+        mRequestStatus(LLIOPipe::STATUS_ERROR),
+        mHaveHTTPStatus(false)
 {
         LLMemType m1(LLMemType::MTYPE_IO_URL_REQUEST);
 }
@@ -572,6 +591,7 @@ void LLURLRequestComplete::header(const std::string& header, const std::string&
 //virtual 
 void LLURLRequestComplete::httpStatus(U32 status, const std::string& reason)
 {
+        mHaveHTTPStatus = true;
 }
 
 //virtual 
diff --git a/linden/indra/llmessage/llurlrequest.h b/linden/indra/llmessage/llurlrequest.h
index 888b962..f965dad 100644
--- a/linden/indra/llmessage/llurlrequest.h
+++ b/linden/indra/llmessage/llurlrequest.h
@@ -327,6 +327,9 @@ public:
         LLURLRequestComplete();
         virtual ~LLURLRequestComplete();
 
+        // The first line of an http response must be the status line
+        // true if we have already parsed this line.
+        bool haveHTTPStatus() const { return mHaveHTTPStatus; }
 protected:
         /* @name LLIOPipe virtual implementations
          */
@@ -345,6 +348,8 @@ protected:
         // value to note if we actually got the response. This value
         // depends on correct useage from the LLURLRequest instance.
         EStatus mRequestStatus;
+
+        bool mHaveHTTPStatus;
 };
 
 
diff --git a/linden/indra/llmessage/message.cpp b/linden/indra/llmessage/message.cpp
index 27f9d12..b20731a 100644
--- a/linden/indra/llmessage/message.cpp
+++ b/linden/indra/llmessage/message.cpp
@@ -754,11 +754,19 @@ BOOL LLMessageSystem::checkMessages( S64 frame_count )
                                 clearReceiveState();
                                 valid_packet = FALSE;
                         }
+
+                        if( valid_packet && mTemplateMessageReader->isUdpBanned())
+                        {
+                                llwarns << "Received UDP black listed message "
+                                                <<  mTemplateMessageReader->getMessageName()
+                                                << " from " << host << llendl;
+                                clearReceiveState();
+                                valid_packet = FALSE;
+                        }
                         
                         if( valid_packet )
                         {
                                 logValidMsg(cdp, host, recv_reliable, recv_resent, (BOOL)(acks>0) );
-
                                 valid_packet = mTemplateMessageReader->readMessage(buffer, host);
                         }