From 99c01ca8e9ca89be5705102e12cdc29abc1cb8ed Mon Sep 17 00:00:00 2001
From: kshithijiyer <kshithij.ki@gmail.com>
Date: Tue, 15 Dec 2020 16:27:41 +0530
Subject: [Test+Lib] Add test to check rebalance impact on acl

Test case:
1. Create a volume, start it and mount it to a client.
2. Create 10 files on the mount point and set acls on the files.
3. Check the acl value and collect arequal-checksum.
4. Add bricks to the volume and start rebalance.
5. Check the value of acl(it should be same as step 3),
   collect and compare arequal-checksum with the one collected
   in step 3

Additional functions added:
a. set_acl(): Set acl rule on a specific file
b. get_acl(): Get all acl rules set to a file
c. delete_acl(): Delete a specific or all acl rules
                 set on a file

Change-Id: Ia420cbcc8daea272cd4a282ae27d24f13b4991fe
Signed-off-by: kshithijiyer <kshithij.ki@gmail.com>
---
 .../glustolibs/gluster/glusterfile.py              | 76 ++++++++++++++++++++++
 1 file changed, 76 insertions(+)

(limited to 'glustolibs-gluster/glustolibs/gluster/glusterfile.py')

diff --git a/glustolibs-gluster/glustolibs/gluster/glusterfile.py b/glustolibs-gluster/glustolibs/gluster/glusterfile.py
index 227ef6f8d..ee9b6040d 100755
--- a/glustolibs-gluster/glustolibs/gluster/glusterfile.py
+++ b/glustolibs-gluster/glustolibs/gluster/glusterfile.py
@@ -569,6 +569,82 @@ def create_link_file(node, file, link, soft=False):
     return True
 
 
+def set_acl(client, rule, fqpath):
+    """Set acl rule on a specific file
+
+    Args:
+     client(str): Host on which the command is executed.
+     rule(str): The acl rule to be set on the file.
+     fqpath (str): The fully-qualified path to the file.
+
+    Returns:
+     (bool): True if command successful else False.
+    """
+    cmd = "setfacl -m {} {}".format(rule, fqpath)
+    ret, _, _ = g.run(client, cmd)
+    if ret:
+        g.log.error('Failed to set rule {} on file {}'.format(rule, fqpath))
+        return False
+    return True
+
+
+def get_acl(client, path, filename):
+    """Get all acl rules set to a file
+
+    Args:
+     client(str): Host on which the command is executed.
+     path (str): The fully-qualified path to the dir where file is present.
+     filename(str): Name of the file for which rules have to be fetched.
+
+    Returns:
+     (dict): A dictionary with the formatted output of the command.
+     (None): In case of failures
+
+    Example:
+    >>> get_acl('dhcp35-4.lab.eng.blr.redhat.com', '/root/', 'file')
+    {'owner': 'root', 'rules': ['user::rw-', 'user:root:rwx', 'group::r--',
+    'mask::rwx', 'other::r--'], 'group': 'root', 'file': 'file'}
+    """
+    cmd = "cd {};getfacl {}".format(path, filename)
+    ret, out, _ = g.run(client, cmd)
+    if ret:
+        return None
+
+    # Generate a dict out of the output
+    output_dict = {}
+    data = out.strip().split('\n')
+    for key, index in (('file', 0), ('owner', 1), ('group', 2)):
+        output_dict[key] = data[index].split(' ')[2]
+    output_dict['rules'] = data[3:]
+
+    return output_dict
+
+
+def delete_acl(client, fqpath, rule=None):
+    """Delete a specific or all acl rules set on a file
+
+    Args:
+     client(str): Host on which the command is executed.
+     fqpath (str): The fully-qualified path to the file.
+
+    Kwargs:
+     rule(str): The acl rule to be removed from the file.
+
+    Returns:
+     (bool): True if command successful else False.
+    """
+    # Remove all acls set on a file
+    cmd = "setfacl -b {}".format(fqpath)
+    # Remove a specific acl of the file
+    if rule:
+        cmd = "setfacl -x {} {}".format(rule, fqpath)
+
+    ret, _, _ = g.run(client, cmd)
+    if ret:
+        return False
+    return True
+
+
 class GlusterFile(object):
     """Class to handle files specific to Gluster (client and backend)"""
     def __init__(self, host, fqpath):
-- 
cgit