summaryrefslogtreecommitdiffstats
path: root/pki/base/ra/lib/perl/PKI/Base/UserStore.pm
diff options
context:
space:
mode:
Diffstat (limited to 'pki/base/ra/lib/perl/PKI/Base/UserStore.pm')
-rw-r--r--pki/base/ra/lib/perl/PKI/Base/UserStore.pm343
1 files changed, 343 insertions, 0 deletions
diff --git a/pki/base/ra/lib/perl/PKI/Base/UserStore.pm b/pki/base/ra/lib/perl/PKI/Base/UserStore.pm
new file mode 100644
index 000000000..c05683792
--- /dev/null
+++ b/pki/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;
generated by cgit (git 2.34.1) at 2024-05-03 17:34:33 +0000