lib/vpwtable/gdbm_write.cc
changeset 2 b3afb9f1e801
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/lib/vpwtable/gdbm_write.cc	Sun Jan 20 00:22:09 2008 +0100
@@ -0,0 +1,86 @@
+// 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 <errno.h>
+#include "vpwtable.h"
+#include <gdbm.h>
+
+class gdbm_vpwtable_writer : public vpwtable_writer
+{
+private:
+  const mystring& tmpname;
+  const mystring& destname;
+  GDBM_FILE out;
+  bool opened;
+public:
+  gdbm_vpwtable_writer(const mystring& filename);
+  ~gdbm_vpwtable_writer();
+  bool operator!() const;
+  bool put(const vpwentry& vpw);
+  bool end();
+  bool abort();
+};
+
+vpwtable_writer* vpwtable::start_write() const
+{
+  return new gdbm_vpwtable_writer(filename);
+}
+
+gdbm_vpwtable_writer::gdbm_vpwtable_writer(const mystring& filename)
+  : tmpname(filename + ".tmp"), destname(filename),
+    out(gdbm_open((char*)filename.c_str(), 0, GDBM_NEWDB|GDBM_FAST, 0600, 0)),
+    opened(true)
+{
+}
+
+gdbm_vpwtable_writer::~gdbm_vpwtable_writer()
+{
+  end();
+}
+
+bool gdbm_vpwtable_writer::operator!() const
+{
+  return opened;
+}
+
+bool gdbm_vpwtable_writer::put(const vpwentry& vpw)
+{
+  mystring name = vpw->name.lower();
+  datum key = { (char*)name.c_str(), name.length() };
+  mystring binary = vpw->to_record();
+  datum data = { (char*)binary.c_str(), binary.length() };
+  return gdbm_store(out, key, data, GDBM_INSERT) == 0;
+}
+
+bool gdbm_vpwtable_writer::end() 
+{
+  if(!opened)
+    return false;
+  gdbm_sync(out);
+  opened = false;
+  return gdbm_close(out) == 0 &&
+    rename(tmpname.c_str(), destname.c_str()) == 0;
+}
+
+bool gdbm_vpwtable_writer::abort()
+{
+  if(!opened)
+    return false;
+  opened = false;
+  gdbm_close(out);
+  return unlink(tmpname.c_str()) == 0;
+}