def ConfigureSegments(objlist):
if not IsConfigAllowed(objlist): return
stub = l2segment_pb2.L2SegmentStub(HalChannel)
msg = l2segment_pb2.L2SegmentRequestMsg
__config(objlist, msg, stub.L2SegmentCreate)
if GlobalOptions.mbt:
SignalingClient.SendSignalingData(msg.__name__)
SignalingClient.Wait()
return
def ConfigureNetworks(objlist):
if not IsConfigAllowed(objlist): return
stub = nw_pb2.NetworkStub(HalChannel)
msg = nw_pb2.NetworkRequestMsg
__config(objlist, msg, stub.NetworkCreate)
if GlobalOptions.mbt:
SignalingClient.SendSignalingData(msg.__name__)
SignalingClient.Wait()
return
def ConfigureAcls(objlist, update = False):
if not IsConfigAllowed(objlist): return
stub = acl_pb2.AclStub(HalChannel)
api = stub.AclUpdate if update else stub.AclCreate
msg = acl_pb2.AclRequestMsg
__config(objlist, msg, api)
if not update and GlobalOptions.mbt:
SignalingClient.SendSignalingData(msg.__name__)
SignalingClient.Wait()
return
def ConfigureTenants(objlist):
if not IsConfigAllowed(objlist): return
stub = vrf_pb2.VrfStub(HalChannel)
msg = vrf_pb2.VrfRequestMsg
__config(objlist, msg,
stub.VrfCreate)
if GlobalOptions.mbt:
SignalingClient.SendSignalingData(msg.__name__)
SignalingClient.Wait()
return
def ConfigureDevice(objlist, update = False):
if not IsConfigAllowed(objlist): return
stub = nic_pb2.NicStub(HalChannel)
msg = nic_pb2.DeviceRequestMsg
api = stub.DeviceUpdate if update else stub.DeviceCreate
__config(objlist, msg, api)
if not update and GlobalOptions.mbt:
SignalingClient.SendSignalingData(msg.__name__)
SignalingClient.Wait()
return
def ConfigureLifs(objlist, update = False):
if not IsConfigAllowed(objlist): return
stub = interface_pb2.InterfaceStub(HalChannel)
api = stub.LifUpdate if update else stub.LifCreate
msg = interface_pb2.LifRequestMsg
__config(objlist, msg, api)
if not update and GlobalOptions.mbt:
SignalingClient.SendSignalingData(msg.__name__)
SignalingClient.Wait()
return
def ConfigureSecurityGroupPolicies(objlist, update = False):
if not IsConfigAllowed(objlist): return
stub = nwsec_pb2.NwSecurityStub(HalChannel)
api = stub.SecurityPolicyCreate
msg = nwsec_pb2.SecurityPolicyRequestMsg
if update: api = stub.SecurityPolicyUpdate
__config(objlist, msg, api)
if not update and GlobalOptions.mbt:
SignalingClient.SendSignalingData(msg.__name__)
SignalingClient.Wait()
return
def init():
global HalChannel
if IsHalDisabled(): return
if 'MBT_GRPC_PORT' in os.environ: # If MBT toggle mode is enabled
# assert GlobalOptions.mbt
port = os.environ['MBT_GRPC_PORT']
elif 'HAL_GRPC_PORT' in os.environ:
port = os.environ['HAL_GRPC_PORT']
else:
port = '50054'
logger.info("Creating GRPC channel to HAL on port %s" %(port))
server = 'localhost:' + port
if 'HAL_GRPC_IP' in os.environ:
server = os.environ['HAL_GRPC_IP'] + ':' + port
HalChannel = grpc.insecure_channel(server)
logger.info("HAL Server IP and Port = ", server)
logger.info("Waiting for HAL to be ready ...")
grpc.channel_ready_future(HalChannel).result()
logger.info("Connected to HAL!")
if GlobalOptions.mbt:
SignalingClient.Connect()
logger.info("Connected to the Model based tester")
return