This is xnu-12377.1.9. See this file in:
/*
* Copyright (c) 2024 Apple Inc. All rights reserved.
*
* @APPLE_OSREFERENCE_LICENSE_HEADER_START@
*
* This file contains Original Code and/or Modifications of Original Code
* as defined in and that are subject to the Apple Public Source License
* Version 2.0 (the 'License'). You may not use this file except in
* compliance with the License. The rights granted to you under the License
* may not be used to create, or enable the creation or redistribution of,
* unlawful or unlicensed copies of an Apple operating system, or to
* circumvent, violate, or enable the circumvention or violation of, any
* terms of an Apple operating system software license agreement.
*
* Please obtain a copy of the License at
* http://www.opensource.apple.com/apsl/ and read it before using this file.
*
* The Original Code and all software distributed under the License are
* distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
* EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
* INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
* Please see the License for the specific language governing rights and
* limitations under the License.
*
* @APPLE_OSREFERENCE_LICENSE_HEADER_END@
*/
#include <darwintest.h>
#include <darwintest_utils.h>
#include <errno.h>
#include <stdio.h>
#include <fcntl.h>
#include <stdlib.h>
#include <string.h>
#include <sys/mman.h>
#include <sys/uio.h>
T_GLOBAL_META(
T_META_NAMESPACE("xnu.vm"),
T_META_RADAR_COMPONENT_NAME("xnu"),
T_META_RADAR_COMPONENT_VERSION("VM"),
T_META_OWNER("francois"),
T_META_CHECK_LEAKS(true),
T_META_RUN_CONCURRENTLY(true));
T_DECL(direct_write_wired,
"test direct write() to wired file", T_META_TAG_VM_PREFERRED)
{
int fd;
int ret;
char tmpf[PATH_MAX] = "";
char *map_addr;
size_t file_size;
ssize_t num_bytes;
char *buf;
T_SETUPBEGIN;
strlcpy(tmpf, dt_tmpdir(), PATH_MAX);
strlcat(tmpf, "/wire_direct_write.txt", PATH_MAX);
T_LOG("file name: <%s>\n", tmpf);
fd = open(tmpf, O_RDWR | O_CREAT | O_TRUNC, 0644);
T_QUIET; T_ASSERT_POSIX_SUCCESS(fd, "open()");
file_size = PAGE_SIZE;
ret = ftruncate(fd, (off_t) file_size);
T_QUIET; T_ASSERT_POSIX_SUCCESS(fd, "ftruncate()");
map_addr = mmap(NULL, file_size, PROT_READ | PROT_WRITE, MAP_FILE | MAP_SHARED, fd, 0);
T_QUIET; T_ASSERT_POSIX_SUCCESS(map_addr, "mmap()");
memset(map_addr, 'a', file_size);
ret = mlock(map_addr, file_size);
T_QUIET; T_ASSERT_POSIX_SUCCESS(ret, "mlock()");
ret = fcntl(fd, F_NOCACHE, true);
T_QUIET; T_ASSERT_POSIX_SUCCESS(ret, "fcntl(F_NOCACHE)");
buf = valloc(file_size);
T_QUIET; T_ASSERT_NOTNULL(buf, "valloc()");
memset(buf, 'b', file_size);
T_SETUPEND;
num_bytes = pwrite(fd, buf, file_size, 0);
T_QUIET; T_EXPECT_POSIX_SUCCESS(num_bytes, "write()");
T_QUIET; T_EXPECT_EQ(num_bytes, (ssize_t) file_size, "wrote <file_size> bytes");
ret = munlock(map_addr, file_size);
T_ASSERT_POSIX_SUCCESS(ret, "munlock()");
T_ASSERT_EQ(map_addr[0], 'b', "write() worked");
T_QUIET; T_ASSERT_POSIX_SUCCESS(close(fd), "close()");
T_QUIET; T_ASSERT_POSIX_SUCCESS(unlink(tmpf), "unlink(%s)", tmpf);
}
#define NUM_IOVS 7
T_DECL(direct_write_wired_vector_upl,
"test direct write() to wired file with vector UIO",
T_META_TAG_VM_PREFERRED)
{
char tmpf[PATH_MAX] = "";
struct iovec w_iovs[NUM_IOVS];
struct iovec r_iovs[NUM_IOVS];
int fd;
ssize_t bytes, expected_bytes;
int w, r;
char *w_base, *r_base;
int w_cnt, r_cnt;
int w_idx, r_idx;
char *map_addr;
int ret;
char *buf;
T_SETUPBEGIN;
expected_bytes = 0;
for (w = 0, r = NUM_IOVS - 1; w < NUM_IOVS; w++, r--) {
w_iovs[w].iov_len = (size_t) ((w + 1) * (int)PAGE_SIZE);
T_QUIET; T_ASSERT_POSIX_SUCCESS(posix_memalign(&w_iovs[w].iov_base, PAGE_SIZE, w_iovs[w].iov_len), "alloc(w_iov_base[%d])", w);
memset(w_iovs[w].iov_base, 'a' + w, w_iovs[w].iov_len);
expected_bytes += w_iovs[w].iov_len;
r_iovs[r].iov_len = w_iovs[w].iov_len;
T_QUIET; T_ASSERT_POSIX_SUCCESS(posix_memalign(&r_iovs[r].iov_base, PAGE_SIZE, r_iovs[r].iov_len), "alloc(r_iov_base[%d])", r);
}
strlcpy(tmpf, dt_tmpdir(), PATH_MAX);
strlcat(tmpf, "/wire_direct_write.txt", PATH_MAX);
T_QUIET; T_ASSERT_LT(strlen(tmpf), (unsigned long)PATH_MAX,
"path exceeds PATH_MAX");
fd = open(tmpf, O_RDWR | O_CREAT | O_TRUNC);
T_QUIET; T_ASSERT_POSIX_SUCCESS(fd, "open()");
/* use F_NOCACHE to get direct I/O */
T_QUIET; T_ASSERT_POSIX_SUCCESS(fcntl(fd, F_NOCACHE, 1), "fcntl(F_NOCACHE)");
buf = valloc(expected_bytes);
T_QUIET; T_ASSERT_POSIX_SUCCESS(buf, "valloc()");
memset(buf, '?', expected_bytes);
bytes = pwrite(fd, buf, expected_bytes, 0);
T_QUIET; T_ASSERT_POSIX_SUCCESS(bytes, "pwrite()");
T_ASSERT_EQ(bytes, expected_bytes,
"pwrite() wrote %ld bytes", bytes);
T_SETUPEND;
map_addr = mmap(NULL, expected_bytes, PROT_READ | PROT_WRITE,
MAP_FILE | MAP_SHARED, fd, 0);
T_QUIET; T_ASSERT_POSIX_SUCCESS(map_addr, "mmap()");
ret = mlock(&map_addr[(NUM_IOVS / 2) * PAGE_SIZE], PAGE_SIZE);
T_ASSERT_POSIX_SUCCESS(ret, "mlock(0x%llx)", (uint64_t)((NUM_IOVS / 2) * PAGE_SIZE));
bytes = pwritev(fd, w_iovs, NUM_IOVS, 0);
T_ASSERT_EQ(bytes, expected_bytes,
"pwritev() wrote %ld bytes", bytes);
bytes = preadv(fd, r_iovs, NUM_IOVS, 0);
T_ASSERT_EQ(bytes, expected_bytes,
"preadv() read %ld bytes", bytes);
w = 0;
w_base = w_iovs[w].iov_base;
w_cnt = (int) w_iovs[w].iov_len;
w_idx = 0;
for (r = 0; r < NUM_IOVS; r++) {
r_base = r_iovs[r].iov_base;
r_cnt = (int) r_iovs[r].iov_len;
for (r_idx = 0; r_idx < r_cnt; r_idx++) {
T_QUIET; T_ASSERT_EQ(r_base[r_idx], w_base[w_idx],
"r_iovs[%d].iov_base[%d]='%c' == w_iovs[%d].iov_base[%d]='%c'",
r, r_idx, (unsigned char)r_base[r_idx],
w, w_idx, (unsigned char) w_base[w_idx]);
if (++w_idx == w_cnt) {
w++;
w_base = w_iovs[w].iov_base;
w_cnt = (int) w_iovs[w].iov_len;
w_idx = 0;
}
}
}
T_QUIET; T_ASSERT_POSIX_SUCCESS(close(fd), "close()");
T_QUIET; T_ASSERT_POSIX_SUCCESS(unlink(tmpf), "unlink(%s)", tmpf);
T_PASS("%s", __FUNCTION__);
}