mirror of
https://github.com/Neo-Desktop/WindowsXPKg
synced 2025-12-23 10:06:48 +02:00
WIP more rewrite/rework changes, major refactoring
This commit is contained in:
@@ -27,145 +27,266 @@
|
||||
#include "pidgen3/BINK2002.h"
|
||||
#include "pidgen3/PIDGEN3.h"
|
||||
|
||||
FNEXPORT BOOL LIBUMSKT_SET_DEBUG_OUTPUT(void *filestream)
|
||||
{
|
||||
char buffer[7];
|
||||
memcpy(buffer, filestream, 6);
|
||||
buffer[6] = 0;
|
||||
auto buff_string = std::string(buffer);
|
||||
std::map<UMSKT_TAG, UMSKT_Value> UMSKT::tags;
|
||||
|
||||
if (strcasecmp("STDOUT", &buffer[0]) != 0)
|
||||
extern "C"
|
||||
{
|
||||
|
||||
/**
|
||||
* Sets debug output to a given C++ File stream
|
||||
* if the memory allocated at filestream is "STDOUT" or "STDERR"
|
||||
* simply use the global vars allocated by *this* C++ runtime.
|
||||
* otherwise, assume that the input pointer is an ABI equivalent std::FILE
|
||||
*
|
||||
* @param char* or std::FILE "filestream"
|
||||
*/
|
||||
EXPORT BOOL UMSKT_SET_DEBUG_OUTPUT(void *filestream)
|
||||
{
|
||||
UMSKT::debug = stdout;
|
||||
return true;
|
||||
}
|
||||
else if (strcasecmp("STDERR", &buffer[0]) != 0)
|
||||
{
|
||||
UMSKT::debug = stderr;
|
||||
return true;
|
||||
}
|
||||
char buffer[7];
|
||||
memcpy(buffer, filestream, 6);
|
||||
buffer[6] = 0;
|
||||
auto buffstring = std::string(buffer);
|
||||
std::transform(buffstring.begin(), buffstring.end(), buffstring.begin(), ::tolower);
|
||||
|
||||
return false;
|
||||
}
|
||||
if (buffstring == "stdout")
|
||||
{
|
||||
UMSKT::debug = stdout;
|
||||
return true;
|
||||
}
|
||||
else if (buffstring == "stderr")
|
||||
{
|
||||
UMSKT::debug = stderr;
|
||||
return true;
|
||||
}
|
||||
else
|
||||
{
|
||||
UMSKT::debug = (std::FILE *)filestream;
|
||||
return true;
|
||||
}
|
||||
|
||||
// ---------------------------------------------
|
||||
|
||||
FNEXPORT void *CONFID_INIT()
|
||||
{
|
||||
auto cid = new ConfirmationID();
|
||||
|
||||
// cid->LoadHyperellipticCurve(0, 0, 0, 0, 0, 0, 0, 0, 0, false, false, 0);
|
||||
|
||||
return cid;
|
||||
}
|
||||
|
||||
FNEXPORT BYTE CONFID_GENERATE(void *cidIn, const char *installation_id_str, char *&confirmation_id, char *productid)
|
||||
{
|
||||
auto *cid((ConfirmationID *)cidIn);
|
||||
|
||||
std::string str, confid(confirmation_id), productids(productid);
|
||||
auto retval = cid->Generate(str, confid, productids);
|
||||
|
||||
return retval;
|
||||
}
|
||||
|
||||
FNEXPORT BYTE CONFID_END(void *cidIn)
|
||||
{
|
||||
auto *cid((ConfirmationID *)cidIn);
|
||||
delete cid;
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
// ---------------------------------------------
|
||||
|
||||
FNEXPORT void *PIDGEN3_BINK1998_INIT(const char *p, const char *a, const char *b, const char *generatorX,
|
||||
const char *generatorY, const char *publicKeyX, const char *publicKeyY,
|
||||
const char *genOrder, const char *privateKey)
|
||||
{
|
||||
|
||||
auto *bink1998 = new BINK1998();
|
||||
|
||||
bink1998->LoadEllipticCurve(p, a, b, generatorX, generatorY, publicKeyX, publicKeyY, genOrder, privateKey);
|
||||
|
||||
return bink1998;
|
||||
}
|
||||
|
||||
FNEXPORT void *PIDGEN3_BINK2002_INIT(const char *p, const char *a, const char *b, const char *generatorX,
|
||||
const char *generatorY, const char *publicKeyX, const char *publicKeyY,
|
||||
const char *genOrder, const char *privateKey, const char *authinfo)
|
||||
{
|
||||
|
||||
auto bink2002 = new BINK2002();
|
||||
|
||||
bink2002->LoadEllipticCurve(p, a, b, generatorX, generatorY, publicKeyX, publicKeyY, genOrder, privateKey);
|
||||
|
||||
return bink2002;
|
||||
}
|
||||
|
||||
FNEXPORT BOOL PIDGEN3_Generate(void *&ptrIn, char *&pKeyOut, int pKeySizeIn)
|
||||
{
|
||||
auto *p3((PIDGEN3 *)ptrIn);
|
||||
|
||||
std::string str;
|
||||
// BOOL retval = p3->Generate(str);
|
||||
|
||||
if (pKeySizeIn > str.length() + 1)
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
memcpy(pKeyOut, &str[0], str.length());
|
||||
pKeyOut[str.length()] = 0;
|
||||
// ---------------------------------------------
|
||||
|
||||
return true;
|
||||
/**
|
||||
*
|
||||
* @param tag
|
||||
* @param value
|
||||
* @param valueSize
|
||||
* @return success
|
||||
*/
|
||||
EXPORT BOOL UMSKT_SET_TAG(UMSKT_TAG tag, char *value, size_t valueSize)
|
||||
{
|
||||
if (valueSize > sizeof(UMSKT_Value))
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
// wipe/set the tag
|
||||
memset(&UMSKT::tags[tag], 0, sizeof(UMSKT_Value));
|
||||
memcpy(&UMSKT::tags[tag], value, valueSize);
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
EXPORT void UMSKT_RESET_TAGS()
|
||||
{
|
||||
UMSKT::tags.clear();
|
||||
}
|
||||
|
||||
// ---------------------------------------------
|
||||
|
||||
EXPORT void *CONFID_INIT()
|
||||
{
|
||||
auto cid = new ConfirmationID();
|
||||
|
||||
// cid->LoadHyperellipticCurve(0, 0, 0, 0, 0, 0, 0, 0, 0, false, false, 0);
|
||||
|
||||
return cid;
|
||||
}
|
||||
|
||||
EXPORT BYTE CONFID_GENERATE(void *cidIn, const char *installation_id_str, char *&confirmation_id, char *productid)
|
||||
{
|
||||
ConfirmationID *cid;
|
||||
try
|
||||
{
|
||||
cid = static_cast<ConfirmationID *>(cidIn);
|
||||
}
|
||||
catch (const std::bad_cast &e)
|
||||
{
|
||||
fmt::print(UMSKT::debug, "{}: input is not a {} - {}", __FUNCTION__, e.what());
|
||||
return -1;
|
||||
}
|
||||
|
||||
for (auto const i : UMSKT::tags)
|
||||
{
|
||||
switch (i.first)
|
||||
{
|
||||
case UMSKT_tag_InstallationID:
|
||||
break;
|
||||
case UMSKT_tag_ProductID:
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
std::string str, confid(confirmation_id), productids(productid);
|
||||
auto retval = cid->Generate(str, confid, productids);
|
||||
|
||||
return retval;
|
||||
}
|
||||
|
||||
EXPORT void CONFID_END(void *cidIn)
|
||||
{
|
||||
auto *cid((ConfirmationID *)cidIn);
|
||||
delete cid;
|
||||
cid = nullptr;
|
||||
cidIn = nullptr;
|
||||
}
|
||||
|
||||
// ---------------------------------------------
|
||||
|
||||
EXPORT void *PIDGEN3_INIT(const char *p, const char *a, const char *b, const char *generatorX,
|
||||
const char *generatorY, const char *publicKeyX, const char *publicKeyY,
|
||||
const char *genOrder, const char *privateKey)
|
||||
{
|
||||
PIDGEN3 *p3;
|
||||
|
||||
if (PIDGEN3::checkFieldStrIsBink1998(p))
|
||||
{
|
||||
p3 = new BINK1998();
|
||||
}
|
||||
else
|
||||
{
|
||||
p3 = new BINK2002();
|
||||
}
|
||||
|
||||
p3->LoadEllipticCurve(p, a, b, generatorX, generatorY, publicKeyX, publicKeyY, genOrder, privateKey);
|
||||
|
||||
return p3;
|
||||
}
|
||||
|
||||
EXPORT BOOL PIDGEN3_Generate(void *&ptrIn, char *&pKeyOut, int pKeySizeIn)
|
||||
{
|
||||
auto *p3((PIDGEN3 *)ptrIn);
|
||||
|
||||
for (auto const i : UMSKT::tags)
|
||||
{
|
||||
switch (i.first)
|
||||
{
|
||||
case UMSKT_tag_isUpgrade:
|
||||
p3->info.isUpgrade = i.second.boolean;
|
||||
break;
|
||||
case UMSKT_tag_ChannelID:
|
||||
p3->info.setChannelID(i.second.dword);
|
||||
break;
|
||||
case UMSKT_tag_Serial:
|
||||
p3->info.setSerial(i.second.dword);
|
||||
break;
|
||||
case UMSKT_tag_AuthData:
|
||||
p3->info.setAuthInfo(i.second.dword);
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
std::string str;
|
||||
BOOL retval = p3->Generate(str);
|
||||
|
||||
assert(pKeySizeIn >= str.length() + 1);
|
||||
|
||||
memcpy(pKeyOut, &str[0], str.length());
|
||||
pKeyOut[str.length()] = 0;
|
||||
|
||||
return retval;
|
||||
}
|
||||
|
||||
EXPORT BOOL PIDGEN3_Validate(void *&ptrIn, char *pKeyIn)
|
||||
{
|
||||
auto *p3((PIDGEN3 *)ptrIn);
|
||||
std::string str(pKeyIn);
|
||||
|
||||
BOOL retval = p3->Validate(str);
|
||||
|
||||
return retval;
|
||||
}
|
||||
|
||||
EXPORT void PIDGEN3_END(void *ptrIn)
|
||||
{
|
||||
auto *p3((PIDGEN3 *)ptrIn);
|
||||
delete p3;
|
||||
ptrIn = nullptr;
|
||||
p3 = nullptr;
|
||||
}
|
||||
|
||||
// ---------------------------------------------
|
||||
|
||||
EXPORT void *PIDGEN2_INIT()
|
||||
{
|
||||
auto p2 = new PIDGEN2();
|
||||
return p2;
|
||||
}
|
||||
|
||||
EXPORT BOOL PIDGEN2_GENERATE(void *ptrIn, char *&keyout)
|
||||
{
|
||||
auto p2 = (PIDGEN2 *)ptrIn;
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
EXPORT void PIDGEN2_END(void *ptrIn)
|
||||
{
|
||||
auto p2 = (PIDGEN2 *)ptrIn;
|
||||
delete p2;
|
||||
p2 = nullptr;
|
||||
ptrIn = nullptr;
|
||||
}
|
||||
|
||||
} // extern "C"
|
||||
|
||||
/**
|
||||
* Convert data between endianness types.
|
||||
*
|
||||
* @param data [in]
|
||||
* @param length [in]
|
||||
**/
|
||||
void UMSKT::endian(BYTE *data, int length)
|
||||
{
|
||||
for (int i = 0; i < length / 2; i++)
|
||||
{
|
||||
BYTE temp = data[i];
|
||||
data[i] = data[length - i - 1];
|
||||
data[length - i - 1] = temp;
|
||||
}
|
||||
}
|
||||
|
||||
FNEXPORT BOOL PIDGEN3_Verify(void *&ptrIn, char *pKeyIn)
|
||||
/**
|
||||
* Converts an OpenSSL BigNumber to it's Little Endian binary equivalent
|
||||
*
|
||||
* @param a [in] BigNumber to convert
|
||||
* @param to [out] char* binary representation
|
||||
* @param tolen [in] length of the char* array
|
||||
*
|
||||
* @return length of number in to
|
||||
**/
|
||||
int UMSKT::BN_bn2lebin(const BIGNUM *a, unsigned char *to, int tolen)
|
||||
{
|
||||
auto *p3((PIDGEN3 *)ptrIn);
|
||||
std::string str(pKeyIn);
|
||||
if (a == nullptr || to == nullptr)
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
|
||||
BOOL retval = p3->Verify(str);
|
||||
int len = BN_bn2bin(a, to);
|
||||
|
||||
return retval;
|
||||
}
|
||||
if (len > tolen)
|
||||
{
|
||||
return -1;
|
||||
}
|
||||
|
||||
FNEXPORT void PIDGEN3_END(void *ptrIn)
|
||||
{
|
||||
auto *p3((PIDGEN3 *)ptrIn);
|
||||
delete p3;
|
||||
}
|
||||
// Choke point inside BN_bn2lebinpad: OpenSSL uses len instead of tolen.
|
||||
endian(to, tolen);
|
||||
|
||||
// ---------------------------------------------
|
||||
|
||||
FNEXPORT BOOL PIDGEN2_INIT()
|
||||
{
|
||||
return true;
|
||||
}
|
||||
|
||||
FNEXPORT BOOL PIDGEN2_GENERATE()
|
||||
{
|
||||
return true;
|
||||
}
|
||||
|
||||
FNEXPORT BOOL PIDGEN2_END()
|
||||
{
|
||||
return true;
|
||||
}
|
||||
|
||||
FNEXPORT BOOL PIDGEN2_GenerateRetail(char *channelID, char *&keyout)
|
||||
{
|
||||
auto P2 = new PIDGEN2();
|
||||
BOOL retval = P2->GenerateRetail(channelID, keyout);
|
||||
delete P2;
|
||||
return retval;
|
||||
}
|
||||
|
||||
FNEXPORT BOOL PIDGEN2_GenerateOEM(char *year, char *day, char *oem, char *&keyout)
|
||||
{
|
||||
auto P2 = new PIDGEN2();
|
||||
BOOL retval = P2->GenerateOEM(year, day, oem, keyout);
|
||||
delete P2;
|
||||
return retval;
|
||||
return len;
|
||||
}
|
||||
Reference in New Issue
Block a user