added \u notation.
[swftools.git] / src / parser.lex
1 %{
2
3 #include <string.h>
4 #include <stdlib.h>
5 #include <stdio.h>
6 #include "../lib/q.h"
7 #include "parser.h"
8
9 //RVALUE         {NUMBER}|{PERCENT}|{NAME}|\"{STRING}\"|{DIM}
10 //<a>.                {printf("<a>%s\n", yytext);}
11 // %x: exclusive, %s: inclusive
12 char*type_names[] = {"twip","number","command","string","assignment","identifier","label","end"};
13 static int line=1;
14 static int column=1;
15
16 mem_t strings;
17 mem_t tokens;
18
19 static void count(char*text, int len, int condition)
20 {
21     int t;
22     for(t=0;t<len;t++) {
23         if(text[t]=='\n') {
24             line++;
25             column=1;
26         } else {
27             column++;
28         }
29     }
30 }
31
32 static char*prefix = 0;
33
34 static char utf8buf[16];
35 static char* getUTF8(unsigned int charnum)
36 {
37     memset(utf8buf, 0, sizeof(utf8buf));
38
39     if(charnum < 0x80) {
40         utf8buf[0] = charnum;
41         return utf8buf;
42     } else if(charnum <0x800) {
43         /* 0000 0080-0000 07FF   110xxxxx 10xxxxxx */
44         utf8buf[0] = 0xc0 | (charnum >> 6);
45         utf8buf[1] = 0x80 | (charnum & 0x3f);
46         return utf8buf;
47     } else if(charnum < 0x10000) {
48         /* 0000 0800-0000 FFFF   1110xxxx 10xxxxxx 10xxxxxx */
49         utf8buf[0] = 0xe0 | (charnum >> 12);
50         utf8buf[1] = 0x80 |((charnum >> 6)&0x3f);
51         utf8buf[2] = 0x80 |((charnum     )&0x3f);
52         return utf8buf;
53     } else if(charnum < 0x200000) {
54         /* 0001 0000-001F FFFF   11110xxx 10xxxxxx 10xxxxxx 10xxxxxx */
55         utf8buf[0] = 0xf0 | (charnum >> 18);
56         utf8buf[1] = 0x80 |((charnum >> 12)&0x3f);
57         utf8buf[2] = 0x80 |((charnum >> 6 )&0x3f);
58         utf8buf[3] = 0x80 |((charnum      )&0x3f);
59         return utf8buf;
60     } else if(charnum < 0x4000000) {
61         /* 0020 0000-03FF FFFF   111110xx 10xxxxxx 10xxxxxx 10xxxxxx 10xxxxxx */
62         utf8buf[0] = 0xf8 | (charnum >> 24);
63         utf8buf[1] = 0x80 |((charnum >> 18)&0x3f);
64         utf8buf[2] = 0x80 |((charnum >> 12)&0x3f);
65         utf8buf[3] = 0x80 |((charnum >> 6 )&0x3f);
66         utf8buf[4] = 0x80 |((charnum      )&0x3f);
67         return utf8buf;
68     } else if(charnum < 0x80000000) {
69         /* 0400 0000-7FFF FFFF   1111110x 10xxxxxx ... 10xxxxxx */
70         utf8buf[0] = 0xfc | (charnum >> 30);
71         utf8buf[1] = 0x80 |((charnum >> 24)&0x3f);
72         utf8buf[2] = 0x80 |((charnum >> 18)&0x3f);
73         utf8buf[3] = 0x80 |((charnum >> 12)&0x3f);
74         utf8buf[4] = 0x80 |((charnum >> 6 )&0x3f);
75         utf8buf[5] = 0x80 |((charnum      )&0x3f);
76         return utf8buf;
77     } else {
78         fprintf(stderr, "Illegal character: 0x%08x\n", charnum);
79         return utf8buf;
80     }
81 }
82
83 static void unescapeString(string_t * tmp)
84 {
85     char *p, *p1;
86     /* fixme - this routine expects the string to be
87        null-terminated */
88
89     for (p1=tmp->str; (p=strchr(p1, '\\')); p1 = p+1) 
90     {
91         int nr=2;
92         int new=1;
93         switch(p[1])
94         {
95             case '\\': p[0] = '\\'; break;
96             case '"': p[0] = '"'; break;
97             case 'b': p[0] = '\b'; break;
98             case 'f': p[0] = '\f'; break;
99             case 'n': p[0] = '\n'; break;
100             case 'r': p[0] = '\r'; break;
101             case 't': p[0] = '\t'; break;
102             case 'x':  case 'u': {
103                 int max=2;
104                 int num=0;
105                 char*utf8;
106                 if(p[1] == 'u')
107                     max = 4;
108                 while(strchr("0123456789abcdefABCDEF", p[nr]) && nr < max+2) {
109                     num <<= 4;
110                     if(p[nr]>='0' && p[nr]<='9') num |= p[nr] - '0';
111                     if(p[nr]>='a' && p[nr]<='f') num |= p[nr] - 'a' + 10;
112                     if(p[nr]>='A' && p[nr]<='F') num |= p[nr] - 'A' + 10;
113                     nr++;
114                 }
115                 utf8 = getUTF8(num);
116                 new = strlen(utf8);
117                 memcpy(p, utf8, new); // do not copy the terminating zero
118                 break;
119             }
120             default:
121                 continue;
122         }
123         tmp->len -= (nr-new); 
124         {
125             int t;
126             char*to=p+new,*from=p+nr;
127             while(*from) {
128                 *to = *from;
129                 to++;
130                 from++;
131             }
132         }
133     }
134 }
135
136 static void store(enum type_t type, int line, int column, char*text, int length)
137 {
138     struct token_t token;
139     string_t tmp;
140     token.type = type;
141     token.line = line;
142     token.column = column;
143     //printf("->%d(%s) %s\n", type, type_names[type], text);fflush(stdout);
144
145     token.text = 0;
146     switch(type) {
147         case END:
148             string_set2(&tmp, "", 0);
149             token.text = (char*)mem_putstring(&strings, tmp);
150         break;
151         case STRING:
152             string_set2(&tmp, text+1, length-2);
153             unescapeString(&tmp);
154             token.text = (char*)mem_putstring(&strings, tmp);
155         break;
156         case TWIP: 
157         case NUMBER: 
158         case IDENTIFIER:
159             string_set2(&tmp, text, length);
160             if(prefix) {
161                 //strcat
162                 token.text = (char*)mem_put(&strings, prefix, strlen(prefix));
163                 mem_putstring(&strings, tmp);
164             } else {
165                 token.text = (char*)mem_putstring(&strings, tmp);
166             }
167             prefix = 0;
168         break;
169         case RAWDATA:
170             string_set2(&tmp, text+1/*:*/, length-5/*.end*/);
171             token.text = (char*)mem_putstring(&strings, tmp);
172         break;
173         case COMMAND:
174             string_set2(&tmp, text+1, length-1);
175             token.text = (char*)mem_putstring(&strings, tmp);
176         break;
177         case ASSIGNMENT: {
178             char*x = &text[length-1];
179             if(x[-1] == '-' || x[-1] == '+')
180                 x--;
181             do{x--;} while(*x==32 || *x==10 || *x==13 || *x=='\t');
182             x++; //first space
183             string_set2(&tmp, text, x-text);
184             token.text = (char*)mem_putstring(&strings, tmp);
185             /*char*y,*x = strchr(text, '=');
186             if(!x) exit(1);
187             y=x;
188             do{y--;} while(*y==32 || *y==10 || *y==13 || *y=='\t');
189             do{x++;} while(*x==32 || *x==10 || *x==13 || *x=='\t');
190             token.text1 = (char*)put(&strings, text, y-text + 1, 1);
191             token.text2 = (char*)put(&strings, x, length-(x-text), 1);*/
192         } break;
193     }
194
195     mem_put(&tokens, &token, sizeof(struct token_t));
196     prefix = 0;
197 }
198
199 #define MAX_INCLUDE_DEPTH 16
200 YY_BUFFER_STATE include_stack[MAX_INCLUDE_DEPTH];
201 int line_stack[MAX_INCLUDE_DEPTH];
202 int column_stack[MAX_INCLUDE_DEPTH];
203 int include_stack_ptr = 0;
204
205 void handleInclude(char*text, int len)
206 {
207     text+=9;len-=9;
208     while(len >=1 && (text[0] == ' ' || text[0] == '\t')) {
209         text++;len--;
210     }
211     while(len >= 1 && (text[len-1] == ' ' || text[len-1] == '\n')) {
212         len--;
213     }
214     if(len >= 2 && text[0] == '"' && text[len-1] == '"') {
215         text++; len-=2;
216     }
217     text[len] = 0;
218     if(include_stack_ptr >= MAX_INCLUDE_DEPTH) {
219         fprintf( stderr, "Includes nested too deeply" );
220         exit( 1 );
221     }
222     include_stack[include_stack_ptr] = YY_CURRENT_BUFFER;
223     line_stack[include_stack_ptr] = line;
224     column_stack[include_stack_ptr] = column;
225     include_stack_ptr++;
226     yyin = fopen(text, "rb");
227     if (!yyin) {
228         fprintf(stderr, "Couldn't open %s\n", text);
229         exit(1);
230     }
231     yy_switch_to_buffer(
232         yy_create_buffer( yyin, YY_BUF_SIZE ) );
233     BEGIN(INITIAL);
234 }
235
236 #define c() {count(yytext, yyleng, YY_START);}
237 #define s(type) {store(type, line, column, yytext, yyleng);}
238 %}
239
240 %s R
241 %x BINARY
242
243 NAME     [a-zA-Z_./](-*[a-zA-Z0-9_./])*
244 TWIP     (-?[0-9]+(\.([0-9]([05])?)?)?)
245 NUMBER   -?[0-9]+(\.[0-9]*)?
246 PERCENT  {NUMBER}%
247 STRING   (\\.|[^\\"\n])*
248 S        [ \n\r\t]
249 RVALUE   \"{STRING}\"|([^ \n\r\t]+)
250
251 %%
252
253 <BINARY>\] {c();BEGIN(0);}
254 <BINARY>.  {c();}
255 <BINARY>\n {c();}
256 {TWIP}/[ \n\r\t]            {s(TWIP);c();BEGIN(0);}
257 {NUMBER}/[ \n\r\t]          {s(NUMBER);c();BEGIN(0);}
258 ^#[^\n]*\n                  {c();}
259 [ \t\r]#[^\n]*\n            {c();}
260 \"{STRING}\"                {s(STRING);c();BEGIN(0);}
261 \"{STRING}$                 {c();printf("unterminated string in line %d: %s\n", line, yytext);exit(1);yyterminate();}
262 {NAME}{S}*\+=               {s(ASSIGNMENT);prefix="<plus>";c();BEGIN(R);}
263 {NAME}{S}*-=                {s(ASSIGNMENT);prefix="<minus>";c();BEGIN(R);}
264 {NAME}{S}*=                 {s(ASSIGNMENT);c();BEGIN(R);}
265 <R>{ /* values which appear only on the right-hand side of assignments, like: x=50% */
266     [^ :\n\t\r]*                    {s(IDENTIFIER);c();BEGIN(0);}
267 }
268 \.include{S}.*\n                    {handleInclude(yytext, yyleng);}
269 \.{NAME}                    {s(COMMAND);c();}
270 :([^.]|\.[^e]|\.e[^n]|\.en[^d]|\.end[^ \n\r\t]|[ \n\r\t])*\.end     {s(RAWDATA);c();}
271 {NAME}                      {s(IDENTIFIER);c();}
272 "["                         {c();BEGIN(BINARY);}
273 {S}                         {c();}
274 .                           {char c,c1=yytext[0];
275                              printf("Syntax error in line %d, %d: %s", line, column, yytext);
276                              while(1) {
277                                  c=input();
278                                  if(c=='\n' || c==EOF) 
279                                      break;
280                                 printf("%c", c);
281                              }
282                              if(c1>='0' && c1<='9')
283                                  printf(" (identifiers must not start with a digit)");
284                              printf("\n");
285                              exit(1);
286                              yyterminate();
287                             }
288 <<EOF>>                     {c();
289                              if ( --include_stack_ptr < 0 ) {
290                                 s(END);
291                                 yyterminate();
292                              } else {
293                                  yy_delete_buffer( YY_CURRENT_BUFFER );
294                                  yy_switch_to_buffer(
295                                       include_stack[include_stack_ptr] );
296                                  column = column_stack[include_stack_ptr];
297                                  line = line_stack[include_stack_ptr];
298                              }
299                             }
300
301 %%
302
303 int yywrap()
304 {
305     return 1;
306 }
307
308 void freeTokens(struct token_t*file)
309 {
310     mem_clear(&strings);
311     mem_clear(&tokens);
312 }
313
314 struct token_t* generateTokens(char*filename)
315 {
316     FILE*fi = fopen(filename, "rb");
317     int t;
318     struct token_t*result;
319     int num;
320     if(!fi) {
321         printf("Couldn't find file %s\n", filename);
322         return 0;
323     }
324     yyin = fi;
325
326     mem_init(&strings);
327     mem_init(&tokens);
328     mem_put(&strings, &t, 1); //hack- make all valid strings start at position >0
329
330     line=1;
331     column=1;
332
333     yylex();
334     yy_delete_buffer(yy_current_buffer);
335
336     result = (struct token_t*)tokens.buffer;
337     num = tokens.pos/sizeof(struct token_t);
338
339     for(t=0;t<tokens.pos/sizeof(struct token_t);t++) {
340         if(result[t].text)
341             result[t].text += (int)strings.buffer;
342     }
343
344     fclose(fi);
345     return result;
346 }
347