You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
452 lines
11 KiB
452 lines
11 KiB
/* ***** BEGIN COPYRIGHT BLOCK ***** |
|
* Copyright (C) 2006 Red Hat, Inc. |
|
* All rights reserved. |
|
* |
|
* This library is free software; you can redistribute it and/or |
|
* modify it under the terms of the GNU Lesser General Public |
|
* License as published by the Free Software Foundation version |
|
* 2.1 of the License. |
|
* |
|
* This library is distributed in the hope that it will be useful, |
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
|
* Lesser General Public License for more details. |
|
* |
|
* You should have received a copy of the GNU Lesser General Public |
|
* License along with this library; if not, write to the Free Software |
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA |
|
* ***** END COPYRIGHT BLOCK ***** */ |
|
|
|
#include <stdio.h> |
|
#include <string.h> |
|
#include "pkcs11.h" |
|
#include "pkcs11n.h" |
|
|
|
/* |
|
* windows specific globing search |
|
*/ |
|
#ifdef WIN32 |
|
#include <windows.h> |
|
#include <winver.h> |
|
#include <winreg.h> |
|
#include <direct.h> |
|
#include <shlobj.h> |
|
|
|
#define PINST_FILE_DATA WIN32_FIND_DATA |
|
#define PINST_ITERATOR HANDLE |
|
#define PINST_FIRST(pattern, data) FindFirstFile(pattern, &data) |
|
#define PINST_PATH(iter, data) (data).cFileName |
|
#define PINST_NEXT(iter, data) FindNextFile(iter, &data) |
|
#define PINST_FREE_ITER(iter, data) FindClose(iter) |
|
#define PINST_INVALID_ITERATOR INVALID_HANDLE_VALUE |
|
#define PINST_IS_DIRECTORY(iter, data) \ |
|
((data).dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) |
|
#define PINST_IS_HIDDEN(iter, data) \ |
|
((data).dwFileAttributes & FILE_ATTRIBUTE_HIDDEN) |
|
#define PINST_FULLPATH(tempPath,path) tempPath |
|
#define PINST_ERROR DWORD |
|
#define PINST_NO_MORE ERROR_NO_MORE_FILES |
|
#define PINST_SET_ERROR(x) SetLastError(x) |
|
#define PINST_GET_ERROR() GetLastError() |
|
#define PINST_FS "\\" |
|
|
|
|
|
/*#define NETSCAPE_KEY "Software\\Netscape\\Netscape Navigator\\Main" */ |
|
#define NETSCAPE_KEY "Software\\Netscape\\Netscape Navigator" |
|
#define NETSCAPE_SUBKEY_1 "Main" |
|
#define NETSCAPE_SUBKEY_2 "Install Directory" |
|
|
|
/* capture the window's error string */ |
|
static void |
|
winPerror(FILE *outFile, DWORD error, const char *msgString) |
|
{ |
|
char buffer[256]; |
|
char *cp; |
|
DWORD ret; |
|
|
|
fprintf(outFile,"*** %s: ",msgString); |
|
sprintf(buffer,"Format message problem, error = %d (0x%x)\n", error, error); |
|
ret=FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM, NULL, error, 0, buffer, |
|
sizeof(buffer), NULL); |
|
for (cp=buffer; *cp; cp++) { |
|
if (*cp == '\r') *cp = ' '; |
|
} |
|
fprintf(outFile, buffer); |
|
} |
|
#endif |
|
|
|
/* |
|
* otherwise we are assuming unix (posix) |
|
*/ |
|
#ifndef PINST_FILE_DATA |
|
#define UNIX |
|
#include <stdlib.h> |
|
#include <limits.h> |
|
#include <glob.h> |
|
#define PINST_FILE_DATA glob_t |
|
#define PINST_ITERATOR int |
|
#define PINST_INVALID_ITERATOR -1 |
|
#define PINST_FIRST(pattern, data) \ |
|
((glob(pattern, GLOB_MARK, NULL, &data) == 0) ? 0 : PINST_INVALID_ITERATOR) |
|
#define PINST_PATH(iter, data) \ |
|
(((data).gl_pathv == NULL) ? 0 : (data).gl_pathv[iter] ) |
|
#define PINST_NEXT(iter, data) (((data).gl_pathc > ++iter) ? iter : 0) |
|
#define PINST_FREE_ITER(iter, data) globfree(&data) |
|
#define PINST_IS_DIRECTORY(iter, data) pinst_isdir(PINST_PATH(iter,data)) |
|
#define PINST_IS_HIDDEN(iter, data) (0) |
|
#define PINST_FULLPATH(tempPath,path) path |
|
#define PINST_ERROR int |
|
#define NO_ERROR 0 |
|
#define PINST_NO_MORE NO_ERROR |
|
#define PINST_SET_ERROR(x) |
|
#define PINST_GET_ERROR() NO_ERROR |
|
#define PINST_FS "/" |
|
|
|
|
|
#define MAX_PATH PATH_MAX |
|
|
|
static int |
|
pinst_isdir(const char *path) |
|
{ |
|
int len = strlen(path); |
|
|
|
|
|
return (len > 0) && (path[len-1] == '/'); |
|
} |
|
|
|
#endif |
|
|
|
|
|
typedef enum _InstType { |
|
Install, |
|
UnInstall, |
|
} InstType; |
|
|
|
typedef enum _DirType { |
|
AppDataDir = 0, |
|
HomeDir, |
|
NetscapeInstallDir, |
|
MaxDirType, |
|
} DirType; |
|
|
|
char *dirPaths[MaxDirType] = { NULL }; |
|
|
|
typedef struct _DirList { |
|
DirType dirType; |
|
char *search; |
|
char *tail; |
|
} DirList; |
|
|
|
DirList dirList[] = { |
|
#ifdef WIN32 |
|
{ AppDataDir, "Mozilla\\Profiles\\*", "*.slt" }, |
|
{ AppDataDir, "Mozilla\\Firefox\\Profiles\\*", NULL }, |
|
{ AppDataDir, "Thunderbird\\Profiles\\*", NULL }, |
|
{ NetscapeInstallDir, "..\\Users\\*", NULL }, |
|
#endif |
|
#ifndef MAC |
|
#ifdef UNIX |
|
{ HomeDir, ".mozilla/firefox/*", NULL }, |
|
{ HomeDir, ".mozilla/*", NULL }, |
|
{ HomeDir, ".thunderbird/*", NULL }, |
|
{ HomeDir, ".netscape", NULL }, |
|
#endif |
|
#endif |
|
#ifdef MAC |
|
|
|
{ HomeDir, "Library/Mozilla/Profiles/*", "*.slt"}, |
|
{ HomeDir, "Library/Application Support/Firefox/Profiles/*", NULL }, |
|
{ HomeDir, "Library/Thunderbird/Profiles/*", NULL }, |
|
|
|
|
|
#endif |
|
|
|
|
|
}; |
|
|
|
int verbose = 0; |
|
|
|
int dirListCount = sizeof(dirList)/sizeof(dirList[0]); |
|
|
|
static void |
|
usage(char *prog) |
|
{ |
|
fprintf(stderr,"usage: %s [-u][-v] [-p path] module\n", prog); |
|
return; |
|
} |
|
|
|
/* Utility printing functions */ |
|
|
|
|
|
|
|
#define CONFIG_TAG "configDir=" |
|
int |
|
installPKCS11(char *dirPath, InstType type, char *module) |
|
{ |
|
char *paramString = (char *)malloc(strlen(dirPath)+sizeof(CONFIG_TAG)+3); |
|
char *cp; |
|
char **rc; |
|
|
|
if (paramString == NULL) { |
|
PINST_SET_ERROR(ERROR_NOT_ENOUGH_MEMORY); |
|
return 0; |
|
} |
|
sprintf(paramString,CONFIG_TAG"\"%s\" ",dirPath); |
|
|
|
/* translate all the \'s to /'s */ |
|
for (cp=paramString; *cp; cp++) { |
|
if (*cp == '\\') *cp='/'; |
|
} |
|
|
|
/* don't call this if you have NSS initialized!!, use SECMOD_AddModule |
|
* or SECMOD_AddUserModule instead */ |
|
rc = (char **) NSC_ModuleDBFunc(type == Install ? |
|
SECMOD_MODULE_DB_FUNCTION_ADD : |
|
SECMOD_MODULE_DB_FUNCTION_DEL, paramString, module); |
|
if (verbose) { |
|
fprintf(stderr, "Install \"%s\" in %s : %s\n", module, dirPath, |
|
rc ? *rc : "Fail" ); |
|
} |
|
|
|
free(paramString); |
|
return 1; |
|
} |
|
|
|
|
|
int |
|
installAllPKCS11(char *dirPath, char *search, char *tail, |
|
InstType type, char *module) |
|
{ |
|
char *searchString; |
|
unsigned long searchStringLen; |
|
int len; |
|
char *tempPath, *fileStart; |
|
PINST_FILE_DATA fileData; |
|
PINST_ITERATOR iter; |
|
PINST_ERROR err = NO_ERROR; |
|
|
|
char *myPath = NULL; |
|
|
|
searchString = (char *)malloc(strlen(dirPath)+2+strlen(search)); |
|
|
|
if (searchString == NULL) { |
|
PINST_SET_ERROR(ERROR_NOT_ENOUGH_MEMORY); |
|
return 0; |
|
} |
|
sprintf(searchString,"%s" PINST_FS "%s",dirPath,search); |
|
|
|
searchStringLen = strlen(searchString); |
|
tempPath=malloc(searchStringLen+MAX_PATH+1); |
|
if (tempPath == NULL) { |
|
free(searchString); |
|
PINST_SET_ERROR(ERROR_NOT_ENOUGH_MEMORY); |
|
return 0; |
|
} |
|
strcpy(tempPath, searchString); |
|
fileStart = strrchr(tempPath, *PINST_FS); |
|
if (fileStart == NULL) { |
|
tempPath[searchStringLen] = *PINST_FS; |
|
fileStart = &tempPath[searchStringLen]; |
|
} |
|
fileStart++; |
|
|
|
iter = PINST_FIRST(searchString, fileData); |
|
free(searchString); |
|
if (iter == PINST_INVALID_ITERATOR) { |
|
/* error set by PINST_FIRST */ |
|
free(tempPath); |
|
return 0; |
|
} |
|
|
|
len=1; |
|
|
|
do { |
|
char *path = PINST_PATH(iter, fileData); |
|
if(!path) |
|
{ |
|
break; |
|
} |
|
|
|
if (!PINST_IS_DIRECTORY(iter, fileData)) { |
|
continue; |
|
} |
|
if (PINST_IS_HIDDEN(iter, fileData)) { |
|
continue; |
|
} |
|
/* skip . and .. */ |
|
if ((path[0] == '.') && ((path[1] == 0) || |
|
(path[1] == '.' && path[2] == 0)) ) { |
|
continue; |
|
} |
|
strcpy(fileStart,path); |
|
|
|
myPath=PINST_FULLPATH(tempPath,path); |
|
if (tail) { |
|
installAllPKCS11(myPath, tail, NULL, type, module); |
|
} else { |
|
installPKCS11(myPath, type, module); |
|
} |
|
} while (PINST_NEXT(iter, fileData)); |
|
free(tempPath); |
|
|
|
err = PINST_GET_ERROR(); |
|
PINST_FREE_ITER(iter,fileData); |
|
|
|
if (err != PINST_NO_MORE) { |
|
/* restore the previous error (in case FindClose trashes it) */ |
|
PINST_SET_ERROR(err); |
|
return 0; |
|
} |
|
return 1; |
|
} |
|
|
|
int main(int argc, char **argv) |
|
{ |
|
char *module = NULL; |
|
char *prog = *argv++; |
|
char *cp; |
|
int argCount = 0; |
|
int i; |
|
InstType type = Install; |
|
char * path = NULL; |
|
#ifdef WIN32 |
|
BOOL brc; |
|
HKEY regKey; |
|
unsigned long lrc; |
|
TCHAR appData[MAX_PATH]; |
|
char netscapeInstall[MAX_PATH]; |
|
unsigned long nsInstallSize = MAX_PATH; |
|
#endif |
|
|
|
/* |
|
* parse the arglist; |
|
*/ |
|
while ((cp = *argv++) != 0) { |
|
if (*cp == '-') { |
|
while (*++cp) switch (*cp) { |
|
case 'i': |
|
type = Install; |
|
break; |
|
case 'u': |
|
type = UnInstall; |
|
break; |
|
case 'v': |
|
verbose = 1; |
|
break; |
|
case 'p': |
|
path = *argv++; |
|
if (path == NULL) { |
|
usage(prog); |
|
return 2; |
|
} |
|
break; |
|
default: |
|
usage(prog); |
|
return 2; |
|
} |
|
} else switch (argCount++) { |
|
case 0: |
|
module = cp; |
|
break; |
|
default: |
|
usage(prog); |
|
return 2; |
|
} |
|
} |
|
|
|
if (module == NULL) { |
|
usage(prog); |
|
} |
|
|
|
if (path) { |
|
installAllPKCS11(path, "", NULL, type, module); |
|
return 0; |
|
} |
|
|
|
#ifdef WIN32 |
|
/* App Data Dir */ |
|
brc = SHGetSpecialFolderPath(NULL, appData, CSIDL_APPDATA, FALSE); |
|
if (brc) { |
|
dirPaths[AppDataDir] = appData; |
|
} else { |
|
if (verbose) { |
|
winPerror(stderr, GetLastError(), "Reading App Directory"); |
|
} |
|
} |
|
|
|
/* Netscape Install Dir */ |
|
lrc = RegOpenKeyEx(HKEY_LOCAL_MACHINE, NETSCAPE_KEY, 0, |
|
KEY_ENUMERATE_SUB_KEYS, ®Key); |
|
if (lrc == ERROR_SUCCESS) { |
|
int i = 0; |
|
TCHAR productName[255]; |
|
HKEY prodKey; |
|
HKEY mainKey; |
|
|
|
while ((lrc = RegEnumKey(regKey, i, productName, sizeof(productName))) |
|
== ERROR_SUCCESS) { |
|
i++; |
|
lrc = RegOpenKeyEx(regKey, productName, 0, |
|
KEY_ENUMERATE_SUB_KEYS, &prodKey); |
|
if (lrc != ERROR_SUCCESS) { |
|
if (verbose) { |
|
winPerror(stderr, GetLastError(), |
|
"Reading Netscape 4.0 prodkey"); |
|
fprintf(stderr,"Product = %s\n",productName); |
|
} |
|
continue; |
|
} |
|
lrc = RegOpenKeyEx(prodKey, NETSCAPE_SUBKEY_1, 0, |
|
KEY_QUERY_VALUE, &mainKey); |
|
if (lrc != ERROR_SUCCESS) { |
|
RegCloseKey(prodKey); |
|
continue; |
|
} |
|
/* open main */ |
|
lrc = RegQueryValueEx(mainKey, NETSCAPE_SUBKEY_2, NULL, NULL, |
|
netscapeInstall, &nsInstallSize); |
|
RegCloseKey(mainKey); |
|
RegCloseKey(prodKey); |
|
if (lrc == ERROR_SUCCESS) { |
|
if (netscapeInstall[nsInstallSize-1] == 0) { |
|
if (verbose) { |
|
fprintf(stderr, |
|
"Found Netscape 4.0 Install directory\n"); |
|
} |
|
dirPaths[NetscapeInstallDir] = netscapeInstall; |
|
break; |
|
} else { |
|
fprintf(stderr, |
|
"Reading Netscape 4.0 key: Value too large\n"); |
|
} |
|
} else { |
|
if (verbose) { |
|
winPerror(stderr, lrc, "Reading Netscape 4.0 key"); |
|
} |
|
} |
|
} |
|
if ((lrc != ERROR_SUCCESS) && (lrc != ERROR_NO_MORE_ITEMS)) { |
|
winPerror(stderr, lrc, "EnumKey on Netscape Registry Key failed"); |
|
} |
|
} else { |
|
if (verbose) { |
|
winPerror(stderr, lrc, "Openning Netscape 4.0 key"); |
|
} |
|
} |
|
#endif |
|
#ifdef UNIX |
|
dirPaths[HomeDir] = getenv("HOME"); |
|
#endif |
|
|
|
/* OK, now search the directories and complete the Install */ |
|
for (i=0; i < dirListCount; i++) { |
|
char *dirPath = dirPaths[dirList[i].dirType]; |
|
if (!dirPath) { |
|
continue; |
|
} |
|
installAllPKCS11(dirPath, dirList[i].search, dirList[i].tail, |
|
type, module); |
|
} |
|
|
|
return 0; |
|
}
|
|
|