【加密】Cocos2d-x PNG图片资源加密（修改版）

【说明】

这篇文章是对上一篇
【Cocos2d-x PNG图片资源加密】的补充和扩展，那篇文章转自【旧时尘安】的博客，文中已经对原理和使用讲解的很清晰，这里只是根据我自己的使用情况做一些小小的功能扩展，也是自己做个整理，以便日后使用。如有侵权，请联系删除。

【链接】

原文地址：http://www.cnblogs.com/zhangpanyi/p/4560297.html

原始工程：https://github.com/zhangpanyi/EncryptPNG

【使用】

修改后的使用有所调整，原文的使用更简洁，这里主要是依照我个人的习惯作出的调整。我在代码中添加了设置密钥和扩展名的接口。

1. 在 cocos 目录下新建文件夹 ext ，将 CCAES.cpp、CCAES.h、CCDecryptImage.cpp、CCDecryptImage.h 拷贝到其中。

2. 在Xcode项目中引用 ext 目录。

注：这里可能会报错，为此我耽误了半天，结果居然是Xcode没有自动引用导致的，我已记录到
【这里】。

3. Android项目需要修改 cocos/Android.mk 文件，将两个cpp文件添加进去即可。

4. 在 CCImage 中调用解密代码，方法与原文一样，这里略有修改，代码见附录。

注：我将对 CCDecryptImage.h 的引用放在了 CCImage.h ，是为了在项目中设置密钥时不需要再引用此头文件。

5. 在项目中设置密钥和扩展名，我是在 AppDelegate.cpp 中设置。

注：如果修改扩展名，需要加密端也做修改，保证两边扩展名一致。

// 设置密钥
const ext::aes_key key = {0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37, 0x38, 0x39};
ext::DecryptImageConfig(key, ".epng");

【代码】

CCImage.cpp

bool Image::initWithImageFile(const std::string& path)
{
......

if (!data.isNull())
{
// 图片文件解密
if (ext::AnalyzeExtension(path)[1] == ext::TARGET_EXTENSION)
{
auto image_data = ext::DecryptImage(path, data);
ret = initWithImageData(&image_data[0], image_data.size());
}
else
{
ret = initWithImageData(data.getBytes(), data.getSize());
}
}
#endif // EMSCRIPTEN

return ret;
}

bool Image::initWithImageFileThreadSafe(const std::string& fullpath)
{
......

if (!data.isNull())
{
// 图片文件解密
if (ext::AnalyzeExtension(fullpath)[1] == ext::TARGET_EXTENSION)
{
auto image_data = ext::DecryptImage(fullpath, data);
ret = initWithImageData(&image_data[0], image_data.size());
}
else
{
ret = initWithImageData(data.getBytes(), data.getSize());
}
}

return ret;
}

<span style="font-family: Arial, Helvetica, sans-serif;">CCDecryptImage.h</span>

#ifndef __CC_DECRYPT_IMAGE_H__
#define __CC_DECRYPT_IMAGE_H__

#include <array>
#include <vector>
#include "CCData.h"
#include "CCAES.h"

namespace ext
{
/* 解密扩展名 */
static std::string TARGET_EXTENSION = ".epng";

/* 解密密钥 */
static aes_key SECRET_KEY = { 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37, 0x38, 0x39 };

/**
* 设置解密秘钥
* @param key   秘钥（AES的16位秘钥）
* @param exten 文件扩展名（需要解密的文件扩展名，如".png"）
*/
void DecryptImageConfig(const aes_key &key, const std::string &exten = TARGET_EXTENSION);

/**
* 解密图片文件（在CCImage中调用）
* @param filename 文件名称
* @param data 文件数据
*/
std::vector<unsigned char> DecryptImage(const std::string &filename, cocos2d::Data &data);

/**
* 分解文件名的扩展名（在CCImage中调用）
* @param file_path 文件名
*/
std::array<std::string, 2> AnalyzeExtension(const std::string &file_path);

}

#endif

CCDecryptImage.cpp

#include "CCDecryptImage.h"

#include <sstream>
#include "ccMacros.h"

#if (CC_TARGET_PLATFORM == CC_PLATFORM_WIN32)
#include <WinSock.h>
#pragma comment(lib, "ws2_32.lib")
#elif (CC_TARGET_PLATFORM == CC_PLATFORM_IOS || CC_TARGET_PLATFORM == CC_PLATFORM_ANDROID)
#include <netinet/in.h>
#endif

namespace ext
{
/* CRC码长度 */
static const uint32_t CRC_SIZE = 4;

/* 文件头部 */
static const unsigned char HEAD_DATA[] = { 0x89, 0x50, 0x4e, 0x47, 0x0d, 0x0a, 0x1a, 0x0a };

/* IEND CRC码 */
static const unsigned char IEND_DATA[] = { 0xae, 0x42, 0x60, 0x82 };

/* 数据块头部（用于验证解密是否成功） */
static const unsigned char BLOCK_HEAD[] = { 0x45, 0x6e, 0x63, 0x72, 0x79, 0x70, 0x74, 0x50, 0x4e, 0x47 };

#pragma pack(push, 1)

struct Block
{
char name[4];
uint32_t pos;
uint32_t size;
};

struct IHDRBlock
{
Block block;
char data[13 + CRC_SIZE];
};

#pragma pack(pop)

/* 解析文件扩展名 */
std::array<std::string, 2> AnalyzeExtension(const std::string &file_path)
{
std::string::size_type pos = file_path.rfind('.');
std::array<std::string, 2> text;
if (std::string::npos != pos)
{
text[1] = file_path.substr(pos);
text[0] = file_path.substr(0, pos);
}
else
{
text[0] = file_path;
}
return text;
}

template <int _Value, typename _Stream>
std::array<char, _Value> ReadSome(_Stream &stream)
{
std::array<char, _Value> buffer;
for (unsigned int i = 0; i < _Value; ++i) buffer[i] = stream.get();
return buffer;
}

/* 解密块 */
void DecryptBlock(std::stringstream &ss, const aes_key &key)
{
const std::streamoff contents_size = ss.tellp() - ss.tellg();
const uint32_t block_size = (uint32_t)(contents_size + AES_BLOCK_SIZE - contents_size % AES_BLOCK_SIZE);
std::vector<uint8_t> buffer;
buffer.resize(block_size);
for (uint32_t i = 0; i < contents_size; ++i) buffer[i] = ss.get();
AES::DecryptData(&buffer[0], block_size, key);
ss.seekg(0); ss.seekp(0);
for (uint32_t i = 0; i < block_size; ++i) ss.put(buffer[i]);
}

/* 解密图片文件 */
std::vector<unsigned char> DecryptImage(const std::string &filename, cocos2d::Data &data)
{
CCAssert(!data.isNull(), "data is null!");

// 获取数据块信息位置
const uint32_t block_start_pos = ntohl(*reinterpret_cast<uint32_t *>(data.getBytes() + data.getSize() - sizeof(uint32_t)));

// 获取数据块信息
std::stringstream block_info;
for (uint32_t i = block_start_pos; i < data.getSize() - sizeof(uint32_t); ++i)
{
block_info.put(*(data.getBytes() + i));
}

// 解密数据块信息
DecryptBlock(block_info, SECRET_KEY);

// 验证数据块信息是否解密成功
auto block_head = ReadSome<sizeof(BLOCK_HEAD)>(block_info);
for (unsigned int i = 0; i < block_head.size(); ++i)
{
if (block_head[i] != BLOCK_HEAD[i])
{
CCAssert(false, "the key is wrong!");
}
}

// 写入文件头信息
std::vector<unsigned char> image_data;
image_data.reserve(data.getSize());
for (auto ch : HEAD_DATA) image_data.push_back(ch);

// 写入数据块信息
while (true)
{
Block block;
memcpy(&block, &(ReadSome<sizeof(Block)>(block_info)[0]), sizeof(Block));
if (block_info.eof())
{
CCAssert(false, "");
CCLOG("the %s file format error!", filename.c_str());
}

// 写入数据块长度和名称
char size_buffer[sizeof(block.size)];
memcpy(size_buffer, &block.size, sizeof(size_buffer));
for (auto ch : size_buffer) image_data.push_back(ch);
for (auto ch : block.name) image_data.push_back(ch);

block.pos = ntohl(block.pos);
block.size = ntohl(block.size);

char block_name[sizeof(block.name) + 1] = { 0 };
memcpy(block_name, block.name, sizeof(block.name));
if (strcmp(block_name, "IHDR") == 0)
{
IHDRBlock ihdr;
memcpy(&ihdr, &block, sizeof(Block));
memcpy(((char *)&ihdr) + sizeof(Block), &ReadSome<sizeof(IHDRBlock) - sizeof(Block)>(block_info)[0], sizeof(IHDRBlock) - sizeof(Block));
for (auto ch : ihdr.data) image_data.push_back(ch);
}
else if (strcmp(block_name, "IEND") == 0)
{
for (auto ch : IEND_DATA) image_data.push_back(ch);
CCLOG("decrypt %s success!", filename.c_str());
break;
}
else
{
for (uint32_t i = 0; i < block.size + CRC_SIZE; ++i)
{
image_data.push_back(*(data.getBytes() + block.pos + i));
}
}
}
return image_data;
}

/* 配置解密秘钥和扩展名 */
void DecryptImageConfig(const aes_key &key, const std::string &exten)
{
SECRET_KEY = key;
TARGET_EXTENSION = exten;
}
}

【附录】

修改版：http://pan.baidu.com/s/1bnCWEyN