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