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 fix interrupted installations
88 cmd = 'dpkg --configure -a'
89 stdout = ssh_no_error(ssh, cmd, sudo=True)
90 print "###TI {}".format(stdout)
91 # Try to remove installed vpp.* packages
92 cmd = 'apt-get purge -y "vpp.*"'
93 stdout = ssh_no_error(ssh, cmd, sudo=True)
94 print "###TI {}".format(stdout)
96 # Create installation directory on DUT
97 cmd = "mkdir {}".format(install_dir)
98 stdout = ssh_no_error(ssh, cmd)
99 print "###TI {}".format(stdout)
101 # Copy packages from local path to installation dir
103 print "###TI scp: {}".format(deb)
104 ssh.scp(local_path=deb, remote_path=install_dir)
106 cmd = "dpkg -l | grep vpp"
107 ret, _, _ = ssh.exec_command(cmd)
109 # Try to fix interrupted installations
110 cmd = 'dpkg --configure -a'
111 stdout = ssh_no_error(ssh, cmd, sudo=True)
112 print "###TI {}".format(stdout)
113 # Try to remove installed vpp.* packages
114 cmd = 'apt-get purge -y "vpp.*"'
115 stdout = ssh_no_error(ssh, cmd, sudo=True)
116 print "###TI {}".format(stdout)
118 # Installation of VPP deb packages
119 cmd = "dpkg -i --force-all {}/*.deb".format(install_dir)
120 stdout = ssh_no_error(ssh, cmd, sudo=True)
121 print "###TI {}".format(stdout)
123 if __name__ == "__main__":