From 16b63f8cdf5b488ab2f4b0e029478b01a2d6b8a8 Mon Sep 17 00:00:00 2001
From: Boris Kolpackov <boris@codesynthesis.com>
Date: Tue, 10 Jul 2018 09:20:25 +0200
Subject: Add support for hashing ifdstream

---
 libbutl/sha1.cxx   | 20 ++++++++++++++++++++
 libbutl/sha1.mxx   | 10 ++++++++++
 libbutl/sha256.cxx | 18 ++++++++++++++++++
 libbutl/sha256.mxx | 10 ++++++++++
 4 files changed, 58 insertions(+)

(limited to 'libbutl')

diff --git a/libbutl/sha1.cxx b/libbutl/sha1.cxx
index 8fdb833..af1f260 100644
--- a/libbutl/sha1.cxx
+++ b/libbutl/sha1.cxx
@@ -42,6 +42,8 @@ extern "C"
 #define SHA1_Update(x, y, z) sha1_loop((x), (const uint8_t *)(y), (z))
 #define SHA1_Final(x, y)     sha1_result((y), (char(&)[20])(x))
 
+#include <cassert>
+
 #ifndef __cpp_lib_modules
 #include <string>
 #include <cstddef>
@@ -53,12 +55,16 @@ extern "C"
 #ifdef __cpp_modules
 module butl.sha1;
 
+// Only imports additional to interface.
 #ifdef __clang__
 #ifdef __cpp_lib_modules
 import std.core;
 #endif
 #endif
 
+import butl.fdstream;
+#else
+#include <libbutl/fdstream.mxx>
 #endif
 
 using namespace std;
@@ -78,6 +84,20 @@ namespace butl
     SHA1_Update (reinterpret_cast<SHA1_CTX*> (buf_), b, n);
   }
 
+  void sha1::
+  append (ifdstream& is)
+  {
+    fdbuf* buf (dynamic_cast<fdbuf*> (is.rdbuf ()));
+    assert (buf != nullptr);
+
+    while (is.peek () != ifdstream::traits_type::eof () && is.good ())
+    {
+      size_t n (buf->egptr () - buf->gptr ());
+      append (buf->gptr (), n);
+      buf->gbump (static_cast<int> (n));
+    }
+  }
+
   const sha1::digest_type& sha1::
   binary () const
   {
diff --git a/libbutl/sha1.mxx b/libbutl/sha1.mxx
index a3b75d7..ea4d240 100644
--- a/libbutl/sha1.mxx
+++ b/libbutl/sha1.mxx
@@ -28,6 +28,8 @@ import std.core;
 
 LIBBUTL_MODEXPORT namespace butl
 {
+  class ifdstream;
+
   // SHA1 checksum calculator.
   //
   // For a single chunk of data a sum can be obtained in one line, for
@@ -64,6 +66,14 @@ LIBBUTL_MODEXPORT namespace butl
     explicit
     sha1 (const char* s): sha1 () {append (s);}
 
+    // Append stream.
+    //
+    void
+    append (ifdstream&);
+
+    explicit
+    sha1 (ifdstream& i): sha1 () {append (i);}
+
     // Extract result.
     //
     // It can be obtained as either a 20-byte binary digest or as a 40-
diff --git a/libbutl/sha256.cxx b/libbutl/sha256.cxx
index a688a15..c4556e7 100644
--- a/libbutl/sha256.cxx
+++ b/libbutl/sha256.cxx
@@ -27,6 +27,8 @@ extern "C"
 #include "sha256c.c"
 }
 
+#include <cassert>
+
 #ifndef __cpp_lib_modules
 #include <string>
 #include <cstddef>
@@ -53,8 +55,10 @@ import std.core;
 #endif
 
 import butl.utility; // *case()
+import butl.fdstream;
 #else
 #include <libbutl/utility.mxx>
+#include <libbutl/fdstream.mxx>
 #endif
 
 using namespace std;
@@ -74,6 +78,20 @@ namespace butl
     SHA256_Update (reinterpret_cast<SHA256_CTX*> (buf_), b, n);
   }
 
+  void sha256::
+  append (ifdstream& is)
+  {
+    fdbuf* buf (dynamic_cast<fdbuf*> (is.rdbuf ()));
+    assert (buf != nullptr);
+
+    while (is.peek () != ifdstream::traits_type::eof () && is.good ())
+    {
+      size_t n (buf->egptr () - buf->gptr ());
+      append (buf->gptr (), n);
+      buf->gbump (static_cast<int> (n));
+    }
+  }
+
   const sha256::digest_type& sha256::
   binary () const
   {
diff --git a/libbutl/sha256.mxx b/libbutl/sha256.mxx
index ba8eeea..c0c8a84 100644
--- a/libbutl/sha256.mxx
+++ b/libbutl/sha256.mxx
@@ -29,6 +29,8 @@ import std.core;
 
 LIBBUTL_MODEXPORT namespace butl
 {
+  class ifdstream;
+
   // SHA256 checksum calculator.
   //
   // For a single chunk of data a sum can be obtained in one line, for
@@ -98,6 +100,14 @@ LIBBUTL_MODEXPORT namespace butl
         append (&x, len);
     }
 
+    // Append stream.
+    //
+    void
+    append (ifdstream&);
+
+    explicit
+    sha256 (ifdstream& i): sha256 () {append (i);}
+
     // Extract result.
     //
     // It can be obtained as either a 32-byte binary digest or as a 64-
-- 
cgit v1.1