/*

 * 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

/*

 * @test

 * @bug 8154556

 * @run testng/othervm -Diters=20000 -XX:TieredStopAtLevel=1 VarHandleTestByteArrayAsShort

 * @run testng/othervm -Diters=20000                         VarHandleTestByteArrayAsShort

 * @run testng/othervm -Diters=20000 -XX:-TieredCompilation  VarHandleTestByteArrayAsShort

 */

import org.testng.annotations.DataProvider;

import org.testng.annotations.Test;

import java.lang.invoke.MethodHandles;

import java.lang.invoke.VarHandle;

import java.nio.ByteBuffer;

import java.nio.ByteOrder;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.EnumSet;

import java.util.List;

import static org.testng.Assert.*;

public class VarHandleTestByteArrayAsShort extends VarHandleBaseByteArrayTest {

    static final int SIZE = Short.BYTES;

    static final short VALUE_1 = (short)0x0102;

    static final short VALUE_2 = (short)0x1112;

    static final short VALUE_3 = (short)0xFFFE;

    @Override

    public List<VarHandleSource> setupVarHandleSources(boolean same) {

        // Combinations of VarHandle byte[] or ByteBuffer

        List<VarHandleSource> vhss = new ArrayList<>();

        for (MemoryMode endianess : List.of(MemoryMode.BIG_ENDIAN, MemoryMode.LITTLE_ENDIAN)) {

            ByteOrder bo = endianess == MemoryMode.BIG_ENDIAN

                    ? ByteOrder.BIG_ENDIAN : ByteOrder.LITTLE_ENDIAN;

            Class<?> arrayType;

            if (same) {

                arrayType = short[].class;

            }

            else {

                arrayType = int[].class;

            }

            VarHandleSource aeh = new VarHandleSource(

                    MethodHandles.byteArrayViewVarHandle(arrayType, bo),

                    endianess, MemoryMode.READ_WRITE);

            vhss.add(aeh);

            VarHandleSource bbh = new VarHandleSource(

                    MethodHandles.byteBufferViewVarHandle(arrayType, bo),

                    endianess, MemoryMode.READ_WRITE);

            vhss.add(bbh);

        }

        return vhss;

    }

    @Test

    public void testEqualsAndHashCode() {

        VarHandle[] vhs1 = setupVarHandleSources(true).stream().

            map(vhs -> vhs.s).toArray(VarHandle[]::new);

        VarHandle[] vhs2 = setupVarHandleSources(true).stream().

            map(vhs -> vhs.s).toArray(VarHandle[]::new);

        for (int i = 0; i < vhs1.length; i++) {

            for (int j = 0; j < vhs1.length; j++) {

                if (i == j) {

                    assertEquals(vhs1[i], vhs1[i]);

                }

                else {

                    assertNotEquals(vhs1[i], vhs1[j]);

                    assertNotEquals(vhs1[i], vhs2[j]);

                }

            }

        }

        VarHandle[] vhs3 = setupVarHandleSources(false).stream().

            map(vhs -> vhs.s).toArray(VarHandle[]::new);

        for (int i = 0; i < vhs1.length; i++) {

            assertNotEquals(vhs1[i], vhs3[i]);

        }

    }

    @Test(dataProvider = "varHandlesProvider")

    public void testIsAccessModeSupported(VarHandleSource vhs) {

        VarHandle vh = vhs.s;

        assertTrue(vh.isAccessModeSupported(VarHandle.AccessMode.GET));

        assertTrue(vh.isAccessModeSupported(VarHandle.AccessMode.SET));

        assertTrue(vh.isAccessModeSupported(VarHandle.AccessMode.GET_VOLATILE));

        assertTrue(vh.isAccessModeSupported(VarHandle.AccessMode.SET_VOLATILE));

        assertTrue(vh.isAccessModeSupported(VarHandle.AccessMode.GET_ACQUIRE));

        assertTrue(vh.isAccessModeSupported(VarHandle.AccessMode.SET_RELEASE));

        assertTrue(vh.isAccessModeSupported(VarHandle.AccessMode.GET_OPAQUE));

        assertTrue(vh.isAccessModeSupported(VarHandle.AccessMode.SET_OPAQUE));

        assertFalse(vh.isAccessModeSupported(VarHandle.AccessMode.COMPARE_AND_SET));

        assertFalse(vh.isAccessModeSupported(VarHandle.AccessMode.COMPARE_AND_EXCHANGE));

        assertFalse(vh.isAccessModeSupported(VarHandle.AccessMode.COMPARE_AND_EXCHANGE_ACQUIRE));

        assertFalse(vh.isAccessModeSupported(VarHandle.AccessMode.COMPARE_AND_EXCHANGE_RELEASE));

        assertFalse(vh.isAccessModeSupported(VarHandle.AccessMode.WEAK_COMPARE_AND_SET_PLAIN));

        assertFalse(vh.isAccessModeSupported(VarHandle.AccessMode.WEAK_COMPARE_AND_SET));

        assertFalse(vh.isAccessModeSupported(VarHandle.AccessMode.WEAK_COMPARE_AND_SET_ACQUIRE));

        assertFalse(vh.isAccessModeSupported(VarHandle.AccessMode.WEAK_COMPARE_AND_SET_RELEASE));

        assertFalse(vh.isAccessModeSupported(VarHandle.AccessMode.GET_AND_SET));

        assertFalse(vh.isAccessModeSupported(VarHandle.AccessMode.GET_AND_SET_ACQUIRE));

        assertFalse(vh.isAccessModeSupported(VarHandle.AccessMode.GET_AND_SET_RELEASE));

        assertFalse(vh.isAccessModeSupported(VarHandle.AccessMode.GET_AND_ADD));

        assertFalse(vh.isAccessModeSupported(VarHandle.AccessMode.GET_AND_ADD_ACQUIRE));

        assertFalse(vh.isAccessModeSupported(VarHandle.AccessMode.GET_AND_ADD_RELEASE));

        assertFalse(vh.isAccessModeSupported(VarHandle.AccessMode.GET_AND_BITWISE_OR));

        assertFalse(vh.isAccessModeSupported(VarHandle.AccessMode.GET_AND_BITWISE_OR_ACQUIRE));

        assertFalse(vh.isAccessModeSupported(VarHandle.AccessMode.GET_AND_BITWISE_OR_RELEASE));

        assertFalse(vh.isAccessModeSupported(VarHandle.AccessMode.GET_AND_BITWISE_AND));

        assertFalse(vh.isAccessModeSupported(VarHandle.AccessMode.GET_AND_BITWISE_AND_ACQUIRE));

        assertFalse(vh.isAccessModeSupported(VarHandle.AccessMode.GET_AND_BITWISE_AND_RELEASE));

        assertFalse(vh.isAccessModeSupported(VarHandle.AccessMode.GET_AND_BITWISE_XOR));

        assertFalse(vh.isAccessModeSupported(VarHandle.AccessMode.GET_AND_BITWISE_XOR_ACQUIRE));

        assertFalse(vh.isAccessModeSupported(VarHandle.AccessMode.GET_AND_BITWISE_XOR_RELEASE));

    }

    @Test(dataProvider = "typesProvider")

    public void testTypes(VarHandle vh, List<java.lang.Class<?>> pts) {

        assertEquals(vh.varType(), short.class);

        assertEquals(vh.coordinateTypes(), pts);

        testTypes(vh);

    }

    @DataProvider

    public Object[][] accessTestCaseProvider() throws Exception {

        List<AccessTestCase<?>> cases = new ArrayList<>();

        for (ByteArrayViewSource<?> bav : bavss) {

            for (VarHandleSource vh : vhss) {

                if (vh.matches(bav)) {

                    if (bav instanceof ByteArraySource) {

                        ByteArraySource bas = (ByteArraySource) bav;

                        cases.add(new VarHandleSourceAccessTestCase(

                                "read write", bav, vh, h -> testArrayReadWrite(bas, h),

                                true));

                        cases.add(new VarHandleSourceAccessTestCase(

                                "null array", bav, vh, h -> testArrayNPE(bas, h),

                                false));

                        cases.add(new VarHandleSourceAccessTestCase(

                                "unsupported", bav, vh, h -> testArrayUnsupported(bas, h),

                                false));

                        cases.add(new VarHandleSourceAccessTestCase(

                                "index out of bounds", bav, vh, h -> testArrayIndexOutOfBounds(bas, h),

                                false));

                        cases.add(new VarHandleSourceAccessTestCase(

                                "misaligned access", bav, vh, h -> testArrayMisalignedAccess(bas, h),

                                false));

                    }

                    else {

                        ByteBufferSource bbs = (ByteBufferSource) bav;

                        if (MemoryMode.READ_WRITE.isSet(bav.memoryModes)) {

                            cases.add(new VarHandleSourceAccessTestCase(

                                    "read write", bav, vh, h -> testArrayReadWrite(bbs, h),

                                    true));

                        }

                        else {

                            cases.add(new VarHandleSourceAccessTestCase(

                                    "read only", bav, vh, h -> testArrayReadOnly(bbs, h),

                                    true));

                        }

                        cases.add(new VarHandleSourceAccessTestCase(

                                "null buffer", bav, vh, h -> testArrayNPE(bbs, h),

                                false));

                        cases.add(new VarHandleSourceAccessTestCase(

                                "unsupported", bav, vh, h -> testArrayUnsupported(bbs, h),

                                false));

                        cases.add(new VarHandleSourceAccessTestCase(

                                "index out of bounds", bav, vh, h -> testArrayIndexOutOfBounds(bbs, h),

                                false));

                        cases.add(new VarHandleSourceAccessTestCase(

                                "misaligned access", bav, vh, h -> testArrayMisalignedAccess(bbs, h),

                                false));

                    }

                }

            }

        }

        // Work around issue with jtreg summary reporting which truncates

        // the String result of Object.toString to 30 characters, hence

        // the first dummy argument

        return cases.stream().map(tc -> new Object[]{tc.toString(), tc}).toArray(Object[][]::new);

    }

    @Test(dataProvider = "accessTestCaseProvider")

    public <T> void testAccess(String desc, AccessTestCase<T> atc) throws Throwable {

        T t = atc.get();

        int iters = atc.requiresLoop() ? ITERS : 1;

        for (int c = 0; c < iters; c++) {

            atc.testAccess(t);

        }

    }

    static void testArrayNPE(ByteArraySource bs, VarHandleSource vhs) {

        VarHandle vh = vhs.s;

        byte[] array = null;

        int ci = 1;

        checkNPE(() -> {

            short x = (short) vh.get(array, ci);

        });

        checkNPE(() -> {

            vh.set(array, ci, VALUE_1);

        });

        checkNPE(() -> {

            short x = (short) vh.getVolatile(array, ci);

        });

        checkNPE(() -> {

            short x = (short) vh.getAcquire(array, ci);

        });

        checkNPE(() -> {

            short x = (short) vh.getOpaque(array, ci);

        });

        checkNPE(() -> {

            vh.setVolatile(array, ci, VALUE_1);

        });

        checkNPE(() -> {

            vh.setRelease(array, ci, VALUE_1);

        });

        checkNPE(() -> {

            vh.setOpaque(array, ci, VALUE_1);

        });

    }

    static void testArrayNPE(ByteBufferSource bs, VarHandleSource vhs) {

        VarHandle vh = vhs.s;

        ByteBuffer array = null;

        int ci = 1;

        checkNPE(() -> {

            short x = (short) vh.get(array, ci);

        });

        checkNPE(() -> {

            vh.set(array, ci, VALUE_1);

        });

        checkNPE(() -> {

            short x = (short) vh.getVolatile(array, ci);

        });

        checkNPE(() -> {

            short x = (short) vh.getAcquire(array, ci);

        });

        checkNPE(() -> {

            short x = (short) vh.getOpaque(array, ci);

        });

        checkNPE(() -> {

            vh.setVolatile(array, ci, VALUE_1);

        });

        checkNPE(() -> {

            vh.setRelease(array, ci, VALUE_1);

        });

        checkNPE(() -> {

            vh.setOpaque(array, ci, VALUE_1);

        });

    }

    static void testArrayUnsupported(ByteArraySource bs, VarHandleSource vhs) {

        VarHandle vh = vhs.s;

        byte[] array = bs.s;

        int ci = 1;

        checkUOE(() -> {

            boolean r = vh.compareAndSet(array, ci, VALUE_1, VALUE_2);

        });

        checkUOE(() -> {

            short r = (short) vh.compareAndExchange(array, ci, VALUE_2, VALUE_1);

        });

        checkUOE(() -> {

            short r = (short) vh.compareAndExchangeAcquire(array, ci, VALUE_2, VALUE_1);

        });

        checkUOE(() -> {

            short r = (short) vh.compareAndExchangeRelease(array, ci, VALUE_2, VALUE_1);

        });

        checkUOE(() -> {

            boolean r = vh.weakCompareAndSetPlain(array, ci, VALUE_1, VALUE_2);

        });

        checkUOE(() -> {

            boolean r = vh.weakCompareAndSet(array, ci, VALUE_1, VALUE_2);

        });

        checkUOE(() -> {

            boolean r = vh.weakCompareAndSetAcquire(array, ci, VALUE_1, VALUE_2);

        });

        checkUOE(() -> {

            boolean r = vh.weakCompareAndSetRelease(array, ci, VALUE_1, VALUE_2);

        });

        checkUOE(() -> {

            short o = (short) vh.getAndSet(array, ci, VALUE_1);

        });

        checkUOE(() -> {

            short o = (short) vh.getAndSetAcquire(array, ci, VALUE_1);

        });

        checkUOE(() -> {

            short o = (short) vh.getAndSetRelease(array, ci, VALUE_1);

        });

        checkUOE(() -> {

            short o = (short) vh.getAndAdd(array, ci, VALUE_1);

        });

        checkUOE(() -> {

            short o = (short) vh.getAndAddAcquire(array, ci, VALUE_1);

        });

        checkUOE(() -> {

            short o = (short) vh.getAndAddRelease(array, ci, VALUE_1);

        });

        checkUOE(() -> {

            short o = (short) vh.getAndBitwiseOr(array, ci, VALUE_1);

        });

        checkUOE(() -> {

            short o = (short) vh.getAndBitwiseOrAcquire(array, ci, VALUE_1);

        });

        checkUOE(() -> {

            short o = (short) vh.getAndBitwiseOrRelease(array, ci, VALUE_1);

        });

        checkUOE(() -> {

            short o = (short) vh.getAndBitwiseAnd(array, ci, VALUE_1);

        });

        checkUOE(() -> {

            short o = (short) vh.getAndBitwiseAndAcquire(array, ci, VALUE_1);

        });

        checkUOE(() -> {

            short o = (short) vh.getAndBitwiseAndRelease(array, ci, VALUE_1);

        });

        checkUOE(() -> {

            short o = (short) vh.getAndBitwiseXor(array, ci, VALUE_1);

        });

        checkUOE(() -> {

            short o = (short) vh.getAndBitwiseXorAcquire(array, ci, VALUE_1);

        });

        checkUOE(() -> {

            short o = (short) vh.getAndBitwiseXorRelease(array, ci, VALUE_1);

        });

    }

    static void testArrayUnsupported(ByteBufferSource bs, VarHandleSource vhs) {

        VarHandle vh = vhs.s;

        ByteBuffer array = bs.s;

        int ci = 0;

        boolean readOnly = MemoryMode.READ_ONLY.isSet(bs.memoryModes);

        if (readOnly) {

            checkROBE(() -> {

                vh.set(array, ci, VALUE_1);

            });

        }

        if (readOnly) {

            checkROBE(() -> {

                vh.setVolatile(array, ci, VALUE_1);

            });

            checkROBE(() -> {

                vh.setRelease(array, ci, VALUE_1);

            });

            checkROBE(() -> {

                vh.setOpaque(array, ci, VALUE_1);

            });

            checkUOE(() -> {

                boolean r = vh.compareAndSet(array, ci, VALUE_1, VALUE_2);

            });

            checkUOE(() -> {

                short r = (short) vh.compareAndExchange(array, ci, VALUE_2, VALUE_1);

            });

            checkUOE(() -> {

                short r = (short) vh.compareAndExchangeAcquire(array, ci, VALUE_2, VALUE_1);

            });

            checkUOE(() -> {

                short r = (short) vh.compareAndExchangeRelease(array, ci, VALUE_2, VALUE_1);

            });

            checkUOE(() -> {

                boolean r = vh.weakCompareAndSetPlain(array, ci, VALUE_1, VALUE_2);

            });

            checkUOE(() -> {

                boolean r = vh.weakCompareAndSet(array, ci, VALUE_1, VALUE_2);

            });

            checkUOE(() -> {

                boolean r = vh.weakCompareAndSetAcquire(array, ci, VALUE_1, VALUE_2);

            });

            checkUOE(() -> {

                boolean r = vh.weakCompareAndSetRelease(array, ci, VALUE_1, VALUE_2);

            });

            checkUOE(() -> {

                short o = (short) vh.getAndSet(array, ci, VALUE_1);

            });

            checkUOE(() -> {

                short o = (short) vh.getAndSetAcquire(array, ci, VALUE_1);

            });

            checkUOE(() -> {

                short o = (short) vh.getAndSetRelease(array, ci, VALUE_1);

            });

            checkUOE(() -> {

                short o = (short) vh.getAndAdd(array, ci, VALUE_1);

            });

            checkUOE(() -> {

                short o = (short) vh.getAndAddAcquire(array, ci, VALUE_1);

            });

            checkUOE(() -> {

                short o = (short) vh.getAndAddRelease(array, ci, VALUE_1);

            });

            checkUOE(() -> {

                short o = (short) vh.getAndBitwiseOr(array, ci, VALUE_1);

            });

            checkUOE(() -> {

                short o = (short) vh.getAndBitwiseOrAcquire(array, ci, VALUE_1);

            });

            checkUOE(() -> {

                short o = (short) vh.getAndBitwiseOrRelease(array, ci, VALUE_1);

            });

            checkUOE(() -> {