/* ***** 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, &regKey);

    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;

}