The Tor Browser / file comparison
comparison: security/nss/cmd/tests/secmodtest.c
security/nss/cmd/tests/secmodtest.c
- branch
- TOR_BUG_9701
- changeset 15
- b8a032363ba2
equal
deleted
inserted
replaced
| -1:000000000000 | 0:a20908865c37 |
|---|---|
| 1 /* This Source Code Form is subject to the terms of the Mozilla Public | |
| 2 * License, v. 2.0. If a copy of the MPL was not distributed with this | |
| 3 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ | |
| 4 | |
| 5 /* | |
| 6 * Regression test for bug 588269 | |
| 7 * | |
| 8 * SECMOD_CloseUserDB should properly close the user DB, and it should | |
| 9 * be possible to later re-add that same user DB as a new slot | |
| 10 */ | |
| 11 | |
| 12 #include "secutil.h" | |
| 13 | |
| 14 #include <stdio.h> | |
| 15 #include <stdlib.h> | |
| 16 #include <string.h> | |
| 17 | |
| 18 #include "nspr.h" | |
| 19 #include "nss.h" | |
| 20 #include "secerr.h" | |
| 21 #include "pk11pub.h" | |
| 22 #include "plgetopt.h" | |
| 23 | |
| 24 void Usage(char *progName) | |
| 25 { | |
| 26 fprintf(stderr, "Usage: %s -d dbDir\n", progName); | |
| 27 exit(1); | |
| 28 } | |
| 29 | |
| 30 SECStatus TestOpenCloseUserDB(char *progName, char *configDir, char *tokenName) | |
| 31 { | |
| 32 char *modspec = NULL; | |
| 33 SECStatus rv = SECSuccess; | |
| 34 PK11SlotInfo *userDbSlot = NULL; | |
| 35 | |
| 36 printf("Loading database <%s> - %s\n", configDir, tokenName); | |
| 37 modspec = PR_smprintf("configDir='%s' tokenDescription='%s'", | |
| 38 configDir, tokenName); | |
| 39 if (!modspec) { | |
| 40 rv = SECFailure; | |
| 41 goto loser; | |
| 42 } | |
| 43 | |
| 44 userDbSlot = SECMOD_OpenUserDB(modspec); | |
| 45 PR_smprintf_free(modspec); | |
| 46 if (!userDbSlot) { | |
| 47 SECU_PrintError(progName, "couldn't open database"); | |
| 48 rv = SECFailure; | |
| 49 goto loser; | |
| 50 } | |
| 51 | |
| 52 printf("Closing database\n"); | |
| 53 rv = SECMOD_CloseUserDB(userDbSlot); | |
| 54 | |
| 55 if (rv != SECSuccess) { | |
| 56 SECU_PrintError(progName, "couldn't close database"); | |
| 57 } | |
| 58 | |
| 59 PK11_FreeSlot(userDbSlot); | |
| 60 | |
| 61 loser: | |
| 62 return rv; | |
| 63 } | |
| 64 | |
| 65 int main(int argc, char **argv) | |
| 66 { | |
| 67 PLOptState *optstate; | |
| 68 PLOptStatus optstatus; | |
| 69 SECStatus rv = SECFailure; | |
| 70 char *progName; | |
| 71 char *dbDir = NULL; | |
| 72 | |
| 73 progName = strrchr(argv[0], '/'); | |
| 74 if (!progName) { | |
| 75 progName = strrchr(argv[0], '\\'); | |
| 76 } | |
| 77 progName = progName ? progName+1 : argv[0]; | |
| 78 | |
| 79 optstate = PL_CreateOptState(argc, argv, "d:"); | |
| 80 while ((optstatus = PL_GetNextOpt(optstate)) == PL_OPT_OK) { | |
| 81 switch (optstate->option) { | |
| 82 case 'd': | |
| 83 dbDir = strdup(optstate->value); | |
| 84 break; | |
| 85 } | |
| 86 } | |
| 87 if (optstatus == PL_OPT_BAD || dbDir == NULL) { | |
| 88 Usage(progName); | |
| 89 } | |
| 90 | |
| 91 rv = NSS_NoDB_Init(NULL); | |
| 92 if (rv != SECSuccess) { | |
| 93 SECU_PrintError(progName, "unable to initialize NSS"); | |
| 94 goto loser; | |
| 95 } | |
| 96 | |
| 97 printf("Open and Close Test 1\n"); | |
| 98 rv = TestOpenCloseUserDB(progName, dbDir, "Test Slot 1"); | |
| 99 if (rv != SECSuccess) { | |
| 100 goto loser; | |
| 101 } | |
| 102 | |
| 103 printf("Open and Close Test 2\n"); | |
| 104 rv = TestOpenCloseUserDB(progName, dbDir, "Test Slot 2"); | |
| 105 if (rv != SECSuccess) { | |
| 106 goto loser; | |
| 107 } | |
| 108 | |
| 109 loser: | |
| 110 if (dbDir) free(dbDir); | |
| 111 | |
| 112 if (NSS_Shutdown() != SECSuccess) { | |
| 113 exit(1); | |
| 114 } | |
| 115 PR_Cleanup(); | |
| 116 | |
| 117 if (rv != SECSuccess) { | |
| 118 exit(1); | |
| 119 } | |
| 120 | |
| 121 return 0; | |
| 122 } |
