xref: /netbsd/sbin/dump/itime.c (revision 6550d01e)
1 /*	$NetBSD: itime.c,v 1.18 2010/03/11 01:32:59 christos Exp $	*/
2 
3 /*-
4  * Copyright (c) 1980, 1993
5  *	The Regents of the University of California.  All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  * 3. Neither the name of the University nor the names of its contributors
16  *    may be used to endorse or promote products derived from this software
17  *    without specific prior written permission.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29  * SUCH DAMAGE.
30  */
31 
32 #include <sys/cdefs.h>
33 #ifndef lint
34 #if 0
35 static char sccsid[] = "@(#)itime.c	8.1 (Berkeley) 6/5/93";
36 #else
37 __RCSID("$NetBSD: itime.c,v 1.18 2010/03/11 01:32:59 christos Exp $");
38 #endif
39 #endif /* not lint */
40 
41 #include <sys/param.h>
42 #include <sys/queue.h>
43 #include <sys/time.h>
44 #include <ufs/ufs/dinode.h>
45 
46 #include <protocols/dumprestore.h>
47 
48 #include <errno.h>
49 #include <fcntl.h>
50 #include <stdio.h>
51 #include <stdlib.h>
52 #include <string.h>
53 #include <time.h>
54 #include <unistd.h>
55 
56 #include "dump.h"
57 
58 struct dumptime {
59 	struct	dumpdates dt_value;
60 	SLIST_ENTRY(dumptime) dt_list;
61 };
62 SLIST_HEAD(dthead, dumptime) dthead = SLIST_HEAD_INITIALIZER(dthead);
63 struct	dumpdates **ddatev = 0;
64 int	nddates = 0;
65 
66 static	void dumprecout(FILE *, struct dumpdates *);
67 static	int getrecord(FILE *, struct dumpdates *);
68 static	int makedumpdate(struct dumpdates *, char *);
69 static	void readdumptimes(FILE *);
70 
71 void
72 initdumptimes(void)
73 {
74 	FILE *df;
75 
76 	if ((df = fopen(dumpdates, "r")) == NULL) {
77 		if (errno != ENOENT) {
78 			msg("WARNING: cannot read %s: %s\n", dumpdates,
79 			    strerror(errno));
80 			return;
81 		}
82 		/*
83 		 * Dumpdates does not exist, make an empty one.
84 		 */
85 		msg("WARNING: no file `%s', making an empty one\n", dumpdates);
86 		if ((df = fopen(dumpdates, "w")) == NULL) {
87 			msg("WARNING: cannot create %s: %s\n", dumpdates,
88 			    strerror(errno));
89 			return;
90 		}
91 		(void) fclose(df);
92 		if ((df = fopen(dumpdates, "r")) == NULL) {
93 			quit("cannot read %s even after creating it: %s\n",
94 			    dumpdates, strerror(errno));
95 			/* NOTREACHED */
96 		}
97 	}
98 	(void) flock(fileno(df), LOCK_SH);
99 	readdumptimes(df);
100 	(void) fclose(df);
101 }
102 
103 static void
104 readdumptimes(FILE *df)
105 {
106 	int i;
107 	struct	dumptime *dtwalk;
108 
109 	for (;;) {
110 		dtwalk = (struct dumptime *)xcalloc(1, sizeof(struct dumptime));
111 		if (getrecord(df, &(dtwalk->dt_value)) < 0) {
112 			free(dtwalk);
113 			break;
114 		}
115 		nddates++;
116 		SLIST_INSERT_HEAD(&dthead, dtwalk, dt_list);
117 	}
118 
119 	/*
120 	 *	arrayify the list, leaving enough room for the additional
121 	 *	record that we may have to add to the ddate structure
122 	 */
123 	ddatev = (struct dumpdates **)
124 		xcalloc((unsigned) (nddates + 1), sizeof(struct dumpdates *));
125 	dtwalk = SLIST_FIRST(&dthead);
126 	for (i = nddates - 1; i >= 0; i--, dtwalk = SLIST_NEXT(dtwalk, dt_list))
127 		ddatev[i] = &dtwalk->dt_value;
128 }
129 
130 void
131 getdumptime(void)
132 {
133 	struct dumpdates *ddp;
134 	int i;
135 	char *fname;
136 
137 	fname = disk;
138 #ifdef FDEBUG
139 	msg("Looking for name %s in dumpdates = %s for level = %c\n",
140 		fname, dumpdates, level);
141 #endif
142 	spcl.c_ddate = 0;
143 	lastlevel = '0';
144 
145 	initdumptimes();
146 	/*
147 	 *	Go find the entry with the same name for a lower increment
148 	 *	and older date.  If we are doing a true incremental, then
149 	 *	we can use any level as a ref point.
150 	 */
151 	ITITERATE(i, ddp) {
152 		if (strncmp(fname, ddp->dd_name, sizeof (ddp->dd_name)) != 0)
153 			continue;
154 		/* trueinc: ostensibly could omit the second clause
155 		 * since if trueinc is set, we don't care about the level
156 		 * at all.
157 		 */
158 		/* if ((!trueinc && (ddp->dd_level >= level)) ||
159 		    (trueinc && (ddp->dd_level > level))) */
160 		if (!trueinc && (ddp->dd_level >= level))
161 			continue;
162 		if (ddp->dd_ddate <= iswap32(spcl.c_ddate))
163 			continue;
164 		spcl.c_ddate = iswap32(ddp->dd_ddate);
165 		lastlevel = ddp->dd_level;
166 	}
167 }
168 
169 void
170 putdumptime(void)
171 {
172 	FILE *df;
173 	struct dumpdates *dtwalk, *dtfound;
174 	int i;
175 	int fd;
176 	char *fname;
177 
178 	if(uflag == 0)
179 		return;
180 	if ((df = fopen(dumpdates, "r+")) == NULL)
181 		quit("cannot rewrite %s: %s\n", dumpdates, strerror(errno));
182 	fd = fileno(df);
183 	(void) flock(fd, LOCK_EX);
184 	fname = disk;
185 	free((char *)ddatev);
186 	ddatev = 0;
187 	nddates = 0;
188 	readdumptimes(df);
189 	if (fseek(df, 0L, 0) < 0)
190 		quit("fseek: %s\n", strerror(errno));
191 	spcl.c_ddate = 0;
192 	ITITERATE(i, dtwalk) {
193 		if (strncmp(fname, dtwalk->dd_name,
194 				sizeof (dtwalk->dd_name)) != 0)
195 			continue;
196 		if (dtwalk->dd_level != level)
197 			continue;
198 		goto found;
199 	}
200 	/*
201 	 *	construct the new upper bound;
202 	 *	Enough room has been allocated.
203 	 */
204 	dtwalk = ddatev[nddates] =
205 		(struct dumpdates *)xcalloc(1, sizeof (struct dumpdates));
206 	nddates += 1;
207   found:
208 	(void) strlcpy(dtwalk->dd_name, fname, sizeof(dtwalk->dd_name));
209 	dtwalk->dd_level = level;
210 	dtwalk->dd_ddate = iswap32(spcl.c_date);
211 	dtfound = dtwalk;
212 
213 	ITITERATE(i, dtwalk) {
214 		dumprecout(df, dtwalk);
215 	}
216 	if (fflush(df))
217 		quit("%s: %s\n", dumpdates, strerror(errno));
218 	if (ftruncate(fd, ftell(df)))
219 		quit("ftruncate (%s): %s\n", dumpdates, strerror(errno));
220 	(void) fclose(df);
221 	msg("level %c dump on %s", level,
222 		spcl.c_date == 0 ? "the epoch\n" : ctime(&dtfound->dd_ddate));
223 }
224 
225 static void
226 dumprecout(FILE *file, struct dumpdates *what)
227 {
228 
229 	if (fprintf(file, DUMPOUTFMT,
230 		    what->dd_name,
231 		    what->dd_level,
232 		    ctime(&what->dd_ddate)) < 0)
233 		quit("%s: %s\n", dumpdates, strerror(errno));
234 }
235 
236 int	recno;
237 
238 static int
239 getrecord(FILE *df, struct dumpdates *ddatep)
240 {
241 	char tbuf[BUFSIZ];
242 
243 	recno = 0;
244 	if ( (fgets(tbuf, sizeof (tbuf), df)) != tbuf)
245 		return(-1);
246 	recno++;
247 	if (makedumpdate(ddatep, tbuf) < 0)
248 		msg("Unknown intermediate format in %s, line %d\n",
249 			dumpdates, recno);
250 
251 #ifdef FDEBUG
252 	msg("getrecord: %s %c %s", ddatep->dd_name, ddatep->dd_level,
253 	    ddatep->dd_ddate == 0 ? "the epoch\n" : ctime(&ddatep->dd_ddate));
254 #endif
255 	return(0);
256 }
257 
258 static int
259 makedumpdate(struct dumpdates *ddp, char *tbuf)
260 {
261 	char un_buf[128];
262 
263 	(void) sscanf(tbuf, DUMPINFMT, ddp->dd_name, &ddp->dd_level, un_buf);
264 	ddp->dd_ddate = unctime(un_buf);
265 	if (ddp->dd_ddate < 0)
266 		return(-1);
267 	return(0);
268 }
269