chatutils.cpp

ICE3.3.0--聊天程序服务器端demo

// **********************************************************************
//
// Copyright (c) 2003-2008 ZeroC, Inc. All rights reserved.
//
// This copy of Chat Demo is licensed to you under the terms
// described in the CHAT_DEMO_LICENSE file included in this
// distribution.
//
// **********************************************************************

#include <ChatUtils.h>
#include <Chat.h>

using namespace std;

static const unsigned int maxMessageSize = 1024;

typedef pair<const string, const string> HtmlEntity;

static const HtmlEntity htmlEntities[] = { HtmlEntity("&quot;", "\""),
                                           HtmlEntity("&#39;", "'"),
                                           HtmlEntity("&lt;", "<"),
                                           HtmlEntity("&gt;", ">"),
                                           HtmlEntity("&amp;", "&")
};

string
ChatUtils::unstripHtml(const string& s)
{
    string out = s;
    for(unsigned int count = 0; count < sizeof(htmlEntities) / sizeof(htmlEntities[0]); ++count)
    {
        for(string::size_type pos = out.find(htmlEntities[count].first);
            pos != string::npos;
            pos = out.find(htmlEntities[count].first, pos))
        {
            out.replace(pos, htmlEntities[count].first.size(), htmlEntities[count].second);
        }
    }
    return out;
}

void
ChatUtils::validateMessage(const string& in)
{
    if(in.size() > maxMessageSize)
    {
        Chat::InvalidMessageException ex;
        ostringstream msg;
        msg << "Message length exceeded, maximum length is " << maxMessageSize << " characters.";
        ex.reason = msg.str();
        throw ex;
    }
}

string
ChatUtils::trim(const string& s)
{
    static const string delims = "\t\r\n ";
    string::size_type last = s.find_last_not_of(delims);
    if(last != string::npos)
    {
        return s.substr(s.find_first_not_of(delims), last+1);
    }
    return s;
}