| ... | ... | @@ -141,3 +141,67 @@ pub fn fileSize(dir: std.fs.Dir, sub_path: string) !usize { |
| 141 | 141 | const s = try f.stat(); |
| 142 | 142 | return s.size; |
| 143 | 143 | } |
| 144 | |
| 145 | pub const HashFn = enum { |
| 146 | blake3, |
| 147 | gimli, |
| 148 | md5, |
| 149 | sha1, |
| 150 | sha224, |
| 151 | sha256, |
| 152 | sha384, |
| 153 | sha512, |
| 154 | sha3_224, |
| 155 | sha3_256, |
| 156 | sha3_384, |
| 157 | sha3_512, |
| 158 | }; |
| 159 | |
| 160 | pub fn hashFile(alloc: *std.mem.Allocator, dir: std.fs.Dir, sub_path: string, comptime algo: HashFn) !string { |
| 161 | const file = try dir.openFile(sub_path, .{}); |
| 162 | defer file.close(); |
| 163 | const hash = std.crypto.hash; |
| 164 | const Algo = switch (algo) { |
| 165 | .blake3 => hash.Blake3, |
| 166 | .gimli => hash.Gimli, |
| 167 | .md5 => hash.Md5, |
| 168 | .sha1 => hash.Sha1, |
| 169 | .sha224 => hash.sha2.Sha224, |
| 170 | .sha256 => hash.sha2.Sha256, |
| 171 | .sha384 => hash.sha2.Sha384, |
| 172 | .sha512 => hash.sha2.Sha512, |
| 173 | .sha3_224 => hash.sha3.Sha3_224, |
| 174 | .sha3_256 => hash.sha3.Sha3_256, |
| 175 | .sha3_384 => hash.sha3.Sha3_384, |
| 176 | .sha3_512 => hash.sha3.Sha3_512, |
| 177 | }; |
| 178 | const h = &Algo.init(.{}); |
| 179 | var out: [Algo.digest_length]u8 = undefined; |
| 180 | var hw = HashWriter(Algo){ .h = h }; |
| 181 | try pipe(file.reader(), hw.writer()); |
| 182 | h.final(&out); |
| 183 | |
| 184 | var res: [Algo.digest_length * 2]u8 = undefined; |
| 185 | var fbs = std.io.fixedBufferStream(&res); |
| 186 | try std.fmt.format(fbs.writer(), "{x}", .{std.fmt.fmtSliceHexLower(&out)}); |
| 187 | return try alloc.dupe(u8, &res); |
| 188 | } |
| 189 | |
| 190 | fn HashWriter(comptime T: type) type { |
| 191 | return struct { |
| 192 | h: *T, |
| 193 | |
| 194 | const Self = @This(); |
| 195 | pub const Error = error{}; |
| 196 | pub const Writer = std.io.Writer(*Self, Error, write); |
| 197 | |
| 198 | fn write(self: *Self, bytes: []const u8) Error!usize { |
| 199 | self.h.update(bytes); |
| 200 | return bytes.len; |
| 201 | } |
| 202 | |
| 203 | pub fn writer(self: *Self) Writer { |
| 204 | return .{ .context = self }; |
| 205 | } |
| 206 | }; |
| 207 | } |
Meghan Denny <hello@nektro.net> 2021-12-12 15:19:50 -08:00