Use correct env for writing local vars

Author: liufengyun

compiler/src/dotty/tools/dotc/transform/init/Objects.scala

@@ -753,6 +753,7 @@ object Objects:
     assert(sym.is(Flags.Mutable), "Writing to immutable variable " + sym.show)
     Env.resolveEnv(sym.enclosingMethod, thisV, summon[Env.Data]) match
     case Some(thisV -> env) =>
+      given Env.Data = env
       val addr = Env.varAddr(sym)
       if addr.owner != State.currentObject then
         errorMutateOtherStaticObject(State.currentObject, addr.owner)
@@ -1092,7 +1093,12 @@ object Objects:
     // init param fields
     klass.paramGetters.foreach { acc =>
       val value = paramsMap(acc.name.toTermName)
-      thisV.initVal(acc, value)
+      if acc.is(Flags.Mutable) then
+        val addr = Heap.fieldVarAddr(thisV, acc, State.currentObject)
+        thisV.initVar(acc, addr)
+        Heap.write(addr, value)
+      else
+        thisV.initVal(acc, value)
       printer.println(acc.show + " initialized with " + value)
     }
 

tests/init/neg/global-irrelevance4.scala

@@ -1,11 +1,15 @@
+import scala.annotation.init
+
 object A:
   class Pair(val f: Int => Unit, val g: () => Int)
   val p: Pair = foo()
 
   def foo(): Pair =
     var x = 6
-    new Pair(y => x = y, () => x)
-
+    new Pair(
+      (y => x = y): @init.expose,   // error
+      () => x
+    )
 
 object B:
-  A.p.f(10)  // error
+  A.p.f(10)