Add a simple update_be to BufferedComputation that takes an integer

and writes the bytes to the stream in big-endian order. Use it in
KDF2, MGF1, and FPE.

1 files changed, 2 insertions, 3 deletions

diff --git a/src/kdf/kdf2/kdf2.cpp b/src/kdf/kdf2/kdf2.cpp
index b10077e35..51b9e41ea 100644
--- a/src/kdf/kdf2/kdf2.cpp
+++ b/src/kdf/kdf2/kdf2.cpp
@@ -6,7 +6,6 @@
 */
 
 #include <botan/kdf2.h>
-#include <botan/get_byte.h>
 
 namespace Botan {
 
@@ -23,9 +22,9 @@ SecureVector<byte> KDF2::derive(size_t out_len,
    while(out_len && counter)
       {
       hash->update(secret, secret_len);
-      for(size_t i = 0; i != 4; ++i)
-         hash->update(get_byte(i, counter));
+      hash->update_be(counter);
       hash->update(P, P_len);
+
       SecureVector<byte> hash_result = hash->final();
 
       size_t added = std::min(hash_result.size(), out_len);