samples/dynalink/BufferIndexingLinkerExporter.java - toolchain/jdk/jdk9_nashorn - Git at Google

 /*
  * Copyright (c) 2015, Oracle and/or its affiliates. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  *
  *   - Redistributions of source code must retain the above copyright
  *     notice, this list of conditions and the following disclaimer.
  *
  *   - Redistributions in binary form must reproduce the above copyright
  *     notice, this list of conditions and the following disclaimer in the
  *     documentation and/or other materials provided with the distribution.
  *
  *   - Neither the name of Oracle nor the names of its
  *     contributors may be used to endorse or promote products derived
  *     from this software without specific prior written permission.
  *
  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
  * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR
  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */

 import static jdk.dynalink.StandardNamespace.ELEMENT;
 import static jdk.dynalink.StandardNamespace.PROPERTY;
 import static jdk.dynalink.StandardOperation.GET;
 import static jdk.dynalink.StandardOperation.SET;

 import java.lang.invoke.MethodHandle;
 import java.lang.invoke.MethodType;
 import java.nio.Buffer;
 import java.nio.ByteBuffer;
 import java.nio.CharBuffer;
 import java.nio.DoubleBuffer;
 import java.nio.FloatBuffer;
 import java.nio.IntBuffer;
 import java.nio.LongBuffer;
 import java.nio.ShortBuffer;
 import java.util.ArrayList;
 import java.util.List;
 import jdk.dynalink.CallSiteDescriptor;
 import jdk.dynalink.NamedOperation;
 import jdk.dynalink.NamespaceOperation;
 import jdk.dynalink.Operation;
 import jdk.dynalink.StandardNamespace;
 import jdk.dynalink.linker.GuardedInvocation;
 import jdk.dynalink.linker.GuardingDynamicLinker;
 import jdk.dynalink.linker.GuardingDynamicLinkerExporter;
 import jdk.dynalink.linker.LinkRequest;
 import jdk.dynalink.linker.LinkerServices;
 import jdk.dynalink.linker.TypeBasedGuardingDynamicLinker;
 import jdk.dynalink.linker.support.Guards;
 import jdk.dynalink.linker.support.Lookup;

 /**
  * This is a dynalink pluggable linker (see http://openjdk.java.net/jeps/276).
  * This linker adds array-like indexing and "length" property to nio Buffer objects.
  */
 public final class BufferIndexingLinkerExporter extends GuardingDynamicLinkerExporter {
     static {
         System.out.println("pluggable dynalink buffer indexing linker loaded");
     }

     private static final MethodHandle BUFFER_LIMIT;
     private static final MethodHandle BYTEBUFFER_GET;
     private static final MethodHandle BYTEBUFFER_PUT;
     private static final MethodHandle CHARBUFFER_GET;
     private static final MethodHandle CHARBUFFER_PUT;
     private static final MethodHandle SHORTBUFFER_GET;
     private static final MethodHandle SHORTBUFFER_PUT;
     private static final MethodHandle INTBUFFER_GET;
     private static final MethodHandle INTBUFFER_PUT;
     private static final MethodHandle LONGBUFFER_GET;
     private static final MethodHandle LONGBUFFER_PUT;
     private static final MethodHandle FLOATBUFFER_GET;
     private static final MethodHandle FLOATBUFFER_PUT;
     private static final MethodHandle DOUBLEBUFFER_GET;
     private static final MethodHandle DOUBLEBUFFER_PUT;

     // guards
     private static final MethodHandle IS_BUFFER;
     private static final MethodHandle IS_BYTEBUFFER;
     private static final MethodHandle IS_CHARBUFFER;
     private static final MethodHandle IS_SHORTBUFFER;
     private static final MethodHandle IS_INTBUFFER;
     private static final MethodHandle IS_LONGBUFFER;
     private static final MethodHandle IS_FLOATBUFFER;
     private static final MethodHandle IS_DOUBLEBUFFER;

     private static final MethodType GUARD_TYPE;

     static {
         final Lookup look = Lookup.PUBLIC;
         BUFFER_LIMIT = look.findVirtual(Buffer.class, "limit", MethodType.methodType(int.class));
         BYTEBUFFER_GET = look.findVirtual(ByteBuffer.class, "get",
                 MethodType.methodType(byte.class, int.class));
         BYTEBUFFER_PUT = look.findVirtual(ByteBuffer.class, "put",
                 MethodType.methodType(ByteBuffer.class, int.class, byte.class));
         CHARBUFFER_GET = look.findVirtual(CharBuffer.class, "get",
                 MethodType.methodType(char.class, int.class));
         CHARBUFFER_PUT = look.findVirtual(CharBuffer.class, "put",
                 MethodType.methodType(CharBuffer.class, int.class, char.class));
         SHORTBUFFER_GET = look.findVirtual(ShortBuffer.class, "get",
                 MethodType.methodType(short.class, int.class));
         SHORTBUFFER_PUT = look.findVirtual(ShortBuffer.class, "put",
                 MethodType.methodType(ShortBuffer.class, int.class, short.class));
         INTBUFFER_GET = look.findVirtual(IntBuffer.class, "get",
                 MethodType.methodType(int.class, int.class));
         INTBUFFER_PUT = look.findVirtual(IntBuffer.class, "put",
                 MethodType.methodType(IntBuffer.class, int.class, int.class));
         LONGBUFFER_GET = look.findVirtual(LongBuffer.class, "get",
                 MethodType.methodType(long.class, int.class));
         LONGBUFFER_PUT = look.findVirtual(LongBuffer.class, "put",
                 MethodType.methodType(LongBuffer.class, int.class, long.class));
         FLOATBUFFER_GET = look.findVirtual(FloatBuffer.class, "get",
                 MethodType.methodType(float.class, int.class));
         FLOATBUFFER_PUT = look.findVirtual(FloatBuffer.class, "put",
                 MethodType.methodType(FloatBuffer.class, int.class, float.class));
         DOUBLEBUFFER_GET = look.findVirtual(DoubleBuffer.class, "get",
                 MethodType.methodType(double.class, int.class));
         DOUBLEBUFFER_PUT = look.findVirtual(DoubleBuffer.class, "put",
                 MethodType.methodType(DoubleBuffer.class, int.class, double.class));

         GUARD_TYPE = MethodType.methodType(boolean.class, Object.class);
         IS_BUFFER = Guards.isInstance(Buffer.class, GUARD_TYPE);
         IS_BYTEBUFFER = Guards.isInstance(ByteBuffer.class, GUARD_TYPE);
         IS_CHARBUFFER = Guards.isInstance(CharBuffer.class, GUARD_TYPE);
         IS_SHORTBUFFER = Guards.isInstance(ShortBuffer.class, GUARD_TYPE);
         IS_INTBUFFER = Guards.isInstance(IntBuffer.class, GUARD_TYPE);
         IS_LONGBUFFER = Guards.isInstance(LongBuffer.class, GUARD_TYPE);
         IS_FLOATBUFFER = Guards.isInstance(FloatBuffer.class, GUARD_TYPE);
         IS_DOUBLEBUFFER = Guards.isInstance(DoubleBuffer.class, GUARD_TYPE);
     }

     @Override
     public List<GuardingDynamicLinker> get() {
         final ArrayList<GuardingDynamicLinker> linkers = new ArrayList<>();
         linkers.add(new TypeBasedGuardingDynamicLinker() {
             @Override
             public boolean canLinkType(final Class<?> type) {
                 return Buffer.class.isAssignableFrom(type);
             }

             @Override
             public GuardedInvocation getGuardedInvocation(final LinkRequest request,
                 final LinkerServices linkerServices) throws Exception {
                 final Object self = request.getReceiver();
                 if (self == null || !canLinkType(self.getClass())) {
                     return null;
                 }

                 final CallSiteDescriptor desc = request.getCallSiteDescriptor();
                 final Operation namedOp = desc.getOperation();
                 final Operation namespaceOp = NamedOperation.getBaseOperation(namedOp);
                 final Operation op = NamespaceOperation.getBaseOperation(namespaceOp);
                 final StandardNamespace ns = StandardNamespace.findFirst(namespaceOp);
                 if (ns == null) {
                     return null;
                 }

                 if (op == GET) {
                     if (ns == ELEMENT) {
                         return linkGetElement(self);
                     } else if (ns == PROPERTY) {
                         final Object name = NamedOperation.getName(desc.getOperation());
                         if ("length".equals(name)) {
                             return linkLength();
                         }
                     }
                 } else if (op == SET && ns == ELEMENT) {
                     return linkSetElement(self);
                 }

                 return null;
             }
         });
         return linkers;
     }

     private static GuardedInvocation linkGetElement(final Object self) {
         MethodHandle method = null;
         MethodHandle guard = null;
         if (self instanceof ByteBuffer) {
             method = BYTEBUFFER_GET;
             guard = IS_BYTEBUFFER;
         } else if (self instanceof CharBuffer) {
             method = CHARBUFFER_GET;
             guard = IS_CHARBUFFER;
         } else if (self instanceof ShortBuffer) {
             method = SHORTBUFFER_GET;
             guard = IS_SHORTBUFFER;
         } else if (self instanceof IntBuffer) {
             method = INTBUFFER_GET;
             guard = IS_INTBUFFER;
         } else if (self instanceof LongBuffer) {
             method = LONGBUFFER_GET;
             guard = IS_LONGBUFFER;
         } else if (self instanceof FloatBuffer) {
             method = FLOATBUFFER_GET;
             guard = IS_FLOATBUFFER;
         } else if (self instanceof DoubleBuffer) {
             method = DOUBLEBUFFER_GET;
             guard = IS_DOUBLEBUFFER;
         }

         return method != null? new GuardedInvocation(method, guard) : null;
     }

     private static GuardedInvocation linkSetElement(final Object self) {
         MethodHandle method = null;
         MethodHandle guard = null;
         if (self instanceof ByteBuffer) {
             method = BYTEBUFFER_PUT;
             guard = IS_BYTEBUFFER;
         } else if (self instanceof CharBuffer) {
             method = CHARBUFFER_PUT;
             guard = IS_CHARBUFFER;
         } else if (self instanceof ShortBuffer) {
             method = SHORTBUFFER_PUT;
             guard = IS_SHORTBUFFER;
         } else if (self instanceof IntBuffer) {
             method = INTBUFFER_PUT;
             guard = IS_INTBUFFER;
         } else if (self instanceof LongBuffer) {
             method = LONGBUFFER_PUT;
             guard = IS_LONGBUFFER;
         } else if (self instanceof FloatBuffer) {
             method = FLOATBUFFER_PUT;
             guard = IS_FLOATBUFFER;
         } else if (self instanceof DoubleBuffer) {
             method = DOUBLEBUFFER_PUT;
             guard = IS_DOUBLEBUFFER;
         }

         return method != null? new GuardedInvocation(method, guard) : null;
     }

     private static GuardedInvocation linkLength() {
         return new GuardedInvocation(BUFFER_LIMIT, IS_BUFFER);
     }
 }
	/*
	* Copyright (c) 2015, Oracle and/or its affiliates. All rights reserved.
	*
	* Redistribution and use in source and binary forms, with or without
	* modification, are permitted provided that the following conditions
	* are met:
	*
	* - Redistributions of source code must retain the above copyright
	* notice, this list of conditions and the following disclaimer.
	*
	* - Redistributions in binary form must reproduce the above copyright
	* notice, this list of conditions and the following disclaimer in the
	* documentation and/or other materials provided with the distribution.
	*
	* - Neither the name of Oracle nor the names of its
	* contributors may be used to endorse or promote products derived
	* from this software without specific prior written permission.
	*
	* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
	* IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
	* THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
	* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
	* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
	* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
	* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
	* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
	* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
	* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
	* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
	*/

	import static jdk.dynalink.StandardNamespace.ELEMENT;
	import static jdk.dynalink.StandardNamespace.PROPERTY;
	import static jdk.dynalink.StandardOperation.GET;
	import static jdk.dynalink.StandardOperation.SET;

	import java.lang.invoke.MethodHandle;
	import java.lang.invoke.MethodType;
	import java.nio.Buffer;
	import java.nio.ByteBuffer;
	import java.nio.CharBuffer;
	import java.nio.DoubleBuffer;
	import java.nio.FloatBuffer;
	import java.nio.IntBuffer;
	import java.nio.LongBuffer;
	import java.nio.ShortBuffer;
	import java.util.ArrayList;
	import java.util.List;
	import jdk.dynalink.CallSiteDescriptor;
	import jdk.dynalink.NamedOperation;
	import jdk.dynalink.NamespaceOperation;
	import jdk.dynalink.Operation;
	import jdk.dynalink.StandardNamespace;
	import jdk.dynalink.linker.GuardedInvocation;
	import jdk.dynalink.linker.GuardingDynamicLinker;
	import jdk.dynalink.linker.GuardingDynamicLinkerExporter;
	import jdk.dynalink.linker.LinkRequest;
	import jdk.dynalink.linker.LinkerServices;
	import jdk.dynalink.linker.TypeBasedGuardingDynamicLinker;
	import jdk.dynalink.linker.support.Guards;
	import jdk.dynalink.linker.support.Lookup;

	/**
	* This is a dynalink pluggable linker (see http://openjdk.java.net/jeps/276).
	* This linker adds array-like indexing and "length" property to nio Buffer objects.
	*/
	public final class BufferIndexingLinkerExporter extends GuardingDynamicLinkerExporter {
	static {
	System.out.println("pluggable dynalink buffer indexing linker loaded");
	}

	private static final MethodHandle BUFFER_LIMIT;
	private static final MethodHandle BYTEBUFFER_GET;
	private static final MethodHandle BYTEBUFFER_PUT;
	private static final MethodHandle CHARBUFFER_GET;
	private static final MethodHandle CHARBUFFER_PUT;
	private static final MethodHandle SHORTBUFFER_GET;
	private static final MethodHandle SHORTBUFFER_PUT;
	private static final MethodHandle INTBUFFER_GET;
	private static final MethodHandle INTBUFFER_PUT;
	private static final MethodHandle LONGBUFFER_GET;
	private static final MethodHandle LONGBUFFER_PUT;
	private static final MethodHandle FLOATBUFFER_GET;
	private static final MethodHandle FLOATBUFFER_PUT;
	private static final MethodHandle DOUBLEBUFFER_GET;
	private static final MethodHandle DOUBLEBUFFER_PUT;

	// guards
	private static final MethodHandle IS_BUFFER;
	private static final MethodHandle IS_BYTEBUFFER;
	private static final MethodHandle IS_CHARBUFFER;
	private static final MethodHandle IS_SHORTBUFFER;
	private static final MethodHandle IS_INTBUFFER;
	private static final MethodHandle IS_LONGBUFFER;
	private static final MethodHandle IS_FLOATBUFFER;
	private static final MethodHandle IS_DOUBLEBUFFER;

	private static final MethodType GUARD_TYPE;

	static {
	final Lookup look = Lookup.PUBLIC;
	BUFFER_LIMIT = look.findVirtual(Buffer.class, "limit", MethodType.methodType(int.class));
	BYTEBUFFER_GET = look.findVirtual(ByteBuffer.class, "get",
	MethodType.methodType(byte.class, int.class));
	BYTEBUFFER_PUT = look.findVirtual(ByteBuffer.class, "put",
	MethodType.methodType(ByteBuffer.class, int.class, byte.class));
	CHARBUFFER_GET = look.findVirtual(CharBuffer.class, "get",
	MethodType.methodType(char.class, int.class));
	CHARBUFFER_PUT = look.findVirtual(CharBuffer.class, "put",
	MethodType.methodType(CharBuffer.class, int.class, char.class));
	SHORTBUFFER_GET = look.findVirtual(ShortBuffer.class, "get",
	MethodType.methodType(short.class, int.class));
	SHORTBUFFER_PUT = look.findVirtual(ShortBuffer.class, "put",
	MethodType.methodType(ShortBuffer.class, int.class, short.class));
	INTBUFFER_GET = look.findVirtual(IntBuffer.class, "get",
	MethodType.methodType(int.class, int.class));
	INTBUFFER_PUT = look.findVirtual(IntBuffer.class, "put",
	MethodType.methodType(IntBuffer.class, int.class, int.class));
	LONGBUFFER_GET = look.findVirtual(LongBuffer.class, "get",
	MethodType.methodType(long.class, int.class));
	LONGBUFFER_PUT = look.findVirtual(LongBuffer.class, "put",
	MethodType.methodType(LongBuffer.class, int.class, long.class));
	FLOATBUFFER_GET = look.findVirtual(FloatBuffer.class, "get",
	MethodType.methodType(float.class, int.class));
	FLOATBUFFER_PUT = look.findVirtual(FloatBuffer.class, "put",
	MethodType.methodType(FloatBuffer.class, int.class, float.class));
	DOUBLEBUFFER_GET = look.findVirtual(DoubleBuffer.class, "get",
	MethodType.methodType(double.class, int.class));
	DOUBLEBUFFER_PUT = look.findVirtual(DoubleBuffer.class, "put",
	MethodType.methodType(DoubleBuffer.class, int.class, double.class));

	GUARD_TYPE = MethodType.methodType(boolean.class, Object.class);
	IS_BUFFER = Guards.isInstance(Buffer.class, GUARD_TYPE);
	IS_BYTEBUFFER = Guards.isInstance(ByteBuffer.class, GUARD_TYPE);
	IS_CHARBUFFER = Guards.isInstance(CharBuffer.class, GUARD_TYPE);
	IS_SHORTBUFFER = Guards.isInstance(ShortBuffer.class, GUARD_TYPE);
	IS_INTBUFFER = Guards.isInstance(IntBuffer.class, GUARD_TYPE);
	IS_LONGBUFFER = Guards.isInstance(LongBuffer.class, GUARD_TYPE);
	IS_FLOATBUFFER = Guards.isInstance(FloatBuffer.class, GUARD_TYPE);
	IS_DOUBLEBUFFER = Guards.isInstance(DoubleBuffer.class, GUARD_TYPE);
	}

	@Override
	public List<GuardingDynamicLinker> get() {
	final ArrayList<GuardingDynamicLinker> linkers = new ArrayList<>();
	linkers.add(new TypeBasedGuardingDynamicLinker() {
	@Override
	public boolean canLinkType(final Class<?> type) {
	return Buffer.class.isAssignableFrom(type);
	}

	@Override
	public GuardedInvocation getGuardedInvocation(final LinkRequest request,
	final LinkerServices linkerServices) throws Exception {
	final Object self = request.getReceiver();
	if (self == null \|\| !canLinkType(self.getClass())) {
	return null;
	}

	final CallSiteDescriptor desc = request.getCallSiteDescriptor();
	final Operation namedOp = desc.getOperation();
	final Operation namespaceOp = NamedOperation.getBaseOperation(namedOp);
	final Operation op = NamespaceOperation.getBaseOperation(namespaceOp);
	final StandardNamespace ns = StandardNamespace.findFirst(namespaceOp);
	if (ns == null) {
	return null;
	}

	if (op == GET) {
	if (ns == ELEMENT) {
	return linkGetElement(self);
	} else if (ns == PROPERTY) {
	final Object name = NamedOperation.getName(desc.getOperation());
	if ("length".equals(name)) {
	return linkLength();
	}
	}
	} else if (op == SET && ns == ELEMENT) {
	return linkSetElement(self);
	}

	return null;
	}
	});
	return linkers;
	}

	private static GuardedInvocation linkGetElement(final Object self) {
	MethodHandle method = null;
	MethodHandle guard = null;
	if (self instanceof ByteBuffer) {
	method = BYTEBUFFER_GET;
	guard = IS_BYTEBUFFER;
	} else if (self instanceof CharBuffer) {
	method = CHARBUFFER_GET;
	guard = IS_CHARBUFFER;
	} else if (self instanceof ShortBuffer) {
	method = SHORTBUFFER_GET;
	guard = IS_SHORTBUFFER;
	} else if (self instanceof IntBuffer) {
	method = INTBUFFER_GET;
	guard = IS_INTBUFFER;
	} else if (self instanceof LongBuffer) {
	method = LONGBUFFER_GET;
	guard = IS_LONGBUFFER;
	} else if (self instanceof FloatBuffer) {
	method = FLOATBUFFER_GET;
	guard = IS_FLOATBUFFER;
	} else if (self instanceof DoubleBuffer) {
	method = DOUBLEBUFFER_GET;
	guard = IS_DOUBLEBUFFER;
	}

	return method != null? new GuardedInvocation(method, guard) : null;
	}

	private static GuardedInvocation linkSetElement(final Object self) {
	MethodHandle method = null;
	MethodHandle guard = null;
	if (self instanceof ByteBuffer) {
	method = BYTEBUFFER_PUT;
	guard = IS_BYTEBUFFER;
	} else if (self instanceof CharBuffer) {
	method = CHARBUFFER_PUT;
	guard = IS_CHARBUFFER;
	} else if (self instanceof ShortBuffer) {
	method = SHORTBUFFER_PUT;
	guard = IS_SHORTBUFFER;
	} else if (self instanceof IntBuffer) {
	method = INTBUFFER_PUT;
	guard = IS_INTBUFFER;
	} else if (self instanceof LongBuffer) {
	method = LONGBUFFER_PUT;
	guard = IS_LONGBUFFER;
	} else if (self instanceof FloatBuffer) {
	method = FLOATBUFFER_PUT;
	guard = IS_FLOATBUFFER;
	} else if (self instanceof DoubleBuffer) {
	method = DOUBLEBUFFER_PUT;
	guard = IS_DOUBLEBUFFER;
	}

	return method != null? new GuardedInvocation(method, guard) : null;
	}

	private static GuardedInvocation linkLength() {
	return new GuardedInvocation(BUFFER_LIMIT, IS_BUFFER);
	}
	}