1 files changed, 180 insertions, 0 deletions
diff --git a/base/ra/lib/perl/PKI/Base/PinStore.pm b/base/ra/lib/perl/PKI/Base/PinStore.pm
new file mode 100644
index 000000000..437d259ff
--- /dev/null
+++ b/base/ra/lib/perl/PKI/Base/PinStore.pm
@@ -0,0 +1,180 @@
+#!/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::PinStore;
+
+use DBI;
+use PKI::Base::TimeTool;
+
+#######################################
+# Constructs a request queue
+#######################################
+sub new {
+  my $self = {};
+  bless ($self);
+  return $self;
+}
+
+#######################################
+# Opens request queue
+#######################################
+sub open {
+  my ($self, $cfg) = @_;
+  $self->{cfg} = $cfg;
+  my $dbfile = $cfg->get("database.dbfile");
+  $self->{dbh} = DBI->connect("dbi:SQLite:dbname=$dbfile","","");
+}
+
+#######################################
+# Creates a new request
+#######################################
+sub generate_random
+{
+    my $low = $_[0];
+    my $high = $_[1];
+
+    my $number = 0;
+
+    if( $low >= $high || $low < 0 || $high < 0 ) {
+        return -1;
+    }
+
+    $number = int( rand( $high -$low +1 ) ) + $low;
+
+    return $number;
+}
+
+
+# arg0 length of string
+# return random string
+sub generate_random_string()
+{
+    my $length_of_randomstring=shift;  # the length of the string
+
+    my @chars=( 'a'..'z','A'..'Z','0'..'9' );
+    my $random_string;
+
+    foreach( 1..$length_of_randomstring ) {
+        $random_string .= $chars[rand @chars];
+    }
+
+    return $random_string;
+}
+
+sub create_pin {
+  my ($self, $key, $rid, $created_by) = @_;
+  my $dbh = $self->{dbh};
+
+  my $pin = &generate_random_string(10);
+  my $timet = PKI::Base::TimeTool->new();
+  my $now = $timet->get_time();
+
+  # delete previous pin
+  my $delete = "delete from pins where key=" . $dbh->quote($key);
+  $dbh->do($delete);
+
+  my $insert = "insert into pins (" .
+                   "key" . "," .
+                   "pin" . "," .
+                   "rid" . "," .
+                   "created_by" . "," .
+                   "created_at" .
+                 ") values (" .
+                   $dbh->quote($key) . "," .
+                   $dbh->quote($pin) . "," .
+                   $dbh->quote($rid) . "," .
+                   $dbh->quote($created_by) . "," .
+                   $dbh->quote($now) .
+                 ")";
+REDO_CREATE_PIN:
+  eval {
+    $dbh->do($insert);
+  };
+  if ($dbh->err == 5) {
+    sleep(1);
+    goto REDO_CREATE_PIN;
+  }
+
+  my $rid = $dbh->func('last_insert_rowid');
+
+#  my $ref = $self->read_pin($rid);
+
+  return $pin;
+}
+
+#######################################
+# Matches pin
+#######################################
+sub match {
+  my ($self, $key, $pin) = @_;
+  my $dbh = $self->{dbh};
+  my $select = "select * from pins " .
+                    "where " .
+                    "key=" . $dbh->quote($key) . " AND " .
+                    "pin=" . $dbh->quote($pin);
+  my $sth = $dbh->prepare($select);
+  $sth->execute();
+  my $ref = $sth->fetchrow_hashref();
+  $sth->finish();
+  if (defined($ref)) {
+    return 1;
+  } else {
+    return 0;
+  }
+}
+
+sub read_pin {
+  my ($self, $key) = @_;
+  my $dbh = $self->{dbh};
+  my $select = "select * from pins " .
+                    "where " .
+                    "key=" . $dbh->quote($key);
+  my $sth = $dbh->prepare($select);
+  $sth->execute();
+  my $ref = $sth->fetchrow_hashref();
+  $sth->finish();
+  return $ref;
+}
+
+#######################################
+# Deletes pin
+#######################################
+sub delete {
+  my ($self, $key) = @_;
+  my $dbh = $self->{dbh};
+  my $cmd = "delete from pins " .
+                    "where " .
+                    "key=" . $dbh->quote($key);
+  $dbh->do($cmd);
+}
+
+#######################################
+# Closes request queue
+#######################################
+sub close {
+  my ($self) = @_;
+  my $dbh = $self->{dbh};
+  $dbh->disconnect();
+}
+
+1;