commands/vpasswd2cdb.cc
changeset 0 6f7a81934006
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/commands/vpasswd2cdb.cc	Wed Jan 16 22:39:43 2008 +0100
@@ -0,0 +1,93 @@
+// Copyright (C) 1999,2000 Bruce Guenter <bruceg@em.ca>
+//
+// This program is free software; you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation; either version 2 of the License, or
+// (at your option) any later version.
+//
+// This program 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 General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program; if not, write to the Free Software
+// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+
+#include <config.h>
+#include <string.h>
+#include "cdb++/cdb++.h"
+#include "cli/cli.h"
+#include "config/configrc.h"
+#include "fdbuf/fdbuf.h"
+#include "mystring/mystring.h"
+#include "vpwentry/vpwentry.h"
+#include "vcommand.h"
+
+const char* cli_program = "vpasswd2cdb";
+const char* cli_help_prefix = "Converts text password tables to CDB format\n";
+const char* cli_help_suffix =
+"Reads in a standard virtual password table in the current directory,
+and writes it out to a CDB table.  The file names for the text and CDB
+tables are determined from the configuration file.";
+const char* cli_args_usage = "";
+const int cli_args_min = 0;
+const int cli_args_max = 0;
+
+static int o_quiet = false;
+
+cli_option cli_options[] = {
+  { 0, "quiet", cli_option::flag, true, &o_quiet,
+    "Suppress all status messages", 0 },
+  {0}
+};
+
+static bool getpw(fdibuf& in, vpwentry& out)
+{
+  mystring buf;
+  if(!in.getline(buf))
+    return false;
+
+  int first = buf.find_first(':');
+  if(first < 0) return false;
+
+  return out.from_record(buf.left(first), buf.c_str() + first);
+}
+
+int cli_main(int, char* [])
+{
+  if(!go_home())
+    return 1;
+
+  fdibuf in(password_file.c_str());
+  if(!in) {
+    if(!o_quiet)
+      ferr << "Can't open password table named '"
+	   << password_file << "'." << endl;
+    return 1;
+  }
+  mystring cdbname = password_file + ".cdb";
+  mystring cdbtmp = cdbname + ".tmp";
+  cdb_writer cdb(cdbtmp, 0600);
+  if(!cdb) {
+    if(!o_quiet)
+      ferr << "Can't open CDB temporary file named '" << cdbtmp << "'."
+	   << endl;
+    return 1;
+  }
+  vpwentry vpw;
+  while(getpw(in, vpw)) {
+    if(!cdb.put(vpw.name, vpw.to_record())) {
+      if(!o_quiet)
+	ferr << "Failed to add record to CDB table." << endl;
+      return 1;
+    }
+  }
+  if(!cdb.end(cdbname)) {
+    if(!o_quiet)
+      ferr << "Failed to finish CDB table into '"
+	   << password_file << ".cdb'." << endl;
+    return 1;
+  }
+  return 0;
+}