轻量级加密lua

     首先是压缩lua文件，代码如下：     

[cpp] view plaincopy

1. int write_file_content(const char* folder)  
2. {  
3.     //获得文件数据，并压缩文件  
4.     FILE* fpin = fopen(folder, "wb+");  
5.     if (fpin == NULL)  
6.     {  
7.         printf("无法读取文件: %s\n", folder);  
8.         return 0;  
9.     }  
10.   
11.     //得到文件大小  
12.     fseek(fpin, 0, SEEK_END);  
13.     unsigned int size = ftell(fpin);  
14.   
15.     //读出文件内容  
16.     fseek(fpin, 0, SEEK_SET);  
17.     void* con = malloc(size);  
18.     int r = fread(con, size, 1, fpin);    
19.   
20.     //进行加密操作  
21.     unsigned long zip_con_size = size * 2;  
22.     void* zip_con = malloc(zip_con_size);  
23.     if (Z_OK != compress((Bytef*)zip_con, &zip_con_size, (Bytef*)con, size))   
24.     {  
25.         printf("压缩 %s 时发生错误\n",folder);  
26.     }  
27.     printf("%s 压缩前大小：%ld 压缩后大小：%ld\n", folder,  size,  zip_con_size);  
28.   
29.     //写文件内容  
30.     fseek(fpin, 0, SEEK_SET);  
31.     int len = fwrite(zip_con, zip_con_size, 1, fpin);  
32.   
33.     //释放资源  
34.     fclose(fpin);  
35.     free(zip_con);  
36.     free(con);    
37.     return 0;  
38. }  

       然后是解密操作，代码如下：

[cpp] view plaincopy

1. void* read_file_content(const char* folder, int& bufflen)  
2. {  
3.     FILE* file = fopen(folder, "wb+");  
4.     if (file)  
5.     {     
6.     {  
7.         printf("无法读取文件: %s\n", folder);  
8.         return 0;  
9.     }  
10.   
11.     //获取文件大小  
12.     fseek(file, 0, SEEK_END);  
13.     unsigned int size = ftell(file);  
14.   
15.     //读出文件内容  
16.     void* con = malloc(size);  
17.     fseek(file, 0, SEEK_SET);  
18.     int len = fread(con, size, 1, file);  
19.   
20.     //解压缩操作  
21.     unsigned long zip_size = size * 4;  
22.     void* zip_con = malloc(zip_size);  
23.     int code = uncompress((Bytef*)zip_con, &zip_size, (Bytef*)con, size);  
24.     if (Z_OK != code)  
25.     {  
26.         printf("解压 %s 时发生错误 :%d\n", folder, code);  
27.         return 0;  
28.     }  
29.   
30.     //释放资源  
31.     fclose(file);  
32.     free(con);  
33.   
34.     //zip_con由外部释放  
35.     bufflen = zip_size;  
36.     return zip_con；  
37. }  

cocos2d-x-lua工程的lua脚本加密

2014/1/26 更新

最近又发现了一个很简单的方法，其实coco2dx已经给我们提供设置loader的方法。

注意：有个局限性，在非android平台下调用pEngine->executeScriptFile是不调用loader的，只有require这种才会调用loader。也就是说你直接executeScriptFile("main.lua")这个脚本不能加密，main.lua里面require的才能加密

 

步骤如下：

1、实现自己的loader(参考int cocos2dx_lua_loader(lua_State *L))

 

#define CODE_MASK 250extern "C"{    int decode_lua_loader(lua_State *L)    {        std::string filename(luaL_checkstring(L, 1));        size_t pos = filename.rfind(".lua");        if (pos != std::string::npos)        {            filename = filename.substr(0, pos);        }                pos = filename.find_first_of(".");        while (pos != std::string::npos)        {            filename.replace(pos, 1, "/");            pos = filename.find_first_of(".");        }        filename.append(".lua");                unsigned long codeBufferSize = 0;        unsigned char* codeBuffer = CCFileUtils::sharedFileUtils()->getFileData(filename.c_str(), "rb", &codeBufferSize);                //-------------decode here        for(int i=0; i<codeBufferSize; i++)        {            codeBuffer[i] ^= CODE_MASK; //xor decode        }        //-------------        if (codeBuffer)        {            if (luaL_loadbuffer(L, (char*)codeBuffer, codeBufferSize, filename.c_str()) != 0)            {                luaL_error(L, "error loading module %s from file %s :\n\t%s",                    lua_tostring(L, 1), filename.c_str(), lua_tostring(L, -1));            }            delete []codeBuffer;        }        else        {            CCLog("can not get file data of %s", filename.c_str());        }                return 1;    }}

 

 

 

2、添加loader

CCScriptEngineManager::sharedManager()->setScriptEngine(pEngine);pEngine->addLuaLoader(decode_lua_loader);//add this

 

我在win32上测试通过

 

----------------------------------------------------------------------------------------分割线----------------------------------------------------------------------------------------

 

发了这篇博客之后好多同学联系我，说感觉还是不太明白，那我再说清楚一些

2014/1/15 更新

转载请注明出处http://www.cnblogs.com/mrblue/admin/EditPosts.aspx?postid=3122543

 

前言部分

1、本文是以cocos2d-x-lua项目为例说明的。

2、我这里只说下我自己遇到的问题和解决方法，当大家也遇到时可以参考下。

3、我所使用的cocos2d-x版本是cocos2d-2.1rc0-x-2.1.3，这个版本使用的是luajit，而不是源生的lua，luajit有很多好处。

------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------

方法一：

这种其实并不是真正意义上的加密，而是用luajit把lua脚本编译成字节码(我之前实验过用luac编译出的字节码不能被luajit执行)。

1）编译luajit，这里并不是编译luajit的库，而是编译luajit的控制台程序，用作后面把lua脚本代码文件编译成字节码用。

　　cd到 coco2dx安装目录/scripting/lua/luajit/LuaJIT-2.0.1

2) 输入make(如果控台显示command not found，那就要先安装make，Mac作为UNIX系却没有提供make，需要打开xCode->xCode(菜单栏)->Open Developer Tool->More Developer Tools->注册/登陆AppID->选择你对应的Command Line Tool->下载安装)

　　这时会在src文件夹下生成可执行文件luajit（这b玩意我找了好久，根本找不到在哪生成的）

3) 有了luajit这个可执行文件，我们就可以拿它编译lua脚本了。

　　这时cd进src目录

　　输入./luajit -b hello1.lua hello1.out

　　这个hello1.lua应该是你要编译的源文件，必要时请带上路径。

　　注意：你不能把luajit这个可执行文件单独拿出来到别处去运行，因为在src里有些它需要依赖的东西，如果你想把luajit移地方，那你把(当前已经在src文件夹了)luajit、jit文件夹、好几个.dasc文件打个包，这样你就可以到其他的地方运行luajit了。

4）把这个hello1.out加到你工程里，就可以直接当成普通的脚本一样运行。

【严重注意】：如果你hello1.lua里面写了require "hello2"那你最好把hello2.lua的生成的文件也命名成hello2.lua，否则再执行hello1.lua的时候就找不到依赖的hello2.lua。当然你也可以require的文件写带后缀名的文件，但这我没有试验过行不行。

方法二：

方法二的前提是你不用luajit，而使用lua。

这样这种方法是真正加密，原理是我们自己替换掉lua的的loader函数。

其实这种方法是因为我之前以为coco2dx用的lua，我在调试lua代码的时候发现的，当然也受到别的大神的启发。

先贴出一段代码

LUALIB_API int luaL_loadfile (lua_State *L, const char *filename) {  LoadF lf;  int status, readstatus;  int c;  int fnameindex = lua_gettop(L) + 1;  /* index of filename on the stack */  lf.extraline = 0;  if (filename == NULL) {    lua_pushliteral(L, "=stdin");    lf.f = stdin;  }  else {    lua_pushfstring(L, "@%s", filename);    lf.f = fopen(filename, "r");    if (lf.f == NULL) return errfile(L, "open", fnameindex);  }  c = getc(lf.f);  if (c == '#') {  /* Unix exec. file? */    lf.extraline = 1;    while ((c = getc(lf.f)) != EOF && c != '\n') ;  /* skip first line */    if (c == '\n') c = getc(lf.f);  }  if (c == LUA_SIGNATURE[0] && filename) {  /* binary file? */    lf.f = freopen(filename, "rb", lf.f);  /* reopen in binary mode */    if (lf.f == NULL) return errfile(L, "reopen", fnameindex);    /* skip eventual `#!...' */   while ((c = getc(lf.f)) != EOF && c != LUA_SIGNATURE[0]) ;    lf.extraline = 0;  }  ungetc(c, lf.f);  status = lua_load(L, getF, &lf, lua_tostring(L, -1));  readstatus = ferror(lf.f);  if (filename) fclose(lf.f);  /* close file (even in case of errors) */  if (readstatus) {    lua_settop(L, fnameindex);  /* ignore results from `lua_load' */    return errfile(L, "read", fnameindex);  }  lua_remove(L, fnameindex);  return status;}

 

每当要加载新的lua文件时都会调用luaL_loadfile，当走到status = lua_load(L, getF, &lf, lua_tostring(L, -1))时就说明要从文件读取内容了，注意这里穿了一个参数getF，它其实是个函数指针，我们把这个贴出来

static const char *getF (lua_State *L, void *ud, size_t *size) {  LoadF *lf = (LoadF *)ud;  (void)L;  if (lf->extraline) {    lf->extraline = 0;    *size = 1;    return "\n";  }  if (feof(lf->f)) return NULL;  *size = fread(lf->buff, 1, sizeof(lf->buff), lf->f);  return (*size > 0) ? lf->buff : NULL;}

这个函数走完了，lf->buff就有了脚本文件的内容，说道这俩大家就明白了把，我们可以在getF里搞点小动作，比如这样

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

#if 0

static const char *getF (lua_State *L, void *ud,size_t *size) {

  LoadF *lf = (LoadF *)ud;

  (void)L;

  if (lf->extraline) {

    lf->extraline = 0;

    *size = 1;

    return "\n";

  }

  if (feof(lf->f))return NULL;

  *size = fread(lf->buff, 1, sizeof(lf->buff), lf->f);

  return (*size > 0) ? lf->buff : NULL;

}

 

#else

int DeCode(char *pBuffer,int nSize);

 

static const char *getF (lua_State *L, void *ud,size_t *size) {

    LoadF *lf = (LoadF *)ud;

    (void)L;

    if (lf->extraline) {

        lf->extraline = 0;

        *size = 1;

        return "\n";

    }

    if (feof(lf->f))return NULL;

    *size = fread(lf->buff, 1, sizeof(lf->buff), lf->f);

     

    DeCode(lf->buff,*size);//decode file content

     

    return (*size > 0) ? lf->buff : NULL;

}

#endif

　　这样这个lua库编译是会通过，但是链接是不通过的，就需要我们在外面实现一下DeCode函数。

那我们写个测试程序

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

#include <stdlib.h>

#include <fstream>

 

#include <Windows.h>

 

extern "C" {

#include "lua.h"

#include "lualib.h"

#include "lauxlib.h"

}

 

using namespace std;

 

#define CODE_MASK 250   //:)

 

extern "C" {

int EnCode(char *pBuffer,int nSize)

{

    for (size_t i=0; i<nSize; i++)

    {

        pBuffer[i] ^= CODE_MASK;

    }

    return nSize;

}

 

int DeCode(char *pBuffer,int nSize)

{

    return EnCode(pBuffer,nSize);

}

 

}

 

string GetCodeFile(const char* file)

{

    char * pBuffer = NULL;

    size_t nSize = 0;

 

    //read file

    FILE *fp = fopen(file,"rb");

    fseek(fp,0,SEEK_END);

    nSize = ftell(fp);

    fseek(fp,0,SEEK_SET);

    pBuffer = new char[nSize];

    nSize = fread(pBuffer,sizeof(char), nSize,fp);

    fclose(fp);

 

    //encode

    EnCode(pBuffer,nSize);

 

    //save file

    string str = file;

    str+=".out";

    FILE *fpw = fopen(str.c_str(),"wb");

    fwrite(pBuffer, nSize, 1, fpw);

    fclose(fpw);

 

    //free memory

    delete pBuffer;

 

    return str;

}

 

int main(int argc,char* argv[])

{

     

    lua_State   *L = lua_open();

    luaL_openlibs(L);

    if (2==argc)

    {

        //do original file

        //luaL_dofile(L, argv[1]);

 

        //do code file

        luaL_dofile(L, GetCodeFile(argv[1]).c_str());

    }

 

    system("PAUSE");

 

    return 0;

}

这种方法我自己测试可行的。怎么加密在你，我这只是说了个简单的异或加密解密，

需要注意的是：实际上我跟踪代码发现lua并不是一次性把整个文件全部加载，而是每次512的字节。也就是说如果你的文件很大的话，加载它是会多次调用getF

 

   如有疑问eMail我,[e-mail]

　如有错误请指正，大家一起进步。