Пример #1
0
def remove_file(filename, server=''):
    """
    This module removes the given file
    @paramater:
        * filename - <str> absolute path name of the file to be created
        * server   - <str> (optional) name of the server to write the
                     file. If not given, the function takes the
                     first node from config file
    @Returns: True, on success
              False, on failure
    """
    if server == '':
        server = tc.nodes[0]

    try:
        conn = tc.get_connection(server, 'root')
        if conn == -1:
            tc.logger.error("Unable to get connection to 'root' of node %s" \
                            % server)
            return False

        if conn.modules.os.path.exists(filename):
            conn.modules.os.remove(filename)

    except:
        tc.logger.error("Exception occured while removing file %s" % filename)
        return False

    finally:
        conn.close()

    return True
Пример #2
0
def create_geo_passwordless_ssh(mnode, snode, gsuser=''):
    """
        Sets up the password less ssh between two specified nodes

        Returns True if successfull and False on failure
    """
    if gsuser == '':
        gsuser = '******'
    loc = "/var/lib/glusterd/geo-replication/"
    mconn = tc.get_connection(mnode, user='******')
    sconn = tc.get_connection(snode, user=gsuser)
    if not mconn.modules.os.path.isfile('/root/.ssh/id_rsa'):
        if not mconn.modules.os.path.isfile('%s/secret.pem' % loc):
            tc.logger.debug("id_rsa not present. Generating with gsec_create")
            ret = tc.run(mnode, "gluster system:: execute gsec_create")
            if ret[0] != 0:
                tc.logger.error("Unable to generate the secret pem file")
                return False
        tc.logger.debug("Copying the secret.pem to id_rsa")
        mconn.modules.shutil.copyfile("%s/secret.pem" % loc, \
                "/root/.ssh/id_rsa")
        tc.logger.debug("Copying the secret.pem.pub to id_rsa.pub")
        mconn.modules.shutil.copyfile("%s/secret.pem.pub" % loc, \
                "/root/.ssh/id_rsa.pub")
    try:
        slocal = sconn.modules.os.path.expanduser('~')
        sfh = sconn.builtin.open("%s/.ssh/authorized_keys" % slocal, "a")
        with mconn.builtin.open("/root/.ssh/id_rsa.pub", 'r') as f:
            for line in f:
                sfh.write(line)
    except:
        tc.logger.error("Unable to establish passwordless ssh %s@%s to %s@%s" \
                % ('root', mnode, gsuser, snode))
        return False
    finally:
        sfh.close()
        mconn.close()
        sconn.close()
    tc.logger.debug("Password less ssh setup from %s@%s to %s@%s is %s" \
            % ('root', mnode, gsuser, snode, 'successfull'))
    return True
Пример #3
0
def write_file(filename, file_contents=" ", create_mode='', filesize='', server=''):
    """
    This module writes the file along with file contents
    @paramater:
        * filename - <str> absolute path name of the file to be created
        * file_contents - <str> (optional) file content
        * create_mode - <str> (optional) mode to create the file
        * filesize - <str> (optional) filesize
        * server   - <str> (optional) name of the server to write the
                     file. If not given, the function takes the
                     first node from config file
    @Returns: True, on success
              False, on failure
    """
    if server == '':
        server = tc.nodes[0]

    if create_mode == '':
        create_mode = 'open'

    if create_mode != 'open':
        try:
            conn = tc.get_connection(server, 'root')
            if conn == -1:
                tc.logger.error("Unable to get connection to 'root' of node %s" \
                                % server)
                return False


            if not conn.modules.os.path.exists(conn.modules.os.path.dirname(filename)):
                conn.modules.os.makedirs(conn.modules.os.path.dirname(filename))
        except:
            tc.logger.error("Exception occured while creating directory  for file %s" % filename)
            return False
        finally:
            conn.close()

    if create_mode == 'open':
        try:
            conn = tc.get_connection(server, 'root')
            if conn == -1:
                tc.logger.error("Unable to get connection to 'root' of node %s" \
                                % server)
                return False

            if not conn.modules.os.path.exists(conn.modules.os.path.dirname(filename)):
                conn.modules.os.makedirs(conn.modules.os.path.dirname(filename))

            with conn.builtin.open(filename, 'w') as _filehandle:
                _filehandle.write(file_contents)
        except:
            tc.logger.error("Exception occured while writing file %s" % filename)
            return False

        finally:
            conn.close()
    elif create_mode == 'echo':
        cmd = "echo " + file_contents + " > " + filename
        ret, _, _ = tc.run(server, cmd)
        if ret != 0:
            return False
    elif create_mode == 'touch':
        cmd = "touch " + filename
        ret, _, _ = tc.run(server, cmd)
        if ret != 0:
            return False
    elif create_mode == 'dd':
        if filesize == '':
            tc.logger.error("Invalid argument for dd cmd. Pass correct \
                             number of parameters")
            return False

        cmd = "dd if=/dev/zero of=%s bs=%s count=1" % (filename, filesize)
        ret, _, _ = tc.run(server, cmd)
        if ret != 0:
            return False

    return True