Whamcloud - gitweb
LU-12682 llite: fake symlink type of foreign file/dir
[fs/lustre-release.git] / lustre / utils / lfs_project.c
index 180447a..9f82b74 100644 (file)
@@ -69,27 +69,60 @@ lfs_project_item_alloc(struct list_head *head, const char *pathname)
                return -ENOMEM;
        }
 
-       strncpy(lpi->lpi_pathname, pathname, sizeof(lpi->lpi_pathname));
+       strncpy(lpi->lpi_pathname, pathname, sizeof(lpi->lpi_pathname) - 1);
        list_add_tail(&lpi->lpi_list, head);
 
        return 0;
 }
 
-static int project_get_xattr(const char *pathname, struct fsxattr *fsx)
+static const char *mode_to_type(mode_t mode)
+{
+       switch (mode & S_IFMT) {
+       case S_IFDIR:   return "dir";
+       case S_IFREG:   return "regular";
+       case S_IFLNK:   return "symlink";
+       case S_IFCHR:   return "char device";
+       case S_IFBLK:   return "block device";
+       case S_IFIFO:   return "fifo";
+       case S_IFSOCK:  return "sock";
+       }
+
+       return "unknown";
+}
+
+static int project_get_xattr(const char *pathname, struct fsxattr *fsx,
+                            struct stat *st)
 {
        int ret, fd;
 
-       fd = open(pathname, O_RDONLY | O_NOCTTY);
+       ret = lstat(pathname, st);
+       if (ret) {
+               fprintf(stderr, "%s: failed to stat '%s': %s\n",
+                       progname, pathname, strerror(errno));
+               return -errno;
+       }
+
+       /* currently, only file and dir supported */
+       if (!S_ISREG(st->st_mode) && !S_ISDIR(st->st_mode)) {
+               errno = ENOTSUP;
+               fprintf(stderr, "%s: unable to get xattr for %s '%s': %s\n",
+                               progname, mode_to_type(st->st_mode), pathname,
+                               strerror(errno));
+               return -errno;
+       }
+
+       fd = open(pathname, O_RDONLY | O_NOCTTY | O_NDELAY);
        if (fd < 0) {
                fprintf(stderr, "%s: failed to open '%s': %s\n",
                        progname, pathname, strerror(errno));
                return -errno;
        }
 
-       ret = ioctl(fd, LL_IOC_FSGETXATTR, fsx);
+       ret = ioctl(fd, FS_IOC_FSGETXATTR, fsx);
        if (ret) {
                fprintf(stderr, "%s: failed to get xattr for '%s': %s\n",
                        progname, pathname, strerror(errno));
+               close(fd);
                return -errno;
        }
        return fd;
@@ -99,17 +132,10 @@ static int
 project_check_one(const char *pathname, struct project_handle_control *phc)
 {
        struct fsxattr fsx;
-       struct stat st;
        int ret;
+       struct stat st;
 
-       ret = stat(pathname, &st);
-       if (ret) {
-               fprintf(stderr, "%s: failed to stat '%s': %s\n",
-                       progname, pathname, strerror(errno));
-               return -errno;
-       }
-
-       ret = project_get_xattr(pathname, &fsx);
+       ret = project_get_xattr(pathname, &fsx, &st);
        if (ret < 0)
                return ret;
 
@@ -119,7 +145,8 @@ project_check_one(const char *pathname, struct project_handle_control *phc)
                phc->projid = fsx.fsx_projid;
        }
 
-       if (!(fsx.fsx_xflags & LL_PROJINHERIT_FL)) {
+       if (!(fsx.fsx_xflags & FS_XFLAG_PROJINHERIT) &&
+           S_ISDIR(st.st_mode)) {
                if (!phc->newline) {
                        printf("%s%c", pathname, '\0');
                        goto out;
@@ -145,14 +172,15 @@ static int
 project_list_one(const char *pathname, struct project_handle_control *phc)
 {
        struct fsxattr fsx;
+       struct stat st;
        int ret;
 
-       ret = project_get_xattr(pathname, &fsx);
+       ret = project_get_xattr(pathname, &fsx, &st);
        if (ret < 0)
                return ret;
 
        printf("%5u %c %s\n", fsx.fsx_projid,
-              (fsx.fsx_xflags & LL_PROJINHERIT_FL) ?
+              (fsx.fsx_xflags & FS_XFLAG_PROJINHERIT) ?
                'P' : '-', pathname);
 
        close(ret);
@@ -163,22 +191,23 @@ static int
 project_set_one(const char *pathname, struct project_handle_control *phc)
 {
        struct fsxattr fsx;
+       struct stat st;
        int fd, ret = 0;
 
-       fd = project_get_xattr(pathname, &fsx);
+       fd = project_get_xattr(pathname, &fsx, &st);
        if (fd < 0)
                return fd;
 
        if ((!phc->set_projid || fsx.fsx_projid == phc->projid) &&
-           (!phc->set_inherit || (fsx.fsx_xflags & LL_PROJINHERIT_FL)))
+           (!phc->set_inherit || (fsx.fsx_xflags & FS_XFLAG_PROJINHERIT)))
                goto out;
 
        if (phc->set_inherit)
-               fsx.fsx_xflags |= LL_PROJINHERIT_FL;
+               fsx.fsx_xflags |= FS_XFLAG_PROJINHERIT;
        if (phc->set_projid)
                fsx.fsx_projid = phc->projid;
 
-       ret = ioctl(fd, LL_IOC_FSSETXATTR, &fsx);
+       ret = ioctl(fd, FS_IOC_FSSETXATTR, &fsx);
        if (ret)
                fprintf(stderr, "%s: failed to set xattr for '%s': %s\n",
                        progname, pathname, strerror(errno));
@@ -191,21 +220,22 @@ static int
 project_clear_one(const char *pathname, struct project_handle_control *phc)
 {
        struct fsxattr fsx;
+       struct stat st;
        int ret = 0, fd;
 
-       fd = project_get_xattr(pathname, &fsx);
+       fd = project_get_xattr(pathname, &fsx, &st);
        if (fd < 0)
                return fd;
 
-       if ((!(fsx.fsx_xflags & LL_PROJINHERIT_FL)) &&
+       if ((!(fsx.fsx_xflags & FS_XFLAG_PROJINHERIT)) &&
             (fsx.fsx_projid == 0 || phc->keep_projid))
                goto out;
 
-       fsx.fsx_xflags &= ~LL_PROJINHERIT_FL;
+       fsx.fsx_xflags &= ~FS_XFLAG_PROJINHERIT;
        if (!phc->keep_projid)
                fsx.fsx_projid = 0;
 
-       ret = ioctl(fd, LL_IOC_FSSETXATTR, &fsx);
+       ret = ioctl(fd, FS_IOC_FSSETXATTR, &fsx);
        if (ret)
                fprintf(stderr, "%s: failed to set xattr for '%s': %s\n",
                        progname, pathname, strerror(errno));
@@ -224,6 +254,7 @@ lfs_project_handle_dir(struct list_head *head, const char *pathname,
        struct dirent *ent;
        DIR *dir;
        int ret = 0;
+       int rc;
 
        dir = opendir(pathname);
        if (dir == NULL) {
@@ -233,30 +264,33 @@ lfs_project_handle_dir(struct list_head *head, const char *pathname,
                return ret;
        }
 
-       while (ret == 0 && (ent = readdir(dir)) != NULL) {
+       while ((ent = readdir(dir)) != NULL) {
                /* skip "." and ".." */
                if (strcmp(ent->d_name, ".") == 0 ||
                    strcmp(ent->d_name, "..") == 0)
                        continue;
 
-               if (strlen(ent->d_name) + strlen(pathname) >=
-                   sizeof(fullname) + 1) {
+               if (strlen(ent->d_name) + strlen(pathname) + 1 >=
+                   sizeof(fullname)) {
                        ret = -ENAMETOOLONG;
                        errno = ENAMETOOLONG;
-                       break;
+                       fprintf(stderr, "%s: ignored too long path: %s/%s\n",
+                                       progname, pathname, ent->d_name);
+                       continue;
                }
                snprintf(fullname, PATH_MAX, "%s/%s", pathname,
                         ent->d_name);
 
-               ret = func(fullname, phc);
-               if (phc->recursive && ret == 0 && ent->d_type == DT_DIR)
-                       ret = lfs_project_item_alloc(head, fullname);
+               rc = func(fullname, phc);
+               if (rc && !ret)
+                       ret = rc;
+               if (phc->recursive && ent->d_type == DT_DIR) {
+                       rc = lfs_project_item_alloc(head, fullname);
+                       if (rc && !ret)
+                               ret = rc;
+               }
        }
 
-       if (ret)
-               fprintf(stderr, "%s: failed to handle dir '%s': %s\n",
-                       progname, pathname, strerror(errno));
-
        closedir(dir);
        return ret;
 }
@@ -270,7 +304,7 @@ static int lfs_project_iterate(const char *pathname,
        struct list_head head;
        struct stat st;
        int ret = 0;
-       bool top_dir = true;
+       int rc = 0;
 
        ret = stat(pathname, &st);
        if (ret) {
@@ -296,16 +330,11 @@ static int lfs_project_iterate(const char *pathname,
        while (!list_empty(&head)) {
                lpi = list_entry(head.next, struct lfs_project_item, lpi_list);
                list_del(&lpi->lpi_list);
-               if (ret == 0) {
-                       ret = lfs_project_handle_dir(&head, lpi->lpi_pathname,
-                                                    phc, func);
-                       /* only ignore ENOENT error if this is
-                        * not top directory. */
-                       if (ret == -ENOENT && !top_dir)
-                               ret = 0;
-               }
+               rc = lfs_project_handle_dir(&head, lpi->lpi_pathname,
+                                            phc, func);
+               if (!ret && rc)
+                       ret = rc;
                free(lpi);
-               top_dir = false;
        }
 
        return ret;