usr.bin/diff/diffdir.c

   1 /*      $OpenBSD: diffdir.c,v 1.45 2015/10/05 20:15:00 millert Exp $    */
   2
   3 /*
   4  * Copyright (c) 2003, 2010 Todd C. Miller <Todd.Miller@courtesan.com>
   5  *
   6  * Permission to use, copy, modify, and distribute this software for any
   7  * purpose with or without fee is hereby granted, provided that the above
   8  * copyright notice and this permission notice appear in all copies.
   9  *
  10  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  11  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  12  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  13  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  14  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  15  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  16  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  17  *
  18  * Sponsored in part by the Defense Advanced Research Projects
  19  * Agency (DARPA) and Air Force Research Laboratory, Air Force
  20  * Materiel Command, USAF, under agreement number F39502-99-1-0512.
  21  */
  22
  23 #include <sys/cdefs.h>
  24 __FBSDID("$FreeBSD$");
  25
  26 #include <sys/stat.h>
  27
  28 #include <dirent.h>
  29 #include <err.h>
  30 #include <errno.h>
  31 #include <fnmatch.h>
  32 #include <stdio.h>
  33 #include <stdlib.h>
  34 #include <string.h>
  35 #include <limits.h>
  36
  37 #include "diff.h"
  38
  39 static int selectfile(const struct dirent *);
  40 static void diffit(struct dirent *, char *, size_t, char *, size_t, int);
  41 static void print_only(const char *, size_t, const char *);
  42
  43 #define d_status        d_type          /* we need to store status for -l */
  44
  45 /*
  46  * Diff directory traversal. Will be called recursively if -r was specified.
  47  */
  48 void
  49 diffdir(char *p1, char *p2, int flags)
  50 {
  51         struct dirent *dent1, **dp1, **edp1, **dirp1 = NULL;
  52         struct dirent *dent2, **dp2, **edp2, **dirp2 = NULL;
  53         size_t dirlen1, dirlen2;
  54         char path1[PATH_MAX], path2[PATH_MAX];
  55         int pos;
  56
  57         edp1 = edp2 = NULL;
  58
  59         dirlen1 = strlcpy(path1, *p1 ? p1 : ".", sizeof(path1));
  60         if (dirlen1 >= sizeof(path1) - 1) {
  61                 warnc(ENAMETOOLONG, "%s", p1);
  62                 status |= 2;
  63                 return;
  64         }
  65         if (path1[dirlen1 - 1] != '/') {
  66                 path1[dirlen1++] = '/';
  67                 path1[dirlen1] = '\0';
  68         }
  69         dirlen2 = strlcpy(path2, *p2 ? p2 : ".", sizeof(path2));
  70         if (dirlen2 >= sizeof(path2) - 1) {
  71                 warnc(ENAMETOOLONG, "%s", p2);
  72                 status |= 2;
  73                 return;
  74         }
  75         if (path2[dirlen2 - 1] != '/') {
  76                 path2[dirlen2++] = '/';
  77                 path2[dirlen2] = '\0';
  78         }
  79
  80         /*
  81          * Get a list of entries in each directory, skipping "excluded" files
  82          * and sorting alphabetically.
  83          */
  84         pos = scandir(path1, &dirp1, selectfile, alphasort);
  85         if (pos == -1) {
  86                 if (errno == ENOENT && (Nflag || Pflag)) {
  87                         pos = 0;
  88                 } else {
  89                         warn("%s", path1);
  90                         goto closem;
  91                 }
  92         }
  93         dp1 = dirp1;
  94         edp1 = dirp1 + pos;
  95
  96         pos = scandir(path2, &dirp2, selectfile, alphasort);
  97         if (pos == -1) {
  98                 if (errno == ENOENT && Nflag) {
  99                         pos = 0;
 100                 } else {
 101                         warn("%s", path2);
 102                         goto closem;
 103                 }
 104         }
 105         dp2 = dirp2;
 106         edp2 = dirp2 + pos;
 107
 108         /*
 109          * If we were given a starting point, find it.
 110          */
 111         if (start != NULL) {
 112                 while (dp1 != edp1 && strcmp((*dp1)->d_name, start) < 0)
 113                         dp1++;
 114                 while (dp2 != edp2 && strcmp((*dp2)->d_name, start) < 0)
 115                         dp2++;
 116         }
 117
 118         /*
 119          * Iterate through the two directory lists, diffing as we go.
 120          */
 121         while (dp1 != edp1 || dp2 != edp2) {
 122                 dent1 = dp1 != edp1 ? *dp1 : NULL;
 123                 dent2 = dp2 != edp2 ? *dp2 : NULL;
 124
 125                 pos = dent1 == NULL ? 1 : dent2 == NULL ? -1 :
 126                     ignore_file_case ? strcasecmp(dent1->d_name, dent2->d_name) :
 127                     strcmp(dent1->d_name, dent2->d_name) ;
 128                 if (pos == 0) {
 129                         /* file exists in both dirs, diff it */
 130                         diffit(dent1, path1, dirlen1, path2, dirlen2, flags);
 131                         dp1++;
 132                         dp2++;
 133                 } else if (pos < 0) {
 134                         /* file only in first dir, only diff if -N */
 135                         if (Nflag) {
 136                                 diffit(dent1, path1, dirlen1, path2, dirlen2,
 137                                     flags);
 138                         } else {
 139                                 print_only(path1, dirlen1, dent1->d_name);
 140                                 status |= 1;
 141                         }
 142                         dp1++;
 143                 } else {
 144                         /* file only in second dir, only diff if -N or -P */
 145                         if (Nflag || Pflag)
 146                                 diffit(dent2, path1, dirlen1, path2, dirlen2,
 147                                     flags);
 148                         else {
 149                                 print_only(path2, dirlen2, dent2->d_name);
 150                                 status |= 1;
 151                         }
 152                         dp2++;
 153                 }
 154         }
 155
 156 closem:
 157         if (dirp1 != NULL) {
 158                 for (dp1 = dirp1; dp1 < edp1; dp1++)
 159                         free(*dp1);
 160                 free(dirp1);
 161         }
 162         if (dirp2 != NULL) {
 163                 for (dp2 = dirp2; dp2 < edp2; dp2++)
 164                         free(*dp2);
 165                 free(dirp2);
 166         }
 167 }
 168
 169 /*
 170  * Do the actual diff by calling either diffreg() or diffdir().
 171  */
 172 static void
 173 diffit(struct dirent *dp, char *path1, size_t plen1, char *path2, size_t plen2,
 174     int flags)
 175 {
 176         flags |= D_HEADER;
 177         strlcpy(path1 + plen1, dp->d_name, PATH_MAX - plen1);
 178         if (stat(path1, &stb1) != 0) {
 179                 if (!(Nflag || Pflag) || errno != ENOENT) {
 180                         warn("%s", path1);
 181                         return;
 182                 }
 183                 flags |= D_EMPTY1;
 184                 memset(&stb1, 0, sizeof(stb1));
 185         }
 186
 187         strlcpy(path2 + plen2, dp->d_name, PATH_MAX - plen2);
 188         if (stat(path2, &stb2) != 0) {
 189                 if (!Nflag || errno != ENOENT) {
 190                         warn("%s", path2);
 191                         return;
 192                 }
 193                 flags |= D_EMPTY2;
 194                 memset(&stb2, 0, sizeof(stb2));
 195                 stb2.st_mode = stb1.st_mode;
 196         }
 197         if (stb1.st_mode == 0)
 198                 stb1.st_mode = stb2.st_mode;
 199
 200         if (S_ISDIR(stb1.st_mode) && S_ISDIR(stb2.st_mode)) {
 201                 if (rflag)
 202                         diffdir(path1, path2, flags);
 203                 else
 204                         printf("Common subdirectories: %s and %s\n",
 205                             path1, path2);
 206                 return;
 207         }
 208         if (!S_ISREG(stb1.st_mode) && !S_ISDIR(stb1.st_mode))
 209                 dp->d_status = D_SKIPPED1;
 210         else if (!S_ISREG(stb2.st_mode) && !S_ISDIR(stb2.st_mode))
 211                 dp->d_status = D_SKIPPED2;
 212         else
 213                 dp->d_status = diffreg(path1, path2, flags, 0);
 214         print_status(dp->d_status, path1, path2, "");
 215 }
 216
 217 /*
 218  * Returns 1 if the directory entry should be included in the
 219  * diff, else 0.  Checks the excludes list.
 220  */
 221 static int
 222 selectfile(const struct dirent *dp)
 223 {
 224         struct excludes *excl;
 225
 226         if (dp->d_fileno == 0)
 227                 return (0);
 228
 229         /* always skip "." and ".." */
 230         if (dp->d_name[0] == '.' && (dp->d_name[1] == '\0' ||
 231             (dp->d_name[1] == '.' && dp->d_name[2] == '\0')))
 232                 return (0);
 233
 234         /* check excludes list */
 235         for (excl = excludes_list; excl != NULL; excl = excl->next)
 236                 if (fnmatch(excl->pattern, dp->d_name, FNM_PATHNAME) == 0)
 237                         return (0);
 238
 239         return (1);
 240 }
 241
 242 void
 243 print_only(const char *path, size_t dirlen, const char *entry)
 244 {
 245         if (dirlen > 1)
 246                 dirlen--;
 247         printf("Only in %.*s: %s\n", (int)dirlen, path, entry);
 248 }