1 // Copyright (C) 1999,2000 Bruce Guenter <bruceg@em.ca>

     2 //

     3 // This program is free software; you can redistribute it and/or modify

     4 // it under the terms of the GNU General Public License as published by

     5 // the Free Software Foundation; either version 2 of the License, or

     6 // (at your option) any later version.

     7 //

     8 // This program is distributed in the hope that it will be useful,

     9 // but WITHOUT ANY WARRANTY; without even the implied warranty of

    10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    11 // GNU General Public License for more details.

    12 //

    13 // You should have received a copy of the GNU General Public License

    14 // along with this program; if not, write to the Free Software

    15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

    16 

    17 #include <config.h>

    18 #include <errno.h>

    19 #include "vpwtable.h"

    20 #include "cdb++/cdb++.h"

    21 

    22 bool vpwtable::put(const vpwentry* vpw, bool onlyadd) const

    23 {

    24   cdb_reader in(filename);

    25   if(!in && errno != ENOENT)

    26     return false;

    27  

    28   mystring cdbtmpname = filename + ".tmp";

    29   cdb_writer out(cdbtmpname, 0600);

    30   if(!out)

    31     return false;

    32 

    33   mystring lname = vpw->name.lower();

    34   if(in) {

    35     bool failed = false;

    36     datum* d;

    37     while(!failed && (d = in.nextrec()) != 0) {

    38       if(vpw && d->key == lname) {

    39 	if(onlyadd)

    40 	  // names compared equal, but we only wanted to add

    41 	  failed = true;

    42 	else if(!out.put(lname, vpw->to_record()))

    43 	  failed = true;

    44 	else

    45 	  vpw = 0;

    46       }

    47       else

    48 	failed = !out.put(d->key, d->data);

    49       delete d;

    50     }

    51     if(failed)

    52       return false;

    53   }

    54   if(vpw) {

    55     if(!onlyadd)

    56       return false;

    57     if(!out.put(lname, vpw->to_record()))

    58       return false;

    59   }

    60   if(!out.end(filename))

    61     return false;

    62   return true;

    63 }