kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 1 | """ |
| 2 | Driver for blank dataplane VMs. Created for SDNIP test. |
| 3 | """ |
timlindberg | ef8d55d | 2013-09-27 12:50:13 -0700 | [diff] [blame] | 4 | import pexpect |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 5 | import sys |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 6 | sys.path.append( "../" ) |
timlindberg | ef8d55d | 2013-09-27 12:50:13 -0700 | [diff] [blame] | 7 | from drivers.common.clidriver import CLI |
| 8 | |
timlindberg | ef8d55d | 2013-09-27 12:50:13 -0700 | [diff] [blame] | 9 | |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 10 | class DPCliDriver( CLI ): |
timlindberg | ef8d55d | 2013-09-27 12:50:13 -0700 | [diff] [blame] | 11 | |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 12 | def __init__( self ): |
| 13 | super( CLI, self ).__init__() |
| 14 | |
| 15 | def connect( self, **connectargs ): |
timlindberg | ef8d55d | 2013-09-27 12:50:13 -0700 | [diff] [blame] | 16 | for key in connectargs: |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 17 | vars( self )[ key ] = connectargs[ key ] |
timlindberg | ef8d55d | 2013-09-27 12:50:13 -0700 | [diff] [blame] | 18 | |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 19 | self.name = self.options[ 'name' ] |
| 20 | self.handle = super( DPCliDriver, self ).connect( user_name=self.user_name, |
| 21 | ip_address=self.ip_address, |
| 22 | port=self.port, |
| 23 | pwd=self.pwd ) |
timlindberg | ef8d55d | 2013-09-27 12:50:13 -0700 | [diff] [blame] | 24 | |
| 25 | if self.handle: |
| 26 | return self.handle |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 27 | else: |
| 28 | main.log.info( "NO HANDLE" ) |
timlindberg | ef8d55d | 2013-09-27 12:50:13 -0700 | [diff] [blame] | 29 | return main.FALSE |
| 30 | |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 31 | def create_interfaces( self, net, number, start ): |
| 32 | """ |
SeanCorcoran | f580d10 | 2013-09-27 15:13:21 -0700 | [diff] [blame] | 33 | Creates a number,specified by 'number,' of subinterfaces on eth0. Ip addresses start at 'net'.'start'.1.1 with a 24 bit netmask. Addresses increment sequentially in the third quad, |
| 34 | therefore all interfaces are in different subnets on the same machine. When the third quad reaches 255, it is reset to 1 and the second quad is incremented. |
| 35 | Every single ip address is placed in a file in /tmp titled 'ip_table{net}.txt' |
| 36 | The file is used by 'pingall_interfaces()' as a fping argument |
| 37 | This method returns true if all interfaces are created without a hitch, and false if a single interface has issues |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 38 | """ |
| 39 | self.handle.sendline( "" ) |
| 40 | self.handle.expect( "\$" ) |
SeanCorcoran | f580d10 | 2013-09-27 15:13:21 -0700 | [diff] [blame] | 41 | |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 42 | self.handle.sendline( "rm /tmp/local_ip.txt" ) |
| 43 | self.handle.expect( "\$" ) |
| 44 | self.handle.sendline( "touch /tmp/local_ip.txt" ) |
| 45 | self.handle.expect( "\$" ) |
timlindberg | ef8d55d | 2013-09-27 12:50:13 -0700 | [diff] [blame] | 46 | |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 47 | main.log.info( "Creating interfaces" ) |
timlindberg | ef8d55d | 2013-09-27 12:50:13 -0700 | [diff] [blame] | 48 | k = 0 |
| 49 | intf = 0 |
| 50 | while number != 0: |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 51 | k = k + 1 |
timlindberg | ef8d55d | 2013-09-27 12:50:13 -0700 | [diff] [blame] | 52 | if k == 256: |
| 53 | k = 1 |
| 54 | start = start + 1 |
| 55 | number = number - 1 |
| 56 | intf = intf + 1 |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 57 | ip = net + "." + str( start ) + "." + str( k ) + ".1" |
| 58 | self.handle.sendline( |
| 59 | "sudo ifconfig eth0:" + str( |
| 60 | intf ) + " " + ip + " netmask 255.255.255.0" ) |
timlindberg | ef8d55d | 2013-09-27 12:50:13 -0700 | [diff] [blame] | 61 | |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 62 | i = self.handle.expect( [ |
| 63 | "\$", |
| 64 | "password", |
| 65 | pexpect.TIMEOUT, |
| 66 | pexpect.EOF ], |
| 67 | timeout=120 ) |
SeanCorcoran | f580d10 | 2013-09-27 15:13:21 -0700 | [diff] [blame] | 68 | |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 69 | if i == 0: |
| 70 | self.handle.sendline( |
| 71 | "echo " + str( ip ) + " >> /tmp/local_ip.txt" ) |
| 72 | self.handle.expect( "\$" ) |
| 73 | elif i == 1: |
| 74 | main.log.info( "Sending sudo password" ) |
| 75 | self.handle.sendline( self.pwd ) |
| 76 | self.handle.expect( "\$" ) |
| 77 | else: |
| 78 | main.log.error( "INTERFACES NOT CREATED" ) |
| 79 | return main.FALSE |
SeanCorcoran | f580d10 | 2013-09-27 15:13:21 -0700 | [diff] [blame] | 80 | |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 81 | def pingall_interfaces( self, netsrc, netstrt, netdst, destlogin, destip ): |
| 82 | """ |
| 83 | Copies the /tmp/ip_table{ net }.txt file from the machine you wish to ping, then runs fping with a source address of { netsrc }.{ netstrt }.1.1 on the copied file. |
SeanCorcoran | f580d10 | 2013-09-27 15:13:21 -0700 | [diff] [blame] | 84 | Check every single response for reachable or unreachable. If all are reachable, function returns true. If a SINGLE host is unreachable, then the function stops and returns false |
| 85 | If fping is not installed, this function will install fping then run the same command |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 86 | """ |
| 87 | self.handle.sendline( "" ) |
| 88 | self.handle.expect( "\$" ) |
SeanCorcoran | f580d10 | 2013-09-27 15:13:21 -0700 | [diff] [blame] | 89 | |
Jon Hall | febb1c7 | 2015-03-05 13:30:09 -0800 | [diff] [blame^] | 90 | self.handle.sendline( "scp " + str( destlogin ) + "@" + |
| 91 | str( destip ) + ":/tmp/local_ip.txt /tmp/ip_table" + |
| 92 | str( netsrc ) + ".txt" ) |
| 93 | |
| 94 | i = self.handle.expect( [ |
| 95 | "100%", |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 96 | "password", |
| 97 | pexpect.TIMEOUT ], |
Jon Hall | febb1c7 | 2015-03-05 13:30:09 -0800 | [diff] [blame^] | 98 | timeout=30 ) |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 99 | |
SeanCorcoran | f580d10 | 2013-09-27 15:13:21 -0700 | [diff] [blame] | 100 | if i == 0: |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 101 | main.log.info( "Copied ping file successfully" ) |
SeanCorcoran | f580d10 | 2013-09-27 15:13:21 -0700 | [diff] [blame] | 102 | elif i == 1: |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 103 | self.handle.sendline( self.pwd ) |
| 104 | self.handle.expect( "100%" ) |
| 105 | main.log.info( "Copied ping file successfully" ) |
SeanCorcoran | f580d10 | 2013-09-27 15:13:21 -0700 | [diff] [blame] | 106 | elif i == 2: |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 107 | main.log.error( "COULD NOT COPY PING FILE FROM " + str( destip ) ) |
SeanCorcoran | f580d10 | 2013-09-27 15:13:21 -0700 | [diff] [blame] | 108 | result = main.FALSE |
| 109 | return result |
SeanCorcoran | f580d10 | 2013-09-27 15:13:21 -0700 | [diff] [blame] | 110 | |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 111 | self.handle.sendline( "" ) |
| 112 | self.handle.expect( "\$" ) |
| 113 | |
| 114 | main.log.info( "Pinging interfaces on the " + str( netdst ) + |
Jon Hall | febb1c7 | 2015-03-05 13:30:09 -0800 | [diff] [blame^] | 115 | " network from " + str( netsrc ) + "." + |
| 116 | str( netstrt ) + ".1.1" ) |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 117 | self.handle.sendline( "sudo fping -S " + str( netsrc ) + "." + |
Jon Hall | febb1c7 | 2015-03-05 13:30:09 -0800 | [diff] [blame^] | 118 | str( netstrt ) + ".1.1 -f /tmp/ip_table" + |
| 119 | str( netdst ) + ".txt" ) |
timlindberg | ef8d55d | 2013-09-27 12:50:13 -0700 | [diff] [blame] | 120 | while 1: |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 121 | i = self.handle.expect( [ |
Jon Hall | febb1c7 | 2015-03-05 13:30:09 -0800 | [diff] [blame^] | 122 | "reachable", |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 123 | "unreachable", |
| 124 | "\$", |
| 125 | "password", |
| 126 | pexpect.TIMEOUT, |
| 127 | "not installed" ], |
Jon Hall | febb1c7 | 2015-03-05 13:30:09 -0800 | [diff] [blame^] | 128 | timeout=45 ) |
timlindberg | ef8d55d | 2013-09-27 12:50:13 -0700 | [diff] [blame] | 129 | if i == 0: |
| 130 | result = main.TRUE |
| 131 | elif i == 1: |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 132 | main.log.error( "An interface was unreachable" ) |
timlindberg | ef8d55d | 2013-09-27 12:50:13 -0700 | [diff] [blame] | 133 | result = main.FALSE |
| 134 | return result |
| 135 | elif i == 2: |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 136 | main.log.info( "All interfaces reachable" ) |
Jon Hall | febb1c7 | 2015-03-05 13:30:09 -0800 | [diff] [blame^] | 137 | result = main.FALSE |
timlindberg | ef8d55d | 2013-09-27 12:50:13 -0700 | [diff] [blame] | 138 | return result |
| 139 | elif i == 3: |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 140 | self.handle.sendline( self.pwd ) |
timlindberg | ef8d55d | 2013-09-27 12:50:13 -0700 | [diff] [blame] | 141 | elif i == 4: |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 142 | main.log.error( "Unable to fping" ) |
timlindberg | ef8d55d | 2013-09-27 12:50:13 -0700 | [diff] [blame] | 143 | result = main.FALSE |
| 144 | return result |
| 145 | elif i == 5: |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 146 | main.log.info( "fping not installed, installing fping" ) |
| 147 | self.handle.sendline( "sudo apt-get install fping" ) |
| 148 | i = self.handle.expect( |
| 149 | [ "password", |
| 150 | "\$", |
| 151 | pexpect.TIMEOUT ], |
| 152 | timeout=60 ) |
timlindberg | ef8d55d | 2013-09-27 12:50:13 -0700 | [diff] [blame] | 153 | if i == 0: |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 154 | self.handle.sendline( self.pwd ) |
| 155 | self.handle.expect( "\$", timeout=30 ) |
| 156 | main.log.info( "fping installed, now pinging interfaces" ) |
| 157 | self.handle.sendline( |
| 158 | "sudo fping -S " + str( |
| 159 | netsrc ) + "." + str( |
| 160 | netstrt ) + ".1.1 -f /tmp/ip_table" + str( |
| 161 | netdst ) + ".txt" ) |
timlindberg | ef8d55d | 2013-09-27 12:50:13 -0700 | [diff] [blame] | 162 | elif i == 1: |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 163 | main.log.info( "fping installed, now pinging interfaces" ) |
| 164 | self.handle.sendline( |
| 165 | "sudo fping -S " + str( |
| 166 | netsrc ) + "." + str( |
| 167 | netstrt ) + ".1.1 -f /tmp/ip_table" + str( |
| 168 | netdst ) + ".txt" ) |
timlindberg | ef8d55d | 2013-09-27 12:50:13 -0700 | [diff] [blame] | 169 | elif i == 2: |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 170 | main.log.error( "Could not install fping" ) |
timlindberg | ef8d55d | 2013-09-27 12:50:13 -0700 | [diff] [blame] | 171 | result = main.FALSE |
| 172 | return result |
| 173 | |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 174 | def disconnect( self ): |
timlindberg | ef8d55d | 2013-09-27 12:50:13 -0700 | [diff] [blame] | 175 | response = '' |
| 176 | try: |
kelvin | 8ec7144 | 2015-01-15 16:57:00 -0800 | [diff] [blame] | 177 | self.handle.sendline( "exit" ) |
| 178 | self.handle.expect( "closed" ) |
Jon Hall | febb1c7 | 2015-03-05 13:30:09 -0800 | [diff] [blame^] | 179 | except Exception: |
| 180 | main.log.exception( "Connection failed to the host" ) |
timlindberg | ef8d55d | 2013-09-27 12:50:13 -0700 | [diff] [blame] | 181 | response = main.FALSE |
| 182 | return response |
| 183 | |