Source code for hemlock.clients.hemlock_runner

#!/usr/bin/env python
#
#   Copyright (c) 2013 In-Q-Tel, Inc/Lab41, All Rights Reserved.
#
#   Licensed under the Apache License, Version 2.0 (the "License");
#   you may not use this file except in compliance with the License.
#   You may obtain a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#   Unless required by applicable law or agreed to in writing, software
#   distributed under the License is distributed on an "AS IS" BASIS,
#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#   See the License for the specific language governing permissions and
#   limitations under the License.

"""
This module gets all needed credentials in preparation for getting data from
clients and into Hemlock's metadata/data store.

Created on 26 August 2013
@author: Charlie Lewis
"""

from hemlock_debugger import Hemlock_Debugger

import hemlock_base

import ast
import datetime
import MySQLdb as mdb
import sys
import time

class Hemlock_Runner():
    """
    This class is responsible for pulling out the credentials from the Hemlock
    server for both the client system and the Hemlock metadata/data store.
    """

    def __init__(self):
        self.log = Hemlock_Debugger()

    def mysql_server(self, debug, server, user, pw, db):
        """
        Connects to the Hemlock MySQL server to get an instance for retrieving
        credentials.

        :param debug: instance of
            :class:`~hemlock.clients.hemlock_debugger.Hemlock_Debugger`
        :param server: server address of the Hemlock MySQL server
        :param user: user account to connect to the Hemlock MySQL server
        :param pw: password of the user account
        :param db: database to connect to in the Hemlock MySQL server
        :return: returns an instance of the mysql connection
        """
        # DEBUG
        # connect to the mysql server
        try:
            m_server = mdb.connect(server, user, pw, db)
        except:
            print "MySQL server failure"
            sys.exit(0)
        return m_server

    def get_creds(self, debug, m_server, client_id, aes_key):
        """
        Retrieves the credentials for both the client system as well as the
        Hemlock server that are stored in the Hemlock MySQL server.

        :param debug: instance of
            :class:`~hemlock.clients.hemlock_debugger.Hemlock_Debugger`
        :param m_server: instance of MySQL connection
        :param client_id: uuid of the client to get credentials for
        :param aes_key: AES key for decrypting the stored credentials
        :return: returns two dictionaries for the client system and Hemlock
            server credentials
        """
        # DEBUG
        # get client_dict
        cur = m_server.cursor()
        data_action = "SELECT AES_DECRYPT(credentials, '"+aes_key+"') from clients where uuid = '"+client_id+"'"
        cur.execute(data_action)
        results = cur.fetchall()
        client_dict = ast.literal_eval(results[0][0])

        # DEBUG
        # get server_dict
        cur = m_server.cursor()
        data_action = "SELECT AES_DECRYPT(credentials, '"+aes_key+"') from hemlock_server"
        cur.execute(data_action)
        results = cur.fetchall()
        server_dict = ast.literal_eval(results[0][0])
        return client_dict, server_dict