#! /usr/bin/env perl

# Copyright 2019 The OpenSSL Project Authors. All Rights Reserved.

#

# Licensed under the Apache License 2.0 (the "License").  You may not use

# this file except in compliance with the License.  You can obtain a copy

# in the file LICENSE in the source distribution or at

# https://www.openssl.org/source/license.html

#Convert CCM CAVS test vectors to a format suitable for evp_test

use strict;

use warnings;

my $alg;

my $mode;

my $keylen;

my $key = "";

my $iv = "";

my $aad = "";

my $ct = "";

my $pt = "";

my $tag = "";

my $aadlen = 0;

my $ptlen = 0;

my $taglen = 0;

my $res = "";

my $intest = 0;

my $fixediv = 0;

while (<STDIN>)

{

    chomp;

    # Pull out the cipher mode from the comment at the beginning of the file

    if(/^#\s*"([^-]+)-\w+" information/) {

        $mode = lc($1);

    # Pull out the key length from the comment at the beginning of the file

    } elsif(/^#\s*(\w+) Keylen: (\d+)/) {

        $alg = lc($1);

        $keylen = $2;

    # Some parameters common to many tests appear as a list in square brackets

    # so parse these

    } elsif(/\[(.*)\]/) {

        my @pairs = split(/, /, $1);

        foreach my $pair (@pairs) {

            $pair =~ /(\w+)\s*=\s*(\d+)/;

            # AAD Length

            if ($1 eq "Alen") {

                $aadlen = $2;

            # Plaintext length

            } elsif ($1 eq "Plen") {

                $ptlen = $2;

            # Tag length

            } elsif ($1 eq "Tlen") {

                $taglen = $2;

            }

        }

    # Key/Value pair

    } elsif (/^\s*(\w+)\s*=\s*(\S.*)\r/) {

        if ($1 eq "Key") {

            $key = $2;

        } elsif ($1 eq "Nonce") {

            $iv = $2;

            if ($intest == 0) {

                $fixediv = 1;

            } else {

                $fixediv = 0;

            }

        } elsif ($1 eq "Adata") {

            $aad = $2;

        } elsif ($1 eq "CT") {

            $ct = substr($2, 0, length($2) - ($taglen * 2));

            $tag = substr($2, $taglen * -2);

        } elsif ($1 eq "Payload") {

            $pt = $2;

        } elsif ($1 eq "Result") {

            if ($2 =~ /Fail/) {

                $res = "CIPHERUPDATE_ERROR";

            }

        } elsif ($1 eq "Count") {

            $intest = 1;

        } elsif ($1 eq "Plen") {

            $ptlen = $2;

        } elsif ($1 eq "Tlen") {

            $taglen = $2;

        } elsif ($1 eq "Alen") {

            $aadlen = $2;

        }

    # Something else - probably just a blank line

    } elsif ($intest) {

        print "Cipher = $alg-$keylen-$mode\n";

        print "Key = $key\n";

        print "IV = $iv\n";

        print "AAD =";

        if ($aadlen > 0) {

            print " $aad";

        }

        print "\nTag =";

        if ($taglen > 0) {

            print " $tag";

        }

        print "\nPlaintext =";

        if ($ptlen > 0) {

            print " $pt";

        }

        print "\nCiphertext = $ct\n";

        if ($res ne "") {

            print "Operation = DECRYPT\n";

            print "Result = $res\n";

        }

        print "\n";

        $res = "";

        if ($fixediv == 0) {

            $iv = "";

        }

        $aad = "";

        $tag = "";

        $pt = "";

        $intest = 0;

    }

}