1 /* 2 * Copyright (c) 2015, 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 package org.graalvm.compiler.core.amd64; 25 26 import static org.graalvm.compiler.asm.amd64.AMD64Assembler.OperandSize.QWORD; 27 import static org.graalvm.compiler.asm.amd64.AMD64Assembler.OperandSize.WORD; 28 29 import java.util.HashMap; 30 import java.util.Map; 31 32 import org.graalvm.compiler.core.common.LIRKind; 33 import org.graalvm.compiler.lir.VirtualStackSlot; 34 import org.graalvm.compiler.lir.amd64.AMD64LIRInstruction; 35 import org.graalvm.compiler.lir.amd64.AMD64Move.AMD64PushPopStackMove; 36 import org.graalvm.compiler.lir.framemap.FrameMapBuilder; 37 import org.graalvm.compiler.lir.gen.LIRGeneratorTool.MoveFactory; 38 39 import jdk.vm.ci.amd64.AMD64Kind; 40 import jdk.vm.ci.code.Architecture; 41 import jdk.vm.ci.code.Register; 42 import jdk.vm.ci.code.RegisterArray; 43 import jdk.vm.ci.code.RegisterConfig; 44 import jdk.vm.ci.meta.AllocatableValue; 45 import jdk.vm.ci.meta.PlatformKind; 46 47 public abstract class AMD64MoveFactoryBase implements MoveFactory { 48 49 private final BackupSlotProvider backupSlotProvider; 50 51 private static class RegisterBackupPair { 52 public final Register register; 53 public final VirtualStackSlot backupSlot; 54 55 RegisterBackupPair(Register register, VirtualStackSlot backupSlot) { 56 this.register = register; 57 this.backupSlot = backupSlot; 58 } 59 } 60 61 public static final class BackupSlotProvider { 62 63 private final FrameMapBuilder frameMapBuilder; 64 private Map<PlatformKind.Key, RegisterBackupPair> categorized; 65 66 public BackupSlotProvider(FrameMapBuilder frameMapBuilder) { 67 this.frameMapBuilder = frameMapBuilder; 68 } 69 70 protected RegisterBackupPair getScratchRegister(PlatformKind kind) { 71 PlatformKind.Key key = kind.getKey(); 72 if (categorized == null) { 73 categorized = new HashMap<>(); 74 } else if (categorized.containsKey(key)) { 75 return categorized.get(key); 76 } 77 78 RegisterConfig registerConfig = frameMapBuilder.getRegisterConfig(); 79 80 RegisterArray availableRegister = registerConfig.filterAllocatableRegisters(kind, registerConfig.getAllocatableRegisters()); 81 assert availableRegister != null && availableRegister.size() > 1; 82 Register scratchRegister = availableRegister.get(0); 83 84 Architecture arch = frameMapBuilder.getCodeCache().getTarget().arch; 85 LIRKind largestKind = LIRKind.value(arch.getLargestStorableKind(scratchRegister.getRegisterCategory())); 86 VirtualStackSlot backupSlot = frameMapBuilder.allocateSpillSlot(largestKind); 87 88 RegisterBackupPair value = new RegisterBackupPair(scratchRegister, backupSlot); 89 categorized.put(key, value); 90 91 return value; 92 } 93 } 94 95 public AMD64MoveFactoryBase(BackupSlotProvider backupSlotProvider) { 96 this.backupSlotProvider = backupSlotProvider; 97 } 98 99 @Override 100 public final AMD64LIRInstruction createStackMove(AllocatableValue result, AllocatableValue input) { 101 AMD64Kind kind = (AMD64Kind) result.getPlatformKind(); 102 switch (kind.getSizeInBytes()) { 103 case 2: 104 return new AMD64PushPopStackMove(WORD, result, input); 105 case 8: 106 return new AMD64PushPopStackMove(QWORD, result, input); 107 default: 108 RegisterBackupPair backup = backupSlotProvider.getScratchRegister(input.getPlatformKind()); 109 Register scratchRegister = backup.register; 110 VirtualStackSlot backupSlot = backup.backupSlot; 111 return createStackMove(result, input, scratchRegister, backupSlot); 112 } 113 } 114 115 public abstract AMD64LIRInstruction createStackMove(AllocatableValue result, AllocatableValue input, Register scratchRegister, AllocatableValue backupSlot); 116 }