Make tests public. Rewrite most tests. Add tests.
[pintos-anon] / src / tests / arc4.pm
1 use strict;
2 use warnings;
3
4 sub arc4_init {
5     my ($key) = @_;
6     my (@s) = 0...255;
7     my ($j) = 0;
8     for my $i (0...255) {
9         $j = ($j + $s[$i] + ord (substr ($key, $i % length ($key), 1))) & 0xff;
10         @s[$i, $j] = @s[$j, $i];
11     }
12     return (0, 0, @s);
13 }
14
15 sub arc4_crypt {
16     my ($arc4, $buf) = @_;
17     my ($i, $j, @s) = @$arc4;
18     my ($out) = "";
19     for my $c (split (//, $buf)) {
20         $i = ($i + 1) & 0xff;
21         $j = ($j + $s[$i]) & 0xff;
22         @s[$i, $j] = @s[$j, $i];
23         $out .= chr (ord ($c) ^ $s[($s[$i] + $s[$j]) & 0xff]);
24     }
25     @$arc4 = ($i, $j, @s);
26     return $out;
27 }
28
29 1;