struct Config {
target_hash: array<u32, 8>,
password_len: u32,
num_passwords: u32,
found_flag: atomic<u32>,
found_password: array<u32, 4>,
mask: array<u32, 16>,
mask_len: u32,
target_hash_extra: array<u32, 8>,
salt: array<u32, 16>,
salt_len: u32,
range_start: u32,
range_end: u32,
num_targets: u32,
}
struct Progress {
count: atomic<u32>,
}
struct TargetEntry {
hash: array<u32, 8>,
hash_extra: array<u32, 8>,
}
struct WordEntry {
chars: array<u32, 5>,
len: u32,
}
@group(0) @binding(0) var<storage, read_write> config: Config;
@group(0) @binding(1) var<storage, read_write> progress: Progress;
@group(0) @binding(2) var<storage, read> word_buf: array<WordEntry>;
@group(0) @binding(3) var<storage, read> targets: array<TargetEntry>;
const K: array<u32, 64> = array<u32, 64>(
0xd76aa478u, 0xe8c7b756u, 0x242070dbu, 0xc1bdceeeu,
0xf57c0fafu, 0x4787c62au, 0xa8304613u, 0xfd469501u,
0x698098d8u, 0x8b44f7afu, 0xffff5bb1u, 0x895cd7beu,
0x6b901122u, 0xfd987193u, 0xa679438eu, 0x49b40821u,
0xf61e2562u, 0xc040b340u, 0x265e5a51u, 0xe9b6c7aau,
0xd62f105du, 0x02441453u, 0xd8a1e681u, 0xe7d3fbc8u,
0x21e1cde6u, 0xc33707d6u, 0xf4d50d87u, 0x455a14edu,
0xa9e3e905u, 0xfcefa3f8u, 0x676f02d9u, 0x8d2a4c8au,
0xfffa3942u, 0x8771f681u, 0x6d9d6122u, 0xfde5380cu,
0xa4beea44u, 0x4bdecfa9u, 0xf6bb4b60u, 0xbebfbc70u,
0x289b7ec6u, 0xeaa127fau, 0xd4ef3085u, 0x04881d05u,
0xd9d4d039u, 0xe6db99e5u, 0x1fa27cf8u, 0xc4ac5665u,
0xf4292244u, 0x432aff97u, 0xab9423a7u, 0xfc93a039u,
0x655b59c3u, 0x8f0ccc92u, 0xffeff47du, 0x85845dd1u,
0x6fa87e4fu, 0xfe2ce6e0u, 0xa3014314u, 0x4e0811a1u,
0xf7537e82u, 0xbd3af235u, 0x2ad7d2bbu, 0xeb86d391u,
);
const S: array<u32, 64> = array<u32, 64>(
7u, 12u, 17u, 22u, 7u, 12u, 17u, 22u,
7u, 12u, 17u, 22u, 7u, 12u, 17u, 22u,
5u, 9u, 14u, 20u, 5u, 9u, 14u, 20u,
5u, 9u, 14u, 20u, 5u, 9u, 14u, 20u,
4u, 11u, 16u, 23u, 4u, 11u, 16u, 23u,
4u, 11u, 16u, 23u, 4u, 11u, 16u, 23u,
6u, 10u, 15u, 21u, 6u, 10u, 15u, 21u,
6u, 10u, 15u, 21u, 6u, 10u, 15u, 21u,
);
fn left_rotate(x: u32, n: u32) -> u32 { return (x << n) | (x >> (32u - n)); }
fn md5_block(state: ptr<function, array<u32, 4>>, block: array<u32, 16>) {
var a = (*state)[0u]; var b = (*state)[1u]; var c = (*state)[2u]; var d = (*state)[3u];
for (var i = 0u; i < 64u; i++) {
var f: u32; var g: u32;
if (i < 16u) { f = (b & c) | ((~b) & d); g = i; }
else if (i < 32u) { f = (d & b) | ((~d) & c); g = (5u * i + 1u) % 16u; }
else if (i < 48u) { f = b ^ c ^ d; g = (3u * i + 5u) % 16u; }
else { f = c ^ (b | (~d)); g = (7u * i) % 16u; }
f = f + a + K[i] + block[g];
let old_b = b; a = d; d = c; c = old_b; b = old_b + left_rotate(f, S[i]);
}
(*state)[0u] += a; (*state)[1u] += b; (*state)[2u] += c; (*state)[3u] += d;
}
fn md5(pwd: array<u32, 5>, len: u32) -> array<u32, 8> {
var state: array<u32, 4> = array<u32, 4>(
0x67452301u, 0xefcdab89u, 0x98badcfeu, 0x10325476u,
);
var block: array<u32, 16>;
for (var i = 0u; i < 16u; i++) { block[i] = 0u; }
// pwd stores chars in reverse order: pwd[0] = last char, pwd[len-1] = first char.
// Each element is a single character code in its own u32 slot.
for (var i = 0u; i < len && i < 16u; i++) {
let ch = pwd[len - 1u - i] & 0xFFu;
let dst_word = i / 4u;
let dst_shift = (i % 4u) * 8u;
block[dst_word] |= ch << dst_shift;
}
// Salt (little-endian, right after password)
for (var i = 0u; i < config.salt_len && i < 16u; i++) {
let ch = config.salt[config.salt_len - 1u - i] & 0xFFu;
let dst_byte = len + i;
let dst_word = dst_byte / 4u;
let dst_shift = (dst_byte % 4u) * 8u;
block[dst_word] |= ch << dst_shift;
}
let pad_byte = len + config.salt_len;
let pad_word = pad_byte / 4u;
let pad_shift = (pad_byte % 4u) * 8u;
block[pad_word] |= 0x80u << pad_shift;
block[14u] = (len + config.salt_len) * 8u;
block[15u] = 0u;
md5_block(&state, block);
return array<u32, 8>(state[0u], state[1u], state[2u], state[3u], 0u, 0u, 0u, 0u);
}
const MAX_DISPATCH_X: u32 = 65535u;
@compute @workgroup_size(128)
fn main(@builtin(global_invocation_id) id: vec3<u32>) {
let index = id.y * MAX_DISPATCH_X * 128u + id.x;
if (index >= config.range_end) { return; }
if (atomicLoad(&config.found_flag) != 0u) { atomicAdd(&progress.count, 1u); return; } let entry = word_buf[index];
let hash = md5(entry.chars, entry.len);
var match_found = false;
for (var t = 0u; t < config.num_targets; t++) {
var t_match = true;
for (var i = 0u; i < 4u; i++) { if (hash[i] != targets[t].hash[i]) { t_match = false; } }
if (t_match) { match_found = true; }
}
if (match_found) {
let prev = atomicCompareExchangeWeak(&config.found_flag, 0u, 1u);
if (prev.old_value == 0u) { config.found_password[0] = index; }
}
atomicAdd(&progress.count, 1u);
}