1/*
2 * Copyright (c) 2011, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23
24/**
25 * @test
26 * @bug 7031343
27 * @summary Provide API changes to support GCM AEAD ciphers
28 * @author Brad Wetmore
29 */
30
31import javax.crypto.AEADBadTagException;
32import javax.crypto.spec.GCMParameterSpec;
33import java.util.Arrays;
34
35public class GCMParameterSpecTest {
36
37    // 16 elements
38    private static byte[] bytes = new byte[] {
39        0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17,
40        0x18, 0x19, 0x1a, 0x1b, 0x1c, 0x1d, 0x1e, 0x1f };
41
42    private static int failed = 0;
43
44    public static void main(String[] args) throws Exception {
45        newGCMParameterSpecFail(-1, bytes);
46        newGCMParameterSpecFail(128, null);
47        newGCMParameterSpecPass(128, bytes);
48
49        newGCMParameterSpecFail(-1, bytes, 2, 4);
50        newGCMParameterSpecFail(128, null, 2, 4);
51        newGCMParameterSpecFail(128, bytes, -2, 4);
52        newGCMParameterSpecFail(128, bytes, 2, -4);
53        newGCMParameterSpecFail(128, bytes, 2, 15);  // one too many
54
55        newGCMParameterSpecPass(128, bytes, 2, 14);  // ok.
56        newGCMParameterSpecPass(96, bytes, 4, 4);
57        newGCMParameterSpecPass(96, bytes, 0, 0);
58
59        // Might as well check the Exception constructors.
60        try {
61            new AEADBadTagException();
62            new AEADBadTagException("Bad Tag Seen");
63        } catch (Exception e) {
64            e.printStackTrace();
65            failed++;
66        }
67
68        if (failed != 0) {
69            throw new Exception("Test(s) failed");
70        }
71    }
72
73    private static void newGCMParameterSpecPass(
74            int tLen, byte[] src) {
75        try {
76            GCMParameterSpec gcmps = new GCMParameterSpec(tLen, src);
77            if (gcmps.getTLen() != tLen) {
78                throw new Exception("tLen's not equal");
79            }
80            if (!Arrays.equals(gcmps.getIV(), src)) {
81                throw new Exception("IV's not equal");
82            }
83        } catch (Exception e) {
84            e.printStackTrace();
85            failed++;
86        }
87    }
88
89    private static void newGCMParameterSpecFail(
90            int tLen, byte[] src) {
91        try {
92            new GCMParameterSpec(tLen, src);
93            new Exception("Didn't Fail as Expected").printStackTrace();
94            failed++;
95        } catch (IllegalArgumentException e) {
96            // swallow
97        }
98    }
99
100    private static void newGCMParameterSpecPass(
101            int tLen, byte[] src, int offset, int len) {
102        try {
103            GCMParameterSpec gcmps =
104                new GCMParameterSpec(tLen, src, offset, len);
105            if (gcmps.getTLen() != tLen) {
106                throw new Exception("tLen's not equal");
107            }
108            if (!Arrays.equals(gcmps.getIV(),
109                    Arrays.copyOfRange(src, offset, offset + len))) {
110                System.out.println(offset + " " + len);
111                System.out.println(Arrays.copyOfRange(src, offset, len)[0]);
112                throw new Exception("IV's not equal");
113            }
114        } catch (Exception e) {
115            e.printStackTrace();
116            failed++;
117        }
118    }
119
120    private static void newGCMParameterSpecFail(
121            int tLen, byte[] src, int offset, int len) {
122        try {
123            new GCMParameterSpec(tLen, src, offset, len);
124            new Exception("Didn't Fail as Expected").printStackTrace();
125            failed++;
126        } catch (IllegalArgumentException e) {
127            // swallow
128        }
129    }
130}
131