├── README.md
├── AESCrypt
├── AESCrypt.vcxproj.filters
├── AESCrypt.vcxproj
└── AESCrypt.cpp
├── AESCrypt.sln
└── .gitignore
/README.md:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/hackerhouse-opensource/AESCrypt/HEAD/README.md
--------------------------------------------------------------------------------
/AESCrypt/AESCrypt.vcxproj.filters:
--------------------------------------------------------------------------------
1 |
2 |
3 |
4 |
5 | {4FC737F1-C7A5-4376-A066-2A32D752A2FF}
6 | cpp;c;cc;cxx;c++;cppm;ixx;def;odl;idl;hpj;bat;asm;asmx
7 |
8 |
9 | {93995380-89BD-4b04-88EB-625FBE52EBFB}
10 | h;hh;hpp;hxx;h++;hm;inl;inc;ipp;xsd
11 |
12 |
13 | {67DA6AB6-F800-4c08-8B7A-83BB121AAD01}
14 | rc;ico;cur;bmp;dlg;rc2;rct;bin;rgs;gif;jpg;jpeg;jpe;resx;tiff;tif;png;wav;mfcribbon-ms
15 |
16 |
17 |
18 |
19 | Source Files
20 |
21 |
22 |
23 |
24 |
25 |
--------------------------------------------------------------------------------
/AESCrypt.sln:
--------------------------------------------------------------------------------
1 |
2 | Microsoft Visual Studio Solution File, Format Version 12.00
3 | # Visual Studio Version 17
4 | VisualStudioVersion = 17.7.34202.233
5 | MinimumVisualStudioVersion = 10.0.40219.1
6 | Project("{8BC9CEB8-8B4A-11D0-8D11-00A0C91BC942}") = "AESCrypt", "AESCrypt\AESCrypt.vcxproj", "{CB18BD1F-815F-4B97-B5FF-213E337D41C0}"
7 | EndProject
8 | Global
9 | GlobalSection(SolutionConfigurationPlatforms) = preSolution
10 | Debug|x64 = Debug|x64
11 | Debug|x86 = Debug|x86
12 | Release|x64 = Release|x64
13 | Release|x86 = Release|x86
14 | EndGlobalSection
15 | GlobalSection(ProjectConfigurationPlatforms) = postSolution
16 | {CB18BD1F-815F-4B97-B5FF-213E337D41C0}.Debug|x64.ActiveCfg = Debug|x64
17 | {CB18BD1F-815F-4B97-B5FF-213E337D41C0}.Debug|x64.Build.0 = Debug|x64
18 | {CB18BD1F-815F-4B97-B5FF-213E337D41C0}.Debug|x86.ActiveCfg = Debug|Win32
19 | {CB18BD1F-815F-4B97-B5FF-213E337D41C0}.Debug|x86.Build.0 = Debug|Win32
20 | {CB18BD1F-815F-4B97-B5FF-213E337D41C0}.Release|x64.ActiveCfg = Release|x64
21 | {CB18BD1F-815F-4B97-B5FF-213E337D41C0}.Release|x64.Build.0 = Release|x64
22 | {CB18BD1F-815F-4B97-B5FF-213E337D41C0}.Release|x86.ActiveCfg = Release|Win32
23 | {CB18BD1F-815F-4B97-B5FF-213E337D41C0}.Release|x86.Build.0 = Release|Win32
24 | EndGlobalSection
25 | GlobalSection(SolutionProperties) = preSolution
26 | HideSolutionNode = FALSE
27 | EndGlobalSection
28 | GlobalSection(ExtensibilityGlobals) = postSolution
29 | SolutionGuid = {C6162175-8FAD-40D7-9EA0-D9405DDBD817}
30 | EndGlobalSection
31 | EndGlobal
32 |
--------------------------------------------------------------------------------
/AESCrypt/AESCrypt.vcxproj:
--------------------------------------------------------------------------------
1 |
2 |
3 |
4 |
5 | Debug
6 | Win32
7 |
8 |
9 | Release
10 | Win32
11 |
12 |
13 | Debug
14 | x64
15 |
16 |
17 | Release
18 | x64
19 |
20 |
21 |
22 | 17.0
23 | Win32Proj
24 | {cb18bd1f-815f-4b97-b5ff-213e337d41c0}
25 | AESCrypt
26 | 10.0
27 |
28 |
29 |
30 | Application
31 | true
32 | v143
33 | Unicode
34 |
35 |
36 | Application
37 | false
38 | v143
39 | true
40 | Unicode
41 |
42 |
43 | Application
44 | true
45 | v143
46 | Unicode
47 |
48 |
49 | Application
50 | false
51 | v143
52 | true
53 | Unicode
54 | false
55 |
56 |
57 |
58 |
59 |
60 |
61 |
62 |
63 |
64 |
65 |
66 |
67 |
68 |
69 |
70 |
71 |
72 |
73 |
74 |
75 |
76 | Level3
77 | true
78 | WIN32;_DEBUG;_CONSOLE;%(PreprocessorDefinitions)
79 | true
80 |
81 |
82 | Console
83 | true
84 |
85 |
86 |
87 |
88 | Level3
89 | true
90 | true
91 | true
92 | WIN32;NDEBUG;_CONSOLE;%(PreprocessorDefinitions)
93 | true
94 |
95 |
96 | Console
97 | true
98 | true
99 | true
100 |
101 |
102 |
103 |
104 | Level3
105 | true
106 | _DEBUG;_CONSOLE;%(PreprocessorDefinitions)
107 | true
108 |
109 |
110 | Console
111 | true
112 |
113 |
114 |
115 |
116 | Level3
117 | true
118 | true
119 | true
120 | NDEBUG;_CONSOLE;%(PreprocessorDefinitions)
121 | true
122 |
123 |
124 | Console
125 | true
126 | true
127 | true
128 | Bcrypt.lib
129 |
130 |
131 |
132 |
133 |
134 |
135 |
136 |
137 |
138 |
139 |
140 |
--------------------------------------------------------------------------------
/.gitignore:
--------------------------------------------------------------------------------
1 | ## Ignore Visual Studio temporary files, build results, and
2 | ## files generated by popular Visual Studio add-ons.
3 | ##
4 | ## Get latest from https://github.com/github/gitignore/blob/main/VisualStudio.gitignore
5 |
6 | # User-specific files
7 | *.rsuser
8 | *.suo
9 | *.user
10 | *.userosscache
11 | *.sln.docstates
12 |
13 | # User-specific files (MonoDevelop/Xamarin Studio)
14 | *.userprefs
15 |
16 | # Mono auto generated files
17 | mono_crash.*
18 |
19 | # Build results
20 | [Dd]ebug/
21 | [Dd]ebugPublic/
22 | [Rr]elease/
23 | [Rr]eleases/
24 | x64/
25 | x86/
26 | [Ww][Ii][Nn]32/
27 | [Aa][Rr][Mm]/
28 | [Aa][Rr][Mm]64/
29 | bld/
30 | [Bb]in/
31 | [Oo]bj/
32 | [Ll]og/
33 | [Ll]ogs/
34 |
35 | # Visual Studio 2015/2017 cache/options directory
36 | .vs/
37 | # Uncomment if you have tasks that create the project's static files in wwwroot
38 | #wwwroot/
39 |
40 | # Visual Studio 2017 auto generated files
41 | Generated\ Files/
42 |
43 | # MSTest test Results
44 | [Tt]est[Rr]esult*/
45 | [Bb]uild[Ll]og.*
46 |
47 | # NUnit
48 | *.VisualState.xml
49 | TestResult.xml
50 | nunit-*.xml
51 |
52 | # Build Results of an ATL Project
53 | [Dd]ebugPS/
54 | [Rr]eleasePS/
55 | dlldata.c
56 |
57 | # Benchmark Results
58 | BenchmarkDotNet.Artifacts/
59 |
60 | # .NET Core
61 | project.lock.json
62 | project.fragment.lock.json
63 | artifacts/
64 |
65 | # ASP.NET Scaffolding
66 | ScaffoldingReadMe.txt
67 |
68 | # StyleCop
69 | StyleCopReport.xml
70 |
71 | # Files built by Visual Studio
72 | *_i.c
73 | *_p.c
74 | *_h.h
75 | *.ilk
76 | *.meta
77 | *.obj
78 | *.iobj
79 | *.pch
80 | *.pdb
81 | *.ipdb
82 | *.pgc
83 | *.pgd
84 | *.rsp
85 | *.sbr
86 | *.tlb
87 | *.tli
88 | *.tlh
89 | *.tmp
90 | *.tmp_proj
91 | *_wpftmp.csproj
92 | *.log
93 | *.tlog
94 | *.vspscc
95 | *.vssscc
96 | .builds
97 | *.pidb
98 | *.svclog
99 | *.scc
100 |
101 | # Chutzpah Test files
102 | _Chutzpah*
103 |
104 | # Visual C++ cache files
105 | ipch/
106 | *.aps
107 | *.ncb
108 | *.opendb
109 | *.opensdf
110 | *.sdf
111 | *.cachefile
112 | *.VC.db
113 | *.VC.VC.opendb
114 |
115 | # Visual Studio profiler
116 | *.psess
117 | *.vsp
118 | *.vspx
119 | *.sap
120 |
121 | # Visual Studio Trace Files
122 | *.e2e
123 |
124 | # TFS 2012 Local Workspace
125 | $tf/
126 |
127 | # Guidance Automation Toolkit
128 | *.gpState
129 |
130 | # ReSharper is a .NET coding add-in
131 | _ReSharper*/
132 | *.[Rr]e[Ss]harper
133 | *.DotSettings.user
134 |
135 | # TeamCity is a build add-in
136 | _TeamCity*
137 |
138 | # DotCover is a Code Coverage Tool
139 | *.dotCover
140 |
141 | # AxoCover is a Code Coverage Tool
142 | .axoCover/*
143 | !.axoCover/settings.json
144 |
145 | # Coverlet is a free, cross platform Code Coverage Tool
146 | coverage*.json
147 | coverage*.xml
148 | coverage*.info
149 |
150 | # Visual Studio code coverage results
151 | *.coverage
152 | *.coveragexml
153 |
154 | # NCrunch
155 | _NCrunch_*
156 | .*crunch*.local.xml
157 | nCrunchTemp_*
158 |
159 | # MightyMoose
160 | *.mm.*
161 | AutoTest.Net/
162 |
163 | # Web workbench (sass)
164 | .sass-cache/
165 |
166 | # Installshield output folder
167 | [Ee]xpress/
168 |
169 | # DocProject is a documentation generator add-in
170 | DocProject/buildhelp/
171 | DocProject/Help/*.HxT
172 | DocProject/Help/*.HxC
173 | DocProject/Help/*.hhc
174 | DocProject/Help/*.hhk
175 | DocProject/Help/*.hhp
176 | DocProject/Help/Html2
177 | DocProject/Help/html
178 |
179 | # Click-Once directory
180 | publish/
181 |
182 | # Publish Web Output
183 | *.[Pp]ublish.xml
184 | *.azurePubxml
185 | # Note: Comment the next line if you want to checkin your web deploy settings,
186 | # but database connection strings (with potential passwords) will be unencrypted
187 | *.pubxml
188 | *.publishproj
189 |
190 | # Microsoft Azure Web App publish settings. Comment the next line if you want to
191 | # checkin your Azure Web App publish settings, but sensitive information contained
192 | # in these scripts will be unencrypted
193 | PublishScripts/
194 |
195 | # NuGet Packages
196 | *.nupkg
197 | # NuGet Symbol Packages
198 | *.snupkg
199 | # The packages folder can be ignored because of Package Restore
200 | **/[Pp]ackages/*
201 | # except build/, which is used as an MSBuild target.
202 | !**/[Pp]ackages/build/
203 | # Uncomment if necessary however generally it will be regenerated when needed
204 | #!**/[Pp]ackages/repositories.config
205 | # NuGet v3's project.json files produces more ignorable files
206 | *.nuget.props
207 | *.nuget.targets
208 |
209 | # Microsoft Azure Build Output
210 | csx/
211 | *.build.csdef
212 |
213 | # Microsoft Azure Emulator
214 | ecf/
215 | rcf/
216 |
217 | # Windows Store app package directories and files
218 | AppPackages/
219 | BundleArtifacts/
220 | Package.StoreAssociation.xml
221 | _pkginfo.txt
222 | *.appx
223 | *.appxbundle
224 | *.appxupload
225 |
226 | # Visual Studio cache files
227 | # files ending in .cache can be ignored
228 | *.[Cc]ache
229 | # but keep track of directories ending in .cache
230 | !?*.[Cc]ache/
231 |
232 | # Others
233 | ClientBin/
234 | ~$*
235 | *~
236 | *.dbmdl
237 | *.dbproj.schemaview
238 | *.jfm
239 | *.pfx
240 | *.publishsettings
241 | orleans.codegen.cs
242 |
243 | # Including strong name files can present a security risk
244 | # (https://github.com/github/gitignore/pull/2483#issue-259490424)
245 | #*.snk
246 |
247 | # Since there are multiple workflows, uncomment next line to ignore bower_components
248 | # (https://github.com/github/gitignore/pull/1529#issuecomment-104372622)
249 | #bower_components/
250 |
251 | # RIA/Silverlight projects
252 | Generated_Code/
253 |
254 | # Backup & report files from converting an old project file
255 | # to a newer Visual Studio version. Backup files are not needed,
256 | # because we have git ;-)
257 | _UpgradeReport_Files/
258 | Backup*/
259 | UpgradeLog*.XML
260 | UpgradeLog*.htm
261 | ServiceFabricBackup/
262 | *.rptproj.bak
263 |
264 | # SQL Server files
265 | *.mdf
266 | *.ldf
267 | *.ndf
268 |
269 | # Business Intelligence projects
270 | *.rdl.data
271 | *.bim.layout
272 | *.bim_*.settings
273 | *.rptproj.rsuser
274 | *- [Bb]ackup.rdl
275 | *- [Bb]ackup ([0-9]).rdl
276 | *- [Bb]ackup ([0-9][0-9]).rdl
277 |
278 | # Microsoft Fakes
279 | FakesAssemblies/
280 |
281 | # GhostDoc plugin setting file
282 | *.GhostDoc.xml
283 |
284 | # Node.js Tools for Visual Studio
285 | .ntvs_analysis.dat
286 | node_modules/
287 |
288 | # Visual Studio 6 build log
289 | *.plg
290 |
291 | # Visual Studio 6 workspace options file
292 | *.opt
293 |
294 | # Visual Studio 6 auto-generated workspace file (contains which files were open etc.)
295 | *.vbw
296 |
297 | # Visual Studio 6 auto-generated project file (contains which files were open etc.)
298 | *.vbp
299 |
300 | # Visual Studio 6 workspace and project file (working project files containing files to include in project)
301 | *.dsw
302 | *.dsp
303 |
304 | # Visual Studio 6 technical files
305 | *.ncb
306 | *.aps
307 |
308 | # Visual Studio LightSwitch build output
309 | **/*.HTMLClient/GeneratedArtifacts
310 | **/*.DesktopClient/GeneratedArtifacts
311 | **/*.DesktopClient/ModelManifest.xml
312 | **/*.Server/GeneratedArtifacts
313 | **/*.Server/ModelManifest.xml
314 | _Pvt_Extensions
315 |
316 | # Paket dependency manager
317 | .paket/paket.exe
318 | paket-files/
319 |
320 | # FAKE - F# Make
321 | .fake/
322 |
323 | # CodeRush personal settings
324 | .cr/personal
325 |
326 | # Python Tools for Visual Studio (PTVS)
327 | __pycache__/
328 | *.pyc
329 |
330 | # Cake - Uncomment if you are using it
331 | # tools/**
332 | # !tools/packages.config
333 |
334 | # Tabs Studio
335 | *.tss
336 |
337 | # Telerik's JustMock configuration file
338 | *.jmconfig
339 |
340 | # BizTalk build output
341 | *.btp.cs
342 | *.btm.cs
343 | *.odx.cs
344 | *.xsd.cs
345 |
346 | # OpenCover UI analysis results
347 | OpenCover/
348 |
349 | # Azure Stream Analytics local run output
350 | ASALocalRun/
351 |
352 | # MSBuild Binary and Structured Log
353 | *.binlog
354 |
355 | # NVidia Nsight GPU debugger configuration file
356 | *.nvuser
357 |
358 | # MFractors (Xamarin productivity tool) working folder
359 | .mfractor/
360 |
361 | # Local History for Visual Studio
362 | .localhistory/
363 |
364 | # Visual Studio History (VSHistory) files
365 | .vshistory/
366 |
367 | # BeatPulse healthcheck temp database
368 | healthchecksdb
369 |
370 | # Backup folder for Package Reference Convert tool in Visual Studio 2017
371 | MigrationBackup/
372 |
373 | # Ionide (cross platform F# VS Code tools) working folder
374 | .ionide/
375 |
376 | # Fody - auto-generated XML schema
377 | FodyWeavers.xsd
378 |
379 | # VS Code files for those working on multiple tools
380 | .vscode/*
381 | !.vscode/settings.json
382 | !.vscode/tasks.json
383 | !.vscode/launch.json
384 | !.vscode/extensions.json
385 | *.code-workspace
386 |
387 | # Local History for Visual Studio Code
388 | .history/
389 |
390 | # Windows Installer files from build outputs
391 | *.cab
392 | *.msi
393 | *.msix
394 | *.msm
395 | *.msp
396 |
397 | # JetBrains Rider
398 | *.sln.iml
399 |
--------------------------------------------------------------------------------
/AESCrypt/AESCrypt.cpp:
--------------------------------------------------------------------------------
1 | /* AESCrypt
2 | * ========
3 | * Takes a file on the command line and encrypts it with AES-256
4 | * using a password and Microsoft Cryptography API. It can also
5 | * take a password and decrypt the file.
6 | *
7 | * https://hacker.house
8 | */
9 | #ifndef ENCRYPTION_TOOL_H
10 | #define ENCRYPTION_TOOL_H
11 |
12 | #include
13 | #include
14 | #include
15 | #include
16 | #include
17 | #include
18 | #include
19 |
20 | #pragma comment(lib, "bcrypt.lib")
21 |
22 | #define NT_SUCCESS(Status) ((NTSTATUS)(Status) >= 0)
23 |
24 | class AESCrypt {
25 | public:
26 | AESCrypt(const wchar_t* filename, const std::wstring& password, const std::wstring& action) :
27 | filename(filename), password(password), action(action) {}
28 |
29 | void Run() {
30 | try {
31 | if (action == L"encrypt") {
32 | EncryptFile();
33 | }
34 | else if (action == L"decrypt") {
35 | DecryptFile();
36 | }
37 | else {
38 | std::wcerr << L"Invalid action. Use 'encrypt' or 'decrypt'." << std::endl;
39 | }
40 | }
41 | catch (const std::exception& ex) {
42 | std::cerr << "Error: " << ex.what() << std::endl;
43 | }
44 | }
45 |
46 | private:
47 | const wchar_t* filename;
48 | std::wstring password;
49 | std::wstring action;
50 |
51 | void PrintError(const wchar_t* msg, NTSTATUS status) {
52 | std::wcerr << L"Error: " << msg << L" (Error code: 0x" << std::hex << status << L")" << std::endl;
53 | }
54 |
55 | void CheckStatus(const wchar_t* msg, NTSTATUS status) {
56 | if (!NT_SUCCESS(status)) {
57 | PrintError(msg, status);
58 | throw std::runtime_error("Operation failed.");
59 | }
60 | }
61 |
62 | void PrintOperationDetails(const wchar_t* operation) {
63 | std::wcout << L"Performing operation: " << operation << std::endl;
64 | }
65 |
66 | void PrintHexDump(const void* data, size_t size) {
67 | const uint8_t* byteData = static_cast(data);
68 | for (size_t i = 0; i < size; ++i) {
69 | std::wcout << std::hex << std::setw(2) << std::setfill(L'0') << static_cast(byteData[i]) << L" ";
70 | if ((i + 1) % 16 == 0) {
71 | std::wcout << std::endl;
72 | }
73 | }
74 | if (size % 16 != 0) {
75 | std::wcout << std::endl;
76 | }
77 | }
78 |
79 | void GenerateRandomBytes(std::vector& buffer) {
80 | PrintOperationDetails(L"Generating random bytes");
81 | BCRYPT_ALG_HANDLE randAlgHandle;
82 | CheckStatus(L"BCryptOpenAlgorithmProvider failed", BCryptOpenAlgorithmProvider(&randAlgHandle, BCRYPT_RNG_ALGORITHM, NULL, 0));
83 | if (!buffer.empty()) {
84 | CheckStatus(L"BCryptGenRandom failed", BCryptGenRandom(randAlgHandle, buffer.data(), static_cast(buffer.size()), 0));
85 | }
86 | BCryptCloseAlgorithmProvider(randAlgHandle, 0);
87 | std::wcout << L"Random bytes generated: " << std::endl;
88 | PrintHexDump(buffer.data(), buffer.size());
89 | }
90 |
91 | std::vector DeriveKey() {
92 | PrintOperationDetails(L"Deriving key from password");
93 | BCRYPT_ALG_HANDLE algHandle;
94 | CheckStatus(L"BCryptOpenAlgorithmProvider failed", BCryptOpenAlgorithmProvider(&algHandle, BCRYPT_SHA256_ALGORITHM, NULL, 0));
95 | BCRYPT_HASH_HANDLE hashHandle;
96 | CheckStatus(L"BCryptCreateHash failed", BCryptCreateHash(algHandle, &hashHandle, NULL, 0, NULL, 0, 0));
97 | CheckStatus(L"BCryptHashData failed", BCryptHashData(hashHandle, reinterpret_cast(const_cast(password.c_str())), static_cast(password.size() * sizeof(wchar_t)), 0));
98 | std::vector keyBuffer(32); // 256 bits for AES-256
99 | CheckStatus(L"BCryptFinishHash failed", BCryptFinishHash(hashHandle, keyBuffer.data(), static_cast(keyBuffer.size()), 0));
100 | BCryptDestroyHash(hashHandle);
101 | BCryptCloseAlgorithmProvider(algHandle, 0);
102 | std::wcout << L"Derived key: " << std::endl;
103 | PrintHexDump(keyBuffer.data(), keyBuffer.size());
104 | return keyBuffer;
105 | }
106 |
107 | std::vector ReadFileContent() {
108 | std::ifstream file(filename, std::ios::binary);
109 | if (!file.is_open()) {
110 | throw std::runtime_error("Error opening file for reading content.");
111 | }
112 | std::vector content((std::istreambuf_iterator(file)), std::istreambuf_iterator());
113 | return content;
114 | }
115 |
116 | void WriteFileContent(const std::vector& content, const wchar_t* outputFilename) {
117 | std::ofstream file(outputFilename, std::ios::binary);
118 | if (!file.is_open()) {
119 | throw std::runtime_error("Error opening file for writing content.");
120 | }
121 | file.write(reinterpret_cast(content.data()), content.size());
122 | }
123 |
124 | void EncryptFile() {
125 | PrintOperationDetails(L"Encrypting file");
126 | std::vector fileContent = ReadFileContent();
127 | std::vector key = DeriveKey();
128 | std::vector iv(16);
129 | GenerateRandomBytes(iv);
130 | std::vector originalIV = iv;
131 | BCRYPT_ALG_HANDLE encAlgHandle;
132 | CheckStatus(L"BCryptOpenAlgorithmProvider failed", BCryptOpenAlgorithmProvider(&encAlgHandle, BCRYPT_AES_ALGORITHM, NULL, 0));
133 | CheckStatus(L"BCryptSetProperty failed", BCryptSetProperty(encAlgHandle, BCRYPT_CHAINING_MODE, (PBYTE)BCRYPT_CHAIN_MODE_CBC, sizeof(BCRYPT_CHAIN_MODE_CBC), 0));
134 | BCRYPT_KEY_HANDLE keyHandle;
135 | CheckStatus(L"BCryptGenerateSymmetricKey failed", BCryptGenerateSymmetricKey(encAlgHandle, &keyHandle, NULL, 0, key.data(), static_cast(key.size()), 0));
136 | DWORD cbCipherText = 0;
137 | CheckStatus(L"BCryptEncrypt (determine size) failed", BCryptEncrypt(
138 | keyHandle,
139 | fileContent.data(),
140 | static_cast(fileContent.size()),
141 | NULL,
142 | iv.data(),
143 | static_cast(iv.size()),
144 | NULL,
145 | 0,
146 | &cbCipherText,
147 | BCRYPT_BLOCK_PADDING
148 | ));
149 | std::vector cipherText(cbCipherText);
150 | CheckStatus(L"BCryptEncrypt failed", BCryptEncrypt(
151 | keyHandle,
152 | fileContent.data(),
153 | static_cast(fileContent.size()),
154 | NULL,
155 | iv.data(),
156 | static_cast(iv.size()),
157 | cipherText.data(),
158 | cbCipherText,
159 | &cbCipherText,
160 | BCRYPT_BLOCK_PADDING
161 | ));
162 | BCryptDestroyKey(keyHandle);
163 | BCryptCloseAlgorithmProvider(encAlgHandle, 0);
164 | std::vector encryptedContent;
165 | encryptedContent.insert(encryptedContent.end(), originalIV.begin(), originalIV.end());
166 | encryptedContent.insert(encryptedContent.end(), cipherText.begin(), cipherText.end());
167 | std::wstring outputFilename = std::wstring(filename) + L".enc";
168 | WriteFileContent(encryptedContent, outputFilename.c_str());
169 | std::wcout << L"Encryption successful." << std::endl;
170 | }
171 |
172 | void DecryptFile() {
173 | PrintOperationDetails(L"Decrypting file");
174 | std::vector encryptedContentBytes = ReadFileContent();
175 | if (encryptedContentBytes.size() < 16) {
176 | throw std::runtime_error("Invalid file format");
177 | }
178 | std::vector iv(encryptedContentBytes.begin(), encryptedContentBytes.begin() + 16);
179 | std::wcout << L"IV: " << std::endl;
180 | PrintHexDump(iv.data(), iv.size());
181 | std::vector cipherText(encryptedContentBytes.begin() + 16, encryptedContentBytes.end());
182 | std::vector key = DeriveKey();
183 | BCRYPT_ALG_HANDLE decAlgHandle;
184 | CheckStatus(L"BCryptOpenAlgorithmProvider failed", BCryptOpenAlgorithmProvider(&decAlgHandle, BCRYPT_AES_ALGORITHM, NULL, 0));
185 | CheckStatus(L"BCryptSetProperty failed", BCryptSetProperty(decAlgHandle, BCRYPT_CHAINING_MODE, (PBYTE)BCRYPT_CHAIN_MODE_CBC, sizeof(BCRYPT_CHAIN_MODE_CBC), 0));
186 | BCRYPT_KEY_HANDLE keyHandle;
187 | CheckStatus(L"BCryptGenerateSymmetricKey failed", BCryptGenerateSymmetricKey(decAlgHandle, &keyHandle, NULL, 0, key.data(), static_cast(key.size()), 0));
188 | std::vector decryptedText(cipherText.size());
189 | DWORD cbData = 0;
190 | NTSTATUS status = BCryptDecrypt(
191 | keyHandle,
192 | cipherText.data(),
193 | static_cast(cipherText.size()),
194 | NULL,
195 | iv.data(),
196 | static_cast(iv.size()),
197 | decryptedText.data(),
198 | static_cast(decryptedText.size()),
199 | &cbData,
200 | BCRYPT_BLOCK_PADDING
201 | );
202 | if (NT_SUCCESS(status)) {
203 | BCryptDestroyKey(keyHandle);
204 | BCryptCloseAlgorithmProvider(decAlgHandle, 0);
205 | decryptedText.resize(cbData);
206 | std::wstring outputFilename = std::wstring(filename) + L".dec";
207 | WriteFileContent(decryptedText, outputFilename.c_str());
208 | std::wcout << L"Decryption successful." << std::endl;
209 | return;
210 | }
211 | BCryptDestroyKey(keyHandle);
212 | BCryptCloseAlgorithmProvider(decAlgHandle, 0);
213 | PrintError(L"BCryptDecrypt failed", status);
214 | throw std::runtime_error("Decryption failed.");
215 | }
216 | };
217 |
218 | int wmain(int argc, wchar_t* argv[]) {
219 | try {
220 | if (argc != 4) {
221 | std::wcerr << L"Usage: " << argv[0] << L" " << std::endl;
222 | return 1;
223 | }
224 | AESCrypt aesCrypt(argv[1], argv[2], argv[3]);
225 | aesCrypt.Run();
226 | }
227 | catch (const std::exception& ex) {
228 | std::cerr << "Error: " << ex.what() << std::endl;
229 | return 1;
230 | }
231 |
232 | return 0;
233 | }
234 | #endif // ENCRYPTION_TOOL_H
--------------------------------------------------------------------------------