| ... | @@ -234,3 +234,56 @@ pub fn random_string(len: usize) ![]const u8 { | ... | @@ -234,3 +234,56 @@ pub fn random_string(len: usize) ![]const u8 { |
| 234 | } | 234 | } |
| 235 | return buf; | 235 | return buf; |
| 236 | } | 236 | } |
| | 237 | |
| | 238 | pub fn parse_split(comptime T: type, delim: []const u8) type { |
| | 239 | return struct { |
| | 240 | const Self = @This(); |
| | 241 | |
| | 242 | id: T, |
| | 243 | string: []const u8, |
| | 244 | |
| | 245 | pub fn do(input: []const u8) !Self { |
| | 246 | const iter = &std.mem.split(input, delim); |
| | 247 | return Self{ |
| | 248 | .id = std.meta.stringToEnum(T, iter.next() orelse return error.IterEmpty) orelse return error.NoMemberFound, |
| | 249 | .string = iter.rest(), |
| | 250 | }; |
| | 251 | } |
| | 252 | }; |
| | 253 | } |
| | 254 | |
| | 255 | pub const HashFn = enum { |
| | 256 | blake3, |
| | 257 | sha256, |
| | 258 | sha512, |
| | 259 | }; |
| | 260 | |
| | 261 | pub fn validate_hash(input: []const u8, file_path: []const u8) !bool { |
| | 262 | const hash = parse_split(HashFn, "-").do(input) catch return false; |
| | 263 | const file = try std.fs.cwd().openFile(file_path, .{}); |
| | 264 | defer file.close(); |
| | 265 | const data = try file.reader().readAllAlloc(gpa, mb); |
| | 266 | return std.mem.eql(u8, hash.string, switch (hash.id) { |
| | 267 | .blake3 => blk: { |
| | 268 | const h = &std.crypto.hash.Blake3.init(.{}); |
| | 269 | var out: [32]u8 = undefined; |
| | 270 | h.update(data); |
| | 271 | h.final(&out); |
| | 272 | break :blk try std.fmt.allocPrint(gpa, "{x}", .{out}); |
| | 273 | }, |
| | 274 | .sha256 => blk: { |
| | 275 | const h = &std.crypto.hash.sha2.Sha256.init(.{}); |
| | 276 | var out: [32]u8 = undefined; |
| | 277 | h.update(data); |
| | 278 | h.final(&out); |
| | 279 | break :blk try std.fmt.allocPrint(gpa, "{x}", .{out}); |
| | 280 | }, |
| | 281 | .sha512 => blk: { |
| | 282 | const h = &std.crypto.hash.sha2.Sha512.init(.{}); |
| | 283 | var out: [64]u8 = undefined; |
| | 284 | h.update(data); |
| | 285 | h.final(&out); |
| | 286 | break :blk try std.fmt.allocPrint(gpa, "{x}", .{out}); |
| | 287 | }, |
| | 288 | }); |
| | 289 | } |
Meghan <hello@nektro.net> 2020-12-15 06:26:28 -08:00