void
MessageData::SetBody(const String &sBody)
{
shared_ptr<MimeBody> pPart = FindPart("text/plain");
if (!pPart)
pPart = CreatePart("text/plain");
String sModifiedBody = sBody;
if (sModifiedBody.Right(2) != _T("\r\n"))
{
// Add carriage return.
sModifiedBody += "\r\n";
}
// Set the text to the part
if (m_bEncodeFields)
pPart->SetUnicodeText(sModifiedBody);
else
pPart->SetRawText(sModifiedBody);
}
void
MessageData::SetHTMLBody(const String &sNewVal)
{
shared_ptr<MimeBody> pHTMLPart = FindPart("text/html");
if (!pHTMLPart)
{
// Create a new part.
pHTMLPart = CreatePart("text/html");
}
String sModifiedBody = sNewVal;
if (sModifiedBody.Right(2) != _T("\r\n"))
{
// Add carriage return.
sModifiedBody += "\r\n";
}
// Set the text to the part
if (m_bEncodeFields)
pHTMLPart->SetUnicodeText(sModifiedBody);
else
pHTMLPart->SetRawText(sModifiedBody);
}
// load a body part from string buffer
int CMimeBody::Load(const char* pszData, int nDataSize)
{
// load header fields
int nSize = CMimeHeader::Load(pszData, nDataSize);
if (nSize <= 0)
return nSize;
const char* pszDataBegin = pszData; // preserve start position
pszData += nSize;
nDataSize -= nSize;
FreeBuffer();
// determine the length of the content
const char* pszEnd = pszData + nDataSize;
int nMediaType = GetMediaType();
if (MEDIA_MULTIPART == nMediaType)
{
// find the begin boundary
string strBoundary = GetBoundary();
if (!strBoundary.empty())
{
strBoundary = "\r\n--" + strBoundary;
pszEnd = ::FindString(pszData-2, strBoundary.c_str(), pszEnd);
if (!pszEnd)
pszEnd = pszData + nDataSize;
else
pszEnd += 2;
}
}
// load content
nSize = (int)(pszEnd - pszData);
if (nSize > 0)
{
CMimeCodeBase* pCoder = CMimeEnvironment::CreateCoder(GetTransferEncoding());
ASSERT(pCoder != NULL);
pCoder->SetInput(pszData, nSize, false);
int nOutput = pCoder->GetOutputLength();
if (AllocateBuffer(nOutput+4))
nOutput = pCoder->GetOutput(m_pbText, nOutput);
else
nOutput = -1;
delete pCoder;
if (nOutput < 0)
return nOutput;
ASSERT(nOutput < m_nTextSize);
m_pbText[nOutput] = 0;
m_nTextSize = nOutput;
pszData += nSize;
nDataSize -= nSize;
}
if (nDataSize <= 0)
return (int)(pszData - pszDataBegin);
// load child body parts
string strBoundary = GetBoundary();
ASSERT(strBoundary.size() > 0);
strBoundary = "\r\n--" + strBoundary;
// look for the first boundary (case sensitive)
pszData -= 2; // go back to CRLF
nDataSize += 2;
pszEnd = pszData + nDataSize;
const char* pszBound1 = ::FindString(pszData, strBoundary.c_str(), pszEnd);
while (pszBound1 != NULL && pszBound1 < pszEnd)
{
const char* pszStart = ::FindString(pszBound1+2, "\r\n", pszEnd);
if (!pszStart)
break;
pszStart += 2;
if (pszBound1[strBoundary.size()] == '-' && pszBound1[strBoundary.size()+1] == '-')
return (int)(pszStart - pszDataBegin); // reach the closing boundary
// look for the next boundary
const char* pszBound2 = ::FindString(pszStart, strBoundary.c_str(), pszEnd);
if (!pszBound2) // overflow, boundary may be truncated
pszBound2 = pszEnd;
int nEntitySize = (int) (pszBound2 - pszStart);
// find the media type of this body part:
CMimeHeader header;
header.Load(pszStart, nEntitySize);
string strMediaType = header.GetMainType();
CMimeBody* pBP = CreatePart(strMediaType.c_str());
int nInputSize = pBP->Load(pszStart, nEntitySize);
if (nInputSize < 0)
{
ErasePart(pBP);
return nInputSize;
}
pszBound1 = pszBound2;
}
return (int)(pszEnd - pszDataBegin);
}
