campaign.py

from selenium import webdriver
from time import sleep
from config import CONFIG
from selenium.webdriver.chrome.options import Options
from login import login
import sys, os
import logging
script = os.path.basename(sys.argv[0])
logging.basicConfig(filename='./files/test_logs/%s.log' % script, format='%(asctime)s - %(levelname)s - %(message)s', level=logging.INFO)
logger = logging.getLogger('test')


if CONFIG.WEBDRIVER == "Chrome":
    _chrome_options = Options()
    _chrome_options.add_argument('--start-fullscreen')
    driver = webdriver.Chrome(chrome_options=_chrome_options)

elif CONFIG.WEBDRIVER == "Firefox":

    driver = webdriver.Firefox()

else:

    print('No valid Webdrive was provided. ')

# driver.set_window_size(1440, 1200)
driver.implicitly_wait(CONFIG.waittime)
# sleep(2)

host = CONFIG.HOST

def campaign_create(driver, cname, csource, cdesc, cobjective):

    """
    driver = webdriver
    cname = string for ttp title
    cdesc =  formatted string description for campaign.  Can be very long and detailed.
    cobjective = formatted objective for campaign  Can be very long and detailed.
    """

    success = False

    while success is False:
        try:

            driver.find_element_by_id("primary-nav-create").click()
            success = True

        except Exception as e:

            logger.critical("Button is not there, retry.")

    driver.find_element_by_xpath("(//li[@permissions='campaign'])").click()

    sleep(2)

    driver.find_element_by_id("modal_value").clear()

    driver.find_element_by_id("modal_value").send_keys(cname)

    # change iframe to access description frame.
    driver.switch_to.frame(driver.find_elements_by_tag_name('iframe')[0])

    if len(cdesc) == 0:

        driver.find_element_by_css_selector('body').send_keys(cname)

    else:

        driver.find_element_by_css_selector('body').send_keys(cdesc)

    # return to primary frame.
    driver.switch_to.default_content()

    # change iframe to access Objectives frame.
    driver.switch_to.frame(driver.find_elements_by_tag_name('iframe')[1])

    if len(cobjective) == 0:

        driver.find_element_by_css_selector('body').send_keys(cname)

    else:

        driver.find_element_by_css_selector('body').send_keys(cobjective)

    # return to primary frame.
    driver.switch_to.default_content()

    # Add source
    driver.find_element_by_xpath("//span[@ng-show='!form.swapSourceOption']").click()
    sleep(1)
    driver.find_element_by_xpath("//sources-form/div/div/input").send_keys(csource)

    driver.find_element_by_xpath(
        "//DIV[@class='modal-footer ng-scope']//BUTTON[@type='submit'][@form='addCustomObjectForm']").click()


# Run test steps here

campaigns = [ ['Attack against Threatq.com', 'wannacry attack', 'Gain full access to Threatq', CONFIG.SOURCE],
              ['Attack against Google.com', 'russian bear', 'Gain full access to Threatq', CONFIG.SOURCE],
              ['Attack against Yahoo.com', 'wannacry attack', 'Gain full access to yahoo', CONFIG.SOURCE],
              ['Attack against Amazon.com', 'wannacry attack', 'Gain full access to amazon', CONFIG.SOURCE],
              ['Attack against Oracle.com', 'crazygal attack', 'Gain full access to Oracle', CONFIG.SOURCE],
              ['Attack against Bobshome.com', 'delete all attack', 'Delete HDD @ bobshome', CONFIG.SOURCE],
              ['Attack against teamunicorndogs.com', 'krack attack', 'Gain full access to teamunicorndogs', CONFIG.SOURCE],
              ['Attack against privatenetwork.com', 'wannacry attack', 'Gain full access to privatenetwork', CONFIG.SOURCE]
            ]



# print (host + ' ' + str(driver))

login(logger, host, driver, usr=CONFIG.USER[0], pssword=CONFIG.USER[1], SAML=CONFIG.SAML)
logger.info('Start Campaign create test.')
for campaign in campaigns:

    campaign_create(driver, campaign[0], campaign[3], campaign[1], campaign[2] )

    print(campaign)
logger.info('End Campaign create test.')
driver.quit()