test/vcpe/vcpe.py

   1 #! /usr/bin/python
   2 import sys
   3 import logging
   4 from vcpecommon import *
   5 import sdcutils
   6 import soutils
   7 from datetime import datetime
   8 import preload
   9 import vcpe_custom_service
  10 import csar_parser
  11 import config_sdnc_so
  12 import json
  13
  14
  15 def config_sniro(vcpecommon, vgmux_svc_instance_uuid, vbrg_svc_instance_uuid):
  16     logger = logging.getLogger(__name__)
  17
  18     logger.info('\n----------------------------------------------------------------------------------')
  19     logger.info('Start to config SNIRO homing emulator')
  20
  21     preloader = preload.Preload(vcpecommon)
  22     template_sniro_data = vcpecommon.find_file('sniro_data', 'json', 'preload_templates')
  23     template_sniro_request = vcpecommon.find_file('sniro_request', 'json', 'preload_templates')
  24
  25     vcperescust_csar = vcpecommon.find_file('rescust', 'csar', 'csar')
  26     parser = csar_parser.CsarParser()
  27     parser.parse_csar(vcperescust_csar)
  28     tunnelxconn_ar_name = None
  29     brg_ar_name = None
  30     vgw_name = None
  31     for model in parser.vnf_models:
  32         logger.info('modelCustomizationName = %s', model['modelCustomizationName'])
  33         if 'tunnel' in model['modelCustomizationName'].lower():
  34             logger.info('tunnel is in %s', model['modelCustomizationName'])
  35             tunnelxconn_ar_name = model['modelCustomizationName']
  36         elif 'brg' in model['modelCustomizationName'].lower():
  37             logger.info('brg is in %s', model['modelCustomizationName'])
  38             brg_ar_name = model['modelCustomizationName']
  39         #elif 'vgw' in model['modelCustomizationName']:
  40         else:
  41             vgw_name = model['modelCustomizationName']
  42
  43     if not (tunnelxconn_ar_name and brg_ar_name and vgw_name):
  44         logger.error('Cannot find all names from %s.', vcperescust_csar)
  45         sys.exit()
  46
  47     preloader.preload_sniro(template_sniro_data, template_sniro_request, tunnelxconn_ar_name, vgw_name, brg_ar_name,
  48                             vgmux_svc_instance_uuid, vbrg_svc_instance_uuid)
  49
  50
  51 def create_one_service(vcpecommon, csar_file, vnf_template_file, preload_dict, suffix, heatbridge=False):
  52     """
  53     :return:  service instance UUID
  54     """
  55     so = soutils.SoUtils(vcpecommon, 'v4')
  56     return so.create_entire_service(csar_file, vnf_template_file, preload_dict, suffix, heatbridge)
  57
  58
  59 def deploy_brg_only():
  60     logging.basicConfig(level=logging.INFO, format='%(message)s')
  61     logger = logging.getLogger(__name__)
  62
  63     vcpecommon = VcpeCommon()
  64     preload_dict = vcpecommon.load_preload_data()
  65 #    name_suffix = preload_dict['${brg_bng_net}'].split('_')[-1]
  66     name_suffix = datetime.now().strftime('%Y%m%d%H%M')
  67
  68     # create multiple services based on the pre-determined order
  69     svc_instance_uuid = vcpecommon.load_object(vcpecommon.svc_instance_uuid_file)
  70     for keyword in ['brg']:
  71         heatbridge = 'gmux' == keyword
  72         csar_file = vcpecommon.find_file(keyword, 'csar', 'csar')
  73         vnf_template_file = vcpecommon.find_file(keyword, 'json', 'preload_templates')
  74         vcpecommon.increase_ip_address_or_vni_in_template(vnf_template_file, ['vbrgemu_private_ip_0'])
  75         svc_instance_uuid[keyword] = create_one_service(vcpecommon, csar_file, vnf_template_file, preload_dict,
  76                                                         name_suffix, heatbridge)
  77         if not svc_instance_uuid[keyword]:
  78             sys.exit()
  79
  80     # Setting up SNIRO
  81     config_sniro(vcpecommon, svc_instance_uuid['gmux'], svc_instance_uuid['brg'])
  82
  83
  84 def deploy_infra():
  85     logger = logging.getLogger(__name__)
  86
  87     vcpecommon = VcpeCommon()
  88
  89     # preload all networks
  90     network_template = vcpecommon.find_file('network', 'json', 'preload_templates')
  91     name_suffix = datetime.now().strftime('%Y%m%d%H%M')
  92     preloader = preload.Preload(vcpecommon)
  93     preload_dict = preloader.preload_all_networks(network_template, name_suffix)
  94     logger.debug('Initial preload dictionary:')
  95     logger.debug(json.dumps(preload_dict, indent=4, sort_keys=True))
  96     if not preload_dict:
  97         logger.error("Failed to preload networks.")
  98         sys.exit()
  99     vcpecommon.save_preload_data(preload_dict)
 100
 101     # create multiple services based on the pre-determined order
 102     svc_instance_uuid = {}
 103     for keyword in ['infra', 'bng', 'gmux', 'brg']:
 104         heatbridge = 'gmux' == keyword
 105         csar_file = vcpecommon.find_file(keyword, 'csar', 'csar')
 106         vnf_template_file = vcpecommon.find_file(keyword, 'json', 'preload_templates')
 107         svc_instance_uuid[keyword] = create_one_service(vcpecommon, csar_file, vnf_template_file, preload_dict,
 108                                                         name_suffix, heatbridge)
 109         if not svc_instance_uuid[keyword]:
 110             sys.exit()
 111
 112     vcpecommon.save_object(svc_instance_uuid, vcpecommon.svc_instance_uuid_file)
 113     # Setting up SNIRO
 114     config_sniro(vcpecommon, svc_instance_uuid['gmux'], svc_instance_uuid['brg'])
 115
 116     print('----------------------------------------------------------------------------------------------------')
 117     print('Congratulations! The following have been completed correctly:')
 118     print(' - Infrastructure Service Instantiation: ')
 119     print('     * 4 VMs:      DHCP, AAA, DNS, Web Server')
 120     print('     * 2 Networks: CPE_PUBLIC, CPE_SIGNAL')
 121     print(' - vBNG Service Instantiation: ')
 122     print('     * 1 VM:       vBNG')
 123     print('     * 2 Networks: BRG_BNG, BNG_MUX')
 124     print(' - vGMUX Service Instantiation: ')
 125     print('     * 1 VM:       vGMUX')
 126     print('     * 1 Network:  MUX_GW')
 127     print(' - vBRG Service Instantiation: ')
 128     print('     * 1 VM:       vBRG')
 129     print(' - Adding vGMUX vServer information to AAI.')
 130     print(' - SNIRO Homing Emulator configuration.')
 131
 132
 133 def deploy_custom_service():
 134     nodes = ['brg', 'mux']
 135     vcpecommon = VcpeCommon(nodes)
 136     custom_service = vcpe_custom_service.CustomService(vcpecommon)
 137
 138     # clean up
 139     host_dic = {k: vcpecommon.hosts[k] for k in nodes}
 140     if False:
 141         if not vcpecommon.delete_vxlan_interfaces(host_dic):
 142             sys.exit()
 143         custom_service.del_all_vgw_stacks(vcpecommon.vgw_name_keyword)
 144
 145     #custom_service.clean_up_sdnc()
 146
 147     # create new service
 148     csar_file = vcpecommon.find_file('rescust', 'csar', 'csar')
 149     vgw_template_file = vcpecommon.find_file('vgw', 'json', 'preload_templates')
 150     vgw_gra_template_file = vcpecommon.find_file('gwgra', 'json', 'preload_templates')
 151     preload_dict = vcpecommon.load_preload_data()
 152     custom_service.create_custom_service(csar_file, vgw_template_file, vgw_gra_template_file, preload_dict)
 153
 154
 155 def closed_loop(lossrate=0):
 156     if lossrate > 0:
 157         while 'y' != raw_input('Please enter docker container "drools" in Policy VM and type "policy stop". Then enter y here: ').lower():
 158             continue
 159     nodes = ['brg', 'mux']
 160     logger = logging.getLogger('__name__')
 161     vcpecommon = VcpeCommon(nodes)
 162     logger.info('Cleaning up vGMUX data reporting settings')
 163     vcpecommon.del_vgmux_ves_mode()
 164     time.sleep(2)
 165     vcpecommon.del_vgmux_ves_collector()
 166
 167     logger.info('Starting vGMUX data reporting to DCAE')
 168     time.sleep(2)
 169     vcpecommon.set_vgmux_ves_collector()
 170
 171     logger.info('Setting vGMUX to report packet loss rate: %s', lossrate)
 172     time.sleep(2)
 173     vcpecommon.set_vgmux_packet_loss_rate(lossrate, vcpecommon.load_vgmux_vnf_name())
 174     if lossrate > 0:
 175         print('Please enter docker container "drools" in Policy VM and type "policy start". Then observe vGMUX being restarted.')
 176
 177
 178 def init_so_sdnc():
 179     logger = logging.getLogger('__name__')
 180     vcpecommon = VcpeCommon()
 181     config_sdnc_so.insert_customer_service_to_so(vcpecommon)
 182     #config_sdnc_so.insert_customer_service_to_sdnc(vcpecommon)
 183     vgw_vfmod_name_index=  0
 184     vcpecommon.save_object(vgw_vfmod_name_index, vcpecommon.vgw_vfmod_name_index_file)
 185
 186 def download_vcpe_service_templates():
 187     vcpecommon = VcpeCommon()
 188     sdc = sdcutils.SdcUtils(vcpecommon)
 189     sdc.get_service_list()
 190
 191 def tmp_sniro():
 192     logger = logging.getLogger(__name__)
 193
 194     vcpecommon = VcpeCommon()
 195
 196     svc_instance_uuid = vcpecommon.load_object(vcpecommon.svc_instance_uuid_file)
 197     # Setting up SNIRO
 198     config_sniro(vcpecommon, svc_instance_uuid['gmux'], svc_instance_uuid['brg'])
 199
 200 def test():
 201     vcpecommon = VcpeCommon()
 202     print("oom-k8s-04 public ip: %s" % (vcpecommon.get_vm_public_ip_by_nova('oom-k8s-04')))
 203
 204 if __name__ == '__main__':
 205     logging.basicConfig(level=logging.DEBUG, format='%(message)s')
 206
 207     print('----------------------------------------------------------------------------------------------------')
 208     print(' vcpe.py:            Brief info about this program')
 209 #    print(' vcpe.py sdc:        Onboard VNFs, design and distribute vCPE services (under development)')
 210     print(' vcpe.py init:       Add customer service data to SDNC and SO DBs.')
 211     print(' vcpe.py infra:      Deploy infrastructure, including DHCP, AAA, DNS, Web Server, vBNG, vGMUX, vBRG.')
 212     print(' vcpe.py brg:        Deploy brg only (for testing after infra succeeds).')
 213     print(' vcpe.py customer:   Deploy customer service, including vGW and VxLANs')
 214     print(' vcpe.py loop:       Test closed loop control')
 215     print('----------------------------------------------------------------------------------------------------')
 216
 217     if len(sys.argv) != 2:
 218         sys.exit()
 219
 220     if sys.argv[1] == 'sdc':
 221         print('Under development')
 222     elif sys.argv[1] == 'init':
 223         if 'y' == raw_input('Ready to add customer service data to SDNC and SO DBs? This is needed only once.'
 224                             'y/n: ').lower():
 225             download_vcpe_service_templates()
 226             init_so_sdnc()
 227     elif sys.argv[1] == 'infra':
 228         if 'y' == raw_input('Ready to deploy infrastructure? y/n: ').lower():
 229             deploy_infra()
 230     elif sys.argv[1] == 'customer':
 231         if 'y' == raw_input('Ready to deploy customer service? y/n: ').lower():
 232             deploy_custom_service()
 233     elif sys.argv[1] == 'loop':
 234         closed_loop(22)
 235     elif sys.argv[1] == 'noloss':
 236         closed_loop(0)
 237     elif sys.argv[1] == 'brg':
 238         deploy_brg_only()
 239     elif sys.argv[1] == 'sniro':
 240         tmp_sniro()
 241     elif sys.argv[1] == 'test':
 242         test()
 243