This commit is contained in:
Manohar
818
c_adlib/ad_ff/native/ProfileManager.cpp
Normal file
818
c_adlib/ad_ff/native/ProfileManager.cpp
Normal file
@@ -0,0 +1,818 @@
|
||||
|
||||
#include "ProfileManager.h"
|
||||
|
||||
|
||||
|
||||
// Private functions....
|
||||
|
||||
ProfileManager::ProfileManager()
|
||||
{
|
||||
isInitialized = 0;
|
||||
libnss = NULL;
|
||||
libplc = NULL;
|
||||
|
||||
profileName = NULL;
|
||||
profilePath = NULL;
|
||||
libraryPath = NULL;
|
||||
masterPassword = NULL;
|
||||
|
||||
|
||||
}
|
||||
|
||||
|
||||
char * ProfileManager::GetFirefoxProfilePath(char *profileName)
|
||||
{
|
||||
|
||||
#ifdef WIN32
|
||||
|
||||
char profileDir[MAX_PATH] = "";
|
||||
char partialPath[] = "Application Data\\Mozilla\\Firefox";
|
||||
char profilePath[MAX_PATH];
|
||||
char line[1024];
|
||||
|
||||
DWORD pathSize = MAX_PATH;
|
||||
char *finalProfilePath = NULL;
|
||||
unsigned int i;
|
||||
HANDLE token;
|
||||
|
||||
|
||||
// Get current user's profile directory
|
||||
if( OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &token) == FALSE )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n GetFirefoxProfilePath : Failed to get current process token ");
|
||||
return 0;
|
||||
}
|
||||
|
||||
if( GetUserProfileDirectory(token, profileDir, &pathSize) == FALSE )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n GetFirefoxProfilePath : Failed to get user profile directory");
|
||||
return 0;
|
||||
}
|
||||
|
||||
PrintMessage(MESG_DEBUG, "\n GetFirefoxProfilePath : User Profile directory = %s", profileDir);
|
||||
|
||||
// Get firefox profile directory
|
||||
strcpy(profilePath, profileDir);
|
||||
strcat(profilePath,"\\");
|
||||
strcat(profilePath,partialPath);
|
||||
strcat(profilePath,"\\profiles.ini");
|
||||
|
||||
PrintMessage(MESG_DEBUG, "\n GetFirefoxProfilePath : Firefox profile.ini path = %s ", profilePath);
|
||||
|
||||
|
||||
#else // Linux platform....
|
||||
|
||||
char profileDir[] ="/.mozilla/firefox";
|
||||
char profileFile[] ="/.mozilla/firefox/profiles.ini";
|
||||
|
||||
char line[1024];
|
||||
char *profilePath = NULL;
|
||||
char *homeDir = NULL;
|
||||
char *finalProfilePath = NULL;
|
||||
unsigned int i;
|
||||
|
||||
// Get home directory
|
||||
homeDir = getenv("HOME");
|
||||
|
||||
if(homeDir == NULL )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n GetFirefoxProfilePath : Unable to get home directory ");
|
||||
return 0;
|
||||
}
|
||||
|
||||
profilePath = (char*) malloc( strlen(homeDir) + strlen(profileFile) + 3 );
|
||||
if( profilePath == NULL )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n GetFirefoxProfilePath : Insufficient memory ");
|
||||
return 0;
|
||||
}
|
||||
strcpy(profilePath,homeDir);
|
||||
strcat(profilePath,profileFile);
|
||||
|
||||
PrintMessage(MESG_DEBUG, "\n GetFirefoxProfilePath : Firefox profile dir path = %s ", profilePath);
|
||||
|
||||
#endif
|
||||
|
||||
// Open the firefox profile setting file
|
||||
FILE *profile = fopen(profilePath, "r");
|
||||
|
||||
if( profile == NULL )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n GetFirefoxProfilePath : Unable to find firefox profile file : %s ", profilePath);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
// Find out the profile path for given profile name
|
||||
char *temp;
|
||||
int isProfileFound = 0;
|
||||
int isRelative = 1;
|
||||
char *path = NULL;
|
||||
|
||||
while(fgets(line, 1024, profile))
|
||||
{
|
||||
// Remove trailing end of line character
|
||||
line[strlen(line)-1]= 0;
|
||||
|
||||
// Convert to smaller case until "=" found....
|
||||
for(i=0; i<strlen(line); i++)
|
||||
{
|
||||
if( line[i] == '=' )
|
||||
break;
|
||||
|
||||
if( line[i] >=65 && line[i]<=90 )
|
||||
line[i]+=32;
|
||||
}
|
||||
|
||||
if( isProfileFound )
|
||||
{
|
||||
|
||||
temp = strchr(line,'=') + 1;
|
||||
|
||||
// check for IsRelative=
|
||||
if( strstr(line, "isrelative=") != NULL )
|
||||
{
|
||||
// Is the value greater than 0 ( ascii 48 )
|
||||
isRelative = (*temp > 48)? 1 : 0 ;
|
||||
continue;
|
||||
}
|
||||
|
||||
// check for Path=
|
||||
if( strstr(line, "path=") != NULL )
|
||||
{
|
||||
path = (char*) malloc(strlen(temp) + 1);
|
||||
if( !path )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n GetFirefoxProfilePath : Insufficient memory ");
|
||||
fclose(profile);
|
||||
return 0;
|
||||
}
|
||||
|
||||
strcpy(path,temp);
|
||||
continue;
|
||||
}
|
||||
|
||||
// Break once you encounter next profile
|
||||
if( strstr(line, "profile") != NULL )
|
||||
{
|
||||
break;
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
|
||||
|
||||
if( strstr(line, "name=") != NULL )
|
||||
{
|
||||
temp = strchr(line,'=') + 1;
|
||||
|
||||
if( STRCMPI(temp, profileName) == 0 )
|
||||
{
|
||||
PrintMessage(MESG_DEBUG, "\n GetFirefoxProfilePath : Profile name [%s] found ", profileName);
|
||||
isProfileFound = 1;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
|
||||
fclose(profile);
|
||||
|
||||
if( !isProfileFound )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n GetFirefoxProfilePath : Specified Profile [%s] not present ", profileName);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
if( !path )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n GetFirefoxProfilePath : Profile [%s] path not found ", profileName);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
if( isRelative )
|
||||
{
|
||||
#ifdef WIN32
|
||||
// if path contains '/' then make it '\\'
|
||||
char *slash = strstr(path,"/");
|
||||
|
||||
if( slash != NULL )
|
||||
*slash = '\\';
|
||||
|
||||
int totalLen = strlen(profileDir) + strlen(partialPath) + strlen(path) + 3 ;
|
||||
finalProfilePath = (char *) malloc(totalLen);
|
||||
|
||||
if( finalProfilePath )
|
||||
{
|
||||
strcpy(finalProfilePath,profileDir);
|
||||
strcat(finalProfilePath,"\\");
|
||||
strcat(finalProfilePath,partialPath);
|
||||
strcat(finalProfilePath,"\\");
|
||||
strcat(finalProfilePath,path);
|
||||
|
||||
PrintMessage(MESG_DEBUG, "\n Final profile path is : %s ", finalProfilePath);
|
||||
}
|
||||
|
||||
#else // Linux platform
|
||||
|
||||
int totalLen = strlen(homeDir) + strlen(profileDir) + strlen(path) + 2 ;
|
||||
finalProfilePath = (char *) malloc(totalLen);
|
||||
|
||||
if( finalProfilePath )
|
||||
{
|
||||
strcpy(finalProfilePath,homeDir);
|
||||
strcat(finalProfilePath,profileDir);
|
||||
strcat(finalProfilePath,"/");
|
||||
strcat(finalProfilePath,path);
|
||||
|
||||
PrintMessage(MESG_DEBUG, "\n Final profile path is : %s ", finalProfilePath);
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
if(path)
|
||||
free(path);
|
||||
}
|
||||
else
|
||||
{
|
||||
finalProfilePath = path;
|
||||
|
||||
}
|
||||
|
||||
if( !finalProfilePath)
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n GetFirefoxProfilePath : Insufficient memory ");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
// check if profile directory exists.....
|
||||
if( !IsDirectoryExists(finalProfilePath) )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n GetFirefoxProfilePath : Profile directory [%s] does not exist", finalProfilePath);
|
||||
free(finalProfilePath);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
|
||||
return finalProfilePath;
|
||||
}
|
||||
|
||||
|
||||
|
||||
char* ProfileManager::GetFirefoxLibPath()
|
||||
{
|
||||
char *firefoxPath = NULL;
|
||||
|
||||
#ifdef WIN32
|
||||
|
||||
char regSubKey[] = "SOFTWARE\\Clients\\StartMenuInternet\\firefox.exe\\shell\\open\\command";
|
||||
char pathValue[MAX_PATH] ="";
|
||||
DWORD pathSize = MAX_PATH;
|
||||
DWORD valueType;
|
||||
HKEY rkey;
|
||||
|
||||
// Open firefox registry key
|
||||
if( RegOpenKeyEx(HKEY_LOCAL_MACHINE, regSubKey, 0, KEY_READ, &rkey) != ERROR_SUCCESS )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n GetFirefoxLibPath : Failed to open the firefox registry key : HKCU\\%s", regSubKey );
|
||||
return NULL;
|
||||
}
|
||||
|
||||
// Read the firefox path value
|
||||
if( RegQueryValueEx(rkey, NULL, 0, &valueType, (unsigned char*)&pathValue, &pathSize) != ERROR_SUCCESS )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n GetFirefoxLibPath : Failed to read the firefox path value from registry ");
|
||||
RegCloseKey(rkey);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
if( pathSize <= 0 || pathValue[0] == 0)
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n GetFirefoxLibPath : Path value read from the registry is empty");
|
||||
RegCloseKey(rkey);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
RegCloseKey(rkey);
|
||||
|
||||
// This path may contain extra double quote....
|
||||
if( pathValue[0] == '\"' )
|
||||
{
|
||||
for(unsigned int i=0; i< strlen(pathValue)-1 ; i++)
|
||||
pathValue[i] = pathValue[i+1];
|
||||
}
|
||||
|
||||
PrintMessage(MESG_DEBUG, "\n GetFirefoxLibPath : Path value read from registry is %s", pathValue);
|
||||
|
||||
StrLwr(pathValue);
|
||||
char *firefox = strstr(pathValue, "firefox.exe");
|
||||
|
||||
if( firefox == NULL )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n GetFirefoxLibPath : Path value read from registry is invalid");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
// remove the firefox.exe from the end...
|
||||
*(firefox-1) = 0;
|
||||
|
||||
firefoxPath = (char*) malloc( strlen(pathValue) + 1);
|
||||
|
||||
if( firefoxPath )
|
||||
{
|
||||
strcpy(firefoxPath, pathValue);
|
||||
|
||||
if( !IsDirectoryExists(firefoxPath) )
|
||||
{
|
||||
free(firefoxPath);
|
||||
return NULL;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
#else // Linux platform...
|
||||
|
||||
char path[]="/opt/MozillaFirefox/lib";
|
||||
|
||||
firefoxPath = (char*) malloc( strlen(path) + 1);
|
||||
|
||||
if( firefoxPath )
|
||||
{
|
||||
strcpy(firefoxPath, path);
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
PrintMessage(MESG_DEBUG, "\n GetFirefoxLibPath : Firefox library path = [%s] ", firefoxPath);
|
||||
|
||||
return firefoxPath;
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
||||
HMODULE ProfileManager::PMLoadLibrary(char *firefoxDir, char *libName)
|
||||
{
|
||||
char *loadPath;
|
||||
HMODULE libtmp = NULL;
|
||||
|
||||
if( firefoxDir != NULL )
|
||||
loadPath = ( char*) malloc(strlen(firefoxDir) + strlen(libName) + 2);
|
||||
else
|
||||
loadPath = ( char*) malloc(strlen(libName) + 1);
|
||||
|
||||
if( !loadPath )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n\n LoadLibrary : Insufficient memory");
|
||||
return 0;
|
||||
}
|
||||
|
||||
if( firefoxDir != NULL )
|
||||
{
|
||||
strcpy(loadPath, firefoxDir);
|
||||
strcat(loadPath, "/");
|
||||
strcat(loadPath, libName);
|
||||
}
|
||||
else
|
||||
{
|
||||
strcpy(loadPath, libName);
|
||||
}
|
||||
|
||||
libtmp = LOADLIBRARY(loadPath);
|
||||
|
||||
if( !libtmp )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n\n LoadLibrary : Failed to load library %s ", loadPath);
|
||||
free(loadPath);
|
||||
return 0;
|
||||
}
|
||||
|
||||
free(loadPath);
|
||||
|
||||
return libtmp;
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
// Public functions...
|
||||
|
||||
|
||||
int ProfileManager::ProfileInit(char *profileName)
|
||||
{
|
||||
|
||||
if( isInitialized == FPM_TRUE)
|
||||
{
|
||||
PrintMessage(MESG_DEBUG, "\n ProfileInit : Firefox profile [%s] is already initialized ", profileName);
|
||||
return FPM_TRUE;
|
||||
}
|
||||
|
||||
|
||||
profilePath = GetFirefoxProfilePath(profileName);
|
||||
libraryPath = GetFirefoxLibPath();
|
||||
|
||||
if( !profilePath || !libraryPath )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n ProfileInit : Failed to find firefox profile or library path ");
|
||||
return FPM_PROFILE_NOT_PRESENT;
|
||||
}
|
||||
|
||||
PrintMessage(MESG_DEBUG, "\n Profile path = [%s] ", profilePath);
|
||||
PrintMessage(MESG_DEBUG, "\n Library path = [%s] ", libraryPath);
|
||||
|
||||
// First try to load from the library path then try to load from default lib path
|
||||
|
||||
// Here we have to first load all dependent libraries and then
|
||||
// load main library , otherwise face the problems...:)
|
||||
if( PMLoadLibrary(libraryPath, NSPR_LIBRARY_NAME) )
|
||||
{
|
||||
if((libplc=PMLoadLibrary(libraryPath, PLC_LIBRARY_NAME)) )
|
||||
{
|
||||
if( PMLoadLibrary(libraryPath, PLDS_LIBRARY_NAME) )
|
||||
{
|
||||
if( PMLoadLibrary(libraryPath, SOFTN_LIBRARY_NAME) )
|
||||
libnss=PMLoadLibrary(libraryPath, NSS_LIBRARY_NAME);
|
||||
}
|
||||
}
|
||||
}
|
||||
else // try to load library from default library path
|
||||
{
|
||||
if( PMLoadLibrary(NULL, NSPR_LIBRARY_NAME) )
|
||||
{
|
||||
if((libplc=PMLoadLibrary(NULL, PLC_LIBRARY_NAME)) )
|
||||
{
|
||||
if( PMLoadLibrary(NULL, PLDS_LIBRARY_NAME) )
|
||||
{
|
||||
if( PMLoadLibrary(NULL, SOFTN_LIBRARY_NAME) )
|
||||
libnss=PMLoadLibrary(NULL, NSS_LIBRARY_NAME);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if( !libnss || !libplc )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n ProfileInit : Failed to load the required library from directory %s", libraryPath);
|
||||
return FPM_LIBRARY_LOAD_FAILED;
|
||||
}
|
||||
|
||||
|
||||
PrintMessage(MESG_DEBUG, "\n ProfileInit : Required libraries are loaded from firefox path successfully..");
|
||||
|
||||
|
||||
// Get all the functions from NSS library
|
||||
NSSInit = (NSS_Init) GETPROCADDRESS(libnss, "NSS_Init");
|
||||
NSSShutdown = (NSS_Shutdown)GETPROCADDRESS(libnss, "NSS_Shutdown");
|
||||
//PK11SetPasswordFunc = (PK11_SetPasswordFunc) GETPROCADDRESS(libnss, "PK11_SetPasswordFunc");
|
||||
PK11GetInternalKeySlot = (PK11_GetInternalKeySlot) GETPROCADDRESS(libnss, "PK11_GetInternalKeySlot");
|
||||
PK11FreeSlot = (PK11_FreeSlot) GETPROCADDRESS(libnss, "PK11_FreeSlot");
|
||||
PK11Authenticate = (PK11_Authenticate) GETPROCADDRESS(libnss, "PK11_Authenticate");
|
||||
PK11SDRDecrypt = (PK11SDR_Decrypt) GETPROCADDRESS(libnss, "PK11SDR_Decrypt");
|
||||
PK11SDREncrypt = (PK11SDR_Encrypt) GETPROCADDRESS(libnss, "PK11SDR_Encrypt");
|
||||
PK11CheckUserPassword = (PK11_CheckUserPassword) GETPROCADDRESS(libnss, "PK11_CheckUserPassword");
|
||||
|
||||
if( !NSSInit || !NSSShutdown || !PK11GetInternalKeySlot || !PK11Authenticate || !PK11SDRDecrypt || !PK11SDREncrypt || !PK11FreeSlot || !PK11CheckUserPassword)
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n\n ProfileInit : Failed to get function address for library %s ", NSS_LIBRARY_NAME);
|
||||
ProfileExit();
|
||||
return FPM_LIBRARY_LOAD_FAILED;
|
||||
}
|
||||
|
||||
// Get all the functions from PLC library
|
||||
PLBase64Encode = ( PL_Base64Encode ) GETPROCADDRESS(libplc, "PL_Base64Encode");
|
||||
PLBase64Decode = ( PL_Base64Decode ) GETPROCADDRESS(libplc, "PL_Base64Decode");
|
||||
|
||||
if( !PLBase64Encode || !PLBase64Decode )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\nProfileInit : Failed to get function address for library %s ", PLC_LIBRARY_NAME);
|
||||
ProfileExit();
|
||||
return FPM_LIBRARY_LOAD_FAILED;
|
||||
}
|
||||
|
||||
|
||||
// Initialize the NSS library
|
||||
if( (*NSSInit) (profilePath) != SECSuccess )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n ProfileInit : Initialization failed , Make sure key3.db and cert8.db");
|
||||
PrintMessage(MESG_ERROR, "\n files are present in the specified directory\n");
|
||||
ProfileExit();
|
||||
|
||||
return FPM_LIBRARY_INIT_FAILED;
|
||||
}
|
||||
|
||||
|
||||
PrintMessage(MESG_DEBUG, "\n ProfileInit : NSS_Init success..");
|
||||
|
||||
// Copy the profile name for later use
|
||||
this->profileName = (char*) malloc(strlen(profileName)+1);
|
||||
if( !this->profileName )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n ProfileInit : Insufficient memory..");
|
||||
return FPM_INSUFFICIENT_MEMORY;
|
||||
}
|
||||
|
||||
strcpy(this->profileName, profileName);
|
||||
|
||||
|
||||
// TODO
|
||||
// Read the preference from the config file present in the profile...
|
||||
|
||||
isInitialized = FPM_TRUE;
|
||||
|
||||
// Setup the function pointers...
|
||||
void *funList[9];
|
||||
|
||||
funList[0] = (void*)NULL ; //PK11SetPasswordFunc;
|
||||
funList[1] = (void*)PK11GetInternalKeySlot;
|
||||
funList[2] = (void*)PK11FreeSlot;
|
||||
funList[3] = (void*)PK11Authenticate;
|
||||
funList[4] = (void*)PK11CheckUserPassword;
|
||||
funList[5] = (void*)PK11SDRDecrypt;
|
||||
funList[6] = (void*)PK11SDREncrypt;
|
||||
funList[7] = (void*)PLBase64Encode;
|
||||
funList[8] = (void*)PLBase64Decode;
|
||||
|
||||
signonManager.SetupFunctions(funList);
|
||||
|
||||
return FPM_TRUE;
|
||||
}
|
||||
|
||||
|
||||
|
||||
void ProfileManager::ProfileExit()
|
||||
{
|
||||
PrintMessage(MESG_DEBUG, "\n ProfileExit : Shutting down the profile %s", profileName);
|
||||
|
||||
if( (isInitialized == FPM_TRUE) && NSSShutdown != NULL )
|
||||
(*NSSShutdown)();
|
||||
|
||||
if( libnss != NULL )
|
||||
FREELIBRARY(libnss);
|
||||
|
||||
if( libplc != NULL )
|
||||
FREELIBRARY(libplc);
|
||||
|
||||
// clean up signon data...
|
||||
signonManager.RemoveSignonData();
|
||||
|
||||
isInitialized = FPM_FALSE;
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
int ProfileManager::IsMasterPasswordSet()
|
||||
{
|
||||
PK11SlotInfo *slot = 0;
|
||||
int retValue = 0;
|
||||
|
||||
slot = (*PK11GetInternalKeySlot)();
|
||||
|
||||
if (!slot)
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n IsMasterPasswordSet PK11_GetInternalKeySlot failed ...");
|
||||
return FPM_FALSE;
|
||||
}
|
||||
|
||||
PrintMessage(MESG_DEBUG, "\n PK11_GetInternalKeySlot SUCCESS ...");
|
||||
|
||||
// Check with empty password....If it succeeds then master password is not set
|
||||
if( (*PK11CheckUserPassword)(slot,"") == SECSuccess )
|
||||
{
|
||||
PrintMessage(MESG_DEBUG, "\n IsMasterPasswordSet : Master password is not set...");
|
||||
retValue = FPM_FALSE;
|
||||
}
|
||||
else
|
||||
{
|
||||
PrintMessage(MESG_DEBUG, "\n IsMasterPasswordSet : Master password is set...");
|
||||
retValue = FPM_TRUE;
|
||||
}
|
||||
|
||||
// Free the slot
|
||||
(*PK11FreeSlot) (slot);
|
||||
|
||||
return retValue;
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
int ProfileManager::CheckMasterPassword(char *masterPassword, int doSave)
|
||||
{
|
||||
PK11SlotInfo *slot = 0;
|
||||
int retValue = 0;
|
||||
|
||||
slot = (*PK11GetInternalKeySlot)();
|
||||
|
||||
if (!slot)
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n CheckMasterPassword PK11_GetInternalKeySlot failed ...");
|
||||
return FPM_FALSE;
|
||||
}
|
||||
|
||||
|
||||
if( (*PK11CheckUserPassword)(slot, masterPassword) == SECSuccess )
|
||||
{
|
||||
// Store the password for internal use...
|
||||
if( doSave )
|
||||
{
|
||||
PrintMessage(MESG_DEBUG, "\n CheckMasterPassword : saving master password for future use ******...");
|
||||
|
||||
this->masterPassword = (char *) malloc( strlen(masterPassword)+1 );
|
||||
|
||||
if( !this->masterPassword )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n CheckMasterPassword : Insufficient memory");
|
||||
(*PK11FreeSlot) (slot); // Free the slot...
|
||||
return FPM_INSUFFICIENT_MEMORY;
|
||||
}
|
||||
|
||||
strcpy(this->masterPassword, masterPassword);
|
||||
}
|
||||
|
||||
PrintMessage(MESG_DEBUG, "\n CheckMasterPassword : specified master password is correct...");
|
||||
retValue = FPM_TRUE;
|
||||
}
|
||||
else
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n CheckMasterPassword : specified master password is wrong...");
|
||||
retValue = FPM_FALSE;
|
||||
}
|
||||
|
||||
// Free the slot
|
||||
(*PK11FreeSlot) (slot);
|
||||
|
||||
return retValue;
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
int ProfileManager::GetSignonData( struct Host **host, int doRefresh)
|
||||
{
|
||||
int retValue;
|
||||
|
||||
//TODO NAG:
|
||||
//check if "remember password" enabled....
|
||||
|
||||
// Reload the signon data from the file...
|
||||
if( doRefresh )
|
||||
{
|
||||
retValue = RefreshSignonData();
|
||||
*host = signonManager.GetHostInfo();
|
||||
return retValue;
|
||||
}
|
||||
|
||||
// if host list is null read again....
|
||||
// TODO NAG : If there is no data , this will cause refresh to happen every time...
|
||||
|
||||
if( signonManager.GetHostInfo() == NULL )
|
||||
{
|
||||
retValue = RefreshSignonData();
|
||||
*host = signonManager.GetHostInfo();
|
||||
return retValue;
|
||||
}
|
||||
|
||||
// Just return the previously loaded data
|
||||
*host = signonManager.GetHostInfo();
|
||||
|
||||
return FPM_TRUE;
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
int ProfileManager::RefreshSignonData()
|
||||
{
|
||||
|
||||
// First check if master password is correct....
|
||||
if( masterPassword )
|
||||
{
|
||||
// Just check for password , don't save....
|
||||
if( CheckMasterPassword(masterPassword, 0 ) != FPM_TRUE )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n RefreshSignonData : master password is wrong..can't proceed with signon decryption" );
|
||||
return FPM_MASTERPASSWORD_WRONG;
|
||||
}
|
||||
}
|
||||
else // check if master password is set...
|
||||
{
|
||||
// master password is set and its not specified
|
||||
if( IsMasterPasswordSet() == FPM_TRUE )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n RefreshSignonData : master password is set..but its not specified " );
|
||||
return FPM_MASTERPASSWORD_WRONG;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
return signonManager.LoadSignonData(profilePath);
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
int ProfileManager::WriteSignonData()
|
||||
{
|
||||
// First check if master password is correct....
|
||||
if( masterPassword )
|
||||
{
|
||||
// Just check for password , don't save....
|
||||
if( CheckMasterPassword(masterPassword, 0 ) != FPM_TRUE )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n WriteSignonData : master password is wrong..can't proceed with signon decryption" );
|
||||
return FPM_MASTERPASSWORD_WRONG;
|
||||
}
|
||||
}
|
||||
else // check if master password is set...
|
||||
{
|
||||
// master password is set and its not specified
|
||||
if( IsMasterPasswordSet() == FPM_TRUE )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n WriteSignonData : master password is set..but its not specified " );
|
||||
return FPM_MASTERPASSWORD_WRONG;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
return signonManager.WriteSignonData(profilePath);
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
int ProfileManager::AddHost(struct Host *host, int doUpdate)
|
||||
{
|
||||
int retValue;
|
||||
|
||||
if( (retValue = signonManager.AddHost(host)) != FPM_TRUE)
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n AddHost : Failed to add the host %s ", host->hostName);
|
||||
return retValue;
|
||||
}
|
||||
|
||||
if( doUpdate > 0 )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n AddHost : Updating the signon file ");
|
||||
return WriteSignonData();
|
||||
}
|
||||
|
||||
return FPM_TRUE;
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
int ProfileManager::ModifyHost(struct Host *host, int doUpdate)
|
||||
{
|
||||
int retValue;
|
||||
|
||||
if( (retValue = signonManager.ModifyHost(host)) != FPM_TRUE)
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n ModifyHost : Failed to modify the host %s ", host->hostName);
|
||||
return retValue;
|
||||
}
|
||||
|
||||
|
||||
if( doUpdate > 0)
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n ModifyHost : Updating the signon file ");
|
||||
return WriteSignonData();
|
||||
}
|
||||
|
||||
return FPM_TRUE;
|
||||
|
||||
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
||||
int ProfileManager::RemoveHost(char *hostName, int doUpdate)
|
||||
{
|
||||
int retValue;
|
||||
|
||||
if( (retValue = signonManager.RemoveHost(hostName)) != FPM_TRUE)
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n RemoveHost : Failed to remove the hsot %s ", hostName);
|
||||
return retValue;
|
||||
}
|
||||
|
||||
if( doUpdate > 0 )
|
||||
{
|
||||
PrintMessage(MESG_ERROR, "\n RemoveHost : Updating the signon file ");
|
||||
return WriteSignonData();
|
||||
}
|
||||
|
||||
return FPM_TRUE;
|
||||
}
|
||||
Reference in New Issue
Block a user