github/RRG-Proxmark3
1
0
Fork 0
mirror of https://github.com/RfidResearchGroup/proxmark3.git synced 2025-08-21 05:43:48 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

style

Browse source
This commit is contained in:
iceman1001 2024-03-25 23:33:44 +01:00
commit 140cc1d957
1 changed files with 116 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

156
tools/hitag2crack/crack5opencl/ht2crack5opencl.c
View file

@ -35,6 +35,13 @@
#include "hitag2.h"
#include "dolphin_macro.h"
#define AEND "\x1b[0m"
#define _RED_(s) "\x1b[31m" s AEND
#define _GREEN_(s) "\x1b[32m" s AEND
#define _YELLOW_(s) "\x1b[33m" s AEND
#define _CYAN_(s) "\x1b[36m" s AEND
#if defined(__MINGW64__)
#define timersub(a, b, result) \
do { \
@ -282,9 +289,13 @@ int main(int argc, char **argv) {
free(platforms_selected);
free(devices_selected);
if (device_types_selected == 0) device_types_selected = CL_DEVICE_TYPE_GPU;
else if (device_types_selected == 1) device_types_selected = CL_DEVICE_TYPE_CPU;
else device_types_selected = CL_DEVICE_TYPE_ALL;
if (device_types_selected == 0) {
device_types_selected = CL_DEVICE_TYPE_GPU;
} else if (device_types_selected == 1) {
device_types_selected = CL_DEVICE_TYPE_CPU;
} else {
device_types_selected = CL_DEVICE_TYPE_ALL;
}
if (show) {
plat_sel[0] = 0xff;
@ -293,17 +304,23 @@ int main(int argc, char **argv) {
}
if (verbose) {
if (plat_sel[0] == 0xff) printf("Platforms selected : ALL\n");
else {
if (plat_sel[0] == 0xff) {
printf("Platforms selected : ALL\n");
} else {
printf("Platforms selected : %u", plat_sel[0]);
for (unsigned int i = 1; i < plat_cnt; i++) printf(", %u", plat_sel[i]);
for (unsigned int i = 1; i < plat_cnt; i++) {
printf(", %u", plat_sel[i]);
}
printf("\n");
}
if (dev_sel[0] == 0xff) printf("Devices selected : ALL\n");
else {
if (dev_sel[0] == 0xff) {
printf("Devices selected : ALL\n");
} else {
printf("Devices selected : %u", dev_sel[0]);
for (unsigned int i = 1; i < dev_cnt; i++) printf(", %u", dev_sel[i]);
for (unsigned int i = 1; i < dev_cnt; i++) {
printf(", %u", dev_sel[i]);
}
printf("\n");
}
@ -312,7 +329,8 @@ int main(int argc, char **argv) {
printf("Profile selected : %u\n", profile_selected);
}
if (!show) {
if (show == false) {
if ((argc - optind) < 5) {
#if DEBUGME > 0
printf("Error: invalid extra arguments\n");
@ -324,41 +342,57 @@ int main(int argc, char **argv) {
switch (e) {
case 0: // UID
if (!strncmp(argv[optind], "0x", 2) || !strncmp(argv[optind], "0X", 2)) {
if (strlen(argv[optind]) != 2 + 8) { printf("Error: invalid UID length\n"); usage(argv[0]); }
if (strlen(argv[optind]) != 2 + 8) {
printf("Error: invalid UID length\n"); usage(argv[0]);
}
uid = (uint32_t) rev32(hexreversetoulong(argv[optind] + 2));
} else {
if (strlen(argv[optind]) != 8) { printf("Error: invalid UID length\n"); usage(argv[0]); }
if (strlen(argv[optind]) != 8) {
printf("Error: invalid UID length\n"); usage(argv[0]);
}
uid = (uint32_t) rev32(hexreversetoulong(argv[optind]));
}
break;
case 1: // nR1
if (!strncmp(argv[optind], "0x", 2) || !strncmp(argv[optind], "0X", 2)) {
if (strlen(argv[optind]) != 2 + 8) { printf("Error: invalid nR1 length\n"); usage(argv[0]); }
if (strlen(argv[optind]) != 2 + 8) {
printf("Error: invalid nR1 length\n"); usage(argv[0]);
}
nR1 = (uint32_t) rev32(hexreversetoulong(argv[optind] + 2));
} else {
if (strlen(argv[optind]) != 8) { printf("Error: invalid nR1 length\n"); usage(argv[0]); }
if (strlen(argv[optind]) != 8) {
printf("Error: invalid nR1 length\n"); usage(argv[0]);
}
nR1 = (uint32_t) rev32(hexreversetoulong(argv[optind]));
}
break;
case 2: // aR1
if (strlen(argv[optind]) != 8) { printf("Error: invalid aR1 length\n"); usage(argv[0]); }
if (strlen(argv[optind]) != 8) {
printf("Error: invalid aR1 length\n"); usage(argv[0]);
}
aR1 = (uint32_t) strtoul(argv[optind], NULL, 16);
break;
case 3: // nR2
if (!strncmp(argv[optind], "0x", 2) || !strncmp(argv[optind], "0X", 2)) {
if (strlen(argv[optind]) != 2 + 8) { printf("Error: invalid nR2 length\n"); usage(argv[0]); }
if (strlen(argv[optind]) != 2 + 8) {
printf("Error: invalid nR2 length\n"); usage(argv[0]);
}
nR2 = (uint32_t) rev32(hexreversetoulong(argv[optind] + 2));
} else {
if (strlen(argv[optind]) != 8) { printf("Error: invalid nR2 length\n"); usage(argv[0]); }
if (strlen(argv[optind]) != 8) {
printf("Error: invalid nR2 length\n"); usage(argv[0]);
}
nR2 = (uint32_t) rev32(hexreversetoulong(argv[optind]));
}
break;
case 4: // aR2
if (strlen(argv[optind]) != 8) { printf("Error: invalid aR2 length\n"); usage(argv[0]); }
if (strlen(argv[optind]) != 8) {
printf("Error: invalid aR2 length\n"); usage(argv[0]);
}
aR2 = (uint32_t) strtoul(argv[optind], NULL, 16);
break;
@ -379,8 +413,10 @@ int main(int argc, char **argv) {
uint32_t checks[4] = { uid, aR2, nR1, nR2 };
if (!show) {
if (verbose) printf("uid: %u, aR2: %u, nR1: %u, nR2: %u\n", checks[0], checks[1], checks[2], checks[3]);
if (show == false) {
if (verbose) {
printf("uid: %u, aR2: %u, nR1: %u, nR2: %u\n", checks[0], checks[1], checks[2], checks[3]);
}
uint32_t target = ~aR1;
// bitslice inverse target bits
@ -419,7 +455,7 @@ int main(int argc, char **argv) {
exit(3);
}
if (!show) {
if (show == false) {
// load OpenCL kernel source
struct stat st;
const char *opencl_kernel = "ht2crack5opencl_kernel.cl";
@ -463,17 +499,23 @@ int main(int argc, char **argv) {
size_t selected_devices_cnt = 0;
compute_platform_ctx_t *cd_ctx = NULL;
if (show) verbose = true;
if (show) {
verbose = true;
}
// now discover and set up compute device(s)
if ((err = discoverDevices(profile_selected, device_types_selected, &ocl_platform_cnt, &selected_platforms_cnt, &selected_devices_cnt, &cd_ctx, plat_sel, plat_cnt, dev_sel, dev_cnt, verbose, show)) != 0) {
printf("Error: discoverDevices() failed\n");
if (err < -5) free(cd_ctx);
if (err < -5) {
free(cd_ctx);
}
MEMORY_FREE_ALL
exit(2);
}
if (verbose) printf("\n");
if (verbose) {
printf("\n");
}
// new selection engine, need to support multi-gpu system (with the same platform)
if (show) {
@ -504,10 +546,14 @@ int main(int argc, char **argv) {
printf("Selected %zu OpenCL Device(s)\n\n", selected_devices_cnt);
for (w = 0; w < ocl_platform_cnt; w++) {
if (!cd_ctx[w].selected) continue;
if (!cd_ctx[w].selected) {
continue;
}
for (q = 0; q < cd_ctx[w].device_cnt; q++) {
if (!cd_ctx[w].device[q].selected) continue;
if (!cd_ctx[w].device[q].selected) {
continue;
}
if (cd_ctx[w].is_apple && !strncmp(cd_ctx[w].device[q].vendor, "Intel", 5)) {
// disable hitag2 with apple platform and not apple device vendor (< Apple M1)
@ -749,7 +795,9 @@ int main(int argc, char **argv) {
blen += 21;
}
if (verbose) printf("[%zu] Building OpenCL program with options (len %zu): %s\n", z, blen, build_options);
if (verbose) {
printf("[%zu] Building OpenCL program with options (len %zu): %s\n", z, blen, build_options);
}
err = clBuildProgram(ctx.programs[z], 1, &ctx.device_ids[z], build_options, NULL, NULL);
@ -868,7 +916,9 @@ int main(int argc, char **argv) {
// dow't allow gws < lws
if (ctx.global_ws[z] < ctx.local_ws[z]) ctx.local_ws[z] = ctx.global_ws[z];
if (opencl_profiling) printf("[%zu] global_ws %zu, local_ws %zu\n", g, ctx.global_ws[z], ctx.local_ws[z]);
if (opencl_profiling) {
printf("[%zu] global_ws %zu, local_ws %zu\n", g, ctx.global_ws[z], ctx.local_ws[z]);
}
if (!ctx.force_hitag2_opencl) {
if (!(matches[z] = (uint64_t *) calloc((uint32_t)(ctx.global_ws[z] * WGS_MATCHES_FACTOR), sizeof(uint64_t)))) {
@ -1128,8 +1178,15 @@ int main(int argc, char **argv) {
printf("Attack 5 - opencl - start (Max Slices %u, %s order", max_step, wu_queue_strdesc(ctx.queue_ctx.queue_type));
if (!verbose) printf(")\n\n");
else printf(", Profile %u, Async Threads %s, HiTag2 key verify on device %s)\n\n", profile, (ctx.thread_sched_type == THREAD_TYPE_ASYNC) ? "yes" : "no", (force_hitag2_opencl) ? "yes" : "no");
if (!verbose) {
printf(")\n\n");
} else {
printf(", Profile %u, Async Threads %s, HiTag2 key verify on device %s)\n\n"
, profile
, (ctx.thread_sched_type == THREAD_TYPE_ASYNC) ? "yes" : "no"
, (force_hitag2_opencl) ? "yes" : "no"
);
}
if (gettimeofday(&cpu_t_start, NULL) == -1) {
printf("Error: gettimeofday(start) failed (%d): %s\n", errno, strerror(errno));
@ -1147,17 +1204,27 @@ int main(int argc, char **argv) {
// if found, show the key here
for (size_t y = 0; y < thread_count; y++) {
if (t_arg[y].r) {
if (verbose) printf("\n");
if (thread_count > 1) printf("[%zu] ", y);
if (verbose) {
printf("\n");
}
if (thread_count > 1) {
printf("[%zu] ", y);
}
printf("\nKey found @ slice %zu/%zu [ \x1b[32m"
, t_arg[y].slice
, t_arg[y].max_slices
);
printf("\nKey found @ slice %zu/%zu: [ ", t_arg[y].slice, t_arg[y].max_slices);
for (int i = 0; i < 6; i++) {
printf("%02X", (uint8_t)(t_arg[y].key & 0xff));
t_arg[y].key = t_arg[y].key >> 8;
}
printf(" ]\n");
printf(AEND " ]\n");
fflush(stdout);
break;
}
@ -1172,14 +1239,21 @@ int main(int argc, char **argv) {
}
}
if (!found) {
printf("\nError. %s\n", (error) ? "something went wrong :(" : "Key not found :|");
if (found == false) {
if (error) {
printf("\nSomething went wrong ( " _RED_("fail") " )\n");
} else {
printf("\nExhausted keyspace ( " _RED_("fail") " )\n");
}
}
printf("\nAttack 5 - opencl - end");
if (show_overall_time) printf(" in %ld.%2ld second(s).\n\n", (long int)cpu_t_result.tv_sec, (long int)cpu_t_result.tv_usec);
else printf("\n");
if (show_overall_time) {
printf(" in " _YELLOW_("%ld.%2ld") " second(s).\n\n", (long int)cpu_t_result.tv_sec, (long int)cpu_t_result.tv_usec);
} else {
printf("\n");
}
fflush(stdout);
@ -1188,14 +1262,16 @@ int main(int argc, char **argv) {
fflush(stdout);
#endif
if (!error && th_ctx.type != THREAD_TYPE_SEQ) thread_stop(&th_ctx);
if (error == false && th_ctx.type != THREAD_TYPE_SEQ) {
thread_stop(&th_ctx);
}
#if DEBUGME > 1
printf("destroy threads\n");
fflush(stdout);
#endif
if (!error) {
if (error == false) {
if ((ret = thread_destroy(&th_ctx)) != 0) {
#if DEBUGME > 0
printf("Warning: thread_destroy() failed (%d): %s\n", ret, thread_strerror(ret));