my-fridajs-example/console.js at dev · nblog/my-fridajs-example · GitHub

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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
///<reference path='C:/Users/r0th3r/OneDrive/Code/index.d.ts'/>


/* https://learn.microsoft.com/windows/win32/api/fileapi/nf-fileapi-closehandle */
const CloseHandle = new NativeFunction(
    Module.getExportByName('kernel32.dll', 'CloseHandle'),
    'bool', ['pointer']);

/* https://learn.microsoft.com/windows/win32/api/fileapi/nf-fileapi-readfile */
const ReadFile = new NativeFunction(
    Module.getExportByName('kernel32.dll', 'ReadFile'),
    'bool', ['pointer', 'pointer', 'uint32', 'pointer', 'pointer']);

/* https://learn.microsoft.com/windows/win32/api/fileapi/nf-fileapi-writefile */
const WriteFile = new NativeFunction(
    Module.getExportByName('kernel32.dll', 'WriteFile'),
    'bool', ['pointer', 'pointer', 'uint32', 'pointer', 'pointer']);

/* https://learn.microsoft.com/windows/win32/api/fileapi/nf-fileapi-createfile2 */
const CreateFile2 = new NativeFunction(
    Module.getExportByName('kernel32.dll', 'CreateFile2'),
    'pointer', ['pointer', 'uint32', 'uint32', 'uint32', 'pointer']);

/* https://learn.microsoft.com/windows/win32/api/fileapi/nf-fileapi-createfilea */
const CreateFileA = new NativeFunction(
    Module.getExportByName('kernel32.dll', 'CreateFileA'),
    'pointer', ['pointer', 'uint32', 'uint32', 'pointer', 'uint32', 'uint32', 'pointer']);

/* https://learn.microsoft.com/windows/win32/api/fileapi/nf-fileapi-createfilew */
const CreateFileW = new NativeFunction(
    Module.getExportByName('kernel32.dll', 'CreateFileW'),
    'pointer', ['pointer', 'uint32', 'uint32', 'pointer', 'uint32', 'uint32', 'pointer']);

const CreateFile = CreateFileW;


/*
const stdin = new NativeFunction(
    Module.getExportByName('msvcrt.dll', '__iob_func'),
    'pointer', ['int'])(0);
const stdout = new NativeFunction(
    Module.getExportByName('msvcrt.dll', '__iob_func'),
    'pointer', ['int'])(1);
const stderr = new NativeFunction(
    Module.getExportByName('msvcrt.dll', '__iob_func'),
    'pointer', ['int'])(2);
*/
const stdin = new NativeFunction(
    Module.getExportByName('ucrtbase.dll', '__acrt_iob_func'),
    'pointer', ['int'])(0);
const stdout = new NativeFunction(
    Module.getExportByName('ucrtbase.dll', '__acrt_iob_func'),
    'pointer', ['int'])(1);
const stderr = new NativeFunction(
    Module.getExportByName('ucrtbase.dll', '__acrt_iob_func'),
    'pointer', ['int'])(2);

/*
const freopen = new NativeFunction(
    Module.getExportByName('msvcrt.dll', 'freopen'),
    'pointer', ['pointer', 'pointer', 'pointer']);
*/
const freopen = new NativeFunction(
    Module.getExportByName('ucrtbase.dll', 'freopen'),
    'pointer', ['pointer', 'pointer', 'pointer']);


/* https://learn.microsoft.com/windows/win32/inputdev/virtual-key-codes */
function KeyState(vkcode, ctrl=false, alt=false, shift=false) {
    const GetAsyncKeyState = new NativeFunction(
        Module.getExportByName('user32.dll', 'GetAsyncKeyState'),
        'int', ['int']);

    const CHECK = 0x8000;

    let vkcodePressed = (GetAsyncKeyState(vkcode) & CHECK) != 0;

    let ctrlPressed = (GetAsyncKeyState(0x11/*VK_CONTROL*/) & CHECK) != 0;
    let altPressed = (GetAsyncKeyState(0x12/*VK_MENU*/) & CHECK) != 0;
    let shiftPressed = (GetAsyncKeyState(0x10/*VK_SHIFT*/) & CHECK) != 0;

    return vkcodePressed &&
    (ctrl == ctrlPressed) &&
    (alt == altPressed) &&
    (shift == shiftPressed);
}

const timerId = setInterval(() => {
    if (KeyState(0xC0/*VK_OEM_3*/, true))
    {
        /*
        const ATTACH_PARENT_PROCESS = -1;
        new NativeFunction(
            Module.getExportByName('kernel32.dll', 'AttachConsole'),
            'bool', ['uint32'])(ATTACH_PARENT_PROCESS);
        */
        let isOk = new NativeFunction(
            Module.getExportByName('kernel32.dll', 'AllocConsole'),
            'bool', [])();

        if (isOk) {
            freopen(Memory.allocUtf8String('CONOUT$'), Memory.allocUtf8String('w'), stdout);
            /* freopen(Memory.allocUtf8String('CONERR$'), Memory.allocUtf8String('w'), stderr); */
            /* freopen(Memory.allocUtf8String('CONIN$'), Memory.allocUtf8String('r'), stdin); */

            /*
            const CP_UTF8 = 65001;
            new NativeFunction(
                Module.getExportByName('kernel32.dll', 'SetConsoleCP'),
                'bool', ['uint32'])(CP_UTF8);
            new NativeFunction(
                Module.getExportByName('kernel32.dll', 'SetConsoleOutputCP'),
                'bool', ['uint32'])(CP_UTF8);

            new NativeFunction(
                Module.getExportByName('kernel32.dll', 'SetConsoleCtrlHandler'),
                'bool', ['pointer', 'bool'])(new NativeCallback((dwCtrlType) => {
                    console.log('Ctrl+C');
                    return false;
                }, 'bool', ['uint32']), true);
            */

            new NativeFunction(
                Module.getExportByName('kernel32.dll', 'SetConsoleTitleA'),
                'bool', ['pointer'])(Memory.allocUtf8String('Debug Console'));
        }
    }

    if (KeyState(0x1B/*VK_ESCAPE*/, true))
    {
        clearInterval(timerId);
    }

    if (KeyState(0x70/*VK_F1*/, true))
    {
        console.log('Ctrl F1');
    }
    if (KeyState(0x71/*VK_F2*/, true))
    {
        console.log('Ctrl F2');
    }
}, 0.3 * 1000);