blowfish/tests/Main2.cpp

#include <iostream>

#include <blowfish/blowfish2.h>

std::string from_uint(uint64_t sh) {
  std::string re("");
  for (int i = 0; i < 8; i++) {
    re += (unsigned char)(sh >> i * 8);
  }
  return re;
}

int main(int argc, char const *argv[]) {
  std::string key("test@pass47");
  std::string message("My name is Avinal and I am cute and smart");
  std::string cipher("");
  int len = message.length();
  int j = sizeof(uint64_t);
  int rem =
      ((len > j * 2) ? (((len / j * 2) + 1) * j * 2 - len) : (j * 2 - len));
  message.append(rem, '\0');
  len = message.length();
  Blowfish2 blowfish(key);
  std::cout << "My message is: " << message << std::endl;
  uint64_t lm, rm;
  for (size_t i = 0; i < len; i += 16) {
    lm = 0;
    rm = 0;
    lm = *reinterpret_cast<uint64_t *>(
        const_cast<char *>(message.substr(i, 8).c_str()));
    rm = *reinterpret_cast<uint64_t *>(
        const_cast<char *>(message.substr(i + 8, 8).c_str()));
    blowfish.encrypt(lm, rm);
    cipher += from_uint(lm) + from_uint(rm);
  }
  std::cout << cipher << std::endl;
  std::string decipher("");
  len = cipher.length();
  std::cout << "length: " << len << std::endl;
  for (size_t i = 0; i < len; i += 16) {
    lm = 0;
    rm = 0;
    lm = *reinterpret_cast<uint64_t *>(
        const_cast<char *>(cipher.substr(i, j).c_str()));
    rm = *reinterpret_cast<uint64_t *>(
        const_cast<char *>(cipher.substr(i + j, j).c_str()));
    blowfish.decrypt(lm, rm);
    decipher += from_uint(lm) + from_uint(rm);
  }

  std::cout << decipher << std::endl;
  if (message == decipher) {
    std::cout << "Test successful!" << std::endl;
    return 0;
  } else {
    return 1;
  }
}