lascii85/lascii85.c at master · LuaDist/lascii85 · GitHub

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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
/*
* lascii85.c
* ascii85 encoding and decoding for Lua 5.1
* Luiz Henrique de Figueiredo <lhf@tecgraf.puc-rio.br>
* 23 Mar 2010 22:25:18
* This code is hereby placed in the public domain.
*/

#include <string.h>

#include "lua.h"
#include "lauxlib.h"

#define MYNAME		"ascii85"
#define MYVERSION	MYNAME " library for " LUA_VERSION " / Mar 2010"

#define uint unsigned int

static void encode(luaL_Buffer *b, uint c1, uint c2, uint c3, uint c4, int n)
{
 unsigned long tuple=c4+256UL*(c3+256UL*(c2+256UL*c1));
 if (tuple==0 && n==4)
  luaL_addlstring(b,"z",1);
 else
 {
  int i;
  char s[5];
  for (i=0; i<5; i++) {
   s[4-i] = '!' + (tuple % 85);
   tuple /= 85;
  }
  luaL_addlstring(b,s,n+1);
 }
}

static int Lencode(lua_State *L)		/** encode(s) */
{
 size_t l;
 const unsigned char *s=(const unsigned char*)luaL_checklstring(L,1,&l);
 luaL_Buffer b;
 int n;
 luaL_buffinit(L,&b);
 luaL_addlstring(&b,"<~",2);
 for (n=l/4; n--; s+=4) encode(&b,s[0],s[1],s[2],s[3],4);
 switch (l%4)
 {
  case 1: encode(&b,s[0],0,0,0,1);		break;
  case 2: encode(&b,s[0],s[1],0,0,2);		break;
  case 3: encode(&b,s[0],s[1],s[2],0,3);	break;
 }
 luaL_addlstring(&b,"~>",2);
 luaL_pushresult(&b);
 return 1;
}

static void decode(luaL_Buffer *b, int c1, int c2, int c3, int c4, int c5, int n)
{
 unsigned long tuple=c5+85L*(c4+85L*(c3+85L*(c2+85L*c1)));
 char s[4];
 switch (--n)
 {
  case 4: s[3]=tuple;
  case 3: s[2]=tuple >> 8;
  case 2: s[1]=tuple >> 16;
  case 1: s[0]=tuple >> 24;
 }
 luaL_addlstring(b,s,n);
}

static int Ldecode(lua_State *L)		/** decode(s) */
{
 size_t l;
 const char *s=luaL_checklstring(L,1,&l);
 luaL_Buffer b;
 int n=0;
 char t[5];
 if (*s++!='<') return 0;
 if (*s++!='~') return 0;
 luaL_buffinit(L,&b);
 for (;;)
 {
  int c=*s++;
  switch (c)
  {
   default:
    if (c<'!' || c>'u') return 0;
    t[n++]=c-'!';
    if (n==5)
    {
     decode(&b,t[0],t[1],t[2],t[3],t[4],5);
     n=0;
    }
    break;
   case 'z':
    luaL_addlstring(&b,"\0\0\0\0",4);
    break;
   case '~':
    if (*s!='>') return 0;
    switch (n)
    {
     case 1: decode(&b,t[0],85,0,0,0,1);		break;
     case 2: decode(&b,t[0],t[1],85,0,0,2);		break;
     case 3: decode(&b,t[0],t[1],t[2],85,0,3);		break;
     case 4: decode(&b,t[0],t[1],t[2],t[3],85,4);	break;
    }
    luaL_pushresult(&b);
    return 1;
   case '\n': case '\r': case '\t': case ' ':
   case '\0': case '\f': case '\b': case 0177:
    break;
  }
 }
 return 0;
}

static const luaL_Reg R[] =
{
	{ "encode",	Lencode	},
	{ "decode",	Ldecode	},
	{ NULL,		NULL	}
};

LUALIB_API int luaopen_ascii85(lua_State *L)
{
 luaL_register(L,MYNAME,R);
 lua_pushliteral(L,"version");			/** version */
 lua_pushliteral(L,MYVERSION);
 lua_settable(L,-3);
 return 1;
}