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