Install/Uninstall Flash start certificate

import os

import ssl

from subprocess import Popen, PIPE

import ctypes

import urllib2

​

CretFlag = 1

CretFlag = itsm.getParameter('CerFlag') ## Parameter must be give as integer at CerFlag label ## 1 -> install certificate, 0 -> uninstall certificate

​

​

class disable_file_system_redirection:

    _disable = ctypes.windll.kernel32.Wow64DisableWow64FsRedirection

    _revert = ctypes.windll.kernel32.Wow64RevertWow64FsRedirection

​

    def __enter__(self):

        self.old_value = ctypes.c_long()

        self.success = self._disable(ctypes.byref(self.old_value))

​

    def __exit__(self, type, value, traceback):

        if self.success:

            self._revert(self.old_value)

​

​

def Download(src_path, URL, fp):

    request = urllib2.Request(URL, headers={'User-Agent': "Magic Browser"})

    try:

        gcontext = ssl.SSLContext(ssl.PROTOCOL_TLSv1)

        parsed = urllib2.urlopen(request, context=gcontext)

    except:

        parsed = urllib2.urlopen(request)

    if not os.path.exists(src_path):

        os.makedirs(src_path)

    with open(fp, 'wb') as f:

        while True:

            chunk = parsed.read(100 * 1000 * 1000)

            if chunk:

                f.write(chunk)

            else:

                break

    return fp

​

​

def InstallCert():

    URL = 'https://script-downloads.comodo.com/Certificate/WebFilter.crt'

    Folder = os.environ['programdata'] + r"\extraction_file"

    if not os.path.exists(Folder):

        os.mkdir(Folder)

    fileName = os.path.basename(URL)

    fp = os.path.join(Folder, fileName)

​

    Download(Folder, URL, fp)

    if os.path.exists(fp):

        command = r'certutil -addstore -f root ' + fp

        with disable_file_system_redirection():

            process = Popen(command, shell=True, stdout=PIPE, stderr=PIPE)

        result, err = process.communicate()

        if err:

            print err

        else:

            print result

​

​

def UnInstallCert():

    cmd = r'Get-ChildItem Cert:\LocalMachine\Root\6E9B9304C504AA77130A4A01367EB8F376B1E82F | Remove-Item'

    with disable_file_system_redirection():

        obj = Popen('powershell "' + cmd + '"', shell=True, stdout=PIPE, stderr=PIPE)

    out, err = obj.communicate()

    return err

​

​

if CretFlag == 1:

    InstallCert()

elif CretFlag == 0:

    res = UnInstallCert()

    if res == "":

        print "Flash Start certificate removed"

    else:

        print res

else:

    print "Invalid parameter"