xref: /dragonfly/sbin/dump/itime.c (revision 2cc9c7f7)
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  * 3. 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  * @(#)itime.c	8.1 (Berkeley) 6/5/93
30  * $FreeBSD: src/sbin/dump/itime.c,v 1.3.2.1 2001/08/01 06:29:35 obrien Exp $
31  */
32 
33 #include <sys/param.h>
34 #include <sys/queue.h>
35 #include <sys/time.h>
36 
37 #include <vfs/ufs/dinode.h>
38 #include <vfs/ufs/fs.h>
39 
40 #include <protocols/dumprestore.h>
41 
42 #include <errno.h>
43 #include <fcntl.h>
44 #include <stdio.h>
45 #include <stdlib.h>
46 #include <string.h>
47 #include <unistd.h>
48 
49 #include "dump.h"
50 
51 struct dumptime {
52 	struct	dumpdates dt_value;
53 	SLIST_ENTRY(dumptime) dt_list;
54 };
55 SLIST_HEAD(dthead, dumptime) dthead = SLIST_HEAD_INITIALIZER(dthead);
56 int	nddates = 0;		/* number of records (might be zero) */
57 struct	dumpdates **ddatev;	/* the arrayfied version */
58 const char *dumpdates;		/* name of the file containing dump date info */
59 int	lastlevel;		/* dump level of previous dump */
60 
61 static	void dumprecout(FILE *, struct dumpdates *);
62 static	int getrecord(FILE *, struct dumpdates *);
63 static	int makedumpdate(struct dumpdates *, char *);
64 static	void readdumptimes(FILE *);
65 
66 void
67 initdumptimes(void)
68 {
69 	FILE *df;
70 
71 	if ((df = fopen(dumpdates, "r")) == NULL) {
72 		if (errno != ENOENT) {
73 			msg("WARNING: cannot read %s: %s\n", dumpdates,
74 			    strerror(errno));
75 			return;
76 		}
77 		/*
78 		 * Dumpdates does not exist, make an empty one.
79 		 */
80 		msg("WARNING: no file `%s', making an empty one\n", dumpdates);
81 		if ((df = fopen(dumpdates, "w")) == NULL) {
82 			msg("WARNING: cannot create %s: %s\n", dumpdates,
83 			    strerror(errno));
84 			return;
85 		}
86 		fclose(df);
87 		if ((df = fopen(dumpdates, "r")) == NULL) {
88 			quit("cannot read %s even after creating it: %s\n",
89 			    dumpdates, strerror(errno));
90 			/* NOTREACHED */
91 		}
92 	}
93 	flock(fileno(df), LOCK_SH);
94 	readdumptimes(df);
95 	fclose(df);
96 }
97 
98 static void
99 readdumptimes(FILE *df)
100 {
101 	int i;
102 	struct dumptime *dtwalk;
103 
104 	for (;;) {
105 		dtwalk = (struct dumptime *)calloc(1, sizeof (struct dumptime));
106 		if (getrecord(df, &(dtwalk->dt_value)) < 0)
107 			break;
108 		nddates++;
109 		SLIST_INSERT_HEAD(&dthead, dtwalk, dt_list);
110 	}
111 
112 	/*
113 	 *	arrayify the list, leaving enough room for the additional
114 	 *	record that we may have to add to the ddate structure
115 	 */
116 	ddatev = (struct dumpdates **)
117 		calloc((unsigned) (nddates + 1), sizeof (struct dumpdates *));
118 	dtwalk = SLIST_FIRST(&dthead);
119 	for (i = nddates - 1; i >= 0; i--, dtwalk = SLIST_NEXT(dtwalk, dt_list))
120 		ddatev[i] = &dtwalk->dt_value;
121 }
122 
123 void
124 getdumptime(void)
125 {
126 	struct dumpdates *ddp;
127 	int i;
128 	char *fname;
129 
130 	fname = disk;
131 #ifdef FDEBUG
132 	msg("Looking for name %s in dumpdates = %s for level = %c\n",
133 		fname, dumpdates, level);
134 #endif
135 	spcl.c_ddate = 0;
136 	lastlevel = '0';
137 
138 	initdumptimes();
139 	/*
140 	 *	Go find the entry with the same name for a lower increment
141 	 *	and older date
142 	 */
143 	ITITERATE(i, ddp) {
144 		if (strncmp(fname, ddp->dd_name, sizeof (ddp->dd_name)) != 0)
145 			continue;
146 		if (ddp->dd_level >= level)
147 			continue;
148 		if (ddp->dd_ddate <= spcl.c_ddate)
149 			continue;
150 		spcl.c_ddate = ddp->dd_ddate;
151 		lastlevel = ddp->dd_level;
152 	}
153 }
154 
155 void
156 putdumptime(void)
157 {
158 	FILE *df;
159 	struct dumpdates *dtwalk;
160 	int i;
161 	int fd;
162 	char *fname;
163 
164 	if(uflag == 0)
165 		return;
166 	if ((df = fopen(dumpdates, "r+")) == NULL)
167 		quit("cannot rewrite %s: %s\n", dumpdates, strerror(errno));
168 	fd = fileno(df);
169 	flock(fd, LOCK_EX);
170 	fname = disk;
171 	free((char *)ddatev);
172 	ddatev = NULL;
173 	nddates = 0;
174 	readdumptimes(df);
175 	if (fseek(df, 0L, 0) < 0)
176 		quit("fseek: %s\n", strerror(errno));
177 	spcl.c_ddate = 0;
178 	ITITERATE(i, dtwalk) {
179 		if (strncmp(fname, dtwalk->dd_name,
180 				sizeof (dtwalk->dd_name)) != 0)
181 			continue;
182 		if (dtwalk->dd_level != level)
183 			continue;
184 		goto found;
185 	}
186 	/*
187 	 *	construct the new upper bound;
188 	 *	Enough room has been allocated.
189 	 */
190 	dtwalk = ddatev[nddates] =
191 		(struct dumpdates *)calloc(1, sizeof (struct dumpdates));
192 	nddates += 1;
193   found:
194 	strncpy(dtwalk->dd_name, fname, sizeof (dtwalk->dd_name));
195 	dtwalk->dd_level = level;
196 	dtwalk->dd_ddate = spcl.c_date;
197 
198 	ITITERATE(i, dtwalk) {
199 		dumprecout(df, dtwalk);
200 	}
201 	if (fflush(df))
202 		quit("%s: %s\n", dumpdates, strerror(errno));
203 	if (ftruncate(fd, ftell(df)))
204 		quit("ftruncate (%s): %s\n", dumpdates, strerror(errno));
205 	fclose(df);
206 	msg("level %c dump on %s", level,
207 		spcl.c_date == 0 ? "the epoch\n" : ctime((const time_t *)&spcl.c_date));
208 }
209 
210 static void
211 dumprecout(FILE *file, struct dumpdates *what)
212 {
213 
214 	if (fprintf(file, DUMPOUTFMT,
215 		    what->dd_name,
216 		    what->dd_level,
217 		    ctime(&what->dd_ddate)) < 0)
218 		quit("%s: %s\n", dumpdates, strerror(errno));
219 }
220 
221 int	recno;
222 
223 static int
224 getrecord(FILE *df, struct dumpdates *ddatep)
225 {
226 	char tbuf[BUFSIZ];
227 
228 	recno = 0;
229 	if ( (fgets(tbuf, sizeof (tbuf), df)) != tbuf)
230 		return(-1);
231 	recno++;
232 	if (makedumpdate(ddatep, tbuf) < 0)
233 		msg("Unknown intermediate format in %s, line %d\n",
234 			dumpdates, recno);
235 
236 #ifdef FDEBUG
237 	msg("getrecord: %s %c %s", ddatep->dd_name, ddatep->dd_level,
238 	    ddatep->dd_ddate == 0 ? "the epoch\n" : ctime(&ddatep->dd_ddate));
239 #endif
240 	return(0);
241 }
242 
243 static int
244 makedumpdate(struct dumpdates *ddp, char *tbuf)
245 {
246 	char un_buf[128];
247 
248 	sscanf(tbuf, DUMPINFMT, ddp->dd_name, &ddp->dd_level, un_buf);
249 	ddp->dd_ddate = unctime(un_buf);
250 	if (ddp->dd_ddate < 0)
251 		return(-1);
252 	return(0);
253 }
254