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
main()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
yyalloc(size_t size,yyscan_t scanner)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. */
check_extra(yyscan_t scanner)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 }
124