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