summaryrefslogtreecommitdiff
path: root/015/ch2.pl
blob: b43e6bf1cd7c189a8ce698ec1b40cb627ae54a21 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
#!/usr/bin/env perl
#
# Write a script to implement Vigenère cipher.
# (https://en.wikipedia.org/wiki/Vigen%C3%A8re_cipher)
################################################################################

use strict;
use warnings;

sub vigenere {
    # Direction: encrypt (1) vs decrypt (-1)
    my ($direction, $text, $key) = @_;

    chomp $text;

    my ($A, $Z) = (ord("a"), ord("z"));
    my $OFFSET = $Z - $A + 1;

    foreach my $i (0 .. length($text)-1) {
        # Only process alpha characters (gets lowered before)
        next unless substr($text, $i, 1) =~ /^[a-zA-Z]$/;

        # Add/subtract the key character to the corresponding one in text
        my $ord = ord(lc(substr($text, $i, 1))) +
            $direction * (ord(substr($key, $i % length($key), 1)) - $A);

        # Adjust if character got out of alpha range
        $ord -= $OFFSET if $ord > $Z;
        $ord += $OFFSET if $ord < $A;

        substr($text, $i, 1) = chr($ord);
    }
    return $text;
}

sub encrypt { vigenere(1, @_); }
sub decrypt { vigenere(-1, @_); }

my $USAGE = "Usage: $0 (enc|dec) key\n";
my $MODE = shift or die $USAGE;
my $KEY = shift or die $USAGE;

my $f;
if ($MODE eq "enc") {
    $f = \&encrypt;
} elsif ($MODE eq "dec") {
    $f = \&decrypt;
} else {
    die $USAGE;
}

while (<STDIN>) {
    print $f->($_, $KEY), "\n";
}