void
MessageAttachmentStripper::_WriteToDisk(shared_ptr<Message> pMessage, MimeBody &oMainMessage, shared_ptr<MimeBody> pBody)
{
if (!pBody)
return;
AnsiString sHeader;
vector<MimeField> oFieldList = oMainMessage.Fields();
vector<MimeField>::iterator iterField = oFieldList.begin();
while (iterField != oFieldList.end())
{
MimeField oField = (*iterField);
AnsiString sName = oField.GetName();
if (sName.CompareNoCase("subject") == 0)
{
String sTextVirusFound = Configuration::Instance()->GetServerMessages()->GetMessage("VIRUS_FOUND");
sHeader += sName;
sHeader += ": " + sTextVirusFound +": ";
sHeader += oField.GetValue();
sHeader += "\r\n";
}
else
if (sName.CompareNoCase("content-type") != 0)
{
sHeader += sName;
sHeader += ": ";
sHeader += oField.GetValue();
sHeader += "\r\n";
}
iterField++;
}
String sBody = pBody->GetRawText();
String sNotification = Configuration::Instance()->GetServerMessages()->GetMessage("VIRUS_ATTACHMENT_REMOVED");
sBody = sNotification + sBody;
String sData = sHeader;
sData += "\r\n";
sData += sBody + "\r\n";
pMessage->SetSize(sData.GetLength());
String fileName = PersistentMessage::GetFileName(pMessage);
FileUtilities::WriteToFile(fileName, sData, false);
}
AnsiString
RelaxedCanonicalization::CanonicalizeHeader(AnsiString header, const std::pair<AnsiString, AnsiString> &signatureField, const std::vector<AnsiString> &fieldsToInclude, AnsiString &fieldList)
{
MimeHeader mimeHeader;
mimeHeader.Load(header, header.GetLength(), true);
std::vector<MimeField> fields = mimeHeader.Fields();
String result;
for(AnsiString headerField : fieldsToInclude)
{
headerField.Trim();
// locate this field, starting from the end.
String value;
for (size_t i = fields.size(); i > 0; i--)
{
size_t fieldIndex = i - 1;
MimeField field = fields[fieldIndex];
if (headerField.CompareNoCase(field.GetName()) == 0)
{
// found
fields.erase(fields.begin() + fieldIndex);
value = field.GetValue();
break;
}
}
if (value.GetLength() > 0)
{
/*
Convert all header field names (not the header field values) to
lowercase. For example, convert "SUBJect: AbC" to "subject: AbC".
Delete any WSP characters remaining before and after the colon
separating the header field name from the header field value. The
colon separator MUST be retained.
*/
String relaxedFieldValue;
String relaxedHeaderName = CanonicalizeHeaderName(headerField);
relaxedFieldValue = CanonicalizeHeaderValue(value);
result += relaxedHeaderName + ":" + relaxedFieldValue + "\r\n";
if (!fieldList.IsEmpty())
fieldList += ":";
fieldList += headerField;
}
}
if (!signatureField.first.IsEmpty())
{
// Don't pick the value from the actual header, use the header we're verifying instead
// If there are more than one DKIM-signature fields in the header, this will be important.
AnsiString relaxedHeaderName = CanonicalizeHeaderName(signatureField.first);
AnsiString relaxedFieldValue = CanonicalizeHeaderValue(signatureField.second);
//and without a trailing CRLF.
result += relaxedHeaderName + ":" + GetDKIMWithoutSignature_(relaxedFieldValue);
}
return result;
}
