Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
using System;


namespace System.Reflection.Metadata.ApplyUpdate.Test
{
public class ReflectionAddNewMethod
{
public string ExistingMethod(string u, double f)
{
return u + f.ToString();;
}

}
}
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,21 @@
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
using System;
using System.Runtime.CompilerServices;
using CancellationToken = System.Threading.CancellationToken;

namespace System.Reflection.Metadata.ApplyUpdate.Test
{
public class ReflectionAddNewMethod
{
public string ExistingMethod(string u, double f)
{
return u + f.ToString();;
}

public double AddedNewMethod(char c, float h, string w, CancellationToken ct = default, [CallerMemberName] string callerName = "")
{
return ((double)Convert.ToInt32(c)) + h;
}
}
}
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
<Project Sdk="Microsoft.NET.Sdk">
<PropertyGroup>
<RootNamespace>System.Runtime.Loader.Tests</RootNamespace>
<TargetFramework>$(NetCoreAppCurrent)</TargetFramework>
<TestRuntime>true</TestRuntime>
<DeltaScript>deltascript.json</DeltaScript>
</PropertyGroup>
<ItemGroup>
<Compile Include="ReflectionAddNewMethod.cs" />
</ItemGroup>
</Project>
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
{
"changes": [
{"document": "ReflectionAddNewMethod.cs", "update": "ReflectionAddNewMethod_v1.cs"},
]
}

83 changes: 83 additions & 0 deletions src/libraries/System.Runtime.Loader/tests/ApplyUpdateTest.cs
Original file line numberDiff line numberDiff line change
Expand Up@@ -3,6 +3,7 @@

using System.Reflection;
using System.Runtime.CompilerServices;
using System.Runtime.InteropServices;
using Xunit;

namespace System.Reflection.Metadata
Expand DownExpand Up@@ -629,5 +630,87 @@ public static void TestReflectionAddNewType()
System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewType.ZExistingClass.ExistingMethod ();
});
}

[ConditionalFact(typeof(ApplyUpdateUtil), nameof(ApplyUpdateUtil.IsSupported))]
public static void TestReflectionAddNewMethod()
{
ApplyUpdateUtil.TestCase(static () =>
{
var ty = typeof(System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod);
var assm = ty.Assembly;

var bindingFlags = BindingFlags.Instance | BindingFlags.Public;
var allMethods = ty.GetMethods(bindingFlags);

int objectMethods = typeof(object).GetMethods(bindingFlags).Length;
Assert.Equal (objectMethods + 1, allMethods.Length);

ApplyUpdateUtil.ApplyUpdate(assm);
ApplyUpdateUtil.ClearAllReflectionCaches();

ty = typeof(System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod);

allMethods = ty.GetMethods(bindingFlags);
Assert.Equal (objectMethods + 2, allMethods.Length);

var mi = ty.GetMethod ("AddedNewMethod");

Assert.NotNull (mi);

var retParm = mi.ReturnParameter;
Assert.NotNull (retParm);
Assert.NotNull (retParm.ParameterType);
Assert.Equal (-1, retParm.Position);

var retCas = retParm.GetCustomAttributes(false);
Assert.NotNull(retCas);
Assert.Equal(0, retCas.Length);

var parms = mi.GetParameters();
Assert.Equal (5, parms.Length);

int parmPos = 0;
foreach (var parm in parms)
{
Assert.NotNull(parm);
Assert.NotNull(parm.ParameterType);
Assert.Equal(parmPos, parm.Position);
Assert.NotNull(parm.Name);

var cas = parm.GetCustomAttributes(false);
foreach (var ca in cas) {
Assert.NotNull (ca);
}

parmPos++;
}

var parmAttrs = parms[4].GetCustomAttributes(false);
Assert.Equal (2, parmAttrs.Length);
bool foundCallerMemberName = false;
bool foundOptional = false;
foreach (var pa in parmAttrs) {
if (typeof (CallerMemberNameAttribute).Equals(pa.GetType()))
{
foundCallerMemberName = true;
}
if (typeof (OptionalAttribute).Equals(pa.GetType()))
{
foundOptional = true;
}
}
Assert.True(foundCallerMemberName);
Assert.True(foundOptional);

// n.b. this typeof() also makes the rest of the test work on Wasm with aggressive trimming.
Assert.Equal (typeof(System.Threading.CancellationToken), parms[3].ParameterType);

Assert.True(parms[3].HasDefaultValue);
Assert.True(parms[4].HasDefaultValue);

Assert.Null(parms[3].DefaultValue);
Assert.Equal(string.Empty, parms[4].DefaultValue);
});
}
}
}
Original file line numberDiff line numberDiff line change
Expand Up@@ -62,6 +62,7 @@
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.AddStaticLambda\System.Reflection.Metadata.ApplyUpdate.Test.AddStaticLambda.csproj" />
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.StaticLambdaRegression\System.Reflection.Metadata.ApplyUpdate.Test.StaticLambdaRegression.csproj" />
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewType\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewType.csproj" />
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod.csproj" />
</ItemGroup>
<ItemGroup Condition="'$(TargetOS)' == 'Browser'">
<WasmFilesToIncludeFromPublishDir Include="$(AssemblyName).dll" />
Expand Down
5 changes: 5 additions & 0 deletions src/mono/mono/component/hot_reload-internals.h
Original file line numberDiff line numberDiff line change
Expand Up@@ -86,4 +86,9 @@ typedef struct _MonoClassMetadataUpdateEvent {
uint32_t token; /* the Event table token where this event was defined. */
} MonoClassMetadataUpdateEvent;

typedef struct _MonoMethodMetadataUpdateParamInfo {
uint32_t first_param_token; /* a Param token */
uint32_t param_count;
} MonoMethodMetadataUpdateParamInfo;

#endif/*_MONO_COMPONENT_HOT_RELOAD_INTERNALS_H*/
12 changes: 11 additions & 1 deletion src/mono/mono/component/hot_reload-stub.c
Original file line numberDiff line numberDiff line change
Expand Up@@ -110,6 +110,9 @@ hot_reload_get_num_methods_added (MonoClass *klass);
static const char *
hot_reload_get_capabilities (void);

static uint32_t
hot_reload_stub_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt);

static MonoComponentHotReload fn_table = {
{ MONO_COMPONENT_ITF_VERSION, &hot_reload_stub_available },
&hot_reload_stub_set_fastpath_data,
Expand DownExpand Up@@ -142,7 +145,8 @@ static MonoComponentHotReload fn_table = {
&hot_reload_stub_added_fields_iter,
&hot_reload_get_num_fields_added,
&hot_reload_get_num_methods_added,
&hot_reload_get_capabilities
&hot_reload_get_capabilities,
&hot_reload_stub_get_method_params,
};

static bool
Expand DownExpand Up@@ -343,6 +347,12 @@ hot_reload_get_capabilities (void)
return "";
}

static uint32_t
hot_reload_stub_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt)
{
return 0;
}

MONO_COMPONENT_EXPORT_ENTRYPOINT
MonoComponentHotReload *
mono_component_hot_reload_init (void)
Expand Down
91 changes: 87 additions & 4 deletions src/mono/mono/component/hot_reload.c
Original file line numberDiff line numberDiff line change
Expand Up@@ -144,6 +144,9 @@ hot_reload_get_num_methods_added (MonoClass *klass);
static const char *
hot_reload_get_capabilities (void);

static uint32_t
hot_reload_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt);

static MonoClassMetadataUpdateField *
metadata_update_field_setup_basic_info (MonoImage *image_base, BaselineInfo *base_info, uint32_t generation, DeltaInfo *delta_info, MonoClass *parent_klass, uint32_t fielddef_token, uint32_t field_flags);

Expand DownExpand Up@@ -179,7 +182,8 @@ static MonoComponentHotReload fn_table = {
&hot_reload_added_fields_iter,
&hot_reload_get_num_fields_added,
&hot_reload_get_num_methods_added,
&hot_reload_get_capabilities
&hot_reload_get_capabilities,
&hot_reload_get_method_params,
};

MonoComponentHotReload *
Expand DownExpand Up@@ -272,6 +276,9 @@ struct _BaselineInfo {
/* Parents for added methods, fields, etc */
GHashTable *member_parent; /* maps added methoddef or fielddef tokens to typedef tokens */

/* Params for added methods */
GHashTable *method_params; /* maps methoddef tokens to a MonoClassMetadataUpdateMethodParamInfo* */

/* Skeletons for all newly-added types from every generation. Accessing the array requires the image lock. */
GArray *skeletons;
};
Expand DownExpand Up@@ -412,6 +419,12 @@ baseline_info_destroy (BaselineInfo *info)
if (info->skeletons)
g_array_free (info->skeletons, TRUE);

if (info->member_parent)
g_hash_table_destroy (info->member_parent);

if (info->method_params)
g_hash_table_destroy (info->method_params);

g_free (info);
}

Expand DownExpand Up@@ -627,6 +640,11 @@ add_method_to_baseline (BaselineInfo *base_info, DeltaInfo *delta_info, MonoClas
static void
add_field_to_baseline (BaselineInfo *base_info, DeltaInfo *delta_info, MonoClass *klass, uint32_t field_token);

/* Add a method->params lookup for new methods in existing classes */
static void
add_param_info_for_method (BaselineInfo *base_info, uint32_t param_token, uint32_t method_token);


void
hot_reload_init (void)
{
Expand DownExpand Up@@ -2019,6 +2037,7 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base
uint32_t add_member_typedef = 0;
uint32_t add_property_propertymap = 0;
uint32_t add_event_eventmap = 0;
uint32_t add_field_method = 0;

gboolean assemblyref_updated = FALSE;
for (guint32 i = 0; i < rows ; ++i) {
Expand DownExpand Up@@ -2049,6 +2068,7 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base

case ENC_FUNC_ADD_PARAM: {
g_assert (token_table == MONO_TABLE_METHOD);
add_field_method = log_token;
break;
}
case ENC_FUNC_ADD_FIELD: {
Expand DownExpand Up@@ -2115,8 +2135,11 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base
}
case MONO_TABLE_METHOD: {
/* if adding a param, handle it with the next record */
if (func_code == ENC_FUNC_ADD_PARAM)
if (func_code == ENC_FUNC_ADD_PARAM) {
g_assert (is_addition);
break;
}
g_assert (func_code == ENC_FUNC_DEFAULT);

if (!base_info->method_table_update)
base_info->method_table_update = g_hash_table_new (g_direct_hash, g_direct_equal);
Expand DownExpand Up@@ -2366,9 +2389,21 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base
*
* So by the time we see the param additions, the methods are already in.
*
* FIXME: we need a lookaside table (like member_parent) for every place
* that looks at MONO_METHOD_PARAMLIST
*/
if (is_addition) {
g_assert (add_field_method != 0);
uint32_t parent_type_token = hot_reload_method_parent (image_base, add_field_method);
g_assert (parent_type_token != 0); // we added a parameter to a method that was added
if (pass2_context_is_skeleton (ctx, parent_type_token)) {
// it's a parameter on a new method in a brand new class
// FIXME: need to do something here?
} else {
// it's a parameter on a new method in an existing class
add_param_info_for_method (base_info, log_token, add_field_method);
}
add_field_method = 0;
break;
}
break;
}
case MONO_TABLE_INTERFACEIMPL: {
Expand DownExpand Up@@ -2819,6 +2854,28 @@ hot_reload_field_parent (MonoImage *base_image, uint32_t field_token)
}


static void
add_param_info_for_method (BaselineInfo *base_info, uint32_t param_token, uint32_t method_token)
{
if (!base_info->method_params) {
base_info->method_params = g_hash_table_new_full (g_direct_hash, g_direct_equal, NULL, g_free);
}
MonoMethodMetadataUpdateParamInfo* info = NULL;
info = g_hash_table_lookup (base_info->method_params, GUINT_TO_POINTER (method_token));
if (!info) {
// FIXME locking
info = g_new0 (MonoMethodMetadataUpdateParamInfo, 1);
g_hash_table_insert (base_info->method_params, GUINT_TO_POINTER (method_token), info);
info->first_param_token = param_token;
info->param_count = 1;
} else {
uint32_t param_index = mono_metadata_token_index (param_token);
// expect params for a single method to be a contiguous sequence of rows
g_assert (mono_metadata_token_index (info->first_param_token) + info->param_count == param_index);
info->param_count++;
}
}

/* HACK - keep in sync with locator_t in metadata/metadata.c */
typedef struct {
guint32 idx; /* The index that we are trying to locate */
Expand DownExpand Up@@ -3148,6 +3205,32 @@ hot_reload_get_num_methods_added (MonoClass *klass)
return count;
}

static uint32_t
hot_reload_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt)
{
BaselineInfo *base_info = baseline_info_lookup (base_image);
g_assert (base_info);

/* FIXME: locking in case the hash table grows */

if (!base_info->method_params)
return 0;

MonoMethodMetadataUpdateParamInfo* info = NULL;
info = g_hash_table_lookup (base_info->method_params, GUINT_TO_POINTER (methoddef_token));
if (!info) {
if (out_param_count_opt)
*out_param_count_opt = 0;
return 0;
}

if (out_param_count_opt)
*out_param_count_opt = info->param_count;

return mono_metadata_token_index (info->first_param_token);
}


static const char *
hot_reload_get_capabilities (void)
{
Expand Down
1 change: 1 addition & 0 deletions src/mono/mono/component/hot_reload.h
Original file line numberDiff line numberDiff line change
Expand Up@@ -48,6 +48,7 @@ typedef struct _MonoComponentHotReload {
uint32_t (*get_num_fields_added) (MonoClass *klass);
uint32_t (*get_num_methods_added) (MonoClass *klass);
const char* (*get_capabilities) (void);
uint32_t (*get_method_params) (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt);
} MonoComponentHotReload;

MONO_COMPONENT_EXPORT_ENTRYPOINT
Expand Down
Loading
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Add copy buttons to all
 blocks\n(function() {\n function addCopyButtons() {\n document.querySelectorAll('pre code').forEach(function(codeBlock) {\n if (codeBlock.parentElement.hasAttribute('data-copy-added')) return;\n codeBlock.parentElement.setAttribute('data-copy-added', 'true');\n \n var btn = document.createElement('button');\n btn.textContent = 'Copy';\n btn.style.cssText = 'position:absolute;top:4px;right:4px;padding:2px 8px;font-size:11px;background:#4ecdc4;border:none;border-radius:4px;color:#1a1a2e;cursor:pointer;opacity:0.7;transition:opacity 0.2s;';\n btn.onmouseover = function() { this.style.opacity = '1'; };\n btn.onmouseout = function() { this.style.opacity = '0.7'; };\n btn.onclick = function() {\n navigator.clipboard.writeText(codeBlock.textContent).then(function() {\n btn.textContent = 'Copied!';\n setTimeout(function() { btn.textContent = 'Copy'; }, 1500);\n });\n };\n codeBlock.parentElement.style.position = 'relative';\n codeBlock.parentElement.appendChild(btn);\n });\n }\n \n addCopyButtons();\n \n // Re-run on dynamic content\n var observer = new MutationObserver(addCopyButtons);\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Add Copy Buttons to Code Blocks");
}
} catch(__e) { console.warn('[Userscript:Add Copy Buttons to Code Blocks]', __e); }
})();
(function(){
try {
var __m = "github.com";
var __re = new RegExp('^' + "github\\.com" + '
Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
using System;


namespace System.Reflection.Metadata.ApplyUpdate.Test
{
public class ReflectionAddNewMethod
{
public string ExistingMethod(string u, double f)
{
return u + f.ToString();;
}

}
}
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,21 @@
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
using System;
using System.Runtime.CompilerServices;
using CancellationToken = System.Threading.CancellationToken;

namespace System.Reflection.Metadata.ApplyUpdate.Test
{
public class ReflectionAddNewMethod
{
public string ExistingMethod(string u, double f)
{
return u + f.ToString();;
}

public double AddedNewMethod(char c, float h, string w, CancellationToken ct = default, [CallerMemberName] string callerName = "")
{
return ((double)Convert.ToInt32(c)) + h;
}
}
}
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
<Project Sdk="Microsoft.NET.Sdk">
<PropertyGroup>
<RootNamespace>System.Runtime.Loader.Tests</RootNamespace>
<TargetFramework>$(NetCoreAppCurrent)</TargetFramework>
<TestRuntime>true</TestRuntime>
<DeltaScript>deltascript.json</DeltaScript>
</PropertyGroup>
<ItemGroup>
<Compile Include="ReflectionAddNewMethod.cs" />
</ItemGroup>
</Project>
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
{
"changes": [
{"document": "ReflectionAddNewMethod.cs", "update": "ReflectionAddNewMethod_v1.cs"},
]
}

83 changes: 83 additions & 0 deletions src/libraries/System.Runtime.Loader/tests/ApplyUpdateTest.cs
Original file line numberDiff line numberDiff line change
Expand Up@@ -3,6 +3,7 @@

using System.Reflection;
using System.Runtime.CompilerServices;
using System.Runtime.InteropServices;
using Xunit;

namespace System.Reflection.Metadata
Expand DownExpand Up@@ -629,5 +630,87 @@ public static void TestReflectionAddNewType()
System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewType.ZExistingClass.ExistingMethod ();
});
}

[ConditionalFact(typeof(ApplyUpdateUtil), nameof(ApplyUpdateUtil.IsSupported))]
public static void TestReflectionAddNewMethod()
{
ApplyUpdateUtil.TestCase(static () =>
{
var ty = typeof(System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod);
var assm = ty.Assembly;

var bindingFlags = BindingFlags.Instance | BindingFlags.Public;
var allMethods = ty.GetMethods(bindingFlags);

int objectMethods = typeof(object).GetMethods(bindingFlags).Length;
Assert.Equal (objectMethods + 1, allMethods.Length);

ApplyUpdateUtil.ApplyUpdate(assm);
ApplyUpdateUtil.ClearAllReflectionCaches();

ty = typeof(System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod);

allMethods = ty.GetMethods(bindingFlags);
Assert.Equal (objectMethods + 2, allMethods.Length);

var mi = ty.GetMethod ("AddedNewMethod");

Assert.NotNull (mi);

var retParm = mi.ReturnParameter;
Assert.NotNull (retParm);
Assert.NotNull (retParm.ParameterType);
Assert.Equal (-1, retParm.Position);

var retCas = retParm.GetCustomAttributes(false);
Assert.NotNull(retCas);
Assert.Equal(0, retCas.Length);

var parms = mi.GetParameters();
Assert.Equal (5, parms.Length);

int parmPos = 0;
foreach (var parm in parms)
{
Assert.NotNull(parm);
Assert.NotNull(parm.ParameterType);
Assert.Equal(parmPos, parm.Position);
Assert.NotNull(parm.Name);

var cas = parm.GetCustomAttributes(false);
foreach (var ca in cas) {
Assert.NotNull (ca);
}

parmPos++;
}

var parmAttrs = parms[4].GetCustomAttributes(false);
Assert.Equal (2, parmAttrs.Length);
bool foundCallerMemberName = false;
bool foundOptional = false;
foreach (var pa in parmAttrs) {
if (typeof (CallerMemberNameAttribute).Equals(pa.GetType()))
{
foundCallerMemberName = true;
}
if (typeof (OptionalAttribute).Equals(pa.GetType()))
{
foundOptional = true;
}
}
Assert.True(foundCallerMemberName);
Assert.True(foundOptional);

// n.b. this typeof() also makes the rest of the test work on Wasm with aggressive trimming.
Assert.Equal (typeof(System.Threading.CancellationToken), parms[3].ParameterType);

Assert.True(parms[3].HasDefaultValue);
Assert.True(parms[4].HasDefaultValue);

Assert.Null(parms[3].DefaultValue);
Assert.Equal(string.Empty, parms[4].DefaultValue);
});
}
}
}
Original file line numberDiff line numberDiff line change
Expand Up@@ -62,6 +62,7 @@
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.AddStaticLambda\System.Reflection.Metadata.ApplyUpdate.Test.AddStaticLambda.csproj" />
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.StaticLambdaRegression\System.Reflection.Metadata.ApplyUpdate.Test.StaticLambdaRegression.csproj" />
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewType\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewType.csproj" />
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod.csproj" />
</ItemGroup>
<ItemGroup Condition="'$(TargetOS)' == 'Browser'">
<WasmFilesToIncludeFromPublishDir Include="$(AssemblyName).dll" />
Expand Down
5 changes: 5 additions & 0 deletions src/mono/mono/component/hot_reload-internals.h
Original file line numberDiff line numberDiff line change
Expand Up@@ -86,4 +86,9 @@ typedef struct _MonoClassMetadataUpdateEvent {
uint32_t token; /* the Event table token where this event was defined. */
} MonoClassMetadataUpdateEvent;

typedef struct _MonoMethodMetadataUpdateParamInfo {
uint32_t first_param_token; /* a Param token */
uint32_t param_count;
} MonoMethodMetadataUpdateParamInfo;

#endif/*_MONO_COMPONENT_HOT_RELOAD_INTERNALS_H*/
12 changes: 11 additions & 1 deletion src/mono/mono/component/hot_reload-stub.c
Original file line numberDiff line numberDiff line change
Expand Up@@ -110,6 +110,9 @@ hot_reload_get_num_methods_added (MonoClass *klass);
static const char *
hot_reload_get_capabilities (void);

static uint32_t
hot_reload_stub_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt);

static MonoComponentHotReload fn_table = {
{ MONO_COMPONENT_ITF_VERSION, &hot_reload_stub_available },
&hot_reload_stub_set_fastpath_data,
Expand DownExpand Up@@ -142,7 +145,8 @@ static MonoComponentHotReload fn_table = {
&hot_reload_stub_added_fields_iter,
&hot_reload_get_num_fields_added,
&hot_reload_get_num_methods_added,
&hot_reload_get_capabilities
&hot_reload_get_capabilities,
&hot_reload_stub_get_method_params,
};

static bool
Expand DownExpand Up@@ -343,6 +347,12 @@ hot_reload_get_capabilities (void)
return "";
}

static uint32_t
hot_reload_stub_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt)
{
return 0;
}

MONO_COMPONENT_EXPORT_ENTRYPOINT
MonoComponentHotReload *
mono_component_hot_reload_init (void)
Expand Down
91 changes: 87 additions & 4 deletions src/mono/mono/component/hot_reload.c
Original file line numberDiff line numberDiff line change
Expand Up@@ -144,6 +144,9 @@ hot_reload_get_num_methods_added (MonoClass *klass);
static const char *
hot_reload_get_capabilities (void);

static uint32_t
hot_reload_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt);

static MonoClassMetadataUpdateField *
metadata_update_field_setup_basic_info (MonoImage *image_base, BaselineInfo *base_info, uint32_t generation, DeltaInfo *delta_info, MonoClass *parent_klass, uint32_t fielddef_token, uint32_t field_flags);

Expand DownExpand Up@@ -179,7 +182,8 @@ static MonoComponentHotReload fn_table = {
&hot_reload_added_fields_iter,
&hot_reload_get_num_fields_added,
&hot_reload_get_num_methods_added,
&hot_reload_get_capabilities
&hot_reload_get_capabilities,
&hot_reload_get_method_params,
};

MonoComponentHotReload *
Expand DownExpand Up@@ -272,6 +276,9 @@ struct _BaselineInfo {
/* Parents for added methods, fields, etc */
GHashTable *member_parent; /* maps added methoddef or fielddef tokens to typedef tokens */

/* Params for added methods */
GHashTable *method_params; /* maps methoddef tokens to a MonoClassMetadataUpdateMethodParamInfo* */

/* Skeletons for all newly-added types from every generation. Accessing the array requires the image lock. */
GArray *skeletons;
};
Expand DownExpand Up@@ -412,6 +419,12 @@ baseline_info_destroy (BaselineInfo *info)
if (info->skeletons)
g_array_free (info->skeletons, TRUE);

if (info->member_parent)
g_hash_table_destroy (info->member_parent);

if (info->method_params)
g_hash_table_destroy (info->method_params);

g_free (info);
}

Expand DownExpand Up@@ -627,6 +640,11 @@ add_method_to_baseline (BaselineInfo *base_info, DeltaInfo *delta_info, MonoClas
static void
add_field_to_baseline (BaselineInfo *base_info, DeltaInfo *delta_info, MonoClass *klass, uint32_t field_token);

/* Add a method->params lookup for new methods in existing classes */
static void
add_param_info_for_method (BaselineInfo *base_info, uint32_t param_token, uint32_t method_token);


void
hot_reload_init (void)
{
Expand DownExpand Up@@ -2019,6 +2037,7 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base
uint32_t add_member_typedef = 0;
uint32_t add_property_propertymap = 0;
uint32_t add_event_eventmap = 0;
uint32_t add_field_method = 0;

gboolean assemblyref_updated = FALSE;
for (guint32 i = 0; i < rows ; ++i) {
Expand DownExpand Up@@ -2049,6 +2068,7 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base

case ENC_FUNC_ADD_PARAM: {
g_assert (token_table == MONO_TABLE_METHOD);
add_field_method = log_token;
break;
}
case ENC_FUNC_ADD_FIELD: {
Expand DownExpand Up@@ -2115,8 +2135,11 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base
}
case MONO_TABLE_METHOD: {
/* if adding a param, handle it with the next record */
if (func_code == ENC_FUNC_ADD_PARAM)
if (func_code == ENC_FUNC_ADD_PARAM) {
g_assert (is_addition);
break;
}
g_assert (func_code == ENC_FUNC_DEFAULT);

if (!base_info->method_table_update)
base_info->method_table_update = g_hash_table_new (g_direct_hash, g_direct_equal);
Expand DownExpand Up@@ -2366,9 +2389,21 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base
*
* So by the time we see the param additions, the methods are already in.
*
* FIXME: we need a lookaside table (like member_parent) for every place
* that looks at MONO_METHOD_PARAMLIST
*/
if (is_addition) {
g_assert (add_field_method != 0);
uint32_t parent_type_token = hot_reload_method_parent (image_base, add_field_method);
g_assert (parent_type_token != 0); // we added a parameter to a method that was added
if (pass2_context_is_skeleton (ctx, parent_type_token)) {
// it's a parameter on a new method in a brand new class
// FIXME: need to do something here?
} else {
// it's a parameter on a new method in an existing class
add_param_info_for_method (base_info, log_token, add_field_method);
}
add_field_method = 0;
break;
}
break;
}
case MONO_TABLE_INTERFACEIMPL: {
Expand DownExpand Up@@ -2819,6 +2854,28 @@ hot_reload_field_parent (MonoImage *base_image, uint32_t field_token)
}


static void
add_param_info_for_method (BaselineInfo *base_info, uint32_t param_token, uint32_t method_token)
{
if (!base_info->method_params) {
base_info->method_params = g_hash_table_new_full (g_direct_hash, g_direct_equal, NULL, g_free);
}
MonoMethodMetadataUpdateParamInfo* info = NULL;
info = g_hash_table_lookup (base_info->method_params, GUINT_TO_POINTER (method_token));
if (!info) {
// FIXME locking
info = g_new0 (MonoMethodMetadataUpdateParamInfo, 1);
g_hash_table_insert (base_info->method_params, GUINT_TO_POINTER (method_token), info);
info->first_param_token = param_token;
info->param_count = 1;
} else {
uint32_t param_index = mono_metadata_token_index (param_token);
// expect params for a single method to be a contiguous sequence of rows
g_assert (mono_metadata_token_index (info->first_param_token) + info->param_count == param_index);
info->param_count++;
}
}

/* HACK - keep in sync with locator_t in metadata/metadata.c */
typedef struct {
guint32 idx; /* The index that we are trying to locate */
Expand DownExpand Up@@ -3148,6 +3205,32 @@ hot_reload_get_num_methods_added (MonoClass *klass)
return count;
}

static uint32_t
hot_reload_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt)
{
BaselineInfo *base_info = baseline_info_lookup (base_image);
g_assert (base_info);

/* FIXME: locking in case the hash table grows */

if (!base_info->method_params)
return 0;

MonoMethodMetadataUpdateParamInfo* info = NULL;
info = g_hash_table_lookup (base_info->method_params, GUINT_TO_POINTER (methoddef_token));
if (!info) {
if (out_param_count_opt)
*out_param_count_opt = 0;
return 0;
}

if (out_param_count_opt)
*out_param_count_opt = info->param_count;

return mono_metadata_token_index (info->first_param_token);
}


static const char *
hot_reload_get_capabilities (void)
{
Expand Down
1 change: 1 addition & 0 deletions src/mono/mono/component/hot_reload.h
Original file line numberDiff line numberDiff line change
Expand Up@@ -48,6 +48,7 @@ typedef struct _MonoComponentHotReload {
uint32_t (*get_num_fields_added) (MonoClass *klass);
uint32_t (*get_num_methods_added) (MonoClass *klass);
const char* (*get_capabilities) (void);
uint32_t (*get_method_params) (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt);
} MonoComponentHotReload;

MONO_COMPONENT_EXPORT_ENTRYPOINT
Expand Down
Loading
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Force GitHub README to respect dark mode\n(function() {\n var style = document.createElement('style');\n style.textContent = '\n .markdown-body {\n color-scheme: dark light;\n }\n .markdown-body pre { background: #161b22 !important; }\n .markdown-body code { background: rgba(110, 118, 129, 0.4) !important; }\n .markdown-body table th, .markdown-body table td { border-color: #30363d !important; }\n .markdown-body img { background: #0d1117; }\n .markdown-body blockquote { border-left-color: #8b949e; }\n .markdown-body hr { border-color: #30363d; }\n ';\n document.head.appendChild(style);\n})();", "GitHub Dark Mode README Fix"); } } catch(__e) { console.warn('[Userscript:GitHub Dark Mode README Fix]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
using System;


namespace System.Reflection.Metadata.ApplyUpdate.Test
{
public class ReflectionAddNewMethod
{
public string ExistingMethod(string u, double f)
{
return u + f.ToString();;
}

}
}
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,21 @@
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
using System;
using System.Runtime.CompilerServices;
using CancellationToken = System.Threading.CancellationToken;

namespace System.Reflection.Metadata.ApplyUpdate.Test
{
public class ReflectionAddNewMethod
{
public string ExistingMethod(string u, double f)
{
return u + f.ToString();;
}

public double AddedNewMethod(char c, float h, string w, CancellationToken ct = default, [CallerMemberName] string callerName = "")
{
return ((double)Convert.ToInt32(c)) + h;
}
}
}
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
<Project Sdk="Microsoft.NET.Sdk">
<PropertyGroup>
<RootNamespace>System.Runtime.Loader.Tests</RootNamespace>
<TargetFramework>$(NetCoreAppCurrent)</TargetFramework>
<TestRuntime>true</TestRuntime>
<DeltaScript>deltascript.json</DeltaScript>
</PropertyGroup>
<ItemGroup>
<Compile Include="ReflectionAddNewMethod.cs" />
</ItemGroup>
</Project>
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
{
"changes": [
{"document": "ReflectionAddNewMethod.cs", "update": "ReflectionAddNewMethod_v1.cs"},
]
}

83 changes: 83 additions & 0 deletions src/libraries/System.Runtime.Loader/tests/ApplyUpdateTest.cs
Original file line numberDiff line numberDiff line change
Expand Up@@ -3,6 +3,7 @@

using System.Reflection;
using System.Runtime.CompilerServices;
using System.Runtime.InteropServices;
using Xunit;

namespace System.Reflection.Metadata
Expand DownExpand Up@@ -629,5 +630,87 @@ public static void TestReflectionAddNewType()
System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewType.ZExistingClass.ExistingMethod ();
});
}

[ConditionalFact(typeof(ApplyUpdateUtil), nameof(ApplyUpdateUtil.IsSupported))]
public static void TestReflectionAddNewMethod()
{
ApplyUpdateUtil.TestCase(static () =>
{
var ty = typeof(System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod);
var assm = ty.Assembly;

var bindingFlags = BindingFlags.Instance | BindingFlags.Public;
var allMethods = ty.GetMethods(bindingFlags);

int objectMethods = typeof(object).GetMethods(bindingFlags).Length;
Assert.Equal (objectMethods + 1, allMethods.Length);

ApplyUpdateUtil.ApplyUpdate(assm);
ApplyUpdateUtil.ClearAllReflectionCaches();

ty = typeof(System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod);

allMethods = ty.GetMethods(bindingFlags);
Assert.Equal (objectMethods + 2, allMethods.Length);

var mi = ty.GetMethod ("AddedNewMethod");

Assert.NotNull (mi);

var retParm = mi.ReturnParameter;
Assert.NotNull (retParm);
Assert.NotNull (retParm.ParameterType);
Assert.Equal (-1, retParm.Position);

var retCas = retParm.GetCustomAttributes(false);
Assert.NotNull(retCas);
Assert.Equal(0, retCas.Length);

var parms = mi.GetParameters();
Assert.Equal (5, parms.Length);

int parmPos = 0;
foreach (var parm in parms)
{
Assert.NotNull(parm);
Assert.NotNull(parm.ParameterType);
Assert.Equal(parmPos, parm.Position);
Assert.NotNull(parm.Name);

var cas = parm.GetCustomAttributes(false);
foreach (var ca in cas) {
Assert.NotNull (ca);
}

parmPos++;
}

var parmAttrs = parms[4].GetCustomAttributes(false);
Assert.Equal (2, parmAttrs.Length);
bool foundCallerMemberName = false;
bool foundOptional = false;
foreach (var pa in parmAttrs) {
if (typeof (CallerMemberNameAttribute).Equals(pa.GetType()))
{
foundCallerMemberName = true;
}
if (typeof (OptionalAttribute).Equals(pa.GetType()))
{
foundOptional = true;
}
}
Assert.True(foundCallerMemberName);
Assert.True(foundOptional);

// n.b. this typeof() also makes the rest of the test work on Wasm with aggressive trimming.
Assert.Equal (typeof(System.Threading.CancellationToken), parms[3].ParameterType);

Assert.True(parms[3].HasDefaultValue);
Assert.True(parms[4].HasDefaultValue);

Assert.Null(parms[3].DefaultValue);
Assert.Equal(string.Empty, parms[4].DefaultValue);
});
}
}
}
Original file line numberDiff line numberDiff line change
Expand Up@@ -62,6 +62,7 @@
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.AddStaticLambda\System.Reflection.Metadata.ApplyUpdate.Test.AddStaticLambda.csproj" />
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.StaticLambdaRegression\System.Reflection.Metadata.ApplyUpdate.Test.StaticLambdaRegression.csproj" />
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewType\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewType.csproj" />
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod.csproj" />
</ItemGroup>
<ItemGroup Condition="'$(TargetOS)' == 'Browser'">
<WasmFilesToIncludeFromPublishDir Include="$(AssemblyName).dll" />
Expand Down
5 changes: 5 additions & 0 deletions src/mono/mono/component/hot_reload-internals.h
Original file line numberDiff line numberDiff line change
Expand Up@@ -86,4 +86,9 @@ typedef struct _MonoClassMetadataUpdateEvent {
uint32_t token; /* the Event table token where this event was defined. */
} MonoClassMetadataUpdateEvent;

typedef struct _MonoMethodMetadataUpdateParamInfo {
uint32_t first_param_token; /* a Param token */
uint32_t param_count;
} MonoMethodMetadataUpdateParamInfo;

#endif/*_MONO_COMPONENT_HOT_RELOAD_INTERNALS_H*/
12 changes: 11 additions & 1 deletion src/mono/mono/component/hot_reload-stub.c
Original file line numberDiff line numberDiff line change
Expand Up@@ -110,6 +110,9 @@ hot_reload_get_num_methods_added (MonoClass *klass);
static const char *
hot_reload_get_capabilities (void);

static uint32_t
hot_reload_stub_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt);

static MonoComponentHotReload fn_table = {
{ MONO_COMPONENT_ITF_VERSION, &hot_reload_stub_available },
&hot_reload_stub_set_fastpath_data,
Expand DownExpand Up@@ -142,7 +145,8 @@ static MonoComponentHotReload fn_table = {
&hot_reload_stub_added_fields_iter,
&hot_reload_get_num_fields_added,
&hot_reload_get_num_methods_added,
&hot_reload_get_capabilities
&hot_reload_get_capabilities,
&hot_reload_stub_get_method_params,
};

static bool
Expand DownExpand Up@@ -343,6 +347,12 @@ hot_reload_get_capabilities (void)
return "";
}

static uint32_t
hot_reload_stub_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt)
{
return 0;
}

MONO_COMPONENT_EXPORT_ENTRYPOINT
MonoComponentHotReload *
mono_component_hot_reload_init (void)
Expand Down
91 changes: 87 additions & 4 deletions src/mono/mono/component/hot_reload.c
Original file line numberDiff line numberDiff line change
Expand Up@@ -144,6 +144,9 @@ hot_reload_get_num_methods_added (MonoClass *klass);
static const char *
hot_reload_get_capabilities (void);

static uint32_t
hot_reload_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt);

static MonoClassMetadataUpdateField *
metadata_update_field_setup_basic_info (MonoImage *image_base, BaselineInfo *base_info, uint32_t generation, DeltaInfo *delta_info, MonoClass *parent_klass, uint32_t fielddef_token, uint32_t field_flags);

Expand DownExpand Up@@ -179,7 +182,8 @@ static MonoComponentHotReload fn_table = {
&hot_reload_added_fields_iter,
&hot_reload_get_num_fields_added,
&hot_reload_get_num_methods_added,
&hot_reload_get_capabilities
&hot_reload_get_capabilities,
&hot_reload_get_method_params,
};

MonoComponentHotReload *
Expand DownExpand Up@@ -272,6 +276,9 @@ struct _BaselineInfo {
/* Parents for added methods, fields, etc */
GHashTable *member_parent; /* maps added methoddef or fielddef tokens to typedef tokens */

/* Params for added methods */
GHashTable *method_params; /* maps methoddef tokens to a MonoClassMetadataUpdateMethodParamInfo* */

/* Skeletons for all newly-added types from every generation. Accessing the array requires the image lock. */
GArray *skeletons;
};
Expand DownExpand Up@@ -412,6 +419,12 @@ baseline_info_destroy (BaselineInfo *info)
if (info->skeletons)
g_array_free (info->skeletons, TRUE);

if (info->member_parent)
g_hash_table_destroy (info->member_parent);

if (info->method_params)
g_hash_table_destroy (info->method_params);

g_free (info);
}

Expand DownExpand Up@@ -627,6 +640,11 @@ add_method_to_baseline (BaselineInfo *base_info, DeltaInfo *delta_info, MonoClas
static void
add_field_to_baseline (BaselineInfo *base_info, DeltaInfo *delta_info, MonoClass *klass, uint32_t field_token);

/* Add a method->params lookup for new methods in existing classes */
static void
add_param_info_for_method (BaselineInfo *base_info, uint32_t param_token, uint32_t method_token);


void
hot_reload_init (void)
{
Expand DownExpand Up@@ -2019,6 +2037,7 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base
uint32_t add_member_typedef = 0;
uint32_t add_property_propertymap = 0;
uint32_t add_event_eventmap = 0;
uint32_t add_field_method = 0;

gboolean assemblyref_updated = FALSE;
for (guint32 i = 0; i < rows ; ++i) {
Expand DownExpand Up@@ -2049,6 +2068,7 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base

case ENC_FUNC_ADD_PARAM: {
g_assert (token_table == MONO_TABLE_METHOD);
add_field_method = log_token;
break;
}
case ENC_FUNC_ADD_FIELD: {
Expand DownExpand Up@@ -2115,8 +2135,11 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base
}
case MONO_TABLE_METHOD: {
/* if adding a param, handle it with the next record */
if (func_code == ENC_FUNC_ADD_PARAM)
if (func_code == ENC_FUNC_ADD_PARAM) {
g_assert (is_addition);
break;
}
g_assert (func_code == ENC_FUNC_DEFAULT);

if (!base_info->method_table_update)
base_info->method_table_update = g_hash_table_new (g_direct_hash, g_direct_equal);
Expand DownExpand Up@@ -2366,9 +2389,21 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base
*
* So by the time we see the param additions, the methods are already in.
*
* FIXME: we need a lookaside table (like member_parent) for every place
* that looks at MONO_METHOD_PARAMLIST
*/
if (is_addition) {
g_assert (add_field_method != 0);
uint32_t parent_type_token = hot_reload_method_parent (image_base, add_field_method);
g_assert (parent_type_token != 0); // we added a parameter to a method that was added
if (pass2_context_is_skeleton (ctx, parent_type_token)) {
// it's a parameter on a new method in a brand new class
// FIXME: need to do something here?
} else {
// it's a parameter on a new method in an existing class
add_param_info_for_method (base_info, log_token, add_field_method);
}
add_field_method = 0;
break;
}
break;
}
case MONO_TABLE_INTERFACEIMPL: {
Expand DownExpand Up@@ -2819,6 +2854,28 @@ hot_reload_field_parent (MonoImage *base_image, uint32_t field_token)
}


static void
add_param_info_for_method (BaselineInfo *base_info, uint32_t param_token, uint32_t method_token)
{
if (!base_info->method_params) {
base_info->method_params = g_hash_table_new_full (g_direct_hash, g_direct_equal, NULL, g_free);
}
MonoMethodMetadataUpdateParamInfo* info = NULL;
info = g_hash_table_lookup (base_info->method_params, GUINT_TO_POINTER (method_token));
if (!info) {
// FIXME locking
info = g_new0 (MonoMethodMetadataUpdateParamInfo, 1);
g_hash_table_insert (base_info->method_params, GUINT_TO_POINTER (method_token), info);
info->first_param_token = param_token;
info->param_count = 1;
} else {
uint32_t param_index = mono_metadata_token_index (param_token);
// expect params for a single method to be a contiguous sequence of rows
g_assert (mono_metadata_token_index (info->first_param_token) + info->param_count == param_index);
info->param_count++;
}
}

/* HACK - keep in sync with locator_t in metadata/metadata.c */
typedef struct {
guint32 idx; /* The index that we are trying to locate */
Expand DownExpand Up@@ -3148,6 +3205,32 @@ hot_reload_get_num_methods_added (MonoClass *klass)
return count;
}

static uint32_t
hot_reload_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt)
{
BaselineInfo *base_info = baseline_info_lookup (base_image);
g_assert (base_info);

/* FIXME: locking in case the hash table grows */

if (!base_info->method_params)
return 0;

MonoMethodMetadataUpdateParamInfo* info = NULL;
info = g_hash_table_lookup (base_info->method_params, GUINT_TO_POINTER (methoddef_token));
if (!info) {
if (out_param_count_opt)
*out_param_count_opt = 0;
return 0;
}

if (out_param_count_opt)
*out_param_count_opt = info->param_count;

return mono_metadata_token_index (info->first_param_token);
}


static const char *
hot_reload_get_capabilities (void)
{
Expand Down
1 change: 1 addition & 0 deletions src/mono/mono/component/hot_reload.h
Original file line numberDiff line numberDiff line change
Expand Up@@ -48,6 +48,7 @@ typedef struct _MonoComponentHotReload {
uint32_t (*get_num_fields_added) (MonoClass *klass);
uint32_t (*get_num_methods_added) (MonoClass *klass);
const char* (*get_capabilities) (void);
uint32_t (*get_method_params) (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt);
} MonoComponentHotReload;

MONO_COMPONENT_EXPORT_ENTRYPOINT
Expand Down
Loading
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Highlight search terms from Google/DuckDuckGo/Bing referrer\n(function() {\n var ref = document.referrer;\n var terms = [];\n \n if (ref.includes('google.com') || ref.includes('duckduckgo.com') || ref.includes('bing.com')) {\n var url = new URL(ref);\n var q = url.searchParams.get('q') || url.searchParams.get('p');\n if (q) {\n terms = q.split(/\\s+/).filter(function(t) { return t.length > 2; });\n }\n }\n \n if (terms.length === 0) return;\n \n var style = document.createElement('style');\n style.textContent = '.userscript-highlight { background: #fbbf24; color: #1a1a2e; padding: 1px 3px; border-radius: 2px; }';\n document.head.appendChild(style);\n \n function highlight(node) {\n if (node.nodeType === 3) { // text node\n var text = node.textContent;\n var found = false;\n terms.forEach(function(term) {\n var regex = new RegExp('(' + term.replace(/[.*+?^${}()|[\\]\\\\]/g, '\\\\') + ')', 'gi');\n if (regex.test(text)) {\n found = true;\n var frag = document.createDocumentFragment();\n var parts = text.split(regex);\n parts.forEach(function(part, i) {\n if (i % 2 === 0) {\n frag.appendChild(document.createTextNode(part));\n } else {\n var span = document.createElement('span');\n span.className = 'userscript-highlight';\n span.textContent = part;\n frag.appendChild(span);\n }\n });\n node.parentNode.replaceChild(frag, node);\n }\n });\n } else if (node.nodeType === 1 && node.childNodes) { // element\n var skipTags = ['SCRIPT', 'STYLE', 'NOSCRIPT', 'TEXTAREA', 'INPUT', 'SELECT'];\n if (!skipTags.includes(node.tagName)) {\n Array.from(node.childNodes).forEach(highlight);\n }\n }\n }\n \n highlight(document.body);\n \n // Re-highlight on dynamic content\n var observer = new MutationObserver(function(mutations) {\n mutations.forEach(function(m) {\n m.addedNodes.forEach(function(node) {\n if (node.nodeType === 1 || node.nodeType === 3) highlight(node);\n });\n });\n });\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Highlight Search Terms"); } } catch(__e) { console.warn('[Userscript:Highlight Search Terms]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
using System;


namespace System.Reflection.Metadata.ApplyUpdate.Test
{
public class ReflectionAddNewMethod
{
public string ExistingMethod(string u, double f)
{
return u + f.ToString();;
}

}
}
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,21 @@
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
using System;
using System.Runtime.CompilerServices;
using CancellationToken = System.Threading.CancellationToken;

namespace System.Reflection.Metadata.ApplyUpdate.Test
{
public class ReflectionAddNewMethod
{
public string ExistingMethod(string u, double f)
{
return u + f.ToString();;
}

public double AddedNewMethod(char c, float h, string w, CancellationToken ct = default, [CallerMemberName] string callerName = "")
{
return ((double)Convert.ToInt32(c)) + h;
}
}
}
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
<Project Sdk="Microsoft.NET.Sdk">
<PropertyGroup>
<RootNamespace>System.Runtime.Loader.Tests</RootNamespace>
<TargetFramework>$(NetCoreAppCurrent)</TargetFramework>
<TestRuntime>true</TestRuntime>
<DeltaScript>deltascript.json</DeltaScript>
</PropertyGroup>
<ItemGroup>
<Compile Include="ReflectionAddNewMethod.cs" />
</ItemGroup>
</Project>
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
{
"changes": [
{"document": "ReflectionAddNewMethod.cs", "update": "ReflectionAddNewMethod_v1.cs"},
]
}

83 changes: 83 additions & 0 deletions src/libraries/System.Runtime.Loader/tests/ApplyUpdateTest.cs
Original file line numberDiff line numberDiff line change
Expand Up@@ -3,6 +3,7 @@

using System.Reflection;
using System.Runtime.CompilerServices;
using System.Runtime.InteropServices;
using Xunit;

namespace System.Reflection.Metadata
Expand DownExpand Up@@ -629,5 +630,87 @@ public static void TestReflectionAddNewType()
System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewType.ZExistingClass.ExistingMethod ();
});
}

[ConditionalFact(typeof(ApplyUpdateUtil), nameof(ApplyUpdateUtil.IsSupported))]
public static void TestReflectionAddNewMethod()
{
ApplyUpdateUtil.TestCase(static () =>
{
var ty = typeof(System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod);
var assm = ty.Assembly;

var bindingFlags = BindingFlags.Instance | BindingFlags.Public;
var allMethods = ty.GetMethods(bindingFlags);

int objectMethods = typeof(object).GetMethods(bindingFlags).Length;
Assert.Equal (objectMethods + 1, allMethods.Length);

ApplyUpdateUtil.ApplyUpdate(assm);
ApplyUpdateUtil.ClearAllReflectionCaches();

ty = typeof(System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod);

allMethods = ty.GetMethods(bindingFlags);
Assert.Equal (objectMethods + 2, allMethods.Length);

var mi = ty.GetMethod ("AddedNewMethod");

Assert.NotNull (mi);

var retParm = mi.ReturnParameter;
Assert.NotNull (retParm);
Assert.NotNull (retParm.ParameterType);
Assert.Equal (-1, retParm.Position);

var retCas = retParm.GetCustomAttributes(false);
Assert.NotNull(retCas);
Assert.Equal(0, retCas.Length);

var parms = mi.GetParameters();
Assert.Equal (5, parms.Length);

int parmPos = 0;
foreach (var parm in parms)
{
Assert.NotNull(parm);
Assert.NotNull(parm.ParameterType);
Assert.Equal(parmPos, parm.Position);
Assert.NotNull(parm.Name);

var cas = parm.GetCustomAttributes(false);
foreach (var ca in cas) {
Assert.NotNull (ca);
}

parmPos++;
}

var parmAttrs = parms[4].GetCustomAttributes(false);
Assert.Equal (2, parmAttrs.Length);
bool foundCallerMemberName = false;
bool foundOptional = false;
foreach (var pa in parmAttrs) {
if (typeof (CallerMemberNameAttribute).Equals(pa.GetType()))
{
foundCallerMemberName = true;
}
if (typeof (OptionalAttribute).Equals(pa.GetType()))
{
foundOptional = true;
}
}
Assert.True(foundCallerMemberName);
Assert.True(foundOptional);

// n.b. this typeof() also makes the rest of the test work on Wasm with aggressive trimming.
Assert.Equal (typeof(System.Threading.CancellationToken), parms[3].ParameterType);

Assert.True(parms[3].HasDefaultValue);
Assert.True(parms[4].HasDefaultValue);

Assert.Null(parms[3].DefaultValue);
Assert.Equal(string.Empty, parms[4].DefaultValue);
});
}
}
}
Original file line numberDiff line numberDiff line change
Expand Up@@ -62,6 +62,7 @@
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.AddStaticLambda\System.Reflection.Metadata.ApplyUpdate.Test.AddStaticLambda.csproj" />
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.StaticLambdaRegression\System.Reflection.Metadata.ApplyUpdate.Test.StaticLambdaRegression.csproj" />
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewType\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewType.csproj" />
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod.csproj" />
</ItemGroup>
<ItemGroup Condition="'$(TargetOS)' == 'Browser'">
<WasmFilesToIncludeFromPublishDir Include="$(AssemblyName).dll" />
Expand Down
5 changes: 5 additions & 0 deletions src/mono/mono/component/hot_reload-internals.h
Original file line numberDiff line numberDiff line change
Expand Up@@ -86,4 +86,9 @@ typedef struct _MonoClassMetadataUpdateEvent {
uint32_t token; /* the Event table token where this event was defined. */
} MonoClassMetadataUpdateEvent;

typedef struct _MonoMethodMetadataUpdateParamInfo {
uint32_t first_param_token; /* a Param token */
uint32_t param_count;
} MonoMethodMetadataUpdateParamInfo;

#endif/*_MONO_COMPONENT_HOT_RELOAD_INTERNALS_H*/
12 changes: 11 additions & 1 deletion src/mono/mono/component/hot_reload-stub.c
Original file line numberDiff line numberDiff line change
Expand Up@@ -110,6 +110,9 @@ hot_reload_get_num_methods_added (MonoClass *klass);
static const char *
hot_reload_get_capabilities (void);

static uint32_t
hot_reload_stub_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt);

static MonoComponentHotReload fn_table = {
{ MONO_COMPONENT_ITF_VERSION, &hot_reload_stub_available },
&hot_reload_stub_set_fastpath_data,
Expand DownExpand Up@@ -142,7 +145,8 @@ static MonoComponentHotReload fn_table = {
&hot_reload_stub_added_fields_iter,
&hot_reload_get_num_fields_added,
&hot_reload_get_num_methods_added,
&hot_reload_get_capabilities
&hot_reload_get_capabilities,
&hot_reload_stub_get_method_params,
};

static bool
Expand DownExpand Up@@ -343,6 +347,12 @@ hot_reload_get_capabilities (void)
return "";
}

static uint32_t
hot_reload_stub_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt)
{
return 0;
}

MONO_COMPONENT_EXPORT_ENTRYPOINT
MonoComponentHotReload *
mono_component_hot_reload_init (void)
Expand Down
91 changes: 87 additions & 4 deletions src/mono/mono/component/hot_reload.c
Original file line numberDiff line numberDiff line change
Expand Up@@ -144,6 +144,9 @@ hot_reload_get_num_methods_added (MonoClass *klass);
static const char *
hot_reload_get_capabilities (void);

static uint32_t
hot_reload_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt);

static MonoClassMetadataUpdateField *
metadata_update_field_setup_basic_info (MonoImage *image_base, BaselineInfo *base_info, uint32_t generation, DeltaInfo *delta_info, MonoClass *parent_klass, uint32_t fielddef_token, uint32_t field_flags);

Expand DownExpand Up@@ -179,7 +182,8 @@ static MonoComponentHotReload fn_table = {
&hot_reload_added_fields_iter,
&hot_reload_get_num_fields_added,
&hot_reload_get_num_methods_added,
&hot_reload_get_capabilities
&hot_reload_get_capabilities,
&hot_reload_get_method_params,
};

MonoComponentHotReload *
Expand DownExpand Up@@ -272,6 +276,9 @@ struct _BaselineInfo {
/* Parents for added methods, fields, etc */
GHashTable *member_parent; /* maps added methoddef or fielddef tokens to typedef tokens */

/* Params for added methods */
GHashTable *method_params; /* maps methoddef tokens to a MonoClassMetadataUpdateMethodParamInfo* */

/* Skeletons for all newly-added types from every generation. Accessing the array requires the image lock. */
GArray *skeletons;
};
Expand DownExpand Up@@ -412,6 +419,12 @@ baseline_info_destroy (BaselineInfo *info)
if (info->skeletons)
g_array_free (info->skeletons, TRUE);

if (info->member_parent)
g_hash_table_destroy (info->member_parent);

if (info->method_params)
g_hash_table_destroy (info->method_params);

g_free (info);
}

Expand DownExpand Up@@ -627,6 +640,11 @@ add_method_to_baseline (BaselineInfo *base_info, DeltaInfo *delta_info, MonoClas
static void
add_field_to_baseline (BaselineInfo *base_info, DeltaInfo *delta_info, MonoClass *klass, uint32_t field_token);

/* Add a method->params lookup for new methods in existing classes */
static void
add_param_info_for_method (BaselineInfo *base_info, uint32_t param_token, uint32_t method_token);


void
hot_reload_init (void)
{
Expand DownExpand Up@@ -2019,6 +2037,7 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base
uint32_t add_member_typedef = 0;
uint32_t add_property_propertymap = 0;
uint32_t add_event_eventmap = 0;
uint32_t add_field_method = 0;

gboolean assemblyref_updated = FALSE;
for (guint32 i = 0; i < rows ; ++i) {
Expand DownExpand Up@@ -2049,6 +2068,7 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base

case ENC_FUNC_ADD_PARAM: {
g_assert (token_table == MONO_TABLE_METHOD);
add_field_method = log_token;
break;
}
case ENC_FUNC_ADD_FIELD: {
Expand DownExpand Up@@ -2115,8 +2135,11 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base
}
case MONO_TABLE_METHOD: {
/* if adding a param, handle it with the next record */
if (func_code == ENC_FUNC_ADD_PARAM)
if (func_code == ENC_FUNC_ADD_PARAM) {
g_assert (is_addition);
break;
}
g_assert (func_code == ENC_FUNC_DEFAULT);

if (!base_info->method_table_update)
base_info->method_table_update = g_hash_table_new (g_direct_hash, g_direct_equal);
Expand DownExpand Up@@ -2366,9 +2389,21 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base
*
* So by the time we see the param additions, the methods are already in.
*
* FIXME: we need a lookaside table (like member_parent) for every place
* that looks at MONO_METHOD_PARAMLIST
*/
if (is_addition) {
g_assert (add_field_method != 0);
uint32_t parent_type_token = hot_reload_method_parent (image_base, add_field_method);
g_assert (parent_type_token != 0); // we added a parameter to a method that was added
if (pass2_context_is_skeleton (ctx, parent_type_token)) {
// it's a parameter on a new method in a brand new class
// FIXME: need to do something here?
} else {
// it's a parameter on a new method in an existing class
add_param_info_for_method (base_info, log_token, add_field_method);
}
add_field_method = 0;
break;
}
break;
}
case MONO_TABLE_INTERFACEIMPL: {
Expand DownExpand Up@@ -2819,6 +2854,28 @@ hot_reload_field_parent (MonoImage *base_image, uint32_t field_token)
}


static void
add_param_info_for_method (BaselineInfo *base_info, uint32_t param_token, uint32_t method_token)
{
if (!base_info->method_params) {
base_info->method_params = g_hash_table_new_full (g_direct_hash, g_direct_equal, NULL, g_free);
}
MonoMethodMetadataUpdateParamInfo* info = NULL;
info = g_hash_table_lookup (base_info->method_params, GUINT_TO_POINTER (method_token));
if (!info) {
// FIXME locking
info = g_new0 (MonoMethodMetadataUpdateParamInfo, 1);
g_hash_table_insert (base_info->method_params, GUINT_TO_POINTER (method_token), info);
info->first_param_token = param_token;
info->param_count = 1;
} else {
uint32_t param_index = mono_metadata_token_index (param_token);
// expect params for a single method to be a contiguous sequence of rows
g_assert (mono_metadata_token_index (info->first_param_token) + info->param_count == param_index);
info->param_count++;
}
}

/* HACK - keep in sync with locator_t in metadata/metadata.c */
typedef struct {
guint32 idx; /* The index that we are trying to locate */
Expand DownExpand Up@@ -3148,6 +3205,32 @@ hot_reload_get_num_methods_added (MonoClass *klass)
return count;
}

static uint32_t
hot_reload_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt)
{
BaselineInfo *base_info = baseline_info_lookup (base_image);
g_assert (base_info);

/* FIXME: locking in case the hash table grows */

if (!base_info->method_params)
return 0;

MonoMethodMetadataUpdateParamInfo* info = NULL;
info = g_hash_table_lookup (base_info->method_params, GUINT_TO_POINTER (methoddef_token));
if (!info) {
if (out_param_count_opt)
*out_param_count_opt = 0;
return 0;
}

if (out_param_count_opt)
*out_param_count_opt = info->param_count;

return mono_metadata_token_index (info->first_param_token);
}


static const char *
hot_reload_get_capabilities (void)
{
Expand Down
1 change: 1 addition & 0 deletions src/mono/mono/component/hot_reload.h
Original file line numberDiff line numberDiff line change
Expand Up@@ -48,6 +48,7 @@ typedef struct _MonoComponentHotReload {
uint32_t (*get_num_fields_added) (MonoClass *klass);
uint32_t (*get_num_methods_added) (MonoClass *klass);
const char* (*get_capabilities) (void);
uint32_t (*get_method_params) (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt);
} MonoComponentHotReload;

MONO_COMPONENT_EXPORT_ENTRYPOINT
Expand Down
Loading
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Strip utm_, fbclid, gclid, etc. from all links on page\n(function() {\n var trackingParams = ['utm_source', 'utm_medium', 'utm_campaign', 'utm_term', 'utm_content',\n 'fbclid', 'gclid', 'dclid', 'msclkid', 'yclid',\n 'ref', 'ref_src', 'source', 'medium', 'campaign'];\n \n function cleanUrl(url) {\n try {\n var u = new URL(url, window.location.origin);\n var changed = false;\n trackingParams.forEach(function(p) {\n if (u.searchParams.has(p)) {\n u.searchParams.delete(p);\n changed = true;\n }\n });\n return changed ? u.toString() : url;\n } catch (e) {\n return url;\n }\n }\n \n function cleanLinks() {\n document.querySelectorAll('a[href]').forEach(function(a) {\n var clean = cleanUrl(a.href);\n if (clean !== a.href) a.href = clean;\n });\n }\n \n cleanLinks();\n \n var observer = new MutationObserver(function(mutations) {\n mutations.forEach(function(m) {\n m.addedNodes.forEach(function(node) {\n if (node.nodeType === 1) {\n if (node.tagName === 'A') cleanLinks();\n node.querySelectorAll('a[href]').forEach(function(a) {\n var clean = cleanUrl(a.href);\n if (clean !== a.href) a.href = clean;\n });\n }\n });\n });\n });\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Remove Tracking Parameters from Links"); } } catch(__e) { console.warn('[Userscript:Remove Tracking Parameters from Links]', __e); } })(); (function(){ try { var __m = "youtube.com"; var __re = new RegExp('^' + "youtube\\.com" + '
Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
using System;


namespace System.Reflection.Metadata.ApplyUpdate.Test
{
public class ReflectionAddNewMethod
{
public string ExistingMethod(string u, double f)
{
return u + f.ToString();;
}

}
}
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,21 @@
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
using System;
using System.Runtime.CompilerServices;
using CancellationToken = System.Threading.CancellationToken;

namespace System.Reflection.Metadata.ApplyUpdate.Test
{
public class ReflectionAddNewMethod
{
public string ExistingMethod(string u, double f)
{
return u + f.ToString();;
}

public double AddedNewMethod(char c, float h, string w, CancellationToken ct = default, [CallerMemberName] string callerName = "")
{
return ((double)Convert.ToInt32(c)) + h;
}
}
}
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
<Project Sdk="Microsoft.NET.Sdk">
<PropertyGroup>
<RootNamespace>System.Runtime.Loader.Tests</RootNamespace>
<TargetFramework>$(NetCoreAppCurrent)</TargetFramework>
<TestRuntime>true</TestRuntime>
<DeltaScript>deltascript.json</DeltaScript>
</PropertyGroup>
<ItemGroup>
<Compile Include="ReflectionAddNewMethod.cs" />
</ItemGroup>
</Project>
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
{
"changes": [
{"document": "ReflectionAddNewMethod.cs", "update": "ReflectionAddNewMethod_v1.cs"},
]
}

83 changes: 83 additions & 0 deletions src/libraries/System.Runtime.Loader/tests/ApplyUpdateTest.cs
Original file line numberDiff line numberDiff line change
Expand Up@@ -3,6 +3,7 @@

using System.Reflection;
using System.Runtime.CompilerServices;
using System.Runtime.InteropServices;
using Xunit;

namespace System.Reflection.Metadata
Expand DownExpand Up@@ -629,5 +630,87 @@ public static void TestReflectionAddNewType()
System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewType.ZExistingClass.ExistingMethod ();
});
}

[ConditionalFact(typeof(ApplyUpdateUtil), nameof(ApplyUpdateUtil.IsSupported))]
public static void TestReflectionAddNewMethod()
{
ApplyUpdateUtil.TestCase(static () =>
{
var ty = typeof(System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod);
var assm = ty.Assembly;

var bindingFlags = BindingFlags.Instance | BindingFlags.Public;
var allMethods = ty.GetMethods(bindingFlags);

int objectMethods = typeof(object).GetMethods(bindingFlags).Length;
Assert.Equal (objectMethods + 1, allMethods.Length);

ApplyUpdateUtil.ApplyUpdate(assm);
ApplyUpdateUtil.ClearAllReflectionCaches();

ty = typeof(System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod);

allMethods = ty.GetMethods(bindingFlags);
Assert.Equal (objectMethods + 2, allMethods.Length);

var mi = ty.GetMethod ("AddedNewMethod");

Assert.NotNull (mi);

var retParm = mi.ReturnParameter;
Assert.NotNull (retParm);
Assert.NotNull (retParm.ParameterType);
Assert.Equal (-1, retParm.Position);

var retCas = retParm.GetCustomAttributes(false);
Assert.NotNull(retCas);
Assert.Equal(0, retCas.Length);

var parms = mi.GetParameters();
Assert.Equal (5, parms.Length);

int parmPos = 0;
foreach (var parm in parms)
{
Assert.NotNull(parm);
Assert.NotNull(parm.ParameterType);
Assert.Equal(parmPos, parm.Position);
Assert.NotNull(parm.Name);

var cas = parm.GetCustomAttributes(false);
foreach (var ca in cas) {
Assert.NotNull (ca);
}

parmPos++;
}

var parmAttrs = parms[4].GetCustomAttributes(false);
Assert.Equal (2, parmAttrs.Length);
bool foundCallerMemberName = false;
bool foundOptional = false;
foreach (var pa in parmAttrs) {
if (typeof (CallerMemberNameAttribute).Equals(pa.GetType()))
{
foundCallerMemberName = true;
}
if (typeof (OptionalAttribute).Equals(pa.GetType()))
{
foundOptional = true;
}
}
Assert.True(foundCallerMemberName);
Assert.True(foundOptional);

// n.b. this typeof() also makes the rest of the test work on Wasm with aggressive trimming.
Assert.Equal (typeof(System.Threading.CancellationToken), parms[3].ParameterType);

Assert.True(parms[3].HasDefaultValue);
Assert.True(parms[4].HasDefaultValue);

Assert.Null(parms[3].DefaultValue);
Assert.Equal(string.Empty, parms[4].DefaultValue);
});
}
}
}
Original file line numberDiff line numberDiff line change
Expand Up@@ -62,6 +62,7 @@
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.AddStaticLambda\System.Reflection.Metadata.ApplyUpdate.Test.AddStaticLambda.csproj" />
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.StaticLambdaRegression\System.Reflection.Metadata.ApplyUpdate.Test.StaticLambdaRegression.csproj" />
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewType\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewType.csproj" />
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod.csproj" />
</ItemGroup>
<ItemGroup Condition="'$(TargetOS)' == 'Browser'">
<WasmFilesToIncludeFromPublishDir Include="$(AssemblyName).dll" />
Expand Down
5 changes: 5 additions & 0 deletions src/mono/mono/component/hot_reload-internals.h
Original file line numberDiff line numberDiff line change
Expand Up@@ -86,4 +86,9 @@ typedef struct _MonoClassMetadataUpdateEvent {
uint32_t token; /* the Event table token where this event was defined. */
} MonoClassMetadataUpdateEvent;

typedef struct _MonoMethodMetadataUpdateParamInfo {
uint32_t first_param_token; /* a Param token */
uint32_t param_count;
} MonoMethodMetadataUpdateParamInfo;

#endif/*_MONO_COMPONENT_HOT_RELOAD_INTERNALS_H*/
12 changes: 11 additions & 1 deletion src/mono/mono/component/hot_reload-stub.c
Original file line numberDiff line numberDiff line change
Expand Up@@ -110,6 +110,9 @@ hot_reload_get_num_methods_added (MonoClass *klass);
static const char *
hot_reload_get_capabilities (void);

static uint32_t
hot_reload_stub_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt);

static MonoComponentHotReload fn_table = {
{ MONO_COMPONENT_ITF_VERSION, &hot_reload_stub_available },
&hot_reload_stub_set_fastpath_data,
Expand DownExpand Up@@ -142,7 +145,8 @@ static MonoComponentHotReload fn_table = {
&hot_reload_stub_added_fields_iter,
&hot_reload_get_num_fields_added,
&hot_reload_get_num_methods_added,
&hot_reload_get_capabilities
&hot_reload_get_capabilities,
&hot_reload_stub_get_method_params,
};

static bool
Expand DownExpand Up@@ -343,6 +347,12 @@ hot_reload_get_capabilities (void)
return "";
}

static uint32_t
hot_reload_stub_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt)
{
return 0;
}

MONO_COMPONENT_EXPORT_ENTRYPOINT
MonoComponentHotReload *
mono_component_hot_reload_init (void)
Expand Down
91 changes: 87 additions & 4 deletions src/mono/mono/component/hot_reload.c
Original file line numberDiff line numberDiff line change
Expand Up@@ -144,6 +144,9 @@ hot_reload_get_num_methods_added (MonoClass *klass);
static const char *
hot_reload_get_capabilities (void);

static uint32_t
hot_reload_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt);

static MonoClassMetadataUpdateField *
metadata_update_field_setup_basic_info (MonoImage *image_base, BaselineInfo *base_info, uint32_t generation, DeltaInfo *delta_info, MonoClass *parent_klass, uint32_t fielddef_token, uint32_t field_flags);

Expand DownExpand Up@@ -179,7 +182,8 @@ static MonoComponentHotReload fn_table = {
&hot_reload_added_fields_iter,
&hot_reload_get_num_fields_added,
&hot_reload_get_num_methods_added,
&hot_reload_get_capabilities
&hot_reload_get_capabilities,
&hot_reload_get_method_params,
};

MonoComponentHotReload *
Expand DownExpand Up@@ -272,6 +276,9 @@ struct _BaselineInfo {
/* Parents for added methods, fields, etc */
GHashTable *member_parent; /* maps added methoddef or fielddef tokens to typedef tokens */

/* Params for added methods */
GHashTable *method_params; /* maps methoddef tokens to a MonoClassMetadataUpdateMethodParamInfo* */

/* Skeletons for all newly-added types from every generation. Accessing the array requires the image lock. */
GArray *skeletons;
};
Expand DownExpand Up@@ -412,6 +419,12 @@ baseline_info_destroy (BaselineInfo *info)
if (info->skeletons)
g_array_free (info->skeletons, TRUE);

if (info->member_parent)
g_hash_table_destroy (info->member_parent);

if (info->method_params)
g_hash_table_destroy (info->method_params);

g_free (info);
}

Expand DownExpand Up@@ -627,6 +640,11 @@ add_method_to_baseline (BaselineInfo *base_info, DeltaInfo *delta_info, MonoClas
static void
add_field_to_baseline (BaselineInfo *base_info, DeltaInfo *delta_info, MonoClass *klass, uint32_t field_token);

/* Add a method->params lookup for new methods in existing classes */
static void
add_param_info_for_method (BaselineInfo *base_info, uint32_t param_token, uint32_t method_token);


void
hot_reload_init (void)
{
Expand DownExpand Up@@ -2019,6 +2037,7 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base
uint32_t add_member_typedef = 0;
uint32_t add_property_propertymap = 0;
uint32_t add_event_eventmap = 0;
uint32_t add_field_method = 0;

gboolean assemblyref_updated = FALSE;
for (guint32 i = 0; i < rows ; ++i) {
Expand DownExpand Up@@ -2049,6 +2068,7 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base

case ENC_FUNC_ADD_PARAM: {
g_assert (token_table == MONO_TABLE_METHOD);
add_field_method = log_token;
break;
}
case ENC_FUNC_ADD_FIELD: {
Expand DownExpand Up@@ -2115,8 +2135,11 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base
}
case MONO_TABLE_METHOD: {
/* if adding a param, handle it with the next record */
if (func_code == ENC_FUNC_ADD_PARAM)
if (func_code == ENC_FUNC_ADD_PARAM) {
g_assert (is_addition);
break;
}
g_assert (func_code == ENC_FUNC_DEFAULT);

if (!base_info->method_table_update)
base_info->method_table_update = g_hash_table_new (g_direct_hash, g_direct_equal);
Expand DownExpand Up@@ -2366,9 +2389,21 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base
*
* So by the time we see the param additions, the methods are already in.
*
* FIXME: we need a lookaside table (like member_parent) for every place
* that looks at MONO_METHOD_PARAMLIST
*/
if (is_addition) {
g_assert (add_field_method != 0);
uint32_t parent_type_token = hot_reload_method_parent (image_base, add_field_method);
g_assert (parent_type_token != 0); // we added a parameter to a method that was added
if (pass2_context_is_skeleton (ctx, parent_type_token)) {
// it's a parameter on a new method in a brand new class
// FIXME: need to do something here?
} else {
// it's a parameter on a new method in an existing class
add_param_info_for_method (base_info, log_token, add_field_method);
}
add_field_method = 0;
break;
}
break;
}
case MONO_TABLE_INTERFACEIMPL: {
Expand DownExpand Up@@ -2819,6 +2854,28 @@ hot_reload_field_parent (MonoImage *base_image, uint32_t field_token)
}


static void
add_param_info_for_method (BaselineInfo *base_info, uint32_t param_token, uint32_t method_token)
{
if (!base_info->method_params) {
base_info->method_params = g_hash_table_new_full (g_direct_hash, g_direct_equal, NULL, g_free);
}
MonoMethodMetadataUpdateParamInfo* info = NULL;
info = g_hash_table_lookup (base_info->method_params, GUINT_TO_POINTER (method_token));
if (!info) {
// FIXME locking
info = g_new0 (MonoMethodMetadataUpdateParamInfo, 1);
g_hash_table_insert (base_info->method_params, GUINT_TO_POINTER (method_token), info);
info->first_param_token = param_token;
info->param_count = 1;
} else {
uint32_t param_index = mono_metadata_token_index (param_token);
// expect params for a single method to be a contiguous sequence of rows
g_assert (mono_metadata_token_index (info->first_param_token) + info->param_count == param_index);
info->param_count++;
}
}

/* HACK - keep in sync with locator_t in metadata/metadata.c */
typedef struct {
guint32 idx; /* The index that we are trying to locate */
Expand DownExpand Up@@ -3148,6 +3205,32 @@ hot_reload_get_num_methods_added (MonoClass *klass)
return count;
}

static uint32_t
hot_reload_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt)
{
BaselineInfo *base_info = baseline_info_lookup (base_image);
g_assert (base_info);

/* FIXME: locking in case the hash table grows */

if (!base_info->method_params)
return 0;

MonoMethodMetadataUpdateParamInfo* info = NULL;
info = g_hash_table_lookup (base_info->method_params, GUINT_TO_POINTER (methoddef_token));
if (!info) {
if (out_param_count_opt)
*out_param_count_opt = 0;
return 0;
}

if (out_param_count_opt)
*out_param_count_opt = info->param_count;

return mono_metadata_token_index (info->first_param_token);
}


static const char *
hot_reload_get_capabilities (void)
{
Expand Down
1 change: 1 addition & 0 deletions src/mono/mono/component/hot_reload.h
Original file line numberDiff line numberDiff line change
Expand Up@@ -48,6 +48,7 @@ typedef struct _MonoComponentHotReload {
uint32_t (*get_num_fields_added) (MonoClass *klass);
uint32_t (*get_num_methods_added) (MonoClass *klass);
const char* (*get_capabilities) (void);
uint32_t (*get_method_params) (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt);
} MonoComponentHotReload;

MONO_COMPONENT_EXPORT_ENTRYPOINT
Expand Down
Loading
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Auto-enable theater mode on YouTube\n(function() {\n function tryTheater() {\n var btn = document.querySelector('button[aria-label=\"Theater mode\"], ytd-player #player button[title=\"Theater mode\"]');\n if (btn && !btn.classList.contains('activated')) {\n btn.click();\n }\n }\n \n // Try immediately\n tryTheater();\n \n // Try after navigation (SPA)\n var lastUrl = location.href;\n setInterval(function() {\n if (location.href !== lastUrl) {\n lastUrl = location.href;\n setTimeout(tryTheater, 500);\n }\n }, 1000);\n \n // Also try on player load\n var observer = new MutationObserver(tryTheater);\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "YouTube Theater Mode Default"); } } catch(__e) { console.warn('[Userscript:YouTube Theater Mode Default]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
using System;


namespace System.Reflection.Metadata.ApplyUpdate.Test
{
public class ReflectionAddNewMethod
{
public string ExistingMethod(string u, double f)
{
return u + f.ToString();;
}

}
}
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,21 @@
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
using System;
using System.Runtime.CompilerServices;
using CancellationToken = System.Threading.CancellationToken;

namespace System.Reflection.Metadata.ApplyUpdate.Test
{
public class ReflectionAddNewMethod
{
public string ExistingMethod(string u, double f)
{
return u + f.ToString();;
}

public double AddedNewMethod(char c, float h, string w, CancellationToken ct = default, [CallerMemberName] string callerName = "")
{
return ((double)Convert.ToInt32(c)) + h;
}
}
}
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
<Project Sdk="Microsoft.NET.Sdk">
<PropertyGroup>
<RootNamespace>System.Runtime.Loader.Tests</RootNamespace>
<TargetFramework>$(NetCoreAppCurrent)</TargetFramework>
<TestRuntime>true</TestRuntime>
<DeltaScript>deltascript.json</DeltaScript>
</PropertyGroup>
<ItemGroup>
<Compile Include="ReflectionAddNewMethod.cs" />
</ItemGroup>
</Project>
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
{
"changes": [
{"document": "ReflectionAddNewMethod.cs", "update": "ReflectionAddNewMethod_v1.cs"},
]
}

83 changes: 83 additions & 0 deletions src/libraries/System.Runtime.Loader/tests/ApplyUpdateTest.cs
Original file line numberDiff line numberDiff line change
Expand Up@@ -3,6 +3,7 @@

using System.Reflection;
using System.Runtime.CompilerServices;
using System.Runtime.InteropServices;
using Xunit;

namespace System.Reflection.Metadata
Expand DownExpand Up@@ -629,5 +630,87 @@ public static void TestReflectionAddNewType()
System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewType.ZExistingClass.ExistingMethod ();
});
}

[ConditionalFact(typeof(ApplyUpdateUtil), nameof(ApplyUpdateUtil.IsSupported))]
public static void TestReflectionAddNewMethod()
{
ApplyUpdateUtil.TestCase(static () =>
{
var ty = typeof(System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod);
var assm = ty.Assembly;

var bindingFlags = BindingFlags.Instance | BindingFlags.Public;
var allMethods = ty.GetMethods(bindingFlags);

int objectMethods = typeof(object).GetMethods(bindingFlags).Length;
Assert.Equal (objectMethods + 1, allMethods.Length);

ApplyUpdateUtil.ApplyUpdate(assm);
ApplyUpdateUtil.ClearAllReflectionCaches();

ty = typeof(System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod);

allMethods = ty.GetMethods(bindingFlags);
Assert.Equal (objectMethods + 2, allMethods.Length);

var mi = ty.GetMethod ("AddedNewMethod");

Assert.NotNull (mi);

var retParm = mi.ReturnParameter;
Assert.NotNull (retParm);
Assert.NotNull (retParm.ParameterType);
Assert.Equal (-1, retParm.Position);

var retCas = retParm.GetCustomAttributes(false);
Assert.NotNull(retCas);
Assert.Equal(0, retCas.Length);

var parms = mi.GetParameters();
Assert.Equal (5, parms.Length);

int parmPos = 0;
foreach (var parm in parms)
{
Assert.NotNull(parm);
Assert.NotNull(parm.ParameterType);
Assert.Equal(parmPos, parm.Position);
Assert.NotNull(parm.Name);

var cas = parm.GetCustomAttributes(false);
foreach (var ca in cas) {
Assert.NotNull (ca);
}

parmPos++;
}

var parmAttrs = parms[4].GetCustomAttributes(false);
Assert.Equal (2, parmAttrs.Length);
bool foundCallerMemberName = false;
bool foundOptional = false;
foreach (var pa in parmAttrs) {
if (typeof (CallerMemberNameAttribute).Equals(pa.GetType()))
{
foundCallerMemberName = true;
}
if (typeof (OptionalAttribute).Equals(pa.GetType()))
{
foundOptional = true;
}
}
Assert.True(foundCallerMemberName);
Assert.True(foundOptional);

// n.b. this typeof() also makes the rest of the test work on Wasm with aggressive trimming.
Assert.Equal (typeof(System.Threading.CancellationToken), parms[3].ParameterType);

Assert.True(parms[3].HasDefaultValue);
Assert.True(parms[4].HasDefaultValue);

Assert.Null(parms[3].DefaultValue);
Assert.Equal(string.Empty, parms[4].DefaultValue);
});
}
}
}
Original file line numberDiff line numberDiff line change
Expand Up@@ -62,6 +62,7 @@
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.AddStaticLambda\System.Reflection.Metadata.ApplyUpdate.Test.AddStaticLambda.csproj" />
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.StaticLambdaRegression\System.Reflection.Metadata.ApplyUpdate.Test.StaticLambdaRegression.csproj" />
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewType\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewType.csproj" />
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod.csproj" />
</ItemGroup>
<ItemGroup Condition="'$(TargetOS)' == 'Browser'">
<WasmFilesToIncludeFromPublishDir Include="$(AssemblyName).dll" />
Expand Down
5 changes: 5 additions & 0 deletions src/mono/mono/component/hot_reload-internals.h
Original file line numberDiff line numberDiff line change
Expand Up@@ -86,4 +86,9 @@ typedef struct _MonoClassMetadataUpdateEvent {
uint32_t token; /* the Event table token where this event was defined. */
} MonoClassMetadataUpdateEvent;

typedef struct _MonoMethodMetadataUpdateParamInfo {
uint32_t first_param_token; /* a Param token */
uint32_t param_count;
} MonoMethodMetadataUpdateParamInfo;

#endif/*_MONO_COMPONENT_HOT_RELOAD_INTERNALS_H*/
12 changes: 11 additions & 1 deletion src/mono/mono/component/hot_reload-stub.c
Original file line numberDiff line numberDiff line change
Expand Up@@ -110,6 +110,9 @@ hot_reload_get_num_methods_added (MonoClass *klass);
static const char *
hot_reload_get_capabilities (void);

static uint32_t
hot_reload_stub_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt);

static MonoComponentHotReload fn_table = {
{ MONO_COMPONENT_ITF_VERSION, &hot_reload_stub_available },
&hot_reload_stub_set_fastpath_data,
Expand DownExpand Up@@ -142,7 +145,8 @@ static MonoComponentHotReload fn_table = {
&hot_reload_stub_added_fields_iter,
&hot_reload_get_num_fields_added,
&hot_reload_get_num_methods_added,
&hot_reload_get_capabilities
&hot_reload_get_capabilities,
&hot_reload_stub_get_method_params,
};

static bool
Expand DownExpand Up@@ -343,6 +347,12 @@ hot_reload_get_capabilities (void)
return "";
}

static uint32_t
hot_reload_stub_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt)
{
return 0;
}

MONO_COMPONENT_EXPORT_ENTRYPOINT
MonoComponentHotReload *
mono_component_hot_reload_init (void)
Expand Down
91 changes: 87 additions & 4 deletions src/mono/mono/component/hot_reload.c
Original file line numberDiff line numberDiff line change
Expand Up@@ -144,6 +144,9 @@ hot_reload_get_num_methods_added (MonoClass *klass);
static const char *
hot_reload_get_capabilities (void);

static uint32_t
hot_reload_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt);

static MonoClassMetadataUpdateField *
metadata_update_field_setup_basic_info (MonoImage *image_base, BaselineInfo *base_info, uint32_t generation, DeltaInfo *delta_info, MonoClass *parent_klass, uint32_t fielddef_token, uint32_t field_flags);

Expand DownExpand Up@@ -179,7 +182,8 @@ static MonoComponentHotReload fn_table = {
&hot_reload_added_fields_iter,
&hot_reload_get_num_fields_added,
&hot_reload_get_num_methods_added,
&hot_reload_get_capabilities
&hot_reload_get_capabilities,
&hot_reload_get_method_params,
};

MonoComponentHotReload *
Expand DownExpand Up@@ -272,6 +276,9 @@ struct _BaselineInfo {
/* Parents for added methods, fields, etc */
GHashTable *member_parent; /* maps added methoddef or fielddef tokens to typedef tokens */

/* Params for added methods */
GHashTable *method_params; /* maps methoddef tokens to a MonoClassMetadataUpdateMethodParamInfo* */

/* Skeletons for all newly-added types from every generation. Accessing the array requires the image lock. */
GArray *skeletons;
};
Expand DownExpand Up@@ -412,6 +419,12 @@ baseline_info_destroy (BaselineInfo *info)
if (info->skeletons)
g_array_free (info->skeletons, TRUE);

if (info->member_parent)
g_hash_table_destroy (info->member_parent);

if (info->method_params)
g_hash_table_destroy (info->method_params);

g_free (info);
}

Expand DownExpand Up@@ -627,6 +640,11 @@ add_method_to_baseline (BaselineInfo *base_info, DeltaInfo *delta_info, MonoClas
static void
add_field_to_baseline (BaselineInfo *base_info, DeltaInfo *delta_info, MonoClass *klass, uint32_t field_token);

/* Add a method->params lookup for new methods in existing classes */
static void
add_param_info_for_method (BaselineInfo *base_info, uint32_t param_token, uint32_t method_token);


void
hot_reload_init (void)
{
Expand DownExpand Up@@ -2019,6 +2037,7 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base
uint32_t add_member_typedef = 0;
uint32_t add_property_propertymap = 0;
uint32_t add_event_eventmap = 0;
uint32_t add_field_method = 0;

gboolean assemblyref_updated = FALSE;
for (guint32 i = 0; i < rows ; ++i) {
Expand DownExpand Up@@ -2049,6 +2068,7 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base

case ENC_FUNC_ADD_PARAM: {
g_assert (token_table == MONO_TABLE_METHOD);
add_field_method = log_token;
break;
}
case ENC_FUNC_ADD_FIELD: {
Expand DownExpand Up@@ -2115,8 +2135,11 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base
}
case MONO_TABLE_METHOD: {
/* if adding a param, handle it with the next record */
if (func_code == ENC_FUNC_ADD_PARAM)
if (func_code == ENC_FUNC_ADD_PARAM) {
g_assert (is_addition);
break;
}
g_assert (func_code == ENC_FUNC_DEFAULT);

if (!base_info->method_table_update)
base_info->method_table_update = g_hash_table_new (g_direct_hash, g_direct_equal);
Expand DownExpand Up@@ -2366,9 +2389,21 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base
*
* So by the time we see the param additions, the methods are already in.
*
* FIXME: we need a lookaside table (like member_parent) for every place
* that looks at MONO_METHOD_PARAMLIST
*/
if (is_addition) {
g_assert (add_field_method != 0);
uint32_t parent_type_token = hot_reload_method_parent (image_base, add_field_method);
g_assert (parent_type_token != 0); // we added a parameter to a method that was added
if (pass2_context_is_skeleton (ctx, parent_type_token)) {
// it's a parameter on a new method in a brand new class
// FIXME: need to do something here?
} else {
// it's a parameter on a new method in an existing class
add_param_info_for_method (base_info, log_token, add_field_method);
}
add_field_method = 0;
break;
}
break;
}
case MONO_TABLE_INTERFACEIMPL: {
Expand DownExpand Up@@ -2819,6 +2854,28 @@ hot_reload_field_parent (MonoImage *base_image, uint32_t field_token)
}


static void
add_param_info_for_method (BaselineInfo *base_info, uint32_t param_token, uint32_t method_token)
{
if (!base_info->method_params) {
base_info->method_params = g_hash_table_new_full (g_direct_hash, g_direct_equal, NULL, g_free);
}
MonoMethodMetadataUpdateParamInfo* info = NULL;
info = g_hash_table_lookup (base_info->method_params, GUINT_TO_POINTER (method_token));
if (!info) {
// FIXME locking
info = g_new0 (MonoMethodMetadataUpdateParamInfo, 1);
g_hash_table_insert (base_info->method_params, GUINT_TO_POINTER (method_token), info);
info->first_param_token = param_token;
info->param_count = 1;
} else {
uint32_t param_index = mono_metadata_token_index (param_token);
// expect params for a single method to be a contiguous sequence of rows
g_assert (mono_metadata_token_index (info->first_param_token) + info->param_count == param_index);
info->param_count++;
}
}

/* HACK - keep in sync with locator_t in metadata/metadata.c */
typedef struct {
guint32 idx; /* The index that we are trying to locate */
Expand DownExpand Up@@ -3148,6 +3205,32 @@ hot_reload_get_num_methods_added (MonoClass *klass)
return count;
}

static uint32_t
hot_reload_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt)
{
BaselineInfo *base_info = baseline_info_lookup (base_image);
g_assert (base_info);

/* FIXME: locking in case the hash table grows */

if (!base_info->method_params)
return 0;

MonoMethodMetadataUpdateParamInfo* info = NULL;
info = g_hash_table_lookup (base_info->method_params, GUINT_TO_POINTER (methoddef_token));
if (!info) {
if (out_param_count_opt)
*out_param_count_opt = 0;
return 0;
}

if (out_param_count_opt)
*out_param_count_opt = info->param_count;

return mono_metadata_token_index (info->first_param_token);
}


static const char *
hot_reload_get_capabilities (void)
{
Expand Down
1 change: 1 addition & 0 deletions src/mono/mono/component/hot_reload.h
Original file line numberDiff line numberDiff line change
Expand Up@@ -48,6 +48,7 @@ typedef struct _MonoComponentHotReload {
uint32_t (*get_num_fields_added) (MonoClass *klass);
uint32_t (*get_num_methods_added) (MonoClass *klass);
const char* (*get_capabilities) (void);
uint32_t (*get_method_params) (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt);
} MonoComponentHotReload;

MONO_COMPONENT_EXPORT_ENTRYPOINT
Expand Down
Loading
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Remove or un-stick sticky/fixed headers that block content\n(function() {\n function unstick() {\n document.querySelectorAll('header, nav, [role=\"banner\"], .header, .navbar, .sticky, .fixed-top, [style*=\"position: fixed\"], [style*=\"position:sticky\"]').forEach(function(el) {\n if (el.style.position === 'fixed' || el.style.position === 'sticky' || \n getComputedStyle(el).position === 'fixed' || getComputedStyle(el).position === 'sticky') {\n el.style.position = 'static';\n el.style.top = 'auto';\n el.style.zIndex = 'auto';\n }\n });\n }\n \n unstick();\n \n var observer = new MutationObserver(unstick);\n observer.observe(document.body, { childList: true, subtree: true, attributes: true, attributeFilter: ['style', 'class'] });\n})();", "Kill Sticky Headers"); } } catch(__e) { console.warn('[Userscript:Kill Sticky Headers]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
using System;


namespace System.Reflection.Metadata.ApplyUpdate.Test
{
public class ReflectionAddNewMethod
{
public string ExistingMethod(string u, double f)
{
return u + f.ToString();;
}

}
}
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,21 @@
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
using System;
using System.Runtime.CompilerServices;
using CancellationToken = System.Threading.CancellationToken;

namespace System.Reflection.Metadata.ApplyUpdate.Test
{
public class ReflectionAddNewMethod
{
public string ExistingMethod(string u, double f)
{
return u + f.ToString();;
}

public double AddedNewMethod(char c, float h, string w, CancellationToken ct = default, [CallerMemberName] string callerName = "")
{
return ((double)Convert.ToInt32(c)) + h;
}
}
}
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
<Project Sdk="Microsoft.NET.Sdk">
<PropertyGroup>
<RootNamespace>System.Runtime.Loader.Tests</RootNamespace>
<TargetFramework>$(NetCoreAppCurrent)</TargetFramework>
<TestRuntime>true</TestRuntime>
<DeltaScript>deltascript.json</DeltaScript>
</PropertyGroup>
<ItemGroup>
<Compile Include="ReflectionAddNewMethod.cs" />
</ItemGroup>
</Project>
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
{
"changes": [
{"document": "ReflectionAddNewMethod.cs", "update": "ReflectionAddNewMethod_v1.cs"},
]
}

83 changes: 83 additions & 0 deletions src/libraries/System.Runtime.Loader/tests/ApplyUpdateTest.cs
Original file line numberDiff line numberDiff line change
Expand Up@@ -3,6 +3,7 @@

using System.Reflection;
using System.Runtime.CompilerServices;
using System.Runtime.InteropServices;
using Xunit;

namespace System.Reflection.Metadata
Expand DownExpand Up@@ -629,5 +630,87 @@ public static void TestReflectionAddNewType()
System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewType.ZExistingClass.ExistingMethod ();
});
}

[ConditionalFact(typeof(ApplyUpdateUtil), nameof(ApplyUpdateUtil.IsSupported))]
public static void TestReflectionAddNewMethod()
{
ApplyUpdateUtil.TestCase(static () =>
{
var ty = typeof(System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod);
var assm = ty.Assembly;

var bindingFlags = BindingFlags.Instance | BindingFlags.Public;
var allMethods = ty.GetMethods(bindingFlags);

int objectMethods = typeof(object).GetMethods(bindingFlags).Length;
Assert.Equal (objectMethods + 1, allMethods.Length);

ApplyUpdateUtil.ApplyUpdate(assm);
ApplyUpdateUtil.ClearAllReflectionCaches();

ty = typeof(System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod);

allMethods = ty.GetMethods(bindingFlags);
Assert.Equal (objectMethods + 2, allMethods.Length);

var mi = ty.GetMethod ("AddedNewMethod");

Assert.NotNull (mi);

var retParm = mi.ReturnParameter;
Assert.NotNull (retParm);
Assert.NotNull (retParm.ParameterType);
Assert.Equal (-1, retParm.Position);

var retCas = retParm.GetCustomAttributes(false);
Assert.NotNull(retCas);
Assert.Equal(0, retCas.Length);

var parms = mi.GetParameters();
Assert.Equal (5, parms.Length);

int parmPos = 0;
foreach (var parm in parms)
{
Assert.NotNull(parm);
Assert.NotNull(parm.ParameterType);
Assert.Equal(parmPos, parm.Position);
Assert.NotNull(parm.Name);

var cas = parm.GetCustomAttributes(false);
foreach (var ca in cas) {
Assert.NotNull (ca);
}

parmPos++;
}

var parmAttrs = parms[4].GetCustomAttributes(false);
Assert.Equal (2, parmAttrs.Length);
bool foundCallerMemberName = false;
bool foundOptional = false;
foreach (var pa in parmAttrs) {
if (typeof (CallerMemberNameAttribute).Equals(pa.GetType()))
{
foundCallerMemberName = true;
}
if (typeof (OptionalAttribute).Equals(pa.GetType()))
{
foundOptional = true;
}
}
Assert.True(foundCallerMemberName);
Assert.True(foundOptional);

// n.b. this typeof() also makes the rest of the test work on Wasm with aggressive trimming.
Assert.Equal (typeof(System.Threading.CancellationToken), parms[3].ParameterType);

Assert.True(parms[3].HasDefaultValue);
Assert.True(parms[4].HasDefaultValue);

Assert.Null(parms[3].DefaultValue);
Assert.Equal(string.Empty, parms[4].DefaultValue);
});
}
}
}
Original file line numberDiff line numberDiff line change
Expand Up@@ -62,6 +62,7 @@
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.AddStaticLambda\System.Reflection.Metadata.ApplyUpdate.Test.AddStaticLambda.csproj" />
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.StaticLambdaRegression\System.Reflection.Metadata.ApplyUpdate.Test.StaticLambdaRegression.csproj" />
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewType\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewType.csproj" />
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod.csproj" />
</ItemGroup>
<ItemGroup Condition="'$(TargetOS)' == 'Browser'">
<WasmFilesToIncludeFromPublishDir Include="$(AssemblyName).dll" />
Expand Down
5 changes: 5 additions & 0 deletions src/mono/mono/component/hot_reload-internals.h
Original file line numberDiff line numberDiff line change
Expand Up@@ -86,4 +86,9 @@ typedef struct _MonoClassMetadataUpdateEvent {
uint32_t token; /* the Event table token where this event was defined. */
} MonoClassMetadataUpdateEvent;

typedef struct _MonoMethodMetadataUpdateParamInfo {
uint32_t first_param_token; /* a Param token */
uint32_t param_count;
} MonoMethodMetadataUpdateParamInfo;

#endif/*_MONO_COMPONENT_HOT_RELOAD_INTERNALS_H*/
12 changes: 11 additions & 1 deletion src/mono/mono/component/hot_reload-stub.c
Original file line numberDiff line numberDiff line change
Expand Up@@ -110,6 +110,9 @@ hot_reload_get_num_methods_added (MonoClass *klass);
static const char *
hot_reload_get_capabilities (void);

static uint32_t
hot_reload_stub_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt);

static MonoComponentHotReload fn_table = {
{ MONO_COMPONENT_ITF_VERSION, &hot_reload_stub_available },
&hot_reload_stub_set_fastpath_data,
Expand DownExpand Up@@ -142,7 +145,8 @@ static MonoComponentHotReload fn_table = {
&hot_reload_stub_added_fields_iter,
&hot_reload_get_num_fields_added,
&hot_reload_get_num_methods_added,
&hot_reload_get_capabilities
&hot_reload_get_capabilities,
&hot_reload_stub_get_method_params,
};

static bool
Expand DownExpand Up@@ -343,6 +347,12 @@ hot_reload_get_capabilities (void)
return "";
}

static uint32_t
hot_reload_stub_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt)
{
return 0;
}

MONO_COMPONENT_EXPORT_ENTRYPOINT
MonoComponentHotReload *
mono_component_hot_reload_init (void)
Expand Down
91 changes: 87 additions & 4 deletions src/mono/mono/component/hot_reload.c
Original file line numberDiff line numberDiff line change
Expand Up@@ -144,6 +144,9 @@ hot_reload_get_num_methods_added (MonoClass *klass);
static const char *
hot_reload_get_capabilities (void);

static uint32_t
hot_reload_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt);

static MonoClassMetadataUpdateField *
metadata_update_field_setup_basic_info (MonoImage *image_base, BaselineInfo *base_info, uint32_t generation, DeltaInfo *delta_info, MonoClass *parent_klass, uint32_t fielddef_token, uint32_t field_flags);

Expand DownExpand Up@@ -179,7 +182,8 @@ static MonoComponentHotReload fn_table = {
&hot_reload_added_fields_iter,
&hot_reload_get_num_fields_added,
&hot_reload_get_num_methods_added,
&hot_reload_get_capabilities
&hot_reload_get_capabilities,
&hot_reload_get_method_params,
};

MonoComponentHotReload *
Expand DownExpand Up@@ -272,6 +276,9 @@ struct _BaselineInfo {
/* Parents for added methods, fields, etc */
GHashTable *member_parent; /* maps added methoddef or fielddef tokens to typedef tokens */

/* Params for added methods */
GHashTable *method_params; /* maps methoddef tokens to a MonoClassMetadataUpdateMethodParamInfo* */

/* Skeletons for all newly-added types from every generation. Accessing the array requires the image lock. */
GArray *skeletons;
};
Expand DownExpand Up@@ -412,6 +419,12 @@ baseline_info_destroy (BaselineInfo *info)
if (info->skeletons)
g_array_free (info->skeletons, TRUE);

if (info->member_parent)
g_hash_table_destroy (info->member_parent);

if (info->method_params)
g_hash_table_destroy (info->method_params);

g_free (info);
}

Expand DownExpand Up@@ -627,6 +640,11 @@ add_method_to_baseline (BaselineInfo *base_info, DeltaInfo *delta_info, MonoClas
static void
add_field_to_baseline (BaselineInfo *base_info, DeltaInfo *delta_info, MonoClass *klass, uint32_t field_token);

/* Add a method->params lookup for new methods in existing classes */
static void
add_param_info_for_method (BaselineInfo *base_info, uint32_t param_token, uint32_t method_token);


void
hot_reload_init (void)
{
Expand DownExpand Up@@ -2019,6 +2037,7 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base
uint32_t add_member_typedef = 0;
uint32_t add_property_propertymap = 0;
uint32_t add_event_eventmap = 0;
uint32_t add_field_method = 0;

gboolean assemblyref_updated = FALSE;
for (guint32 i = 0; i < rows ; ++i) {
Expand DownExpand Up@@ -2049,6 +2068,7 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base

case ENC_FUNC_ADD_PARAM: {
g_assert (token_table == MONO_TABLE_METHOD);
add_field_method = log_token;
break;
}
case ENC_FUNC_ADD_FIELD: {
Expand DownExpand Up@@ -2115,8 +2135,11 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base
}
case MONO_TABLE_METHOD: {
/* if adding a param, handle it with the next record */
if (func_code == ENC_FUNC_ADD_PARAM)
if (func_code == ENC_FUNC_ADD_PARAM) {
g_assert (is_addition);
break;
}
g_assert (func_code == ENC_FUNC_DEFAULT);

if (!base_info->method_table_update)
base_info->method_table_update = g_hash_table_new (g_direct_hash, g_direct_equal);
Expand DownExpand Up@@ -2366,9 +2389,21 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base
*
* So by the time we see the param additions, the methods are already in.
*
* FIXME: we need a lookaside table (like member_parent) for every place
* that looks at MONO_METHOD_PARAMLIST
*/
if (is_addition) {
g_assert (add_field_method != 0);
uint32_t parent_type_token = hot_reload_method_parent (image_base, add_field_method);
g_assert (parent_type_token != 0); // we added a parameter to a method that was added
if (pass2_context_is_skeleton (ctx, parent_type_token)) {
// it's a parameter on a new method in a brand new class
// FIXME: need to do something here?
} else {
// it's a parameter on a new method in an existing class
add_param_info_for_method (base_info, log_token, add_field_method);
}
add_field_method = 0;
break;
}
break;
}
case MONO_TABLE_INTERFACEIMPL: {
Expand DownExpand Up@@ -2819,6 +2854,28 @@ hot_reload_field_parent (MonoImage *base_image, uint32_t field_token)
}


static void
add_param_info_for_method (BaselineInfo *base_info, uint32_t param_token, uint32_t method_token)
{
if (!base_info->method_params) {
base_info->method_params = g_hash_table_new_full (g_direct_hash, g_direct_equal, NULL, g_free);
}
MonoMethodMetadataUpdateParamInfo* info = NULL;
info = g_hash_table_lookup (base_info->method_params, GUINT_TO_POINTER (method_token));
if (!info) {
// FIXME locking
info = g_new0 (MonoMethodMetadataUpdateParamInfo, 1);
g_hash_table_insert (base_info->method_params, GUINT_TO_POINTER (method_token), info);
info->first_param_token = param_token;
info->param_count = 1;
} else {
uint32_t param_index = mono_metadata_token_index (param_token);
// expect params for a single method to be a contiguous sequence of rows
g_assert (mono_metadata_token_index (info->first_param_token) + info->param_count == param_index);
info->param_count++;
}
}

/* HACK - keep in sync with locator_t in metadata/metadata.c */
typedef struct {
guint32 idx; /* The index that we are trying to locate */
Expand DownExpand Up@@ -3148,6 +3205,32 @@ hot_reload_get_num_methods_added (MonoClass *klass)
return count;
}

static uint32_t
hot_reload_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt)
{
BaselineInfo *base_info = baseline_info_lookup (base_image);
g_assert (base_info);

/* FIXME: locking in case the hash table grows */

if (!base_info->method_params)
return 0;

MonoMethodMetadataUpdateParamInfo* info = NULL;
info = g_hash_table_lookup (base_info->method_params, GUINT_TO_POINTER (methoddef_token));
if (!info) {
if (out_param_count_opt)
*out_param_count_opt = 0;
return 0;
}

if (out_param_count_opt)
*out_param_count_opt = info->param_count;

return mono_metadata_token_index (info->first_param_token);
}


static const char *
hot_reload_get_capabilities (void)
{
Expand Down
1 change: 1 addition & 0 deletions src/mono/mono/component/hot_reload.h
Original file line numberDiff line numberDiff line change
Expand Up@@ -48,6 +48,7 @@ typedef struct _MonoComponentHotReload {
uint32_t (*get_num_fields_added) (MonoClass *klass);
uint32_t (*get_num_methods_added) (MonoClass *klass);
const char* (*get_capabilities) (void);
uint32_t (*get_method_params) (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt);
} MonoComponentHotReload;

MONO_COMPONENT_EXPORT_ENTRYPOINT
Expand Down
Loading
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Universal Dark Mode - works on any site\n(function() {\n var enabled = true;\n \n function applyDarkMode() {\n if (!enabled) return;\n \n // Create style element if it doesn't exist\n var style = document.getElementById('universal-dark-mode-style');\n if (!style) {\n style = document.createElement('style');\n style.id = 'universal-dark-mode-style';\n document.head.appendChild(style);\n }\n \n // Dark mode CSS - inverts colors but preserves images/video\n style.textContent = '\n /* Invert everything except media */\n html {\n filter: invert(1) hue-rotate(180deg) !important;\n background: #1a1a2e !important;\n }\n \n /* Restore images, videos, iframes, canvas */\n img, video, iframe, canvas, svg, picture, [style*=\"background-image\"] {\n filter: invert(1) hue-rotate(180deg) !important;\n }\n \n /* Preserve specific elements that should not be inverted */\n .no-dark-mode, .no-dark-mode *,\n [data-theme=\"light\"], [data-theme=\"light\"],\n .ace_editor, .ace_editor *,\n .CodeMirror, .CodeMirror *,\n .monaco-editor, .monaco-editor *,\n .markdown-body pre, .markdown-body pre *,\n .highlight, .highlight *,\n pre code, pre code * {\n filter: none !important;\n }\n \n /* Fix common UI elements */\n .modal, .popup, .dropdown-menu, .tooltip, .popover {\n filter: invert(1) hue-rotate(180deg) !important;\n background: #2d2d44 !important;\n border-color: #444 !important;\n }\n \n /* Scrollbars */\n ::-webkit-scrollbar { background: #1a1a2e !important; }\n ::-webkit-scrollbar-thumb { background: #444 !important; }\n ::-webkit-scrollbar-thumb:hover { background: #555 !important; }\n \n /* Selection */\n ::selection { background: #4ecdc4 !important; color: #1a1a2e !important; }\n ::-moz-selection { background: #4ecdc4 !important; color: #1a1a2e !important; }\n ';\n }\n \n function removeDarkMode() {\n var style = document.getElementById('universal-dark-mode-style');\n if (style) style.remove();\n }\n \n // Toggle with Alt+Shift+D\n document.addEventListener('keydown', function(e) {\n if (e.altKey && e.shiftKey && e.key === 'D') {\n e.preventDefault();\n enabled = !enabled;\n if (enabled) {\n applyDarkMode();\n console.log('[Universal Dark Mode] Enabled');\n } else {\n removeDarkMode();\n console.log('[Universal Dark Mode] Disabled');\n }\n }\n });\n \n // Apply on load\n applyDarkMode();\n \n // Re-apply on dynamic content\n var observer = new MutationObserver(function(mutations) {\n if (enabled && !document.getElementById('universal-dark-mode-style')) {\n applyDarkMode();\n }\n });\n observer.observe(document.head, { childList: true });\n \n console.log('[Universal Dark Mode] Loaded - Press Alt+Shift+D to toggle');\n})();", "Universal Dark Mode"); } } catch(__e) { console.warn('[Userscript:Universal Dark Mode]', __e); } })(); })();
Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
using System;


namespace System.Reflection.Metadata.ApplyUpdate.Test
{
public class ReflectionAddNewMethod
{
public string ExistingMethod(string u, double f)
{
return u + f.ToString();;
}

}
}
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,21 @@
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
using System;
using System.Runtime.CompilerServices;
using CancellationToken = System.Threading.CancellationToken;

namespace System.Reflection.Metadata.ApplyUpdate.Test
{
public class ReflectionAddNewMethod
{
public string ExistingMethod(string u, double f)
{
return u + f.ToString();;
}

public double AddedNewMethod(char c, float h, string w, CancellationToken ct = default, [CallerMemberName] string callerName = "")
{
return ((double)Convert.ToInt32(c)) + h;
}
}
}
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
<Project Sdk="Microsoft.NET.Sdk">
<PropertyGroup>
<RootNamespace>System.Runtime.Loader.Tests</RootNamespace>
<TargetFramework>$(NetCoreAppCurrent)</TargetFramework>
<TestRuntime>true</TestRuntime>
<DeltaScript>deltascript.json</DeltaScript>
</PropertyGroup>
<ItemGroup>
<Compile Include="ReflectionAddNewMethod.cs" />
</ItemGroup>
</Project>
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
{
"changes": [
{"document": "ReflectionAddNewMethod.cs", "update": "ReflectionAddNewMethod_v1.cs"},
]
}

83 changes: 83 additions & 0 deletions src/libraries/System.Runtime.Loader/tests/ApplyUpdateTest.cs
Original file line numberDiff line numberDiff line change
Expand Up@@ -3,6 +3,7 @@

using System.Reflection;
using System.Runtime.CompilerServices;
using System.Runtime.InteropServices;
using Xunit;

namespace System.Reflection.Metadata
Expand DownExpand Up@@ -629,5 +630,87 @@ public static void TestReflectionAddNewType()
System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewType.ZExistingClass.ExistingMethod ();
});
}

[ConditionalFact(typeof(ApplyUpdateUtil), nameof(ApplyUpdateUtil.IsSupported))]
public static void TestReflectionAddNewMethod()
{
ApplyUpdateUtil.TestCase(static () =>
{
var ty = typeof(System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod);
var assm = ty.Assembly;

var bindingFlags = BindingFlags.Instance | BindingFlags.Public;
var allMethods = ty.GetMethods(bindingFlags);

int objectMethods = typeof(object).GetMethods(bindingFlags).Length;
Assert.Equal (objectMethods + 1, allMethods.Length);

ApplyUpdateUtil.ApplyUpdate(assm);
ApplyUpdateUtil.ClearAllReflectionCaches();

ty = typeof(System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod);

allMethods = ty.GetMethods(bindingFlags);
Assert.Equal (objectMethods + 2, allMethods.Length);

var mi = ty.GetMethod ("AddedNewMethod");

Assert.NotNull (mi);

var retParm = mi.ReturnParameter;
Assert.NotNull (retParm);
Assert.NotNull (retParm.ParameterType);
Assert.Equal (-1, retParm.Position);

var retCas = retParm.GetCustomAttributes(false);
Assert.NotNull(retCas);
Assert.Equal(0, retCas.Length);

var parms = mi.GetParameters();
Assert.Equal (5, parms.Length);

int parmPos = 0;
foreach (var parm in parms)
{
Assert.NotNull(parm);
Assert.NotNull(parm.ParameterType);
Assert.Equal(parmPos, parm.Position);
Assert.NotNull(parm.Name);

var cas = parm.GetCustomAttributes(false);
foreach (var ca in cas) {
Assert.NotNull (ca);
}

parmPos++;
}

var parmAttrs = parms[4].GetCustomAttributes(false);
Assert.Equal (2, parmAttrs.Length);
bool foundCallerMemberName = false;
bool foundOptional = false;
foreach (var pa in parmAttrs) {
if (typeof (CallerMemberNameAttribute).Equals(pa.GetType()))
{
foundCallerMemberName = true;
}
if (typeof (OptionalAttribute).Equals(pa.GetType()))
{
foundOptional = true;
}
}
Assert.True(foundCallerMemberName);
Assert.True(foundOptional);

// n.b. this typeof() also makes the rest of the test work on Wasm with aggressive trimming.
Assert.Equal (typeof(System.Threading.CancellationToken), parms[3].ParameterType);

Assert.True(parms[3].HasDefaultValue);
Assert.True(parms[4].HasDefaultValue);

Assert.Null(parms[3].DefaultValue);
Assert.Equal(string.Empty, parms[4].DefaultValue);
});
}
}
}
Original file line numberDiff line numberDiff line change
Expand Up@@ -62,6 +62,7 @@
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.AddStaticLambda\System.Reflection.Metadata.ApplyUpdate.Test.AddStaticLambda.csproj" />
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.StaticLambdaRegression\System.Reflection.Metadata.ApplyUpdate.Test.StaticLambdaRegression.csproj" />
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewType\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewType.csproj" />
<ProjectReference Include="ApplyUpdate\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod\System.Reflection.Metadata.ApplyUpdate.Test.ReflectionAddNewMethod.csproj" />
</ItemGroup>
<ItemGroup Condition="'$(TargetOS)' == 'Browser'">
<WasmFilesToIncludeFromPublishDir Include="$(AssemblyName).dll" />
Expand Down
5 changes: 5 additions & 0 deletions src/mono/mono/component/hot_reload-internals.h
Original file line numberDiff line numberDiff line change
Expand Up@@ -86,4 +86,9 @@ typedef struct _MonoClassMetadataUpdateEvent {
uint32_t token; /* the Event table token where this event was defined. */
} MonoClassMetadataUpdateEvent;

typedef struct _MonoMethodMetadataUpdateParamInfo {
uint32_t first_param_token; /* a Param token */
uint32_t param_count;
} MonoMethodMetadataUpdateParamInfo;

#endif/*_MONO_COMPONENT_HOT_RELOAD_INTERNALS_H*/
12 changes: 11 additions & 1 deletion src/mono/mono/component/hot_reload-stub.c
Original file line numberDiff line numberDiff line change
Expand Up@@ -110,6 +110,9 @@ hot_reload_get_num_methods_added (MonoClass *klass);
static const char *
hot_reload_get_capabilities (void);

static uint32_t
hot_reload_stub_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt);

static MonoComponentHotReload fn_table = {
{ MONO_COMPONENT_ITF_VERSION, &hot_reload_stub_available },
&hot_reload_stub_set_fastpath_data,
Expand DownExpand Up@@ -142,7 +145,8 @@ static MonoComponentHotReload fn_table = {
&hot_reload_stub_added_fields_iter,
&hot_reload_get_num_fields_added,
&hot_reload_get_num_methods_added,
&hot_reload_get_capabilities
&hot_reload_get_capabilities,
&hot_reload_stub_get_method_params,
};

static bool
Expand DownExpand Up@@ -343,6 +347,12 @@ hot_reload_get_capabilities (void)
return "";
}

static uint32_t
hot_reload_stub_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt)
{
return 0;
}

MONO_COMPONENT_EXPORT_ENTRYPOINT
MonoComponentHotReload *
mono_component_hot_reload_init (void)
Expand Down
91 changes: 87 additions & 4 deletions src/mono/mono/component/hot_reload.c
Original file line numberDiff line numberDiff line change
Expand Up@@ -144,6 +144,9 @@ hot_reload_get_num_methods_added (MonoClass *klass);
static const char *
hot_reload_get_capabilities (void);

static uint32_t
hot_reload_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt);

static MonoClassMetadataUpdateField *
metadata_update_field_setup_basic_info (MonoImage *image_base, BaselineInfo *base_info, uint32_t generation, DeltaInfo *delta_info, MonoClass *parent_klass, uint32_t fielddef_token, uint32_t field_flags);

Expand DownExpand Up@@ -179,7 +182,8 @@ static MonoComponentHotReload fn_table = {
&hot_reload_added_fields_iter,
&hot_reload_get_num_fields_added,
&hot_reload_get_num_methods_added,
&hot_reload_get_capabilities
&hot_reload_get_capabilities,
&hot_reload_get_method_params,
};

MonoComponentHotReload *
Expand DownExpand Up@@ -272,6 +276,9 @@ struct _BaselineInfo {
/* Parents for added methods, fields, etc */
GHashTable *member_parent; /* maps added methoddef or fielddef tokens to typedef tokens */

/* Params for added methods */
GHashTable *method_params; /* maps methoddef tokens to a MonoClassMetadataUpdateMethodParamInfo* */

/* Skeletons for all newly-added types from every generation. Accessing the array requires the image lock. */
GArray *skeletons;
};
Expand DownExpand Up@@ -412,6 +419,12 @@ baseline_info_destroy (BaselineInfo *info)
if (info->skeletons)
g_array_free (info->skeletons, TRUE);

if (info->member_parent)
g_hash_table_destroy (info->member_parent);

if (info->method_params)
g_hash_table_destroy (info->method_params);

g_free (info);
}

Expand DownExpand Up@@ -627,6 +640,11 @@ add_method_to_baseline (BaselineInfo *base_info, DeltaInfo *delta_info, MonoClas
static void
add_field_to_baseline (BaselineInfo *base_info, DeltaInfo *delta_info, MonoClass *klass, uint32_t field_token);

/* Add a method->params lookup for new methods in existing classes */
static void
add_param_info_for_method (BaselineInfo *base_info, uint32_t param_token, uint32_t method_token);


void
hot_reload_init (void)
{
Expand DownExpand Up@@ -2019,6 +2037,7 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base
uint32_t add_member_typedef = 0;
uint32_t add_property_propertymap = 0;
uint32_t add_event_eventmap = 0;
uint32_t add_field_method = 0;

gboolean assemblyref_updated = FALSE;
for (guint32 i = 0; i < rows ; ++i) {
Expand DownExpand Up@@ -2049,6 +2068,7 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base

case ENC_FUNC_ADD_PARAM: {
g_assert (token_table == MONO_TABLE_METHOD);
add_field_method = log_token;
break;
}
case ENC_FUNC_ADD_FIELD: {
Expand DownExpand Up@@ -2115,8 +2135,11 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base
}
case MONO_TABLE_METHOD: {
/* if adding a param, handle it with the next record */
if (func_code == ENC_FUNC_ADD_PARAM)
if (func_code == ENC_FUNC_ADD_PARAM) {
g_assert (is_addition);
break;
}
g_assert (func_code == ENC_FUNC_DEFAULT);

if (!base_info->method_table_update)
base_info->method_table_update = g_hash_table_new (g_direct_hash, g_direct_equal);
Expand DownExpand Up@@ -2366,9 +2389,21 @@ apply_enclog_pass2 (Pass2Context *ctx, MonoImage *image_base, BaselineInfo *base
*
* So by the time we see the param additions, the methods are already in.
*
* FIXME: we need a lookaside table (like member_parent) for every place
* that looks at MONO_METHOD_PARAMLIST
*/
if (is_addition) {
g_assert (add_field_method != 0);
uint32_t parent_type_token = hot_reload_method_parent (image_base, add_field_method);
g_assert (parent_type_token != 0); // we added a parameter to a method that was added
if (pass2_context_is_skeleton (ctx, parent_type_token)) {
// it's a parameter on a new method in a brand new class
// FIXME: need to do something here?
} else {
// it's a parameter on a new method in an existing class
add_param_info_for_method (base_info, log_token, add_field_method);
}
add_field_method = 0;
break;
}
break;
}
case MONO_TABLE_INTERFACEIMPL: {
Expand DownExpand Up@@ -2819,6 +2854,28 @@ hot_reload_field_parent (MonoImage *base_image, uint32_t field_token)
}


static void
add_param_info_for_method (BaselineInfo *base_info, uint32_t param_token, uint32_t method_token)
{
if (!base_info->method_params) {
base_info->method_params = g_hash_table_new_full (g_direct_hash, g_direct_equal, NULL, g_free);
}
MonoMethodMetadataUpdateParamInfo* info = NULL;
info = g_hash_table_lookup (base_info->method_params, GUINT_TO_POINTER (method_token));
if (!info) {
// FIXME locking
info = g_new0 (MonoMethodMetadataUpdateParamInfo, 1);
g_hash_table_insert (base_info->method_params, GUINT_TO_POINTER (method_token), info);
info->first_param_token = param_token;
info->param_count = 1;
} else {
uint32_t param_index = mono_metadata_token_index (param_token);
// expect params for a single method to be a contiguous sequence of rows
g_assert (mono_metadata_token_index (info->first_param_token) + info->param_count == param_index);
info->param_count++;
}
}

/* HACK - keep in sync with locator_t in metadata/metadata.c */
typedef struct {
guint32 idx; /* The index that we are trying to locate */
Expand DownExpand Up@@ -3148,6 +3205,32 @@ hot_reload_get_num_methods_added (MonoClass *klass)
return count;
}

static uint32_t
hot_reload_get_method_params (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt)
{
BaselineInfo *base_info = baseline_info_lookup (base_image);
g_assert (base_info);

/* FIXME: locking in case the hash table grows */

if (!base_info->method_params)
return 0;

MonoMethodMetadataUpdateParamInfo* info = NULL;
info = g_hash_table_lookup (base_info->method_params, GUINT_TO_POINTER (methoddef_token));
if (!info) {
if (out_param_count_opt)
*out_param_count_opt = 0;
return 0;
}

if (out_param_count_opt)
*out_param_count_opt = info->param_count;

return mono_metadata_token_index (info->first_param_token);
}


static const char *
hot_reload_get_capabilities (void)
{
Expand Down
1 change: 1 addition & 0 deletions src/mono/mono/component/hot_reload.h
Original file line numberDiff line numberDiff line change
Expand Up@@ -48,6 +48,7 @@ typedef struct _MonoComponentHotReload {
uint32_t (*get_num_fields_added) (MonoClass *klass);
uint32_t (*get_num_methods_added) (MonoClass *klass);
const char* (*get_capabilities) (void);
uint32_t (*get_method_params) (MonoImage *base_image, uint32_t methoddef_token, uint32_t *out_param_count_opt);
} MonoComponentHotReload;

MONO_COMPONENT_EXPORT_ENTRYPOINT
Expand Down
Loading