/*
* 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 "sysdeps.h"
#include <android-base/test_utils.h>
TEST(sysdeps_win32, adb_getenv) {
// Insert all test env vars before first call to adb_getenv() which will
// read the env var block only once.
ASSERT_EQ(0, _putenv("SYSDEPS_WIN32_TEST_UPPERCASE=1"));
ASSERT_EQ(0, _putenv("sysdeps_win32_test_lowercase=2"));
ASSERT_EQ(0, _putenv("Sysdeps_Win32_Test_MixedCase=3"));
// UTF-16 value
ASSERT_EQ(0, _wputenv(L"SYSDEPS_WIN32_TEST_UNICODE=\u00a1\u0048\u006f\u006c"
L"\u0061\u0021\u03b1\u03b2\u03b3\u0061\u006d\u0062"
L"\u0075\u006c\u014d\u043f\u0440\u0438\u0432\u0435"
L"\u0442"));
// Search for non-existant env vars.
EXPECT_STREQ(nullptr, adb_getenv("SYSDEPS_WIN32_TEST_NONEXISTANT"));
// Search for existing env vars.
// There is no test for an env var with a value of a zero-length string
// because _putenv() does not support inserting such an env var.
// Search for env var that is uppercase.
EXPECT_STREQ("1", adb_getenv("SYSDEPS_WIN32_TEST_UPPERCASE"));
EXPECT_STREQ("1", adb_getenv("sysdeps_win32_test_uppercase"));
EXPECT_STREQ("1", adb_getenv("Sysdeps_Win32_Test_Uppercase"));
// Search for env var that is lowercase.
EXPECT_STREQ("2", adb_getenv("SYSDEPS_WIN32_TEST_LOWERCASE"));
EXPECT_STREQ("2", adb_getenv("sysdeps_win32_test_lowercase"));
EXPECT_STREQ("2", adb_getenv("Sysdeps_Win32_Test_Lowercase"));
// Search for env var that is mixed-case.
EXPECT_STREQ("3", adb_getenv("SYSDEPS_WIN32_TEST_MIXEDCASE"));
EXPECT_STREQ("3", adb_getenv("sysdeps_win32_test_mixedcase"));
EXPECT_STREQ("3", adb_getenv("Sysdeps_Win32_Test_MixedCase"));
// Check that UTF-16 was converted to UTF-8.
EXPECT_STREQ("\xc2\xa1\x48\x6f\x6c\x61\x21\xce\xb1\xce\xb2\xce\xb3\x61\x6d"
"\x62\x75\x6c\xc5\x8d\xd0\xbf\xd1\x80\xd0\xb8\xd0\xb2\xd0\xb5"
"\xd1\x82",
adb_getenv("SYSDEPS_WIN32_TEST_UNICODE"));
// Check an env var that should always be set.
const char* path_val = adb_getenv("PATH");
EXPECT_NE(nullptr, path_val);
if (path_val != nullptr) {
EXPECT_GT(strlen(path_val), 0U);
}
}
void TestAdbStrError(int err, const char* expected) {
errno = 12345;
const char* result = adb_strerror(err);
// Check that errno is not overwritten.
EXPECT_EQ(12345, errno);
EXPECT_STREQ(expected, result);
}
TEST(sysdeps_win32, adb_strerror) {
// Test an error code that should not have a mapped string. Use an error
// code that is not used by the internal implementation of adb_strerror().
TestAdbStrError(-2, "Unknown error");
// adb_strerror() uses -1 internally, so test that it can still be passed
// as a parameter.
TestAdbStrError(-1, "Unknown error");
// Test very big, positive unknown error.
TestAdbStrError(1000000, "Unknown error");
// Test success case.
// Wine returns "Success" for strerror(0), Windows returns "No error", so accept both.
std::string success = adb_strerror(0);
EXPECT_TRUE(success == "Success" || success == "No error") << "strerror(0) = " << success;
// Test error that regular strerror() should have a string for.
TestAdbStrError(EPERM, "Operation not permitted");
// Test error that regular strerror() doesn't have a string for, but that
// adb_strerror() returns.
TestAdbStrError(ECONNRESET, "Connection reset by peer");
}
TEST(sysdeps_win32, unix_isatty) {
// stdin and stdout should be consoles. Use CONIN$ and CONOUT$ special files
// so that we can test this even if stdin/stdout have been redirected. Read
// permissions are required for unix_isatty().
int conin_fd = unix_open("CONIN$", O_RDONLY);
int conout_fd = unix_open("CONOUT$", O_RDWR);
for (const int fd : {conin_fd, conout_fd}) {
EXPECT_TRUE(fd >= 0);
EXPECT_EQ(1, unix_isatty(fd));
EXPECT_EQ(0, unix_close(fd));
}
// nul returns 1 from isatty(), make sure unix_isatty() corrects that.
for (auto flags : {O_RDONLY, O_RDWR}) {
int nul_fd = unix_open("nul", flags);
EXPECT_TRUE(nul_fd >= 0);
EXPECT_EQ(0, unix_isatty(nul_fd));
EXPECT_EQ(0, unix_close(nul_fd));
}
// Check a real file, both read-write and read-only.
TemporaryFile temp_file;
EXPECT_TRUE(temp_file.fd >= 0);
EXPECT_EQ(0, unix_isatty(temp_file.fd));
int temp_file_ro_fd = unix_open(temp_file.path, O_RDONLY);
EXPECT_TRUE(temp_file_ro_fd >= 0);
EXPECT_EQ(0, unix_isatty(temp_file_ro_fd));
EXPECT_EQ(0, unix_close(temp_file_ro_fd));
// Check a real OS pipe.
int pipe_fds[2];
EXPECT_EQ(0, _pipe(pipe_fds, 64, _O_BINARY));
EXPECT_EQ(0, unix_isatty(pipe_fds[0]));
EXPECT_EQ(0, unix_isatty(pipe_fds[1]));
EXPECT_EQ(0, _close(pipe_fds[0]));
EXPECT_EQ(0, _close(pipe_fds[1]));
// Make sure an invalid FD is handled correctly.
EXPECT_EQ(0, unix_isatty(-1));
}
void TestParseCompleteUTF8(const char* buf, const size_t buf_size,
const size_t expected_complete_bytes,
const std::vector<char>& expected_remaining_bytes) {
std::vector<char> remaining_bytes;
const size_t complete_bytes = internal::ParseCompleteUTF8(buf, buf + buf_size,
&remaining_bytes);
EXPECT_EQ(expected_complete_bytes, complete_bytes);
EXPECT_EQ(expected_remaining_bytes, remaining_bytes);
}
TEST(sysdeps_win32, ParseCompleteUTF8) {
const std::vector<std::vector<char>> multi_byte_sequences = {
{ '\xc2', '\xa9' }, // 2 byte UTF-8 sequence
{ '\xe1', '\xb4', '\xa8' }, // 3 byte UTF-8 sequence
{ '\xf0', '\x9f', '\x98', '\x80' }, // 4 byte UTF-8 sequence
};
std::vector<std::vector<char>> all_sequences = {
{}, // 0 bytes
{ '\0' }, // NULL byte
{ 'a' }, // 1 byte UTF-8 sequence
};
all_sequences.insert(all_sequences.end(), multi_byte_sequences.begin(),
multi_byte_sequences.end());
// Vary a prefix of bytes in front of the sequence that we're actually interested in parsing.
for (const auto& prefix : all_sequences) {
// Parse (prefix + one byte of the sequence at a time)
for (const auto& seq : multi_byte_sequences) {
std::vector<char> buffer(prefix);
// For every byte of the sequence except the last
for (size_t i = 0; i < seq.size() - 1; ++i) {
buffer.push_back(seq[i]);
// When parsing an incomplete UTF-8 sequence, the amount of the buffer preceding
// the start of the incomplete UTF-8 sequence is valid. The remaining bytes are the
// bytes of the incomplete UTF-8 sequence.
TestParseCompleteUTF8(buffer.data(), buffer.size(), prefix.size(),
std::vector<char>(seq.begin(), seq.begin() + i + 1));
}
// For the last byte of the sequence
buffer.push_back(seq.back());
TestParseCompleteUTF8(buffer.data(), buffer.size(), buffer.size(), std::vector<char>());
}
// Parse (prefix (aka sequence) + invalid trailing bytes) to verify that the invalid
// trailing bytes are immediately "returned" to prevent them from being stuck in some
// buffer.
std::vector<char> buffer(prefix);
for (size_t i = 0; i < 8; ++i) {
buffer.push_back(0x80); // trailing byte
TestParseCompleteUTF8(buffer.data(), buffer.size(), buffer.size(), std::vector<char>());
}
}
}