Bug 1496124 - Populate public values for imported private keys, r=mt

1 files changed, 17 insertions, 1 deletions

diff --git a/gtests/common/gtests.cc b/gtests/common/gtests.cc
index bd5a97a8e..7e585791b 100644
--- a/gtests/common/gtests.cc
+++ b/gtests/common/gtests.cc
@@ -10,7 +10,23 @@
 int main(int argc, char **argv) {
   ::testing::InitGoogleTest(&argc, argv);
 
-  if (NSS_NoDB_Init(nullptr) != SECSuccess) {
+  const char *workdir = "";
+  uint32_t flags = NSS_INIT_READONLY;
+
+  for (int i = 0; i < argc; i++) {
+    if (!strcmp(argv[i], "-d")) {
+      if (i + 1 >= argc) {
+        PR_fprintf(PR_STDERR, "Usage: %s [-d <dir> [-w]]\n", argv[0]);
+        exit(2);
+      }
+      workdir = argv[i + 1];
+      i++;
+    } else if (!strcmp(argv[i], "-w")) {
+      flags &= ~NSS_INIT_READONLY;
+    }
+  }
+
+  if (NSS_Initialize(workdir, "", "", SECMOD_DB, flags) != SECSuccess) {
     return 1;
   }
   if (NSS_SetDomesticPolicy() != SECSuccess) {