3 # Copyright (c) 2016 Cisco and/or its affiliates.
4 # Licensed under the Apache License, Version 2.0 (the "License");
5 # you may not use this file except in compliance with the License.
6 # You may obtain a copy of the License at:
8 # http://www.apache.org/licenses/LICENSE-2.0
10 # Unless required by applicable law or agreed to in writing, software
11 # distributed under the License is distributed on an "AS IS" BASIS,
12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 # See the License for the specific language governing permissions and
14 # limitations under the License.
16 """This script provides copy and installation of VPP build deb packages.
17 As destinations are used all DUT nodes from the topology file."""
23 from resources.libraries.python.ssh import SSH
25 def ssh_no_error(ssh, cmd, sudo=False):
26 """Execute a command over ssh channel, and log and exit if the command
29 :param ssh: SSH() object connected to a node.
30 :param cmd: Command line to execute on remote node.
31 :type ssh: SSH() object
33 :return: stdout from the SSH command.
38 ret, stdo, stde = ssh.exec_command_sudo(cmd)
40 ret, stdo, stde = ssh.exec_command(cmd)
43 print 'Command execution failed: "{}"'.format(cmd)
44 print 'stdout: {0}'.format(stdo)
45 print 'stderr: {0}'.format(stde)
46 raise RuntimeError('Unexpected ssh command failure')
52 """Copy and installation of VPP packages."""
54 parser = argparse.ArgumentParser()
55 parser.add_argument("-t", "--topo", required=True,
57 parser.add_argument("-d", "--directory", required=True,
58 help="Installation directory")
59 parser.add_argument("-p", "--packages", required=False, nargs='+',
60 help="Packages paths to copy")
61 parser.add_argument("-c", "--cancel", help="Cancel installation",
63 args = parser.parse_args()
64 topology_file = args.topo
65 packages = args.packages
66 install_dir = args.directory
67 cancel_installation = args.cancel
69 work_file = open(topology_file)
70 topology = load(work_file.read())['nodes']
74 if topology[node]['type'] == "DUT":
75 print "###TI host: {}".format(topology[node]['host'])
76 ssh.connect(topology[node])
78 if cancel_installation:
79 # Remove installation directory on DUT
80 cmd = "rm -r {}".format(install_dir)
81 stdout = ssh_no_error(ssh, cmd)
82 print "###TI {}".format(stdout)
84 cmd = "dpkg -l | grep vpp"
85 ret, _, _ = ssh.exec_command(cmd)
87 # Try to remove installed vpp.* packages
88 cmd = 'apt-get purge -y "vpp.*"'
89 stdout = ssh_no_error(ssh, cmd, sudo=True)
90 print "###TI {}".format(stdout)
92 # Create installation directory on DUT
93 cmd = "mkdir {}".format(install_dir)
94 stdout = ssh_no_error(ssh, cmd)
95 print "###TI {}".format(stdout)
97 # Copy packages from local path to installation dir
99 print "###TI scp: {}".format(deb)
100 ssh.scp(local_path=deb, remote_path=install_dir)
102 cmd = "dpkg -l | grep vpp"
103 ret, _, _ = ssh.exec_command(cmd)
105 # Try to remove installed vpp.* packages
106 cmd = 'apt-get purge -y "vpp.*"'
107 stdout = ssh_no_error(ssh, cmd, sudo=True)
108 print "###TI {}".format(stdout)
110 # Installation of VPP deb packages
111 cmd = "dpkg -i --force-all {}/*.deb".format(install_dir)
112 stdout = ssh_no_error(ssh, cmd, sudo=True)
113 print "###TI {}".format(stdout)
115 if __name__ == "__main__":