1 /*

     2  * Copyright (c) 2015, 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 import java.io.Serializable;

    25 import java.security.KeyPair;

    26 import java.security.KeyPairGenerator;

    27 import java.security.Signature;

    28 import java.security.SignedObject;

    29 

    30 /*

    31  * @test

    32  * @bug 8050374

    33  * @summary Checks if a signed object is a copy of an original object

    34  */

    35 public class Copy {

    36 

    37     private static final String DSA = "DSA";

    38     private static final int KEY_SIZE = 512;

    39     private static final int MAGIC = 123;

    40 

    41     public static void main(String args[]) throws Exception {

    42         KeyPairGenerator kg = KeyPairGenerator.getInstance(DSA);

    43         kg.initialize(KEY_SIZE);

    44         KeyPair kp = kg.genKeyPair();

    45 

    46         Signature signature = Signature.getInstance(DSA);

    47         Test original = new Test();

    48         SignedObject so = new SignedObject(original, kp.getPrivate(),

    49                 signature);

    50         System.out.println("Signature algorithm: " + so.getAlgorithm());

    51 

    52         signature = Signature.getInstance(DSA, "SUN");

    53         if (!so.verify(kp.getPublic(), signature)) {

    54             throw new RuntimeException("Verification failed");

    55         }

    56 

    57         kg = KeyPairGenerator.getInstance(DSA);

    58         kg.initialize(KEY_SIZE);

    59         kp = kg.genKeyPair();

    60 

    61         if (so.verify(kp.getPublic(), signature)) {

    62             throw new RuntimeException("Unexpected success");

    63         }

    64 

    65         Object copy = so.getObject();

    66         if (!original.equals(copy)) {

    67             throw new RuntimeException("Signed object is not equal "

    68                     + "to original one: " + copy);

    69         }

    70 

    71         /*

    72          * The signed object is a copy of an original one.

    73          * Once the copy is made, further manipulation

    74          * of the original object shouldn't has any effect on the copy.

    75          */

    76         original.set(MAGIC - 1);

    77         copy = so.getObject();

    78         if (original.equals(copy)) {

    79             throw new RuntimeException("Signed object is not a copy "

    80                     + "of original one: " + copy);

    81         }

    82 

    83         System.out.println("Test passed");

    84     }

    85 

    86     private static class Test implements Serializable {

    87         private int number = MAGIC;

    88 

    89         public int get() {

    90             return number;

    91         }

    92 

    93         public void set(int magic) {

    94             this.number = magic;

    95         }

    96 

    97         @Override

    98         public int hashCode() {

    99             return number;

   100         }

   101 

   102         @Override

   103         public boolean equals(Object obj) {

   104             if (obj == null) {

   105                 return false;

   106             }

   107 

   108             if (!(obj instanceof Test)) {

   109                 return false;

   110             }

   111 

   112             Test other = (Test) obj;

   113             return number == other.number;

   114         }

   115 

   116         @Override

   117         public String toString() {

   118             return "" + number;

   119         }

   120     }

   121 }