source: proto/xmlschema/pablo_template_avx.cpp @ 3532

Last change on this file since 3532 was 2219, checked in by shiyangy, 7 years ago

project checkin

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