Home | History | Annotate | Line # | Download | only in tests 
      1 /*
      2  * This file is part of flex.
      3  *
      4  * Redistribution and use in source and binary forms, with or without
      5  * modification, are permitted provided that the following conditions
      6  * are met:
      7  *
      8  * 1. Redistributions of source code must retain the above copyright
      9  *    notice, this list of conditions and the following disclaimer.
     10  * 2. Redistributions in binary form must reproduce the above copyright
     11  *    notice, this list of conditions and the following disclaimer in the
     12  *    documentation and/or other materials provided with the distribution.
     13  *
     14  * Neither the name of the University nor the names of its contributors
     15  * may be used to endorse or promote products derived from this software
     16  * without specific prior written permission.
     17  *
     18  * THIS SOFTWARE IS PROVIDED ``AS IS'' AND WITHOUT ANY EXPRESS OR
     19  * IMPLIED WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED
     20  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     21  * PURPOSE.
     22  */
     23 
     24 %{
     25 /* A file to build "scanner.c". */
     26 /* This tests that we can use "yyextra".
     27    We buffer all input into a growable array, then print it.
     28    We run diff on the input and output.
     29 */
     30 
     31 #include <stdio.h>
     32 #include <stdlib.h>
     33 #include "config.h"
     34 
     35 
     36 /* We'll store the entire input in this buffer, growing as necessary. */
     37 struct Check {
     38     char foo;
     39     char *bar;
     40     char qux;
     41 };
     42 
     43 /* Save char into junk array at next position. */
     44 static void check_extra ( yyscan_t  scanner );
     45 
     46 %}
     47 
     48 %option 8bit  prefix="test"
     49 %option nounput nomain noyywrap nodefault noinput
     50 %option warn
     51 %option extra-type="struct Check *"
     52 %option reentrant
     53 %option noyyalloc
     54 
     55 
     56 %%
     57 
     58 .|\r|\n  { check_extra (yyscanner); }
     59 
     60 %%
     61 
     62 int main(void);
     63 
     64 int
     65 main (void)
     66 {
     67     yyscan_t scanner;
     68     struct Check check;
     69 
     70     check.foo = 'a';
     71     check.bar = NULL;
     72     check.qux = 'z';
     73 
     74     testlex_init_extra(&check, &scanner);
     75     testset_in(stdin, scanner);
     76     testset_out(stdout, scanner);
     77 
     78     /* Test to confirm that testalloc was called from
     79      * testlex_init_extra with the testextra argument.
     80      */
     81     check_extra(scanner);
     82 
     83     testlex(scanner);
     84 
     85     testlex_destroy(scanner);
     86     return 0;
     87 }
     88 
     89 void *testalloc(size_t size, yyscan_t scanner)
     90 {
     91     struct Check *check;
     92     check = testget_extra(scanner);
     93 
     94     if (!check->bar)
     95         check->bar = "Hello World";
     96 
     97     check_extra(scanner);
     98 
     99     return malloc(size);
    100 }
    101 
    102 /* Save char into junk array at next position. */
    103 static void check_extra(yyscan_t  scanner)
    104 {
    105     struct Check *check;
    106     check = testget_extra(scanner);
    107 
    108     if (check->foo != 'a') {
    109         fprintf(stderr, "foo is not 'a'\n");
    110         exit(1);
    111     }
    112     if (strcmp(check->bar, "Hello World") != 0) {
    113         fprintf(stderr, "bar is not Hello World\n");
    114         exit(1);
    115     }
    116     if (check->qux != 'z') {
    117         fprintf(stderr, "qux is not 'z'\n");
    118         exit(1);
    119     }
    120 }
    121