Source code
Revision control
Copy as Markdown
Other Tools
/* vim: set ts=2 et sw=2 tw=80: */
/* 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 "nsIMsgAccountManager.h"
#include "nsServiceManagerUtils.h"
#include "nsIINIParser.h"
#include "nsISmtpService.h"
#include "nsISmtpServer.h"
#include "nsIPop3IncomingServer.h"
#include "nsNetUtil.h"
#include "nsString.h"
#include "msgCore.h"
#include "nsBeckySettings.h"
#include "nsBeckyStringBundle.h"
#include "nsBeckyUtils.h"
NS_IMPL_ISUPPORTS(nsBeckySettings, nsIImportSettings)
nsresult nsBeckySettings::Create(nsIImportSettings** aImport) {
NS_ENSURE_ARG_POINTER(aImport);
NS_ADDREF(*aImport = new nsBeckySettings());
return NS_OK;
}
nsBeckySettings::nsBeckySettings() {}
nsBeckySettings::~nsBeckySettings() {}
NS_IMETHODIMP
nsBeckySettings::AutoLocate(char16_t** aDescription, nsIFile** aLocation,
bool* _retval) {
NS_ENSURE_ARG_POINTER(aDescription);
NS_ENSURE_ARG_POINTER(aLocation);
NS_ENSURE_ARG_POINTER(_retval);
*aDescription = nsBeckyStringBundle::GetStringByName("BeckyImportName");
*aLocation = nullptr;
*_retval = false;
nsCOMPtr<nsIFile> location;
nsresult rv =
nsBeckyUtils::GetDefaultMailboxINIFile(getter_AddRefs(location));
if (NS_FAILED(rv))
location = do_CreateInstance(NS_LOCAL_FILE_CONTRACTID, &rv);
else
*_retval = true;
location.forget(aLocation);
return NS_OK;
}
NS_IMETHODIMP
nsBeckySettings::SetLocation(nsIFile* aLocation) {
mLocation = aLocation;
return NS_OK;
}
nsresult nsBeckySettings::CreateParser() {
if (!mLocation) {
nsresult rv =
nsBeckyUtils::GetDefaultMailboxINIFile(getter_AddRefs(mLocation));
NS_ENSURE_SUCCESS(rv, rv);
}
// nsIINIParser accepts only UTF-8 encoding, so we need to convert the file
// first.
nsresult rv;
rv = nsBeckyUtils::ConvertToUTF8File(mLocation,
getter_AddRefs(mConvertedFile));
NS_ENSURE_SUCCESS(rv, rv);
return nsBeckyUtils::CreateINIParserForFile(mConvertedFile,
getter_AddRefs(mParser));
}
nsresult nsBeckySettings::CreateSmtpServer(const nsCString& aUserName,
const nsCString& aServerName,
nsISmtpServer** aServer,
bool* existing) {
nsresult rv;
nsCOMPtr<nsISmtpService> smtpService =
do_GetService("@mozilla.org/messengercompose/smtp;1", &rv);
NS_ENSURE_SUCCESS(rv, rv);
nsCOMPtr<nsISmtpServer> server;
rv = smtpService->FindServer(aUserName, aServerName, getter_AddRefs(server));
if (NS_FAILED(rv) || !server) {
rv = smtpService->CreateServer(getter_AddRefs(server));
NS_ENSURE_SUCCESS(rv, rv);
server->SetHostname(aServerName);
server->SetUsername(aUserName);
*existing = false;
} else {
*existing = true;
}
server.forget(aServer);
return NS_OK;
}
nsresult nsBeckySettings::CreateIncomingServer(const nsCString& aUserName,
const nsCString& aServerName,
const nsCString& aProtocol,
nsIMsgIncomingServer** aServer) {
nsresult rv;
nsCOMPtr<nsIMsgAccountManager> accountManager =
do_GetService("@mozilla.org/messenger/account-manager;1", &rv);
NS_ENSURE_SUCCESS(rv, rv);
nsCOMPtr<nsIMsgIncomingServer> incomingServer;
accountManager->FindServer(aUserName, aServerName, aProtocol, 0,
getter_AddRefs(incomingServer));
if (!incomingServer) {
rv = accountManager->CreateIncomingServer(aUserName, aServerName, aProtocol,
getter_AddRefs(incomingServer));
NS_ENSURE_SUCCESS(rv, rv);
}
incomingServer.forget(aServer);
return NS_OK;
}
nsresult nsBeckySettings::SetupSmtpServer(nsISmtpServer** aServer) {
nsresult rv;
nsAutoCString userName, serverName;
mParser->GetString("Account"_ns, "SMTPServer"_ns, serverName);
mParser->GetString("Account"_ns, "UserID"_ns, userName);
nsCOMPtr<nsISmtpServer> server;
bool existing = false;
rv =
CreateSmtpServer(userName, serverName, getter_AddRefs(server), &existing);
NS_ENSURE_SUCCESS(rv, rv);
// If we already have an existing server, do not touch it's settings.
if (existing) {
server.forget(aServer);
return NS_OK;
}
nsAutoCString value;
rv = mParser->GetString("Account"_ns, "SMTPPort"_ns, value);
int32_t port = 25;
if (NS_SUCCEEDED(rv)) {
nsresult errorCode;
port = value.ToInteger(&errorCode, 10);
}
server->SetPort(port);
mParser->GetString("Account"_ns, "SSLSMTP"_ns, value);
if (value.EqualsLiteral("1")) server->SetSocketType(nsMsgSocketType::SSL);
mParser->GetString("Account"_ns, "SMTPAUTH"_ns, value);
if (value.EqualsLiteral("1")) {
mParser->GetString("Account"_ns, "SMTPAUTHMODE"_ns, value);
nsMsgAuthMethodValue authMethod = nsMsgAuthMethod::none;
if (value.EqualsLiteral("1")) {
authMethod = nsMsgAuthMethod::passwordEncrypted;
} else if (value.EqualsLiteral("2") || value.EqualsLiteral("4") ||
value.EqualsLiteral("6")) {
authMethod = nsMsgAuthMethod::passwordCleartext;
} else {
authMethod = nsMsgAuthMethod::anything;
}
server->SetAuthMethod(authMethod);
}
server.forget(aServer);
return NS_OK;
}
nsresult nsBeckySettings::SetPop3ServerProperties(
nsIMsgIncomingServer* aServer) {
nsCOMPtr<nsIPop3IncomingServer> pop3Server = do_QueryInterface(aServer);
nsAutoCString value;
mParser->GetString("Account"_ns, "POP3Auth"_ns,
value); // 0: plain, 1: APOP, 2: CRAM-MD5, 3: NTLM
nsMsgAuthMethodValue authMethod;
if (value.IsEmpty() || value.EqualsLiteral("0")) {
authMethod = nsMsgAuthMethod::passwordCleartext;
} else if (value.EqualsLiteral("1")) {
authMethod = nsMsgAuthMethod::old;
} else if (value.EqualsLiteral("2")) {
authMethod = nsMsgAuthMethod::passwordEncrypted;
} else if (value.EqualsLiteral("3")) {
authMethod = nsMsgAuthMethod::NTLM;
} else {
authMethod = nsMsgAuthMethod::none;
}
aServer->SetAuthMethod(authMethod);
mParser->GetString("Account"_ns, "LeaveServer"_ns, value);
if (value.EqualsLiteral("1")) {
pop3Server->SetLeaveMessagesOnServer(true);
nsresult rv = mParser->GetString("Account"_ns, "KeepDays"_ns, value);
if (NS_FAILED(rv)) return NS_OK;
nsresult errorCode;
int32_t leftDays = value.ToInteger(&errorCode, 10);
if (NS_SUCCEEDED(errorCode)) {
pop3Server->SetNumDaysToLeaveOnServer(leftDays);
pop3Server->SetDeleteByAgeFromServer(true);
}
}
return NS_OK;
}
nsresult nsBeckySettings::SetupIncomingServer(nsIMsgIncomingServer** aServer) {
nsAutoCString value;
mParser->GetString("Account"_ns, "Protocol"_ns, value);
nsCString protocol;
if (value.EqualsLiteral("1")) {
protocol = "imap"_ns;
} else {
protocol = "pop3"_ns;
}
nsAutoCString userName, serverName;
mParser->GetString("Account"_ns, "MailServer"_ns, serverName);
mParser->GetString("Account"_ns, "UserID"_ns, userName);
nsresult rv;
nsCOMPtr<nsIMsgIncomingServer> server;
rv = CreateIncomingServer(userName, serverName, protocol,
getter_AddRefs(server));
NS_ENSURE_SUCCESS(rv, rv);
bool isSecure = false;
int32_t port = 0;
nsresult errorCode;
if (protocol.EqualsLiteral("pop3")) {
SetPop3ServerProperties(server);
rv = mParser->GetString("Account"_ns, "POP3Port"_ns, value);
if (NS_SUCCEEDED(rv))
port = value.ToInteger(&errorCode, 10);
else
port = 110;
mParser->GetString("Account"_ns, "SSLPOP"_ns, value);
if (value.EqualsLiteral("1")) isSecure = true;
} else if (protocol.EqualsLiteral("imap")) {
rv = mParser->GetString("Account"_ns, "IMAP4Port"_ns, value);
if (NS_SUCCEEDED(rv))
port = value.ToInteger(&errorCode, 10);
else
port = 143;
mParser->GetString("Account"_ns, "SSLIMAP"_ns, value);
if (value.EqualsLiteral("1")) isSecure = true;
}
server->SetPort(port);
if (isSecure) server->SetSocketType(nsMsgSocketType::SSL);
mParser->GetString("Account"_ns, "CheckInt"_ns, value);
if (value.EqualsLiteral("1")) server->SetDoBiff(true);
rv = mParser->GetString("Account"_ns, "CheckEvery"_ns, value);
if (NS_SUCCEEDED(rv)) {
int32_t minutes = value.ToInteger(&errorCode, 10);
if (NS_SUCCEEDED(errorCode)) server->SetBiffMinutes(minutes);
}
server.forget(aServer);
return NS_OK;
}
nsresult nsBeckySettings::CreateIdentity(nsIMsgIdentity** aIdentity) {
nsAutoCString email, fullName, identityName, bccAddress;
mParser->GetString("Account"_ns, "Name"_ns, identityName);
mParser->GetString("Account"_ns, "YourName"_ns, fullName);
mParser->GetString("Account"_ns, "MailAddress"_ns, email);
mParser->GetString("Account"_ns, "PermBcc"_ns, bccAddress);
nsresult rv;
nsCOMPtr<nsIMsgAccountManager> accountManager =
do_GetService("@mozilla.org/messenger/account-manager;1", &rv);
NS_ENSURE_SUCCESS(rv, rv);
nsCOMPtr<nsIMsgIdentity> identity;
rv = accountManager->CreateIdentity(getter_AddRefs(identity));
NS_ENSURE_SUCCESS(rv, rv);
identity->SetLabel(NS_ConvertUTF8toUTF16(identityName));
identity->SetFullName(NS_ConvertUTF8toUTF16(fullName));
identity->SetEmail(email);
if (!bccAddress.IsEmpty()) {
identity->SetDoBcc(true);
identity->SetDoBccList(bccAddress);
}
identity.forget(aIdentity);
return NS_OK;
}
nsresult nsBeckySettings::CreateAccount(nsIMsgIdentity* aIdentity,
nsIMsgIncomingServer* aIncomingServer,
nsIMsgAccount** aAccount) {
nsresult rv;
nsCOMPtr<nsIMsgAccountManager> accountManager =
do_GetService("@mozilla.org/messenger/account-manager;1", &rv);
NS_ENSURE_SUCCESS(rv, rv);
nsCOMPtr<nsIMsgAccount> account;
rv = accountManager->CreateAccount(getter_AddRefs(account));
NS_ENSURE_SUCCESS(rv, rv);
rv = account->AddIdentity(aIdentity);
NS_ENSURE_SUCCESS(rv, rv);
rv = account->SetIncomingServer(aIncomingServer);
NS_ENSURE_SUCCESS(rv, rv);
account.forget(aAccount);
return NS_OK;
}
nsresult nsBeckySettings::RemoveConvertedFile() {
if (mConvertedFile) {
bool exists;
mConvertedFile->Exists(&exists);
if (exists) mConvertedFile->Remove(false);
mConvertedFile = nullptr;
}
return NS_OK;
}
#define NS_RETURN_IF_FAILED_WITH_REMOVE_CONVERTED_FILE(expr, rv) \
if (NS_FAILED(expr)) { \
RemoveConvertedFile(); \
return rv; \
}
NS_IMETHODIMP
nsBeckySettings::Import(nsIMsgAccount** aLocalMailAccount, bool* _retval) {
NS_ENSURE_ARG_POINTER(aLocalMailAccount);
NS_ENSURE_ARG_POINTER(_retval);
nsresult rv = CreateParser();
NS_RETURN_IF_FAILED_WITH_REMOVE_CONVERTED_FILE(rv, rv);
nsCOMPtr<nsIMsgIncomingServer> incomingServer;
rv = SetupIncomingServer(getter_AddRefs(incomingServer));
NS_RETURN_IF_FAILED_WITH_REMOVE_CONVERTED_FILE(rv, rv);
nsCOMPtr<nsISmtpServer> smtpServer;
rv = SetupSmtpServer(getter_AddRefs(smtpServer));
NS_RETURN_IF_FAILED_WITH_REMOVE_CONVERTED_FILE(rv, rv);
nsCOMPtr<nsIMsgIdentity> identity;
rv = CreateIdentity(getter_AddRefs(identity));
NS_RETURN_IF_FAILED_WITH_REMOVE_CONVERTED_FILE(rv, rv);
nsAutoCString smtpKey;
smtpServer->GetKey(getter_Copies(smtpKey));
identity->SetSmtpServerKey(smtpKey);
nsCOMPtr<nsIMsgAccount> account;
rv = CreateAccount(identity, incomingServer, getter_AddRefs(account));
NS_RETURN_IF_FAILED_WITH_REMOVE_CONVERTED_FILE(rv, rv);
RemoveConvertedFile();
if (aLocalMailAccount) account.forget(aLocalMailAccount);
*_retval = true;
return NS_OK;
}