From: Maxim Mamontov <faust.madf@gmail.com>
Date: Sat, 13 Sep 2014 21:56:48 +0000 (+0300)
Subject: Handle short string correctly in blowfish encription.
X-Git-Url: https://git.stg.codes/stg.git/commitdiff_plain/ed18fae7caed32c29b04753096d923bbf6ebb6ca?ds=sidebyside

Handle short string correctly in blowfish encription.
---

diff --git a/stglibs/crypto.lib/blowfish.c b/stglibs/crypto.lib/blowfish.c
index 90c93b9d..e2f275e6 100644
--- a/stglibs/crypto.lib/blowfish.c
+++ b/stglibs/crypto.lib/blowfish.c
@@ -504,7 +504,16 @@ void EncryptString(void * d, const void * s, size_t length, const BLOWFISH_CTX *
 size_t pos = 0;
 while (pos < length)
     {
-    EncryptBlock(d + pos, s + pos, ctx);
+    if (pos + 8 < length)
+        EncryptBlock(d + pos, s + pos, ctx);
+    else
+        {
+        // Short string, use 0-padded buffer.
+        char buf[8];
+        memset(buf, 0, sizeof(buf));
+        memcpy(buf, s + pos, length - pos);
+        EncryptBlock(d + pos, buf, ctx);
+        }
     pos += 8;
     }
 }