/*
 * Copyright 2003-2007 Sun Microsystems, Inc.  All Rights Reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
 * CA 95054 USA or visit www.sun.com if you need additional information or
 * have any questions.
 */

/**
 * @test
 * @bug 4794996
 * @summary Ensure basic functionality of these new ECC classes.
 * @author Valerie Peng
 */
import java.math.BigInteger;
import java.util.Arrays;
import java.security.*;
import java.security.spec.*;

public class ECCBasic {
    private static final BigInteger ZERO = BigInteger.ZERO;
    private static final BigInteger ONE = BigInteger.ONE;
    private static final BigInteger TEN = BigInteger.TEN;
    private static final ECFieldFp FP = new ECFieldFp(TEN);
    private static final int F2M_M = 4;
    private static final int[] F2M_KS;
    static {
        F2M_KS = new int[1];
        F2M_KS[0] = 1;
    }
    private static final BigInteger F2M_RP = BigInteger.valueOf(19);
    private static final ECFieldF2m F2M = new ECFieldF2m(F2M_M, F2M_KS);
    private static final EllipticCurve CURVE = new EllipticCurve
        (new ECFieldFp(TEN), ONE, ONE);
    private static final ECPoint POINT = new ECPoint(ONE, TEN);
    private static final BigInteger ORDER = ONE;
    private static final int COFACTOR = 3;
    private static final ECParameterSpec PARAMS = new ECParameterSpec
        (CURVE, POINT, ORDER, COFACTOR);
    private static final ECGenParameterSpec GENPARAMS =
        new ECGenParameterSpec("prime192v1");
    private static final ECPrivateKeySpec PRIV_KEY = new ECPrivateKeySpec
        (TEN, PARAMS);
    private static final ECPublicKeySpec PUB_KEY = new ECPublicKeySpec
        (POINT, PARAMS);

    private static void testECFieldFp() throws Exception {
        System.out.println("Testing ECFieldFp(BigInteger)");
        try {
            new ECFieldFp(ZERO);
            throw new Exception("...should throw IAE");
        } catch (IllegalArgumentException iae) {
            System.out.println("...expected IAE thrown");
        }
        try {
            new ECFieldFp(null);
            throw new Exception("...should throw NPE");
        } catch (NullPointerException npe) {
            System.out.println("...expected NPE thrown");
        }
        if (TEN.equals(FP.getP()) == false) {
            throw new Exception("...error in getP()");
        }
        if (FP.getFieldSize() != TEN.bitLength()) {
            throw new Exception("...error in getFieldSize()");
        }
    }
    private static void testECFieldF2m() throws Exception {
        // invalid parameters
        int mBad = 0;
        int[] ksBad;
        for (int i=0; i<7; i++) {
            System.out.print("Testing ECFieldF2m");
            try {
                switch (i) {
                case 0:
                    System.out.println("(int)");
                    new ECFieldF2m(mBad);
                    break;
                case 1:
                    System.out.println("(int, BigInteger)#1");
                    new ECFieldF2m(mBad, F2M_RP);
                    break;
                case 2:
                    System.out.println("(int, BigInteger)#2");
                    new ECFieldF2m(mBad, F2M_RP);
                    break;
                case 3:
                    System.out.println("(int, int[])#1");
                    new ECFieldF2m(mBad, F2M_KS);
                    break;
                case 4:
                    ksBad = new int[2];
                    System.out.println("(int, int[])#2");
                    new ECFieldF2m(F2M_M, ksBad);
                    break;
                case 5:
                    ksBad = new int[1];
                    ksBad[0] = 5;
                    System.out.println("(int, int[])#3");
                    new ECFieldF2m(F2M_M, ksBad);
                    break;
                case 6:
                    ksBad = new int[3];
                    ksBad[0] = 1;
                    ksBad[1] = 2;
                    ksBad[2] = 3;
                    System.out.println("(int, int[])#4");
                    new ECFieldF2m(F2M_M, ksBad);
                    break;
                }
                throw new Exception("...should throw IAE");
            } catch (IllegalArgumentException iae) {
                System.out.println("...expected IAE thrown");
            }
        }
        for (int i=0; i<2; i++) {
            System.out.print("Testing ECFieldF2m");
            try {
                switch (i) {
                case 0:
                    System.out.println("(int, BigInteger)");
                    new ECFieldF2m(F2M_M, (BigInteger) null);
                    break;
                case 1:
                    System.out.println("(int, int[])#1");
                    new ECFieldF2m(F2M_M, (int[]) null);
                    break;
                }
                throw new Exception("...should throw NPE");
            } catch (NullPointerException npe) {
                System.out.println("...expected NPE thrown");
            }
        }
        if (F2M_RP.compareTo(F2M.getReductionPolynomial()) != 0) {
            throw new Exception("...error in getReductionPolynomial()");
        }
        ECFieldF2m field = new ECFieldF2m(F2M_M, F2M_RP);
        if (!(Arrays.equals(F2M_KS,
                            field.getMidTermsOfReductionPolynomial()))) {
            throw new Exception("...error in getMidTermsOfReductionPolynomial()");
        }
        if (field.getFieldSize() != F2M_M) {
            throw new Exception("...error in getFieldSize()");
        }
    }
    private static void testECParameterSpec() throws Exception {
        System.out.println("Testing ECParameterSpec(...)");
        for (int i = 0; i < 2; i++) {
            try {
                switch (i) {
                case 0:
                    System.out.println("with zero order");
                    new ECParameterSpec(CURVE, POINT, ZERO, COFACTOR);
                    break;
                case 1:
                    System.out.println("with negative cofactor");
                    new ECParameterSpec(CURVE, POINT, ORDER, -1);
                    break;
                }
                throw new Exception("...should throw IAE");
            } catch (IllegalArgumentException iae) {
                System.out.println("...expected IAE thrown");
            }
        }
        for (int i = 0; i < 3; i++) {
            try {
                switch (i) {
                case 0:
                    System.out.println("with null curve");
                    new ECParameterSpec(null, POINT, ORDER, COFACTOR);
                    break;
                case 1:
                    System.out.println("with null generator");
                    new ECParameterSpec(CURVE, null, ORDER, COFACTOR);
                    break;
                case 2:
                    System.out.println("with null order");
                    new ECParameterSpec(CURVE, POINT, null, COFACTOR);
                    break;
                }
                throw new Exception("...should throw NPE");
            } catch (NullPointerException npe) {
                System.out.println("...expected NPE thrown");
            }
        }
        if (!(CURVE.equals(PARAMS.getCurve()))) {
            throw new Exception("...error in getCurve()");
        }
        if (!(POINT.equals(PARAMS.getGenerator()))) {
            throw new Exception("...error in getGenerator()");
        }
        if (!(ORDER.equals(PARAMS.getOrder()))) {
            throw new Exception("...error in getOrder()");
        }
        if (COFACTOR != PARAMS.getCofactor()) {
            throw new Exception("...error in getCofactor()");
        }
    }
    private static void testECGenParameterSpec() throws Exception {
        System.out.println("Testing ECGenParameterSpec(String)");
        try {
            new ECGenParameterSpec(null);
            throw new Exception("...should throw NPE");
        } catch (NullPointerException npe) {
            System.out.println("...expected NPE thrown");
        }
        if (!("prime192v1".equals(GENPARAMS.getName()))) {
            throw new Exception("...error in getName()");
        }
    }
    private static void testECPoint() throws Exception {
        System.out.println("Testing ECParameterSpec(...)");
        for (int i = 0; i < 2; i++) {
            try {
                switch (i) {
                case 0:
                    System.out.println("with null x-coordinate");
                    new ECPoint(null, TEN);
                    break;
                case 1:
                    System.out.println("with null y-coordinate");
                    new ECPoint(ONE, null);
                    break;
                }
                throw new Exception("...should throw NPE");
            } catch (NullPointerException npe) {
                System.out.println("...expected NPE thrown");
            }
        }
        if (!(ONE.equals(POINT.getAffineX()))) {
            throw new Exception("...error in getAffineX()");
        }
        if (!(TEN.equals(POINT.getAffineY()))) {
            throw new Exception("...error in getAffineY()");
        }
    }
    private static void testECPublicKeySpec() throws Exception {
        System.out.println("Testing ECPublicKeySpec(...)");
        for (int i = 0; i < 2; i++) {
            try {
                switch (i) {
                case 0:
                    System.out.println("with null public value");
                    new ECPublicKeySpec(null, PARAMS);
                    break;
                case 1:
                    System.out.println("with null params");
                    new ECPublicKeySpec(POINT, null);
                    break;
                }
                throw new Exception("...should throw NPE");
            } catch (NullPointerException npe) {
                System.out.println("...expected NPE thrown");
            }
        }
        if (!(POINT.equals(PUB_KEY.getW()))) {
            throw new Exception("...error in getW()");
        }
        if (!(PARAMS.equals(PUB_KEY.getParams()))) {
            throw new Exception("...error in getParams()");
        }
    }
    private static void testECPrivateKeySpec() throws Exception {
        System.out.println("Testing ECPrivateKeySpec(...)");
        for (int i = 0; i < 2; i++) {
            try {
                switch (i) {
                case 0:
                    System.out.println("with null private value");
                    new ECPrivateKeySpec(null, PARAMS);
                    break;
                case 1:
                    System.out.println("with null param");
                    new ECPrivateKeySpec(ONE, null);
                    break;
                }
                throw new Exception("...should throw NPE");
            } catch (NullPointerException npe) {
                System.out.println("...expected NPE thrown");
            }
        }
        if (!(TEN.equals(PRIV_KEY.getS()))) {
            throw new Exception("...error in getS()");
        }
        if (!(PARAMS.equals(PRIV_KEY.getParams()))) {
            throw new Exception("...error in getParams()");
        }
    }
    private static void testEllipticCurve() throws Exception {
        System.out.println("Testing EllipticCurve(...)");
        for (int j = 0; j < 2; j++) {
            BigInteger a = ONE;
            BigInteger b = ONE;
            if (j == 0) { // test a out of range
                System.out.println("with a's value out of range");
                a = BigInteger.valueOf(20);
            } else { // test b out of range
                System.out.println("with b's value out of range");
                b = BigInteger.valueOf(20);
            }
            System.out.println(">> over ECFieldFp");
            for (int i = 0; i < 3; i++) {
                try {
                    switch (i) {
                    case 0:
                        new EllipticCurve(FP, a, b);
                        break;
                    case 1:
                        new EllipticCurve(FP, a, b, null);
                        break;
                    case 2:
                        new EllipticCurve(FP, a, b, new byte[8]);
                        break;
                    }
                    throw new Exception("...should throw IAE");
                } catch (IllegalArgumentException iae) {
                    System.out.println("...expected IAE thrown for #" + (i+1));
                }
            }
            System.out.println(">> over ECFieldF2m");
            for (int i = 0; i < 3; i++) {
                try {
                    switch (i) {
                    case 0:
                        new EllipticCurve(F2M, a, b);
                        break;
                    case 1:
                        new EllipticCurve(F2M, a, b, null);
                        break;
                    case 2:
                        new EllipticCurve(F2M, a, b, new byte[8]);
                        break;
                    }
                    throw new Exception("...should throw IAE");
                } catch (IllegalArgumentException iae) {
                    System.out.println("...expected IAE thrown for #" + (i+1));
                }
            }
        }
        for (int i = 0; i < 6; i++) {
            try {
                switch (i) {
                case 0:
                    new EllipticCurve(null, ONE, TEN);
                    break;
                case 1:
                    new EllipticCurve(FP, null, TEN);
                    break;
                case 2:
                    new EllipticCurve(FP, ONE, null);
                    break;
                case 3:
                    new EllipticCurve(null, ONE, TEN, null);
                    break;
                case 4:
                    new EllipticCurve(FP, null, TEN, null);
                    break;
                case 5:
                    new EllipticCurve(FP, ONE, null, null);
                    break;
                }
                throw new Exception("...should throw NPE");
            } catch (NullPointerException npe) {
                System.out.println("...expected NPE thrown");
            }
        }
        if (!(FP.equals(CURVE.getField()))) {
            throw new Exception("...error in getField()");
        }
        if (!(ONE.equals(CURVE.getA()))) {
            throw new Exception("...error in getA()");
        }
        if (!(ONE.equals(CURVE.getB()))) {
            throw new Exception("...error in getB()");
        }
        if (!(CURVE.equals(new EllipticCurve(FP, ONE, ONE, null)))) {
            throw new Exception("...error in equals()");
        }
    }
    private static void testAllHashCode() throws Exception {
        // make sure no unexpected exception when hashCode() is called.
        FP.hashCode();
        F2M.hashCode();
        GENPARAMS.hashCode();
        PARAMS.hashCode();
        POINT.hashCode();
        PRIV_KEY.hashCode();
        PUB_KEY.hashCode();
        CURVE.hashCode();
    }
    public static void main(String[] argv) throws Exception {
        testECFieldFp();
        testECFieldF2m();
        testECParameterSpec();
        testECGenParameterSpec();
        testECPoint();
        testECPrivateKeySpec();
        testECPublicKeySpec();
        testEllipticCurve();
        testAllHashCode();
        System.out.println("All Test Passed");
    }
}