Skip to content

Commit d422157

Browse files
rakachanrakachan
committed
Added the org.graalvm.continuations.IdentityHashCodes class, providing utilities for restoring identity hashcodes
1 parent 753b81f commit d422157

File tree

16 files changed

+316
-23
lines changed

16 files changed

+316
-23
lines changed

‎espresso/CHANGELOG.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
### User-visible changes
55
* Added experimental support for JVMCI. It can be enabled with the `java.EnableJVMCI` option.
66
* Added experimentation support for `-javaagent`. It can also be enabled from the polyglot API with `java.JavaAgent.$i` option set to `/path/to/jar=agent-options` where `$i` starts at 0 and increments by 1 for each extra java agent.
7+
* Added the `org.graalvm.continuations.IdentityHashCodes` class, providing utilities for restoring identity hashcodes. This may be used for more properly deserializing continuations.
78

89
## Version 24.2.0
910
### User-visible changes

‎espresso/src/com.oracle.truffle.espresso/src/com/oracle/truffle/espresso/EspressoLanguage.java

Lines changed: 21 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,6 +119,7 @@ public final class EspressoLanguage extends TruffleLanguage<EspressoContext> imp
119119
@CompilationFinal private SignatureSymbols signatureSymbols;
120120

121121
private final StaticProperty arrayProperty = new DefaultStaticProperty("array");
122+
private final StaticProperty arrayHashCodeProperty = new DefaultStaticProperty("ihashcode");
122123
// This field should be final, but creating a shape requires a fully-initialized instance of
123124
// TruffleLanguage.
124125
@CompilationFinal //
@@ -144,6 +145,7 @@ public final class EspressoLanguage extends TruffleLanguage<EspressoContext> imp
144145
@CompilationFinal private boolean eagerFrameAnalysis;
145146
@CompilationFinal private boolean internalJvmciEnabled;
146147
@CompilationFinal private boolean useEspressoLibs;
148+
@CompilationFinal private boolean continuum;
147149
// endregion Options
148150

149151
// region Allocation
@@ -242,6 +244,7 @@ private void initializeOptions(final TruffleLanguage.Env env) {
242244
previewEnabled = env.getOptions().get(EspressoOptions.EnablePreview);
243245
whiteBoxEnabled = env.getOptions().get(EspressoOptions.WhiteBoxAPI);
244246
internalJvmciEnabled = env.getOptions().get(EspressoOptions.EnableJVMCI);
247+
continuum = env.getOptions().get(EspressoOptions.Continuum);
245248

246249
EspressoOptions.GuestFieldOffsetStrategyEnum strategy = env.getOptions().get(EspressoOptions.GuestFieldOffsetStrategy);
247250
guestFieldOffsetStrategy = switch (strategy) {
@@ -339,6 +342,7 @@ protected boolean areOptionsCompatible(OptionValues oldOptions, OptionValues new
339342
isOptionCompatible(newOptions, oldOptions, EspressoOptions.EnablePreview) &&
340343
isOptionCompatible(newOptions, oldOptions, EspressoOptions.WhiteBoxAPI) &&
341344
isOptionCompatible(newOptions, oldOptions, EspressoOptions.EnableJVMCI) &&
345+
isOptionCompatible(newOptions, oldOptions, EspressoOptions.Continuum) &&
342346
isOptionCompatible(newOptions, oldOptions, EspressoOptions.GuestFieldOffsetStrategy) &&
343347
isOptionCompatible(newOptions, oldOptions, EspressoOptions.UseEspressoLibs);
344348
}
@@ -481,6 +485,14 @@ public StaticProperty getArrayProperty() {
481485
return arrayProperty;
482486
}
483487

488+
public StaticProperty getArrayHashCodeProperty() {
489+
if (!continuum) {
490+
CompilerDirectives.transferToInterpreterAndInvalidate();
491+
throw EspressoError.shouldNotReachHere("Accessing array hash code property without continuum set up.");
492+
}
493+
return arrayHashCodeProperty;
494+
}
495+
484496
public StaticShape<StaticObjectFactory> getArrayShape() {
485497
assert fullyInitialized : "Array shape accessed before language is fully initialized";
486498
return arrayShape;
@@ -489,7 +501,11 @@ public StaticShape<StaticObjectFactory> getArrayShape() {
489501
@TruffleBoundary
490502
private StaticShape<StaticObjectFactory> createArrayShape() {
491503
assert arrayShape == null;
492-
return StaticShape.newBuilder(this).property(arrayProperty, Object.class, true).build(StaticObject.class, StaticObjectFactory.class);
504+
StaticShape.Builder builder = StaticShape.newBuilder(this).property(arrayProperty, Object.class, true);
505+
if (continuum) {
506+
builder.property(arrayHashCodeProperty, int.class, false);
507+
}
508+
return builder.build(StaticObject.class, StaticObjectFactory.class);
493509
}
494510

495511
public StaticProperty getForeignProperty() {
@@ -570,6 +586,10 @@ public boolean useEspressoLibs() {
570586
return useEspressoLibs;
571587
}
572588

589+
public boolean isContinuumEnabled() {
590+
return continuum;
591+
}
592+
573593
public EspressoLanguageCache getLanguageCache() {
574594
return languageCache;
575595
}

‎espresso/src/com.oracle.truffle.espresso/src/com/oracle/truffle/espresso/descriptors/EspressoSymbols.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -928,6 +928,8 @@ public static class Names {
928928
public static final Symbol<Name> entrySet = SYMBOLS.putName("entrySet");
929929
public static final Symbol<Name> hasNext = SYMBOLS.putName("hasNext");
930930
public static final Symbol<Name> toArray = SYMBOLS.putName("toArray");
931+
// j.l.Object
932+
public static final Symbol<Name> HIDDEN_SYSTEM_IHASHCODE = SYMBOLS.putName("0HIDDEN_SYSTEM_IHASHCODE");
931933
// MemberName
932934
public static final Symbol<Name> HIDDEN_VMINDEX = SYMBOLS.putName("0HIDDEN_VMINDEX");
933935
public static final Symbol<Name> HIDDEN_VMTARGET = SYMBOLS.putName("0HIDDEN_VMTARGET");

‎espresso/src/com.oracle.truffle.espresso/src/com/oracle/truffle/espresso/impl/Klass.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,7 @@
8383
import com.oracle.truffle.espresso.meta.InteropKlassesDispatch;
8484
import com.oracle.truffle.espresso.meta.Meta;
8585
import com.oracle.truffle.espresso.meta.MetaUtil;
86+
import com.oracle.truffle.espresso.nodes.interop.IHashCodeNode;
8687
import com.oracle.truffle.espresso.nodes.interop.InteropUnwrapNode;
8788
import com.oracle.truffle.espresso.nodes.interop.InteropUnwrapNodeGen;
8889
import com.oracle.truffle.espresso.nodes.interop.LookupDeclaredMethod;
@@ -106,7 +107,6 @@
106107
import com.oracle.truffle.espresso.shared.meta.TypeAccess;
107108
import com.oracle.truffle.espresso.substitutions.JavaType;
108109
import com.oracle.truffle.espresso.vm.InterpreterToVM;
109-
import com.oracle.truffle.espresso.vm.VM;
110110

111111
@ExportLibrary(InteropLibrary.class)
112112
public abstract class Klass extends ContextAccessImpl implements KlassRef, TruffleObject, EspressoType, TypeAccess<Klass, Method, Field> {
@@ -607,11 +607,11 @@ static TriState doOther(@SuppressWarnings("unused") Klass receiver, @SuppressWar
607607
}
608608

609609
@ExportMessage
610-
int identityHashCode() {
610+
int identityHashCode(@Cached IHashCodeNode iHashCodeNode) {
611611
// In unit tests, Truffle performs additional sanity checks, this assert causes stack
612612
// overflow.
613613
// assert InteropLibrary.getUncached().hasIdentity(this);
614-
return VM.JVM_IHashCode(mirror(), null /*- path where language is needed is never reached through here. */);
614+
return iHashCodeNode.execute(mirror());
615615
}
616616

617617
// endregion ### Identity/hashCode

‎espresso/src/com.oracle.truffle.espresso/src/com/oracle/truffle/espresso/impl/LinkedKlassFieldLayout.java

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424

2525
import static com.oracle.truffle.espresso.classfile.Constants.ACC_FINAL;
2626
import static com.oracle.truffle.espresso.classfile.Constants.ACC_HIDDEN;
27+
import static com.oracle.truffle.espresso.classfile.Constants.ACC_VOLATILE;
2728
import static java.util.Map.entry;
2829

2930
import java.util.HashSet;
@@ -178,6 +179,9 @@ private static class HiddenField {
178179
private static final int NO_ADDITIONAL_FLAGS = 0;
179180
private static final HiddenField[] EMPTY = new HiddenField[0];
180181
private static final Map<Symbol<Type>, HiddenField[]> REGISTRY = Map.ofEntries(
182+
entry(Types.java_lang_Object, new HiddenField[]{
183+
new HiddenField(Names.HIDDEN_SYSTEM_IHASHCODE, Types._int, EspressoLanguage::isContinuumEnabled, ACC_VOLATILE),
184+
}),
181185
entry(Types.java_lang_invoke_MemberName, new HiddenField[]{
182186
new HiddenField(Names.HIDDEN_VMTARGET),
183187
new HiddenField(Names.HIDDEN_VMINDEX)
@@ -197,7 +201,6 @@ private static class HiddenField {
197201
// All references (including strong) get an extra hidden field, this
198202
// simplifies the code for weak/soft/phantom/final references.
199203
entry(Types.java_lang_ref_Reference, new HiddenField[]{
200-
201204
new HiddenField(Names.HIDDEN_HOST_REFERENCE)
202205
}),
203206
entry(Types.java_lang_Throwable, new HiddenField[]{

‎espresso/src/com.oracle.truffle.espresso/src/com/oracle/truffle/espresso/meta/Meta.java

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,6 +99,7 @@ public Meta(EspressoContext context) {
9999
// Object and Class (+ Class fields) must be initialized before all other classes in order
100100
// to eagerly create the guest Class instances.
101101
java_lang_Object = knownKlass(Types.java_lang_Object);
102+
HIDDEN_SYSTEM_IHASHCODE = context.getLanguage().isContinuumEnabled() ? java_lang_Object.requireHiddenField(Names.HIDDEN_SYSTEM_IHASHCODE) : null;
102103
// Cloneable must be loaded before Serializable.
103104
java_lang_Cloneable = knownKlass(Types.java_lang_Cloneable);
104105
java_lang_Class = knownKlass(Types.java_lang_Class);
@@ -1342,7 +1343,7 @@ public void postSystemInit() {
13421343
}
13431344

13441345
// Continuations
1345-
boolean continuumSupport = getContext().getEspressoEnv().Continuum;
1346+
boolean continuumSupport = getLanguage().isContinuumEnabled();
13461347
this.continuum = continuumSupport ? new ContinuumSupport() : null;
13471348
}
13481349

@@ -1375,6 +1376,11 @@ private DiffVersionLoadHelper diff() {
13751376

13761377
public final ObjectKlass java_lang_Object;
13771378
public final ArrayKlass java_lang_Object_array;
1379+
/*
1380+
* Though only used when Continuum is enabled, the hashcode is used during VM initialization, so
1381+
* it cannot be put in the ContinuumSupport object.
1382+
*/
1383+
public final Field HIDDEN_SYSTEM_IHASHCODE;
13781384

13791385
public final ObjectKlass java_lang_String;
13801386
public final ArrayKlass java_lang_String_array;

‎espresso/src/com.oracle.truffle.espresso/src/com/oracle/truffle/espresso/nodes/interop/IHashCodeNode.java

Lines changed: 45 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,45 @@
1+
/*
2+
* Copyright (c) 2025, 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+
package com.oracle.truffle.espresso.nodes.interop;
24+
25+
import com.oracle.truffle.api.dsl.Bind;
26+
import com.oracle.truffle.api.dsl.GenerateUncached;
27+
import com.oracle.truffle.api.dsl.Specialization;
28+
import com.oracle.truffle.espresso.EspressoLanguage;
29+
import com.oracle.truffle.espresso.nodes.EspressoNode;
30+
import com.oracle.truffle.espresso.runtime.EspressoContext;
31+
import com.oracle.truffle.espresso.runtime.staticobject.StaticObject;
32+
import com.oracle.truffle.espresso.vm.VM;
33+
34+
@GenerateUncached
35+
public abstract class IHashCodeNode extends EspressoNode {
36+
public abstract int execute(StaticObject obj);
37+
38+
@Specialization
39+
public static int doCached(StaticObject obj,
40+
@Bind("getLanguage()") EspressoLanguage lang,
41+
@Bind("getContext()") EspressoContext ctx) {
42+
assert !obj.isForeignObject();
43+
return VM.JVM_IHashCode(obj, ctx.getMeta(), lang);
44+
}
45+
}

‎espresso/src/com.oracle.truffle.espresso/src/com/oracle/truffle/espresso/runtime/EspressoEnv.java

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,6 @@ public final class EspressoEnv {
9292
public final boolean SoftExit;
9393
public final boolean AllowHostExit;
9494
public final boolean Polyglot;
95-
public final boolean Continuum;
9695
public final boolean BuiltInPolyglotCollections;
9796
public final boolean HotSwapAPI;
9897
public final boolean UseBindingsLoader;
@@ -169,7 +168,6 @@ public EspressoEnv(EspressoContext context, TruffleLanguage.Env env) {
169168
this.multiThreadingDisabled = multiThreadingDisabledReason;
170169
this.NativeAccessAllowed = env.isNativeAccessAllowed();
171170
this.Polyglot = env.getOptions().get(EspressoOptions.Polyglot);
172-
this.Continuum = env.getOptions().get(EspressoOptions.Continuum);
173171
this.HotSwapAPI = env.getOptions().get(EspressoOptions.HotSwapAPI);
174172
this.BuiltInPolyglotCollections = env.getOptions().get(EspressoOptions.BuiltInPolyglotCollections);
175173
this.polyglotTypeMappings = new PolyglotTypeMappings(env.getOptions().get(EspressoOptions.PolyglotInterfaceMappings), env.getOptions().get(EspressoOptions.PolyglotTypeConverters),

‎espresso/src/com.oracle.truffle.espresso/src/com/oracle/truffle/espresso/runtime/dispatch/staticobject/BaseInterop.java

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -41,10 +41,10 @@
4141
import com.oracle.truffle.espresso.impl.Klass;
4242
import com.oracle.truffle.espresso.meta.EspressoError;
4343
import com.oracle.truffle.espresso.meta.Meta;
44+
import com.oracle.truffle.espresso.nodes.interop.IHashCodeNode;
4445
import com.oracle.truffle.espresso.runtime.dispatch.messages.GenerateInteropNodes;
4546
import com.oracle.truffle.espresso.runtime.dispatch.messages.Shareable;
4647
import com.oracle.truffle.espresso.runtime.staticobject.StaticObject;
47-
import com.oracle.truffle.espresso.vm.VM;
4848

4949
/**
5050
* BaseInterop (isNull, is/asString, meta-instance, identity, exceptions, toDisplayString) Support
@@ -209,10 +209,14 @@ public static TriState isIdenticalOrUndefined(StaticObject receiver, Object othe
209209
}
210210

211211
@ExportMessage
212-
public static int identityHashCode(StaticObject object) {
212+
public static int identityHashCode(StaticObject object,
213+
@Cached IHashCodeNode iHashCodeNode) {
213214
object.checkNotForeign();
215+
if (StaticObject.isNull(object)) {
216+
return 0;
217+
}
214218
// Working with espresso objects here, guaranteed to have identity.
215-
return VM.JVM_IHashCode(object, null /*- path where language is needed is never reached through here. */);
219+
return iHashCodeNode.execute(object);
216220
}
217221

218222
// endregion ### Identity/hashCode

‎espresso/src/com.oracle.truffle.espresso/src/com/oracle/truffle/espresso/substitutions/ModuleExtension.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@
3737
*/
3838
public final class ModuleExtension {
3939
private static final ModuleExtension[] ESPRESSO_EXTENSION_MODULES = {
40-
new Builder("org.graalvm.continuations", "continuations.jar", (context) -> context.getEspressoEnv().Continuum).build(),
40+
new Builder("org.graalvm.continuations", "continuations.jar", (context) -> context.getLanguage().isContinuumEnabled()).build(),
4141
new Builder("espresso.hotswap", "hotswap.jar", (context) -> context.getEspressoEnv().JDWPOptions != null).build(),
4242
new Builder("espresso.polyglot", "espresso-polyglot.jar", (context) -> context.getEspressoEnv().Polyglot).build(),
4343
new Builder("jdk.graal.compiler.espresso", "espresso-graal.jar", (context) -> context.getLanguage().isInternalJVMCIEnabled()) //

‎espresso/src/com.oracle.truffle.espresso/src/com/oracle/truffle/espresso/substitutions/continuations/Target_java_lang_invoke_LambdaMetafactory.java

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232
import com.oracle.truffle.api.dsl.Cached;
3333
import com.oracle.truffle.api.dsl.Specialization;
3434
import com.oracle.truffle.api.nodes.DirectCallNode;
35+
import com.oracle.truffle.espresso.EspressoLanguage;
3536
import com.oracle.truffle.espresso.meta.Meta;
3637
import com.oracle.truffle.espresso.runtime.EspressoContext;
3738
import com.oracle.truffle.espresso.runtime.staticobject.StaticObject;
@@ -71,12 +72,12 @@ StaticObject doCached(
7172
@Bind("getMeta()") Meta meta,
7273
@Cached("create(meta.java_lang_invoke_LambdaMetafactory_altMetafactory.getCallTargetNoSubstitution())") DirectCallNode altMetafactory,
7374
@Cached("create(meta.java_lang_invoke_LambdaMetafactory_metafactory.getCallTargetNoSubstitution())") DirectCallNode original,
74-
@Bind("getContext()") EspressoContext context) {
75-
if (context.getEspressoEnv().Continuum) {
75+
@Bind("getLanguage()") EspressoLanguage lang) {
76+
if (lang.isContinuumEnabled()) {
7677
// altMetafactory has a curious calling convention, apparently designed for
7778
// extensibility.
78-
StaticObject extraArgsRef = context.getAllocator().createNewReferenceArray(meta.java_lang_Object, 4);
79-
StaticObject[] extraArgs = extraArgsRef.unwrap(context.getLanguage());
79+
StaticObject extraArgsRef = lang.getAllocator().createNewReferenceArray(meta.java_lang_Object, 4);
80+
StaticObject[] extraArgs = extraArgsRef.unwrap(lang);
8081
extraArgs[0] = interfaceMethodType;
8182
extraArgs[1] = implementation;
8283
extraArgs[2] = dynamicMethodType;
@@ -106,7 +107,7 @@ StaticObject doCached(
106107
@Bind("getMeta()") Meta meta,
107108
@Cached("create(meta.java_lang_invoke_LambdaMetafactory_altMetafactory.getCallTargetNoSubstitution())") DirectCallNode original,
108109
@Bind("getContext()") EspressoContext context) {
109-
if (context.getEspressoEnv().Continuum) {
110+
if (context.getLanguage().isContinuumEnabled()) {
110111
StaticObject[] extraArgs = args.unwrap(context.getLanguage());
111112
extraArgs[3] = meta.boxInteger(meta.unboxInteger(extraArgs[3]) | LambdaMetafactory.FLAG_SERIALIZABLE);
112113
return (StaticObject) original.call(caller, interfaceMethodName, factoryType, StaticObject.wrap(extraArgs, meta));

0 commit comments

Comments
 (0)