Code to handle some weird siutations and bad WIMs

[wimlib] / src / sha1.h

diff --git a/src/sha1.h b/src/sha1.h
index db524e6de60d5a579804ace47682423e73dc9d60..c44b9f40f76ce46cd78d7c12eec3c86a17d28498 100644 (file)
--- a/src/sha1.h
+++ b/src/sha1.h
@@ -27,21 +27,50 @@ static inline bool hashes_equal(const u8 h1[SHA1_HASH_SIZE],
 }
 
 /* Prints a hash code field. */
-static inline void print_hash(const u8 hash[])
+static inline void print_hash(const u8 hash[SHA1_HASH_SIZE])
 {
        print_byte_field(hash, SHA1_HASH_SIZE);
 }
 
-extern int sha1sum(const char *filename, void *md);
+static inline bool is_zero_hash(const u8 hash[SHA1_HASH_SIZE])
+{
+       for (u8 i = 0; i < SHA1_HASH_SIZE / 4; i++)
+               if (((u32*)hash)[i])
+                       return false;
+       return true;
+}
 
-#ifdef WITH_LIBCRYPTO
-#include <openssl/sha.h>
-static inline void sha1_buffer(const void *buffer, size_t len, void *md)
+static void zero_hash(u8 hash[SHA1_HASH_SIZE])
 {
-       SHA1(buffer, len, md);
+       memset(hash, 0, SHA1_HASH_SIZE);
 }
-#else
-extern void sha1_buffer(const void *buffer, size_t len, void *md);
-#endif
+
+
+#ifdef WITH_LIBCRYPTO
+
+#include <openssl/sha.h>
+#define sha1_buffer   SHA1
+#define sha1_init     SHA1_Init
+#define sha1_update   SHA1_Update
+#define sha1_final    SHA1_Final
+
+#else /* WITH_LIBCRYPTO */
+
+typedef struct {
+    u32 state[5];
+    u32 count[2];
+    u8  buffer[64];
+} SHA_CTX;
+
+extern void sha1_buffer(const u8 buffer[], size_t len, u8 hash[SHA1_HASH_SIZE]);
+extern void sha1_init(SHA_CTX *ctx);
+extern void sha1_update(SHA_CTX *ctx, const u8 data[], size_t len);
+extern void sha1_final(u8 hash[SHA1_HASH_SIZE], SHA_CTX *ctx);
+
+#endif /* !WITH_LIBCRYPTO */
+
+extern int sha1sum(const char *filename, u8 hash[SHA1_HASH_SIZE]);
+
+
 
 #endif /* _WIMLIB_SHA1_H */