mirror of
https://github.com/golang/sys.git
synced 2026-01-29 07:02:06 +03:00
123459f096
- Add a wrapper function around fcntl to handle different operation
types and new fcntl implementation that accepts uintptr as an arg.
- Add support for calling mmap/munmap with address pointers.
- Add accompanying tests for new functions.
Change-Id: If5e77aa4cf2cccfd431de4f3bd0c5014a761e167
GitHub-Last-Rev: 07e32a4ab7
GitHub-Pull-Request: golang/sys#216
Reviewed-on: https://go-review.googlesource.com/c/sys/+/610296
Reviewed-by: Ian Lance Taylor <iant@google.com>
Reviewed-by: Dmitri Shuralyov <dmitshur@google.com>
Auto-Submit: Dmitri Shuralyov <dmitshur@golang.org>
TryBot-Bypass: Dmitri Shuralyov <dmitshur@golang.org>
89 lines
2.1 KiB
Go
89 lines
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)
|
|
}
|
|
}
|