main.c

#define  _CRT_SECURE_NO_WARNINGS
#include "rlmpubkey.h"
#include <Windows.h>
#include "resource.h"

int replacepubkey(PubkeyInfo* pki);
int createsign(PubkeyInfo* pki, char* rlmsign);
PubkeyInfo* init();
int listFiles(char* dir, PubkeyInfo* pki, const char* isv_name);

int help() 
{
	printf("Usage Instructions\n");
	printf("In RLM 12 new version, RLM_ISV_NAME parameter is hidden, need to find it through license file.\n");
	
	printf("RLMpubkey.exe ISV\n");
	printf("Example: For a license file content as below, ISV is demo, command is \"RLMpubkey.exe demo\".\n");
	printf("HOST THIS_HOST 68f7283e08b7 \nISV demo \nLICENSE demo f1 15.1 permanent 1 hostid = 68f7283e08b7 ...\n");
	return 0;
}

int freelist(PubkeyInfo* pki)
{
	PubkeyInfo* tmp =pki->next,*tmq;
	for (; tmp != NULL;)
	{
		tmq = tmp;
		tmp = tmp->next;
		free(tmq);
	}
	return 1;
}

int main(int argc, char* argv[])
{
	// Handle command line arguments
	const char* isv_name = NULL;
	if (argc >= 2) {
		isv_name = argv[1];
		printf("Using ISV name from command line: %s\n", isv_name);
	} else {
		printf("No ISV name provided in command line.\n");
		printf("Usage: %s <ISV_NAME>\n", argv[0]);
		help();
		return 0;
	}

	PubkeyInfo* pfirst = init(),*ptmp;

	/* Search for public key files that need to be replaced */
	listFiles(localdir, pfirst, isv_name);
	if (pfirst->next ==NULL)
	{
		printf("\nNo files need to be replaced, program exit.\n");
		freelist(pfirst);
		return 0;
	}
	listpubkey(pfirst);

	/* RC resource loading */
	HRSRC hRsrc = FindResource(NULL, MAKEINTRESOURCE(IDR_EXE1), TEXT("EXE"));
	DWORD dwSize = SizeofResource(NULL, hRsrc);
	HGLOBAL hGlobal = LoadResource(NULL, hRsrc);
	LPVOID pBuffer = LockResource(hGlobal);

	/* Generate keygen */
	if (!createsign(pfirst, (unsigned char*)pBuffer)) {
        printf("Failed to create signature.\n");
		FreeResource(hGlobal);
		freelist(pfirst);
        return 0;
    }

	/* Replace file public key */
	replacepubkey(pfirst);

	/* Release resources */
	FreeResource(hGlobal);
	freelist(pfirst);

	return 1;
}