/* -*- 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 The Browser Profile Migrator. * * The Initial Developer of the Original Code is Ben Goodger. * Portions created by the Initial Developer are Copyright (C) 2004 * the Initial Developer. All Rights Reserved. * * Contributor(s): *  Ben Goodger <ben@bengoodger.com> *  Scott MacGregor <mscott@mozilla.org> * * 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 ***** */#include "nsAppDirectoryServiceDefs.h"#include "nsCRT.h"#include "nsIFile.h"#include "nsIInputStream.h"#include "nsILineInputStream.h"#include "nsInt64.h"#include "nsIPrefBranch.h"#include "nsIPrefLocalizedString.h"#include "nsIPrefService.h"#include "nsIRDFService.h"#include "nsIRegistry.h"#include "nsIServiceManager.h"#include "nsISupportsArray.h"#include "nsISupportsPrimitives.h"#include "nsIURL.h"#include "nsNetscapeProfileMigratorBase.h"#include "nsNetUtil.h"#include "nsReadableUtils.h"#include "nsXPIDLString.h"#include "prtime.h"#include "prprf.h"#include "nsVoidArray.h"static NS_DEFINE_CID(kStringBundleServiceCID, NS_STRINGBUNDLESERVICE_CID);#define MIGRATION_BUNDLE "chrome://messenger/locale/migration/migration.properties"#define FILE_NAME_PREFS_5X NS_LITERAL_STRING("prefs.js")///////////////////////////////////////////////////////////////////////////////// nsNetscapeProfileMigratorBasensNetscapeProfileMigratorBase::nsNetscapeProfileMigratorBase(){  nsCOMPtr<nsIStringBundleService> bundleService(do_GetService(kStringBundleServiceCID));  bundleService->CreateBundle(MIGRATION_BUNDLE, getter_AddRefs(mBundle));  // create the array we'll be using to keep track of the asynchronous file copy routines  mFileCopyTransactions = new nsVoidArray();  mFileCopyTransactionIndex = 0;}nsresultnsNetscapeProfileMigratorBase::GetProfileDataFromRegistry(nsILocalFile* aRegistryFile,                                                          nsISupportsArray* aProfileNames,                                                          nsISupportsArray* aProfileLocations){  nsresult rv = NS_OK;  // Open It  nsCOMPtr<nsIRegistry> reg(do_CreateInstance("@mozilla.org/registry;1"));  reg->Open(aRegistryFile);  nsRegistryKey profilesTree;  rv = reg->GetKey(nsIRegistry::Common, NS_LITERAL_STRING("Profiles").get(), &profilesTree);  if (NS_FAILED(rv)) return rv;  nsCOMPtr<nsIEnumerator> keys;  reg->EnumerateSubtrees(profilesTree, getter_AddRefs(keys));  keys->First();  while (keys->IsDone() != NS_OK) {    nsCOMPtr<nsISupports> key;    keys->CurrentItem(getter_AddRefs(key));    nsCOMPtr<nsIRegistryNode> node(do_QueryInterface(key));    nsRegistryKey profile;    node->GetKey(&profile);    // "migrated" is "yes" for all valid Seamonkey profiles. It is only "no"    // for 4.x profiles.     nsXPIDLString isMigrated;    reg->GetString(profile, NS_LITERAL_STRING("migrated").get(), getter_Copies(isMigrated));    if (isMigrated.Equals(NS_LITERAL_STRING("no"))) {      keys->Next();      continue;    }    // Get the profile name and add it to the names array    nsXPIDLString profileName;    node->GetName(getter_Copies(profileName));    // Get the profile location and add it to the locations array    nsXPIDLString directory;    reg->GetString(profile, NS_LITERAL_STRING("directory").get(), getter_Copies(directory));    nsCOMPtr<nsILocalFile> dir;#ifdef XP_MACOSX    rv = NS_NewNativeLocalFile(EmptyCString(), PR_TRUE, getter_AddRefs(dir));    if (NS_FAILED(rv)) return rv;    dir->SetPersistentDescriptor(NS_LossyConvertUCS2toASCII(directory));#else    rv = NS_NewLocalFile(directory, PR_TRUE, getter_AddRefs(dir));    if (NS_FAILED(rv)) return rv;#endif    PRBool exists;    dir->Exists(&exists);    if (exists) {      nsCOMPtr<nsISupportsString> profileNameString(do_CreateInstance("@mozilla.org/supports-string;1"));      profileNameString->SetData(profileName);      aProfileNames->AppendElement(profileNameString);      aProfileLocations->AppendElement(dir);    }    keys->Next();  }  return rv;}#define GETPREF(xform, method, value) \  nsresult rv = aBranch->method(xform->sourcePrefName, value); \  if (NS_SUCCEEDED(rv)) \    xform->prefHasValue = PR_TRUE; \  return rv;#define SETPREF(xform, method, value) \  if (xform->prefHasValue) { \    return aBranch->method(xform->targetPrefName ? xform->targetPrefName : xform->sourcePrefName, value); \  } \  return NS_OK;nsresult nsNetscapeProfileMigratorBase::GetString(void* aTransform, nsIPrefBranch* aBranch){  PrefTransform* xform = (PrefTransform*)aTransform;  GETPREF(xform, GetCharPref, &xform->stringValue);}nsresult nsNetscapeProfileMigratorBase::SetString(void* aTransform, nsIPrefBranch* aBranch){  PrefTransform* xform = (PrefTransform*)aTransform;  SETPREF(xform, SetCharPref, xform->stringValue);}nsresultnsNetscapeProfileMigratorBase::GetWString(void* aTransform, nsIPrefBranch* aBranch){  PrefTransform* xform = (PrefTransform*)aTransform;  nsCOMPtr<nsIPrefLocalizedString> prefValue;  nsresult rv = aBranch->GetComplexValue(xform->sourcePrefName,                                          NS_GET_IID(nsIPrefLocalizedString),                                         getter_AddRefs(prefValue));  if (NS_SUCCEEDED(rv) && prefValue) {    nsXPIDLString data;    prefValue->ToString(getter_Copies(data));    xform->stringValue = ToNewCString(NS_ConvertUCS2toUTF8(data));    xform->prefHasValue = PR_TRUE;  }  return rv;}nsresult nsNetscapeProfileMigratorBase::SetWStringFromASCII(void* aTransform, nsIPrefBranch* aBranch){  PrefTransform* xform = (PrefTransform*)aTransform;  if (xform->prefHasValue) {    nsCOMPtr<nsIPrefLocalizedString> pls(do_CreateInstance("@mozilla.org/pref-localizedstring;1"));    nsAutoString data; data.AssignWithConversion(xform->stringValue);    pls->SetData(data.get());    return aBranch->SetComplexValue(xform->targetPrefName ? xform->targetPrefName : xform->sourcePrefName, NS_GET_IID(nsIPrefLocalizedString), pls);  }  return NS_OK;}nsresultnsNetscapeProfileMigratorBase::SetWString(void* aTransform, nsIPrefBranch* aBranch){  PrefTransform* xform = (PrefTransform*)aTransform;  if (xform->prefHasValue) {    nsCOMPtr<nsIPrefLocalizedString> pls(do_CreateInstance("@mozilla.org/pref-localizedstring;1"));    nsAutoString data = NS_ConvertUTF8toUCS2(xform->stringValue);    pls->SetData(data.get());    return aBranch->SetComplexValue(xform->targetPrefName ? xform->targetPrefName : xform->sourcePrefName, NS_GET_IID(nsIPrefLocalizedString), pls);  }  return NS_OK;}nsresult nsNetscapeProfileMigratorBase::GetBool(void* aTransform, nsIPrefBranch* aBranch){  PrefTransform* xform = (PrefTransform*)aTransform;  GETPREF(xform, GetBoolPref, &xform->boolValue);}nsresult nsNetscapeProfileMigratorBase::SetBool(void* aTransform, nsIPrefBranch* aBranch){  PrefTransform* xform = (PrefTransform*)aTransform;  SETPREF(xform, SetBoolPref, xform->boolValue);}nsresult nsNetscapeProfileMigratorBase::GetInt(void* aTransform, nsIPrefBranch* aBranch){  PrefTransform* xform = (PrefTransform*)aTransform;  GETPREF(xform, GetIntPref, &xform->intValue);}nsresult nsNetscapeProfileMigratorBase::SetInt(void* aTransform, nsIPrefBranch* aBranch){  PrefTransform* xform = (PrefTransform*)aTransform;  SETPREF(xform, SetIntPref, xform->intValue);}nsresultnsNetscapeProfileMigratorBase::CopyFile(const nsAString& aSourceFileName, const nsAString& aTargetFileName){  nsCOMPtr<nsIFile> sourceFile;  mSourceProfile->Clone(getter_AddRefs(sourceFile));  sourceFile->Append(aSourceFileName);  PRBool exists = PR_FALSE;  sourceFile->Exists(&exists);  if (!exists)    return NS_OK;  nsCOMPtr<nsIFile> targetFile;  mTargetProfile->Clone(getter_AddRefs(targetFile));    targetFile->Append(aTargetFileName);  targetFile->Exists(&exists);  if (exists)    targetFile->Remove(PR_FALSE);  return sourceFile->CopyTo(mTargetProfile, aTargetFileName);}nsresultnsNetscapeProfileMigratorBase::GetSignonFileName(PRBool aReplace, char** aFileName){  nsresult rv;  if (aReplace) {    // Find out what the signons file was called, this is stored in a pref    // in Seamonkey.    nsCOMPtr<nsIPrefService> psvc(do_GetService(NS_PREFSERVICE_CONTRACTID));    psvc->ResetPrefs();    nsCOMPtr<nsIFile> sourcePrefsName;    mSourceProfile->Clone(getter_AddRefs(sourcePrefsName));    sourcePrefsName->Append(FILE_NAME_PREFS_5X);    psvc->ReadUserPrefs(sourcePrefsName);    nsCOMPtr<nsIPrefBranch> branch(do_QueryInterface(psvc));    rv = branch->GetCharPref("signon.SignonFileName", aFileName);  }  else     rv = LocateSignonsFile(aFileName);  return rv;}nsresultnsNetscapeProfileMigratorBase::LocateSignonsFile(char** aResult){  nsCOMPtr<nsISimpleEnumerator> entries;  nsresult rv = mSourceProfile->GetDirectoryEntries(getter_AddRefs(entries));  if (NS_FAILED(rv)) return rv;  nsCAutoString fileName;  do {    PRBool hasMore = PR_FALSE;    rv = entries->HasMoreElements(&hasMore);    if (NS_FAILED(rv) || !hasMore) break;    nsCOMPtr<nsISupports> supp;    rv = entries->GetNext(getter_AddRefs(supp));    if (NS_FAILED(rv)) break;    nsCOMPtr<nsIFile> currFile(do_QueryInterface(supp));    nsCOMPtr<nsIURI> uri;    rv = NS_NewFileURI(getter_AddRefs(uri), currFile);    if (NS_FAILED(rv)) break;    nsCOMPtr<nsIURL> url(do_QueryInterface(uri));    nsCAutoString extn;    url->GetFileExtension(extn);    if (extn.EqualsIgnoreCase("s")) {      url->GetFileName(fileName);      break;    }  }  while (1);  *aResult = ToNewCString(fileName);  return NS_OK;}// helper function, copies the contents of srcDir into destDir.// destDir will be created if it doesn't exist.nsresult nsNetscapeProfileMigratorBase::RecursiveCopy(nsIFile* srcDir, nsIFile* destDir){  nsresult rv;  PRBool isDir;    rv = srcDir->IsDirectory(&isDir);  if (NS_FAILED(rv)) return rv;  if (!isDir) return NS_ERROR_INVALID_ARG;    PRBool exists;  rv = destDir->Exists(&exists);  if (NS_SUCCEEDED(rv) && !exists)    rv = destDir->Create(nsIFile::DIRECTORY_TYPE, 0775);  if (NS_FAILED(rv)) return rv;    PRBool hasMore = PR_FALSE;  nsCOMPtr<nsISimpleEnumerator> dirIterator;  rv = srcDir->GetDirectoryEntries(getter_AddRefs(dirIterator));  if (NS_FAILED(rv)) return rv;    rv = dirIterator->HasMoreElements(&hasMore);  if (NS_FAILED(rv)) return rv;    nsCOMPtr<nsIFile> dirEntry;    while (hasMore)  {    rv = dirIterator->GetNext((nsISupports**)getter_AddRefs(dirEntry));    if (NS_SUCCEEDED(rv))    {      rv = dirEntry->IsDirectory(&isDir);      if (NS_SUCCEEDED(rv))      {        if (isDir)        {          nsCOMPtr<nsIFile> destClone;          rv = destDir->Clone(getter_AddRefs(destClone));          if (NS_SUCCEEDED(rv))          {            nsCOMPtr<nsILocalFile> newChild(do_QueryInterface(destClone));            nsAutoString leafName;            dirEntry->GetLeafName(leafName);            newChild->AppendRelativePath(leafName);            rv = newChild->Exists(&exists);            if (NS_SUCCEEDED(rv) && !exists)              rv = newChild->Create(nsIFile::DIRECTORY_TYPE, 0775);            rv = RecursiveCopy(dirEntry, newChild);          }        }        else        {          // we aren't going to do any actual file copying here. Instead, add this to our          // file transaction list so we can copy files asynchronously...          fileTransactionEntry* fileEntry = new fileTransactionEntry;          fileEntry->srcFile = dirEntry;          fileEntry->destFile = destDir;          mFileCopyTransactions->AppendElement((void*) fileEntry);        }      }          }    rv = dirIterator->HasMoreElements(&hasMore);    if (NS_FAILED(rv)) return rv;  }    return rv;}