88a507932e
reformat the source in an unified way. (improve later with clang-format config) fix order of includes use proper delete[] operator
123 lines
2.8 KiB
C++
123 lines
2.8 KiB
C++
#include "keyinfo.h"
|
|
#include "scsiencrypt.h"
|
|
#include <config.h>
|
|
#include <iostream>
|
|
#include <sstream>
|
|
#include <string.h>
|
|
#include <string>
|
|
|
|
Keyinfo::Keyinfo() {
|
|
valid = false;
|
|
check = "";
|
|
key = NULL;
|
|
keySize = 0;
|
|
}
|
|
void Keyinfo::load(std::string hexinput) {
|
|
valid = true;
|
|
if (hexinput.size() < 2) {
|
|
valid = false;
|
|
std::cout << "Key input too short!\n";
|
|
return;
|
|
}
|
|
// parse for invalid characters
|
|
for (unsigned int i = 0; i < hexinput.size(); i++) {
|
|
switch ((unsigned char)hexinput.at(i)) {
|
|
case '0':
|
|
case '1':
|
|
case '2':
|
|
case '3':
|
|
case '4':
|
|
case '5':
|
|
case '6':
|
|
case '7':
|
|
case '8':
|
|
case '9':
|
|
case 'a':
|
|
case 'b':
|
|
case 'c':
|
|
case 'd':
|
|
case 'e':
|
|
case 'f':
|
|
case 'A':
|
|
case 'B':
|
|
case 'C':
|
|
case 'D':
|
|
case 'E':
|
|
case 'F':
|
|
break;
|
|
default:
|
|
std::cout << "Invalid character '" << hexinput.at(i)
|
|
<< "' found in key!\n";
|
|
valid = false;
|
|
return;
|
|
}
|
|
}
|
|
// delete the key if its already allocated
|
|
if (key != NULL)
|
|
delete[] key;
|
|
// check that the input size is divisible by 2
|
|
if (hexinput.size() % 2 != 0) {
|
|
valid = false;
|
|
std::cout << "Each hexadecimal byte must consist of 2 digits!\n";
|
|
return;
|
|
}
|
|
// convert the hex input to a char*
|
|
loadKey(hexinput);
|
|
// load the check value
|
|
loadCheck();
|
|
// check for oversized key
|
|
if (keySize == 0 || keySize > SSP_KEY_LENGTH) {
|
|
std::cout << "Key size cannot exceed " << (SSP_KEY_LENGTH * 8)
|
|
<< " bits!\n";
|
|
std::cout << "Provided key is " << (keySize * 8) << " bits in length.\n";
|
|
valid = false;
|
|
return;
|
|
}
|
|
std::cout << "Provided key length is " << (keySize * 8) << " bits.\n";
|
|
std::cout << "Key checksum is " << check << ".\n";
|
|
}
|
|
void Keyinfo::loadCheck() {
|
|
int i;
|
|
int chk = 0;
|
|
for (i = 0; i < keySize; i++) {
|
|
chk += ((int)key[i]) * (i + 1);
|
|
}
|
|
std::stringstream retval;
|
|
retval << std::hex << chk;
|
|
check = retval.str();
|
|
}
|
|
Keyinfo::~Keyinfo() { delete[] key; }
|
|
void Keyinfo::loadKey(std::string str) {
|
|
int length = str.size();
|
|
// make sure the input string has an even digit numbers
|
|
if (length % 2 == 1) {
|
|
str = "0" + str;
|
|
length++;
|
|
}
|
|
|
|
// allocate memory for the output array
|
|
key = new char[length / 2];
|
|
memset(key, 0, (length / 2) + 1);
|
|
keySize = length / 2;
|
|
|
|
std::stringstream sstr(str);
|
|
for (int i = 0; i < keySize; i++) {
|
|
char ch1, ch2;
|
|
sstr >> ch1 >> ch2;
|
|
int dig1 = 0, dig2 = 0;
|
|
if (isdigit(ch1))
|
|
dig1 = ch1 - '0';
|
|
else if (ch1 >= 'A' && ch1 <= 'F')
|
|
dig1 = ch1 - 'A' + 10;
|
|
else if (ch1 >= 'a' && ch1 <= 'f')
|
|
dig1 = ch1 - 'a' + 10;
|
|
if (isdigit(ch2))
|
|
dig2 = ch2 - '0';
|
|
else if (ch2 >= 'A' && ch2 <= 'F')
|
|
dig2 = ch2 - 'A' + 10;
|
|
else if (ch2 >= 'a' && ch2 <= 'f')
|
|
dig2 = ch2 - 'a' + 10;
|
|
key[i] = dig1 * 16 + dig2;
|
|
}
|
|
}
|