Text file src/runtime/testdata/testprogcgo/bigstack_windows.c

     1  // Copyright 2018 The Go Authors. All rights reserved.
     2  // Use of this source code is governed by a BSD-style
     3  // license that can be found in the LICENSE file.
     4  
     5  // This test source is used by both TestBigStackCallbackCgo (linked
     6  // directly into the Go binary) and TestBigStackCallbackSyscall
     7  // (compiled into a DLL).
     8  
     9  #include <windows.h>
    10  #include <stdio.h>
    11  
    12  #ifndef STACK_SIZE_PARAM_IS_A_RESERVATION
    13  #define STACK_SIZE_PARAM_IS_A_RESERVATION 0x00010000
    14  #endif
    15  
    16  typedef void callback(char*);
    17  
    18  // Allocate a stack that's much larger than the default.
    19  static const int STACK_SIZE = 16<<20;
    20  
    21  static callback *bigStackCallback;
    22  
    23  static void useStack(int bytes) {
    24  	// Windows doesn't like huge frames, so we grow the stack 64k at a time.
    25  	char x[64<<10];
    26  	if (bytes < sizeof x) {
    27  		bigStackCallback(x);
    28  	} else {
    29  		useStack(bytes - sizeof x);
    30  	}
    31  }
    32  
    33  static DWORD WINAPI threadEntry(LPVOID lpParam) {
    34  	useStack(STACK_SIZE - (128<<10));
    35  	return 0;
    36  }
    37  
    38  void bigStack(callback *cb) {
    39  	bigStackCallback = cb;
    40  	HANDLE hThread = CreateThread(NULL, STACK_SIZE, threadEntry, NULL, STACK_SIZE_PARAM_IS_A_RESERVATION, NULL);
    41  	if (hThread == NULL) {
    42  		fprintf(stderr, "CreateThread failed\n");
    43  		exit(1);
    44  	}
    45  	WaitForSingleObject(hThread, INFINITE);
    46  }
    47  

View as plain text