1 /* 2 * Copyright (c) 2019, 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 package org.graalvm.compiler.hotspot.replacements; 26 27 import static org.graalvm.compiler.hotspot.GraalHotSpotVMConfigBase.INJECTED_INTRINSIC_CONTEXT; 28 import static org.graalvm.compiler.hotspot.GraalHotSpotVMConfigBase.INJECTED_METAACCESS; 29 import static org.graalvm.compiler.hotspot.replacements.CipherBlockChainingSubstitutions.aesCryptType; 30 import static org.graalvm.compiler.hotspot.replacements.CipherBlockChainingSubstitutions.embeddedCipherOffset; 31 import static org.graalvm.compiler.nodes.PiNode.piCastNonNull; 32 33 import org.graalvm.compiler.api.replacements.ClassSubstitution; 34 import org.graalvm.compiler.api.replacements.Fold; 35 import org.graalvm.compiler.api.replacements.Fold.InjectedParameter; 36 import org.graalvm.compiler.api.replacements.MethodSubstitution; 37 import org.graalvm.compiler.hotspot.HotSpotBackend; 38 import org.graalvm.compiler.nodes.ComputeObjectAddressNode; 39 import org.graalvm.compiler.nodes.extended.RawLoadNode; 40 import org.graalvm.compiler.nodes.graphbuilderconf.IntrinsicContext; 41 import org.graalvm.compiler.replacements.ReplacementsUtil; 42 import org.graalvm.compiler.word.Word; 43 import jdk.internal.vm.compiler.word.LocationIdentity; 44 import jdk.internal.vm.compiler.word.WordFactory; 45 46 import jdk.vm.ci.meta.JavaKind; 47 import jdk.vm.ci.meta.ResolvedJavaType; 48 49 @ClassSubstitution(className = "com.sun.crypto.provider.CounterMode", optional = true) 50 public class CounterModeSubstitutions { 51 52 @MethodSubstitution(isStatic = false) 53 static int implCrypt(Object receiver, byte[] in, int inOff, int len, byte[] out, int outOff) { 54 Object realReceiver = piCastNonNull(receiver, HotSpotReplacementsUtil.methodHolderClass(INJECTED_INTRINSIC_CONTEXT)); 55 Object embeddedCipher = RawLoadNode.load(realReceiver, embeddedCipherOffset(INJECTED_INTRINSIC_CONTEXT), JavaKind.Object, LocationIdentity.any()); 56 Object aesCipher = piCastNonNull(embeddedCipher, aesCryptType(INJECTED_INTRINSIC_CONTEXT)); 57 58 Word srcAddr = WordFactory.unsigned(ComputeObjectAddressNode.get(in, ReplacementsUtil.getArrayBaseOffset(INJECTED_METAACCESS, JavaKind.Byte) + inOff)); 59 Word dstAddr = WordFactory.unsigned(ComputeObjectAddressNode.get(out, ReplacementsUtil.getArrayBaseOffset(INJECTED_METAACCESS, JavaKind.Byte) + outOff)); 60 Word usedPtr = WordFactory.unsigned(ComputeObjectAddressNode.get(realReceiver, usedOffset(INJECTED_INTRINSIC_CONTEXT))); 61 62 int cntOffset = counterOffset(INJECTED_INTRINSIC_CONTEXT); 63 int encCntOffset = encCounterOffset(INJECTED_INTRINSIC_CONTEXT); 64 Object kObject = RawLoadNode.load(aesCipher, AESCryptSubstitutions.kOffset(INJECTED_INTRINSIC_CONTEXT), JavaKind.Object, LocationIdentity.any()); 65 Object cntObj = RawLoadNode.load(realReceiver, cntOffset, JavaKind.Object, LocationIdentity.any()); 66 Object encCntObj = RawLoadNode.load(realReceiver, encCntOffset, JavaKind.Object, LocationIdentity.any()); 67 68 Word kPtr = Word.objectToTrackedPointer(kObject).add(ReplacementsUtil.getArrayBaseOffset(INJECTED_METAACCESS, JavaKind.Int)); 69 Word cntPtr = Word.objectToTrackedPointer(cntObj).add(ReplacementsUtil.getArrayBaseOffset(INJECTED_METAACCESS, JavaKind.Byte)); 70 Word encCntPtr = Word.objectToTrackedPointer(encCntObj).add(ReplacementsUtil.getArrayBaseOffset(INJECTED_METAACCESS, JavaKind.Byte)); 71 72 return HotSpotBackend.counterModeAESCrypt(srcAddr, dstAddr, kPtr, cntPtr, len, encCntPtr, usedPtr); 73 } 74 75 static ResolvedJavaType counterModeType(IntrinsicContext context) { 76 return HotSpotReplacementsUtil.getType(context, "Lcom/sun/crypto/provider/CounterMode;"); 77 } 78 79 @Fold 80 static int counterOffset(@InjectedParameter IntrinsicContext context) { 81 return HotSpotReplacementsUtil.getFieldOffset(counterModeType(context), "counter"); 82 } 83 84 @Fold 85 static int encCounterOffset(@InjectedParameter IntrinsicContext context) { 86 return HotSpotReplacementsUtil.getFieldOffset(counterModeType(context), "encryptedCounter"); 87 } 88 89 @Fold 90 static int usedOffset(@InjectedParameter IntrinsicContext context) { 91 return HotSpotReplacementsUtil.getFieldOffset(counterModeType(context), "used"); 92 } 93 }