aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/src/wincompat/init.c
blob: 8d862ff6d7dde51652d675c8b2169cc786daf0a6 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (C) 2015-2019 Jason A. Donenfeld <Jason@zx2c4.com>. All Rights Reserved.
 */

#include <winsock2.h>
#include <windows.h>

#ifndef ENABLE_VIRTUAL_TERMINAL_PROCESSING
#define ENABLE_VIRTUAL_TERMINAL_PROCESSING 0x4
#endif

__attribute__((constructor)) static void init(void)
{
	char *colormode;
	DWORD console_mode;
	HANDLE stdout_handle;
	WSADATA wsaData;
	WSAStartup(MAKEWORD(2, 2), &wsaData);

	stdout_handle = GetStdHandle(STD_OUTPUT_HANDLE); // We don't close this.
	if (stdout_handle == INVALID_HANDLE_VALUE)
		goto no_color;
	if (!GetConsoleMode(stdout_handle, &console_mode))
		goto no_color;
	if (!SetConsoleMode(stdout_handle, ENABLE_VIRTUAL_TERMINAL_PROCESSING | console_mode))
		goto no_color;
	return;

no_color:
	colormode = getenv("WG_COLOR_MODE");
	if (!colormode)
		putenv("WG_COLOR_MODE=never");
}

__attribute__((destructor)) static void deinit(void)
{
	WSACleanup();
}