up

Author: kebiao

Plugins/kbengine_ue4_plugins/Source/KBEnginePlugins/Engine/ClientSDKUpdateUI.cpp

@@ -0,0 +1,62 @@
+#include "ClientSDKUpdateUI.h"
+#include "Engine.h"
+#include "KBEvent.h"
+#include "KBEngine.h"
+#include "KBEventTypes.h"
+#include "SlateOptMacros.h"
+
+BEGIN_SLATE_FUNCTION_BUILD_OPTIMIZATION
+void SClientSDKUpdateUI::Construct(const FArguments& args)
+{
+	FTextBlockStyle textStyle = FCoreStyle::Get().GetWidgetStyle< FTextBlockStyle >("NormalText");
+	textStyle.SetFont(FSlateFontInfo("Veranda", 26));
+	textStyle.SetColorAndOpacity(FLinearColor::White);
+
+	ChildSlot
+		[
+			SNew(SOverlay)
+			/*+ SOverlay::Slot()
+		.HAlign(HAlign_Center)
+		.VAlign(VAlign_Top)
+		[
+			SNew(STextBlock)
+			.ColorAndOpacity(FLinearColor::White)
+		.ShadowColorAndOpacity(FLinearColor::Black)
+		.ShadowOffset(FIntPoint(-1, 1))
+		.Font(FSlateFontInfo("Arial", 26))
+		.Text(FText::FromString("Main Menu"))
+		]*/
+	+ SOverlay::Slot()
+		.HAlign(HAlign_Center)
+		.VAlign(VAlign_Center)
+		[
+			SNew(SVerticalBox)
+			+ SVerticalBox::Slot()
+			.Padding(20)
+			[
+				SNew(SButton)
+				.Text(FText::FromString("Update SDK"))
+				.TextStyle(&textStyle)
+				.ContentPadding(FMargin(30.0, 15.0))
+				.ButtonColorAndOpacity(FLinearColor::Green)
+				.HAlign(HAlign_Center)
+				.VAlign(VAlign_Top)
+				.OnClicked(this, &SClientSDKUpdateUI::UpdateSDKClicked)
+			]
+		]
+		];
+}
+END_SLATE_FUNCTION_BUILD_OPTIMIZATION
+
+FReply SClientSDKUpdateUI::UpdateSDKClicked()
+{
+	if (GEngine)
+	{
+		GEngine->AddOnScreenDebugMessage(-1, 3.f, FColor::Yellow, TEXT("Update SDK!"));
+	}
+
+	UKBEventData_onDownloadSDK* pEventData = NewObject<UKBEventData_onDownloadSDK>();
+	pEventData->isDownload = true;
+	KBENGINE_EVENT_FIRE(KBEventTypes::onDownloadSDK, pEventData);
+	return FReply::Handled();
+}

Plugins/kbengine_ue4_plugins/Source/KBEnginePlugins/Engine/ClientSDKUpdateUI.h

@@ -0,0 +1,19 @@
+#pragma once
+
+#include "KBEnginePluginsPrivatePCH.h"
+
+
+class KBENGINEPLUGINS_API SClientSDKUpdateUI : public SCompoundWidget
+{
+
+public:
+	SLATE_BEGIN_ARGS(SClientSDKUpdateUI){}
+	
+	SLATE_END_ARGS()
+
+	void Construct(const FArguments& args);
+
+	FReply UpdateSDKClicked();
+
+
+};

Plugins/kbengine_ue4_plugins/Source/KBEnginePlugins/Engine/EncryptionFilter.cpp

@@ -5,35 +5,37 @@
 #include "PacketSenderKCP.h"
 #include "Engine/KBDebug.h"
 
-#ifndef KBENGINE_NO_CRYPTO
-#include "cryptlib.h"
-#include "rdrand.h"
-#include "secblock.h"
-#endif
+EncryptionFilter::~EncryptionFilter()
+{
+}
 
-BlowfishFilter::BlowfishFilter(int keySize):
+BlowfishFilter::BlowfishFilter(int keySize) :
 	isGood_(false),
 	pPacket_(new MemoryStream()),
 	pEncryptStream_(new MemoryStream()),
 	packetLen_(0),
-	padSize_(0)
+	padSize_(0),
+	key_(),
+	keySize_(0),
+	pBlowFishKey_(NULL)
 {
-#ifndef KBENGINE_NO_CRYPTO
-	key_.Init(0, keySize);
+	unsigned char buf[20] = "";
+	RAND_bytes(buf, 20);
+	key_ = (char *)buf;
+	keySize_ = key_.Len();
 
-	CryptoPP::RDRAND rng;
-	rng.GenerateBlock(key_.GetData(), key_.Num());
 	init();
-#endif
 }
 
-BlowfishFilter::BlowfishFilter(const TArray<uint8>& key):
+BlowfishFilter::BlowfishFilter(const FString & key) :
 	isGood_(false),
-	key_(key),
 	pPacket_(new MemoryStream()),
 	pEncryptStream_(new MemoryStream()),
 	packetLen_(0),
-	padSize_(0)
+	padSize_(0),
+	key_(key),
+	keySize_(key_.Len()),
+	pBlowFishKey_(NULL)
 {
 	init();
 }
@@ -42,31 +44,32 @@ BlowfishFilter::~BlowfishFilter()
 {
 	KBE_SAFE_RELEASE(pPacket_);
 	KBE_SAFE_RELEASE(pEncryptStream_);
+	KBE_SAFE_RELEASE(pBlowFishKey_);
 }
 
 bool BlowfishFilter::init()
 {
-#ifndef KBENGINE_NO_CRYPTO
-	if (key_.Num() >= encripter.MinKeyLength() && key_.Num() <= encripter.MaxKeyLength())
+	pBlowFishKey_ = new BF_KEY;
+
+	if (MIN_KEY_SIZE <= keySize_ && keySize_ <= MAX_KEY_SIZE)
 	{
-		encripter.SetKey(key_.GetData(), key_.Num());
-		decripter.SetKey(key_.GetData(), key_.Num());
+
+		BF_set_key(this->pBlowFishKey(), key_.Len(), reinterpret_cast<const unsigned char*>(TCHAR_TO_ANSI(*key_)));
 		isGood_ = true;
 	}
 	else
 	{
-		ERROR_MSG("BlowfishFilter::init: invalid length %d", key_.Num());
+		ERROR_MSG("BlowfishFilter::init: invalid length %d", key_.Len());
 		isGood_ = false;
 	}
-#endif
+
 	return isGood_;
 }
 
 void BlowfishFilter::encrypt(MemoryStream *pMemoryStream)
 {
 	// BlowFish 每次只能加密和解密8字节数据
 	// 不足8字节则填充0
-#ifndef KBENGINE_NO_CRYPTO
 	uint8 padSize = 0;
 
 	if (pMemoryStream->length() % BLOCK_SIZE != 0)
@@ -86,12 +89,10 @@ void BlowfishFilter::encrypt(MemoryStream *pMemoryStream)
 
 	pMemoryStream->swap(*pEncryptStream_);
 	pEncryptStream_->clear(false);
-#endif
 }
 
 void BlowfishFilter::encrypt(uint8 *buf, MessageLengthEx len)
 {
-#ifndef KBENGINE_NO_CRYPTO
 	if (len % BLOCK_SIZE != 0)
 	{
 		ERROR_MSG("BlowfishFilter::encrypt: Input length (%d) is not a multiple of block size ", len);
@@ -113,9 +114,8 @@ void BlowfishFilter::encrypt(uint8 *buf, MessageLengthEx len)
 			prevBlock = *(uint64*)(data + i);
 		}
 
-		encripter.ProcessData(data + i, data + i, BLOCK_SIZE);
+		BF_ecb_encrypt(data + i, data + i, this->pBlowFishKey(), BF_ENCRYPT);
 	}
-#endif
 }
 
 void BlowfishFilter::encrypt(uint8 *buf, MessageLengthEx offset, MessageLengthEx len)
@@ -130,7 +130,6 @@ void BlowfishFilter::decrypt(MemoryStream *pMemoryStream)
 
 void BlowfishFilter::decrypt(uint8 *buf, MessageLengthEx len)
 {
-#ifndef KBENGINE_NO_CRYPTO
 	if (len % BLOCK_SIZE != 0)
 	{
 		ERROR_MSG("BlowfishFilter::decrypt: Input length (%d) is not a multiple of block size ", len);
@@ -141,16 +140,15 @@ void BlowfishFilter::decrypt(uint8 *buf, MessageLengthEx len)
 	uint64 prevBlock = 0;
 	for (uint32 i = 0; i < len; i += BLOCK_SIZE)
 	{
-		decripter.ProcessData(data + i, data + i, BLOCK_SIZE);
+		BF_ecb_encrypt(data + i, data + i, this->pBlowFishKey(), BF_DECRYPT);
 
 		if (prevBlock != 0)
 		{
 			*(uint64*)(data + i) = *(uint64*)(data + i) ^ (prevBlock);
 		}
-		
+
 		prevBlock = *(uint64*)(data + i);
 	}
-#endif
 }
 
 void BlowfishFilter::decrypt(uint8 *buf, MessageLengthEx offset, MessageLengthEx len)
@@ -160,21 +158,18 @@ void BlowfishFilter::decrypt(uint8 *buf, MessageLengthEx offset, MessageLengthEx
 
 bool BlowfishFilter::send(PacketSenderBase* pPacketSender, MemoryStream *pPacket)
 {
-#ifndef KBENGINE_NO_CRYPTO
 	if (!isGood_)
 	{
 		ERROR_MSG("BlowfishFilter::send: Dropping packet due to invalid filter");
 		return false;
 	}
 
 	encrypt(pPacket);
-#endif
 	return pPacketSender->send(pPacket);;
 }
 
 bool BlowfishFilter::recv(MessageReader* pMessageReader, MemoryStream *pPacket)
 {
-#ifndef KBENGINE_NO_CRYPTO
 	if (!isGood_)
 	{
 		ERROR_MSG("BlowfishFilter::recv: Dropping packet due to invalid filter");
@@ -185,7 +180,7 @@ bool BlowfishFilter::recv(MessageReader* pMessageReader, MemoryStream *pPacket)
 	uint32 len = pPacket->length();
 	uint16 packeLen = pPacket->readUint16();
 
-	if ( 0 == pPacket_->length() && len > MIN_PACKET_SIZE && packeLen - 1 == len - 3)
+	if (0 == pPacket_->length() && len > MIN_PACKET_SIZE && packeLen - 1 == len - 3)
 	{
 		int padSize = pPacket->readUint8();
 		decrypt(pPacket);
@@ -267,12 +262,6 @@ bool BlowfishFilter::recv(MessageReader* pMessageReader, MemoryStream *pPacket)
 		packetLen_ = 0;
 		padSize_ = 0;
 	}
-#else
-	if (pMessageReader)
-	{
-		pMessageReader->process(pPacket->data() + pPacket->rpos(), 0, pPacket->length());
-	}
-#endif
-	
 	return true;
 }
+

Plugins/kbengine_ue4_plugins/Source/KBEnginePlugins/Engine/EncryptionFilter.h

@@ -1,14 +1,21 @@
 #pragma once
 #include "KBECommon.h"
 
-#ifndef KBENGINE_NO_CRYPTO
-// https://stackoverflow.com/questions/51416259/unreal-engine-4-20-build-error-in-plugin-adaptive-unity-build-disabling-pch-f
-#pragma warning(disable:4668)   // x  is not defined as a preprocessor macro, replacing with '0' for '#if/#elif'
-#include "blowfish.h"
-#pragma warning(default:4668)  
-#include "modes.h"
+#if PLATFORM_WINDOWS
+#include "WindowsHWrapper.h"
+#include "AllowWindowsPlatformTypes.h"
 #endif
 
+#define UI UI_ST
+THIRD_PARTY_INCLUDES_START
+#include "openssl/rand.h"
+#include "openssl/blowfish.h"
+THIRD_PARTY_INCLUDES_END
+#undef UI
+
+#if PLATFORM_WINDOWS
+#include "HideWindowsPlatformTypes.h"
+#endif
 
 class MemoryStream;
 class PacketSenderBase;
@@ -17,8 +24,9 @@ class MessageReader;
 class EncryptionFilter
 {
 public:
-	virtual ~EncryptionFilter() {}
 
+	EncryptionFilter() {}
+	virtual ~EncryptionFilter();
 	virtual void encrypt(MemoryStream *pMemoryStream) = 0;
 	virtual void encrypt(uint8 *buf, MessageLengthEx len) = 0;
 	virtual void encrypt(uint8 *buf, MessageLengthEx offset, MessageLengthEx len) = 0;
@@ -39,11 +47,20 @@ class BlowfishFilter : public EncryptionFilter
 	static const uint32 BLOCK_SIZE = 64 / 8;
 	static const uint32 MIN_PACKET_SIZE = (sizeof(MessageLength) + 1 + BLOCK_SIZE);
 
-	BlowfishFilter(int keySize=16);
-	BlowfishFilter(const TArray<uint8>& key);
+
+	// key的最小和最大大小
+	static const int MIN_KEY_SIZE = 32 / 8;
+	static const int MAX_KEY_SIZE = 448 / 8;
+
+	// 默认key的大小
+	static const int DEFAULT_KEY_SIZE = 128 / 8;
+
+	BlowfishFilter(const FString & key);
+	BlowfishFilter(int keySize = DEFAULT_KEY_SIZE);
+
 	virtual ~BlowfishFilter();
 
-	virtual void encrypt(MemoryStream *pMemoryStream) ;
+	virtual void encrypt(MemoryStream *pMemoryStream);
 	virtual void encrypt(uint8 *buf, MessageLengthEx len);
 	virtual void encrypt(uint8 *buf, MessageLengthEx offset, MessageLengthEx len);
 
@@ -54,22 +71,29 @@ class BlowfishFilter : public EncryptionFilter
 	virtual bool send(PacketSenderBase *pPacketSender, MemoryStream *pPacket);
 	virtual bool recv(MessageReader *pMessageReader, MemoryStream *pPacket);
 
-	TArray<uint8>& key() {
-		return key_;
+
+	BF_KEY * pBlowFishKey() { return (BF_KEY*)pBlowFishKey_; }
+
+	TArray<uint8> key() 
+	{
+		TArray<uint8> keyArray;
+		keyArray.SetNum(key_.Len());
+		memcpy(keyArray.GetData(), TCHAR_TO_ANSI(*key_), key_.Len());
+
+		return keyArray;
 	}
 
 private:
 	bool init();
 
 private:
 	bool			isGood_;
-	TArray<uint8>	key_;
 	MemoryStream*	pPacket_;
 	MemoryStream*	pEncryptStream_;
 	MessageLength	packetLen_;
 	uint8			padSize_;
-#ifndef KBENGINE_NO_CRYPTO
-	CryptoPP::ECB_Mode<CryptoPP::Blowfish>::Encryption encripter;
-	CryptoPP::ECB_Mode<CryptoPP::Blowfish>::Decryption decripter;
-#endif
+
+	FString key_;
+	int keySize_;
+	void * pBlowFishKey_;
 };