Bug 143 Update ping pong to use charm
[osm/devops.git] / src / nsd / ping_pong_ns / scripts / start_traffic.py
1 #!/usr/bin/env python3
2
3 ############################################################################
4 # Copyright 2016 RIFT.IO Inc #
5 # #
6 # Licensed under the Apache License, Version 2.0 (the "License"); #
7 # you may not use this file except in compliance with the License. #
8 # You may obtain a copy of the License at #
9 # #
10 # http://www.apache.org/licenses/LICENSE-2.0 #
11 # #
12 # Unless required by applicable law or agreed to in writing, software #
13 # distributed under the License is distributed on an "AS IS" BASIS, #
14 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. #
15 # See the License for the specific language governing permissions and #
16 # limitations under the License. #
17 ############################################################################
18
19
20 import argparse
21 import logging
22 import os
23 import subprocess
24 import sys
25 import time
26
27 import yaml
28
29
30 def start_traffic(yaml_cfg, logger):
31 '''Use curl and set admin status to enable on pong and ping vnfs'''
32
33 curl_fmt = 'curl -D /dev/stdout -H "Accept: application/vnd.yang.data' \
34 '+xml" -H "Content-Type: application/vnd.yang.data+json" ' \
35 '-X POST -d "{{ {data} }}" http://{mgmt_ip}:' \
36 '{mgmt_port}/api/v1/{vnf_type}/{url}'
37
38 def setup_service(mgmt_ip, port, vnf_type, service_ip, service_port):
39 data = '\\"ip\\":\\"{}\\", \\"port\\":5555'.format(service_ip)
40 curl_cmd = curl_fmt.format(
41 mgmt_ip=mgmt_ip,
42 mgmt_port=port,
43 vnf_type=vnf_type,
44 data=data,
45 url='server'
46 )
47
48 logger.debug("Executing cmd: %s", curl_cmd)
49 proc = subprocess.run(curl_cmd, shell=True,
50 stdout=subprocess.PIPE,
51 stderr=subprocess.PIPE)
52
53 logger.debug("Process: {}".format(proc))
54
55 return proc.returncode
56
57 def enable_service(mgmt_ip, port, vnf_type):
58 curl_cmd = curl_fmt.format(
59 mgmt_ip=mgmt_ip,
60 mgmt_port=port,
61 vnf_type=vnf_type,
62 data='\\"enable\\":true',
63 url='adminstatus/state'
64 )
65
66 logger.debug("Executing cmd: %s", curl_cmd)
67 proc = subprocess.run(curl_cmd, shell=True,
68 stdout=subprocess.PIPE,
69 stderr=subprocess.PIPE)
70
71 logger.debug("Process: {}".format(proc))
72
73 return proc.returncode
74
75 # Get port from user parameter
76 service_port = yaml_cfg['parameter']['port']
77
78 service_ip = None
79 # Enable pong service first
80 for index, vnfr in yaml_cfg['vnfr'].items():
81 logger.debug("VNFR {}: {}".format(index, vnfr))
82
83 def get_cp_ip(cp_name):
84 for cp in vnfr['connection_point']:
85 if cp['name'].endswith(cp_name):
86 return cp['ip_address']
87
88 # Check if it is pong vnf
89 if 'pong_vnf' in vnfr['name']:
90 vnf_type = 'pong'
91 mgmt_ip = vnfr['mgmt_ip_address']
92 port = vnfr['mgmt_port']
93 service_ip = get_cp_ip('cp1')
94
95 max_tries = 60
96 tries = 0
97 while tries < max_tries:
98 rc = setup_service(mgmt_ip, port, vnf_type, service_ip, service_port)
99 tries += 1
100 if rc != 0:
101 logger.error("Setup service for pong failed ({}): {}".
102 format(tries, rc))
103 if rc != 7:
104 return rc
105 else:
106 time.sleep(1) # Sleep for 1 seconds
107
108 rc = enable_service(mgmt_ip, port, vnf_type)
109 if rc != 0:
110 logger.error("Enable service for pong failed: {}".
111 format(rc))
112 return rc
113 break
114
115 # Add a delay to provide pong port to come up
116 time.sleep(1)
117
118 # Enable ping service next
119 for index, vnfr in yaml_cfg['vnfr'].items():
120 logger.debug("VNFR {}: {}".format(index, vnfr))
121
122 # Check if it is pong vnf
123 if 'ping_vnf' in vnfr['name']:
124 vnf_type = 'ping'
125 mgmt_ip = vnfr['mgmt_ip_address']
126 port = vnfr['mgmt_port']
127 if service_ip is None:
128 logger.error("Did not find pong ip!!")
129 return 1
130
131 max_tries = 30
132 tries = 0
133 while tries < max_tries:
134 rc = setup_service(mgmt_ip, port, vnf_type, service_ip, service_port)
135 tries += 1
136 if rc != 0:
137 logger.error("Setup service for ping failed ({}): {}".
138 format(tries, rc))
139 if rc != 7:
140 return rc
141 else:
142 time.sleep(1) # Sleep for 1 seconds
143
144 rc = enable_service(mgmt_ip, port, vnf_type)
145 if rc != 0:
146 logger.error("Enable service for ping failed: {}".
147 format(rc))
148 break
149
150 return rc
151
152
153 def main(argv=sys.argv[1:]):
154 try:
155 parser = argparse.ArgumentParser()
156 parser.add_argument("yaml_cfg_file", type=argparse.FileType('r'))
157 parser.add_argument("-q", "--quiet", dest="verbose", action="store_false")
158 args = parser.parse_args()
159
160 run_dir = os.path.join(os.environ['RIFT_INSTALL'], "var/run/rift")
161 if not os.path.exists(run_dir):
162 os.makedirs(run_dir)
163 log_file = "{}/ping_pong_start_traffic-{}.log".format(run_dir, time.strftime("%Y%m%d%H%M%S"))
164
165 # logging.basicConfig(filename=log_file, level=logging.DEBUG)
166 logger = logging.getLogger('ping-pong-start-traffic')
167 logger.setLevel(logging.DEBUG)
168
169 fh = logging.FileHandler(log_file)
170 fh.setLevel(logging.DEBUG)
171
172 ch = logging.StreamHandler()
173 if args.verbose:
174 ch.setLevel(logging.DEBUG)
175 else:
176 ch.setLevel(logging.INFO)
177
178 # create formatter and add it to the handlers
179 formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
180 fh.setFormatter(formatter)
181 ch.setFormatter(formatter)
182 logger.addHandler(fh)
183 logger.addHandler(ch)
184
185 except Exception as e:
186 logger.exception("Exception in {}: {}".format(__file__, e))
187 sys.exit(1)
188
189 try:
190 logger.debug("Input file: {}".format(args.yaml_cfg_file.name))
191 yaml_str = args.yaml_cfg_file.read()
192 yaml_cfg = yaml.load(yaml_str)
193 logger.debug("Input YAML: {}".format(yaml_cfg))
194
195 rc = start_traffic(yaml_cfg, logger)
196 logger.info("Return code: {}".format(rc))
197 sys.exit(rc)
198
199 except Exception as e:
200 logger.exception("Exception in {}: {}".format(__file__, e))
201 sys.exit(1)
202
203 if __name__ == "__main__":
204 main()