blob: 38af1a9bc29129724870cd7a8a903e7b4692da95 [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
alshabib9a044c12014-11-17 10:36:36 -080012path = os.path.join( '/home', 'mininet', 'mininet','util', 'vm' )
Brian O'Connor37f0cc72014-06-06 20:03:05 -070013path = 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
Jonathan Hart4a7ae2b2014-12-04 23:12:43 -080020onos_version='1.0.0'
21
Brian O'Connor37f0cc72014-06-06 20:03:05 -070022# Import required functions from the Mininet build script
Jonathan Hart8b3b1832015-03-31 10:02:22 -070023from build import bootAndRun, Prompt, OVFOSNameID, generateOVF, log, build, run
Brian O'Connor37f0cc72014-06-06 20:03:05 -070024
25def installONOS( vm, prompt=Prompt ):
26 # start with sendline
27 #TODO consider resizing the HDD
Jonathan Hart53433b52015-03-26 17:23:34 -070028 url = 'https://gerrit.onosproject.org/gitweb?p=onos-vm.git;a=blob_plain;hb=refs/heads/new-onos-tutorial;f=vm-setup-lxc.sh'
Jonathan Harta09e5352015-03-26 17:30:36 -070029 vm.sendline( ' wget -O vm-setup-lxc.sh "%s" | bash' % url ) # space prefix is used to avoid entry in history
Brian O'Connor5b18ef52014-08-05 02:25:53 -070030 vm.expect( prompt, timeout=20 )
Jonathan Hart53433b52015-03-26 17:23:34 -070031 vm.sendline( 'bash vm-setup-lxc.sh' )
alshabibfed22522014-12-09 00:49:59 -080032 vm.expect( prompt, timeout=3600 )
alshabibcc130252014-11-17 10:19:35 -080033
Jonathan Hart1cc5a1a2015-03-31 16:54:11 -070034 url = 'https://gerrit.onosproject.org/gitweb?p=onos-vm.git;a=blob_plain;hb=refs/heads/new-onos-tutorial;f=onos-setup-lxc.sh'
Jonathan Hartb1e22112015-03-31 16:42:03 -070035 vm.sendline( ' wget -O onos-setup-lxc.sh "%s" | bash' % url ) # space prefix is used to avoid entry in history
36 vm.expect( prompt, timeout=20 )
37 vm.sendline( 'bash onos-setup-lxc.sh' )
38 vm.expect(prompt, timeout=3600)
Jonathan Hart1cc5a1a2015-03-31 16:54:11 -070039
alshabibbca6bf22014-11-15 16:38:45 -080040
Brian O'Connor37f0cc72014-06-06 20:03:05 -070041def vmdk2size( vmdk ):
42 "Return virtual disk size (in bytes) of vmdk image"
43 output = check_output( [ 'file', vmdk ] )
44 assert 'disk image' in output
45 return os.stat( vmdk )[ ST_SIZE ]
46
alshabib0ecf4c42014-12-11 15:11:44 -080047def buildONOS( image, out, flavor='trusty64server', installMemory=2048, ovfMemory=4096 ):
Brian O'Connor37f0cc72014-06-06 20:03:05 -070048 if not image:
49 # build VM from flavor
50 #build( flavor )
51 pass
52
53 bootAndRun( image, runFunction=installONOS, memory=installMemory, outputFile=out )
54
55 size = vmdk2size( out + '.vmdk' ) # get vmdk size
56 ovfname = out
57 osname, osid = OVFOSNameID( flavor )
58 print osname, osid
59
60 ovf = generateOVF( name=ovfname, osname=osname, osid=osid,
alshabibd2b24562014-11-18 19:24:00 -080061 diskname=out+'.vmdk', disksize=size, mem=ovfMemory, cpus=2,
62 vmname='ONOS-Tutorial', vminfo='ONOS Tutorial VM' )
Brian O'Connor37f0cc72014-06-06 20:03:05 -070063 log( '* Generated OVF descriptor file', ovf )
64
65if __name__ == '__main__':
Brian O'Connor3ca84022014-08-11 19:46:54 -070066 parser = argparse.ArgumentParser( description='Mininet Tutorial VM build script' )
Brian O'Connor37f0cc72014-06-06 20:03:05 -070067 parser.add_argument( '-i', '--image', metavar='image', required=True,
68 help='Build from an existing VM image' )
Brian O'Connor3ca84022014-08-11 19:46:54 -070069 parser.add_argument( '-f', '--flavor', default='trusty64server',
70 help='VM flavor to build (e.g. trusty64server)' )
Brian O'Connor37f0cc72014-06-06 20:03:05 -070071 parser.add_argument( 'scripts', nargs='*',
Brian O'Connor3ca84022014-08-11 19:46:54 -070072 help='VM flavor(s) to build (e.g. trusty64server)' )
Brian O'Connor37f0cc72014-06-06 20:03:05 -070073 parser.add_argument( '-z', '--zip', action='store_true',
74 help='archive .ovf and .vmdk into .zip file' )
75 parser.add_argument( '-o', '--out', required=True,
76 help='output file for test image (vmdk)' )
77 args = parser.parse_args()
78
79 print args
Brian O'Connor37f0cc72014-06-06 20:03:05 -070080
81 buildONOS( args.image, args.out )
82
83 if args.zip:
84 log( '* Generating .zip file' )
85 run( 'zip %s-ovf.zip %s %s' % ( args.out, args.out + '.ovf', args.out + '.vmdk' ) )