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