mirror of
https://github.com/nodejs/node.git
synced 2024-11-22 11:12:18 +00:00
fcf27b12e3
GetPageSize() and OverrunGuardedBuffer currently use non-fatal EXPECT_* macros because GoogleTest does not allow the fatal variants ASSERT_* in non-void returning functions (i.e., in this file, nowhere outside of the TEST itself). The EXPECT_* macros continue execution upon failure, but we really don't want that (and static analysis apparently does not like it either). Since we cannot use GoogleTest's ASSERT_* here, use our own CHECK_* instead of EXPECT_* outside of the TEST. Hopefully, this will finally pacify static analysis. Refs: https://github.com/nodejs/node/pull/44666 PR-URL: https://github.com/nodejs/node/pull/44795 Reviewed-By: Darshan Sen <raisinten@gmail.com> Reviewed-By: James M Snell <jasnell@gmail.com> Reviewed-By: Erick Wendel <erick.workspace@gmail.com> Reviewed-By: Michael Dawson <midawson@redhat.com>
134 lines
4.2 KiB
C++
134 lines
4.2 KiB
C++
#include "crypto/crypto_clienthello-inl.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
// If the test is being compiled with an address sanitizer enabled, it should
|
|
// catch the memory violation, so do not use a guard page.
|
|
#ifdef __SANITIZE_ADDRESS__
|
|
#define NO_GUARD_PAGE
|
|
#elif defined(__has_feature)
|
|
#if __has_feature(address_sanitizer)
|
|
#define NO_GUARD_PAGE
|
|
#endif
|
|
#endif
|
|
|
|
// If the test is running without an address sanitizer, see if we can use
|
|
// mprotect() or VirtualProtect() to cause a segmentation fault when spatial
|
|
// safety is violated.
|
|
#if !defined(NO_GUARD_PAGE)
|
|
#ifdef __linux__
|
|
#include <sys/mman.h>
|
|
#include <unistd.h>
|
|
#if defined(_SC_PAGE_SIZE) && defined(PROT_NONE) && defined(PROT_READ) && \
|
|
defined(PROT_WRITE)
|
|
#define USE_MPROTECT
|
|
#endif
|
|
#elif defined(_WIN32) && defined(_MSC_VER)
|
|
#include <Windows.h>
|
|
#include <memoryapi.h>
|
|
#define USE_VIRTUALPROTECT
|
|
#endif
|
|
#endif
|
|
|
|
#if defined(USE_MPROTECT)
|
|
size_t GetPageSize() {
|
|
int page_size = sysconf(_SC_PAGE_SIZE);
|
|
CHECK_GE(page_size, 1);
|
|
return page_size;
|
|
}
|
|
#elif defined(USE_VIRTUALPROTECT)
|
|
size_t GetPageSize() {
|
|
SYSTEM_INFO system_info;
|
|
GetSystemInfo(&system_info);
|
|
return system_info.dwPageSize;
|
|
}
|
|
#endif
|
|
|
|
template <size_t N>
|
|
class OverrunGuardedBuffer {
|
|
public:
|
|
OverrunGuardedBuffer() {
|
|
#if defined(USE_MPROTECT) || defined(USE_VIRTUALPROTECT)
|
|
size_t page = GetPageSize();
|
|
CHECK_GE(page, N);
|
|
#endif
|
|
#ifdef USE_MPROTECT
|
|
// Place the packet right before a guard page, which, when accessed, causes
|
|
// a segmentation fault.
|
|
alloc_base = static_cast<uint8_t*>(aligned_alloc(page, 2 * page));
|
|
CHECK_NOT_NULL(alloc_base);
|
|
uint8_t* second_page = alloc_base + page;
|
|
CHECK_EQ(mprotect(second_page, page, PROT_NONE), 0);
|
|
data_base = second_page - N;
|
|
#elif defined(USE_VIRTUALPROTECT)
|
|
// On Windows, it works almost the same way.
|
|
alloc_base = static_cast<uint8_t*>(
|
|
VirtualAlloc(nullptr, 2 * page, MEM_COMMIT, PAGE_READWRITE));
|
|
CHECK_NOT_NULL(alloc_base);
|
|
uint8_t* second_page = alloc_base + page;
|
|
DWORD old_prot;
|
|
CHECK_NE(VirtualProtect(second_page, page, PAGE_NOACCESS, &old_prot), 0);
|
|
CHECK_EQ(old_prot, PAGE_READWRITE);
|
|
data_base = second_page - N;
|
|
#else
|
|
// Place the packet in a regular allocated buffer. The bug causes undefined
|
|
// behavior, which might crash the process, and when it does not, address
|
|
// sanitizers and valgrind will catch it.
|
|
alloc_base = static_cast<uint8_t*>(malloc(N));
|
|
CHECK_NOT_NULL(alloc_base);
|
|
data_base = alloc_base;
|
|
#endif
|
|
}
|
|
|
|
OverrunGuardedBuffer(const OverrunGuardedBuffer& other) = delete;
|
|
OverrunGuardedBuffer& operator=(const OverrunGuardedBuffer& other) = delete;
|
|
|
|
~OverrunGuardedBuffer() {
|
|
#if defined(USE_MPROTECT) || defined(USE_VIRTUALPROTECT)
|
|
size_t page = GetPageSize();
|
|
#endif
|
|
#ifdef USE_VIRTUALPROTECT
|
|
VirtualFree(alloc_base, 2 * page, MEM_RELEASE);
|
|
#else
|
|
#ifdef USE_MPROTECT
|
|
// Revert page protection such that the memory can be free()'d.
|
|
uint8_t* second_page = alloc_base + page;
|
|
CHECK_EQ(mprotect(second_page, page, PROT_READ | PROT_WRITE), 0);
|
|
#endif
|
|
free(alloc_base);
|
|
#endif
|
|
}
|
|
|
|
uint8_t* data() {
|
|
return data_base;
|
|
}
|
|
|
|
private:
|
|
uint8_t* alloc_base;
|
|
uint8_t* data_base;
|
|
};
|
|
|
|
// Test that ClientHelloParser::ParseHeader() does not blindly trust the client
|
|
// to send a valid frame length and subsequently does not read out-of-bounds.
|
|
TEST(NodeCrypto, ClientHelloParserParseHeaderOutOfBoundsRead) {
|
|
using node::crypto::ClientHelloParser;
|
|
|
|
// This is the simplest packet triggering the bug.
|
|
const uint8_t packet[] = {0x16, 0x03, 0x01, 0x00, 0x00};
|
|
OverrunGuardedBuffer<sizeof(packet)> buffer;
|
|
memcpy(buffer.data(), packet, sizeof(packet));
|
|
|
|
// Let the ClientHelloParser parse the packet. This should not lead to a
|
|
// segmentation fault or to undefined behavior.
|
|
node::crypto::ClientHelloParser parser;
|
|
bool end_cb_called = false;
|
|
parser.Start([](void* arg, auto hello) { GTEST_FAIL(); },
|
|
[](void* arg) {
|
|
bool* end_cb_called = static_cast<bool*>(arg);
|
|
EXPECT_FALSE(*end_cb_called);
|
|
*end_cb_called = true;
|
|
},
|
|
&end_cb_called);
|
|
parser.Parse(buffer.data(), sizeof(packet));
|
|
EXPECT_TRUE(end_cb_called);
|
|
}
|