File Coverage

blib/lib/Crypt/OFB.pm

Criterion	Covered	Total	%
statement	9	62	14.5
branch	0	26	0.0
condition	0	12	0.0
subroutine	3	9	33.3
pod	5	6	83.3
total	17	115	14.7

package Crypt::OFB;

########################################
# general module startup things
########################################

use strict;
use vars qw(@ISA @EXPORT @EXPORT_OK $VERSION);

require Exporter;

use Crypt::ECB;

@ISA=qw(Crypt::ECB);

@EXPORT_OK = qw(encrypt decrypt encrypt_hex decrypt_hex);

$VERSION   = 0.01;

########################################
# basic methods
########################################

#
# sets iv if given
# returns iv
#
sub iv (\$;$)
{
    my $crypt = shift;
    $crypt->{Iv} = shift if @_;
    return $crypt->{Iv};
}

#
# calls the crypting module
# returns the en-/decrypted data
#
sub crypt (\$;$)
{
    my $crypt = shift;
    my $data  = shift || $_ || '';

    my $errmsg = $crypt->{Errstring};
    my $bs     = $crypt->{Blocksize};
    my $mode   = $crypt->{Mode};

    die $errmsg if $errmsg;

    unless ($mode)
    {
	die "You tried to use crypt() without calling start()"
	  . " before. Use '\$your_obj->start(\$mode)' first,"
	  . " \$mode being one of 'decrypt' or 'encrypt'.\n";
    }

    $data = $crypt->{buffer}.$data;

    # data is split into blocks of proper size which is reported
    # by the cipher module
    my @blocks = $data=~/(.{1,$bs})/gs;

    $crypt->{buffer} = pop @blocks;

    my $cipher = $crypt->_getcipher;
    my $text = '';

    # OFB Implementation here.
    my $skey = $crypt->iv;
    foreach my $block (@blocks) {
	$skey  = $cipher->encrypt($skey);
	$text .= $block ^ $skey;
    }
    #
    return $text;
}

########################################
# convenience functions/methods
########################################

#
# magic convenience encrypt function/method
#
sub encrypt ($$;$$)
{
    if (ref($_[0]) =~ /^Crypt/)
    {
	my $crypt = shift;

	$crypt->start('encrypt') || die $crypt->errstring;

	my $text = $crypt->crypt(shift)
	         . $crypt->finish;

	return $text;
    }
    else
    {
	my ($key, $cipher, $data, $padstyle) = @_;

	my $crypt = Crypt::OFB->new($key);

	$crypt->padding($padstyle || 0);
	$crypt->cipher($cipher)  || die $crypt->errstring;
	$crypt->start('encrypt') || die $crypt->errstring;

	my $text = $crypt->crypt($data || $_)
	         . $crypt->finish;

	return $text;
    }
}

#
# magic convenience decrypt function/method
#
sub decrypt ($$;$$)
{
    if (ref($_[0]) =~ /^Crypt/)
    {
	my $crypt = shift;

	$crypt->start('decrypt') || die $crypt->errstring;

	my $text = $crypt->crypt(shift)
	         . $crypt->finish;

	return $text;
    }
    else
    {
	my ($key, $cipher, $data, $padstyle) = @_;

	my $crypt = Crypt::OFB->new($key);

	$crypt->padding($padstyle || 0);
	$crypt->cipher($cipher)  || die $crypt->errstring;
	$crypt->start('decrypt') || die $crypt->errstring;

	my $text = $crypt->crypt($data || $_)
	         . $crypt->finish;

	return $text;
    }
}

#
# calls encrypt, returns hex packed data
#
sub encrypt_hex ($$;$$)
{
    if (ref($_[0]) =~ /^Crypt/)
    {
	my $crypt = shift;
	join('',unpack('H*',$crypt->encrypt(shift)));
    }
    else
    {
	join('',unpack('H*',encrypt($_[0], $_[1], $_[2], $_[3])));
    }
}

#
# calls decrypt, expected input is hex packed
#
sub decrypt_hex ($$;$$)
{
    if (ref($_[0]) =~ /^Crypt/)
    {
	my $crypt = shift;
	$crypt->decrypt(pack('H*',shift));
    }
    else
    {
	decrypt($_[0], $_[1], pack('H*',$_[2]), $_[3]);
    }
}


########################################
# finally, to satisfy require
########################################

1;
__END__