From 6475baa50e5f08f6e76d8cc0beb9bc5f6ea3f134 Mon Sep 17 00:00:00 2001
From: malloc <aleco@aroltd.com>
Date: Thu, 15 Mar 2018 17:02:10 -0500
Subject: [PATCH] squid64

---
 server/src/crypto/base64.cpp | 36 ++++++++++++++++++++++++++++++++++++
 server/src/crypto/base64.hpp | 13 +++++++++++++
 server/src/crypto/sha1.cpp   | 32 ++++++++++++++++++++++++++------
 server/src/crypto/sha1.hpp   |  4 +++-
 server/src/main.cpp          |  5 ++++-
 5 files changed, 82 insertions(+), 8 deletions(-)

diff --git a/server/src/crypto/base64.cpp b/server/src/crypto/base64.cpp
index 8b13789..6cec9d5 100644
--- a/server/src/crypto/base64.cpp
+++ b/server/src/crypto/base64.cpp
@@ -1 +1,37 @@
+#include "base64.hpp"
 
+std::string sosc::cgc::base64_encode(const std::string& data) {
+    return base64_decode(data.c_str());
+}
+
+std::string sosc::cgc::base64_encode(const void* raw, size_t length) {
+    uint8_t* data = (uint8_t*)raw;
+    const std::string char_table
+        = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
+        
+    std::string encoded;
+    for(size_t i = 0; i < length; i += 3) {
+        if(i > 0 && ((i / 3) * 4) % 76 == 0)
+            encoded += "\r\n";
+        
+        int index = 0, padding = 0;
+        for(int j = 0; j < 3; ++j) {
+            if(i + j < length)
+                index += data[i + j] << (8 * (2 - j));
+            else
+                ++padding;
+        }
+                
+        for(int j = 0; j < 4 - padding; ++j)
+            encoded += char_table[(index >> (6 * (3 - j))) & 0x3F];
+            
+        for(int j = 0; j < padding; ++j)
+            encoded += '=';
+    }
+    
+    return encoded;
+}
+
+std::string base64_decode(const std::string& data) {
+    
+}
diff --git a/server/src/crypto/base64.hpp b/server/src/crypto/base64.hpp
index 8b13789..6c04310 100644
--- a/server/src/crypto/base64.hpp
+++ b/server/src/crypto/base64.hpp
@@ -1 +1,14 @@
+#ifndef SOSC_CRYPTO_BASE64
+#define SOSC_CRYPTO_BASE64
 
+#include <string>
+
+namespace sosc {
+namespace cgc {
+std::string base64_encode(const std::string& data);
+std::string base64_encode(const void* raw, size_t length);
+
+std::string base64_decode(const std::string& data);
+}}
+
+#endif
diff --git a/server/src/crypto/sha1.cpp b/server/src/crypto/sha1.cpp
index 1100e65..0ad96d8 100644
--- a/server/src/crypto/sha1.cpp
+++ b/server/src/crypto/sha1.cpp
@@ -48,19 +48,19 @@ static uint32_t K(uint8_t t) {
         return 0xCA62C1D6;
 }
 
-std::string sosc::cgc::sha1(const std::string& data) {
+std::string sosc::cgc::sha1(const std::string& data, bool raw) {
     int last_block = data.length() % 64, 
         pad_size = SHA1_BLOCK_SIZE - last_block;
         
     if(last_block >= 55)
         pad_size += SHA1_BLOCK_SIZE;
     
-    std::string padding(0, pad_size);
+    std::string padding(pad_size, 0);
     padding[0] = 0x80;
     
     for(int i = 0; i < 8; ++i) {
         padding[padding.length() - i - 1] =
-            (padding.length() >> (8 * i)) & 0xFF;
+            ((data.length() * 8ull) >> (8 * i)) & 0xFF;
     }
            
     uint32_t H[5] = {
@@ -76,8 +76,10 @@ std::string sosc::cgc::sha1(const std::string& data) {
         for(int j = 0; j < 16; ++j) {
             W[j] = 0;
             for(int k = 0; k < 4; ++k) {
-                W[j] |= (((i + j + k < data.length()) 
-                    ? data : padding)[i + j + k] >> (8 * (3 - i))) & 0xFF;
+                W[j] |= (((i + j * 4 + k < data.length()) 
+                    ? data[i + j * 4 + k] 
+                    : padding[(i + j * 4 + k) - data.length()]) & 0xFF)
+                        << (8 * (3 - k));
             }
         }
         
@@ -100,5 +102,23 @@ std::string sosc::cgc::sha1(const std::string& data) {
             H[j] += O[j];
     }
     
-    return "";
+    if(raw) {
+        std::string hash(20, 0);
+        for(int i = 0; i < 5; ++i) {
+            for(int j = 0; j < 4; ++j)
+                hash[i * 4 + j] = (H[i] >> (8 * (3 - j))) & 0xFF;
+        }
+        
+        return hash;
+    } else {
+        std::stringstream stream;
+        for(int i = 0; i < 5; ++i) {
+            stream << std::setfill('0') 
+                << std::setw(8)
+                << std::hex 
+                << H[i];
+        }
+            
+        return stream.str();
+    }
 }
diff --git a/server/src/crypto/sha1.hpp b/server/src/crypto/sha1.hpp
index 337424f..a742025 100644
--- a/server/src/crypto/sha1.hpp
+++ b/server/src/crypto/sha1.hpp
@@ -3,10 +3,12 @@
 
 #include <cstdint>
 #include <string>
+#include <sstream>
+#include <iomanip>
 
 namespace sosc {
 namespace cgc {
-std::string sha1(const std::string& data);
+std::string sha1(const std::string& data, bool raw = false);
 }}
 
 #endif
diff --git a/server/src/main.cpp b/server/src/main.cpp
index 063bf1d..739cf26 100644
--- a/server/src/main.cpp
+++ b/server/src/main.cpp
@@ -6,6 +6,7 @@
 #include "utils/time.hpp"
 #include "sock/tcpsock.hpp"
 #include "crypto/sha1.hpp"
+#include "crypto/base64.hpp"
 
 int main(int argc, char **argv) {
     //auto sock = sosc::TcpClient();
@@ -29,7 +30,9 @@ int main(int argc, char **argv) {
             std::cout << got << std::endl;
     }*/
     
-    std::string a = sosc::cgc::sha1("test");
+    //std::string a = sosc::cgc::sha1("test", true);
+    
+    std::cout << sosc::cgc::base64_encode("this is a longer test");
     
     return 0;
 }