pkw%us.ibm.com b900d5843e Bug 270248 - Cleanup string usage in wallet code
r=cbiesinger@gmx.at, sr=darin@meer.net


git-svn-id: svn://10.0.0.236/trunk@166760 18797224-902f-48f8-a5cc-f745e15eee43
2004-12-15 19:29:54 +00:00

3843 lines
124 KiB
C++

/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* ***** BEGIN LICENSE BLOCK *****
* Version: MPL 1.1/GPL 2.0/LGPL 2.1
*
* The contents of this file are subject to the Mozilla Public License Version
* 1.1 (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
* http://www.mozilla.org/MPL/
*
* Software distributed under the License is distributed on an "AS IS" basis,
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
* for the specific language governing rights and limitations under the
* License.
*
* The Original Code is Mozilla Communicator client code.
*
* The Initial Developer of the Original Code is
* Netscape Communications Corporation.
* Portions created by the Initial Developer are Copyright (C) 1998
* the Initial Developer. All Rights Reserved.
*
* Contributor(s):
*
* Alternatively, the contents of this file may be used under the terms of
* either the GNU General Public License Version 2 or later (the "GPL"), or
* the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
* in which case the provisions of the GPL or the LGPL are applicable instead
* of those above. If you wish to allow use of your version of this file only
* under the terms of either the GPL or the LGPL, and not to allow others to
* use your version of this file under the terms of the MPL, indicate your
* decision by deleting the provisions above and replace them with the notice
* and other provisions required by the GPL or the LGPL. If you do not delete
* the provisions above, a recipient may use your version of this file under
* the terms of any one of the MPL, the GPL or the LGPL.
*
* ***** END LICENSE BLOCK ***** */
/*
wallet.cpp
*/
#include "wallet.h"
#include "singsign.h"
#include "nsNetUtil.h"
#include "nsILineInputStream.h"
#include "nsReadableUtils.h"
#include "nsUnicharUtils.h"
#include "nsILocalFile.h"
#include "nsIServiceManager.h"
#include "nsIDocument.h"
#include "nsIDOMHTMLDocument.h"
#include "nsIDOMHTMLCollection.h"
#include "nsIDOMHTMLFormElement.h"
#include "nsIDOMHTMLInputElement.h"
#include "nsIDOMHTMLSelectElement.h"
#include "nsIDOMHTMLOptionElement.h"
#include "nsIURL.h"
#include "nsIDOMWindowCollection.h"
#include "nsIPrompt.h"
#include "nsIWindowWatcher.h"
#include "nsAppDirectoryServiceDefs.h"
#include "nsIStringBundle.h"
#include "prmem.h"
#include "prprf.h"
#include "nsIContent.h"
#include "nsIObserverService.h"
#include "nsIWalletService.h"
#include <time.h>
#include "prlong.h"
#include "prinrval.h"
#include "prlog.h"
//
// To enable logging (see prlog.h for full details):
//
// set NSPR_LOG_MODULES=nsWallet:5
// set NSPR_LOG_FILE=nspr.log
//
PRLogModuleInfo* gWalletLog = nsnull;
/********************************************************/
/* The following data and procedures are for preference */
/********************************************************/
static const char pref_Caveat[] = "wallet.caveat";
static const char pref_captureForms[] = "wallet.captureForms";
static const char pref_enabled[] = "wallet.enabled";
static const char pref_WalletSchemaValueFileName[] = "wallet.SchemaValueFileName";
static PRBool wallet_captureForms = PR_FALSE;
static void
wallet_SetFormsCapturingPref(PRBool x)
{
/* do nothing if new value of pref is same as current value */
if (x == wallet_captureForms) {
return;
}
/* change the pref */
wallet_captureForms = x;
}
int PR_CALLBACK
wallet_FormsCapturingPrefChanged(const char * newpref, void * data)
{
PRBool x;
x = SI_GetBoolPref(pref_captureForms, PR_TRUE);
wallet_SetFormsCapturingPref(x);
return 0; /* this is PREF_NOERROR but we no longer include prefapi.h */
}
static void
wallet_RegisterCapturePrefCallbacks(void)
{
PRBool x;
static PRBool first_time = PR_TRUE;
if(first_time)
{
first_time = PR_FALSE;
x = SI_GetBoolPref(pref_captureForms, PR_TRUE);
wallet_SetFormsCapturingPref(x);
SI_RegisterCallback(pref_captureForms, wallet_FormsCapturingPrefChanged, NULL);
}
}
static PRBool
wallet_GetFormsCapturingPref(void)
{
wallet_RegisterCapturePrefCallbacks();
return wallet_captureForms;
}
static PRBool
wallet_GetEnabledPref(void)
{
/* This pref is not in the prefs panel. It's purpose is to remove wallet from all UI */
static PRBool first_time = PR_TRUE;
static PRBool enabled = PR_TRUE;
if (first_time) {
first_time = PR_FALSE;
PRBool x = SI_GetBoolPref(pref_enabled, PR_TRUE);
enabled = x;
}
return enabled;
}
/***************************************************/
/* The following declarations define the data base */
/***************************************************/
#define WALLET_FREE(_ptr) { nsMemory::Free((void*)_ptr); (_ptr) = nsnull; }
#define WALLET_FREEIF(_ptr) if (_ptr) WALLET_FREE(_ptr)
enum PlacementType {DUP_IGNORE, DUP_OVERWRITE, DUP_BEFORE, DUP_AFTER, AT_END, BY_LENGTH};
#define LIST_COUNT(list) ((list) ? (list)->Count() : 0)
MOZ_DECL_CTOR_COUNTER(wallet_Sublist)
class wallet_Sublist {
public:
wallet_Sublist()
{
MOZ_COUNT_CTOR(wallet_Sublist);
}
~wallet_Sublist()
{
WALLET_FREEIF(item);
MOZ_COUNT_DTOR(wallet_Sublist);
}
const char* item;
};
/*
* The data structure below consists of mapping tables that map one item into another.
* The actual interpretation of the items depend on which table we are in. For
* example, if in the field-to-schema table, item1 is a field name and item2 is a
* schema name. Whereas in the schema-to-value table, item1 is a schema name and
* item2 is a value. Therefore this generic data structure refers to them simply as
* item1 and item2.
*/
MOZ_DECL_CTOR_COUNTER(wallet_MapElement)
class wallet_MapElement {
public:
wallet_MapElement() : itemList(nsnull)
{
MOZ_COUNT_CTOR(wallet_MapElement);
}
~wallet_MapElement()
{
WALLET_FREEIF(item1);
WALLET_FREEIF(item2);
if (itemList) {
PRInt32 count = LIST_COUNT(itemList);
wallet_Sublist * sublistPtr;
for (PRInt32 i=0; i<count; i++) {
sublistPtr = NS_STATIC_CAST(wallet_Sublist*, itemList->ElementAt(i));
delete sublistPtr;
}
delete itemList;
}
MOZ_COUNT_DTOR(wallet_MapElement);
}
const char* item1;
const char* item2;
nsVoidArray * itemList;
};
/* Purpose of this class is to speed up startup time on the mac
*
* These strings are used over and over again inside an inner loop. Rather
* then allocating them and then deallocating them, they will be allocated
* only once and left sitting on the heap
*/
MOZ_DECL_CTOR_COUNTER(wallet_HelpMac)
class wallet_HelpMac {
public:
wallet_HelpMac() {
MOZ_COUNT_CTOR(wallet_HelpMac);
}
~wallet_HelpMac() {
MOZ_COUNT_DTOR(wallet_HelpMac);
}
nsCString item1;
nsCString item2;
nsCString item3;
};
wallet_HelpMac * helpMac;
static nsVoidArray * wallet_FieldToSchema_list = 0;
static nsVoidArray * wallet_VcardToSchema_list = 0;
static nsVoidArray * wallet_SchemaToValue_list = 0;
static nsVoidArray * wallet_SchemaConcat_list = 0;
static nsVoidArray * wallet_SchemaStrings_list = 0;
static nsVoidArray * wallet_PositionalSchema_list = 0;
static nsVoidArray * wallet_StateSchema_list = 0;
static nsVoidArray * wallet_URL_list = 0;
static nsVoidArray * wallet_DistinguishedSchema_list = 0;
#define NO_CAPTURE(x) x[0]
#define NO_PREVIEW(x) x[1]
MOZ_DECL_CTOR_COUNTER(wallet_PrefillElement)
class wallet_PrefillElement {
public:
wallet_PrefillElement() : inputElement(nsnull), selectElement(nsnull)
{
schema = nsnull;
MOZ_COUNT_CTOR(wallet_PrefillElement);
}
~wallet_PrefillElement()
{
WALLET_FREEIF(schema);
NS_IF_RELEASE(inputElement);
NS_IF_RELEASE(selectElement);
MOZ_COUNT_DTOR(wallet_PrefillElement);
}
nsIDOMHTMLInputElement* inputElement;
nsIDOMHTMLSelectElement* selectElement;
char* schema;
nsString value;
PRInt32 selectIndex;
PRUint32 count;
};
nsIURI * wallet_lastUrl = NULL;
/***********************************************************/
/* The following routines are for diagnostic purposes only */
/***********************************************************/
#ifdef DEBUG_morse
static void
wallet_Pause(){
fprintf(stdout,"%cpress y to continue\n", '\007');
char c;
for (;;) {
c = getchar();
if (tolower(c) == 'y') {
fprintf(stdout,"OK\n");
break;
}
}
while (c != '\n') {
c = getchar();
}
}
static void
wallet_DumpAutoString(const nsString& as){
fprintf(stdout, "%s\n", NS_LossyConvertUCS2toASCII(as).get());
}
static void
wallet_Dump(nsVoidArray * list) {
wallet_MapElement * mapElementPtr;
PRInt32 count = LIST_COUNT(list);
for (PRInt32 i=0; i<count; i++) {
mapElementPtr = NS_STATIC_CAST(wallet_MapElement*, list->ElementAt(i));
fprintf(stdout, "%s %s \n", (mapElementPtr->item1), (mapElementPtr->item2));
wallet_Sublist * sublistPtr;
PRInt32 count2 = LIST_COUNT(mapElementPtr->itemList);
for (PRInt32 i2=0; i2<count2; i2++) {
sublistPtr = NS_STATIC_CAST(wallet_Sublist*, mapElementPtr->itemList->ElementAt(i2));
fprintf(stdout, " %s \n", (sublistPtr->item));
}
}
wallet_Pause();
}
/******************************************************************/
/* The following diagnostic routines are for timing purposes only */
/******************************************************************/
const PRInt32 timing_max = 1000;
PRInt64 timings [timing_max];
char timingID [timing_max];
PRInt32 timing_index = 0;
PRInt64 stopwatch = LL_Zero();
PRInt64 stopwatchBase;
PRBool stopwatchRunning = PR_FALSE;
static void
wallet_ClearTiming() {
timing_index = 0;
LL_I2L(timings[timing_index++], PR_IntervalNow());
}
static void
wallet_DumpTiming() {
PRInt32 i, r4;
PRInt64 r1, r2, r3;
for (i=1; i<timing_index; i++) {
LL_SUB(r1, timings[i], timings[i-1]);
LL_I2L(r2, 100);
LL_DIV(r3, r1, r2);
LL_L2I(r4, r3);
fprintf(stdout, "time %c = %ld\n", timingID[i], (long)r4);
if (i%20 == 0) {
wallet_Pause();
}
}
wallet_Pause();
}
static void
wallet_AddTiming(char c) {
if (timing_index<timing_max) {
timingID[timing_index] = c;
// note: PR_IntervalNow returns a 32 bit value!
LL_I2L(timings[timing_index++], PR_IntervalNow());
}
}
static void
wallet_ClearStopwatch() {
stopwatch = LL_Zero();
stopwatchRunning = PR_FALSE;
}
static void
wallet_ResumeStopwatch() {
if (!stopwatchRunning) {
// note: PR_IntervalNow returns a 32 bit value!
LL_I2L(stopwatchBase, PR_IntervalNow());
stopwatchRunning = PR_TRUE;
}
}
static void
wallet_PauseStopwatch() {
PRInt64 r1, r2;
if (stopwatchRunning) {
// note: PR_IntervalNow returns a 32 bit value!
LL_I2L(r1, PR_IntervalNow());
LL_SUB(r2, r1, stopwatchBase);
LL_ADD(stopwatch, stopwatch, r2);
stopwatchRunning = PR_FALSE;
}
}
static void
wallet_DumpStopwatch() {
PRInt64 r1, r2;
PRInt32 r3;
if (stopwatchRunning) {
// note: PR_IntervalNow returns a 32 bit value!
LL_I2L(r1, PR_IntervalNow());
LL_SUB(r2, r1, stopwatchBase);
LL_ADD(stopwatch, stopwatch, r2);
LL_I2L(stopwatchBase, PR_IntervalNow());
}
LL_I2L(r1, 100);
LL_DIV(r2, stopwatch, r1);
LL_L2I(r3, r2);
fprintf(stdout, "stopwatch = %ld\n", (long)r3);
}
#endif /* DEBUG_morse */
/*************************************************************************/
/* The following routines are used for accessing strings to be localized */
/*************************************************************************/
#define PROPERTIES_URL "chrome://communicator/locale/wallet/wallet.properties"
PRUnichar *
Wallet_Localize(const char* genericString) {
nsresult ret;
nsAutoString v;
/* create a bundle for the localization */
nsCOMPtr<nsIStringBundleService> pStringService = do_GetService(NS_STRINGBUNDLE_CONTRACTID, &ret);
if (NS_FAILED(ret)) {
#ifdef DEBUG
printf("cannot get string service\n");
#endif
return ToNewUnicode(v);
}
nsCOMPtr<nsIStringBundle> bundle;
ret = pStringService->CreateBundle(PROPERTIES_URL, getter_AddRefs(bundle));
if (NS_FAILED(ret)) {
#ifdef DEBUG
printf("cannot create instance\n");
#endif
return ToNewUnicode(v);
}
/* localize the given string */
NS_ConvertASCIItoUTF16 strtmp(genericString);
PRUnichar *ptrv = nsnull;
ret = bundle->GetStringFromName(strtmp.get(), &ptrv);
if (NS_FAILED(ret)) {
#ifdef DEBUG
printf("cannot get string from name\n");
#endif
return ToNewUnicode(v);
}
v = ptrv;
nsCRT::free(ptrv);
/* convert # to newlines */
PRUint32 i;
for (i=0; i<v.Length(); i++) {
if (v.CharAt(i) == '#') {
v.SetCharAt('\n', i);
}
}
return ToNewUnicode(v);
}
/**********************/
/* Modal dialog boxes */
/**********************/
PRBool
Wallet_Confirm(PRUnichar * szMessage, nsIDOMWindowInternal* window)
{
PRBool retval = PR_TRUE; /* default value */
nsresult res;
nsCOMPtr<nsIPrompt> dialog;
window->GetPrompter(getter_AddRefs(dialog));
if (!dialog) {
return retval;
}
const nsAutoString message( szMessage );
retval = PR_FALSE; /* in case user exits dialog by clicking X */
res = dialog->Confirm(nsnull, message.get(), &retval);
return retval;
}
PRBool
Wallet_ConfirmYN(PRUnichar * szMessage, nsIDOMWindowInternal* window) {
nsresult res;
nsCOMPtr<nsIPrompt> dialog;
window->GetPrompter(getter_AddRefs(dialog));
if (!dialog) {
return PR_FALSE;
}
PRInt32 buttonPressed = 1; /* in case user exits dialog by clickin X */
PRUnichar * confirm_string = Wallet_Localize("Confirm");
res = dialog->ConfirmEx(confirm_string, szMessage,
(nsIPrompt::BUTTON_TITLE_YES * nsIPrompt::BUTTON_POS_0) +
(nsIPrompt::BUTTON_TITLE_NO * nsIPrompt::BUTTON_POS_1),
nsnull, nsnull, nsnull, nsnull, nsnull, &buttonPressed);
WALLET_FREE(confirm_string);
return (buttonPressed == 0);
}
PRInt32
Wallet_3ButtonConfirm(PRUnichar * szMessage, nsIDOMWindowInternal* window)
{
nsresult res;
nsCOMPtr<nsIPrompt> dialog;
window->GetPrompter(getter_AddRefs(dialog));
if (!dialog) {
return 0; /* default value is NO */
}
PRInt32 buttonPressed = 1; /* default of NO if user exits dialog by clickin X */
PRUnichar * never_string = Wallet_Localize("Never");
PRUnichar * confirm_string = Wallet_Localize("Confirm");
res = dialog->ConfirmEx(confirm_string, szMessage,
(nsIPrompt::BUTTON_TITLE_YES * nsIPrompt::BUTTON_POS_0) +
(nsIPrompt::BUTTON_TITLE_NO * nsIPrompt::BUTTON_POS_1) +
(nsIPrompt::BUTTON_TITLE_IS_STRING * nsIPrompt::BUTTON_POS_2),
nsnull, nsnull, never_string, nsnull, nsnull, &buttonPressed);
WALLET_FREE(never_string);
WALLET_FREE(confirm_string);
return buttonPressed;
}
static void
wallet_Alert(PRUnichar * szMessage, nsIDOMWindowInternal* window)
{
nsresult res;
nsCOMPtr<nsIPrompt> dialog;
window->GetPrompter(getter_AddRefs(dialog));
if (!dialog) {
return; // XXX should return the error
}
const nsAutoString message( szMessage );
PRUnichar * title = Wallet_Localize("CaveatTitle");
res = dialog->Alert(title, message.get());
WALLET_FREE(title);
return; // XXX should return the error
}
static void
wallet_Alert(PRUnichar * szMessage, nsIPrompt* dialog)
{
nsresult res;
const nsAutoString message( szMessage );
PRUnichar * title = Wallet_Localize("CaveatTitle");
res = dialog->Alert(title, message.get());
WALLET_FREE(title);
return; // XXX should return the error
}
PRBool
Wallet_CheckConfirmYN
(PRUnichar * szMessage, PRUnichar * szCheckMessage, PRBool* checkValue,
nsIDOMWindowInternal* window) {
nsresult res;
nsCOMPtr<nsIPrompt> dialog;
window->GetPrompter(getter_AddRefs(dialog));
if (!dialog) {
return PR_FALSE;
}
PRInt32 buttonPressed = 1; /* in case user exits dialog by clickin X */
PRUnichar * confirm_string = Wallet_Localize("Confirm");
res = dialog->ConfirmEx(confirm_string, szMessage,
(nsIPrompt::BUTTON_TITLE_YES * nsIPrompt::BUTTON_POS_0) +
(nsIPrompt::BUTTON_TITLE_NO * nsIPrompt::BUTTON_POS_1),
nsnull, nsnull, nsnull, szCheckMessage, checkValue, &buttonPressed);
if (NS_FAILED(res)) {
*checkValue = 0;
}
if (*checkValue!=0 && *checkValue!=1) {
NS_ASSERTION(PR_FALSE, "Bad result from checkbox");
*checkValue = 0; /* this should never happen but it is happening!!! */
}
WALLET_FREE(confirm_string);
return (buttonPressed == 0);
}
/*******************************************************/
/* The following routines are for Encyption/Decryption */
/*******************************************************/
#include "nsISecretDecoderRing.h"
nsISecretDecoderRing* gSecretDecoderRing;
PRBool gEncryptionFailure = PR_FALSE;
PRInt32 gReencryptionLevel = 0;
static nsresult
wallet_CryptSetup() {
if (!gSecretDecoderRing)
{
/* Get a secret decoder ring */
nsresult rv = NS_OK;
nsCOMPtr<nsISecretDecoderRing> secretDecoderRing
= do_CreateInstance("@mozilla.org/security/sdr;1", &rv);
if (NS_FAILED(rv)) {
return NS_ERROR_FAILURE;
}
gSecretDecoderRing = secretDecoderRing.get();
NS_ADDREF(gSecretDecoderRing);
}
return NS_OK;
}
#define PREFIX "~"
#include "plbase64.h"
static nsresult EncryptString (const char * text, char *& crypt) {
/* use SecretDecoderRing if encryption pref is set */
nsresult rv;
if (SI_GetBoolPref(pref_Crypto, PR_FALSE)) {
rv = wallet_CryptSetup();
if (NS_SUCCEEDED(rv)) {
rv = gSecretDecoderRing->EncryptString(text, &crypt);
}
if (NS_FAILED(rv)) {
gEncryptionFailure = PR_TRUE;
}
return rv;
}
/* otherwise do our own obscuring using Base64 encoding */
char * crypt0 = PL_Base64Encode(text, 0, NULL);
if (!crypt0) {
return NS_ERROR_FAILURE;
}
PRUint32 PREFIX_len = sizeof (PREFIX) - 1;
PRUint32 crypt0_len = PL_strlen(crypt0);
crypt = (char *)PR_Malloc(PREFIX_len + crypt0_len + 1);
PRUint32 i;
for (i=0; i<PREFIX_len; i++) {
crypt[i] = PREFIX[i];
}
for (i=0; i<crypt0_len; i++) {
crypt[PREFIX_len+i] = crypt0[i];
}
crypt[PREFIX_len + crypt0_len] = '\0';
WALLET_FREE(crypt0);
return NS_OK;
}
static nsresult DecryptString (const char * crypt, char *& text) {
/* treat zero-length crypt string as a special case */
if (crypt[0] == '\0') {
text = (char *)PR_Malloc(1);
text[0] = '\0';
return NS_OK;
}
/* use SecretDecoderRing if crypt doesn't starts with prefix */
if (crypt[0] != PREFIX[0]) {
if ((gReencryptionLevel == 0) && !SI_GetBoolPref(pref_Crypto, PR_FALSE)) {
/*
* User's data is encrypted but pref says it's not.
* This should never occur but it has been observed.
* Consequence of it happening is that user will be asked for master password
* when doing such mundane things as opening edit menu or context menu.
*
* Note that we do not want to make this test if we are in the middle of
* reencypting the entire database (i.e., while execute wallet_ReencryptAll).
* In that case the pref has already been changed and this test will always
* fail. That is why we test the gReencryptionLevel indicator.
*/
NS_ASSERTION(PR_FALSE, "wallet.crypto pref is set incorrectly");
return NS_ERROR_FAILURE;
}
nsresult rv = wallet_CryptSetup();
if (NS_SUCCEEDED(rv)) {
rv = gSecretDecoderRing->DecryptString(crypt, &text);
}
if (NS_FAILED(rv)) {
gEncryptionFailure = PR_TRUE;
}
return rv;
}
/* otherwise do our own de-obscuring */
PRUint32 PREFIX_len = sizeof(PREFIX) - 1;
if (PL_strlen(crypt) == PREFIX_len) {
text = (char *)PR_Malloc(1);
text[0] = '\0';
return NS_OK;
}
text = PL_Base64Decode(&crypt[PREFIX_len], 0, NULL);
if (!text) {
return NS_ERROR_FAILURE;
}
return NS_OK;
}
void
WLLT_ExpirePassword(PRBool* status) {
nsresult rv = wallet_CryptSetup();
if (NS_SUCCEEDED(rv)) {
rv = gSecretDecoderRing->LogoutAndTeardown();
}
*status = NS_SUCCEEDED(rv);
}
void
WLLT_ExpirePasswordOnly(PRBool* status) {
nsresult rv = wallet_CryptSetup();
if (NS_SUCCEEDED(rv)) {
rv = gSecretDecoderRing->Logout();
}
*status = NS_SUCCEEDED(rv);
}
PRBool changingPassword = PR_FALSE;
void
WLLT_ChangePassword(PRBool* status) {
nsresult rv = wallet_CryptSetup();
if (NS_SUCCEEDED(rv)) {
changingPassword = PR_TRUE;
rv = gSecretDecoderRing->ChangePassword();
changingPassword = PR_FALSE;
}
*status = NS_SUCCEEDED(rv);
}
nsresult
wallet_Encrypt(const nsCString& text, nsCString& crypt) {
/* encrypt text to crypt */
char * cryptCString = nsnull;
nsresult rv = EncryptString(text.get(), cryptCString);
if NS_FAILED(rv) {
return rv;
}
crypt = cryptCString;
WALLET_FREE(cryptCString);
return NS_OK;
}
nsresult
wallet_Decrypt(const nsCString& crypt, nsCString& text) {
/* decrypt crypt to text */
char * textCString = nsnull;
nsresult rv = DecryptString(crypt.get(), textCString);
if NS_FAILED(rv) {
return rv;
}
text = textCString;
WALLET_FREE(textCString);
return NS_OK;
}
nsresult
Wallet_Encrypt (const nsAString& textUCS2, nsAString& cryptUCS2) {
nsCAutoString cryptUTF8;
nsresult rv = wallet_Encrypt(NS_ConvertUCS2toUTF8(textUCS2), cryptUTF8);
CopyUTF8toUTF16(cryptUTF8, cryptUCS2);
return rv;
}
nsresult
Wallet_Decrypt(const nsAString& cryptUCS2, nsAString& textUCS2) {
nsCAutoString textUTF8;
nsresult rv = wallet_Decrypt(NS_ConvertUCS2toUTF8(cryptUCS2), textUTF8);
CopyUTF8toUTF16(textUTF8, textUCS2);
return rv;
}
/**********************************************************/
/* The following routines are for accessing the data base */
/**********************************************************/
/*
* clear out the designated list
*/
static void
wallet_Clear(nsVoidArray ** list) {
if (*list == wallet_SchemaToValue_list || *list == wallet_URL_list) {
/* the other lists were allocated in blocks and need to be deallocated the same way */
wallet_MapElement * mapElementPtr;
PRInt32 count = LIST_COUNT((*list));
for (PRInt32 i=count-1; i>=0; i--) {
mapElementPtr = NS_STATIC_CAST(wallet_MapElement*, (*list)->ElementAt(i));
delete mapElementPtr;
}
}
delete (*list);
*list = nsnull;
}
/*
* allocate another mapElement
* We are going to buffer up allocations because it was found that alocating one
* element at a time was very inefficient on the mac
*/
static nsVoidArray * wallet_MapElementAllocations_list = 0;
const PRInt32 kAllocBlockElems = 500;
static PRInt32 wallet_NextAllocSlot = kAllocBlockElems;
static nsresult
wallet_AllocateMapElement(wallet_MapElement*& mapElement) {
static wallet_MapElement* mapElementTable;
if (wallet_NextAllocSlot >= kAllocBlockElems) {
mapElementTable = new wallet_MapElement[kAllocBlockElems];
if (!mapElementTable) {
return NS_ERROR_OUT_OF_MEMORY;
}
if(!wallet_MapElementAllocations_list) {
wallet_MapElementAllocations_list = new nsVoidArray();
}
if(wallet_MapElementAllocations_list) {
wallet_MapElementAllocations_list->AppendElement(mapElementTable);
}
wallet_NextAllocSlot = 0;
}
mapElement = &mapElementTable[wallet_NextAllocSlot++];
return NS_OK;
}
static void
wallet_DeallocateMapElements() {
wallet_MapElement * mapElementPtr;
PRInt32 count = LIST_COUNT(wallet_MapElementAllocations_list);
// initialize remainder of last allocated block so we don't crash on []delete
for (PRInt32 j=wallet_NextAllocSlot; j<kAllocBlockElems; j++) {
mapElementPtr =
NS_STATIC_CAST(wallet_MapElement*,
(wallet_MapElementAllocations_list)->ElementAt(count-1));
mapElementPtr[j].item1 = nsnull;
mapElementPtr[j].item2 = nsnull;
mapElementPtr[j].itemList = nsnull;
}
for (PRInt32 i=count-1; i>=0; i--) {
mapElementPtr =
NS_STATIC_CAST(wallet_MapElement*, (wallet_MapElementAllocations_list)->ElementAt(i));
delete [] mapElementPtr;
}
delete wallet_MapElementAllocations_list;
wallet_MapElementAllocations_list = nsnull;
wallet_NextAllocSlot = kAllocBlockElems;
}
/*
* add an entry to the designated list
*/
static PRBool
wallet_WriteToList(
const char* item1,
const char* item2,
nsVoidArray* itemList,
nsVoidArray*& list,
PRBool obscure,
PlacementType placement = DUP_BEFORE) {
wallet_MapElement * mapElementPtr;
PRBool added_to_list = PR_FALSE;
wallet_MapElement * mapElement = nsnull;
if (list == wallet_FieldToSchema_list || list == wallet_SchemaStrings_list ||
list == wallet_PositionalSchema_list || list == wallet_StateSchema_list ||
list == wallet_SchemaConcat_list || list == wallet_DistinguishedSchema_list ||
list == wallet_VcardToSchema_list) {
wallet_AllocateMapElement(mapElement);
} else {
mapElement = new wallet_MapElement;
}
if (!mapElement) {
return PR_FALSE;
}
nsCAutoString item1UTF8(item1); ToLowerCase(item1UTF8);
mapElement->item1 = ToNewCString(item1UTF8);
mapElement->item2 = PL_strdup(item2);
if (obscure) {
char * crypt = nsnull;
if (NS_FAILED(EncryptString(mapElement->item2, crypt))) {
delete mapElement;
return PR_FALSE;
}
WALLET_FREEIF(mapElement->item2);
mapElement->item2 = crypt;
}
/* make sure the list exists */
if(!list) {
list = new nsVoidArray();
if(!list) {
delete mapElement;
return PR_FALSE;
}
}
mapElement->itemList = itemList;
// note: we didn't want to assign itemList sooner because if we delete mapElement
// above, we would be wiping out the itemList input parameter
/*
* Add new entry to the list in alphabetical order by item1.
* If identical value of item1 exists, use "placement" parameter to
* determine what to do
*/
if (AT_END==placement) {
list->AppendElement(mapElement);
return PR_TRUE;
}
PRInt32 count = LIST_COUNT(list);
for (PRInt32 i=0; i<count; i++) {
mapElementPtr = NS_STATIC_CAST(wallet_MapElement*, list->ElementAt(i));
if (BY_LENGTH==placement) {
if (LIST_COUNT(mapElementPtr->itemList) < LIST_COUNT(itemList)) {
list->InsertElementAt(mapElement, i);
added_to_list = PR_TRUE;
break;
} else if (LIST_COUNT(mapElementPtr->itemList) == LIST_COUNT(itemList)) {
if (itemList) {
wallet_Sublist * sublistPtr;
wallet_Sublist * sublistPtr2;
sublistPtr = NS_STATIC_CAST(wallet_Sublist*, mapElementPtr->itemList->ElementAt(0));
sublistPtr2 = NS_STATIC_CAST(wallet_Sublist*, itemList->ElementAt(0));
if(PL_strlen(sublistPtr->item) < PL_strlen(sublistPtr2->item)) {
list->InsertElementAt(mapElement, i);
added_to_list = PR_TRUE;
break;
}
} else if (PL_strlen(mapElementPtr->item2) < PL_strlen(mapElement->item2)) {
list->InsertElementAt(mapElement, i);
added_to_list = PR_TRUE;
break;
}
}
} else if(!PL_strcmp(mapElementPtr->item1, mapElement->item1)) {
if (DUP_OVERWRITE==placement) {
mapElementPtr->item2 = PL_strdup(item2);
mapElementPtr->itemList = itemList;
mapElement->itemList = nsnull; // else delete might delete itemList input parameter
delete mapElement;
} else if (DUP_BEFORE==placement) {
list->InsertElementAt(mapElement, i);
}
if (DUP_AFTER!=placement) {
added_to_list = PR_TRUE;
break;
}
} else if(PL_strcmp(mapElementPtr->item1, mapElement->item1)>=0) {
list->InsertElementAt(mapElement, i);
added_to_list = PR_TRUE;
break;
}
}
if (!added_to_list) {
list->AppendElement(mapElement);
}
return PR_TRUE;
}
/*
* fetch an entry from the designated list
*/
static PRBool
wallet_ReadFromList(
const nsACString& item1,
nsACString& item2,
nsVoidArray*& itemList,
nsVoidArray*& list,
PRBool obscure,
PRInt32& index)
{
if (!list || (index == -1)) {
return PR_FALSE;
}
/* find item1 in the list */
wallet_MapElement * mapElementPtr;
PRInt32 count = LIST_COUNT(list);
for (PRInt32 i=index; i<count; i++) {
mapElementPtr = NS_STATIC_CAST(wallet_MapElement*, list->ElementAt(i));
if(item1.Equals(mapElementPtr->item1, nsCaseInsensitiveCStringComparator())) {
if (obscure) {
char * plaintext = nsnull;
if (NS_FAILED(DecryptString(mapElementPtr->item2, plaintext))) {
return PR_FALSE;
}
item2 = plaintext;
} else {
item2 = mapElementPtr->item2;
}
itemList = mapElementPtr->itemList;
index = i+1;
if (index == count) {
index = -1;
}
return PR_TRUE;
}
}
index = 0;
return PR_FALSE;
}
PRBool
wallet_ReadFromList(
const nsACString& item1,
nsACString& item2,
nsVoidArray*& itemList,
nsVoidArray*& list,
PRBool obscure)
{
PRInt32 index = 0;
return wallet_ReadFromList(item1, item2, itemList, list, obscure, index);
}
/************************************************************/
/* The following routines are for unlocking the stored data */
/************************************************************/
char* schemaValueFileName = nsnull;
static const char URLFileName[] = "URL.tbl";
static const char allFileName[] = "wallet.tbl";
static const char fieldSchemaFileName[] = "FieldSchema.tbl";
static const char vcardSchemaFileName[] = "VcardSchema.tbl";
static const char schemaConcatFileName[] = "SchemaConcat.tbl";
static const char schemaStringsFileName[] = "SchemaStrings.tbl";
static const char positionalSchemaFileName[] = "PositionalSchema.tbl";
static const char stateSchemaFileName[] = "StateSchema.tbl";
static const char distinguishedSchemaFileName[] = "DistinguishedSchema.tbl";
/******************************************************/
/* The following routines are for accessing the files */
/******************************************************/
nsresult Wallet_ProfileDirectory(nsIFile** aFile) {
/* return the profile */
return NS_GetSpecialDirectory(NS_APP_USER_PROFILE_50_DIR, aFile);
}
nsresult Wallet_DefaultsDirectory(nsIFile** aFile) {
nsresult res;
nsCOMPtr<nsIFile> file;
res = NS_GetSpecialDirectory(NS_APP_DEFAULTS_50_DIR, getter_AddRefs(file));
if (NS_FAILED(res)) return res;
res = file->AppendNative(NS_LITERAL_CSTRING("wallet"));
if (NS_FAILED(res)) return res;
NS_ADDREF(*aFile = file);
return NS_OK;
}
char *
Wallet_RandomName(char* suffix)
{
/* pick the current time as the random number */
time_t curTime = time(NULL);
/* take 8 least-significant digits + three-digit suffix as the file name */
char name[13];
PR_snprintf(name, 13, "%lu.%s", ((int)curTime%100000000), suffix);
return PL_strdup(name);
}
/*
* get a line from a file. stream must implement nsILineInputStream.
* return error if end of file reached
* strip carriage returns and line feeds from end of line
* free with nsMemory::Free
*/
nsresult
wallet_GetLine(nsIInputStream* strm, nsACString &line)
{
line.Truncate();
nsCOMPtr<nsILineInputStream> lis(do_QueryInterface(strm));
NS_ENSURE_TRUE(lis, NS_ERROR_UNEXPECTED);
PRBool more;
nsresult rv = lis->ReadLine(line, &more);
if (NS_FAILED(rv))
return rv;
// Assume that we are past EOF if more==FALSE and line is empty
// this may be wrong if the file ends with an empty line, though
if (!more && line.IsEmpty())
return NS_ERROR_FAILURE;
return NS_OK;
}
static PRBool
wallet_GetHeader(nsIInputStream* strm)
{
nsCAutoString format;
/* format revision number */
if (NS_FAILED(wallet_GetLine(strm, format))) {
return PR_FALSE;
}
return format.EqualsLiteral(HEADER_VERSION);
}
/*
* Write a line-feed to a file
*/
static void
wallet_EndLine(nsIOutputStream* strm) {
static const char nl = '\n';
PRUint32 dummy;
strm->Write(&nl, 1, &dummy);
}
/*
* Write a line to a file
*/
void
wallet_PutLine(nsIOutputStream* strm, const char* line) {
PRUint32 dummy;
strm->Write(line, strlen(line), &dummy);
wallet_EndLine(strm);
}
static void
wallet_PutHeader(nsIOutputStream* strm) {
/* format revision number */
wallet_PutLine(strm, HEADER_VERSION);
}
#define WALLET_NULL(_ptr) (!(_ptr) || !(_ptr)[0])
/*
* write contents of designated list into designated file
*/
static void
wallet_WriteToFile(const char * filename, nsVoidArray* list) {
wallet_MapElement * mapElementPtr;
/* make sure the list exists */
if(!list) {
return;
}
/* open output stream */
nsCOMPtr<nsIFile> file;
nsresult rv = Wallet_ProfileDirectory(getter_AddRefs(file));
if (NS_FAILED(rv)) {
return;
}
file->AppendNative(nsDependentCString(filename));
nsCOMPtr<nsIOutputStream> fileOutputStream;
rv = NS_NewSafeLocalFileOutputStream(getter_AddRefs(fileOutputStream),
file,
-1,
0600);
if (NS_FAILED(rv))
return;
nsCOMPtr<nsIOutputStream> strm;
rv = NS_NewBufferedOutputStream(getter_AddRefs(strm), fileOutputStream, 4096);
if (NS_FAILED(rv))
return;
/* put out the header */
if (!PL_strcmp(filename, schemaValueFileName)) {
wallet_PutHeader(strm);
}
/* traverse the list */
PRInt32 count = LIST_COUNT(list);
for (PRInt32 i=0; i<count; i++) {
mapElementPtr = NS_STATIC_CAST(wallet_MapElement*, list->ElementAt(i));
wallet_PutLine(strm, (*mapElementPtr).item1);
if (!WALLET_NULL((*mapElementPtr).item2)) {
wallet_PutLine(strm, (*mapElementPtr).item2);
} else {
wallet_Sublist * sublistPtr;
PRInt32 count2 = LIST_COUNT(mapElementPtr->itemList);
for (PRInt32 j=0; j<count2; j++) {
sublistPtr = NS_STATIC_CAST(wallet_Sublist*, mapElementPtr->itemList->ElementAt(j));
wallet_PutLine(strm, (*sublistPtr).item);
}
}
wallet_EndLine(strm);
}
// All went ok. Maybe except for problems in Write(), but the stream detects
// that for us
nsCOMPtr<nsISafeOutputStream> safeStream = do_QueryInterface(strm);
NS_ASSERTION(safeStream, "expected a safe output stream!");
if (safeStream) {
rv = safeStream->Finish();
if (NS_FAILED(rv)) {
NS_WARNING("failed to save wallet file! possible dataloss");
return;
}
}
}
/*
* Read contents of designated file into designated list
*/
static void
wallet_ReadFromFile
(const char * filename, nsVoidArray*& list, PRBool localFile, PlacementType placement = AT_END) {
/* open input stream */
nsCOMPtr<nsIFile> file;
nsresult rv;
if (localFile) {
rv = Wallet_ProfileDirectory(getter_AddRefs(file));
} else {
rv = Wallet_DefaultsDirectory(getter_AddRefs(file));
}
if (NS_FAILED(rv)) {
return;
}
file->AppendNative(nsDependentCString(filename));
nsCOMPtr<nsIInputStream> strm;
rv = NS_NewLocalFileInputStream(getter_AddRefs(strm), file);
if (NS_FAILED(rv))
return;
/* read in the header */
if (!PL_strcmp(filename, schemaValueFileName)) {
if (!wallet_GetHeader(strm)) {
/* something's wrong -- ignore the file */
return;
}
}
for (;;) {
if (NS_FAILED(wallet_GetLine(strm, helpMac->item1))) {
/* end of file reached */
break;
}
/* Distinguished schema list is a list of single entries, not name/value pairs */
if (!PL_strcmp(filename, distinguishedSchemaFileName)) {
nsVoidArray* dummy = NULL;
wallet_WriteToList(helpMac->item1.get(), helpMac->item1.get(), dummy, list, PR_FALSE, placement);
continue;
}
if (NS_FAILED(wallet_GetLine(strm, helpMac->item2))) {
/* unexpected end of file reached */
break;
}
if (helpMac->item2.IsEmpty()) {
/* the value must have been deleted */
nsVoidArray* dummy = NULL;
wallet_WriteToList(helpMac->item1.get(), helpMac->item2.get(), dummy, list, PR_FALSE, placement);
continue;
}
if (NS_FAILED(wallet_GetLine(strm, helpMac->item3))) {
/* end of file reached */
nsVoidArray* dummy = NULL;
wallet_WriteToList(helpMac->item1.get(), helpMac->item2.get(), dummy, list, PR_FALSE, placement);
return;
}
if (helpMac->item3.IsEmpty()) {
/* just a pair of values, no need for a sublist */
nsVoidArray* dummy = NULL;
wallet_WriteToList(helpMac->item1.get(), helpMac->item2.get(), dummy, list, PR_FALSE, placement);
} else {
/* need to create a sublist and put item2 and item3 onto it */
nsVoidArray * itemList = new nsVoidArray();
if (!itemList) {
break;
}
wallet_Sublist * sublist = new wallet_Sublist;
if (!sublist) {
break;
}
sublist->item = ToNewCString(helpMac->item2);
itemList->AppendElement(sublist);
sublist = new wallet_Sublist;
if (!sublist) {
delete itemList;
break;
}
sublist->item = ToNewCString(helpMac->item3);
itemList->AppendElement(sublist);
/* add any following items to sublist up to next blank line */
for (;;) {
/* get next item for sublist */
if (NS_FAILED(wallet_GetLine(strm, helpMac->item3))) {
/* end of file reached */
wallet_WriteToList(helpMac->item1.get(), nsnull, itemList, list, PR_FALSE, placement);
return;
}
if (helpMac->item3.IsEmpty()) {
/* blank line reached indicating end of sublist */
wallet_WriteToList(helpMac->item1.get(), nsnull, itemList, list, PR_FALSE, placement);
break;
}
/* add item to sublist */
sublist = new wallet_Sublist;
if (!sublist) {
delete itemList;
break;
}
sublist->item = ToNewCString(helpMac->item3);
itemList->AppendElement(sublist);
}
}
}
}
/*********************************************************************/
/* The following are utility routines for the main wallet processing */
/*********************************************************************/
void
Wallet_GiveCaveat(nsIDOMWindowInternal* window, nsIPrompt* dialog) {
/* test for first capturing of data ever and give caveat if so */
if (!SI_GetBoolPref(pref_Caveat, PR_FALSE)) {
SI_SetBoolPref(pref_Caveat, PR_TRUE);
PRUnichar * message = Wallet_Localize("Caveat");
if (window) {
wallet_Alert(message, window);
} else {
wallet_Alert(message, dialog);
}
WALLET_FREE(message);
}
}
static void
wallet_GetHostFile(nsIURI * url, nsString& outHostFile)
{
outHostFile.Truncate(0);
nsCAutoString host;
nsresult rv = url->GetHost(host);
if (NS_FAILED(rv)) {
return;
}
NS_ConvertUTF8toUCS2 urlName(host);
nsCAutoString file;
rv = url->GetPath(file);
if (NS_FAILED(rv)) {
return;
}
AppendUTF8toUTF16(file, urlName);
PRInt32 queryPos = urlName.FindChar('?');
PRInt32 stringEnd = (queryPos == kNotFound) ? urlName.Length() : queryPos;
urlName.Left(outHostFile, stringEnd);
}
static void
Strip(const nsString& textUCS2, nsCString& stripText) {
NS_ConvertUCS2toUTF8 textUTF8(textUCS2);
// above line is equivalen to the following (who would have guessed it?)
// nsCAutoString textUTF8 = NS_ConvertUCS2toUTF8(textUCS2);
for (PRUint32 i=0; i<textUTF8.Length(); i++) {
char c = textUTF8.CharAt(i);
if (nsCRT::IsAsciiAlpha(c) || nsCRT::IsAsciiDigit(c) || c>'~') {
stripText += c;
}
}
}
/*
* given a displayable text, get the schema
*/
static void
TextToSchema(
const nsString& text,
nsACString& schema)
{
/* return if no SchemaStrings list exists */
if (!wallet_SchemaStrings_list) {
return;
}
/* try each schema entry in schemastring table to see if it's acceptable */
wallet_MapElement * mapElementPtr;
PRInt32 count = LIST_COUNT(wallet_SchemaStrings_list);
for (PRInt32 i=0; i<count; i++) {
/* get each string associated with this schema */
PRBool isSubstring = PR_TRUE;
mapElementPtr = NS_STATIC_CAST(wallet_MapElement*, wallet_SchemaStrings_list->ElementAt(i));
wallet_Sublist * sublistPtr;
PRInt32 count2 = LIST_COUNT(mapElementPtr->itemList);
if (count2) {
for (PRInt32 i2=0; i2<count2; i2++) {
/* see if displayable text contains this string */
sublistPtr = NS_STATIC_CAST(wallet_Sublist*, mapElementPtr->itemList->ElementAt(i2));
if (text.Find(sublistPtr->item, PR_TRUE) == -1) {
/* displayable text does not contain this string, reject this schema */
isSubstring = PR_FALSE;
break;
}
}
} else if (text.Find(mapElementPtr->item2, PR_TRUE) == -1) {
/* displayable text does not contain this string, reject this schema */
isSubstring = PR_FALSE;
}
if (isSubstring) {
/* all strings were contained in the displayable text, accept this schema */
schema.Assign(mapElementPtr->item1);
return;
}
}
}
/*
* given a field name, get the value
*/
static nsresult
FieldToValue(
const nsString& field,
nsACString& schema,
nsString& valueUCS2,
nsVoidArray*& itemList,
PRInt32& index)
{
/* return if no SchemaToValue list exists or if all values previous used */
if (!wallet_SchemaToValue_list || index == -1) {
return NS_ERROR_FAILURE;
}
/* if no schema name is given, fetch schema name from field/schema tables */
nsVoidArray* dummy;
nsCAutoString stripField;
if (schema.IsEmpty()) {
Strip(field, stripField);
}
if (!schema.IsEmpty() ||
wallet_ReadFromList(stripField, schema, dummy, wallet_FieldToSchema_list, PR_FALSE)) {
/* schema name found, now attempt to fetch value from schema/value table */
nsCAutoString valueUTF8;
PRInt32 index2 = index;
if ((index >= 0) &&
wallet_ReadFromList
(schema, valueUTF8, itemList, wallet_SchemaToValue_list, PR_TRUE, index2)) {
/* value found, prefill it into form and return */
CopyUTF8toUTF16(valueUTF8, valueUCS2);
index = index2;
return NS_OK;
} else {
/* value not found, see if concatenation rule exists */
nsVoidArray * itemList2;
nsCAutoString valueUTF8b;
if (index > 0) {
index = 0;
}
PRInt32 index0 = index;
PRInt32 index00 = index;
PRInt32 index4 = 0;
while (wallet_ReadFromList(schema, valueUTF8b, itemList2, wallet_SchemaConcat_list, PR_FALSE, index4)) {
/* concatenation rules exist, generate value as a concatenation */
nsCAutoString concatenatedValueUTF8;
wallet_Sublist * sublistPtr;
concatenatedValueUTF8.SetLength(0);
nsCAutoString valueUTF8c;
PRInt32 index00max = index0;
if (!valueUTF8b.IsEmpty()) {
/* single item on rhs of concatenation rule */
PRInt32 index5 = 0;
PRInt32 j;
PRBool failed = PR_FALSE;
for (j=0; j>index0; j -= 2) {
if (!wallet_ReadFromList(valueUTF8b, valueUTF8c, dummy, wallet_SchemaToValue_list, PR_TRUE, index5)) {
failed = PR_TRUE;
break;
}
index00 += 2;
}
if (!failed && wallet_ReadFromList(valueUTF8b, valueUTF8c, dummy, wallet_SchemaToValue_list, PR_TRUE, index5)) {
/* found an unused value for the single rhs item */
concatenatedValueUTF8 += valueUTF8c;
index00 += 2;
}
index00max = index00;
}
/* process each item in a multi-rhs rule */
PRInt32 count = LIST_COUNT(itemList2);
for (PRInt32 i=0; i<count; i++) {
sublistPtr = NS_STATIC_CAST(wallet_Sublist*, itemList2->ElementAt(i));
/* skip over values found previously */
/* note: a returned index of -1 means not-found. So we will use the
* negative even numbers (-2, -4, -6) to designate found as a concatenation
* where -2 means first value of each concatenation, -4 means second value, etc.
*/
index00 = index0;
PRInt32 index3 = 0;
PRBool failed = PR_FALSE;
nsCAutoString valueUTF8d; valueUTF8d.Assign(sublistPtr->item);
for (PRInt32 j=0; j>index0; j -= 2) {
if (!wallet_ReadFromList(valueUTF8d, valueUTF8, dummy, wallet_SchemaToValue_list, PR_TRUE, index3)) {
/* all values of next multi-rhs item were used previously */
failed = PR_TRUE;
break;
}
index00 += 2;
}
if (!failed && wallet_ReadFromList(valueUTF8d, valueUTF8, dummy, wallet_SchemaToValue_list, PR_TRUE, index3)) {
if (!concatenatedValueUTF8.IsEmpty()) {
concatenatedValueUTF8 += " ";
}
/* found an unused value for the multi-rhs item */
concatenatedValueUTF8 += valueUTF8;
index00 += 2;
}
if (index00 > index00max) {
index00max = index00;
}
}
itemList = nsnull;
if (!concatenatedValueUTF8.IsEmpty()) {
/* a new value was found */
index -= 2;
CopyUTF8toUTF16(concatenatedValueUTF8, valueUCS2);
return NS_OK;
}
/* all values from this concat rule were used, go on to next concat rule */
index0 = index00max;
}
/* no more concat rules, indicate failure */
index = -1;
return NS_ERROR_FAILURE;
}
} else {
/* schema name not found, use field name as schema name and fetch value */
PRInt32 index2 = index;
nsAutoString localSchemaUCS2;
wallet_GetHostFile(wallet_lastUrl, localSchemaUCS2);
localSchemaUCS2.AppendLiteral(":");
localSchemaUCS2.Append(field);
NS_ConvertUTF16toUTF8 localSchemaUTF8(localSchemaUCS2);
nsCAutoString valueUTF8;
if (wallet_ReadFromList
(localSchemaUTF8, valueUTF8, itemList, wallet_SchemaToValue_list, PR_TRUE, index2)) {
/* value found, prefill it into form */
schema = localSchemaUTF8;
index = index2;
CopyUTF8toUTF16(valueUTF8, valueUCS2);
return NS_OK;
}
}
index = -1;
return NS_ERROR_FAILURE;
}
static nsresult
wallet_GetSelectIndex(
nsIDOMHTMLSelectElement* selectElement,
const nsString& value,
PRInt32& index)
{
PRUint32 length;
selectElement->GetLength(&length);
nsCOMPtr<nsIDOMHTMLOptionsCollection> options;
selectElement->GetOptions(getter_AddRefs(options));
if (options) {
PRUint32 numOptions;
options->GetLength(&numOptions);
for (PRUint32 optionX = 0; optionX < numOptions; optionX++) {
nsCOMPtr<nsIDOMNode> optionNode;
options->Item(optionX, getter_AddRefs(optionNode));
if (optionNode) {
nsCOMPtr<nsIDOMHTMLOptionElement> optionElement(do_QueryInterface(optionNode));
if (optionElement) {
nsAutoString optionValue;
nsAutoString optionText;
optionElement->GetValue(optionValue);
optionElement->GetText(optionText);
nsAutoString valueLC( value );
ToLowerCase(valueLC);
ToLowerCase(optionValue);
ToLowerCase(optionText);
optionText.Trim(" \n\t\r");
if (valueLC==optionValue || valueLC==optionText) {
index = optionX;
return NS_OK;
}
}
}
}
}
return NS_ERROR_FAILURE;
}
void
wallet_StepForwardOrBack
(nsIDOMNode*& elementNode, nsString& text, PRBool& atInputOrSelect, PRBool& atEnd, PRBool goForward) {
nsresult result;
atInputOrSelect = PR_FALSE;
atEnd = PR_FALSE;
/* try getting next/previous sibling */
nsCOMPtr<nsIDOMNode> sibling;
if (goForward) {
result = elementNode->GetNextSibling(getter_AddRefs(sibling));
} else {
result = elementNode->GetPreviousSibling(getter_AddRefs(sibling));
}
if ((NS_FAILED(result)) || !sibling) {
/* no next/previous siblings, try getting parent */
nsCOMPtr<nsIDOMNode> parent;
result = elementNode->GetParentNode(getter_AddRefs(parent));
if ((NS_FAILED(result)) || !parent) {
/* no parent, we've reached the top of the tree */
atEnd = PR_TRUE;
} else {
/* parent obtained */
elementNode = parent;
}
return;
}
/* sibling obtained */
elementNode = sibling;
while (PR_TRUE) {
/* if we've reached a SELECT or non-hidden INPUT tag, we're done */
/*
* There is a subtle difference here between going forward and going backwards.
*
* When going forward we are trying to find out how many consecutive <input> elements are not separated
* by displayed text. That is important for determing, for example, if we have a three-input phone-number
* field. In that case, we want to consider only input tags have type="text" or no type ("text" by default).
*
* When going backwards we want to find the text between the current <input> element and any preceding
* visible <input> element. That would include such things as type="button", type="submit" etc. The
* only thing it would exclude is type="hidden".
*/
nsCOMPtr<nsIDOMHTMLInputElement> inputElement(do_QueryInterface(elementNode, &result));
if ((NS_SUCCEEDED(result)) && (inputElement)) {
nsAutoString type;
result = inputElement->GetType(type);
if (goForward) {
if (NS_SUCCEEDED(result) &&
(type.IsEmpty() ||
type.LowerCaseEqualsLiteral("text"))) {
/* at <input> element and it's type is either "text" or is missing ("text" by default) */
atInputOrSelect = PR_TRUE;
return;
}
} else {
if (NS_SUCCEEDED(result) &&
!type.LowerCaseEqualsLiteral("hidden")) {
/* at <input> element and it's type is not "hidden" */
atInputOrSelect = PR_TRUE;
return;
}
}
} else {
nsCOMPtr<nsIDOMHTMLSelectElement> selectElement(do_QueryInterface(elementNode));
if (selectElement) {
atInputOrSelect = PR_TRUE;
return;
}
}
/* if we've reached a #text node, append it to accumulated text */
nsAutoString siblingNameUCS2;
result = elementNode->GetNodeName(siblingNameUCS2);
if (siblingNameUCS2.LowerCaseEqualsLiteral("#text")) {
nsAutoString siblingValue;
result = elementNode->GetNodeValue(siblingValue);
text.Append(siblingValue);
}
/* if we've reached a SCRIPT node, don't fetch its siblings */
if (siblingNameUCS2.LowerCaseEqualsLiteral("script")) {
return;
}
/* try getting first/last child */
nsCOMPtr<nsIDOMNode> child;
if (goForward) {
result = elementNode->GetFirstChild(getter_AddRefs(child));
} else {
result = elementNode->GetLastChild(getter_AddRefs(child));
}
if ((NS_FAILED(result)) || !child) {
/* no children, we're done with this node */
return;
}
/* child obtained */
elementNode = child;
}
return;
}
//#include "nsIUGenCategory.h"
//#include "nsUnicharUtilCIID.h"
//static NS_DEFINE_IID(kUnicharUtilCID, NS_UNICHARUTIL_CID);
//#include "nsICaseConversion.h"
//static nsICaseConversion* gCaseConv = nsnull;
static void
wallet_ResolvePositionalSchema(nsIDOMNode* elementNode, nsACString& schema) {
static PRInt32 numerator = 0;
static PRInt32 denominator = 0;
static nsCString lastPositionalSchema;
/* return if no PositionalSchema list exists */
if (!wallet_PositionalSchema_list) {
schema.SetLength(0);
return;
}
if (!schema.IsEmpty()) {
numerator = 0;
denominator = 0;
lastPositionalSchema.Assign(schema);
} else if (numerator < denominator) {
schema.Assign(lastPositionalSchema);
} else {
schema.SetLength(0);
return;
}
/* search PositionalSchema list for our positional schema */
wallet_MapElement * mapElementPtr;
PRInt32 count = LIST_COUNT(wallet_PositionalSchema_list);
for (PRInt32 i=0; i<count; i++) {
mapElementPtr = NS_STATIC_CAST(wallet_MapElement*, wallet_PositionalSchema_list->ElementAt(i));
if (schema.Equals(mapElementPtr->item1, nsCaseInsensitiveCStringComparator())) {
/* found our positional schema in the list */
/* A "position set" is a set of continuous <input> or <select> fields
* with no displayable text between them. For example: zipcode [ ]-[ ].
* We need to determine how many elements are in the current set (denominator)
* and which of those elements we are currently up to (numerator). From that
* we can identify our position with the fraction x/y meaning the xth element
* out of a set of y. We use that fraction when consulting the positionalSchema list
* to determine which schema should be used.
*
* So for example, the positionalSchema list for %zip might be:
*
* 1/1 Home.PostalCode
* 1/2 Home.PostalCode.Prefix
* 2/2 Home.PostalCode.Suffix
*
* The positionalSchema list also contains fractions with no denominators, for example x/.
* That means the xth element out of a set of any length. These entries come last in
* the positionalSchema list so they can match only if no match for a specific length is
* found. As an example, the positionalSchema list for %phone might be:
*
* 1/1 Home.Phone
* 1/2 Home.Phone.LocCode
* 2/2 Home.Phone.Number
* 1/ Home.Phone.LocCode
* 2/ Home.Phone.Number.Prefix
* 3/ Home.Phone.Number.Suffix
*/
if (numerator < denominator) {
/* this is a continuation of previous position set */
numerator++;
} else {
/* start a new position set */
numerator = 1; /* start with first element */
/* determine how many elements in current position set (denominator) */
denominator = 1; /* assume that's the only element */
PRBool atInputOrSelect = PR_FALSE;
PRBool charFound = PR_FALSE;
while (!charFound) {
nsAutoString text;
PRBool atEnd;
wallet_StepForwardOrBack
(elementNode, text, atInputOrSelect, atEnd, PR_TRUE); /* step forward */
if (atEnd) {
break;
}
PRUnichar c;
for (PRUint32 j=0; j<text.Length(); j++) {
c = text.CharAt(j);
/* break out if an alphanumeric character is found */
// nsresult res = CallGetService(kUnicharUtilCID, &gCaseConv);
//
// nsIUGenCategory* intl = nsnull;
// nsresult rv = CallGetService(kUnicharUtilCID, &intl);
// Whaaaaaa, intl is never released here!
// if (NS_SUCCEEDED(rv) && intl) {
// PRBool accept;
// rv = intl->Is(c, intl->kUGenCategory_Number, &accept);
// if (NS_FAILED(rv) || !accept) {
// rv = intl->Is(c, intl->kUGenCategory_Letter, &accept);
// }
// if (NS_OK(rv) && accept) {
// charFound = PR_TRUE;
// break;
// }
// } else {
// /* failed to get the i18n interfaces, so just treat latin characters */
if (nsCRT::IsAsciiAlpha(c) || nsCRT::IsAsciiDigit(c)) {
charFound = PR_TRUE;
break;
// }
}
}
if (!charFound && atInputOrSelect) {
/* add one more element to position set */
denominator++;
}
}
}
nsCAutoString fractionString; /* of form 2/5 meaning 2nd in a 5-element set */
nsCAutoString fractionStringWithoutDenominator; /* of form 2/ meaning 2nd in any-length set */
fractionString.SetLength(0);
fractionString.AppendInt(numerator);
fractionString.Append("/");
fractionStringWithoutDenominator.Assign(fractionString);
fractionString.AppendInt(denominator);
/* use positionalSchema list to obtain schema */
wallet_Sublist * sublistPtr;
PRInt32 count2 = LIST_COUNT(mapElementPtr->itemList);
for (PRInt32 j=0; j<count2; j=j+2) {
sublistPtr = NS_STATIC_CAST(wallet_Sublist*, mapElementPtr->itemList->ElementAt(j));
if (!PL_strcmp(sublistPtr->item, fractionString.get()) ||
!PL_strcmp(sublistPtr->item, fractionStringWithoutDenominator.get())) {
sublistPtr = NS_STATIC_CAST(wallet_Sublist*, mapElementPtr->itemList->ElementAt(j+1));
schema.Assign(sublistPtr->item);
return;
}
}
}
}
}
const char* previousElementState = nsnull;
static nsIDOMNode* previousElementNode;
static void
wallet_InitializeStateTesting() {
previousElementNode = nsnull;
previousElementState = nsnull;
}
static void
wallet_ResolveStateSchema(nsIDOMNode* elementNode, nsACString& schema) {
/* return if no StateSchema list exists */
if (!wallet_StateSchema_list) {
return;
}
/* search state schema list for our state schema */
wallet_MapElement * mapElementPtr;
PRInt32 count = LIST_COUNT(wallet_StateSchema_list);
for (PRInt32 i=0; i<count; i++) {
mapElementPtr = NS_STATIC_CAST(wallet_MapElement*, wallet_StateSchema_list->ElementAt(i));
if (schema.Equals(mapElementPtr->item1, nsCaseInsensitiveCStringComparator())) {
/* found our state schema in the list */
/* A state-schema entry consists of a set of possible states and the schema associated
* with each state. For example, for the state-schema $phone we might have
*
* ship ShipTo.Phone
* bill BillTo.Phone
* * Home.Phone
*
* This means that if we are in the "ship" state, the schema is ShipTo.Phone, if in the
* "bill" state it is BillTo.Phone, and if in no identifiable state it is Home.Phone.
*
* So we will start stepping backwards through the dom tree
* obtaining text at each step. If the text contains a substring for one of
* the states, then that is the state we are in and we take the associated
* schema. If the text does not contain any of the states, we continue
* stepping back until we get to a preceding node for which we knew the state.
* If none is found, stop when we get to the beginning of the tree.
*/
nsIDOMNode* localElementNode = elementNode;
PRBool atEnd = PR_FALSE;
PRBool atInputOrSelect = PR_FALSE;
while (!atEnd) {
/* get next text in the dom */
nsAutoString text;
wallet_StepForwardOrBack(localElementNode, text, atInputOrSelect, atEnd, PR_FALSE);
/* see if it's a node we already saved the state for */
if (localElementNode == previousElementNode) {
previousElementNode = elementNode;
/* step through the list of states to see if any are the state of the previous Node */
wallet_Sublist * sublistPtr;
PRInt32 count2 = LIST_COUNT(mapElementPtr->itemList);
PRInt32 j;
for (j=0; j<count2; j=j+2) {
sublistPtr = NS_STATIC_CAST(wallet_Sublist*, mapElementPtr->itemList->ElementAt(j));
if (!PL_strcasecmp(sublistPtr->item, previousElementState)) {
previousElementState = sublistPtr->item;
sublistPtr = NS_STATIC_CAST(wallet_Sublist*, mapElementPtr->itemList->ElementAt(j+1));
schema.Assign(sublistPtr->item);
return;
}
/* test to see if we obtained the catch-all (*) state.
* Note: the catch-all must be the last entry in the list
*/
if (!PL_strcmp(sublistPtr->item, "*")) {
sublistPtr = NS_STATIC_CAST(wallet_Sublist*, mapElementPtr->itemList->ElementAt(j+1));
schema.Assign(sublistPtr->item);
return;
}
}
/* no catch-all state specified, return no schema */
return;
}
/* step through the list of states to see if any are in the text */
wallet_Sublist * sublistPtr;
PRInt32 count2 = LIST_COUNT(mapElementPtr->itemList);
for (PRInt32 j=0; j<count2; j=j+2) {
sublistPtr = NS_STATIC_CAST(wallet_Sublist*, mapElementPtr->itemList->ElementAt(j));
/* next state obtained, test to see if it is in the text */
if (text.Find(sublistPtr->item, PR_TRUE) != -1) {
previousElementState = sublistPtr->item;
previousElementNode = elementNode;
sublistPtr = NS_STATIC_CAST(wallet_Sublist*, mapElementPtr->itemList->ElementAt(j+1));
schema.Assign(sublistPtr->item);
return;
}
}
}
/* state not found, so take the catch-all (*) state */
wallet_Sublist * sublistPtr;
PRInt32 count2 = LIST_COUNT(mapElementPtr->itemList);
for (PRInt32 j=0; j<count2; j=j+2) {
sublistPtr = NS_STATIC_CAST(wallet_Sublist*, mapElementPtr->itemList->ElementAt(j));
if (!PL_strcmp(sublistPtr->item, "*")) {
previousElementNode = localElementNode;
sublistPtr = NS_STATIC_CAST(wallet_Sublist*, mapElementPtr->itemList->ElementAt(j+1));
schema.Assign(sublistPtr->item);
previousElementNode = elementNode;
return;
}
}
/* no catch-all state specified, return no schema */
previousElementNode = elementNode;
return;
}
}
/* This is an error. It means that a state-schema (entry starting with a $)
* was obtained from the SchemaStrings table or the PositionalSchema table
* but there was no entry for that state-schema in the StateSchema table.
*/
NS_ASSERTION(PR_FALSE, "Undefined state in SchemaStrings table");
}
static void
wallet_GetSchemaFromDisplayableText
(nsIDOMNode* elementNode, nsACString& schema, PRBool skipStateChecking) {
static nsCString lastSchema;
static nsIDOMNode* lastElementNode;
/* return if this is the same as the last element node */
if (elementNode == lastElementNode) {
schema.Assign(lastSchema);
return;
}
lastElementNode = elementNode;
nsIDOMNode* localElementNode = elementNode;
PRBool atInputOrSelect = PR_FALSE;
PRBool atEnd = PR_FALSE;
PRBool someTextFound = PR_FALSE;
while (!atEnd && !atInputOrSelect) {
/* step back and get text found in a preceding node */
nsAutoString text;
wallet_StepForwardOrBack(localElementNode, text, atInputOrSelect, atEnd, PR_FALSE);
/* strip off non-alphanumerics */
PRUint32 i;
PRUnichar c;
nsAutoString temp;
for (i=0; i<text.Length(); i++) {
c = text.CharAt(i);
if (nsCRT::IsAsciiAlpha(c) || nsCRT::IsAsciiDigit(c)) {
temp.Append(c);
}
}
text = temp;
/* done if we've obtained enough text from which to determine the schema */
if (!text.IsEmpty()) {
someTextFound = PR_TRUE;
TextToSchema(text, schema);
if (!schema.IsEmpty()) {
/* schema found, process positional schema if any */
if (schema.First() == '%') {
wallet_ResolvePositionalSchema(elementNode, schema);
}
/* process state schema if any */
if (!skipStateChecking && !schema.IsEmpty() && schema.First() == '$') {
wallet_ResolveStateSchema(elementNode, schema);
}
lastSchema.Assign(schema);
return;
}
}
}
/* no displayable text found, see if we are inside a position set */
if (!someTextFound) {
wallet_ResolvePositionalSchema(elementNode, schema);
}
/* process state schema if any */
/* The current routine is called for each field whose value is to be captured,
* even if there is no value entered for that field. We do this because we need
* to call ResolvePositionalSchema above even for null values. If we didn't
* make that call, we would fail to recognize fields in a positional set if any
* preceding fields in that set were blank. For example:
*
* name (first, middle, last): [William] [ ] [Clinton]
*
* With that said, at least we can skip the call to ResolveStateSchema in this
* case. That call could be very time consuming because it involves looking
* looking backwards through all preceding text (possibly all the way to the
* beginning of the document) just to determine the state. That is the purpose
* of the skipStateChecking argument.
*/
if (!skipStateChecking && !schema.IsEmpty() && schema.First() == '$') {
wallet_ResolveStateSchema(elementNode, schema);
}
lastSchema.Assign(schema);
return;
}
nsresult
wallet_GetPrefills(
nsIDOMNode* elementNode,
nsIDOMHTMLInputElement*& inputElement,
nsIDOMHTMLSelectElement*& selectElement,
nsACString& schema,
nsString& value,
PRInt32& selectIndex,
PRInt32& index)
{
nsresult result;
nsCAutoString localSchema; localSchema.Assign(schema);
/* get prefills for input element */
result = elementNode->QueryInterface(NS_GET_IID(nsIDOMHTMLInputElement), (void**)&inputElement);
if ((NS_SUCCEEDED(result)) && (nsnull != inputElement)) {
nsAutoString type;
result = inputElement->GetType(type);
if (NS_SUCCEEDED(result) &&
(type.IsEmpty() ||
type.LowerCaseEqualsLiteral("text"))) {
nsAutoString field;
result = inputElement->GetName(field);
if (NS_SUCCEEDED(result)) {
nsVoidArray* itemList;
/* try to get schema name from vcard attribute if it exists */
if (localSchema.IsEmpty()) {
nsCOMPtr<nsIDOMElement> element = do_QueryInterface(elementNode);
if (element) {
nsAutoString vcard; vcard.AssignLiteral("VCARD_NAME");
nsAutoString vcardValueUCS2;
result = element->GetAttribute(vcard, vcardValueUCS2);
if (NS_OK == result) {
nsVoidArray* dummy;
wallet_ReadFromList(NS_ConvertUCS2toUTF8(vcardValueUCS2), localSchema, dummy,
wallet_VcardToSchema_list, PR_FALSE);
}
}
}
/* try to get schema name from displayable text if possible */
if (localSchema.IsEmpty()) {
wallet_GetSchemaFromDisplayableText(inputElement, localSchema, PR_FALSE);
}
/*
* if schema name was obtained then get value from schema name,
* otherwise get value from field name by using mapping tables to get schema name
*/
if (NS_SUCCEEDED(FieldToValue(field, localSchema, value, itemList, index))) {
if (value.IsEmpty() && nsnull != itemList) {
/* pick first of a set of synonymous values */
const char* encryptedValue = ((wallet_Sublist *)itemList->ElementAt(0))->item;
char* valueCString = nsnull;
if (NS_FAILED(DecryptString(encryptedValue, valueCString))) {
NS_RELEASE(inputElement);
return NS_ERROR_FAILURE;
}
CopyUTF8toUTF16(valueCString, value);
}
selectElement = nsnull;
selectIndex = -1;
schema = localSchema;
return NS_OK;
}
}
}
NS_RELEASE(inputElement);
return NS_ERROR_FAILURE;
}
/* get prefills for dropdown list */
result = elementNode->QueryInterface(NS_GET_IID(nsIDOMHTMLSelectElement), (void**)&selectElement);
if ((NS_SUCCEEDED(result)) && (nsnull != selectElement)) {
nsAutoString field;
result = selectElement->GetName(field);
if (NS_SUCCEEDED(result)) {
/* try to get schema name from displayable text if possible */
if (localSchema.IsEmpty()) {
wallet_GetSchemaFromDisplayableText(selectElement, localSchema, PR_FALSE);
}
nsVoidArray* itemList;
if (NS_SUCCEEDED(FieldToValue(field, localSchema, value, itemList, index))) {
if (!value.IsEmpty()) {
/* no synonym list, just one value to try */
result = wallet_GetSelectIndex(selectElement, value, selectIndex);
if (NS_SUCCEEDED(result)) {
/* value matched one of the values in the drop-down list */
inputElement = nsnull;
schema = localSchema;
return NS_OK;
}
} else {
/* synonym list exists, try each value */
for (PRInt32 i=0; i<LIST_COUNT(itemList); i++) {
CopyUTF8toUTF16(((wallet_Sublist *)itemList->ElementAt(i))->item, value);
result = wallet_GetSelectIndex(selectElement, value, selectIndex);
if (NS_SUCCEEDED(result)) {
/* value matched one of the values in the drop-down list */
// No Release() here?
inputElement = nsnull;
schema = localSchema;
return NS_OK;
}
}
}
}
}
NS_RELEASE(selectElement);
}
return NS_ERROR_FAILURE;
}
/*
* termination for wallet session
*/
void
Wallet_ReleaseAllLists() {
wallet_Clear(&wallet_FieldToSchema_list); /* otherwise we will duplicate the list */
wallet_Clear(&wallet_VcardToSchema_list); /* otherwise we will duplicate the list */
wallet_Clear(&wallet_SchemaConcat_list); /* otherwise we will duplicate the list */
wallet_Clear(&wallet_SchemaStrings_list); /* otherwise we will duplicate the list */
wallet_Clear(&wallet_PositionalSchema_list); /* otherwise we will duplicate the list */
wallet_Clear(&wallet_StateSchema_list); /* otherwise we will duplicate the list */
wallet_Clear(&wallet_DistinguishedSchema_list); /* otherwise we will duplicate the list */
wallet_DeallocateMapElements();
delete helpMac;
helpMac = nsnull;
}
//#define WALLET_CHECK_FOOTPRINT
#ifdef WALLET_CHECK_FOOTPRINT
PRInt32
wallet_Size(nsVoidArray * list) {
PRInt32 size = 0;
wallet_MapElement * mapElementPtr;
PRInt32 count = LIST_COUNT(list);
for (PRInt32 i=0; i<count; i++) {
mapElementPtr = NS_STATIC_CAST(wallet_MapElement*, list->ElementAt(i));
size += sizeof(wallet_MapElement*);
size += sizeof(wallet_MapElement);
size += PL_strlen(mapElementPtr->item1);
size += PL_strlen(mapElementPtr->item2);
wallet_Sublist * sublistPtr;
PRInt32 count2 = LIST_COUNT(mapElementPtr->itemList);
for (PRInt32 i2=0; i2<count2; i2++) {
sublistPtr = NS_STATIC_CAST(wallet_Sublist*, mapElementPtr->itemList->ElementAt(i2));
size += sizeof(wallet_Sublist);
size += PL_strlen(sublistPtr->item);
}
}
return size;
}
#endif
/*
* initialization for wallet session (done only once)
*/
static PRBool wallet_tablesInitialized = PR_FALSE;
static PRBool wallet_ValuesReadIn = PR_FALSE;
static PRBool namesInitialized = PR_FALSE;
static PRBool wallet_URLListInitialized = PR_FALSE;
static void
wallet_Initialize(PRBool unlockDatabase=PR_TRUE) {
#ifdef DEBUG_morse
//wallet_ClearStopwatch();
//wallet_ResumeStopwatch();
#endif
if (!wallet_tablesInitialized) {
#ifdef DEBUG_morse
//wallet_PauseStopwatch();
//wallet_DumpStopwatch();
#endif
// printf("******** start profile\n");
// ProfileStart();
Wallet_ReleaseAllLists();
helpMac = new wallet_HelpMac; /* to speed up startup time on the mac */
wallet_ReadFromFile(distinguishedSchemaFileName, wallet_DistinguishedSchema_list, PR_FALSE);
wallet_ReadFromFile(fieldSchemaFileName, wallet_FieldToSchema_list, PR_FALSE);
wallet_ReadFromFile(vcardSchemaFileName, wallet_VcardToSchema_list, PR_FALSE);
wallet_ReadFromFile(schemaConcatFileName, wallet_SchemaConcat_list, PR_FALSE);
wallet_ReadFromFile(schemaStringsFileName, wallet_SchemaStrings_list, PR_FALSE, BY_LENGTH);
wallet_ReadFromFile(positionalSchemaFileName, wallet_PositionalSchema_list, PR_FALSE);
wallet_ReadFromFile(stateSchemaFileName, wallet_StateSchema_list, PR_FALSE);
#ifdef WALLET_CHECK_FOOTPRINT
PRInt32 totalSize = 0;
PRInt32 size;
size = wallet_Size(wallet_FieldToSchema_list);
totalSize += size;
printf("FieldToSchema: %d\n", size);
size = wallet_Size(wallet_VcardToSchema_list);
totalSize += size;
printf("VcardToSchema: %d\n", size);
size = wallet_Size(wallet_SchemaConcat_list);
totalSize += size;
printf("SchemaConcat: %d\n", size);
size = wallet_Size(wallet_SchemaStrings_list);
totalSize += size;
printf("SchemaStrings: %d\n", size);
size = wallet_Size(wallet_PositionalSchema_list);
totalSize += size;
printf("PositionalSchema: %d\n", size);
size = wallet_Size(wallet_StateSchema_list);
totalSize += size;
printf("StateSchema: %d\n", size);
size = wallet_Size(wallet_DistinguishedSchema_list);
totalSize += size;
printf("DistinguishedSchema: %d\n", size);
printf("Total size: %d\n", totalSize);
#endif
/* Note that we sort the SchemaString list by length instead of alphabetically. To see
* why that's necessary, consider the following example:
*
* Card.Name: requires "card" and "name" both be present
* Name: requires "name"
*
* So we want to check for a match on one with more strings (Card.Name in this case) before
* checking for a match with the one containing less strings.
*/
// ProfileStop();
// printf("****** end profile\n");
wallet_tablesInitialized = PR_TRUE;
}
if (!unlockDatabase) {
return;
}
if (!namesInitialized) {
SI_GetCharPref(pref_WalletSchemaValueFileName, &schemaValueFileName);
if (!schemaValueFileName) {
schemaValueFileName = Wallet_RandomName("w");
SI_SetCharPref(pref_WalletSchemaValueFileName, schemaValueFileName);
}
SI_InitSignonFileName();
namesInitialized = PR_TRUE;
}
if (!wallet_ValuesReadIn) {
wallet_Clear(&wallet_SchemaToValue_list); /* otherwise we will duplicate the list */
wallet_ReadFromFile(schemaValueFileName, wallet_SchemaToValue_list, PR_TRUE);
wallet_ValuesReadIn = PR_TRUE;
}
#if DEBUG
// fprintf(stdout,"Field to Schema table \n");
// wallet_Dump(wallet_FieldToSchema_list);
// fprintf(stdout,"Vcard to Schema table \n");
// wallet_Dump(wallet_VcardToSchema_list);
// fprintf(stdout,"SchemaConcat table \n");
// wallet_Dump(wallet_SchemaConcat_list);
// fprintf(stdout,"SchemaStrings table \n");
// wallet_Dump(wallet_SchemaStrings_list);
// fprintf(stdout,"PositionalSchema table \n");
// wallet_Dump(wallet_PositionalSchema_list);
// fprintf(stdout,"StateSchema table \n");
// wallet_Dump(wallet_StateSchema_list);
// fprintf(stdout,"Schema to Value table \n");
// wallet_Dump(wallet_SchemaToValue_list);
#endif
}
static void
wallet_InitializeURLList() {
if (!wallet_URLListInitialized) {
wallet_Clear(&wallet_URL_list);
wallet_ReadFromFile(URLFileName, wallet_URL_list, PR_TRUE);
wallet_URLListInitialized = PR_TRUE;
}
}
/*
* initialization for current URL
*/
static void
wallet_InitializeCurrentURL(nsIDocument * doc) {
/* get url */
nsIURI *url = doc->GetDocumentURI();
if (wallet_lastUrl == url) {
return;
} else {
if (wallet_lastUrl) {
//?? NS_RELEASE(lastUrl);
}
wallet_lastUrl = url;
}
}
#define SEPARATOR "#*%$"
static nsresult
wallet_GetNextInString(const nsString& str, nsString& head, nsString& tail) {
PRInt32 separator = str.Find(SEPARATOR);
if (separator == -1) {
return NS_ERROR_FAILURE;
}
str.Left(head, separator);
str.Mid(tail, separator+sizeof(SEPARATOR)-1, str.Length() - (separator+sizeof(SEPARATOR)-1));
return NS_OK;
}
static void
wallet_ReleasePrefillElementList(nsVoidArray * wallet_PrefillElement_list) {
if (wallet_PrefillElement_list) {
wallet_PrefillElement * prefillElementPtr;
PRInt32 count = LIST_COUNT(wallet_PrefillElement_list);
for (PRInt32 i=count-1; i>=0; i--) {
prefillElementPtr = NS_STATIC_CAST(wallet_PrefillElement*, wallet_PrefillElement_list->ElementAt(i));
delete prefillElementPtr;
}
delete wallet_PrefillElement_list;
wallet_PrefillElement_list = nsnull;
}
}
#define BREAK PRUnichar('\001')
nsVoidArray * wallet_list;
nsAutoString wallet_url;
void
WLLT_GetPrefillListForViewer(nsAString& aPrefillList)
{
wallet_Initialize(PR_FALSE); /* to initialize helpMac */
wallet_PrefillElement * prefillElementPtr;
nsAutoString buffer;
PRInt32 count = LIST_COUNT(wallet_list);
for (PRInt32 i=0; i<count; i++) {
prefillElementPtr = NS_STATIC_CAST(wallet_PrefillElement*, wallet_list->ElementAt(i));
buffer.Append(BREAK);
buffer.AppendInt(prefillElementPtr->count,10);
buffer.Append(BREAK);
AppendUTF8toUTF16(prefillElementPtr->schema, buffer);
buffer.Append(BREAK);
buffer.Append(prefillElementPtr->value);
}
buffer.Append(BREAK);
buffer += wallet_url.get();
aPrefillList = buffer;
}
static void
wallet_FreeURL(wallet_MapElement *url) {
if(!url) {
return;
}
wallet_URL_list->RemoveElement(url);
PR_Free(url);
}
static const char permission_NoCapture_NoPreview[] = "yy";
static const char permission_NoCapture_Preview[] = "yn";
static const char permission_Capture_NoPreview[] = "ny";
static const char permission_Capture_Preview[] = "nn";
void
Wallet_SignonViewerReturn(const nsAString& results)
{
wallet_MapElement *url;
nsAutoString gone;
char oldPermissionChar;
/* step through all nopreviews and delete those that are in the sequence */
{
SI_FindValueInArgs(results, NS_LITERAL_STRING("|goneP|"), gone);
}
PRInt32 count = LIST_COUNT(wallet_URL_list);
while (count>0) {
count--;
url = NS_STATIC_CAST(wallet_MapElement*, wallet_URL_list->ElementAt(count));
if (url && SI_InSequence(gone, count)) {
/* clear the NO_PREVIEW indicator */
oldPermissionChar = NO_CAPTURE(url->item2);
WALLET_FREEIF (url->item2);
if (oldPermissionChar == 'y') {
url->item2 = PL_strdup(permission_NoCapture_Preview);
} else {
url->item2 = PL_strdup(permission_Capture_Preview);
}
if (!PL_strcmp(url->item2, permission_Capture_Preview)) {
wallet_FreeURL(url);
}
wallet_WriteToFile(URLFileName, wallet_URL_list);
}
}
/* step through all nocaptures and delete those that are in the sequence */
{
SI_FindValueInArgs(results, NS_LITERAL_STRING("|goneC|"), gone);
}
PRInt32 count2 = LIST_COUNT(wallet_URL_list);
while (count2>0) {
count2--;
url = NS_STATIC_CAST(wallet_MapElement*, wallet_URL_list->ElementAt(count2));
if (url && SI_InSequence(gone, count2)) {
/* clear the NO_CAPTURE indicator */
oldPermissionChar = NO_PREVIEW(url->item2);
WALLET_FREEIF (url->item2);
if (oldPermissionChar == 'y') {
url->item2 = PL_strdup(permission_Capture_NoPreview);
} else {
url->item2 = PL_strdup(permission_Capture_Preview);
}
if (!PL_strcmp(url->item2, permission_Capture_Preview)) {
wallet_FreeURL(url);
}
wallet_WriteToFile(URLFileName, wallet_URL_list);
}
}
}
/*
* see if user wants to capture data on current page
*/
static PRBool
wallet_OKToCapture(const nsAFlatCString& url, nsIDOMWindowInternal* window) {
/* exit if pref is not set */
if (!wallet_GetFormsCapturingPref() || !wallet_GetEnabledPref()) {
return PR_FALSE;
}
/* see if this url is already on list of url's for which we don't want to capture */
wallet_InitializeURLList();
nsVoidArray* dummy;
nsCAutoString urlPermissions;
if (wallet_ReadFromList(url, urlPermissions, dummy, wallet_URL_list, PR_FALSE)) {
if (NO_CAPTURE(urlPermissions) == 'y') {
return PR_FALSE;
}
}
/* ask user if we should capture the values on this form */
PRUnichar * message = Wallet_Localize("WantToCaptureForm?");
PRInt32 button = Wallet_3ButtonConfirm(message, window);
if (button == NEVER_BUTTON) {
/* add URL to list with NO_CAPTURE indicator set */
if (NO_PREVIEW(urlPermissions) == 'y') {
urlPermissions = permission_NoCapture_NoPreview;
} else {
urlPermissions = permission_NoCapture_Preview;
}
if (wallet_WriteToList(url.get(), urlPermissions.get(), dummy, wallet_URL_list, PR_FALSE, DUP_OVERWRITE)) {
wallet_WriteToFile(URLFileName, wallet_URL_list);
/* Notify signon manager dialog to update its display */
nsCOMPtr<nsIObserverService> os(do_GetService("@mozilla.org/observer-service;1"));
if (os) {
os->NotifyObservers(nsnull, "signonChanged", NS_LITERAL_STRING("nocaptures").get());
}
}
}
WALLET_FREE(message);
return (button == YES_BUTTON);
}
/*
* capture the value of a form element
*/
static PRBool
wallet_Capture(nsIDocument* doc, const nsString& field, const nsString& value, nsACString& schema)
{
/* do nothing if there is no value */
if (value.IsEmpty()) {
return PR_FALSE;
}
/* read in the mappings if they are not already present */
wallet_Initialize();
wallet_InitializeCurrentURL(doc);
NS_ConvertUTF16toUTF8 valueCString(value);
nsCAutoString oldValue;
/* is there a mapping from this field name to a schema name */
nsCAutoString localSchema; localSchema.Assign(schema);
nsVoidArray* dummy;
nsCAutoString stripField;
if (localSchema.IsEmpty()) {
Strip(field, stripField);
}
if (!localSchema.IsEmpty() ||
(wallet_ReadFromList(stripField, localSchema, dummy,
wallet_FieldToSchema_list, PR_FALSE))) {
/* field to schema mapping already exists */
/* is this a new value for the schema */
PRInt32 index = 0;
PRInt32 lastIndex = index;
while(wallet_ReadFromList(localSchema, oldValue, dummy, wallet_SchemaToValue_list, PR_TRUE, index)) {
PRBool isNewValue = !oldValue.Equals(valueCString.get());
if (!isNewValue) {
/*
* Remove entry from wallet_SchemaToValue_list and then reinsert. This will
* keep multiple values in that list for the same field ordered with
* most-recently-used first. That's useful since the first such entry
* is the default value used for pre-filling.
*/
wallet_MapElement * mapElement =
(wallet_MapElement *) (wallet_SchemaToValue_list->ElementAt(lastIndex));
wallet_SchemaToValue_list->RemoveElementAt(lastIndex);
wallet_WriteToList(
mapElement->item1,
mapElement->item2,
mapElement->itemList,
wallet_SchemaToValue_list,
PR_FALSE); /* note: obscure=false, otherwise we will obscure an obscured value */
delete mapElement;
return PR_TRUE;
}
lastIndex = index;
}
/* this is a new value so store it */
dummy = nsnull;
if (wallet_WriteToList(localSchema.get(), valueCString.get(), dummy, wallet_SchemaToValue_list, PR_TRUE)) {
wallet_WriteToFile(schemaValueFileName, wallet_SchemaToValue_list);
}
} else {
/* no field to schema mapping so assume schema name is same as field name */
/* is this a new value for the schema */
PRInt32 index = 0;
PRInt32 lastIndex = index;
nsAutoString concatParamUCS2;
wallet_GetHostFile(wallet_lastUrl, concatParamUCS2);
concatParamUCS2.AppendLiteral(":");
concatParamUCS2.Append(field);
NS_ConvertUTF16toUTF8 concatParamUTF8(concatParamUCS2);
while(wallet_ReadFromList
(concatParamUTF8, oldValue, dummy, wallet_SchemaToValue_list, PR_TRUE, index)) {
PRBool isNewValue = !oldValue.Equals(valueCString.get());
if (!isNewValue) {
/*
* Remove entry from wallet_SchemaToValue_list and then reinsert. This will
* keep multiple values in that list for the same field ordered with
* most-recently-used first. That's useful since the first such entry
* is the default value used for pre-filling.
*/
wallet_MapElement * mapElement =
(wallet_MapElement *) (wallet_SchemaToValue_list->ElementAt(lastIndex));
wallet_SchemaToValue_list->RemoveElementAt(lastIndex);
wallet_WriteToList(
mapElement->item1,
mapElement->item2,
mapElement->itemList,
wallet_SchemaToValue_list,
PR_FALSE); /* note: obscure=false, otherwise we will obscure an obscured value */
delete mapElement;
return PR_TRUE;
}
lastIndex = index;
//??? aren't these next four lines redundant?
wallet_GetHostFile(wallet_lastUrl, concatParamUCS2);
concatParamUCS2.AppendLiteral(":");
concatParamUCS2.Append(field);
CopyUTF16toUTF8(concatParamUCS2, concatParamUTF8);
}
/* this is a new value so store it */
dummy = nsnull;
nsAutoString hostFileFieldUCS2;
wallet_GetHostFile(wallet_lastUrl, hostFileFieldUCS2);
hostFileFieldUCS2.AppendLiteral(":");
hostFileFieldUCS2.Append(field);
if (wallet_WriteToList
(NS_ConvertUCS2toUTF8(hostFileFieldUCS2).get(), valueCString.get(), dummy,
wallet_SchemaToValue_list, PR_TRUE)) {
wallet_WriteToFile(schemaValueFileName, wallet_SchemaToValue_list);
}
}
return PR_TRUE;
}
/***************************************************************/
/* The following are the interface routines seen by other dlls */
/***************************************************************/
void
WLLT_GetNopreviewListForViewer(nsAString& aNopreviewList)
{
wallet_Initialize(PR_FALSE); /* to initialize helpMac */
nsAutoString buffer;
wallet_MapElement *url;
wallet_InitializeURLList();
PRInt32 count = LIST_COUNT(wallet_URL_list);
for (PRInt32 i=0; i<count; i++) {
url = NS_STATIC_CAST(wallet_MapElement*, wallet_URL_list->ElementAt(i));
if (NO_PREVIEW(url->item2) == 'y') {
buffer.Append(BREAK);
AppendUTF8toUTF16(url->item1, buffer);
}
}
aNopreviewList = buffer;
}
void
WLLT_GetNocaptureListForViewer(nsAString& aNocaptureList)
{
nsAutoString buffer;
wallet_MapElement *url;
wallet_InitializeURLList();
PRInt32 count = LIST_COUNT(wallet_URL_list);
for (PRInt32 i=0; i<count; i++) {
url = NS_STATIC_CAST(wallet_MapElement*, wallet_URL_list->ElementAt(i));
if (NO_CAPTURE(url->item2) == 'y') {
buffer.Append(BREAK);
AppendUTF8toUTF16(url->item1, buffer);
}
}
aNocaptureList = buffer;
}
void
WLLT_PostEdit(const nsAString& walletList)
{
nsCOMPtr<nsIFile> file;
nsresult rv = Wallet_ProfileDirectory(getter_AddRefs(file));
if (NS_FAILED(rv)) {
return;
}
nsAutoString tail( walletList );
nsAutoString head, temp;
PRInt32 separator;
/* get first item in list */
separator = tail.FindChar(BREAK);
if (-1 == separator) {
return;
}
tail.Left(head, separator);
tail.Mid(temp, separator+1, tail.Length() - (separator+1));
tail = temp;
/* return if OK button was not pressed */
if (!head.EqualsLiteral("OK")) {
return;
}
file->AppendNative(nsDependentCString(schemaValueFileName));
/* open SchemaValue file */
nsCOMPtr<nsIOutputStream> fileOutputStream;
rv = NS_NewSafeLocalFileOutputStream(getter_AddRefs(fileOutputStream),
file,
-1,
0600);
if (NS_FAILED(rv))
return;
nsCOMPtr<nsIOutputStream> strm;
rv = NS_NewBufferedOutputStream(getter_AddRefs(strm), fileOutputStream, 4096);
if (NS_FAILED(rv))
return;
/* write the values in the walletList to the file */
wallet_PutHeader(strm);
for (;;) {
separator = tail.FindChar(BREAK);
if (-1 == separator) {
break;
}
tail.Left(head, separator);
tail.Mid(temp, separator+1, tail.Length() - (separator+1));
tail = temp;
wallet_PutLine(strm, NS_ConvertUCS2toUTF8(head).get());
}
/* close the file and read it back into the SchemaToValue list */
// All went ok. Maybe except for problems in Write(), but the stream detects
// that for us
nsCOMPtr<nsISafeOutputStream> safeStream = do_QueryInterface(strm);
NS_ASSERTION(safeStream, "expected a safe output stream!");
if (safeStream) {
rv = safeStream->Finish();
if (NS_FAILED(rv)) {
NS_WARNING("failed to save wallet file! possible dataloss");
return;
}
}
strm = nsnull;
fileOutputStream = nsnull;
wallet_Clear(&wallet_SchemaToValue_list);
wallet_ReadFromFile(schemaValueFileName, wallet_SchemaToValue_list, PR_TRUE);
}
void
WLLT_PreEdit(nsAString& walletList)
{
wallet_Initialize();
walletList.Assign(BREAK);
wallet_MapElement * mapElementPtr;
PRInt32 count = LIST_COUNT(wallet_SchemaToValue_list);
for (PRInt32 i=0; i<count; i++) {
mapElementPtr = NS_STATIC_CAST(wallet_MapElement*, wallet_SchemaToValue_list->ElementAt(i));
AppendUTF8toUTF16(mapElementPtr->item1, walletList);
walletList.Append(BREAK);
if (!WALLET_NULL(mapElementPtr->item2)) {
AppendUTF8toUTF16(mapElementPtr->item2, walletList);
walletList.Append(BREAK);
} else {
wallet_Sublist * sublistPtr;
PRInt32 count2 = LIST_COUNT(mapElementPtr->itemList);
for (PRInt32 i2=0; i2<count2; i2++) {
sublistPtr = NS_STATIC_CAST(wallet_Sublist*, mapElementPtr->itemList->ElementAt(i2));
AppendUTF8toUTF16(sublistPtr->item, walletList);
walletList.Append(BREAK);
}
}
walletList.Append(BREAK);
}
}
void
WLLT_DeleteAll() {
wallet_Initialize();
wallet_Clear(&wallet_SchemaToValue_list);
wallet_WriteToFile(schemaValueFileName, wallet_SchemaToValue_list);
SI_DeleteAll();
}
void
WLLT_ClearUserData() {
wallet_ValuesReadIn = PR_FALSE;
namesInitialized = PR_FALSE;
wallet_URLListInitialized = PR_FALSE;
}
void
WLLT_DeletePersistentUserData() {
if (schemaValueFileName && schemaValueFileName[0]) {
nsCOMPtr<nsIFile> file;
nsresult rv = Wallet_ProfileDirectory(getter_AddRefs(file));
if (NS_SUCCEEDED(rv)) {
rv = file->AppendNative(nsDependentCString(schemaValueFileName));
if (NS_SUCCEEDED(rv))
file->Remove(PR_FALSE);
}
}
}
int PR_CALLBACK
wallet_ReencryptAll(const char * newpref, void* window) {
PRUnichar * message;
/* prevent reentry for the case that the user doesn't supply correct master password */
if (gReencryptionLevel != 0) {
return 0; /* this is PREF_NOERROR but we no longer include prefapi.h */
}
gReencryptionLevel ++;
PRInt32 count = LIST_COUNT(wallet_SchemaToValue_list);
PRInt32 i = 0;
char* plainText = nsnull;
/* logout first so there is no conversion unless user knows the master password */
if (!changingPassword) {
nsresult rv = wallet_CryptSetup();
if (NS_SUCCEEDED(rv)) {
rv = gSecretDecoderRing->Logout();
}
if (NS_FAILED(rv)) {
goto fail;
}
wallet_Initialize();
}
wallet_MapElement * mapElementPtr;
gEncryptionFailure = PR_FALSE;
for (i=0; i<count && !gEncryptionFailure; i++) {
mapElementPtr = NS_STATIC_CAST(wallet_MapElement*, wallet_SchemaToValue_list->ElementAt(i));
char * crypt = nsnull;
if (!WALLET_NULL(mapElementPtr->item2)) {
if (NS_FAILED(DecryptString(mapElementPtr->item2, plainText))) {
goto fail;
}
if (NS_FAILED(EncryptString(plainText, crypt))) {
goto fail;
}
mapElementPtr->item2 = crypt;
} else {
wallet_Sublist * sublistPtr;
PRInt32 count2 = LIST_COUNT(mapElementPtr->itemList);
for (PRInt32 i2=0; i2<count2; i2++) {
sublistPtr = NS_STATIC_CAST(wallet_Sublist*, mapElementPtr->itemList->ElementAt(i2));
if (NS_FAILED(DecryptString(sublistPtr->item, plainText))) {
goto fail;
}
if (NS_FAILED(EncryptString(plainText, crypt))) {
goto fail;
}
sublistPtr->item = crypt;
}
}
}
wallet_WriteToFile(schemaValueFileName, wallet_SchemaToValue_list);
if (!SINGSIGN_ReencryptAll()) {
goto fail;
}
/* force a rewriting of prefs.js to make sure pref_Crypto got updated
*
* Note: In the event of a crash after changing this pref (either way), the user
* could get misled as to what state his storage was in. If the crash occurred
* after changing to encrypted, he could think he was encrypting in the future (because
* he remembered changed to encypting at one time) but his new values are only being
* obscurred. If the crash occurred after changing to obscured, later on he might
* think his store was encrypted (because he checked the pref panel and that's what
* it told him) whereas some of the earlier values are actually obscured and so not
* protected. For both these reasons, we force this rewriting of the prefs file now.
*/
SI_SetBoolPref(pref_Crypto, SI_GetBoolPref(pref_Crypto, PR_TRUE));
// message = Wallet_Localize("Converted");
// wallet_Alert(message, (nsIDOMWindowInternal *)window);
// WALLET_FREE(message);
gReencryptionLevel--;
return 0; /* this is PREF_NOERROR but we no longer include prefapi.h */
fail:
/* toggle the pref back to its previous value */
SI_SetBoolPref(pref_Crypto, !SI_GetBoolPref(pref_Crypto, PR_TRUE));
/* alert the user to the failure */
message = Wallet_Localize("NotConverted");
wallet_Alert(message, (nsIDOMWindowInternal *)window);
WALLET_FREE(message);
gReencryptionLevel--;
return 1;
}
void
WLLT_InitReencryptCallback(nsIDOMWindowInternal* window) {
static PRBool registered = PR_FALSE;
static nsIDOMWindowInternal* lastWindow;
if (registered) {
SI_UnregisterCallback(pref_Crypto, wallet_ReencryptAll, lastWindow);
}
SI_RegisterCallback(pref_Crypto, wallet_ReencryptAll, window);
lastWindow = window;
registered = PR_TRUE;
}
static void
wallet_DecodeVerticalBars(nsString& s) {
s.ReplaceSubstring(NS_LITERAL_STRING("^2").get(), NS_LITERAL_STRING("|").get());
s.ReplaceSubstring(NS_LITERAL_STRING("^1").get(), NS_LITERAL_STRING("^").get());
}
/*
* return after previewing a set of prefills
*/
void
WLLT_PrefillReturn(const nsAString& results)
{
nsAutoString fillins;
nsAutoString urlName;
nsAutoString skip;
nsAutoString next;
/* get values that are in environment variables */
SI_FindValueInArgs(results, NS_LITERAL_STRING("|fillins|"), fillins);
SI_FindValueInArgs(results, NS_LITERAL_STRING("|skip|"), skip);
SI_FindValueInArgs(results, NS_LITERAL_STRING("|url|"), urlName);
wallet_DecodeVerticalBars(fillins);
wallet_DecodeVerticalBars(urlName);
/* add url to url list if user doesn't want to preview this page in the future */
if (skip.EqualsLiteral("true")) {
NS_ConvertUTF16toUTF8 url(urlName);
nsVoidArray* dummy;
nsCAutoString urlPermissions("nn");
wallet_ReadFromList(url, urlPermissions, dummy, wallet_URL_list, PR_FALSE);
/* add URL to list with NO_PREVIEW indicator set */
if (NO_CAPTURE(urlPermissions) == 'y') {
urlPermissions = permission_NoCapture_NoPreview;
} else {
urlPermissions = permission_Capture_NoPreview;
}
if (wallet_WriteToList(url.get(), urlPermissions.get(), dummy, wallet_URL_list, PR_FALSE, DUP_OVERWRITE)) {
wallet_WriteToFile(URLFileName, wallet_URL_list);
/* Notify signon manager dialog to update its display */
nsCOMPtr<nsIObserverService> os(do_GetService("@mozilla.org/observer-service;1"));
if (os) {
os->NotifyObservers(nsnull, "signonChanged", NS_LITERAL_STRING("nopreviews").get());
}
}
}
/* process the list, doing the fillins */
if (fillins.Length() == 0) { /* user pressed CANCEL */
wallet_ReleasePrefillElementList(wallet_list);
wallet_list = nsnull;
return;
}
/*
* note: there are two lists involved here and we are stepping through both of them.
* One is the pre-fill list that was generated when we walked through the html content.
* For each pre-fillable item, it contains n entries, one for each possible value that
* can be prefilled for that field. The first entry for each field can be identified
* because it has a non-zero count field (in fact, the count is the number of entries
* for that field), all subsequent entries for the same field have a zero count field.
* The other is the fillin list which was generated by the html dialog that just
* finished. It contains one entry for each pre-fillable item specificying that
* particular value that should be prefilled for that item.
*/
wallet_PrefillElement * mapElementPtr;
/* step through pre-fill list */
PRInt32 count = LIST_COUNT(wallet_list);
for (PRInt32 i=0; i<count; i++) {
mapElementPtr = NS_STATIC_CAST(wallet_PrefillElement*, wallet_list->ElementAt(i));
/* advance in fillins list each time a new schema name in pre-fill list is encountered */
if (mapElementPtr->count != 0) {
/* count != 0 indicates a new schema name */
nsAutoString tail;
if (NS_FAILED(wallet_GetNextInString(fillins, next, tail))) {
break;
}
fillins = tail;
if (PL_strcmp(NS_ConvertUCS2toUTF8(next).get(), mapElementPtr->schema)) {
break; /* something's wrong so stop prefilling */
}
wallet_GetNextInString(fillins, next, tail);
fillins = tail;
}
if (next == mapElementPtr->value) {
/*
* Remove entry from wallet_SchemaToValue_list and then reinsert. This will
* keep multiple values in that list for the same field ordered with
* most-recently-used first. That's useful since the first such entry
* is the default value used for pre-filling.
*/
/*
* Test for mapElementPtr->count being zero is an optimization that avoids us from doing a
* reordering if the current entry already was first
*/
if (mapElementPtr->count == 0) {
nsCAutoString oldvalueUTF8;
PRInt32 index = 0;
PRInt32 lastIndex = index;
nsVoidArray* dummy;
while(wallet_ReadFromList(nsDependentCString(mapElementPtr->schema),
oldvalueUTF8,
dummy,
wallet_SchemaToValue_list,
PR_TRUE,
index)) {
if (oldvalueUTF8.Equals(NS_ConvertUCS2toUTF8(mapElementPtr->value).get())) {
wallet_MapElement * mapElement =
(wallet_MapElement *) (wallet_SchemaToValue_list->ElementAt(lastIndex));
wallet_SchemaToValue_list->RemoveElementAt(lastIndex);
wallet_WriteToList(
mapElement->item1,
mapElement->item2,
mapElement->itemList,
wallet_SchemaToValue_list,
PR_FALSE); /* note: obscure=false, otherwise we will obscure an obscured value */
delete mapElement;
break;
}
lastIndex = index;
}
}
}
/* Change the value */
if (!next.IsEmpty()) {
if (mapElementPtr->inputElement) {
mapElementPtr->inputElement->SetValue(next);
} else {
nsresult result;
result = wallet_GetSelectIndex(mapElementPtr->selectElement, next, mapElementPtr->selectIndex);
if (NS_SUCCEEDED(result)) {
mapElementPtr->selectElement->SetSelectedIndex(mapElementPtr->selectIndex);
} else {
mapElementPtr->selectElement->SetSelectedIndex(0);
}
}
}
}
/* Release the prefill list that was generated when we walked thru the html content */
wallet_ReleasePrefillElementList(wallet_list);
wallet_list = nsnull;
}
/*
* get the form elements on the current page and prefill them if possible
*/
static void
wallet_TraversalForPrefill
(nsIDOMWindow* win, nsVoidArray* wallet_PrefillElement_list, nsString& urlName) {
nsresult result;
if (nsnull != win) {
nsCOMPtr<nsIDOMDocument> domdoc;
result = win->GetDocument(getter_AddRefs(domdoc));
if (NS_SUCCEEDED(result)) {
nsCOMPtr<nsIDocument> doc = do_QueryInterface(domdoc);
if (doc) {
nsIURI *url = doc->GetDocumentURI();
if (url) {
wallet_GetHostFile(url, urlName);
}
wallet_Initialize();
wallet_InitializeCurrentURL(doc);
nsCOMPtr<nsIDOMHTMLDocument> htmldoc = do_QueryInterface(doc);
if (htmldoc) {
nsCOMPtr<nsIDOMHTMLCollection> forms;
htmldoc->GetForms(getter_AddRefs(forms));
if (forms) {
wallet_InitializeStateTesting();
PRUint32 numForms;
forms->GetLength(&numForms);
for (PRUint32 formX = 0; (formX < numForms) && !gEncryptionFailure; formX++) {
nsCOMPtr<nsIDOMNode> formNode;
forms->Item(formX, getter_AddRefs(formNode));
if (formNode) {
nsCOMPtr<nsIDOMHTMLFormElement> formElement = do_QueryInterface(formNode);
if (formElement) {
nsCOMPtr<nsIDOMHTMLCollection> elements;
result = formElement->GetElements(getter_AddRefs(elements));
if (elements) {
/* got to the form elements at long last */
PRUint32 numElements;
elements->GetLength(&numElements);
for (PRUint32 elementX = 0; (elementX < numElements) && !gEncryptionFailure; elementX++) {
nsCOMPtr<nsIDOMNode> elementNode;
elements->Item(elementX, getter_AddRefs(elementNode));
if (elementNode) {
wallet_PrefillElement * prefillElement;
PRInt32 index = 0;
wallet_PrefillElement * firstElement = nsnull;
PRUint32 numberOfElements = 0;
for (; !gEncryptionFailure;) {
/* loop to allow for multiple values */
/* first element in multiple-value group will have its count
* field set to the number of elements in group. All other
* elements in group will have count field set to 0
*/
prefillElement = new wallet_PrefillElement;
nsCAutoString schemaUTF8;
if (NS_SUCCEEDED(wallet_GetPrefills
(elementNode,
prefillElement->inputElement,
prefillElement->selectElement,
schemaUTF8,
prefillElement->value,
prefillElement->selectIndex,
index))) {
/* another value found */
prefillElement->schema = ToNewCString(schemaUTF8);
if (nsnull == firstElement) {
firstElement = prefillElement;
}
numberOfElements++;
prefillElement->count = 0;
wallet_PrefillElement_list->AppendElement(prefillElement);
} else {
/* value not found, stop looking for more values */
delete prefillElement;
break;
}
} // for
if (numberOfElements>0) {
firstElement->count = numberOfElements;
}
}
} // for
}
}
}
}
}
}
}
}
}
nsCOMPtr<nsIDOMWindowCollection> frames;
win->GetFrames(getter_AddRefs(frames));
if (frames) {
PRUint32 numFrames;
frames->GetLength(&numFrames);
for (PRUint32 frameX = 0; (frameX < numFrames) && !gEncryptionFailure; frameX++) {
nsCOMPtr<nsIDOMWindow> frameNode;
frames->Item(frameX, getter_AddRefs(frameNode));
if (frameNode) {
wallet_TraversalForPrefill(frameNode, wallet_PrefillElement_list, urlName);
}
}
}
}
nsresult
WLLT_PrefillOneElement
(nsIDOMWindowInternal* win, nsIDOMNode* elementNode, nsAString& compositeValue)
{
nsIDOMHTMLInputElement* inputElement;
nsIDOMHTMLSelectElement* selectElement;
nsCAutoString schema;
nsString value;
PRInt32 selectIndex = 0;
PRInt32 index = 0;
if (nsnull != win) {
nsCOMPtr<nsIDOMDocument> domdoc;
nsresult result = win->GetDocument(getter_AddRefs(domdoc));
if (NS_SUCCEEDED(result)) {
nsCOMPtr<nsIDocument> doc = do_QueryInterface(domdoc);
if (doc) {
wallet_Initialize(PR_TRUE);
wallet_InitializeCurrentURL(doc);
wallet_InitializeStateTesting();
while (NS_SUCCEEDED(wallet_GetPrefills
(elementNode,
inputElement,
selectElement,
schema,
value,
selectIndex,
index))) {
compositeValue.Append(BREAK);
compositeValue.Append(value);
}
}
}
}
return NS_OK;
}
nsresult
WLLT_Prefill(nsIPresShell* shell, PRBool quick, nsIDOMWindowInternal* win)
{
/* do not prefill if preview window is open in some other browser window */
if (wallet_list) {
return NS_ERROR_FAILURE;
}
/* create list of elements that can be prefilled */
nsVoidArray *wallet_PrefillElement_list=new nsVoidArray();
if (!wallet_PrefillElement_list) {
return NS_ERROR_FAILURE;
}
nsAutoString urlName;
gEncryptionFailure = PR_FALSE;
wallet_TraversalForPrefill(win, wallet_PrefillElement_list, urlName);
/* return if no elements were put into the list */
if (LIST_COUNT(wallet_PrefillElement_list) == 0) {
if (!gEncryptionFailure) {
PRUnichar * message = Wallet_Localize("noPrefills");
wallet_Alert(message, win);
WALLET_FREE(message);
}
// Shouldn't wallet_PrefillElement_list be deleted here?
return NS_ERROR_FAILURE; // indicates to caller not to display preview screen
}
/* prefill each element using the list */
/* determine if url is on list of urls that should not be previewed */
PRBool noPreview = PR_FALSE;
if (!quick) {
wallet_InitializeURLList();
nsVoidArray* dummy;
nsCAutoString urlPermissions;
if (!urlName.IsEmpty()) {
wallet_ReadFromList
(NS_ConvertUCS2toUTF8(urlName), urlPermissions, dummy, wallet_URL_list, PR_FALSE);
noPreview = (NO_PREVIEW(urlPermissions) == 'y');
}
}
/* determine if preview is necessary */
if (noPreview || quick) {
/* prefill each element without any preview for user verification */
wallet_PrefillElement * mapElementPtr;
PRInt32 count = LIST_COUNT(wallet_PrefillElement_list);
for (PRInt32 i=0; i<count; i++) {
mapElementPtr = NS_STATIC_CAST(wallet_PrefillElement*, wallet_PrefillElement_list->ElementAt(i));
if (mapElementPtr->count) {
if (mapElementPtr->inputElement) {
mapElementPtr->inputElement->SetValue(mapElementPtr->value);
} else {
mapElementPtr->selectElement->SetSelectedIndex(mapElementPtr->selectIndex);
}
}
}
/* go thru list just generated and release everything */
wallet_ReleasePrefillElementList(wallet_PrefillElement_list);
return NS_ERROR_FAILURE; // indicates to caller not to display preview screen
} else {
/* let user preview and verify the prefills first */
wallet_list = wallet_PrefillElement_list;
wallet_url = urlName;
#ifdef DEBUG_morse
////wallet_DumpStopwatch();
////wallet_ClearStopwatch();
//wallet_DumpTiming();
//wallet_ClearTiming();
#endif
return NS_OK; // indicates that caller is to display preview screen
}
}
static PRBool
wallet_CaptureInputElement(nsIDOMNode* elementNode, nsIDocument* doc) {
nsresult result;
PRBool captured = PR_FALSE;
nsCOMPtr<nsIDOMHTMLInputElement> inputElement = do_QueryInterface(elementNode);
if (inputElement) {
/* it's an input element */
nsAutoString type;
result = inputElement->GetType(type);
if (NS_SUCCEEDED(result) &&
(type.IsEmpty() ||
type.LowerCaseEqualsLiteral("text"))) {
nsAutoString field;
result = inputElement->GetName(field);
if (NS_SUCCEEDED(result)) {
nsAutoString value;
result = inputElement->GetValue(value);
if (NS_SUCCEEDED(result)) {
/* get schema name from vcard attribute if it exists */
nsCAutoString schema;
nsCOMPtr<nsIDOMElement> element = do_QueryInterface(elementNode);
if (element) {
nsAutoString vcardName; vcardName.AssignLiteral("VCARD_NAME");
nsAutoString vcardValueUCS2;
result = element->GetAttribute(vcardName, vcardValueUCS2);
if (NS_OK == result) {
nsVoidArray* dummy;
wallet_ReadFromList(NS_ConvertUCS2toUTF8(vcardValueUCS2), schema, dummy,
wallet_VcardToSchema_list, PR_FALSE);
}
}
if (schema.IsEmpty()) {
/* get schema from displayable text if possible */
wallet_GetSchemaFromDisplayableText(inputElement, schema, value.IsEmpty());
}
if (wallet_Capture(doc, field, value, schema)) {
captured = PR_TRUE;
}
}
}
}
}
return captured;
}
static PRBool
wallet_CaptureSelectElement(nsIDOMNode* elementNode, nsIDocument* doc) {
nsresult result;
PRBool captured = PR_FALSE;
nsCOMPtr<nsIDOMHTMLSelectElement> selectElement = do_QueryInterface(elementNode);
if (selectElement) {
/* it's a dropdown list */
nsAutoString field;
result = selectElement->GetName(field);
if (NS_SUCCEEDED(result)) {
PRUint32 length;
selectElement->GetLength(&length);
nsCOMPtr<nsIDOMHTMLOptionsCollection> options;
selectElement->GetOptions(getter_AddRefs(options));
if (options) {
PRInt32 selectedIndex;
result = selectElement->GetSelectedIndex(&selectedIndex);
if (NS_SUCCEEDED(result)) {
nsCOMPtr<nsIDOMNode> optionNode;
options->Item(selectedIndex, getter_AddRefs(optionNode));
if (optionNode) {
nsCOMPtr<nsIDOMHTMLOptionElement> optionElement(do_QueryInterface(optionNode));
if (optionElement) {
nsAutoString optionValue;
nsAutoString optionText;
PRBool valueOK = NS_SUCCEEDED(optionElement->GetValue(optionValue))
&& optionValue.Length();
PRBool textOK = NS_SUCCEEDED(optionElement->GetText(optionText))
&& optionText.Length();
if (valueOK || textOK) {
/* get schema name from vcard attribute if it exists */
nsCAutoString schema;
nsCOMPtr<nsIDOMElement> element = do_QueryInterface(elementNode);
if (element) {
nsAutoString vcardName; vcardName.AssignLiteral("VCARD_NAME");
nsAutoString vcardValueUCS2;
result = element->GetAttribute(vcardName, vcardValueUCS2);
if (NS_OK == result) {
nsVoidArray* dummy;
wallet_ReadFromList(NS_ConvertUCS2toUTF8(vcardValueUCS2), schema, dummy,
wallet_VcardToSchema_list, PR_FALSE);
}
}
if (schema.IsEmpty()) {
/* get schema from displayable text if possible */
wallet_GetSchemaFromDisplayableText(selectElement, schema, (!valueOK && !textOK));
}
if (valueOK && wallet_Capture(doc, field, optionValue, schema)) {
captured = PR_TRUE;
}
optionText.Trim(" \n\t\r");
if (textOK && wallet_Capture(doc, field, optionText, schema)) {
captured = PR_TRUE;
}
}
}
}
}
}
}
}
return captured;
}
static void
wallet_TraversalForRequestToCapture(nsIDOMWindow* win, PRInt32& captureCount) {
nsresult result;
if (nsnull != win) {
nsCOMPtr<nsIDOMDocument> domdoc;
result = win->GetDocument(getter_AddRefs(domdoc));
if (NS_SUCCEEDED(result)) {
nsCOMPtr<nsIDocument> doc = do_QueryInterface(domdoc);
if (doc) {
wallet_Initialize();
wallet_InitializeCurrentURL(doc);
nsCOMPtr<nsIDOMHTMLDocument> htmldoc = do_QueryInterface(doc);
if (htmldoc) {
nsCOMPtr<nsIDOMHTMLCollection> forms;
htmldoc->GetForms(getter_AddRefs(forms));
if (forms) {
wallet_InitializeStateTesting();
PRUint32 numForms;
forms->GetLength(&numForms);
for (PRUint32 formX = 0; (formX < numForms) && !gEncryptionFailure; formX++) {
nsCOMPtr<nsIDOMNode> formNode;
forms->Item(formX, getter_AddRefs(formNode));
if (formNode) {
nsCOMPtr<nsIDOMHTMLFormElement> formElement = do_QueryInterface(formNode);
if (formElement) {
nsCOMPtr<nsIDOMHTMLCollection> elements;
result = formElement->GetElements(getter_AddRefs(elements));
if (elements) {
/* got to the form elements at long last */
/* now find out how many text fields are on the form */
PRUint32 numElements;
elements->GetLength(&numElements);
for (PRUint32 elementY = 0; (elementY < numElements) && !gEncryptionFailure; elementY++) {
nsCOMPtr<nsIDOMNode> elementNode;
elements->Item(elementY, getter_AddRefs(elementNode));
if (elementNode) {
if (wallet_CaptureInputElement(elementNode, doc)) {
captureCount++;
}
if (wallet_CaptureSelectElement(elementNode, doc)) {
captureCount++;
}
}
}
}
}
}
}
}
}
}
}
}
nsCOMPtr<nsIDOMWindowCollection> frames;
win->GetFrames(getter_AddRefs(frames));
if (frames) {
PRUint32 numFrames;
frames->GetLength(&numFrames);
for (PRUint32 frameX = 0; (frameX < numFrames) && !gEncryptionFailure; frameX++)
{
nsCOMPtr<nsIDOMWindow> frameNode;
frames->Item(frameX, getter_AddRefs(frameNode));
if (frameNode) {
wallet_TraversalForRequestToCapture(frameNode, captureCount);
}
}
}
}
void
WLLT_RequestToCapture(nsIPresShell* shell, nsIDOMWindowInternal* win, PRUint32* status) {
PRInt32 captureCount = 0;
gEncryptionFailure = PR_FALSE;
wallet_TraversalForRequestToCapture(win, captureCount);
PRUnichar * message;
if (gEncryptionFailure) {
message = Wallet_Localize("UnableToCapture");
*status = 0;
} else if (captureCount) {
/* give caveat if this is the first time data is being captured */
Wallet_GiveCaveat(win, nsnull);
message = Wallet_Localize("Captured");
*status = 0;
} else {
message = Wallet_Localize("NotCaptured");
*status = +1;
}
wallet_Alert(message, win);
WALLET_FREE(message);
}
static PRBool
wallet_IsNewValue(nsIDOMNode* elementNode, nsString valueOnForm) {
if (valueOnForm.Equals(EmptyString())) {
return PR_FALSE;
}
nsIDOMHTMLInputElement* inputElement;
nsIDOMHTMLSelectElement* selectElement;
nsCAutoString schema;
nsAutoString valueSaved;
PRInt32 selectIndex = 0;
PRInt32 index = 0;
while (NS_SUCCEEDED(wallet_GetPrefills
(elementNode, inputElement, selectElement, schema, valueSaved, selectIndex, index))) {
if (valueOnForm.Equals(valueSaved)) {
return PR_FALSE;
}
}
return PR_TRUE;
}
void
WLLT_OnSubmit(nsIContent* currentForm, nsIDOMWindowInternal* window) {
nsCOMPtr<nsIDOMHTMLFormElement> currentFormNode(do_QueryInterface(currentForm));
/* get url name as ascii string */
nsAutoString strippedURLNameUCS2;
nsCOMPtr<nsIDocument> doc = currentForm->GetDocument();
if (!doc) {
return;
}
nsIURI *docURL = doc->GetDocumentURI();
if (!docURL) {
return;
}
wallet_GetHostFile(docURL, strippedURLNameUCS2);
NS_ConvertUTF16toUTF8 strippedURLNameUTF8(strippedURLNameUCS2);
/* get to the form elements */
nsCOMPtr<nsIDOMHTMLDocument> htmldoc(do_QueryInterface(doc));
if (htmldoc == nsnull) {
return;
}
nsCOMPtr<nsIDOMHTMLCollection> forms;
nsresult rv = htmldoc->GetForms(getter_AddRefs(forms));
if (NS_FAILED(rv) || (forms == nsnull)) {
return;
}
PRUint32 numForms;
forms->GetLength(&numForms);
for (PRUint32 formX = 0; formX < numForms; formX++) {
nsCOMPtr<nsIDOMNode> formNode;
forms->Item(formX, getter_AddRefs(formNode));
if (nsnull != formNode) {
nsCOMPtr<nsIDOMHTMLFormElement> formElement(do_QueryInterface(formNode));
if ((nsnull != formElement)) {
nsCOMPtr<nsIDOMHTMLCollection> elements;
rv = formElement->GetElements(getter_AddRefs(elements));
if ((NS_SUCCEEDED(rv)) && (nsnull != elements)) {
/* got to the form elements at long last */
nsVoidArray * signonData = new nsVoidArray();
si_SignonDataStruct * data;
PRUint32 numElements;
elements->GetLength(&numElements);
PRBool OKToPrompt = PR_FALSE;
PRInt32 passwordcount = 0;
PRInt32 hits = 0;
wallet_Initialize(PR_FALSE);
wallet_InitializeCurrentURL(doc);
wallet_InitializeStateTesting();
PRBool newValueFound = PR_FALSE;
for (PRUint32 elementX = 0; elementX < numElements; elementX++) {
nsCOMPtr<nsIDOMNode> elementNode;
elements->Item(elementX, getter_AddRefs(elementNode));
if (nsnull != elementNode) {
nsCOMPtr<nsIDOMHTMLSelectElement> selectElement(do_QueryInterface(elementNode));
if ((NS_SUCCEEDED(rv)) && (nsnull != selectElement)) {
if (passwordcount == 0 && !newValueFound && !OKToPrompt) {
nsAutoString valueOnForm;
rv = selectElement->GetValue(valueOnForm);
if (NS_SUCCEEDED(rv) && wallet_IsNewValue (elementNode, valueOnForm)) {
newValueFound = PR_TRUE;
if (hits > 1) {
OKToPrompt = PR_TRUE;
}
}
}
}
nsCOMPtr<nsIDOMHTMLInputElement> inputElement(do_QueryInterface(elementNode));
if ((NS_SUCCEEDED(rv)) && (nsnull != inputElement)) {
nsAutoString type;
rv = inputElement->GetType(type);
if (NS_SUCCEEDED(rv)) {
PRBool isText = (type.IsEmpty() || type.LowerCaseEqualsLiteral("text"));
PRBool isPassword = type.LowerCaseEqualsLiteral("password");
// don't save password if field was left blank
if (isPassword) {
nsAutoString val;
(void) inputElement->GetValue(val);
if (val.IsEmpty()) {
isPassword = PR_FALSE;
}
}
// Do not store this 'password' form element if the 'autocomplete = off'
// attribute is present, unless the 'wallet.crypto.autocompleteoverride'
// preference is enabled. (The "autocomplete" property is a Microsoft
// extension to HTML.)
if (isPassword && !SI_GetBoolPref(pref_AutoCompleteOverride, PR_FALSE)) {
nsAutoString val;
(void) inputElement->GetAttribute(NS_LITERAL_STRING("autocomplete"), val);
if (val.LowerCaseEqualsLiteral("off")) {
isPassword = PR_FALSE;
} else {
(void) formElement->GetAttribute(NS_LITERAL_STRING("autocomplete"), val);
if (val.LowerCaseEqualsLiteral("off")) {
isPassword = PR_FALSE;
}
}
}
if (isPassword) {
passwordcount++;
OKToPrompt = PR_FALSE;
}
if (isText) {
if (passwordcount == 0 && !newValueFound && !OKToPrompt) {
nsAutoString valueOnForm;
rv = inputElement->GetValue(valueOnForm);
if (NS_SUCCEEDED(rv) && wallet_IsNewValue (elementNode, valueOnForm)) {
newValueFound = PR_TRUE;
if (hits > 1) {
OKToPrompt = PR_TRUE;
}
}
}
}
if (isText || isPassword) {
nsAutoString value;
rv = inputElement->GetValue(value);
if (NS_SUCCEEDED(rv)) {
nsAutoString field;
rv = inputElement->GetName(field);
if (NS_SUCCEEDED(rv)) {
data = new si_SignonDataStruct;
data->value = value;
if (!field.IsEmpty() && field.CharAt(0) == '\\') {
/*
* Note that data saved for browser-generated logins (e.g. http
* authentication) use artificial field names starting with
* \= (see USERNAMEFIELD and PASSWORDFIELD in singsign.cpp). To
* avoid mistakes whereby saved logins for http authentication is
* then prefilled into a field on the html form at the same URL,
* we will prevent html field names from starting with \=. We
* do that by doubling up a backslash if it appears in the first
* character position
*/
data->name = nsAutoString('\\');
data->name.Append(field);
} else {
data->name = field;
}
data->isPassword = isPassword;
signonData->AppendElement(data);
if (passwordcount == 0 && !OKToPrompt) {
/* get schema from field */
nsCAutoString schema;
nsVoidArray* dummy;
nsCAutoString stripField;
/* try to get schema from displayable text */
if (schema.IsEmpty()) {
wallet_GetSchemaFromDisplayableText(inputElement, schema, PR_FALSE);
}
/* no schema found, so try to get it from field name */
if (schema.IsEmpty()) {
Strip(field, stripField);
wallet_ReadFromList
(stripField, schema,
dummy, wallet_FieldToSchema_list, PR_FALSE);
}
/* if schema found, see if it is in distinguished schema list */
if (!schema.IsEmpty()) {
/* see if schema is in distinguished list */
wallet_MapElement * mapElementPtr;
PRInt32 count = LIST_COUNT(wallet_DistinguishedSchema_list);
/* test for at least two distinguished schemas and no passwords */
for (PRInt32 i=0; i<count; i++) {
mapElementPtr = NS_STATIC_CAST
(wallet_MapElement*, wallet_DistinguishedSchema_list->ElementAt(i));
if (schema.Equals(mapElementPtr->item1, nsCaseInsensitiveCStringComparator()) && !value.IsEmpty()) {
hits++;
if (hits > 1 && newValueFound) {
OKToPrompt = PR_TRUE;
break;
}
}
}
}
}
}
}
}
}
}
}
}
/* save login if appropriate */
if (currentFormNode == formNode) {
nsCOMPtr<nsIPrompt> dialog;
nsCOMPtr<nsIWindowWatcher> wwatch(do_GetService(NS_WINDOWWATCHER_CONTRACTID));
if (wwatch)
wwatch->GetNewPrompter(0, getter_AddRefs(dialog));
if (dialog) {
SINGSIGN_RememberSignonData(dialog, docURL, signonData, window);
}
}
PRInt32 count2 = signonData->Count();
for (PRInt32 i=count2-1; i>=0; i--) {
data = NS_STATIC_CAST(si_SignonDataStruct*, signonData->ElementAt(i));
delete data;
}
delete signonData;
/* save form if it meets all necessary conditions */
if (wallet_GetFormsCapturingPref() &&
(OKToPrompt) && wallet_OKToCapture(strippedURLNameUTF8, window)) {
/* give caveat if this is the first time data is being captured */
Wallet_GiveCaveat(window, nsnull);
/* conditions all met, now save it */
for (PRUint32 elementY = 0; elementY < numElements; elementY++) {
nsIDOMNode* elementNode = nsnull;
elements->Item(elementY, &elementNode);
if (nsnull != elementNode) {
wallet_CaptureInputElement(elementNode, doc);
wallet_CaptureSelectElement(elementNode, doc);
}
}
}
}
}
}
}
}