Play viking, get axe ... take axe to useless deadwood that has existed way
[dragonfly.git] / usr.sbin / ctm / ctm / ctm_pass2.c
1 /*
2  * ----------------------------------------------------------------------------
3  * "THE BEER-WARE LICENSE" (Revision 42):
4  * <phk@login.dknet.dk> wrote this file.  As long as you retain this notice you
5  * can do whatever you want with this stuff. If we meet some day, and you think
6  * this stuff is worth it, you can buy me a beer in return.   Poul-Henning Kamp
7  * ----------------------------------------------------------------------------
8  *
9  * $FreeBSD: src/usr.sbin/ctm/ctm/ctm_pass2.c,v 1.18.2.1 2001/07/05 07:46:57 kris Exp $
10  * $DragonFly: src/usr.sbin/ctm/ctm/Attic/ctm_pass2.c,v 1.2 2003/06/17 04:29:53 dillon Exp $
11  *
12  */
13
14 #include "ctm.h"
15 #define BADREAD 32
16
17 /*---------------------------------------------------------------------------*/
18 /* Pass2 -- Validate the incoming CTM-file.
19  */
20
21 int
22 Pass2(FILE *fd)
23 {
24     u_char *p,*q,*md5=0;
25     MD5_CTX ctx;
26     int i,j,sep,cnt,fdesc;
27     u_char *trash=0,*name=0;
28     struct CTM_Syntax *sp;
29     struct stat st;
30     int ret = 0;
31     int match = 0;
32     char md5_1[33];
33     struct CTM_Filter *filter;
34     FILE *ed = NULL;
35     static char *template = NULL;
36
37     if(Verbose>3)
38         printf("Pass2 -- Checking if CTM-patch will apply\n");
39     MD5Init (&ctx);
40
41     GETFIELD(p,' '); if(strcmp("CTM_BEGIN",p)) WRONG
42     GETFIELD(p,' '); if(strcmp(Version,p)) WRONG
43     GETFIELD(p,' '); if(strcmp(Name,p)) WRONG
44     /* XXX Lookup name in /etc/ctm,conf, read stuff */
45     GETFIELD(p,' '); if(strcmp(Nbr,p)) WRONG
46     /* XXX Verify that this is the next patch to apply */
47     GETFIELD(p,' '); if(strcmp(TimeStamp,p)) WRONG
48     GETFIELD(p,'\n'); if(strcmp(Prefix,p)) WRONG
49     /* XXX drop or use ? */
50
51     for(;;) {
52         Delete(trash);
53         Delete(name);
54         Delete(md5);
55         cnt = -1;
56
57         /* if a filter list was specified, check file name against
58            the filters specified 
59            if no filter was given operate on all files. */
60         match = (FilterList ? 
61                     !(FilterList->Action) : CTM_FILTER_ENABLE);
62
63         GETFIELD(p,' ');
64
65         if (p[0] != 'C' || p[1] != 'T' || p[2] != 'M') WRONG
66
67         if(!strcmp(p+3,"_END"))
68             break;
69
70         for(sp=Syntax;sp->Key;sp++)
71             if(!strcmp(p+3,sp->Key))
72                 goto found;
73         WRONG
74     found:
75         for(i=0;(j = sp->List[i]);i++) {
76             if (sp->List[i+1] && (sp->List[i+1] & CTM_F_MASK) != CTM_F_Bytes)
77                 sep = ' ';
78             else
79                 sep = '\n';
80
81             switch (j & CTM_F_MASK) {
82                 case CTM_F_Name:
83                     GETNAMECOPY(name,sep,j,0);
84                     /* If `keep' was specified, we won't remove any files,
85                        so don't check if the file exists */
86                     if (KeepIt &&
87                         (!strcmp(sp->Key,"FR") || !strcmp(sp->Key,"DR"))) {
88                         match = CTM_FILTER_DISABLE;
89                         break;
90                     }
91
92                     for (filter = FilterList; filter; filter = filter->Next)                            if (0 == regexec(&filter->CompiledRegex, name,
93                                     0, 0, 0)) {
94                                     match = filter->Action;
95                             }
96
97                     if (CTM_FILTER_DISABLE == match)
98                             break;      /* should ignore this file */
99
100                     /* XXX Check DR DM rec's for parent-dir */
101                     if(j & CTM_Q_Name_New) {
102                         /* XXX Check DR FR rec's for item */
103                         if(-1 != stat(name,&st)) {
104                             fprintf(stderr,"  %s: %s exists.\n",
105                                 sp->Key,name);
106                             ret |= Exit_Forcible;
107                         }
108                         break;
109                     }
110                     if(-1 == stat(name,&st)) {
111                         fprintf(stderr,"  %s: %s doesn't exist.\n",
112                             sp->Key,name);
113                         if (sp->Key[1] == 'R')
114                             ret |= Exit_Forcible;
115                         else
116                             ret |= Exit_NotOK;
117                         break;
118                     }
119                     if (SetTime && getuid() && (getuid() != st.st_uid)) {
120                             fprintf(stderr,
121                                 "  %s: %s not mine, cannot set time.\n",
122                                 sp->Key,name);
123                             ret |= Exit_NotOK;
124                     }
125                     if (j & CTM_Q_Name_Dir) {
126                         if((st.st_mode & S_IFMT) != S_IFDIR) {
127                             fprintf(stderr,
128                                 "  %s: %s exist, but isn't dir.\n",
129                                 sp->Key,name);
130                             ret |= Exit_NotOK;
131                         }
132                         break;
133                     }
134                     if (j & CTM_Q_Name_File) {
135                         if((st.st_mode & S_IFMT) != S_IFREG) {
136                             fprintf(stderr,
137                                 "  %s: %s exist, but isn't file.\n",
138                                 sp->Key,name);
139                             ret |= Exit_NotOK;
140                         }
141                         break;
142                     }
143                     break;
144                 case CTM_F_Uid:
145                 case CTM_F_Gid:
146                 case CTM_F_Mode:
147                     GETFIELD(p,sep);
148                     break;
149                 case CTM_F_MD5:
150                     if(!name) WRONG
151                     if(j & CTM_Q_MD5_Before) {
152                         char *tmp;
153                         GETFIELD(p,sep);
154                         if(match && (st.st_mode & S_IFMT) == S_IFREG &&
155                           (tmp = MD5File(name,md5_1)) != NULL &&
156                           strcmp(tmp,p)) {
157                             fprintf(stderr,"  %s: %s md5 mismatch.\n",
158                                 sp->Key,name);
159                             GETFIELDCOPY(md5,sep);
160                             if(md5 != NULL && strcmp(tmp,md5) == 0) {
161                                 fprintf(stderr,"  %s: %s already applied.\n",
162                                         sp->Key,name);
163                                 match = CTM_FILTER_DISABLE;
164                             } else if(j & CTM_Q_MD5_Force) {
165                                 if(Force)
166                                     fprintf(stderr,"  Can and will force.\n");
167                                 else
168                                     fprintf(stderr,"  Could have forced.\n");
169                                 ret |= Exit_Forcible;
170                             } else {
171                                 ret |= Exit_NotOK;
172                             }
173                         }
174                         break;
175                     } else if(j & CTM_Q_MD5_After) {
176                         if(md5 == NULL) {
177                             GETFIELDCOPY(md5,sep);
178                         }
179                         break;
180                     }
181                     /* Unqualified MD5 */
182                     WRONG
183                     break;
184                 case CTM_F_Count:
185                     GETBYTECNT(cnt,sep);
186                     break;
187                 case CTM_F_Bytes:
188                     if(cnt < 0) WRONG
189                     GETDATA(trash,cnt);
190                     if (!match)
191                         break;
192                     if (!template) {
193                         if (asprintf(&template, "%s/CTMclientXXXXXX",
194                                 TmpDir) == -1) {
195                             fprintf(stderr, "  %s: malloc failed.\n",
196                                 sp->Key);
197                             ret |= Exit_Mess;
198                             return ret;
199                         }
200                     }
201                     if(!strcmp(sp->Key,"FN")) {
202                         if ((p = strdup(template)) == NULL) {
203                             fprintf(stderr, "  %s: malloc failed.\n",
204                                 sp->Key);
205                             ret |= Exit_Mess;
206                             return ret;
207                         }
208                         if ((fdesc = mkstemp(p)) == -1) {
209                             fprintf(stderr, "  %s: mkstemp failed.\n",
210                                 sp->Key);
211                             ret |= Exit_Mess;
212                             Free(p);
213                             return ret;
214                         }
215                         if (close(fdesc) == -1) {
216                             fprintf(stderr, "  %s: close failed.\n",
217                                 sp->Key);
218                             ret |= Exit_Mess;
219                             unlink(p);
220                             Free(p);
221                             return ret;
222                         }
223                         j = ctm_edit(trash,cnt,name,p);
224                         if(j) {
225                             fprintf(stderr,"  %s: %s edit returned %d.\n",
226                                 sp->Key,name,j);
227                             ret |= j;
228                             unlink(p);
229                             Free(p);
230                             return ret;
231                         } else if(strcmp(md5,MD5File(p,md5_1))) {
232                             fprintf(stderr,"  %s: %s edit fails.\n",
233                                 sp->Key,name);
234                             ret |= Exit_Mess;
235                             unlink(p);
236                             Free(p);
237                             return ret;
238                         }
239                         unlink(p);
240                         Free(p);
241                     } else if (!strcmp(sp->Key,"FE")) {
242                         if ((p = strdup(template)) == NULL) {
243                             fprintf(stderr, "  %s: malloc failed.\n",
244                                 sp->Key);
245                             ret |= Exit_Mess;
246                             return ret;
247                         }
248                         if ((fdesc = mkstemp(p)) == -1) {
249                             fprintf(stderr, "  %s: mkstemp failed.\n",
250                                 sp->Key);
251                             ret |= Exit_Mess;
252                             Free(p);
253                             return ret;
254                         }
255                         if (close(fdesc) == -1) {
256                             fprintf(stderr, "  %s: close failed.\n",
257                                 sp->Key);
258                             ret |= Exit_Mess;
259                             unlink(p);
260                             Free(p);
261                             return ret;
262                         }
263                         ed = popen("ed","w");
264                         if (!ed) {
265                             WRONG
266                         }
267                         fprintf(ed,"e %s\n", name);
268                         if (cnt != fwrite(trash,1,cnt,ed)) {
269                             warn("%s", name);
270                             pclose(ed);
271                             WRONG
272                         }
273                         fprintf(ed,"w %s\n",p);
274                         if (pclose(ed)) {
275                             warn("%s", p);
276                             WRONG
277                         }
278                         if(strcmp(md5,MD5File(p,md5_1))) {
279                             fprintf(stderr,"%s %s MD5 didn't come out right\n",
280                                 sp->Key, name);
281                             WRONG
282                         }
283                         unlink(p);
284                         Free(p);
285                     }
286
287                     break;
288                 default: WRONG
289             }
290         }
291     }
292
293     Delete(trash);
294     Delete(name);
295     Delete(md5);
296
297     q = MD5End (&ctx,md5_1);
298     GETFIELD(p,'\n');                   /* <MD5> */
299     if(strcmp(q,p)) WRONG
300     if (-1 != getc(fd)) WRONG
301     return ret;
302 }