external/bsd/flex/dist/tests/test-alloc-extra/scanner.l

   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 /* Special yyalloc */
  47 void *yyalloc ( size_t size, yyscan_t  scanner );
  48
  49 %}
  50
  51 %option 8bit outfile="scanner.c" prefix="test"
  52 %option nounput nomain noyywrap nodefault
  53 %option warn
  54 %option extra-type="struct Check *"
  55 %option reentrant
  56 %option noyyalloc
  57
  58
  59 %%
  60
  61 .|\r|\n  { check_extra (yyscanner); }
  62
  63 %%
  64
  65 int main(void);
  66
  67 int
  68 main ()
  69 {
  70     yyscan_t scanner;
  71     struct Check check;
  72
  73     check.foo = 'a';
  74     check.bar = NULL;
  75     check.qux = 'z';
  76
  77     testlex_init_extra(&check, &scanner);
  78     testset_in(stdin, scanner);
  79     testset_out(stdout, scanner);
  80
  81     /* Test to confirm that yyalloc was called from
  82      * yylex_init_extra with the yyextra argument.
  83      */
  84     check_extra(scanner);
  85
  86     testlex(scanner);
  87
  88     testlex_destroy(scanner);
  89     return 0;
  90 }
  91
  92 void *yyalloc(size_t size, yyscan_t scanner)
  93 {
  94     struct Check *check;
  95     check = testget_extra(scanner);
  96
  97     if (!check->bar)
  98         check->bar = "Hello World";
  99
 100     check_extra(scanner);
 101
 102     return malloc(size);
 103 }
 104
 105 /* Save char into junk array at next position. */
 106 static void check_extra(yyscan_t  scanner)
 107 {
 108     struct Check *check;
 109     check = testget_extra(scanner);
 110
 111     if (check->foo != 'a') {
 112         fprintf(stderr, "foo is not 'a'\n");
 113         exit(1);
 114     }
 115     if (strcmp(check->bar, "Hello World") != 0) {
 116         fprintf(stderr, "bar is not Hello World\n");
 117         exit(1);
 118     }
 119     if (check->qux != 'z') {
 120         fprintf(stderr, "qux is not 'z'\n");
 121         exit(1);
 122     }
 123 }