You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
tdepim/kmail/kmmsgpart.cpp

606 lines
18 KiB

// kmmsgpart.cpp
#include <config.h>
#include <kmimemagic.h>
#include <kmimetype.h>
#include <kdebug.h>
#include <kmdcodec.h>
#include "kmmsgpart.h"
#include "kmkernel.h"
#include "kmmessage.h"
#include "globalsettings.h"
#include "util.h"
#include <kasciistringtools.h>
#include <kmime_charfreq.h>
#include <kmime_codecs.h>
#include <mimelib/enum.h>
#include <mimelib/utility.h>
#include <mimelib/string.h>
#include <kiconloader.h>
#include <tqtextcodec.h>
#include <assert.h>
using namespace KMime;
//-----------------------------------------------------------------------------
KMMessagePart::KMMessagePart()
: mType("text"), mSubtype("plain"), mCte("7bit"), mBodyDecodedSize(0),
mParent(0), mLoadHeaders(false), mLoadPart(false)
{
}
//-----------------------------------------------------------------------------
KMMessagePart::KMMessagePart( TQDataStream & stream )
: mParent(0), mLoadHeaders(false), mLoadPart(false)
{
unsigned long size;
stream >> mOriginalContentTypeStr >> mName >> mContentDescription
>> mContentDisposition >> mCte >> size >> mPartSpecifier;
KPIM::kAsciiToLower( mContentDisposition.data() );
KPIM::kAsciiToUpper( mOriginalContentTypeStr.data() );
// set the type
int sep = mOriginalContentTypeStr.tqfind('/');
mType = mOriginalContentTypeStr.left(sep);
mSubtype = mOriginalContentTypeStr.mid(sep+1);
mBodyDecodedSize = size;
}
//-----------------------------------------------------------------------------
KMMessagePart::~KMMessagePart()
{
}
//-----------------------------------------------------------------------------
void KMMessagePart::clear()
{
mOriginalContentTypeStr = TQCString();
mType = "text";
mSubtype = "plain";
mCte = "7bit";
mContentDescription = TQCString();
mContentDisposition = TQCString();
mBody.truncate( 0 );
mAdditionalCTypeParamStr = TQCString();
mName = TQString();
mParameterAttribute = TQCString();
mParameterValue = TQString();
mCharset = TQCString();
mPartSpecifier = TQString();
mBodyDecodedSize = 0;
mParent = 0;
mLoadHeaders = false;
mLoadPart = false;
}
//-----------------------------------------------------------------------------
void KMMessagePart::duplicate( const KMMessagePart & msgPart )
{
// copy the data of msgPart
*this = msgPart;
// detach the explicitely shared TQByteArray
mBody.detach();
}
//-----------------------------------------------------------------------------
int KMMessagePart::decodedSize(void) const
{
if (mBodyDecodedSize < 0)
mBodyDecodedSize = bodyDecodedBinary().size();
return mBodyDecodedSize;
}
//-----------------------------------------------------------------------------
void KMMessagePart::setBody(const TQCString &aStr)
{
KMail::Util::setFromTQCString( mBody, aStr );
int enc = cte();
if (enc == DwMime::kCte7bit || enc == DwMime::kCte8bit || enc == DwMime::kCteBinary)
mBodyDecodedSize = mBody.size();
else
mBodyDecodedSize = -1; // Can't know the decoded size
}
void KMMessagePart::setBody(const DwString &aStr)
{
mBody.duplicate( aStr.c_str(), aStr.length() );
int enc = cte();
if (enc == DwMime::kCte7bit || enc == DwMime::kCte8bit || enc == DwMime::kCteBinary)
mBodyDecodedSize = mBody.size();
else
mBodyDecodedSize = -1; // Can't know the decoded size
}
void KMMessagePart::setBody(const TQByteArray &aStr)
{
mBody = aStr;
int enc = cte();
if (enc == DwMime::kCte7bit || enc == DwMime::kCte8bit || enc == DwMime::kCteBinary)
mBodyDecodedSize = mBody.size();
else
mBodyDecodedSize = -1; // Can't know the decoded size
}
void KMMessagePart::setBodyFromUnicode( const TQString & str ) {
TQCString encoding = KMMsgBase::autoDetectCharset( charset(), KMMessage::preferredCharsets(), str );
if ( encoding.isEmpty() )
encoding = "utf-8";
const TQTextCodec * codec = KMMsgBase::codecForName( encoding );
assert( codec );
TQValueList<int> dummy;
setCharset( encoding );
setBodyAndGuessCte( codec->fromUnicode( str ), dummy, false /* no 8bit */ );
}
const TQTextCodec * KMMessagePart::codec() const {
const TQTextCodec * c = KMMsgBase::codecForName( charset() );
if ( !c ) {
// Ok, no override and nothing in the message, let's use the fallback
// the user configured
c = KMMsgBase::codecForName( GlobalSettings::self()->fallbackCharacterEncoding().latin1() );
}
if ( !c )
// no charset means us-ascii (RFC 2045), so using local encoding should
// be okay
c = kmkernel->networkCodec();
assert( c );
return c;
}
TQString KMMessagePart::bodyToUnicode(const TQTextCodec* codec) const {
if ( !codec )
// No codec was given, so try the charset in the mail
codec = this->codec();
assert( codec );
return codec->toUnicode( bodyDecoded() );
}
void KMMessagePart::setCharset( const TQCString & c ) {
if ( type() != DwMime::kTypeText )
kdWarning()
<< "KMMessagePart::setCharset(): trying to set a charset for a non-textual mimetype." << endl
<< "Fix this caller:" << endl
<< "====================================================================" << endl
<< kdBacktrace( 5 ) << endl
<< "====================================================================" << endl;
mCharset = c;
}
//-----------------------------------------------------------------------------
void KMMessagePart::setBodyEncoded(const TQCString& aStr)
{
mBodyDecodedSize = aStr.size() - 1; // same as aStr.length() but faster - assuming no embedded nuls
switch (cte())
{
case DwMime::kCteQuotedPrintable:
case DwMime::kCteBase64:
{
Codec * codec = Codec::codecForName( cteStr() );
assert( codec );
// we can't use the convenience function here, since aStr is not
// a TQByteArray...:
mBody.resize( codec->maxEncodedSizeFor( mBodyDecodedSize ) );
TQCString::ConstIterator iit = aStr.data();
TQCString::ConstIterator iend = aStr.data() + mBodyDecodedSize;
TQByteArray::Iterator oit = mBody.begin();
TQByteArray::ConstIterator oend = mBody.end();
if ( !codec->encode( iit, iend, oit, oend ) )
kdWarning(5006) << codec->name()
<< " codec lies about it's maxEncodedSizeFor( "
<< mBodyDecodedSize << " ). Result truncated!" << endl;
mBody.truncate( oit - mBody.begin() );
break;
}
default:
kdWarning(5006) << "setBodyEncoded: unknown encoding '" << cteStr()
<< "'. Assuming binary." << endl;
// fall through
case DwMime::kCte7bit:
case DwMime::kCte8bit:
case DwMime::kCteBinary:
// This is slow and memory hungry - consider using setBodyEncodedBinary instead!
mBody.duplicate( aStr.data(), mBodyDecodedSize );
break;
}
}
void KMMessagePart::setBodyAndGuessCte(const TQByteArray& aBuf,
TQValueList<int> & allowedCte,
bool allow8Bit,
bool willBeSigned )
{
mBodyDecodedSize = aBuf.size();
CharFreq cf( aBuf ); // save to pass null arrays...
allowedCte = KMMessage::determineAllowedCtes( cf, allow8Bit, willBeSigned );
#ifndef NDEBUG
DwString dwCte;
DwCteEnumToStr(allowedCte[0], dwCte);
kdDebug(5006) << "CharFreq returned " << cf.type() << "/"
<< cf.printableRatio() << " and I chose "
<< dwCte.c_str() << endl;
#endif
setCte( allowedCte[0] ); // choose best fitting
setBodyEncodedBinary( aBuf );
}
void KMMessagePart::setBodyAndGuessCte(const TQCString& aBuf,
TQValueList<int> & allowedCte,
bool allow8Bit,
bool willBeSigned )
{
mBodyDecodedSize = aBuf.size() - 1; // same as aStr.length() but faster - assuming no embedded nuls
CharFreq cf( aBuf.data(), mBodyDecodedSize ); // save to pass null strings
allowedCte = KMMessage::determineAllowedCtes( cf, allow8Bit, willBeSigned );
#ifndef NDEBUG
DwString dwCte;
DwCteEnumToStr(allowedCte[0], dwCte);
kdDebug(5006) << "CharFreq returned " << cf.type() << "/"
<< cf.printableRatio() << " and I chose "
<< dwCte.c_str() << endl;
#endif
setCte( allowedCte[0] ); // choose best fitting
setBodyEncoded( aBuf );
}
//-----------------------------------------------------------------------------
void KMMessagePart::setBodyEncodedBinary(const TQByteArray& aStr)
{
mBodyDecodedSize = aStr.size();
if (aStr.isEmpty())
{
mBody.resize(0);
return;
}
switch (cte())
{
case DwMime::kCteQuotedPrintable:
case DwMime::kCteBase64:
{
Codec * codec = Codec::codecForName( cteStr() );
assert( codec );
// Nice: We can use the convenience function :-)
mBody = codec->encode( aStr );
// QP encoding does CRLF -> LF conversion, which can change the size after decoding again
// and a size mismatch triggers an assert in various other methods
mBodyDecodedSize = -1;
break;
}
default:
kdWarning(5006) << "setBodyEncodedBinary: unknown encoding '" << cteStr()
<< "'. Assuming binary." << endl;
// fall through
case DwMime::kCte7bit:
case DwMime::kCte8bit:
case DwMime::kCteBinary:
//mBody.duplicate( aStr );
mBody = aStr;
// Caller has to detach before it modifies aStr!
break;
}
}
void KMMessagePart::setMessageBody( const TQByteArray& aBuf )
{
CharFreq cf( aBuf ); // it's safe to pass null arrays
mBodyDecodedSize = aBuf.size();
int cte;
switch ( cf.type() ) {
case CharFreq::SevenBitText:
case CharFreq::SevenBitData:
cte = DwMime::kCte7bit;
break;
case CharFreq::EightBitText:
case CharFreq::EightBitData:
cte = DwMime::kCte8bit;
break;
default:
kdWarning(5006) << "Calling " << k_funcinfo
<< " with something containing neither 7 nor 8 bit text!"
<< " Fix this caller: " << kdBacktrace() << endl;
}
setCte( cte );
setBodyEncodedBinary( aBuf );
}
//-----------------------------------------------------------------------------
TQByteArray KMMessagePart::bodyDecodedBinary() const
{
if (mBody.isEmpty()) return TQByteArray();
TQByteArray result;
switch (cte())
{
case DwMime::kCte7bit:
case DwMime::kCte8bit:
case DwMime::kCteBinary:
result.duplicate(mBody);
break;
default:
if ( const Codec * codec = Codec::codecForName( cteStr() ) )
// Nice: we can use the convenience function :-)
result = codec->decode( mBody );
else {
kdWarning(5006) << "bodyDecodedBinary: unknown encoding '" << cteStr()
<< "'. Assuming binary." << endl;
result.duplicate(mBody);
}
}
assert( mBodyDecodedSize < 0
|| (unsigned int)mBodyDecodedSize == result.size() );
if ( mBodyDecodedSize < 0 )
mBodyDecodedSize = result.size(); // cache the decoded size.
return result;
}
TQCString KMMessagePart::bodyDecoded(void) const
{
if (mBody.isEmpty()) return TQCString("");
bool decodeBinary = false;
TQCString result;
int len;
switch (cte())
{
case DwMime::kCte7bit:
case DwMime::kCte8bit:
case DwMime::kCteBinary:
{
decodeBinary = true;
break;
}
default:
if ( const Codec * codec = Codec::codecForName( cteStr() ) ) {
// We can't use the codec convenience functions, since we must
// return a TQCString, not a TQByteArray:
int bufSize = codec->maxDecodedSizeFor( mBody.size() ) + 1; // trailing NUL
result.resize( bufSize );
TQByteArray::ConstIterator iit = mBody.begin();
TQCString::Iterator oit = result.begin();
TQCString::ConstIterator oend = result.begin() + bufSize;
if ( !codec->decode( iit, mBody.end(), oit, oend ) )
kdWarning(5006) << codec->name()
<< " lies about it's maxDecodedSizeFor( "
<< mBody.size() << " ). Result truncated!" << endl;
len = oit - result.begin();
result.truncate( len ); // adds trailing NUL
} else {
kdWarning(5006) << "bodyDecoded: unknown encoding '" << cteStr()
<< "'. Assuming binary." << endl;
decodeBinary = true;
}
}
if ( decodeBinary ) {
len = mBody.size();
KMail::Util::setFromByteArray( result, mBody );
}
// Calls length -> slow
//kdWarning( result.length() != (unsigned int)len, 5006 )
// << "KMMessagePart::bodyDecoded(): body is binary but used as text!" << endl;
result = result.tqreplace( "\r\n", "\n" ); // CRLF -> LF conversion
assert( mBodyDecodedSize < 0 || mBodyDecodedSize == len );
if ( mBodyDecodedSize < 0 )
mBodyDecodedSize = len; // cache decoded size
return result;
}
//-----------------------------------------------------------------------------
void KMMessagePart::magicSetType(bool aAutoDecode)
{
KMimeMagic::self()->setFollowLinks( true ); // is it necessary ?
const TQByteArray body = ( aAutoDecode ) ? bodyDecodedBinary() : mBody ;
KMimeMagicResult * result = KMimeMagic::self()->findBufferType( body );
TQString mimetype = result->mimeType();
const int sep = mimetype.tqfind('/');
mType = mimetype.left(sep).latin1();
mSubtype = mimetype.mid(sep+1).latin1();
}
//-----------------------------------------------------------------------------
TQString KMMessagePart::iconName( int size ) const
{
TQCString mimeType( mType + "/" + mSubtype );
KPIM::kAsciiToLower( mimeType.data() );
TQString fileName =
KMimeType::mimeType( mimeType )->icon( TQString(), false );
if ( fileName.isEmpty() )
{
fileName = this->fileName();
if ( fileName.isEmpty() ) fileName = this->name();
if ( !fileName.isEmpty() )
{
fileName = KMimeType::findByPath( "/tmp/"+fileName, 0, true )->icon( TQString(), true );
}
}
fileName =
KGlobal::instance()->iconLoader()->iconPath( fileName, size );
return fileName;
}
//-----------------------------------------------------------------------------
int KMMessagePart::type() const {
return DwTypeStrToEnum(DwString(mType));
}
//-----------------------------------------------------------------------------
void KMMessagePart::setType(int aType)
{
DwString dwType;
DwTypeEnumToStr(aType, dwType);
mType = dwType.c_str();
}
//-----------------------------------------------------------------------------
int KMMessagePart::subtype() const {
return DwSubtypeStrToEnum(DwString(mSubtype));
}
//-----------------------------------------------------------------------------
void KMMessagePart::setSubtype(int aSubtype)
{
DwString dwSubtype;
DwSubtypeEnumToStr(aSubtype, dwSubtype);
mSubtype = dwSubtype.c_str();
}
//-----------------------------------------------------------------------------
TQCString KMMessagePart::parameterAttribute(void) const
{
return mParameterAttribute;
}
//-----------------------------------------------------------------------------
TQString KMMessagePart::parameterValue(void) const
{
return mParameterValue;
}
//-----------------------------------------------------------------------------
void KMMessagePart::setParameter(const TQCString &attribute,
const TQString &value)
{
mParameterAttribute = attribute;
mParameterValue = value;
}
//-----------------------------------------------------------------------------
TQCString KMMessagePart::contentTransferEncodingStr(void) const
{
return mCte;
}
//-----------------------------------------------------------------------------
int KMMessagePart::contentTransferEncoding(void) const
{
return DwCteStrToEnum(DwString(mCte));
}
//-----------------------------------------------------------------------------
void KMMessagePart::setContentTransferEncodingStr(const TQCString &aStr)
{
mCte = aStr;
}
//-----------------------------------------------------------------------------
void KMMessagePart::setContentTransferEncoding(int aCte)
{
DwString dwCte;
DwCteEnumToStr(aCte, dwCte);
mCte = dwCte.c_str();
}
//-----------------------------------------------------------------------------
TQString KMMessagePart::contentDescription(void) const
{
return KMMsgBase::decodeRFC2047String(mContentDescription, charset());
}
//-----------------------------------------------------------------------------
void KMMessagePart::setContentDescription(const TQString &aStr)
{
TQCString encoding = KMMsgBase::autoDetectCharset(charset(),
KMMessage::preferredCharsets(), aStr);
if (encoding.isEmpty()) encoding = "utf-8";
mContentDescription = KMMsgBase::encodeRFC2047String(aStr, encoding);
}
//-----------------------------------------------------------------------------
TQString KMMessagePart::fileName(void) const
{
TQCString str;
// Allow for multiple filname*0, filename*1, ... params (defined by RFC 2231)
// in the Content-Disposision
if ( mContentDisposition.tqcontains( "filename*", false ) ) {
// It's RFC 2231 encoded, so extract the file name with the 2231 method
str = KMMsgBase::extractRFC2231HeaderField( mContentDisposition, "filename" );
return KMMsgBase::decodeRFC2231String(str);
} else {
// Standard RFC 2047-encoded
// search the start of the filename
int startOfFilename = mContentDisposition.tqfind("filename=", 0, false);
if (startOfFilename < 0)
return TQString();
startOfFilename += 9;
// search the end of the filename
int endOfFilename;
if ( '"' == mContentDisposition[startOfFilename] ) {
startOfFilename++; // the double quote isn't part of the filename
endOfFilename = mContentDisposition.tqfind('"', startOfFilename) - 1;
}
else {
endOfFilename = mContentDisposition.tqfind(';', startOfFilename) - 1;
}
if (endOfFilename < 0)
endOfFilename = 32767;
const TQCString str = mContentDisposition.mid(startOfFilename,
endOfFilename-startOfFilename+1)
.stripWhiteSpace();
return KMMsgBase::decodeRFC2047String(str);
}
return TQString();
}
TQCString KMMessagePart::body() const
{
return TQCString( mBody.data(), mBody.size() + 1 ); // space for trailing NUL
}
DwString KMMessagePart::dwBody() const
{
return KMail::Util::dwString( mBody );
}