Revision control
Copy as Markdown
Other Tools
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
#include <sys/stat.h>
#include "msgCore.h"
#include "nsImapMailDatabase.h"
#include "nsDBFolderInfo.h"
#include "nsMsgMessageFlags.h"
const char* kPendingHdrsScope =
"ns:msg:db:row:scope:pending:all"; // scope for all offine ops table
const char* kPendingHdrsTableKind = "ns:msg:db:table:kind:pending";
struct mdbOid gAllPendingHdrsTableOID;
nsImapMailDatabase::nsImapMailDatabase()
: m_pendingHdrsRowScopeToken(0), m_pendingHdrsTableKindToken(0) {}
nsImapMailDatabase::~nsImapMailDatabase() {}
NS_IMETHODIMP nsImapMailDatabase::GetSummaryValid(bool* aResult) {
NS_ENSURE_ARG_POINTER(aResult);
if (m_dbFolderInfo) {
uint32_t version;
m_dbFolderInfo->GetVersion(&version);
*aResult = (GetCurVersion() == version);
} else
*aResult = false;
return NS_OK;
}
NS_IMETHODIMP nsImapMailDatabase::SetSummaryValid(bool valid) {
if (m_dbFolderInfo) {
m_dbFolderInfo->SetVersion(valid ? GetCurVersion() : 0);
Commit(nsMsgDBCommitType::kLargeCommit);
}
return NS_OK;
}
// We override this to avoid our parent class (nsMailDatabase)'s
// grabbing of the folder semaphore, and bailing on failure.
NS_IMETHODIMP nsImapMailDatabase::DeleteMessages(
nsTArray<nsMsgKey> const& nsMsgKeys, nsIDBChangeListener* instigator) {
return nsMsgDatabase::DeleteMessages(nsMsgKeys, instigator);
}
nsresult nsImapMailDatabase::AdjustExpungedBytesOnDelete(nsIMsgDBHdr* msgHdr) {
uint32_t msgFlags;
msgHdr->GetFlags(&msgFlags);
if (msgFlags & nsMsgMessageFlags::Offline && m_dbFolderInfo) {
uint32_t size = 0;
(void)msgHdr->GetOfflineMessageSize(&size);
return m_dbFolderInfo->ChangeExpungedBytes(size);
}
return NS_OK;
}
NS_IMETHODIMP nsImapMailDatabase::ForceClosed() {
m_mdbAllPendingHdrsTable = nullptr;
return nsMailDatabase::ForceClosed();
}
nsresult nsImapMailDatabase::GetAllPendingHdrsTable() {
nsresult rv = NS_OK;
if (!m_mdbAllPendingHdrsTable)
rv = GetTableCreateIfMissing(kPendingHdrsScope, kPendingHdrsTableKind,
getter_AddRefs(m_mdbAllPendingHdrsTable),
m_pendingHdrsRowScopeToken,
m_pendingHdrsTableKindToken);
return rv;
}
NS_IMETHODIMP nsImapMailDatabase::AddNewHdrToDB(nsIMsgDBHdr* newHdr,
bool notify) {
nsresult rv = nsMsgDatabase::AddNewHdrToDB(newHdr, notify);
if (NS_SUCCEEDED(rv)) rv = UpdatePendingAttributes(newHdr);
return rv;
}
NS_IMETHODIMP nsImapMailDatabase::UpdatePendingAttributes(
nsIMsgDBHdr* aNewHdr) {
nsresult rv = GetAllPendingHdrsTable();
NS_ENSURE_SUCCESS(rv, rv);
mdb_count numPendingHdrs = 0;
m_mdbAllPendingHdrsTable->GetCount(GetEnv(), &numPendingHdrs);
if (numPendingHdrs > 0) {
mdbYarn messageIdYarn;
nsCOMPtr<nsIMdbRow> pendingRow;
mdbOid outRowId;
nsCString messageId;
aNewHdr->GetMessageId(messageId);
messageIdYarn.mYarn_Buf = (void*)messageId.get();
messageIdYarn.mYarn_Fill = messageId.Length();
messageIdYarn.mYarn_Form = 0;
messageIdYarn.mYarn_Size = messageIdYarn.mYarn_Fill;
m_mdbStore->FindRow(GetEnv(), m_pendingHdrsRowScopeToken,
m_messageIdColumnToken, &messageIdYarn, &outRowId,
getter_AddRefs(pendingRow));
if (pendingRow) {
mdb_count numCells;
mdbYarn cellYarn;
mdb_column cellColumn;
uint32_t existingFlags;
pendingRow->GetCount(GetEnv(), &numCells);
aNewHdr->GetFlags(&existingFlags);
// iterate over the cells in the pending hdr setting properties on the
// aNewHdr. we skip cell 0, which is the messageId;
nsMsgHdr* msgHdr =
static_cast<nsMsgHdr*>(aNewHdr); // closed system, cast ok
nsIMdbRow* row = msgHdr->GetMDBRow();
for (mdb_count cellIndex = 1; cellIndex < numCells; cellIndex++) {
nsresult err =
pendingRow->SeekCellYarn(GetEnv(), cellIndex, &cellColumn, nullptr);
if (NS_SUCCEEDED(err)) {
err = pendingRow->AliasCellYarn(GetEnv(), cellColumn, &cellYarn);
if (NS_SUCCEEDED(err)) {
if (row) row->AddColumn(GetEnv(), cellColumn, &cellYarn);
}
}
}
// We might have changed some cached values, so force a refresh.
msgHdr->ClearCachedValues();
uint32_t resultFlags;
msgHdr->OrFlags(existingFlags, &resultFlags);
m_mdbAllPendingHdrsTable->CutRow(GetEnv(), pendingRow);
pendingRow->CutAllColumns(GetEnv());
}
}
return rv;
}
nsresult nsImapMailDatabase::GetRowForPendingHdr(nsIMsgDBHdr* pendingHdr,
nsIMdbRow** row) {
nsresult rv = GetAllPendingHdrsTable();
NS_ENSURE_SUCCESS(rv, rv);
mdbYarn messageIdYarn;
nsCOMPtr<nsIMdbRow> pendingRow;
mdbOid outRowId;
nsCString messageId;
pendingHdr->GetMessageId(messageId);
messageIdYarn.mYarn_Buf = (void*)messageId.get();
messageIdYarn.mYarn_Fill = messageId.Length();
messageIdYarn.mYarn_Form = 0;
messageIdYarn.mYarn_Size = messageIdYarn.mYarn_Fill;
rv = m_mdbStore->FindRow(GetEnv(), m_pendingHdrsRowScopeToken,
m_messageIdColumnToken, &messageIdYarn, &outRowId,
getter_AddRefs(pendingRow));
if (!pendingRow)
rv = m_mdbStore->NewRow(GetEnv(), m_pendingHdrsRowScopeToken,
getter_AddRefs(pendingRow));
NS_ENSURE_SUCCESS(rv, rv);
if (pendingRow) {
// now we need to add cells to the row to remember the messageid, property
// and property value, and flags. Then, when hdrs are added to the db, we'll
// check if they have a matching message-id, and if so, set the property and
// flags
// XXX we already fetched messageId from the pending hdr, could it have
// changed by the time we get here?
nsCString messageId;
pendingHdr->GetMessageId(messageId);
// we're just going to ignore messages without a message-id. They should be
// rare. If SPAM messages often didn't have message-id's, they'd be filtered
// on the server, most likely, and spammers would then start putting in
// message-id's.
if (!messageId.IsEmpty()) {
extern const char* kMessageIdColumnName;
m_mdbAllPendingHdrsTable->AddRow(GetEnv(), pendingRow);
// make sure this is the first cell so that when we ignore the first
// cell in nsImapMailDatabase::AddNewHdrToDB, we're ignoring the right one
(void)SetProperty(pendingRow, kMessageIdColumnName, messageId.get());
pendingRow.forget(row);
} else
return NS_ERROR_FAILURE;
}
return rv;
}
NS_IMETHODIMP nsImapMailDatabase::SetAttributeOnPendingHdr(
nsIMsgDBHdr* pendingHdr, const char* property, const char* propertyVal) {
NS_ENSURE_ARG_POINTER(pendingHdr);
nsCOMPtr<nsIMdbRow> pendingRow;
nsresult rv = GetRowForPendingHdr(pendingHdr, getter_AddRefs(pendingRow));
NS_ENSURE_SUCCESS(rv, rv);
return SetProperty(pendingRow, property, propertyVal);
}
NS_IMETHODIMP
nsImapMailDatabase::SetUint32AttributeOnPendingHdr(nsIMsgDBHdr* pendingHdr,
const char* property,
uint32_t propertyVal) {
NS_ENSURE_ARG_POINTER(pendingHdr);
nsCOMPtr<nsIMdbRow> pendingRow;
nsresult rv = GetRowForPendingHdr(pendingHdr, getter_AddRefs(pendingRow));
NS_ENSURE_SUCCESS(rv, rv);
return SetUint32Property(pendingRow, property, propertyVal);
}
NS_IMETHODIMP
nsImapMailDatabase::SetUint64AttributeOnPendingHdr(nsIMsgDBHdr* aPendingHdr,
const char* aProperty,
uint64_t aPropertyVal) {
NS_ENSURE_ARG_POINTER(aPendingHdr);
nsCOMPtr<nsIMdbRow> pendingRow;
nsresult rv = GetRowForPendingHdr(aPendingHdr, getter_AddRefs(pendingRow));
NS_ENSURE_SUCCESS(rv, rv);
return SetUint64Property(pendingRow, aProperty, aPropertyVal);
}