зеркало из https://github.com/golang/sys.git
89 строки
2.1 KiB
Go
89 строки
2.1 KiB
Go
// Copyright 2020 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.
|
|
|
|
//go:build zos && s390x
|
|
|
|
// This test is based on mmap_unix_test, but tweaked for z/OS, which does not support memadvise
|
|
// or anonymous mmapping.
|
|
|
|
package unix_test
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"golang.org/x/sys/unix"
|
|
)
|
|
|
|
func TestMmap(t *testing.T) {
|
|
tempdir := t.TempDir()
|
|
filename := filepath.Join(tempdir, "memmapped_file")
|
|
|
|
destination, err := os.OpenFile(filename, os.O_WRONLY|os.O_CREATE, 0700)
|
|
if err != nil {
|
|
t.Fatal("os.Create:", err)
|
|
return
|
|
}
|
|
|
|
fmt.Fprintf(destination, "%s\n", "0 <- Flipped between 0 and 1 when test runs successfully")
|
|
fmt.Fprintf(destination, "%s\n", "//Do not change contents - mmap test relies on this")
|
|
destination.Close()
|
|
|
|
fd, err := unix.Open(filename, unix.O_RDWR, 0777)
|
|
if err != nil {
|
|
t.Fatalf("Open: %v", err)
|
|
}
|
|
|
|
b, err := unix.Mmap(fd, 0, 8, unix.PROT_READ, unix.MAP_SHARED)
|
|
if err != nil {
|
|
t.Fatalf("Mmap: %v", err)
|
|
}
|
|
|
|
if err := unix.Mprotect(b, unix.PROT_READ|unix.PROT_WRITE); err != nil {
|
|
t.Fatalf("Mprotect: %v", err)
|
|
}
|
|
|
|
// Flip flag in test file via mapped memory
|
|
flagWasZero := true
|
|
if b[0] == '0' {
|
|
b[0] = '1'
|
|
} else if b[0] == '1' {
|
|
b[0] = '0'
|
|
flagWasZero = false
|
|
}
|
|
|
|
if err := unix.Msync(b, unix.MS_SYNC); err != nil {
|
|
t.Fatalf("Msync: %v", err)
|
|
}
|
|
|
|
// Read file from FS to ensure flag flipped after msync
|
|
buf, err := os.ReadFile(filename)
|
|
if err != nil {
|
|
t.Fatalf("Could not read mmapped file from disc for test: %v", err)
|
|
}
|
|
if flagWasZero && buf[0] != '1' || !flagWasZero && buf[0] != '0' {
|
|
t.Error("Flag flip in MAP_SHARED mmapped file not visible")
|
|
}
|
|
|
|
if err := unix.Munmap(b); err != nil {
|
|
t.Fatalf("Munmap: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestMmapPtr(t *testing.T) {
|
|
p, err := unix.MmapPtr(-1, 0, nil, uintptr(2*unix.Getpagesize()),
|
|
unix.PROT_READ|unix.PROT_WRITE, unix.MAP_ANON|unix.MAP_PRIVATE)
|
|
if err != nil {
|
|
t.Fatalf("MmapPtr: %v", err)
|
|
}
|
|
|
|
*(*byte)(p) = 42
|
|
|
|
if err := unix.MunmapPtr(p, uintptr(2*unix.Getpagesize())); err != nil {
|
|
t.Fatalf("MunmapPtr: %v", err)
|
|
}
|
|
}
|