Brian O'Connor | 37f0cc7 | 2014-06-06 20:03:05 -0700 | [diff] [blame] | 1 | #!/usr/bin/python |
| 2 | |
| 3 | import os |
| 4 | import sys |
| 5 | import re |
| 6 | import argparse |
| 7 | from stat import ST_MODE, ST_SIZE |
| 8 | from subprocess import check_output |
| 9 | |
| 10 | # Add the Mininet build script directory to the path, so we can import it |
| 11 | #TODO do a better job finding Mininet |
| 12 | path = os.path.join( '~', 'mininet', 'util', 'vm' ) |
| 13 | path = os.path.expanduser( path ) |
| 14 | if os.path.isdir( path ): |
| 15 | sys.path.append( path ) |
| 16 | else: |
| 17 | print 'Cannot find Mininet build script. Exiting...' |
| 18 | sys.exit( 1 ) |
| 19 | |
| 20 | # Import required functions from the Mininet build script |
| 21 | from build import bootAndRun, Prompt, OVFOSNameID, generateOVF, log, build, run |
| 22 | |
| 23 | def installONOS( vm, prompt=Prompt ): |
| 24 | # start with sendline |
| 25 | #TODO consider resizing the HDD |
Brian O'Connor | 3ca8402 | 2014-08-11 19:46:54 -0700 | [diff] [blame] | 26 | url = 'https://gerrit.onlab.us/gitweb?p=ONOS-VM.git;a=blob_plain;hb=refs/heads/mininet-tutorial;f=vm-setup.sh' |
Brian O'Connor | 4c074f4 | 2014-08-12 13:39:34 -0700 | [diff] [blame] | 27 | vm.sendline( ' wget -O vm-setup.sh "%s" | bash' % url ) # space prefix is used to avoid entry in history |
Brian O'Connor | 5b18ef5 | 2014-08-05 02:25:53 -0700 | [diff] [blame] | 28 | vm.expect( prompt, timeout=20 ) |
| 29 | vm.sendline( 'bash vm-setup.sh' ) |
Brian O'Connor | 37f0cc7 | 2014-06-06 20:03:05 -0700 | [diff] [blame] | 30 | # end with expect prompt |
Brian O'Connor | 10d941e | 2014-06-12 19:55:09 -0700 | [diff] [blame] | 31 | vm.expect( prompt, timeout=1200 ) |
Brian O'Connor | 37f0cc7 | 2014-06-06 20:03:05 -0700 | [diff] [blame] | 32 | |
alshabib | bca6bf2 | 2014-11-15 16:38:45 -0800 | [diff] [blame] | 33 | vm.spawn( 'git clone https://gerrit.onlab.us/onos-next/' ) |
| 34 | vm.expect("Username for 'https://gerrit.onlab.us':", timeout=30) |
| 35 | vm.sendline('vm') |
| 36 | vm.expect("Password for 'https://vm@gerrit.onlab.us':", timeout=30) |
| 37 | vm.sendline('lAEi5gM/iFF3') |
| 38 | |
| 39 | vm.expect( prompt, timeout=1200 ) |
| 40 | |
| 41 | vm.sendline('ssh 127.0.0.1') |
| 42 | vm.expect('.*connecting (yes/no)?') |
| 43 | vm.sendline('yes') |
| 44 | vm.sendline('exit') |
| 45 | |
| 46 | vm.expect( prompt, timeout=1200 ) |
| 47 | |
| 48 | vm.sendline( 'bash onos-setup.sh' ) |
| 49 | |
| 50 | |
| 51 | |
Brian O'Connor | 37f0cc7 | 2014-06-06 20:03:05 -0700 | [diff] [blame] | 52 | def vmdk2size( vmdk ): |
| 53 | "Return virtual disk size (in bytes) of vmdk image" |
| 54 | output = check_output( [ 'file', vmdk ] ) |
| 55 | assert 'disk image' in output |
| 56 | return os.stat( vmdk )[ ST_SIZE ] |
| 57 | |
Brian O'Connor | 3ca8402 | 2014-08-11 19:46:54 -0700 | [diff] [blame] | 58 | def buildONOS( image, out, flavor='trusty64server', installMemory=1024, ovfMemory=1024 ): |
Brian O'Connor | 37f0cc7 | 2014-06-06 20:03:05 -0700 | [diff] [blame] | 59 | if not image: |
| 60 | # build VM from flavor |
| 61 | #build( flavor ) |
| 62 | pass |
| 63 | |
| 64 | bootAndRun( image, runFunction=installONOS, memory=installMemory, outputFile=out ) |
| 65 | |
| 66 | size = vmdk2size( out + '.vmdk' ) # get vmdk size |
| 67 | ovfname = out |
| 68 | osname, osid = OVFOSNameID( flavor ) |
| 69 | print osname, osid |
| 70 | |
| 71 | ovf = generateOVF( name=ovfname, osname=osname, osid=osid, |
Brian O'Connor | 3ca8402 | 2014-08-11 19:46:54 -0700 | [diff] [blame] | 72 | diskname=out+'.vmdk', disksize=size, mem=ovfMemory, cpus=1, |
| 73 | vmname='Mininet-Tutorial', vminfo='Mininet Tutorial VM' ) |
Brian O'Connor | 37f0cc7 | 2014-06-06 20:03:05 -0700 | [diff] [blame] | 74 | log( '* Generated OVF descriptor file', ovf ) |
| 75 | |
| 76 | if __name__ == '__main__': |
Brian O'Connor | 3ca8402 | 2014-08-11 19:46:54 -0700 | [diff] [blame] | 77 | parser = argparse.ArgumentParser( description='Mininet Tutorial VM build script' ) |
Brian O'Connor | 37f0cc7 | 2014-06-06 20:03:05 -0700 | [diff] [blame] | 78 | parser.add_argument( '-i', '--image', metavar='image', required=True, |
| 79 | help='Build from an existing VM image' ) |
Brian O'Connor | 3ca8402 | 2014-08-11 19:46:54 -0700 | [diff] [blame] | 80 | parser.add_argument( '-f', '--flavor', default='trusty64server', |
| 81 | help='VM flavor to build (e.g. trusty64server)' ) |
Brian O'Connor | 37f0cc7 | 2014-06-06 20:03:05 -0700 | [diff] [blame] | 82 | parser.add_argument( 'scripts', nargs='*', |
Brian O'Connor | 3ca8402 | 2014-08-11 19:46:54 -0700 | [diff] [blame] | 83 | help='VM flavor(s) to build (e.g. trusty64server)' ) |
Brian O'Connor | 37f0cc7 | 2014-06-06 20:03:05 -0700 | [diff] [blame] | 84 | parser.add_argument( '-z', '--zip', action='store_true', |
| 85 | help='archive .ovf and .vmdk into .zip file' ) |
| 86 | parser.add_argument( '-o', '--out', required=True, |
| 87 | help='output file for test image (vmdk)' ) |
| 88 | args = parser.parse_args() |
| 89 | |
| 90 | print args |
| 91 | |
| 92 | buildONOS( args.image, args.out ) |
| 93 | |
| 94 | if args.zip: |
| 95 | log( '* Generating .zip file' ) |
| 96 | run( 'zip %s-ovf.zip %s %s' % ( args.out, args.out + '.ovf', args.out + '.vmdk' ) ) |