82 lines
2.3 KiB
Python
82 lines
2.3 KiB
Python
# * Licensed to the OpenAirInterface (OAI) Software Alliance under one or more
|
|
# * contributor license agreements. See the NOTICE file distributed with
|
|
# * this work for additional information regarding copyright ownership.
|
|
# * The OpenAirInterface Software Alliance licenses this file to You under
|
|
# * the OAI Public License, Version 1.1 (the "License"); you may not use this file
|
|
# * except in compliance with the License.
|
|
# * You may obtain a copy of the License at
|
|
# *
|
|
# * http://www.openairinterface.org/?page_id=698
|
|
# *
|
|
# * Unless required by applicable law or agreed to in writing, software
|
|
# * distributed under the License is distributed on an "AS IS" BASIS,
|
|
# * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# * See the License for the specific language governing permissions and
|
|
# * limitations under the License.
|
|
# *-------------------------------------------------------------------------------
|
|
# * For more information about the OpenAirInterface (OAI) Software Alliance:
|
|
# * contact@openairinterface.org
|
|
# */
|
|
#---------------------------------------------------------------------
|
|
#
|
|
# Required Python Version
|
|
# Python 3.x
|
|
#
|
|
#---------------------------------------------------------------------
|
|
|
|
#to use isfile
|
|
import os
|
|
import sys
|
|
import logging
|
|
#to create a SSH object locally in the methods
|
|
import sshconnection
|
|
#time.sleep
|
|
import time
|
|
|
|
|
|
import re
|
|
import subprocess
|
|
|
|
from datetime import datetime
|
|
|
|
|
|
class AS_UE:
|
|
|
|
def __init__(self,Module):
|
|
#create attributes as in the UE dictionary
|
|
for k, v in Module.items():
|
|
setattr(self, k, v)
|
|
|
|
|
|
|
|
|
|
#-----------------$
|
|
#PUBLIC Methods$
|
|
#-----------------$
|
|
|
|
def WaitEndScenario(self):
|
|
logging.debug('waiting for scenario duration')
|
|
time.sleep(int(self.Duration))
|
|
|
|
def KillASUE(self):
|
|
mySSH = sshconnection.SSHConnection()
|
|
mySSH.open(self.HostIPAddress, self.HostUsername, self.HostPassword)
|
|
mySSH.command('killall --signal SIGKILL lteue-avx2', '#', 5)
|
|
mySSH.close()
|
|
|
|
def RunScenario(self):
|
|
mySSH = sshconnection.SSHConnection()
|
|
mySSH.open(self.HostIPAddress, self.HostUsername, self.HostPassword)
|
|
|
|
logging.debug("Deleting old artifacts :")
|
|
cmd='rm -rf ' + self.Ping + ' ' + self.UELog
|
|
mySSH.command(cmd,'#',5)
|
|
logging.debug("Running scenario :")
|
|
cmd='echo $USER; nohup '+self.Cmd + ' ' + self.Config + ' &'
|
|
mySSH.command(cmd,'#',5)
|
|
|
|
mySSH.close()
|
|
|
|
|
|
|