added export method to PublicKey

Author: mlarue

lib/Crypt/Cryptoki/Object.pm

@@ -2,19 +2,56 @@ package Crypt::Cryptoki::Object;
 use strict;
 use Moo;
 use Carp;
+use List::MoreUtils qw(zip);
 
 use Crypt::Cryptoki::Raw qw(rv_to_str CKR_OK);
 
 has 'session' => ( is => 'ro', required => 1 );
 has 'id' => ( is => 'ro', required => 1 );
+has 'attributes' => ( is => 'lazy' );
+
+sub _attribute_map {{
+}}
+
+sub _build_attributes {
+	my ( $self ) = @_;
+	my @attr_names = keys %{$self->_attribute_map};
+	my @attrs = $self->get_attributes(@attr_names);
+	return { zip @attr_names, @attrs };
+}
+
+sub hex_attributes {
+	my ( $self ) = @_;
+	my $attrs = $self->attributes;
+	+{ map { $_ => unpack('H*',$attrs->{$_}) } keys %$attrs };
+}
 
 sub destroy {
-        my ( $self ) = @_;
-        my $rv = $self->session->slot->ctx->_fl->C_DestroyObject($self->session->id,$self->id);
-        if ( $rv != CKR_OK ) {
-                croak rv_to_str($rv);
-        }
-        return 1;
+	my ( $self ) = @_;
+	my $rv = $self->session->slot->ctx->_fl->C_DestroyObject($self->session->id,$self->id);
+	if ( $rv != CKR_OK ) {
+		croak rv_to_str($rv);
+	}
+	return 1;
+}
+
+sub get_attributes {
+	my ( $self, @attributes ) = @_;
+
+	my @get_attributes_template;
+	for (@attributes) {
+		exists $self->_attribute_map->{$_} or croak 'illegal attribute';
+		push @get_attributes_template, [ $self->_attribute_map->{$_}, '' ];
+	}
+
+	my $rv = $self->session->slot->ctx->_fl->C_GetAttributeValue(
+		$self->session->id,$self->id,\@get_attributes_template
+	);
+	if ( $rv != CKR_OK ) {
+		croak rv_to_str($rv);
+	}
+
+	return map { ''.$_->[1] } @get_attributes_template;
 }
 
 1;

lib/Crypt/Cryptoki/PublicKey.pm

@@ -4,35 +4,55 @@ use Moo;
 extends 'Crypt::Cryptoki::Key';
 use Carp;
 
-use Crypt::Cryptoki::Raw qw(rv_to_str CKR_OK CKM_RSA_PKCS NULL_PTR);
+use Crypt::Cryptoki::Raw qw(rv_to_str CKR_OK NULL_PTR
+	CKM_RSA_PKCS 
+	CKA_MODULUS CKA_PUBLIC_EXPONENT
+);
+
+sub _attribute_map {{
+	modulus 		=> CKA_MODULUS,
+	public_exponent => CKA_PUBLIC_EXPONENT
+}};
+
+sub export_as_string {
+	my ( $self ) = @_;
+	my $attrs = $self->attributes;
+	require Crypt::OpenSSL::Bignum;
+	require Crypt::OpenSSL::RSA;
+	my $n = Crypt::OpenSSL::Bignum->new_from_bin($attrs->{modulus});
+	my $e = Crypt::OpenSSL::Bignum->new_from_bin($attrs->{public_exponent});
+	my $rsa_pub = Crypt::OpenSSL::RSA->new_key_from_parameters($n,$e);
+	$rsa_pub->use_pkcs1_padding;
+	$rsa_pub->get_public_key_string;
+}
 
 sub encrypt {
-        my ( $self, $plain_text_ref, $plain_text_len ) = @_;
+	my ( $self, $plain_text_ref, $plain_text_len ) = @_;
 
-        my $rv = $self->session->slot->ctx->_fl->C_EncryptInit(
+	my $rv = $self->session->slot->ctx->_fl->C_EncryptInit(
 		$self->session->id, 
 		[ CKM_RSA_PKCS, NULL_PTR, 0 ], 
 		$self->id
 	);
-        if ( $rv != CKR_OK ) {
-                croak rv_to_str($rv);
-        }
+	if ( $rv != CKR_OK ) {
+		croak rv_to_str($rv);
+	}
 
 	my $encrypted_text = '';
 	my $encrypted_text_len = -1;
 
-        $rv = $self->session->slot->ctx->_fl->C_Encrypt(
+	$rv = $self->session->slot->ctx->_fl->C_Encrypt(
 		$self->session->id, 
 		$$plain_text_ref,
 		$plain_text_len,
 		$encrypted_text,
 		$encrypted_text_len
 	);
-        if ( $rv != CKR_OK ) {
-                croak rv_to_str($rv);
-        }
+	if ( $rv != CKR_OK ) {
+		croak rv_to_str($rv);
+	}
 
-        return ( \$encrypted_text, $encrypted_text_len );
+	return ( \$encrypted_text, $encrypted_text_len );
 }
 
 1;

t/cryptoki.t

@@ -2,6 +2,7 @@ use strict;
 use warnings;
 
 use Test::More;
+use Try::Tiny;
 
 use_ok 'Crypt::Cryptoki';
 use_ok 'Crypt::Cryptoki::Template';
@@ -52,13 +53,18 @@ my $t_private = Crypt::Cryptoki::Template->new(
 
 my ( $public_key, $private_key ) = $session->generate_key_pair($t_public,$t_private);
 
-my $plain_text = 'plain text';
-my ( $encrypted_text_ref, $len ) = $public_key->encrypt(\$plain_text, length($plain_text));
+try {
+	my $plain_text = 'plain text';
+	my ( $encrypted_text_ref, $len ) = $public_key->encrypt(\$plain_text, length($plain_text));
 
-my ( $encrypted_text_ref ) = $private_key->decrypt($encrypted_text_ref, $len);
-diag $$encrypted_text_ref;
+	my ( $encrypted_text_ref ) = $private_key->decrypt($encrypted_text_ref, $len);
+	diag $$encrypted_text_ref;
 
-$public_key->destroy;
-$private_key->destroy;
+	diag explain $public_key->hex_attributes;
+	diag $public_key->export_as_string;
+} catch { diag $_; 0 };
+
+ok $public_key->destroy;
+ok $private_key->destroy;
 
 done_testing();