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 }