platform_bionic/tests/sys_uio_test.cpp
Elliott Hughes 95646e6666 Add ASSERT_ERRNO and EXPECT_ERRNO (and use them).
We've talked about this many times in the past, but partners struggle to
understand "expected 38, got 22" in these contexts, and I always have to
go and check the header files just to be sure I'm sure.

I actually think the glibc geterrorname_np() function (which would
return "ENOSYS" rather than "Function not implemented") would be more
helpful, but I'll have to go and implement that first, and then come
back.

Being forced to go through all our errno assertions did also make me
want to use a more consistent style for our ENOSYS assertions in
particular --- there's a particularly readable idiom, and I'll also come
back and move more of those checks to the most readable idiom.

I've added a few missing `errno = 0`s before tests, and removed a few
stray `errno = 0`s from tests that don't actually make assertions about
errno, since I had to look at every single reference to errno anyway.

Test: treehugger
Change-Id: Iba7c56f2adc30288c3e00ade106635e515e88179
2023-09-21 14:15:59 -07:00

151 lines
4.3 KiB
C++

/*
* Copyright (C) 2015 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include <gtest/gtest.h>
#include <errno.h>
#include <string.h>
#include <sys/types.h>
#include <sys/uio.h>
#include <unistd.h>
#include <android-base/file.h>
#include "utils.h"
TEST(sys_uio, readv_writev) {
TemporaryFile tf;
char buf1[] = "hello";
char buf2[] = "world";
iovec ios[] = { { buf1, 5 }, { buf2, 5 } };
ASSERT_EQ(10, writev(tf.fd, ios, 2));
ASSERT_EQ(0, lseek(tf.fd, 0, SEEK_SET));
memset(buf1, '1', sizeof(buf1));
memset(buf2, '2', sizeof(buf2));
ASSERT_EQ(10, readv(tf.fd, ios, 2));
buf1[5] = buf2[5] = '\0';
ASSERT_STREQ("hello", buf1);
ASSERT_STREQ("world", buf2);
}
template <typename ReadFn, typename WriteFn>
void TestPreadVPwriteV(ReadFn read_fn, WriteFn write_fn) {
TemporaryFile tf;
char buf[] = "world";
iovec ios[] = { { buf, 5 } };
ASSERT_EQ(5, write_fn(tf.fd, ios, 1, 5));
ASSERT_EQ(0, lseek(tf.fd, 0, SEEK_CUR));
strcpy(buf, "hello");
ASSERT_EQ(5, write_fn(tf.fd, ios, 1, 0));
ASSERT_EQ(0, lseek(tf.fd, 0, SEEK_CUR));
ASSERT_EQ(5, read_fn(tf.fd, ios, 1, 5));
ASSERT_STREQ("world", buf);
ASSERT_EQ(5, read_fn(tf.fd, ios, 1, 0));
ASSERT_STREQ("hello", buf);
}
TEST(sys_uio, preadv_pwritev) {
TestPreadVPwriteV(preadv, pwritev);
}
TEST(sys_uio, preadv64_pwritev64) {
TestPreadVPwriteV(preadv64, pwritev64);
}
template <typename ReadFn, typename WriteFn>
void TestPreadV2PwriteV2(ReadFn read_fn, WriteFn write_fn) {
TemporaryFile tf;
char buf[] = "world";
iovec ios[] = {{buf, 5}};
ASSERT_EQ(5, write_fn(tf.fd, ios, 1, 5, 0)) << strerror(errno);
ASSERT_EQ(0, lseek(tf.fd, 0, SEEK_CUR));
strcpy(buf, "hello");
ASSERT_EQ(5, write_fn(tf.fd, ios, 1, 0, 0)) << strerror(errno);
ASSERT_EQ(0, lseek(tf.fd, 0, SEEK_CUR));
ASSERT_EQ(5, read_fn(tf.fd, ios, 1, 5, 0)) << strerror(errno);
ASSERT_STREQ("world", buf);
ASSERT_EQ(5, read_fn(tf.fd, ios, 1, 0, 0)) << strerror(errno);
ASSERT_STREQ("hello", buf);
}
TEST(sys_uio, preadv2_pwritev2) {
#if defined(__BIONIC__)
TestPreadV2PwriteV2(preadv2, pwritev2);
#else
GTEST_SKIP() << "preadv2/pwritev2 not available";
#endif
}
TEST(sys_uio, preadv64v2_pwritev64v2) {
#if defined(__BIONIC__)
TestPreadV2PwriteV2(preadv64v2, pwritev64v2);
#else
GTEST_SKIP() << "preadv2/pwritev2 not available";
#endif
}
TEST(sys_uio, process_vm_readv) {
ASSERT_EQ(0, process_vm_readv(0, nullptr, 0, nullptr, 0, 0));
// Test that we can read memory from our own process
char src[1024] = "This is the source buffer containing some data";
char dst[1024] = "";
iovec remote = { src, sizeof src };
iovec local = { dst, sizeof dst };
ASSERT_EQ(ssize_t(sizeof src), process_vm_readv(getpid(), &local, 1, &remote, 1, 0));
// Check whether data was copied (in the correct direction)
ASSERT_EQ('T', dst[0]);
ASSERT_EQ(0, memcmp(src, dst, sizeof src));
// Reading from non-allocated memory should return an error
remote = { nullptr, sizeof dst };
errno = 0;
ASSERT_EQ(-1, process_vm_readv(getpid(), &local, 1, &remote, 1, 0));
ASSERT_ERRNO(EFAULT);
}
TEST(sys_uio, process_vm_writev) {
ASSERT_EQ(0, process_vm_writev(0, nullptr, 0, nullptr, 0, 0));
// Test that we can read memory from our own process
char src[1024] = "This is the source buffer containing some data";
char dst[1024] = "";
iovec remote = { dst, sizeof dst };
iovec local = { src, sizeof src };
ASSERT_EQ(ssize_t(sizeof src), process_vm_writev(getpid(), &local, 1, &remote, 1, 0));
// Check whether data was copied (in the correct direction)
ASSERT_EQ('T', dst[0]);
ASSERT_EQ(0, memcmp(src, dst, sizeof src));
// Writing to non-allocated memory should return an error
remote = { nullptr, sizeof dst };
errno = 0;
ASSERT_EQ(-1, process_vm_writev(getpid(), &local, 1, &remote, 1, 0));
ASSERT_ERRNO(EFAULT);
}