mirror of
https://github.com/trustedsec/COFFLoader
synced 2026-06-08 17:55:49 +00:00
Matt Ehrnschwender
8eca447fea
Change '__restrict__' modifier in `printf` to use the C99 keyword or the MSVC/GNU modifier if compiling with a C++ compiler
53 lines
1.5 KiB
C
53 lines
1.5 KiB
C
#include <windows.h>
|
|
#include <stdio.h>
|
|
#include <lm.h>
|
|
#include <dsgetdc.h>
|
|
#include "beacon.h"
|
|
|
|
#ifdef __cplusplus
|
|
#if defined(_MSC_VER) || defined(__GNUC__)
|
|
#define restrict __restrict
|
|
#else
|
|
#define restrict
|
|
#endif // defined(_MSC_VER) || defined(__GNUC__)
|
|
#endif // __cplusplus
|
|
|
|
DECLSPEC_IMPORT DWORD WINAPI NETAPI32$DsGetDcNameA(LPVOID, LPVOID, LPVOID, LPVOID, ULONG, LPVOID);
|
|
DECLSPEC_IMPORT DWORD WINAPI NETAPI32$NetApiBufferFree(LPVOID);
|
|
WINBASEAPI int __cdecl MSVCRT$printf(const char *restrict _Format,...);
|
|
|
|
char* TestGlobalString = "This is a global string";
|
|
/* Can't do stuff like "int testvalue;" in a coff file, because it assumes that
|
|
* the symbol is like any function, so you would need to allocate a section of bss
|
|
* (without knowing the size of it), and then resolve the symbol to that. So safer
|
|
* to just not support that */
|
|
int testvalue = 0;
|
|
|
|
int test(void){
|
|
MSVCRT$printf("Test String from test\n");
|
|
testvalue = 1;
|
|
return 0;
|
|
}
|
|
|
|
int test2(void){
|
|
MSVCRT$printf("Test String from test2\n");
|
|
return 0;
|
|
}
|
|
|
|
|
|
void go(char * args, unsigned long alen) {
|
|
DWORD dwRet;
|
|
PDOMAIN_CONTROLLER_INFO pdcInfo;
|
|
BeaconPrintf(1, "This GlobalString \"%s\"\n", TestGlobalString);
|
|
MSVCRT$printf("Test Value: %d\n", testvalue);
|
|
(void)test();
|
|
MSVCRT$printf("Test ValueBack: %d\n", testvalue);
|
|
(void)test2();
|
|
dwRet = NETAPI32$DsGetDcNameA(NULL, NULL, NULL, NULL, 0, &pdcInfo);
|
|
if (ERROR_SUCCESS == dwRet) {
|
|
MSVCRT$printf("%s", pdcInfo->DomainName);
|
|
}
|
|
|
|
NETAPI32$NetApiBufferFree(pdcInfo);
|
|
}
|