30
|
1 /*-
|
|
2 * Copyright (c) 2010 The NetBSD Foundation, Inc.
|
|
3 * All rights reserved.
|
|
4 *
|
|
5 * This code is derived from software contributed to The NetBSD Foundation
|
|
6 * by David A. Holland.
|
|
7 *
|
|
8 * Redistribution and use in source and binary forms, with or without
|
|
9 * modification, are permitted provided that the following conditions
|
|
10 * are met:
|
|
11 * 1. Redistributions of source code must retain the above copyright
|
|
12 * notice, this list of conditions and the following disclaimer.
|
|
13 * 2. Redistributions in binary form must reproduce the above copyright
|
|
14 * notice, this list of conditions and the following disclaimer in the
|
|
15 * documentation and/or other materials provided with the distribution.
|
|
16 *
|
|
17 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
|
|
18 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
|
|
19 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
|
20 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
|
|
21 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
22 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
23 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
24 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
25 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
26 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
27 * POSSIBILITY OF SUCH DAMAGE.
|
|
28 */
|
|
29
|
27
|
30 #include <string.h>
|
20
|
31 #include <unistd.h>
|
|
32 #include <fcntl.h>
|
|
33 #include <err.h>
|
|
34
|
|
35 #include "utils.h"
|
|
36 #include "mode.h"
|
21
|
37 #include "place.h"
|
20
|
38 #include "output.h"
|
|
39
|
|
40 static int outputfd = -1;
|
21
|
41 static bool incomment = false;
|
27
|
42 static char *linebuf;
|
|
43 static size_t linebufpos, linebufmax;
|
|
44 static struct place linebufplace;
|
20
|
45
|
|
46 static
|
|
47 void
|
|
48 output_open(void)
|
|
49 {
|
24
|
50 if (mode.output_file == NULL) {
|
|
51 outputfd = STDOUT_FILENO;
|
|
52 } else {
|
|
53 outputfd = open(mode.output_file, O_WRONLY|O_CREAT|O_TRUNC,
|
|
54 0664);
|
|
55 if (outputfd < 0) {
|
|
56 warn("%s", mode.output_file);
|
|
57 die();
|
|
58 }
|
20
|
59 }
|
|
60 }
|
|
61
|
21
|
62 static
|
20
|
63 void
|
21
|
64 dowrite(const char *buf, size_t len)
|
20
|
65 {
|
|
66 size_t done;
|
|
67 ssize_t result;
|
|
68 static unsigned write_errors = 0;
|
|
69
|
22
|
70 if (!mode.do_output) {
|
|
71 return;
|
|
72 }
|
|
73
|
20
|
74 if (outputfd < 0) {
|
|
75 output_open();
|
|
76 }
|
|
77
|
|
78 done = 0;
|
|
79 while (done < len) {
|
|
80 result = write(outputfd, buf+done, len-done);
|
|
81 if (result == -1) {
|
|
82 warn("%s: write", mode.output_file);
|
|
83 complain_failed();
|
|
84 write_errors++;
|
|
85 if (write_errors > 5) {
|
|
86 warnx("%s: giving up", mode.output_file);
|
|
87 die();
|
|
88 }
|
|
89 /* XXX is this really a good idea? */
|
|
90 sleep(1);
|
|
91 }
|
|
92 done += (size_t)result;
|
|
93 }
|
|
94 }
|
|
95
|
27
|
96
|
|
97 static
|
20
|
98 void
|
27
|
99 filter_output(const struct place *p, const char *buf, size_t len)
|
21
|
100 {
|
|
101 size_t pos, start;
|
|
102 struct place p2;
|
|
103
|
|
104 start = 0;
|
|
105 for (pos = 0; pos < len - 1; pos++) {
|
|
106 if (buf[pos] == '/' && buf[pos+1] == '*') {
|
|
107 if (incomment && warns.nestcomment) {
|
|
108 p2 = *p;
|
|
109 p2.column += pos;
|
|
110 complain(p, "Warning: %c%c within comment",
|
|
111 '/', '*');
|
|
112 if (mode.werror) {
|
|
113 complain_failed();
|
|
114 }
|
|
115 } else if (!incomment) {
|
|
116 if (pos > start) {
|
|
117 dowrite(buf + start, pos - start);
|
|
118 }
|
|
119 start = pos;
|
|
120 pos += 2;
|
|
121 incomment = true;
|
|
122 /* cancel out the loop's pos++ */
|
|
123 pos--;
|
|
124 continue;
|
|
125 }
|
|
126 } else if (buf[pos] == '*' && buf[pos+1] == '/') {
|
|
127 if (incomment) {
|
|
128 pos += 2;
|
|
129 if (mode.output_retain_comments) {
|
|
130 dowrite(buf + start, pos - start);
|
|
131 }
|
|
132 start = pos;
|
|
133 incomment = false;
|
|
134 /* cancel out the loop's pos++ */
|
|
135 pos--;
|
|
136 continue;
|
|
137 }
|
|
138 }
|
|
139 }
|
24
|
140 pos++;
|
|
141
|
|
142 if (pos > start) {
|
|
143 if (!incomment || mode.output_retain_comments) {
|
|
144 dowrite(buf + start, pos - start);
|
|
145 }
|
|
146 }
|
21
|
147 }
|
|
148
|
|
149 void
|
27
|
150 output(const struct place *p, const char *buf, size_t len)
|
|
151 {
|
|
152 if (linebufpos + len > linebufmax) {
|
|
153 if (linebufmax == 0) {
|
|
154 linebufmax = 64;
|
|
155 }
|
|
156 while (linebufpos + len > linebufmax) {
|
|
157 linebufmax *= 2;
|
|
158 }
|
|
159 linebuf = dorealloc(linebuf, linebufmax);
|
|
160 }
|
|
161 if (linebufpos == 0) {
|
|
162 linebufplace = *p;
|
|
163 }
|
|
164 memcpy(linebuf + linebufpos, buf, len);
|
|
165 linebufpos += len;
|
|
166
|
|
167 if (len == 1 && buf[0] == '\n') {
|
|
168 filter_output(&linebufplace, linebuf, linebufpos);
|
|
169 linebufpos = 0;
|
|
170 }
|
|
171 }
|
|
172
|
|
173 void
|
20
|
174 output_eof(void)
|
|
175 {
|
24
|
176 if (mode.output_file != NULL && outputfd >= 0) {
|
20
|
177 close(outputfd);
|
|
178 }
|
|
179 outputfd = -1;
|
|
180 }
|