xref: /illumos-gate/usr/src/cmd/make/bin/state.cc (revision 4c28a617e3922d92a58e813a5b955eb526b9c386)
1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License (the "License").
6  * You may not use this file except in compliance with the License.
7  *
8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9  * or http://www.opensolaris.org/os/licensing.
10  * See the License for the specific language governing permissions
11  * and limitations under the License.
12  *
13  * When distributing Covered Code, include this CDDL HEADER in each
14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15  * If applicable, add the following below this CDDL HEADER, with the
16  * fields enclosed by brackets "[]" replaced with your own identifying
17  * information: Portions Copyright [yyyy] [name of copyright owner]
18  *
19  * CDDL HEADER END
20  */
21 /*
22  * Copyright 2004 Sun Microsystems, Inc. All rights reserved.
23  * Use is subject to license terms.
24  */
25 
26 /*
27  *	state.c
28  *
29  *	This file contains the routines that write the .make.state file
30  */
31 
32 /*
33  * Included files
34  */
35 #include <mk/defs.h>
36 #include <mksh/misc.h>		/* errmsg() */
37 #include <setjmp.h>		/* setjmp() */
38 #include <unistd.h>		/* getpid() */
39 #include <errno.h>		/* errno    */
40 #include <locale.h>		/* MB_CUR_MAX    */
41 
42 /*
43  * Defined macros
44  */
45 #define LONGJUMP_VALUE 17
46 #define XFWRITE(string, length, fd) {if (fwrite(string, 1, length, fd) == 0) \
47 					longjmp(long_jump, LONGJUMP_VALUE);}
48 #define XPUTC(ch, fd) { \
49 	if (putc((int) ch, fd) == EOF) \
50 		longjmp(long_jump, LONGJUMP_VALUE); \
51 	}
52 #define XFPUTS(string, fd) fputs(string, fd)
53 
54 /*
55  * typedefs & structs
56  */
57 
58 /*
59  * Static variables
60  */
61 
62 /*
63  * File table of contents
64  */
65 static char * escape_target_name(Name np)
66 {
67 	if(np->dollar) {
68 		int len = strlen(np->string_mb);
69 		char * buff = (char*)malloc(2 * len);
70 		int pos = 0;
71 		wchar_t wc;
72 		int pp = 0;
73 		while(pos < len) {
74 			int n = mbtowc(&wc, np->string_mb + pos, MB_CUR_MAX);
75 			if(n < 0) { // error - this shouldn't happen
76 				(void)free(buff);
77 				return strdup(np->string_mb);
78 			}
79 			if(wc == dollar_char) {
80 				buff[pp] = '\\'; pp++;
81 				buff[pp] = '$'; pp++;
82 			} else {
83 				for(int j=0;j<n;j++) {
84 					buff[pp] = np->string_mb[pos+j]; pp++;
85 				}
86 			}
87 			pos += n;
88 		}
89 		buff[pp] = '\0';
90 		return buff;
91 	} else {
92 		return strdup(np->string_mb);
93 	}
94 }
95 
96 static	void		print_auto_depes(register Dependency dependency, register FILE *fd, register Boolean built_this_run, register int *line_length, register char *target_name, jmp_buf long_jump);
97 
98 /*
99  *	write_state_file(report_recursive, exiting)
100  *
101  *	Write a new version of .make.state
102  *
103  *	Parameters:
104  *		report_recursive	Should only be done at end of run
105  *		exiting			true if called from the exit handler
106  *
107  *	Global variables used:
108  *		built_last_make_run The Name ".BUILT_LAST_MAKE_RUN", written
109  *		command_changed	If no command changed we do not need to write
110  *		current_make_version The Name "<current version>", written
111  *		do_not_exec_rule If -n is on we do not write statefile
112  *		hashtab		The hashtable that contains all names
113  *		keep_state	If .KEEP_STATE is no on we do not write file
114  *		make_state	The Name ".make.state", used for opening file
115  *		make_version	The Name ".MAKE_VERSION", written
116  *		recursive_name	The Name ".RECURSIVE", written
117  *		rewrite_statefile Indicates that something changed
118  */
119 
120 void
121 write_state_file(int, Boolean exiting)
122 {
123 	register FILE		*fd;
124 	int			lock_err;
125 	char			buffer[MAXPATHLEN];
126 	char			make_state_tempfile[MAXPATHLEN];
127 	jmp_buf			long_jump;
128 	register int		attempts = 0;
129 	Name_set::iterator	np, e;
130 	register Property	lines;
131 	register int		m;
132 	Dependency		dependency;
133 	register Boolean	name_printed;
134 	Boolean			built_this_run = false;
135 	char			*target_name;
136 	int			line_length;
137 	register Cmd_line	cp;
138 
139 
140 	if (!rewrite_statefile ||
141 	    !command_changed ||
142 	    !keep_state ||
143 	    do_not_exec_rule ||
144 	    (report_dependencies_level > 0)) {
145 		return;
146 	}
147 	/* Lock the file for writing. */
148  	make_state_lockfile = getmem(strlen(make_state->string_mb) + strlen(".lock") + 1);
149  	(void) sprintf(make_state_lockfile,
150  	               "%s.lock",
151  	               make_state->string_mb);
152 	if (lock_err = file_lock(make_state->string_mb,
153 				 make_state_lockfile,
154 				 (int *) &make_state_locked, 0)) {
155  		retmem_mb(make_state_lockfile);
156 		make_state_lockfile = NULL;
157 
158 		/*
159 		 * We need to make sure that we are not being
160 		 * called by the exit handler so we don't call
161 		 * it again.
162 		 */
163 
164 		if (exiting) {
165 			(void) sprintf(buffer, "%s/.make.state.%d.XXXXXX", tmpdir, getpid());
166 			report_pwd = true;
167 			warning(gettext("Writing to %s"), buffer);
168 			int fdes = mkstemp(buffer);
169 			if ((fdes < 0) || (fd = fdopen(fdes, "w")) == NULL) {
170 				fprintf(stderr,
171 					gettext("Could not open statefile `%s': %s"),
172 					buffer,
173 					errmsg(errno));
174 				return;
175 			}
176 		} else {
177 			report_pwd = true;
178 			fatal(gettext("Can't lock .make.state"));
179 		}
180 	}
181 
182 	(void) sprintf(make_state_tempfile,
183 	               "%s.tmp",
184 	               make_state->string_mb);
185 	/* Delete old temporary statefile (in case it exists) */
186 	(void) unlink(make_state_tempfile);
187 	if ((fd = fopen(make_state_tempfile, "w")) == NULL) {
188 		lock_err = errno; /* Save it! unlink() can change errno */
189 		(void) unlink(make_state_lockfile);
190  		retmem_mb(make_state_lockfile);
191 		make_state_lockfile = NULL;
192 		make_state_locked = false;
193 		fatal(gettext("Could not open temporary statefile `%s': %s"),
194 		      make_state_tempfile,
195 		      errmsg(lock_err));
196 	}
197 	/*
198 	 * Set a trap for failed writes. If a write fails, the routine
199 	 * will try saving the .make.state file under another name in /tmp.
200 	 */
201 	if (setjmp(long_jump)) {
202 		(void) fclose(fd);
203 		if (attempts++ > 5) {
204 			if ((make_state_lockfile != NULL) &&
205 			    make_state_locked) {
206 				(void) unlink(make_state_lockfile);
207  				retmem_mb(make_state_lockfile);
208 				make_state_lockfile = NULL;
209 				make_state_locked = false;
210 			}
211 			fatal(gettext("Giving up on writing statefile"));
212 		}
213 		sleep(10);
214 		(void) sprintf(buffer, "%s/.make.state.%d.XXXXXX", tmpdir, getpid());
215 		int fdes = mkstemp(buffer);
216 		if ((fdes < 0) || (fd = fdopen(fdes, "w")) == NULL) {
217 			fatal(gettext("Could not open statefile `%s': %s"),
218 			      buffer,
219 			      errmsg(errno));
220 		}
221 		warning(gettext("Initial write of statefile failed. Trying again on %s"),
222 			buffer);
223 	}
224 
225 	/* Write the version stamp. */
226 	XFWRITE(make_version->string_mb,
227 		strlen(make_version->string_mb),
228 		fd);
229 	XPUTC(colon_char, fd);
230 	XPUTC(tab_char, fd);
231 	XFWRITE(current_make_version->string_mb,
232 		strlen(current_make_version->string_mb),
233 		fd);
234 	XPUTC(newline_char, fd);
235 
236 	/*
237 	 * Go through all the targets, dump their dependencies and
238 	 * command used.
239 	 */
240 	for (np = hashtab.begin(), e = hashtab.end(); np != e; np++) {
241 		/*
242 		 * If the target has no command used nor dependencies,
243 		 * we can go to the next one.
244 		 */
245 		if ((lines = get_prop(np->prop, line_prop)) == NULL) {
246 			continue;
247 		}
248 		/* If this target is a special target, don't print. */
249 		if (np->special_reader != no_special) {
250 			continue;
251 		}
252 		/*
253 		 * Find out if any of the targets dependencies should
254 		 * be written to .make.state.
255 		 */
256 		for (m = 0, dependency = lines->body.line.dependencies;
257 		     dependency != NULL;
258 		     dependency = dependency->next) {
259 			if (m = !dependency->stale
260 			    && (dependency->name != force)
261 #ifndef PRINT_EXPLICIT_DEPEN
262 			    && dependency->automatic
263 #endif
264 			    ) {
265 				break;
266 			}
267 		}
268 		/* Only print if dependencies listed. */
269 		if (m || (lines->body.line.command_used != NULL)) {
270 			name_printed = false;
271 			/*
272 			 * If this target was built during this make run,
273 			 * we mark it.
274 			 */
275 			built_this_run = false;
276 			if (np->has_built) {
277 				built_this_run = true;
278 				XFWRITE(built_last_make_run->string_mb,
279 					strlen(built_last_make_run->string_mb),
280 					fd);
281 				XPUTC(colon_char, fd);
282 				XPUTC(newline_char, fd);
283 			}
284 			/* If the target has dependencies, we dump them. */
285 			target_name = escape_target_name(np);
286 			if (np->has_long_member_name) {
287 				target_name =
288 				  get_prop(np->prop, long_member_name_prop)
289 				    ->body.long_member_name.member_name->
290 				      string_mb;
291 			}
292 			if (m) {
293 				XFPUTS(target_name, fd);
294 				XPUTC(colon_char, fd);
295 				XFPUTS("\t", fd);
296 				name_printed = true;
297 				line_length = 0;
298 				for (dependency =
299 				     lines->body.line.dependencies;
300 				     dependency != NULL;
301 				     dependency = dependency->next) {
302 					print_auto_depes(dependency,
303 							 fd,
304 							 built_this_run,
305 							 &line_length,
306 							 target_name,
307 							 long_jump);
308 				}
309 				XFPUTS("\n", fd);
310 			}
311 			/* If there is a command used, we dump it. */
312 			if (lines->body.line.command_used != NULL) {
313 				/*
314 				 * Only write the target name if it
315 				 * wasn't done for the dependencies.
316 				 */
317 				if (!name_printed) {
318 					XFPUTS(target_name, fd);
319 					XPUTC(colon_char, fd);
320 					XPUTC(newline_char, fd);
321 				}
322 				/*
323 				 * Write the command lines.
324 				 * Prefix each textual line with a tab.
325 				 */
326 				for (cp = lines->body.line.command_used;
327 				     cp != NULL;
328 				     cp = cp->next) {
329 					char		*csp;
330 					int		n;
331 
332 					XPUTC(tab_char, fd);
333 					if (cp->command_line != NULL) {
334 						for (csp = cp->
335 						           command_line->
336 						           string_mb,
337 						     n = strlen(cp->
338 						                command_line->
339 						                string_mb);
340 						     n > 0;
341 						     n--, csp++) {
342 							XPUTC(*csp, fd);
343 							if (*csp ==
344 							    (int) newline_char) {
345 								XPUTC(tab_char,
346 								      fd);
347 							}
348 						}
349 					}
350 					XPUTC(newline_char, fd);
351 				}
352 			}
353 			(void)free(target_name);
354 		}
355 	}
356 	if (fclose(fd) == EOF) {
357 		longjmp(long_jump, LONGJUMP_VALUE);
358 	}
359 	if (attempts == 0) {
360 		if (unlink(make_state->string_mb) != 0 && errno != ENOENT) {
361 			lock_err = errno; /* Save it! unlink() can change errno */
362 			/* Delete temporary statefile */
363 			(void) unlink(make_state_tempfile);
364 			(void) unlink(make_state_lockfile);
365 	 		retmem_mb(make_state_lockfile);
366 			make_state_lockfile = NULL;
367 			make_state_locked = false;
368 			fatal(gettext("Could not delete old statefile `%s': %s"),
369 			      make_state->string_mb,
370 			      errmsg(lock_err));
371 		}
372 		if (rename(make_state_tempfile, make_state->string_mb) != 0) {
373 			lock_err = errno; /* Save it! unlink() can change errno */
374 			/* Delete temporary statefile */
375 			(void) unlink(make_state_tempfile);
376 			(void) unlink(make_state_lockfile);
377 	 		retmem_mb(make_state_lockfile);
378 			make_state_lockfile = NULL;
379 			make_state_locked = false;
380 			fatal(gettext("Could not rename `%s' to `%s': %s"),
381 			      make_state_tempfile,
382 			      make_state->string_mb,
383 			      errmsg(lock_err));
384 		}
385 	}
386 	if ((make_state_lockfile != NULL) && make_state_locked) {
387 		(void) unlink(make_state_lockfile);
388  		retmem_mb(make_state_lockfile);
389 		make_state_lockfile = NULL;
390 		make_state_locked = false;
391 	}
392 }
393 
394 /*
395  *	print_auto_depes(dependency, fd, built_this_run,
396  *			 line_length, target_name, long_jump)
397  *
398  *	Will print a dependency list for automatic entries.
399  *
400  *	Parameters:
401  *		dependency	The dependency to print
402  *		fd		The file to print it to
403  *		built_this_run	If on we prefix each line with .BUILT_THIS...
404  *		line_length	Pointer to line length var that we update
405  *		target_name	We need this when we restart line
406  *		long_jump	setjmp/longjmp buffer used for IO error action
407  *
408  *	Global variables used:
409  *		built_last_make_run The Name ".BUILT_LAST_MAKE_RUN", written
410  *		force		The Name " FORCE", compared against
411  */
412 static void
413 print_auto_depes(register Dependency dependency, register FILE *fd, register Boolean built_this_run, register int *line_length, register char *target_name, jmp_buf long_jump)
414 {
415 	if (!dependency->automatic ||
416 	    dependency->stale ||
417 	    (dependency->name == force)) {
418 		return;
419 	}
420 	XFWRITE(dependency->name->string_mb,
421 		strlen(dependency->name->string_mb),
422 		fd);
423 	/*
424 	 * Check if the dependency line is too long.
425 	 * If so, break it and start a new one.
426 	 */
427 	if ((*line_length += (int) strlen(dependency->name->string_mb) + 1) > 450) {
428 		*line_length = 0;
429 		XPUTC(newline_char, fd);
430 		if (built_this_run) {
431 			XFPUTS(built_last_make_run->string_mb, fd);
432 			XPUTC(colon_char, fd);
433 			XPUTC(newline_char, fd);
434 		}
435 		XFPUTS(target_name, fd);
436 		XPUTC(colon_char, fd);
437 		XPUTC(tab_char, fd);
438 	} else {
439 		XFPUTS(" ", fd);
440 	}
441 	return;
442 }
443 
444 
445