blob: 3695fdaeafb20f21796c61270386d3f7879628ef [file] [log] [blame]
Brian O'Connor37f0cc72014-06-06 20:03:05 -07001#!/usr/bin/python
2
3import os
4import sys
5import re
6import argparse
7from stat import ST_MODE, ST_SIZE
8from 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
12path = os.path.join( '~', 'mininet', 'util', 'vm' )
13path = os.path.expanduser( path )
14if os.path.isdir( path ):
15 sys.path.append( path )
16else:
17 print 'Cannot find Mininet build script. Exiting...'
18 sys.exit( 1 )
19
20# Import required functions from the Mininet build script
21from build import bootAndRun, Prompt, OVFOSNameID, generateOVF, log, build, run
22
23def installONOS( vm, prompt=Prompt ):
24 # start with sendline
25 #TODO consider resizing the HDD
26 url = 'https://gerrit.onlab.us/gitweb?p=ONOS-VM.git;a=blob_plain;hb=refs/heads/master;f=vm-setup.sh'
Brian O'Connor5b18ef52014-08-05 02:25:53 -070027 vm.sendline( 'wget -O vm-setup.sh "%s" | bash' % url )
28 vm.expect( prompt, timeout=20 )
29 vm.sendline( 'bash vm-setup.sh' )
Brian O'Connor37f0cc72014-06-06 20:03:05 -070030 # end with expect prompt
Brian O'Connor10d941e2014-06-12 19:55:09 -070031 vm.expect( prompt, timeout=1200 )
Brian O'Connor37f0cc72014-06-06 20:03:05 -070032
33def vmdk2size( vmdk ):
34 "Return virtual disk size (in bytes) of vmdk image"
35 output = check_output( [ 'file', vmdk ] )
36 assert 'disk image' in output
37 return os.stat( vmdk )[ ST_SIZE ]
38
39def buildONOS( image, out, flavor='raring64server', installMemory=1024, ovfMemory=2048 ):
40 if not image:
41 # build VM from flavor
42 #build( flavor )
43 pass
44
45 bootAndRun( image, runFunction=installONOS, memory=installMemory, outputFile=out )
46
47 size = vmdk2size( out + '.vmdk' ) # get vmdk size
48 ovfname = out
49 osname, osid = OVFOSNameID( flavor )
50 print osname, osid
51
52 ovf = generateOVF( name=ovfname, osname=osname, osid=osid,
53 diskname=out+'.vmdk', disksize=size, mem=2048, cpus=4,
54 vmname='ONOS-VM', vminfo='An ONOS VM' )
55 log( '* Generated OVF descriptor file', ovf )
56
57if __name__ == '__main__':
58 parser = argparse.ArgumentParser( description='ONOS VM build script' )
59 parser.add_argument( '-i', '--image', metavar='image', required=True,
60 help='Build from an existing VM image' )
61 parser.add_argument( '-f', '--flavor', default='raring64server',
62 help='VM flavor to build (e.g. raring64server)' )
63 parser.add_argument( 'scripts', nargs='*',
64 help='VM flavor(s) to build (e.g. raring64server)' )
65 parser.add_argument( '-z', '--zip', action='store_true',
66 help='archive .ovf and .vmdk into .zip file' )
67 parser.add_argument( '-o', '--out', required=True,
68 help='output file for test image (vmdk)' )
69 args = parser.parse_args()
70
71 print args
72
73 buildONOS( args.image, args.out )
74
75 if args.zip:
76 log( '* Generating .zip file' )
77 run( 'zip %s-ovf.zip %s %s' % ( args.out, args.out + '.ovf', args.out + '.vmdk' ) )