#!perl -w

use Test::More tests => 12;

{
   package LenDigest;
   require Digest::base;
   use vars qw(@ISA);
   @ISA = qw(Digest::base);

   sub new {
	my $class = shift;
	my $str = "";
	bless \$str, $class;
   }

   sub add {
	my $self = shift;
	$$self .= join("", @_);
	return $self;
   }

   sub digest {
	my $self = shift;
	my $len = length($$self);
	my $first = ($len > 0) ? substr($$self, 0, 1) : "X";
	$$self = "";
	return sprintf "$first%04d", $len;
   }
}

my $ctx = LenDigest->new;
is($ctx->digest, "X0000");

my $EBCDIC = ord('A') == 193;

if ($EBCDIC) {
    is($ctx->hexdigest, "e7f0f0f0f0");
    is($ctx->b64digest, "5/Dw8PA");
} else {
    is($ctx->hexdigest, "5830303030");
    is($ctx->b64digest, "WDAwMDA");
}

$ctx->add("foo");
is($ctx->digest, "f0003");

$ctx->add("foo");
is($ctx->hexdigest, $EBCDIC ? "86f0f0f0f3" : "6630303033");

$ctx->add("foo");
is($ctx->b64digest, $EBCDIC ? "hvDw8PM" : "ZjAwMDM");

open(F, ">xxtest$$") || die;
binmode(F);
print F "abc" x 100, "\n";
close(F) || die;

open(F, "xxtest$$") || die;
$ctx->addfile(*F);
close(F);
unlink("xxtest$$") || warn;

is($ctx->digest, "a0301");

eval {
    $ctx->add_bits("1010");
};
like($@, '/^Number of bits must be multiple of 8/');

$ctx->add_bits($EBCDIC ? "11100100" : "01010101");
is($ctx->digest, "U0001");

eval {
    $ctx->add_bits("abc", 12);
};
like($@, '/^Number of bits must be multiple of 8/');

$ctx->add_bits("abc", 16);
is($ctx->digest, "a0002");

$ctx->add_bits("abc", 32);
is($ctx->digest, "a0003");