source: proto/SymbolTable/symtab_stl_template.cpp @ 1228

Last change on this file since 1228 was 1228, checked in by vla24, 8 years ago

Integrated symbol table with xmlwf. There are various implementations for the symbol table, please read /proto/SymbolTable/README_SymbolTable for more information.

File size: 14.8 KB
Line 
1#define TEMPLATED_SIMD_LIB
2
3#include <stdio.h>
4#include <stdlib.h>
5#include <errno.h>
6#include <sys/types.h>
7#include <sys/stat.h>
8#include <../lib_simd.h>
9#include <symtab.h>
10
11#include <queue>
12#include <string>
13
14#define DEBUG 0
15#define BLOCK_SIZE (sizeof(SIMD_type) * 8)
16#define SEGMENT_BLOCKS 12
17#define BUFFER_SIZE (BLOCK_SIZE * SEGMENT_BLOCKS)
18#define OVERLAP_BUFSIZE 2*(sizeof(SIMD_type))
19
20typedef long ScanBlock;
21typedef SIMD_type BytePack;
22typedef SIMD_type BitBlock;
23
24using namespace std;
25
26#include <../carryQ.h>
27#include <xmldecl.h>
28#include <xml_error.c>
29#include <xmldecl.c>
30#include <namechars.h>
31
32#include <../perflib/perfsec.h>
33#include <../s2p.h>
34
35#include <TagMatcher.h>
36#include <LineColTracker.h>
37
38#ifdef BUFFER_PROFILING
39        BOM_Table * parser_timer;
40
41#elif CODE_CLOCKER
42        #define NUM_EVENTS 1
43        int Events[NUM_EVENTS] = {PAPI_TOT_CYC};
44        //int Events[NUM_EVENTS] = {PAPI_L2_DCM};
45        //int Events[NUM_EVENTS] = {PAPI_TOT_CYC, PAPI_BR_MSP};
46        int cal_size = 20;
47        CC * parser_timer = new CC(Events,NUM_EVENTS,cal_size);
48#else
49        void * parser_timer;
50#endif
51
52int block_base=0;
53int buffer_base=0;
54int buffer_last;
55char * source;
56LineColTracker tracker;
57TagMatcher matcher;
58BitBlock EOF_mask = simd_const_1(1);
59
60queue <size_t> elem_starts_buf;
61queue <size_t> elem_ends_buf;
62vector <int> gids;
63SymbolTable symbol_table;
64
65static inline int StreamScan(ScanBlock * stream, int blk_count, int ProcessPos(int)) {
66        int blk;
67        int block_pos = 0;
68        for (blk = 0; blk < blk_count; blk++) {
69                ScanBlock s = stream[blk];
70                while(s) {
71                        int code = (ProcessPos(cfzl(s) + block_pos));
72                        if (code) return code;
73                        s = s & (s-1);  // clear rightmost bit.
74                }
75                block_pos += 8 * sizeof(ScanBlock);
76        }
77        return 0;
78}
79
80
81static inline void ReportError(const char * error_msg, int error_pos_in_block) {
82  int error_line, error_column;
83  tracker.get_Line_and_Column(error_pos_in_block, error_line, error_column);
84  fprintf(stderr, "%s at line %i, column %i\n", error_msg, error_line, error_column);
85}
86
87static inline int ElemStrt_check(int pos) {
88        int block_pos = block_base + pos;
89        elem_starts_buf.push(buffer_base + block_pos);
90        return 0;
91}
92
93static inline int ElemEnd_check(int pos) {
94        int block_pos = block_base + pos;
95        elem_ends_buf.push(buffer_base + block_pos);
96        return 0;
97}
98
99static inline int NameStrt_check(int pos) {
100        int block_pos = block_base + pos;
101        if(XML_10_UTF8_NameStrt_bytes((unsigned char*)&source[block_pos]) == 0){
102              ReportError("name start error", pos);
103              exit(-1);
104        }
105        return 0;
106}
107
108static inline int Name_check(int pos) {
109        int block_pos = block_base + pos;
110        if(XML_10_UTF8_NameChar_bytes((unsigned char*)&source[block_pos]) == 0){
111              ReportError("name error", pos);
112              exit(-1);
113        }
114        return 0;
115}
116
117static inline int PIName_check(int pos) {
118        int block_pos = block_base + pos;
119        int file_pos = block_pos+buffer_base;
120        printf ("%s:%i\n",__FUNCTION__,pos);
121        if (at_XxMmLll<ASCII>((unsigned char*)&source[block_pos]) && (source[block_pos+3]=='?' || source[block_pos+3]<= ' ')) {
122              // "<?xml" legal at start of file.
123              if ((file_pos == 2) && at_XmlDecl_start<ASCII>((unsigned char*)&source[0])) return 0;
124              ReportError("[Xx][Mm][Ll] illegal as PI name", pos);
125              exit(-1);
126        }
127        return 0;
128}
129
130static inline int CD_check(int pos) {
131        int block_pos = block_base + pos;
132        if (!at_CDATA1<ASCII>((unsigned char*)&source[block_pos])){
133              ReportError("CDATA error", pos);
134              exit(-1);
135        }
136        return 0;
137}
138
139static inline int GenRef_check(int pos) {
140        int block_pos = block_base + pos;
141        unsigned char* s = (unsigned char*)&source[block_pos];
142        if (!(at_Ref_gt<ASCII>(s)||at_Ref_lt<ASCII>(s)||at_Ref_amp<ASCII>(s)||at_Ref_quot<ASCII>(s)||at_Ref_apos<ASCII>(s))){
143              ReportError("Undefined reference", pos);
144              exit(-1);
145        }
146        return 0;
147}
148
149static inline int HexRef_check(int pos) {
150        int block_pos = block_base + pos;
151        unsigned char* s = (unsigned char*)&source[block_pos];
152        int ch_val = 0;
153        while(at_HexDigit<ASCII>(s)){
154          ch_val = HexVal<ASCII>(s[0]) + (ch_val<<4);
155          if (ch_val> 0x10FFFF ){
156            ReportError("Illegal character reference", pos);
157            exit(-1);
158          }
159          s++;
160        }
161        if ((ch_val == 0x0) || ((ch_val | 0x7FF) == 0xDFFF)|| ((ch_val | 0x1) == 0xFFFF)){
162          ReportError("Illegal character reference", pos);
163          exit(-1);
164        }
165        else if (((ch_val < 0x20) && (ch_val != 0x9) && (ch_val != 0xD) && (ch_val != 0xA))){
166          ReportError("Illegal XML 1.0 character reference", pos);
167          exit(-1);
168        }
169        return 0;
170}
171
172static inline int DecRef_check(int pos) {
173        int block_pos = block_base + pos;
174        unsigned char* s = (unsigned char*)&source[block_pos];
175        int ch_val = 0;
176        while(at_HexDigit<ASCII>(s)){
177          ch_val = DigitVal<ASCII>(s[0]) + ch_val*10;
178          if (ch_val> 0x10FFFF ){
179            ReportError("Illegal character reference", pos);
180            exit(-1);
181          }
182          s++;
183        }
184        if ((ch_val == 0x0) || ((ch_val | 0x7FF) == 0xDFFF)|| ((ch_val | 0x1) == 0xFFFF)){
185          ReportError("Illegal character reference", pos);
186          exit(-1);
187        }
188        else if (((ch_val < 0x20) && (ch_val != 0x9) && (ch_val != 0xD) && (ch_val != 0xA))){
189          ReportError("Illegal XML 1.0 character reference", pos);
190          exit(-1);
191        }
192        return 0;
193}
194
195static inline int AttRef_check(int pos) {
196        int block_pos = block_base + pos;
197        unsigned char* s = (unsigned char*)&source[block_pos];
198        int ch_val = 0;
199        if(s[0]=='#'){
200          s++;
201          if(s[0]=='x' || s[0]=='X'){
202            s++;
203            while(at_HexDigit<ASCII>(s)){
204              ch_val = HexVal<ASCII>(s[0]) + (ch_val<<4);
205              s++;
206            }
207          }
208          else{
209            while(at_HexDigit<ASCII>(s)){
210              ch_val = DigitVal<ASCII>(s[0]) + ch_val*10;
211              s++;
212            }
213          }
214          if (ch_val==60){
215            ReportError("Attribute values contain '<' characters after reference expansion", pos);
216            exit(-1);
217          }
218        }
219        else if(at_Ref_lt<ASCII>(s)){
220          ReportError("Attribute values contain '<' characters after reference expansion", pos);
221          exit(-1);
222        }
223        return 0;
224}
225
226
227
228@global
229
230static inline void s2p_do_block(BytePack U8[], Basis_bits & basis_bits) {
231  s2p(U8[0], U8[1], U8[2], U8[3], U8[4], U8[5], U8[6], U8[7],
232        basis_bits.bit_0, basis_bits.bit_1, basis_bits.bit_2, basis_bits.bit_3, basis_bits.bit_4, basis_bits.bit_5, basis_bits.bit_6, basis_bits.bit_7);
233}
234
235static inline void s2p_do_final_block(BytePack U8[], Basis_bits & basis_bits, BitBlock EOF_mask) {
236  s2p_do_block(U8, basis_bits);
237  basis_bits.bit_0 = simd_and(basis_bits.bit_0, EOF_mask);
238  basis_bits.bit_1 = simd_and(basis_bits.bit_1, EOF_mask);
239  basis_bits.bit_2 = simd_and(basis_bits.bit_2, EOF_mask);
240  basis_bits.bit_3 = simd_and(basis_bits.bit_3, EOF_mask);
241  basis_bits.bit_4 = simd_and(basis_bits.bit_4, EOF_mask);
242  basis_bits.bit_5 = simd_and(basis_bits.bit_5, EOF_mask);
243  basis_bits.bit_6 = simd_and(basis_bits.bit_6, EOF_mask);
244  basis_bits.bit_7 = simd_and(basis_bits.bit_7, EOF_mask);
245}
246
247static inline void do_symbol_table_lookup()
248{
249    while( !elem_starts_buf.empty() && !elem_ends_buf.empty() )
250    {
251        int start = elem_starts_buf.front();
252        int end = elem_ends_buf.front();
253        elem_starts_buf.pop();
254        elem_ends_buf.pop();
255        int length = end - start;
256
257        int gid = -1;
258
259
260        //lookup or insert to symbol table
261#if DEBUG
262        char* symbol = new char[length+1];
263        strncpy ( symbol, source + start - buffer_base, length );
264        symbol[length] ='\0';
265        printf ("start: %i[%i] | end: %i[%i] | length: %i | symbol: %s\n", start, start-buffer_base, end, end-buffer_base, length, symbol );
266
267        delete symbol; symbol = 0;
268#endif
269
270        gid = symbol_table.Lookup_or_Insert_Name(source + start - buffer_base, length);
271
272        gids.push_back(gid);
273    }
274}
275
276static inline void postprocess_do_block(Lex & lex, CtCDPI_Callouts & ctCDPI_Callouts, Ref_Callouts & ref_Callouts, Check_streams & check_streams, Tag_Callouts & tag_Callouts, int chars_avail){
277
278    //NOTE: We are NOT handling the case for an extremely long symbol name (length > 1541).
279    //      TagMatcher will crash if we feed a long symbol name.
280    //      Sample file: test/long_sym_name.xml
281
282            tracker.StoreNewlines(lex.LF);
283
284            if ( bitblock_has_bit(tag_Callouts.ElemName_starts))
285            {
286                StreamScan((ScanBlock *) &tag_Callouts.ElemName_starts, sizeof(BitBlock)/sizeof(ScanBlock), ElemStrt_check);
287            }
288
289            if ( bitblock_has_bit(tag_Callouts.ElemName_ends) )
290            {
291                StreamScan((ScanBlock *) &tag_Callouts.ElemName_ends, sizeof(BitBlock)/sizeof(ScanBlock), ElemEnd_check);
292            }
293
294            do_symbol_table_lookup();
295
296                if (bitblock_has_bit(simd_or(check_streams.non_ascii_name_starts, check_streams.non_ascii_names))) {
297                  StreamScan((ScanBlock *) &check_streams.non_ascii_name_starts, sizeof(BitBlock)/sizeof(ScanBlock), NameStrt_check);
298                  StreamScan((ScanBlock *) &check_streams.non_ascii_names, sizeof(BitBlock)/sizeof(ScanBlock), Name_check);
299                }
300
301                if (bitblock_has_bit(ctCDPI_Callouts.PI_name_starts)){
302                  StreamScan((ScanBlock *) &ctCDPI_Callouts.PI_name_starts, sizeof(BitBlock)/sizeof(ScanBlock), PIName_check);
303                }
304
305                if (bitblock_has_bit(ctCDPI_Callouts.CD_starts)){
306                  StreamScan((ScanBlock *) &ctCDPI_Callouts.CD_starts, sizeof(BitBlock)/sizeof(ScanBlock), CD_check);
307                }
308
309                if (bitblock_has_bit(ref_Callouts.GenRef_starts)){
310                  StreamScan((ScanBlock *) &ref_Callouts.GenRef_starts, sizeof(BitBlock)/sizeof(ScanBlock), GenRef_check);
311                }
312
313                if (bitblock_has_bit(ref_Callouts.DecRef_starts)){
314                  StreamScan((ScanBlock *) &ref_Callouts.DecRef_starts, sizeof(BitBlock)/sizeof(ScanBlock), DecRef_check);
315                }
316
317                if (bitblock_has_bit(ref_Callouts.HexRef_starts)){
318                  StreamScan((ScanBlock *) &ref_Callouts.HexRef_starts, sizeof(BitBlock)/sizeof(ScanBlock), HexRef_check);
319                }
320
321                if (bitblock_has_bit(check_streams.att_refs)){
322                  StreamScan((ScanBlock *) &check_streams.att_refs, sizeof(BitBlock)/sizeof(ScanBlock), AttRef_check);
323                }
324
325                if (bitblock_has_bit(check_streams.error_mask)) {
326                  int errpos = count_forward_zeroes(check_streams.error_mask);
327                  ReportError("error found", errpos);
328              exit(-1);
329                }
330
331                matcher.store_streams(check_streams.tag_marks, check_streams.name_follows, check_streams.misc_mask, chars_avail);
332                tracker.AdvanceBlock();
333}
334
335static inline void print_GIDS()
336{
337    int span_count = gids.size();
338    for(int i=0;i<span_count;i++) {
339             cout << gids[i] << " ";
340    }
341}
342
343void do_process(FILE *infile, FILE *outfile) {
344
345@decl
346
347  int buf_pos = 0;
348  int block_pos = 0;
349  int errpos = 0;
350  int chars_avail = 0;
351  int check_pos = 0;
352  int chars_read = 0;
353  BytePack buf[(BUFFER_SIZE+BLOCK_SIZE+OVERLAP_BUFSIZE*2)/sizeof(SIMD_type)];
354
355  char * srcbuf = ((char *) buf) + OVERLAP_BUFSIZE;
356  buffer_base = buf_pos;
357  source = srcbuf;
358  chars_read = fread((void *)srcbuf, 1, BUFFER_SIZE + OVERLAP_BUFSIZE, infile);
359  chars_avail = chars_read;
360  if (chars_avail > BUFFER_SIZE) chars_avail = BUFFER_SIZE;
361
362  matcher.setSrc(srcbuf);
363
364  if(chars_read<4){
365    fprintf(stderr,"File is too short. Not well formed.\n");
366    exit(-1);
367  }
368
369  Entity_Info * e = new Entity_Info;
370  e->AnalyzeSignature((unsigned char *)srcbuf);
371
372  if (e->code_unit_base == ASCII) {
373
374    XML_Decl_Parser<ASCII> decl_parser((unsigned char *)srcbuf);
375
376    decl_parser.ReadXMLInfo(*e);
377
378    if (e->code_unit_size != SingleByte || (e->has_encoding_decl && (!at_UTF_8(e->encoding)))){
379        fprintf(stderr,"Sorry, this xmlwf demo only works for UTF-8.\n");
380        exit(-1);
381    }
382  }
383  else {
384    fprintf(stderr,"Sorry, this xmlwf demo does not process EBCDIC.\n");
385        exit(-1);
386  }
387
388  if (e->content_start != 0) {
389        memmove(&srcbuf[0], &srcbuf[e->content_start], chars_read - e->content_start);
390        buf_pos = e->content_start;
391        buffer_base = buf_pos;
392        if (chars_avail == BUFFER_SIZE) {
393                chars_read = chars_read - e->content_start +
394                             fread(&srcbuf[chars_read-e->content_start], 1, e->content_start, infile);
395                chars_avail = chars_read;
396                if (chars_avail > BUFFER_SIZE) chars_avail = BUFFER_SIZE;
397        }
398        else {
399          chars_read -=e->content_start;
400          chars_avail -=e->content_start;
401        }
402  }
403
404@stream_stmts
405
406/* Full Buffers */
407    int block_segment_num = 0;
408    while (chars_avail == BUFFER_SIZE) {
409      PERF_SEC_START(parser_timer);
410      for (int blk = 0; blk < SEGMENT_BLOCKS; blk++) {
411          block_base = blk*BLOCK_SIZE;
412          s2p_do_block((BytePack *) &srcbuf[block_base], basis_bits);
413          @block_stmts
414          postprocess_do_block(lex, ctCDPI_Callouts, ref_Callouts, check_streams, tag_Callouts, chars_avail);
415      }
416      matcher.StreamScan(chars_avail);
417      matcher.Advance_buffer();
418      tracker.Advance_buffer();
419      PERF_SEC_END(parser_timer, chars_avail);
420           
421      int bytes_left = chars_read - chars_avail;
422      memmove(buf, &srcbuf[BUFFER_SIZE - OVERLAP_BUFSIZE], bytes_left + OVERLAP_BUFSIZE);
423      chars_read = fread(&srcbuf[bytes_left],1, BUFFER_SIZE + OVERLAP_BUFSIZE - bytes_left, infile) + bytes_left;
424      chars_avail = chars_read;
425      if (chars_avail > BUFFER_SIZE) chars_avail = BUFFER_SIZE;
426      buf_pos += chars_avail;
427      buffer_base = buf_pos;
428      block_segment_num++;
429    }
430/* Final Partial Buffer */
431    PERF_SEC_START(parser_timer);
432
433    block_pos = 0;
434    int remaining = chars_avail;
435/* Full Blocks */
436    while (remaining >= BLOCK_SIZE) {
437          block_base = block_pos;
438          s2p_do_block((BytePack *) &srcbuf[block_pos], basis_bits);
439          @block_stmts
440          postprocess_do_block(lex, ctCDPI_Callouts, ref_Callouts, check_streams, tag_Callouts, chars_avail);
441          block_pos += BLOCK_SIZE;
442          remaining -= BLOCK_SIZE;
443    }
444    block_base = block_pos;
445    if (remaining > 0 || @any_carry) {
446          EOF_mask = sisd_srl(simd_const_1(1),sisd_from_int(BLOCK_SIZE-remaining));
447          s2p_do_final_block((BytePack *) &srcbuf[block_pos], basis_bits, EOF_mask);
448          @final_block_stmts
449          postprocess_do_block(lex, ctCDPI_Callouts, ref_Callouts, check_streams, tag_Callouts, chars_avail);
450    }
451    buf_pos += chars_avail;
452    buffer_base = buf_pos;
453
454    matcher.StreamScan(chars_avail);
455    matcher.Advance_buffer();
456    tracker.Advance_buffer();
457
458    PERF_SEC_END(parser_timer, chars_avail);
459    if (matcher.depth != 0) {
460      fprintf(stderr, "tag matching error (depth %i) at position %i\n", matcher.depth, buffer_base);
461      exit(-1);
462    }
463
464//      print_GIDS();
465}
466
467
468
469int
470main(int argc, char * argv[]) {
471        char * infilename, * outfilename;
472        FILE *infile, *outfile;
473        struct stat fileinfo;
474
475        if (argc < 2) {
476                printf("Usage: %s <filename> [<outputfile>]\n", argv[0]);
477                exit(-1);
478        }
479
480        infilename = argv[1];
481        stat(infilename, &fileinfo);
482        infile = fopen(infilename, "rb");
483        if (!infile) {
484                fprintf(stderr, "Error: cannot open %s for input.\n", infilename);
485                exit(-1);
486        }
487
488        if (argc < 3) outfile = stdout;
489        else {
490                outfilename = argv[2];
491                outfile = fopen(outfilename, "wb");
492                if (!outfile) {
493                        fprintf(stderr, "Error: cannot open %s for writing.\n", outfilename);
494                        exit(-1);
495                }
496        }
497
498//      PERF_SEC_BIND(1);
499
500        PERF_SEC_INIT(parser_timer);
501
502        do_process(infile, outfile);
503
504        PERF_SEC_DUMP(parser_timer);
505
506        PERF_SEC_DESTROY(parser_timer);
507
508        fclose(infile);
509        fclose(outfile);
510
511        printf ("Done procressing\n");
512        return(0);
513}
Note: See TracBrowser for help on using the repository browser.