liyujie
2025-08-28 d9927380ed7c8366f762049be9f3fee225860833
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
// Copyright 2016 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
 
package ssa
 
import (
   "cmd/compile/internal/types"
   "testing"
)
 
func TestWriteBarrierStoreOrder(t *testing.T) {
   // Make sure writebarrier phase works even StoreWB ops are not in dependency order
   c := testConfig(t)
   ptrType := c.config.Types.BytePtr
   fun := c.Fun("entry",
       Bloc("entry",
           Valu("start", OpInitMem, types.TypeMem, 0, nil),
           Valu("sb", OpSB, c.config.Types.Uintptr, 0, nil),
           Valu("sp", OpSP, c.config.Types.Uintptr, 0, nil),
           Valu("v", OpConstNil, ptrType, 0, nil),
           Valu("addr1", OpAddr, ptrType, 0, nil, "sb"),
           Valu("wb2", OpStore, types.TypeMem, 0, ptrType, "addr1", "v", "wb1"),
           Valu("wb1", OpStore, types.TypeMem, 0, ptrType, "addr1", "v", "start"), // wb1 and wb2 are out of order
           Goto("exit")),
       Bloc("exit",
           Exit("wb2")))
 
   CheckFunc(fun.f)
   writebarrier(fun.f)
   CheckFunc(fun.f)
}
 
func TestWriteBarrierPhi(t *testing.T) {
   // Make sure writebarrier phase works for single-block loop, where
   // a Phi op takes the store in the same block as argument.
   // See issue #19067.
   c := testConfig(t)
   ptrType := c.config.Types.BytePtr
   fun := c.Fun("entry",
       Bloc("entry",
           Valu("start", OpInitMem, types.TypeMem, 0, nil),
           Valu("sb", OpSB, c.config.Types.Uintptr, 0, nil),
           Valu("sp", OpSP, c.config.Types.Uintptr, 0, nil),
           Goto("loop")),
       Bloc("loop",
           Valu("phi", OpPhi, types.TypeMem, 0, nil, "start", "wb"),
           Valu("v", OpConstNil, ptrType, 0, nil),
           Valu("addr", OpAddr, ptrType, 0, nil, "sb"),
           Valu("wb", OpStore, types.TypeMem, 0, ptrType, "addr", "v", "phi"), // has write barrier
           Goto("loop")))
 
   CheckFunc(fun.f)
   writebarrier(fun.f)
   CheckFunc(fun.f)
}