* Copyright (c) 2005, 2016, Oracle and/or its affiliates. All rights reserved.

 * @run main/othervm TestPRF

 * @run main/othervm TestPRF sm policy

import java.io.BufferedReader;

import java.nio.file.Files;

import java.nio.file.Paths;

import java.util.Arrays;

import javax.crypto.spec.SecretKeySpec;

import sun.security.internal.spec.TlsPrfParameterSpec;

    private static final int PREFIX_LENGTH = "prf-output: ".length();

        main(new TestPRF(), args);

    @Override

        try (BufferedReader reader = Files.newBufferedReader(

                Paths.get(BASE, "prfdata.txt"))) {

            int n = 0;

            int lineNumber = 0;

            byte[] secret = null;

            String label = null;

            byte[] seed = null;

            int length = 0;

            byte[] output = null;

            while (true) {

                String line = reader.readLine();

                lineNumber++;

                if (line == null) {

                    break;

                }

                if (line.startsWith("prf-") == false) {

                    continue;

                }

                String data = line.substring(PREFIX_LENGTH);

                if (line.startsWith("prf-secret:")) {

                    secret = parse(data);

                } else if (line.startsWith("prf-label:")) {

                    label = data;

                } else if (line.startsWith("prf-seed:")) {

                    seed = parse(data);

                } else if (line.startsWith("prf-length:")) {

                    length = Integer.parseInt(data);

                } else if (line.startsWith("prf-output:")) {

                    output = parse(data);

                    System.out.print(".");

                    n++;

                    KeyGenerator kg =

                        KeyGenerator.getInstance("SunTlsPrf", provider);

                    SecretKey inKey;

                        inKey = null;

                    } else {

                        inKey = new SecretKeySpec(secret, "Generic");

                    TlsPrfParameterSpec spec =

                        new TlsPrfParameterSpec(inKey, label, seed, length,

                            null, -1, -1);

                    SecretKey key;

                    try {

                        kg.init(spec);

                        key = kg.generateKey();

                    } catch (Exception e) {

                        if (secret == null) {

                            // This fails on Solaris, but since we never call this

                            // API for this case in JSSE, ignore the failure.

                            // (SunJSSE uses the CKM_TLS_KEY_AND_MAC_DERIVE

                            // mechanism)

                            System.out.print("X");

                            continue;

                        }

                        System.out.println();

                        throw new Exception("Error on line: " + lineNumber, e);

                    }

                    byte[] enc = key.getEncoded();

                    if (Arrays.equals(output, enc) == false) {

                        System.out.println();

                        System.out.println("expected: " + toString(output));

                        System.out.println("actual:   " + toString(enc));

                        throw new Exception("mismatch line: " + lineNumber);

                    }

                } else {

                    throw new Exception("Unknown line: " + line);

            }

            if (n == 0) {

                throw new Exception("no tests");

            System.out.println();

            System.out.println("OK: " + n + " tests");