1/*
2 * Copyright (c) 2000, 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/* @test
25 * @summary Test proper functioning of class-defined writeReplace methods.
26 */
27
28import java.io.*;
29
30public class WriteReplace {
31
32    static class ReplaceMe implements Serializable {
33        private Object obj;
34        private boolean writeReplaceCalled = false;
35
36        public ReplaceMe(Object obj) {
37            this.obj = obj;
38        }
39
40        private Object writeReplace() throws ObjectStreamException {
41            if (writeReplaceCalled) {
42                throw new Error("multiple calls to writeReplace");
43            }
44            writeReplaceCalled = true;
45            return obj;
46        }
47    }
48
49    public static void main(String[] args) throws Exception {
50        final int nobjs = 10;
51        final int nrounds = 10;
52
53        Object common = "foo";
54        ReplaceMe[] objs = new ReplaceMe[nobjs];
55        for (int i = 0; i < nobjs; i++) {
56            objs[i] = new ReplaceMe(common);
57        }
58
59        ByteArrayOutputStream bout = new ByteArrayOutputStream();
60        ObjectOutputStream oout = new ObjectOutputStream(bout);
61        for (int i = 0; i < nrounds; i++) {
62            for (int j = 0; j < nobjs; j++) {
63                oout.writeObject(objs[j]);
64            }
65        }
66
67        oout.flush();
68        ByteArrayInputStream bin =
69            new ByteArrayInputStream(bout.toByteArray());
70        ObjectInputStream oin = new ObjectInputStream(bin);
71        common = null;
72
73        for (int i = 0; i < nrounds; i++) {
74            for (int j = 0; j < nobjs; j++) {
75                if (common == null) {
76                    common = oin.readObject();
77                } else {
78                    if (oin.readObject() != common) {
79                        throw new Error("incorrect replacement object");
80                    }
81                }
82            }
83        }
84    }
85}
86