source: branches/2.2/security/crl_admin/crl_admin.py @ 3232

#!/usr/bin/python
# -*- coding: utf-8 -*-

import os, popen2, fcntl, select, time, sys
from string import replace

# Lista com as urls das CRLs, e onde colocar(path) os arquivos obtidos ....
CRL_urls = []
Confs = {}

#Confs['dirtemp']         # path para a pasta temp para conter arquivos auxiliares....
#Confs['CAfile']          # Arquivo com cadeia dos certificados das CAs, para verificacao das CRLs.
#Confs['CRLs']            # path para a pasta onde as CRLs sao salvas
#Confs['arquivos_crls']   # path para o arquivo de configuracao contendo urls das crls e paths onde serao baixadas...
#Confs['log']             # Arquivo onde sera grada log de execucao da qtualizacao/verificacao das crls.
                          # Deixe 'log' igual a vazio para ver as msgs de execucao no terminal.......
#Confs['lenMax']          # Tamanho maximo do arquivo de log de atualização das crls antes do rotate.....
#Confs['bkpNum']          # Numero de arquivos de log de atualização das crls mantidos pelo rotate....

def ler_arquivo_com_configuracao():
    # Esta funcao le o arquivo com configuracao geral(linguagen php) para tratar certs.
    import os,sys
    BASE = os.path.realpath(__file__).split(os.sep + 'security')[0] # BASE igual a pasta inicial(raiz) do Expresso
    os.chdir(BASE + '/security/classes')
    # Esta funcao le o arquivo com configuracao geral(linguagen php) para tratar certs.
    conf_file = BASE + '/security/classes/Verifica_Certificado_conf.php'
    e = open(conf_file)
    r = e.read()
    aux1 = r.split('\n')
    # primeiro recupera BASE ...
    for linha in aux1:
        linha = linha.strip()
        if linha[0:16] == "$GLOBALS['BASE']":
            Confs['BASE'] = BASE
            break
    # Agora os demais ...
    for linha in aux1:
        linha = linha.strip()
        if linha[0:10] == "$GLOBALS['":
          if linha[0:16] != "$GLOBALS['BASE']":
            aux2 = linha.split(';')
            if aux2[0] != '':
                aux2a = aux2[0].split("'")
                aux3 = aux2[0].split("=")
                Confs[aux2a[1]] = aux3[1].replace(' ','')
    # Finalmente trata as ocorrencias de BASE ...
    for chave in Confs.keys():
        if chave != 'BASE':
            aux = Confs[chave].replace("$GLOBALS['BASE'].",Confs['BASE'])
            Confs[chave] = aux.replace("'",'')
    return

def ler_conf():
    # Esta funcao le o arquivo passado como parametro e gera a lista CRL_urls.
    # O arquivo he esperado no formato:
    # url ( url = aponta onde buscar a crl,  uma por linha.
    e = open(Confs['arquivos_crls'])
    r = e.read()
    aux1 = r.split('\n')
    for linha in aux1:
        if linha[0:1] != '#':
            if linha.strip() != '':
                # Faz split com ';' para manter compatibilidade com arquivos formato antigo ...
                CRL_urls.append([linha.split(';')[0].strip(),Confs['CRLs']])
    return


theOutput = [] 

def fazlog(mL,dados):
    for i in dados:
        aux = i.split('\n')
        for x in aux:
            mL.info(x)

def makeNonBlocking(fd): 
    fl = fcntl.fcntl(fd, fcntl.F_GETFL)
    try:
        fcntl.fcntl(fd, fcntl.F_SETFL, fl | os.O_NDELAY)
    except AttributeError:
        fcntl.fcntl(fd, fcntl.F_SETFL, fl | fcntl.FNDELAY)
  
def ExeCmd(command): 
    theOutput = [] 
    child = popen2.Popen3(command, 1)           # Captura a stdout e a stderr do comando... 
    child.tochild.close()                               # Não estamos "falando(escrevendo para a stdin do comando) com o processo filho... 
    outfile = child.fromchild 
    outfd = outfile.fileno() 
    errfile = child.childerr 
    errfd = errfile.fileno() 
    makeNonBlocking(outfd)                              # Não queremos um deadlock !!!!( Dica do Python Cookbook v1)
    makeNonBlocking(errfd) 
    outdata = errdata = '' 
    outeof = erreof = 0 
    while 1: 
        ready = select.select([outfd,errfd],[],[]) # Aguarda a entrada(saida do comando) 
        if outfd in ready[0]: 
           outchunk = outfile.read() 
           if outchunk == '': outeof = 1 
           outdata = outdata + outchunk 
        if errfd in ready[0]: 
           errchunk = errfile.read() 
           if errchunk == '': erreof = 1 
           errdata = errdata + errchunk 
        if outeof and erreof: break 
        select.select([],[],[],.1)                              #Da um tempo para os buffers serem preenchidosl 
    err = child.wait() 
    #if err != 0: 
    #   raise RuntimeError, '%s failed w/ exit code %d\n%s' % (command, err, errdata) 
    theOutput.append(outdata) 
    theOutput.append(errdata) 
    return theOutput 

# No arquivo crl_admin_confg esta o path (arquivo) que aponta para a configuracao das urls(CRL_urls) das crls que devem ser obtidas.
# Este arquivo tambem informa o arquivo onde sera gravada a log(log), e o arquivo contendo 
# a cadeia de certificados para verificar as crls(CAfile).
# Apos o import, executar a funcao ler_conf() .....
#from crl_admin_confg import *
ler_arquivo_com_configuracao()
try:
    ler_arquivo_com_configuracao()
except:
    print 'Erro lendo arquivos de configuracao(ERR-01X)';
    sys.exit(1);

try:
    # A execucao da funcao a seguir carrega a lista das CRLs (CRL_urls) que devem ser processadas...
    ler_conf()
except:
    print 'Erro lendo arquivos de configuracao(ERR-02X)';
    sys.exit(1);

CAfile =  Confs['CAs']
CRLs = Confs['CRLs']
arquivo = Confs['arquivos_crls']
log = Confs['log']      # Deixe log = '' para saida da log na console .....
lenMax = int(Confs['lenMax']) # Tamanho maximo do arquivo de log para iniciar o rotate....
bkpNum = int(Confs['bkpNum']) # Número de arquivos de log mantidos pelo rotate....
import logging
import logging.handlers

# Prepara para fazer a log das atualizações das crls ...
fm = logging.Formatter('%(asctime)s %(levelname)s %(message)s')
mL = logging.getLogger('ML')
mL.setLevel(logging.DEBUG)
hd = logging.handlers.RotatingFileHandler(log, mode='a', maxBytes=lenMax, backupCount=bkpNum)
hd.setFormatter(fm)
mL.addHandler(hd)

if len(sys.argv) > 1:
    if os.path.isfile(sys.argv[1]):
        Confs['arquivos_crls'] = sys.argv[1]
    else:
        mL.critical('Erro lendo arquivo de configuracao' + sys.argv[1] + ': Nao localizado.\n')
        sys.exit(2)

mL.info('Processando arquivo de urls para obter crls: ' +Confs['arquivos_crls'] + ' .')

for crl in CRL_urls:
    arquivo = crl[1].strip() + os.path.split(crl[0])[1].strip()
    url = crl[0].strip()
    # Obtendo o arquivo com wget ...
    mL.info('Buscando a CRL: ' + url)
    saida_wget = ExeCmd("wget --timeout=10 --tries=1 " + url + " -O " + arquivo)
    # Abaixo estamos comandando a execucao "openssl crl" passando o arquivo crl .
    # O resultado esta em saida. Tem de ser 'verify OK' .
    if os.path.exists(arquivo) and os.path.getsize(arquivo) > 0:
        mL.info('Verificando ' + arquivo + '(' + str(os.path.getsize(arquivo)) + ' Bytes)')
        saida = ExeCmd('openssl crl -CAfile ' + CAfile + '  -in ' + arquivo + ' -inform DER -noout')
        # usa a funcao fazlog porque saida he um array.....
        fazlog(mL,saida)
        try:
            aux1 = replace(saida[1],'\n','')
            if not aux1 == 'verify OK':
                mL.critical('Erro verificando a CRL ' + arquivo + '\n')
                try:
                    os.remove(arquivo)
                except:
                    pass
                continue
            # A crl foi verificada e est OK....
            # Atuaiza a data do arquivo...
            saida = ExeCmd('touch ' + arquivo)
        except:
            mL.critical('Erro processando o status da verificacao da CRL ' + arquivo + '\n')
            try:
                os.remove(arquivo)
            except:
                pass
            continue
    else:
            # usa a funcao fazlog porque saida_wget he um array.....
            fazlog(mL,saida_wget)
            mL.critical('Nao foi possivel obter a CRL ' + url + '\n')
            try:
                os.remove(arquivo)
            except:
                pass