压缩存档的一个方案

bt

lua版的存档未经压缩，尺寸比较大，文件分为多个，分享很不方便。

因为lua版中的存档代码很多地方是分开写的，如果直接在内存中压缩，程序要改的地方可能比较多。

那么我们可以增加两个压缩和解压缩的函数，在存档时先按照以前的方案写，然后压缩文件，再将未压缩的文件删除，读档时也类似。这样可以少改一些代码。

首先下载附件，里面是头文件和库文件，需要在visual studio里面配置好，这个就不细讲了。
要改的代码如下：

luafun.c增加这两个函数（请自己在jymain.h里添加声明，以及所需的头文件）。

1. int Byte_unzip(lua_State *pL)
   
2. {
   
3.     const char *filenamezip = lua_tostring(pL, 1);
   
4.     int n = lua_gettop(pL);
   
5.     int i;
   
6.     unzFile zip;
   
7.     zip = unzOpen(filenamezip);
   
8.     for (i = 2; i <= n; i++)
   
9.     {
   
10.         const char *filename = lua_tostring(pL, i);
    
11.         unz_file_info zi = { 0 };
    
12.         if (unzLocateFile(zip, filename, NULL) != UNZ_OK)
    
13.         {
    
14.             break;
    
15.         }
    
16.         char s[100];
    
17.         unzGetCurrentFileInfo(zip, &zi, s, 100, NULL, 0, NULL, 0);
    
18.         unzOpenCurrentFile(zip);
    
19. 
    
20.         FILE *fp;
    
21.         if ((fp = fopen(filename, "wb")) == NULL)
    
22.         {
    
23.             break;
    
24.         }
    
25.         void* buf = malloc(8192);
    
26.         int c = 0;
    
27.         while (c < zi.uncompressed_size)
    
28.         {
    
29.             int len = zi.uncompressed_size - c;
    
30.             if (len > 8192) len = 8192;
    
31.             unzReadCurrentFile(zip, buf, 8192);
    
32.             fwrite(buf, len, 1, fp);
    
33.             c += 8192;
    
34.         }
    
35.         fclose(fp);
    
36.         free(buf);
    
37.     }
    
38.     unzClose(zip);
    
39.     return 0;
    
40. }
    
41. 
    
42. 
    
43. int Byte_zip(lua_State *pL)
    
44. {
    
45.     zip_fileinfo zi = { 0 };
    
46.     const char *filenamezip = lua_tostring(pL, 1);
    
47.     int n = lua_gettop(pL);
    
48.     int i;   
    
49.     zipFile zip = zipOpen(filenamezip, APPEND_STATUS_CREATE);
    
50.     for (i = 2; i <= n; i++)
    
51.     {
    
52.         const char *filename = lua_tostring(pL, i);
    
53.         FILE *fp;
    
54.         if ((fp = fopen(filename, "rb")) == NULL)
    
55.         {
    
56.             break;
    
57.         }
    
58.         fseek(fp, 0, SEEK_END);
    
59.         int length = ftell(fp);
    
60.         fseek(fp, 0, 0);
    
61.         char* s = malloc(length + 1);
    
62.         fread(s, length, 1, fp);
    
63.         fclose(fp);
    
64.         get_file_date(filenamezip, &zi.dos_date);
    
65.         //zi.dos_date = time(NULL);
    
66.         zipOpenNewFileInZip(zip, filename, &zi, NULL, 0, NULL, 0, NULL, Z_DEFLATED, Z_DEFAULT_COMPRESSION);
    
67.         zipWriteInFileInZip(zip, s, length);
    
68.         zipCloseFileInZip(zip);
    
69.         free(s);
    
70.     }
    
71.     zipClose(zip, NULL);
    
72.     return 0;
    
73. }
    
74. 
    
75. int Byte_loadfilefromzip(lua_State *pL)
    
76. {
    
77.     char *p = (char *)lua_touserdata(pL, 1);
    
78.     const char *filenamezip = lua_tostring(pL, 2);
    
79.     const char *filename = lua_tostring(pL, 3);
    
80.     int start = (int)lua_tonumber(pL, 4);
    
81.     int length = (int)lua_tonumber(pL, 5);
    
82. 
    
83.     unzFile zip;
    
84.     unz_file_info zi = { 0 };
    
85.     zip = unzOpen(filenamezip);
    
86.     if (unzLocateFile(zip, filename, NULL) != UNZ_OK)
    
87.     {
    
88.         return 0;
    
89.     }
    
90.     char s[100];
    
91.     unzGetCurrentFileInfo(zip, &zi, s, 100, NULL, 0, NULL, 0);
    
92.     unzOpenCurrentFile(zip);
    
93. 
    
94.     const int size_buf = 8192;
    
95.     void* buf = malloc(size_buf);
    
96.     void* buf2 = malloc(zi.uncompressed_size);
    
97.     int c = 0;
    
98.     while (c < zi.uncompressed_size)
    
99.     {
    
100.         int len = zi.uncompressed_size - c;
     
101.         if (len > size_buf) len = size_buf;
     
102.         unzReadCurrentFile(zip, buf, size_buf);
     
103.         memcpy((char*)buf2+c,buf,len);
     
104.         c += size_buf;
     
105.     }
     
106.     unzClose(zip);
     
107.     memcpy(p, (char*)buf2+start,length);
     
108.     free(buf);
     
109.     free(buf2);
     
110.     return 0;
     
111. }
     
112. 
     

复制代码

jymain.c里面初始化lua函数名的地方当然也要改一下：

1. static const struct luaL_Reg bytelib [] = {
   
2.       {"create", Byte_create},
   
3.       {"loadfile", Byte_loadfile},
   
4.       {"savefile", Byte_savefile},
   
5.       {"unzip", Byte_unzip},
   
6.       {"zip", Byte_zip},
   
7.       {"get16", Byte_get16},
   
8.       {"set16", Byte_set16},
   
9.       {"getu16", Byte_getu16},
   
10.       {"setu16", Byte_setu16},
    
11.       {"get32", Byte_get32},
    
12.       {"set32", Byte_set32},
    
13.       {"getstr", Byte_getstr},
    
14.       {"setstr", Byte_setstr},
    
15.           {NULL, NULL}
    
16. };

复制代码

这样就添加了两个专用于压缩或者解压的函数Byte.zip和Byte.unzip。它们的参数个数都是2个到无限多个。用法如下：

1. Byte.unzip('data/save/1.zip', 'r.grp','d.grp','s.grp')
   
2. Byte.zip('data/save/1.zip', 'r.grp','d.grp','s.grp')

复制代码

某些情况下如果只需要解压r.grp，也可以写:

1. Byte.unzip('data/save/1.zip', 'r.grp')

复制代码

在任务完成后可以用os.remove删除文件。
但是需要注意，压缩的时候是新建一个压缩包，故一定要一次将所有文件都写完。

westcoast

可以很6

zhangyebo1991

非常有必要的功能

airliukai12

顶顶顶顶顶