diff options
Diffstat (limited to 'base/ra/lib/perl/PKI/Base/UserStore.pm')
-rw-r--r-- | base/ra/lib/perl/PKI/Base/UserStore.pm | 343 |
1 files changed, 343 insertions, 0 deletions
diff --git a/base/ra/lib/perl/PKI/Base/UserStore.pm b/base/ra/lib/perl/PKI/Base/UserStore.pm new file mode 100644 index 000000000..c05683792 --- /dev/null +++ b/base/ra/lib/perl/PKI/Base/UserStore.pm @@ -0,0 +1,343 @@ +#!/usr/bin/perl +# +# --- BEGIN COPYRIGHT BLOCK --- +# 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; version 2 of the License. +# +# 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., +# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. +# +# Copyright (C) 2007 Red Hat, Inc. +# All rights reserved. +# --- END COPYRIGHT BLOCK --- +# +# +# +# +package PKI::Base::UserStore; + +use DBI; +use PKI::Base::TimeTool; + +####################################### +# Constructs a request queue +####################################### +sub new { + my $self = {}; + bless ($self); + return $self; +} + +####################################### +# Opens this store +####################################### +sub open { + my ($self, $cfg) = @_; + $self->{cfg} = $cfg; + my $dbfile = $cfg->get("database.dbfile"); + $self->{dbh} = DBI->connect("dbi:SQLite:dbname=$dbfile","",""); + my $timeout = $self->{dbh}->func("busy_timeout"); + $self->{dbh}->func($timeout * 10, "busy_timeout"); +} + +####################################### +# Maps user +####################################### +sub map_user { + my ($self, $certificate) = @_; + my $dbh = $self->{dbh}; + my $select = "select * from users " . + "where " . + "certificate=" . $dbh->quote($certificate); + my $sth = $dbh->prepare($select); + $sth->execute(); + my $ref = $sth->fetchrow_hashref(); + $sth->finish(); + return $ref; +} + +####################################### +# Gets roles of the given user +####################################### +sub get_roles { + my ($self, $uid) = @_; + my $dbh = $self->{dbh}; + my $select = "select * from roles " . + "where " . + "uid=" . $dbh->quote($uid); + my @roles; + my $sth = $dbh->prepare($select); + $sth->execute(); + while (my $ref = $sth->fetchrow_hashref()) { + push(@roles, $ref->{'gid'}); + } + $sth->finish(); + return @roles; +} + + +####################################### +# Reads a user +####################################### +sub read_group { + my ($self, $gid) = @_; + my $dbh = $self->{dbh}; + my $select = "select * from groups " . + "where gid=" . $dbh->quote($gid); + my $sth = $dbh->prepare($select); + $sth->execute(); + my $ref = $sth->fetchrow_hashref(); + $sth->finish(); + return $ref; +} + +sub read_user { + my ($self, $uid) = @_; + my $dbh = $self->{dbh}; + my $select = "select * from users " . + "where uid=" . $dbh->quote($uid); + my $sth = $dbh->prepare($select); + $sth->execute(); + my $ref = $sth->fetchrow_hashref(); + $sth->finish(); + return $ref; +} + +sub set_user { + my ($self, $uid, $name, $value) = @_; + my $dbh = $self->{dbh}; + + my $timet = PKI::Base::TimeTool->new(); + my $now = $timet->get_time(); + my $update = "update users set " . + $name . "=" . $dbh->quote($value) . "," . + "updated_at=" . $dbh->quote($now) . " " . + "where uid=" . $dbh->quote($uid); + $dbh->do($update); + + my $select = "select * from users " . + "where uid=" . $dbh->quote($uid); + my $sth = $dbh->prepare($select); + $sth->execute(); + my $ref = $sth->fetchrow_hashref(); + $sth->finish(); + + return $ref; +} + +####################################### +# Lists all members in the given group +####################################### +sub list_all_members { + my ($self, $gid) = @_; + my $dbh = $self->{dbh}; + my $select = "select * from roles where " . + "gid=" . $dbh->quote($gid) . " " . + "order by uid desc "; + my $sth = $dbh->prepare($select); + $sth->execute(); + my @reqs; + while (my $ref = $sth->fetchrow_hashref()) { + push(@reqs, $ref); + } + $sth->finish(); + return @reqs; +} + +####################################### +# Lists +####################################### +sub list_all_non_members { + my ($self, $gid) = @_; + my $dbh = $self->{dbh}; + # find members of the given group + my $select1 = "select * from roles where " . + "gid=" . $dbh->quote($gid); + my $sth1 = $dbh->prepare($select1); + $sth1->execute(); + my $filter = ""; + while (my $ref1 = $sth1->fetchrow_hashref()) { + if ($filter eq "") { + $filter = "uid<>" . $dbh->quote($ref1->{'uid'}); + } else { + $filter = $filter . " AND " . "uid<>" . $dbh->quote($ref1->{'uid'}); + } + } + $sth1->finish(); + + my $select; + if ($filter eq "") { + $select = "select * from users " . + "order by uid desc "; + } else { + $select = "select * from users where (" . + $filter . ") " . + "order by uid desc "; + } + my $sth = $dbh->prepare($select); + $sth->execute(); + my @reqs; + while (my $ref = $sth->fetchrow_hashref()) { + push(@reqs, $ref); + } + $sth->finish(); + return @reqs; +} + +sub delete_user { + my ($self, $userid) = @_; + my $dbh = $self->{dbh}; + + my $cmd = "delete from roles where uid=" . $dbh->quote($userid); + $dbh->do($cmd); + $cmd = "delete from users where uid=" . $dbh->quote($userid); + $dbh->do($cmd); +} + +sub add_user_to_group { + my ($self, $gid, $userid) = @_; + my $dbh = $self->{dbh}; + + my $timet = PKI::Base::TimeTool->new(); + my $now = $timet->get_time(); + + my $cmd = "insert into roles (" . + "gid" . "," . + "uid" . + ") values (" . + $dbh->quote($gid) . "," . + $dbh->quote($userid) . + ")"; + $dbh->do($cmd); +} + +sub delete_user_from_group { + my ($self, $gid, $userid) = @_; + my $dbh = $self->{dbh}; + + my $timet = PKI::Base::TimeTool->new(); + my $now = $timet->get_time(); + + my $cmd = "delete from roles where " . + "gid=" . $dbh->quote($gid) . " AND " . + "uid=" . $dbh->quote($userid); + $dbh->do($cmd); +} + +sub add_user { + my ($self, $userid, $name, $email, $certificate) = @_; + my $dbh = $self->{dbh}; + + my $timet = PKI::Base::TimeTool->new(); + my $now = $timet->get_time(); + + my $cmd = "insert into users (" . + "uid" . "," . + "name" . "," . + "email" . "," . + "certificate" . "," . + "created_at" . + ") values (" . + $dbh->quote($userid) . "," . + $dbh->quote($name) . "," . + $dbh->quote($email) . "," . + $dbh->quote($certificate) . "," . + $dbh->quote($now) . + ")"; +REDO_ADD_USER: + eval { + $dbh->do($cmd); + }; + if ($dbh->err == 5) { + sleep(1); + goto REDO_ADD_USER; + } +} + +sub add_group { + my ($self, $gid, $name) = @_; + my $dbh = $self->{dbh}; + + my $timet = PKI::Base::TimeTool->new(); + my $now = $timet->get_time(); + + my $cmd = "insert into groups (" . + "gid" . "," . + "name" . "," . + "created_at" . + ") values (" . + $dbh->quote($gid) . "," . + $dbh->quote($name) . "," . + $dbh->quote($now) . + ")"; +REDO_ADD_GROUP: + eval { + $dbh->do($cmd); + }; + if ($dbh->err == 5) { + sleep(1); + goto REDO_ADD_GROUP; + } +} + +sub delete_group { + my ($self, $gid) = @_; + my $dbh = $self->{dbh}; + + my $timet = PKI::Base::TimeTool->new(); + my $now = $timet->get_time(); + + my $cmd = "delete from roles where gid=" . $dbh->quote($gid); + $dbh->do($cmd); + $cmd = "delete from groups where gid=" . $dbh->quote($gid); + $dbh->do($cmd); +} + +sub list_users { + my ($self, $startpos, $maxcount) = @_; + my $dbh = $self->{dbh}; + my $select = "select * from users " . + "order by uid desc " . + "limit $startpos, $maxcount"; + my $sth = $dbh->prepare($select); + $sth->execute(); + my @reqs; + while (my $ref = $sth->fetchrow_hashref()) { + push(@reqs, $ref); + } + $sth->finish(); + return @reqs; +} + +sub list_groups { + my ($self, $startpos, $maxcount) = @_; + my $dbh = $self->{dbh}; + my $select = "select * from groups " . + "order by gid desc " . + "limit $startpos, $maxcount"; + my $sth = $dbh->prepare($select); + $sth->execute(); + my @reqs; + while (my $ref = $sth->fetchrow_hashref()) { + push(@reqs, $ref); + } + $sth->finish(); + return @reqs; +} +####################################### +# Closes this store +####################################### +sub close { + my ($self) = @_; + my $dbh = $self->{dbh}; + $dbh->disconnect(); +} + +1; |