Kevin Guerineau
138 lines
4.0 KiB
Python
138 lines
4.0 KiB
Python
#!/usr/bin/python3
|
|
|
|
import subprocess
|
|
import jinja2
|
|
import os
|
|
import configparser
|
|
import sys
|
|
from colorama import Fore, Style
|
|
import shutil
|
|
import time
|
|
|
|
config = configparser.ConfigParser()
|
|
config.read('samba-pki-tools.ini')
|
|
|
|
# # write config file
|
|
# jinja_env = jinja2.Environment(loader=jinja2.FileSystemLoader(template_3cx))
|
|
# template = jinja_env.get_template("3CXVoipPhone.j2")
|
|
# template_variables = {
|
|
# "name": "TIS (%s)" % get_current_user(), }
|
|
|
|
|
|
class Printing():
|
|
|
|
def information(string):
|
|
print(Fore.CYAN + '[i] ' + string)
|
|
return print(Style.RESET_ALL)
|
|
|
|
def success(string):
|
|
print(Fore.GREEN + '[+] '+ string)
|
|
return print(Style.RESET_ALL)
|
|
|
|
def warning(string):
|
|
print(Fore.YELLOW + '[!] ' + string)
|
|
return print(Style.RESET_ALL)
|
|
|
|
def error(string):
|
|
print(Fore.RED + '[-] ' + string)
|
|
return print(Style.RESET_ALL)
|
|
|
|
class TisPKI:
|
|
|
|
def pki_dir():
|
|
return config.get('general','pki_dir')
|
|
|
|
def root_ca_certfile():
|
|
return os.path.join(TisPKI.pki_dir(),'certs','root_ca.crt')
|
|
|
|
def root_ca_keyfile():
|
|
return os.path.join(TisPKI.pki_dir(),'private','root_ca.key')
|
|
|
|
def root_ca_crlfile():
|
|
return os.path.join(TisPKI.root_crl_path(),'root_ca.crl')
|
|
|
|
def root_ca_configfile():
|
|
return os.path.join(TisPKI.root_config_path(),'openssl_root_ca.ini')
|
|
|
|
def root_keyout_path():
|
|
return os.path.join(TisPKI.pki_dir(),'private')
|
|
|
|
def root_csr_path():
|
|
return os.path.join(TisPKI.pki_dir(),'csr')
|
|
|
|
def root_cert_path():
|
|
return os.path.join(TisPKI.pki_dir(),'certs')
|
|
|
|
def root_p12_path():
|
|
return os.path.join(TisPKI.pki_dir(),'p12')
|
|
|
|
def root_config_path():
|
|
return os.path.join(TisPKI.pki_dir(),'config')
|
|
|
|
def root_crl_path():
|
|
return os.path.join(TisPKI.pki_dir(),'crl')
|
|
|
|
|
|
def intermediate_ca():
|
|
if config.getboolean('openssl_config','create_intermediate'):
|
|
return True
|
|
|
|
def pki_intermediate_dir(name):
|
|
name = name.replace(' ','_')
|
|
return os.path.join(config.get('general','pki_dir'),f'{name}_intermediate_ca')
|
|
|
|
def intermediate_cert_path(name):
|
|
return os.path.join(TisPKI.pki_intermediate_dir(name),'certs')
|
|
|
|
def intermediate_keyout_path(name):
|
|
return os.path.join(TisPKI.pki_intermediate_dir(name),'private')
|
|
|
|
def intermediate_csr_path(name):
|
|
return os.path.join(TisPKI.pki_intermediate_dir(name),'csr')
|
|
|
|
def intermediate_p12_path(name):
|
|
return os.path.join(TisPKI.pki_intermediate_dir(name),'p12')
|
|
|
|
def intermediate_config_path(name):
|
|
return os.path.join(TisPKI.pki_intermediate_dir(name),'config')
|
|
|
|
def intermediate_crl_path(name):
|
|
return os.path.join(TisPKI.pki_intermediate_dir(name),'crl')
|
|
|
|
def intermediate_ca_certfile(name):
|
|
name = name.replace(' ','_')
|
|
return os.path.join(TisPKI.intermediate_cert_path(name),f'{name}_intermediate_ca.crt')
|
|
|
|
def intermediate_ca_keyfile(name):
|
|
name = name.replace(' ','_')
|
|
return os.path.join(TisPKI.intermediate_keyout_path(name),f'{name}_intermediate_ca.key')
|
|
|
|
def intermediate_ca_crlfile(name):
|
|
name = name.replace(' ','_')
|
|
return os.path.join(TisPKI.intermediate_crl_path(name),f'{name}.crl')
|
|
|
|
|
|
|
|
def check_directories(path,verbose=False):
|
|
if verbose:
|
|
Printing.information('Check directories')
|
|
|
|
directories_list = ['certs','config','crl','private','csr','p12','newcerts']
|
|
|
|
if not os.path.isdir(path):
|
|
if verbose:
|
|
Printing.information(f'Create { path } directory')
|
|
os.makedirs(path)
|
|
|
|
for directory in directories_list:
|
|
directory_path = os.path.join(path,directory)
|
|
if not os.path.isdir(directory_path):
|
|
if verbose:
|
|
Printing.information(f'Create { directory_path } directory')
|
|
os.makedirs(directory_path)
|
|
|
|
if not os.path.isfile(os.path.join(path,'index.txt')):
|
|
with open(os.path.join(path,'index.txt'),'w') as file:
|
|
pass
|
|
|