Make inline_getxattr() and inline_listxattr() unified as
iter_inline_xattr(), shared_getxattr() and shared_listxattr() unified as
iter_shared_xattr().

After the unification, both iter_inline_xattr() and iter_shared_xattr()
return 0 on success, and negative error on failure.

One thing worth noting is that, the logic of returning it->buffer_ofs
when there's no shared xattrs in shared_listxattr() is moved to
erofs_listxattr() to make the unification possible.  The only difference
is that, semantically the old behavior will return ENOATTR rather than
it->buffer_ofs if ENOATTR encountered when listxattr is parsing upon a
specific shared xattr, while now the new behavior will return
it->buffer_ofs in this case.  This is not an issue, as listxattr upon a
specific xattr won't return ENOATTR.

Signed-off-by: Jingbo Xu <[email protected]>
---
 fs/erofs/xattr.c | 210 ++++++++++++++++++-----------------------------
 1 file changed, 81 insertions(+), 129 deletions(-)

diff --git a/fs/erofs/xattr.c b/fs/erofs/xattr.c
index 264561ccaa44..196b2eb59e29 100644
--- a/fs/erofs/xattr.c
+++ b/fs/erofs/xattr.c
@@ -7,18 +7,8 @@
 #include <linux/security.h>
 #include "xattr.h"
 
-static inline erofs_blk_t erofs_xattr_blkaddr(struct super_block *sb,
-                                             unsigned int xattr_id)
-{
-       return EROFS_SB(sb)->xattr_blkaddr +
-              erofs_blknr(sb, xattr_id * sizeof(__u32));
-}
-
-static inline unsigned int erofs_xattr_blkoff(struct super_block *sb,
-                                             unsigned int xattr_id)
-{
-       return erofs_blkoff(sb, xattr_id * sizeof(__u32));
-}
+static int erofs_getxattr(struct inode *inode, int index, const char *name,
+                         void *buffer, size_t buffer_size);
 
 struct erofs_xattr_iter {
        struct super_block *sb;
@@ -33,6 +23,8 @@ struct erofs_xattr_iter {
        int index;
        struct qstr name;
        struct dentry *dentry;
+       struct inode *inode;
+       bool getxattr;
 };
 
 static inline int erofs_xattr_iter_fixup(struct erofs_xattr_iter *it)
@@ -168,30 +160,6 @@ struct xattr_iter_handlers {
                      unsigned int len);
 };
 
-static int inline_xattr_iter_begin(struct erofs_xattr_iter *it,
-                                  struct inode *inode)
-{
-       struct erofs_inode *const vi = EROFS_I(inode);
-       unsigned int xattr_header_sz, inline_xattr_ofs;
-
-       xattr_header_sz = sizeof(struct erofs_xattr_ibody_header) +
-                         sizeof(u32) * vi->xattr_shared_count;
-       if (xattr_header_sz >= vi->xattr_isize) {
-               DBG_BUGON(xattr_header_sz > vi->xattr_isize);
-               return -ENOATTR;
-       }
-
-       inline_xattr_ofs = vi->inode_isize + xattr_header_sz;
-
-       it->blkaddr = erofs_blknr(it->sb, erofs_iloc(inode) + inline_xattr_ofs);
-       it->ofs = erofs_blkoff(it->sb, erofs_iloc(inode) + inline_xattr_ofs);
-       it->kaddr = erofs_read_metabuf(&it->buf, inode->i_sb, it->blkaddr,
-                                      EROFS_KMAP);
-       if (IS_ERR(it->kaddr))
-               return PTR_ERR(it->kaddr);
-       return vi->xattr_isize - xattr_header_sz;
-}
-
 /*
  * Regardless of success or failure, `xattr_foreach' will end up with
  * `ofs' pointing to the next xattr item rather than an arbitrary position.
@@ -323,46 +291,6 @@ static const struct xattr_iter_handlers 
find_xattr_handlers = {
        .value = xattr_copyvalue
 };
 
-static int inline_getxattr(struct inode *inode, struct erofs_xattr_iter *it)
-{
-       int ret;
-       unsigned int remaining;
-
-       ret = inline_xattr_iter_begin(it, inode);
-       if (ret < 0)
-               return ret;
-
-       remaining = ret;
-       while (remaining) {
-               ret = xattr_foreach(it, &find_xattr_handlers, &remaining);
-               if (ret != -ENOATTR)
-                       break;
-       }
-       return ret ? ret : it->buffer_ofs;
-}
-
-static int shared_getxattr(struct inode *inode, struct erofs_xattr_iter *it)
-{
-       struct erofs_inode *const vi = EROFS_I(inode);
-       struct super_block *const sb = it->sb;
-       unsigned int i, xsid;
-       int ret = -ENOATTR;
-
-       for (i = 0; i < vi->xattr_shared_count; ++i) {
-               xsid = vi->xattr_shared_xattrs[i];
-               it->blkaddr = erofs_xattr_blkaddr(sb, xsid);
-               it->ofs = erofs_xattr_blkoff(sb, xsid);
-               it->kaddr = erofs_read_metabuf(&it->buf, sb, it->blkaddr, 
EROFS_KMAP);
-               if (IS_ERR(it->kaddr))
-                       return PTR_ERR(it->kaddr);
-
-               ret = xattr_foreach(it, &find_xattr_handlers, NULL);
-               if (ret != -ENOATTR)
-                       break;
-       }
-       return ret ? ret : it->buffer_ofs;
-}
-
 static bool erofs_xattr_user_list(struct dentry *dentry)
 {
        return test_opt(&EROFS_SB(dentry->d_sb)->opt, XATTR_USER);
@@ -373,37 +301,6 @@ static bool erofs_xattr_trusted_list(struct dentry *dentry)
        return capable(CAP_SYS_ADMIN);
 }
 
-static int erofs_getxattr(struct inode *inode, int index, const char *name,
-                         void *buffer, size_t buffer_size)
-{
-       int ret;
-       struct erofs_xattr_iter it;
-
-       if (!name)
-               return -EINVAL;
-       if (strlen(name) > EROFS_NAME_LEN)
-               return -ERANGE;
-
-       ret = erofs_init_inode_xattrs(inode);
-       if (ret)
-               return ret;
-
-       it = (struct erofs_xattr_iter) {
-               .buf         = __EROFS_BUF_INITIALIZER,
-               .sb          = inode->i_sb,
-               .name        = QSTR_INIT(name, strlen(name)),
-               .index       = index,
-               .buffer      = buffer,
-               .buffer_size = buffer_size,
-       };
-
-       ret = inline_getxattr(inode, &it);
-       if (ret == -ENOATTR)
-               ret = shared_getxattr(inode, &it);
-       erofs_put_metabuf(&it.buf);
-       return ret;
-}
-
 static int erofs_xattr_generic_get(const struct xattr_handler *handler,
                                   struct dentry *unused, struct inode *inode,
                                   const char *name, void *buffer, size_t size)
@@ -521,45 +418,96 @@ static const struct xattr_iter_handlers 
list_xattr_handlers = {
        .value = NULL
 };
 
-static int inline_listxattr(struct erofs_xattr_iter *it)
+static int erofs_iter_inline_xattr(struct erofs_xattr_iter *it)
 {
+       struct erofs_inode *const vi = EROFS_I(it->inode);
+       const struct xattr_iter_handlers *op;
+       unsigned int xattr_header_sz, remaining;
+       erofs_off_t pos;
        int ret;
-       unsigned int remaining;
 
-       ret = inline_xattr_iter_begin(it, d_inode(it->dentry));
-       if (ret < 0)
-               return ret;
+       xattr_header_sz = sizeof(struct erofs_xattr_ibody_header) +
+                         sizeof(u32) * vi->xattr_shared_count;
+       if (xattr_header_sz >= vi->xattr_isize) {
+               DBG_BUGON(xattr_header_sz > vi->xattr_isize);
+               return -ENOATTR;
+       }
+
+       pos = erofs_iloc(it->inode) + vi->inode_isize + xattr_header_sz;
+       it->blkaddr = erofs_blknr(it->sb, pos);
+       it->ofs = erofs_blkoff(it->sb, pos);
+       it->kaddr = erofs_read_metabuf(&it->buf, it->sb, it->blkaddr, 
EROFS_KMAP);
+       if (IS_ERR(it->kaddr))
+               return PTR_ERR(it->kaddr);
+
+       remaining = vi->xattr_isize - xattr_header_sz;
+       op = it->getxattr ? &find_xattr_handlers : &list_xattr_handlers;
 
-       remaining = ret;
        while (remaining) {
-               ret = xattr_foreach(it, &list_xattr_handlers, &remaining);
-               if (ret)
+               ret = xattr_foreach(it, op, &remaining);
+               if ((it->getxattr && ret != -ENOATTR) || (!it->getxattr && ret))
                        break;
        }
-       return ret ? ret : it->buffer_ofs;
+       return ret;
 }
 
-static int shared_listxattr(struct erofs_xattr_iter *it)
+static int erofs_iter_shared_xattr(struct erofs_xattr_iter *it)
 {
-       struct inode *const inode = d_inode(it->dentry);
-       struct erofs_inode *const vi = EROFS_I(inode);
+       struct erofs_inode *const vi = EROFS_I(it->inode);
        struct super_block *const sb = it->sb;
+       const struct xattr_iter_handlers *op;
        unsigned int i, xsid;
-       int ret = 0;
+       int ret = -ENOATTR;
+
+       op = it->getxattr ? &find_xattr_handlers : &list_xattr_handlers;
 
        for (i = 0; i < vi->xattr_shared_count; ++i) {
                xsid = vi->xattr_shared_xattrs[i];
-               it->blkaddr = erofs_xattr_blkaddr(sb, xsid);
-               it->ofs = erofs_xattr_blkoff(sb, xsid);
+               it->blkaddr = EROFS_SB(sb)->xattr_blkaddr +
+                             erofs_blknr(sb, xsid * sizeof(__u32));
+               it->ofs = erofs_blkoff(sb, xsid * sizeof(__u32));
                it->kaddr = erofs_read_metabuf(&it->buf, sb, it->blkaddr, 
EROFS_KMAP);
                if (IS_ERR(it->kaddr))
                        return PTR_ERR(it->kaddr);
 
-               ret = xattr_foreach(it, &list_xattr_handlers, NULL);
-               if (ret)
+               ret = xattr_foreach(it, op, NULL);
+               if ((it->getxattr && ret != -ENOATTR) || (!it->getxattr && ret))
                        break;
        }
-       return ret ? ret : it->buffer_ofs;
+       return ret;
+}
+
+static int erofs_getxattr(struct inode *inode, int index, const char *name,
+                         void *buffer, size_t buffer_size)
+{
+       int ret;
+       struct erofs_xattr_iter it;
+
+       if (!name)
+               return -EINVAL;
+       if (strlen(name) > EROFS_NAME_LEN)
+               return -ERANGE;
+
+       ret = erofs_init_inode_xattrs(inode);
+       if (ret)
+               return ret;
+
+       it = (struct erofs_xattr_iter) {
+               .buf         = __EROFS_BUF_INITIALIZER,
+               .sb          = inode->i_sb,
+               .inode       = inode,
+               .name        = QSTR_INIT(name, strlen(name)),
+               .index       = index,
+               .buffer      = buffer,
+               .buffer_size = buffer_size,
+               .getxattr    = true,
+       };
+
+       ret = erofs_iter_inline_xattr(&it);
+       if (ret == -ENOATTR)
+               ret = erofs_iter_shared_xattr(&it);
+       erofs_put_metabuf(&it.buf);
+       return ret ? ret : it.buffer_ofs;
 }
 
 ssize_t erofs_listxattr(struct dentry *dentry,
@@ -578,15 +526,19 @@ ssize_t erofs_listxattr(struct dentry *dentry,
                .buf         = __EROFS_BUF_INITIALIZER,
                .sb          = dentry->d_sb,
                .dentry      = dentry,
+               .inode       = d_inode(dentry),
                .buffer      = buffer,
                .buffer_size = buffer_size,
+               .getxattr    = false,
        };
 
-       ret = inline_listxattr(&it);
-       if (ret >= 0 || ret == -ENOATTR)
-               ret = shared_listxattr(&it);
+       ret = erofs_iter_inline_xattr(&it);
+       if (!ret || ret == -ENOATTR)
+               ret = erofs_iter_shared_xattr(&it);
+       if (ret == -ENOATTR)
+               ret = 0;
        erofs_put_metabuf(&it.buf);
-       return ret;
+       return ret ? ret : it.buffer_ofs;
 }
 
 #ifdef CONFIG_EROFS_FS_POSIX_ACL
-- 
2.19.1.6.gb485710b

Reply via email to