Lua 通过 alien 库调用 zlib 压缩/解压

http://blog.csdn.net/kowity/article/details/7256376

上次的文章（http://blog.csdn.net/kowity/article/details/7229815）讲述了上个在 Lua 中调用 zlib 压缩、解压功能的库。其实通过使用 Lua 的 alien 库就可以直接调用外部 dll 中的函数，并不需要专门为 Lua 写一个插件。

调用 zlib 压缩、解压的代码如下（参考了 LuaJIT 的部分实现方式（http://luajit.org/ext_ffi_tutorial.html））：

[python]  view plain copy

1. require("alien")  
2.   
3. local zlib = alien.load("zlib1")  
4. zlib.compressBound:types('long', 'long')  
5. zlib.compress2:types('int', 'pointer', 'pointer', 'string', 'long', 'int')  
6. zlib.uncompress:types('int', 'pointer', 'pointer', 'string', 'long')  
7.   
8. local function compress(txt)  
9.   local n = zlib.compressBound(#txt)  
10.   local buf = alien.buffer(n)  
11.   local buflen = alien.buffer('0000')  
12.   local res = zlib.compress2(buf, buflen, txt, #txt, 9)  
13.   assert(res == 0)  
14.   -- 返回压缩结果和压缩后的长度  
15.   return buf, alien.toulong(buflen:topointer())  
16. end  
17.   
18. local function uncompress(comp, comp_len, n)  
19.   local buf = alien.buffer(n)  
20.   local buflen = alien.buffer('0000')  
21.   local res = zlib.uncompress(buf, buflen, comp, comp_len)  
22.   assert(res == 0)  
23.   -- 返回解压后在缓冲区中有效的部分  
24.   return tostr(buf, alien.toulong(buflen:topointer()))  
25.     
26. end  
27.   
28. -- 有符号数转无符号数  
29. function toUnsigned(num)  
30.     local n  
31.     if num < 0 then  
32.         n = 256 + num  
33.     else  
34.         n = num  
35.     end  
36.       
37.     return n  
38. end  
39.   
40. function tostr(buf, len)  
41.     local str  
42.     for i = 1, len do   
43.         -- Lua 把 buf 里面的数当成有符号数了，  
44.         -- 导致读出来的有负数  
45.         local val = toUnsigned(buf[i])  
46.         if i == 1 then  
47.             str = string.char(val)  
48.         else  
49.             str = str .. string.char(val)  
50.         end  
51.     end  
52.       
53.     return str  
54. end  
55.   
56. local txt = string.rep("ab\0cd", 100)  
57. print("Uncompressed size: ", #txt)  
58. local c, c_len = compress(txt)  
59. print("Compressed size: ", c_len)  
60. local txt2 = uncompress(c, c_len, #txt)  
61. assert(txt2 == txt)