mingw-w64-crt/ssp/stack_chk_guard.c

/**
 * This file has no copyright assigned and is placed in the Public Domain.
 * This file is part of the mingw-w64 runtime package.
 * No warranty is given; refer to the file DISCLAIMER.PD within this package.
 */

#define _CRT_RAND_S
#include <stdlib.h>
#include <stdint.h>

uintptr_t __stack_chk_guard = 0;

#if defined(__GNUC__) && __GNUC__ >= 9 && !defined(__clang__)
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wprio-ctor-dtor"
#endif

#if defined __SSP__ || defined __SSP_STRONG__ || defined __SSP_ALL__
// This function requires `no_stack_protector` because it changes the
// value of `__stack_chk_guard`, causing stack checks to fail before
// returning from this function.
# if (defined __GNUC__ && __GNUC__ >= 11) || (defined __clang__ && __clang_major__ >= 7)
__attribute__((__no_stack_protector__))
# else
#  error This file can not be built with stack protector enabled. Remove \
         -fstack-protector* options from CFLAGS.
# endif
#endif
__attribute__((constructor(0)))
static void __cdecl init(void)
{
  unsigned int ui;
  if (__stack_chk_guard != 0)
    return;

  // Call rand_s from the CRT; this uses a high quality random source
  // RtlGenRandom internally. This function is available since XP, and is
  // callable in WinStore mode too (since it's from the CRT).
  // In the case of msvcrt.dll, our import library provides a small wrapper
  // which tries to load the function dynamically, and falls back on
  // using RtlGenRandom if not available.
  if (rand_s(&ui) == 0) {
    __stack_chk_guard = ui;
#if __SIZEOF_POINTER__ > 4
    rand_s(&ui);
    __stack_chk_guard = __stack_chk_guard << 32 | ui;
#endif
    return;
  }

  // If rand_s failed (it shouldn't), hardcode a nonzero default stack guard.
#if __SIZEOF_POINTER__ > 4
  __stack_chk_guard = 0xdeadbeefdeadbeefULL;
#else
  __stack_chk_guard = 0xdeadbeef;
#endif
}

#if defined(__GNUC__) && __GNUC__ >= 9 && !defined(__clang__)
#pragma GCC diagnostic pop
#endif