Source file test/uintptrescapes.dir/main.go

     1  // Copyright 2016 The Go Authors. All rights reserved.
     2  // Use of this source code is governed by a BSD-style
     3  // license that can be found in the LICENSE file.
     4  
     5  package main
     6  
     7  import (
     8  	"fmt"
     9  	"os"
    10  	"sync"
    11  	"unsafe"
    12  
    13  	"./a"
    14  )
    15  
    16  func F1() int {
    17  	var buf [1024]int
    18  	a.F1(uintptr(unsafe.Pointer(&buf[0])))
    19  	return buf[0]
    20  }
    21  
    22  func F2() int {
    23  	var buf [1024]int
    24  	a.F2(uintptr(unsafe.Pointer(&buf[0])))
    25  	return buf[0]
    26  }
    27  
    28  var t = a.GetT()
    29  
    30  func M1() int {
    31  	var buf [1024]int
    32  	t.M1(uintptr(unsafe.Pointer(&buf[0])))
    33  	return buf[0]
    34  }
    35  
    36  func M2() int {
    37  	var buf [1024]int
    38  	t.M2(uintptr(unsafe.Pointer(&buf[0])))
    39  	return buf[0]
    40  }
    41  
    42  func main() {
    43  	// Use different goroutines to force stack growth.
    44  	var wg sync.WaitGroup
    45  	wg.Add(4)
    46  	c := make(chan bool, 4)
    47  
    48  	go func() {
    49  		defer wg.Done()
    50  		b := F1()
    51  		if b != 42 {
    52  			fmt.Printf("F1: got %d, expected 42\n", b)
    53  			c <- false
    54  		}
    55  	}()
    56  
    57  	go func() {
    58  		defer wg.Done()
    59  		b := F2()
    60  		if b != 42 {
    61  			fmt.Printf("F2: got %d, expected 42\n", b)
    62  			c <- false
    63  		}
    64  	}()
    65  
    66  	go func() {
    67  		defer wg.Done()
    68  		b := M1()
    69  		if b != 42 {
    70  			fmt.Printf("M1: got %d, expected 42\n", b)
    71  			c <- false
    72  		}
    73  	}()
    74  
    75  	go func() {
    76  		defer wg.Done()
    77  		b := M2()
    78  		if b != 42 {
    79  			fmt.Printf("M2: got %d, expected 42\n", b)
    80  			c <- false
    81  		}
    82  	}()
    83  
    84  	wg.Wait()
    85  
    86  	select {
    87  	case <-c:
    88  		os.Exit(1)
    89  	default:
    90  	}
    91  }
    92  

View as plain text