adb/sysdeps_win32_test.cpp - android-core - Git at Google

 /*
  * 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>());
         }
     }
 }
	/*
	* 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>());
	}
	}
	}