def stop(self):
        logger.verbose('sliver_libvirt: {} stop'.format(self.name))

        # Remove the ebtables rule before stopping 
        bwlimit.ebtables("-D INPUT -i veth{} -j mark --set-mark {}"
                         .format(self.xid, self.xid))

        try:
            self.dom.destroy()
        except:
            logger.log_exc("in sliver_libvirt.stop", name=self.name)
    def start(self, delay=0):
        '''Just start the sliver'''
        logger.verbose('sliver_libvirt: {} start'.format(self.name))

        # TD: Added OpenFlow rules to avoid OpenVSwitch-based slivers'
        # auto-configuration issues when IPv6 auto-config and/or DHCP are
        # available in the node's network. Sliver configuration is static.
        if os.path.exists('/usr/bin/ovs-ofctl'):
            logger.log('Adding OpenFlow rules to prevent IPv6 auto-config and DHCP in OpenVSwitch slivers')
            # IPv6 ICMP Router Solicitation and Advertisement
            logger.log_call([ '/usr/bin/ovs-ofctl', 'add-flow', 'public0', 'priority=100,icmp6,icmp_type=133,idle_timeout=0,hard_timeout=0,actions=drop' ])
            logger.log_call([ '/usr/bin/ovs-ofctl', 'add-flow', 'public0', 'priority=100,icmp6,icmp_type=134,idle_timeout=0,hard_timeout=0,actions=drop' ])
            # IPv4 DHCP
            logger.log_call([ '/usr/bin/ovs-ofctl', 'add-flow', 'public0', 'priority=101,udp,nw_src=0.0.0.0,nw_dst=255.255.255.255,tp_src=68,tp_dst=67,idle_timeout=0,hard_timeout=0,actions=drop' ])
            logger.log_call([ '/usr/bin/ovs-ofctl', 'add-flow', 'public0', 'priority=101,udp,tp_src=67,tp_dst=68,idle_timeout=0,hard_timeout=0,actions=drop' ])
        else:
            logger.log('NOTE: /usr/bin/ovs-ofctl not found!')

        # Check if it's running to avoid throwing an exception if the
        # domain was already running
        if not self.is_running():
            try:
                # create actually means start
                self.dom.create()
            except Exception as e:
                # XXX smbaker: attempt to resolve slivers that are stuck in
                #   "failed to allocate free veth".
                if "ailed to allocate free veth" in str(e):
                     logger.log("failed to allocate free veth on {}".format(self.name))
                     self.repair_veth()
                     logger.log("trying dom.create again")
                     self.dom.create()
                else:
                    raise
        else:
            logger.verbose('sliver_libvirt: sliver {} already started'.format(self.name))

        # After the VM is started... we can play with the virtual interface
        # Create the ebtables rule to mark the packets going out from the virtual
        # interface to the actual device so the filter canmatch against the mark
        bwlimit.ebtables("-A INPUT -i veth{} -j mark --set-mark {}"
                         .format(self.xid, self.xid))

        # TD: Turn off SCTP checksum offloading. It is currently not working. FIXME: check for a kernel fix!
        result = logger.log_call(['/usr/sbin/lxcsu', '-r', self.name, '--', '/usr/sbin/ethtool', '-K', 'eth0', 'tx-checksum-sctp', 'off'])
        if not result:
            logger.log('unable to apply SCTP checksum bug work-around for %s' % self.name)

        # TD: Work-around for missing interface configuration: ensure that networking service is running.
        result = logger.log_call(['/usr/sbin/lxcsu', '-r', self.name, '/sbin/service', 'network', 'restart'])
        if not result:
            logger.log('unable to restart networking service for %s' % self.name)
    def stop(self):
        logger.verbose('sliver_libvirt: %s stop'%(self.name))

        # Remove the ebtables rule before stopping 
        bwlimit.ebtables("-D INPUT -i veth%d -j mark --set-mark %d" % \
            (self.xid, self.xid))

        try:
            self.dom.destroy()
        except:
            logger.verbose('sliver_libvirt: Domain %s not running ' \
                           'UNEXPECTED: %s'%(self.name, sys.exc_info()[1]))
            print 'sliver_libvirt: Domain %s not running ' \
                  'UNEXPECTED: %s'%(self.name, sys.exc_info()[1])
    def start(self, delay=0):
        ''' Just start the sliver '''
        logger.verbose('sliver_libvirt: %s start'%(self.name))

        # Check if it's running to avoid throwing an exception if the
        # domain was already running, create actually means start
        if not self.is_running():
            self.dom.create()
        else:
            logger.verbose('sliver_libvirt: sliver %s already started'%(self.name))

        # After the VM is started... we can play with the virtual interface
        # Create the ebtables rule to mark the packets going out from the virtual
        # interface to the actual device so the filter canmatch against the mark
        bwlimit.ebtables("-A INPUT -i veth%d -j mark --set-mark %d" % \
            (self.xid, self.xid))
Exemple #5
0
def create_lxc(name, rec=None):
        ''' Create dirs, copy fs image, creat_lxc '''
        logger.verbose ('sliver_lxc: %s create'%(name))
        conn = libvirt.open("lxc://")
       
	try: 
            p = conn.lookupByName(name)
            logger.log("create_lxc: there is already a running vm %s!"%(name))
            return
	except:
            logger.log("create a new sliver %s"%(name))
		#return

        # Get the type of image from vref myplc tags specified as:
        # pldistro = lxc
        # fcdistro = squeeze
        # arch x86_64

        arch = 'x86_64'
        #tags = rec['rspec']['tags']
        #if 'arch' in tags:
        #    arch = tags['arch']
        #    if arch == 'i386':
        #        arch = 'i686'

        vref = "lxc-f18-x86_64"
	#vref = rec['vref']
        #if vref is None:
        #    vref = "lxc-f14-x86_64"
        #    logger.log("sliver_libvirt: %s: WARNING - no vref attached, using hard-wired default %s" % (name,vref))

        refImgDir    = os.path.join('/vservers/.lvref', vref)
        containerDir = os.path.join('/vservers', name)

        # check the template exists -- there's probably a better way..
        if not os.path.isdir(refImgDir):
            logger.log('creat_lxc: %s: ERROR Could not create sliver - reference image %s not found' % (name,vref))
            logger.log('creat_lxc: %s: ERROR Expected reference image in %s'%(name,refImgDir))
            return

        # Snapshot the reference image fs (assume the reference image is in its own
        # subvolume)
        command = ['btrfs', 'subvolume', 'snapshot', refImgDir, containerDir]
        if not logger.log_call(command, timeout=15*60):
            logger.log('creat_lxc: ERROR Could not create BTRFS snapshot at', containerDir)
            return
        command = ['chmod', '755', containerDir]
        logger.log_call(command, timeout=15*60)

        # TODO: set quotas...

        # Set hostname. A valid hostname cannot have '_'
        #with open(os.path.join(containerDir, 'etc/hostname'), 'w') as f:
        #    print >>f, name.replace('_', '-')

        # Add slices group if not already present
        try:
            group = grp.getgrnam('slices')
        except:
            command = ['/usr/sbin/groupadd', 'slices']
            logger.log_call(command, timeout=15*60)

        # Add unix account (TYPE is specified in the subclass)
        command = ['/usr/sbin/useradd', '-g', 'slices', '-s', '/usr/sbin/vsh', name, '-p', '*']
        logger.log_call(command, timeout=15*60)
        command = ['mkdir', '/home/%s/.ssh'%name]
        logger.log_call(command, timeout=15*60)

        # Create PK pair keys to connect from the host to the guest without
        # password... maybe remove the need for authentication inside the
        # guest?
        command = ['su', '-s', '/bin/bash', '-c', 'ssh-keygen -t rsa -N "" -f /home/%s/.ssh/id_rsa'%(name)]
        logger.log_call(command, timeout=60)

        command = ['chown', '-R', '%s.slices'%name, '/home/%s/.ssh'%name]
        logger.log_call(command, timeout=30)

        command = ['mkdir', '%s/root/.ssh'%containerDir]
        logger.log_call(command, timeout=10)

        command = ['cp', '/home/%s/.ssh/id_rsa.pub'%name, '%s/root/.ssh/authorized_keys'%containerDir]
        logger.log_call(command, timeout=30)

        logger.log("creating /etc/slicename file in %s" % os.path.join(containerDir,'etc/slicename'))
        try:
            file(os.path.join(containerDir,'etc/slicename'), 'w').write(name)
        except:
            logger.log_exc("exception while creating /etc/slicename")

        try:
            file(os.path.join(containerDir,'etc/slicefamily'), 'w').write(vref)
        except:
            logger.log_exc("exception while creating /etc/slicefamily")

        uid = None
        try:
            uid = getpwnam(name).pw_uid
        except KeyError:
            # keyerror will happen if user id was not created successfully
            logger.log_exc("exception while getting user id")

        if uid is not None:
            logger.log("uid is %d" % uid)
            command = ['mkdir', '%s/home/%s' % (containerDir, name)]
            logger.log_call(command, timeout=10)
            command = ['chown', name, '%s/home/%s' % (containerDir, name)]
            logger.log_call(command, timeout=10)
            etcpasswd = os.path.join(containerDir, 'etc/passwd')
            etcgroup = os.path.join(containerDir, 'etc/group')
            if os.path.exists(etcpasswd):
                # create all accounts with gid=1001 - i.e. 'slices' like it is in the root context
                slices_gid=1001
                logger.log("adding user %(name)s id %(uid)d gid %(slices_gid)d to %(etcpasswd)s" % (locals()))
                try:
                    file(etcpasswd,'a').write("%(name)s:x:%(uid)d:%(slices_gid)d::/home/%(name)s:/bin/bash\n" % locals())
                except:
                    logger.log_exc("exception while updating %s"%etcpasswd)
                logger.log("adding group slices with gid %(slices_gid)d to %(etcgroup)s"%locals())
                try:
                    file(etcgroup,'a').write("slices:x:%(slices_gid)d\n"%locals())
                except:
                    logger.log_exc("exception while updating %s"%etcgroup)
            sudoers = os.path.join(containerDir, 'etc/sudoers')
            if os.path.exists(sudoers):
                try:
                    file(sudoers,'a').write("%s ALL=(ALL) NOPASSWD: ALL\n" % name)
                except:
                    logger.log_exc("exception while updating /etc/sudoers")

        # customizations for the user environment - root or slice uid
        # we save the whole business in /etc/planetlab.profile 
        # and source this file for both root and the slice uid's .profile
        # prompt for slice owner, + LD_PRELOAD for transparently wrap bind
        pl_profile=os.path.join(containerDir,"etc/planetlab.profile")
        ld_preload_text="""# by default, we define this setting so that calls to bind(2),
# when invoked on 0.0.0.0, get transparently redirected to the public interface of this node
# see https://svn.planet-lab.org/wiki/LxcPortForwarding"""
        usrmove_path_text="""# VM's before Features/UsrMove need /bin and /sbin in their PATH"""
        usrmove_path_code="""
pathmunge () {
        if ! echo $PATH | /bin/egrep -q "(^|:)$1($|:)" ; then
           if [ "$2" = "after" ] ; then
              PATH=$PATH:$1
           else
              PATH=$1:$PATH
           fi
        fi
}
pathmunge /bin after
pathmunge /sbin after
unset pathmunge
"""
        with open(pl_profile,'w') as f:
            f.write("export PS1='%s@\H \$ '\n"%(name))
            f.write("%s\n"%ld_preload_text)
            f.write("export LD_PRELOAD=/etc/planetlab/lib/bind_public.so\n")
            f.write("%s\n"%usrmove_path_text)
            f.write("%s\n"%usrmove_path_code)

        # make sure this file is sourced from both root's and slice's .profile
        enforced_line = "[ -f /etc/planetlab.profile ] && source /etc/planetlab.profile\n"
        for path in [ 'root/.profile', 'home/%s/.profile'%name ]:
            from_root=os.path.join(containerDir,path)
            # if dir is not yet existing let's forget it for now
            if not os.path.isdir(os.path.dirname(from_root)): continue
            found=False
            try: 
                contents=file(from_root).readlines()
                for content in contents:
                    if content==enforced_line: found=True
            except IOError: pass
            if not found:
                with open(from_root,"a") as user_profile:
                    user_profile.write(enforced_line)
                # in case we create the slice's .profile when writing
                if from_root.find("/home")>=0:
                    command=['chown','%s:slices'%name,from_root]
                    logger.log_call(command,timeout=5)

        # Lookup for xid and create template after the user is created so we
        # can get the correct xid based on the name of the slice
        xid = bwlimit.get_xid(name)

        # Template for libvirt sliver configuration
        template_filename_sliceimage = os.path.join('/vservers/.lvref','lxc_template.xml')
        if os.path.isfile (template_filename_sliceimage):
            logger.log("WARNING: using compat template %s"%template_filename_sliceimage)
            template_filename=template_filename_sliceimage
        else:
            logger.log("Cannot find XML template %s"%template_filename_sliceimage)
            return

        interfaces = get_interfaces_xml(rec)

        try:
            with open(template_filename) as f:
                template = Template(f.read())
                xml  = template.substitute(name=name, xid=xid, interfaces=interfaces, arch=arch)
        except IOError:
            logger.log('Failed to parse or use XML template file %s'%template_filename)
            return

        # Lookup for the sliver before actually
        # defining it, just in case it was already defined.
        try:
            dom = conn.lookupByName(name)
        except:
            dom = conn.defineXML(xml)

	def debuginfo(dom):
        	''' Helper method to get a "nice" output of the info struct for debug'''
        	[state, maxmem, mem, ncpu, cputime] = dom.info()
	        return '%s is %s, maxmem = %s, mem = %s, ncpu = %s, cputime = %s' % (dom.name(), STATES.get(state, state), maxmem, mem, ncpu, cputime)

        logger.verbose('create_lxc: %s -> %s'%(name, debuginfo(dom)))

	# the sliver has been created
	# then configure the sliver, write keys to authorized_keys file
	configure(name, rec)

	# finally, start the sliver
	dom.create()
	logger.verbose('create_lxc:%s start'%(name))

	# After the VM is started... we can play with the virtual interface
        # Create the ebtables rule to mark the packets going out from the virtual
        # interface to the actual device so the filter canmatch against the mark
        bwlimit.ebtables("-A INPUT -i veth%d -j mark --set-mark %d" % \
            (xid, xid))
def create_lxc(name, rec=None):
    ''' Create dirs, copy fs image, creat_lxc '''
    logger.verbose('sliver_lxc: %s create' % (name))
    conn = libvirt.open("lxc://")

    try:
        p = conn.lookupByName(name)
        logger.log("create_lxc: there is already a running vm %s!" % (name))
        return
    except:
        logger.log("create a new sliver %s" % (name))
#return

# Get the type of image from vref myplc tags specified as:
# pldistro = lxc
# fcdistro = squeeze
# arch x86_64

    arch = 'x86_64'
    #tags = rec['rspec']['tags']
    #if 'arch' in tags:
    #    arch = tags['arch']
    #    if arch == 'i386':
    #        arch = 'i686'

    vref = "lxc-f18-x86_64"
    #vref = rec['vref']
    #if vref is None:
    #    vref = "lxc-f14-x86_64"
    #    logger.log("sliver_libvirt: %s: WARNING - no vref attached, using hard-wired default %s" % (name,vref))

    refImgDir = os.path.join('/vservers/.lvref', vref)
    containerDir = os.path.join('/vservers', name)

    # check the template exists -- there's probably a better way..
    if not os.path.isdir(refImgDir):
        logger.log(
            'creat_lxc: %s: ERROR Could not create sliver - reference image %s not found'
            % (name, vref))
        logger.log('creat_lxc: %s: ERROR Expected reference image in %s' %
                   (name, refImgDir))
        return

    # Snapshot the reference image fs (assume the reference image is in its own
    # subvolume)
    command = ['btrfs', 'subvolume', 'snapshot', refImgDir, containerDir]
    if not logger.log_call(command, timeout=15 * 60):
        logger.log('creat_lxc: ERROR Could not create BTRFS snapshot at',
                   containerDir)
        return
    command = ['chmod', '755', containerDir]
    logger.log_call(command, timeout=15 * 60)

    # TODO: set quotas...

    # Set hostname. A valid hostname cannot have '_'
    #with open(os.path.join(containerDir, 'etc/hostname'), 'w') as f:
    #    print >>f, name.replace('_', '-')

    # Add slices group if not already present
    try:
        group = grp.getgrnam('slices')
    except:
        command = ['/usr/sbin/groupadd', 'slices']
        logger.log_call(command, timeout=15 * 60)

    # Add unix account (TYPE is specified in the subclass)
    command = [
        '/usr/sbin/useradd', '-g', 'slices', '-s', '/usr/sbin/vsh', name, '-p',
        '*'
    ]
    logger.log_call(command, timeout=15 * 60)
    command = ['mkdir', '/home/%s/.ssh' % name]
    logger.log_call(command, timeout=15 * 60)

    # Create PK pair keys to connect from the host to the guest without
    # password... maybe remove the need for authentication inside the
    # guest?
    command = [
        'su', '-s', '/bin/bash', '-c',
        'ssh-keygen -t rsa -N "" -f /home/%s/.ssh/id_rsa' % (name)
    ]
    logger.log_call(command, timeout=60)

    command = ['chown', '-R', '%s.slices' % name, '/home/%s/.ssh' % name]
    logger.log_call(command, timeout=30)

    command = ['mkdir', '%s/root/.ssh' % containerDir]
    logger.log_call(command, timeout=10)

    command = [
        'cp',
        '/home/%s/.ssh/id_rsa.pub' % name,
        '%s/root/.ssh/authorized_keys' % containerDir
    ]
    logger.log_call(command, timeout=30)

    logger.log("creating /etc/slicename file in %s" %
               os.path.join(containerDir, 'etc/slicename'))
    try:
        file(os.path.join(containerDir, 'etc/slicename'), 'w').write(name)
    except:
        logger.log_exc("exception while creating /etc/slicename")

    try:
        file(os.path.join(containerDir, 'etc/slicefamily'), 'w').write(vref)
    except:
        logger.log_exc("exception while creating /etc/slicefamily")

    uid = None
    try:
        uid = getpwnam(name).pw_uid
    except KeyError:
        # keyerror will happen if user id was not created successfully
        logger.log_exc("exception while getting user id")

    if uid is not None:
        logger.log("uid is %d" % uid)
        command = ['mkdir', '%s/home/%s' % (containerDir, name)]
        logger.log_call(command, timeout=10)
        command = ['chown', name, '%s/home/%s' % (containerDir, name)]
        logger.log_call(command, timeout=10)
        etcpasswd = os.path.join(containerDir, 'etc/passwd')
        etcgroup = os.path.join(containerDir, 'etc/group')
        if os.path.exists(etcpasswd):
            # create all accounts with gid=1001 - i.e. 'slices' like it is in the root context
            slices_gid = 1001
            logger.log(
                "adding user %(name)s id %(uid)d gid %(slices_gid)d to %(etcpasswd)s"
                % (locals()))
            try:
                file(etcpasswd, 'a').write(
                    "%(name)s:x:%(uid)d:%(slices_gid)d::/home/%(name)s:/bin/bash\n"
                    % locals())
            except:
                logger.log_exc("exception while updating %s" % etcpasswd)
            logger.log(
                "adding group slices with gid %(slices_gid)d to %(etcgroup)s" %
                locals())
            try:
                file(etcgroup,
                     'a').write("slices:x:%(slices_gid)d\n" % locals())
            except:
                logger.log_exc("exception while updating %s" % etcgroup)
        sudoers = os.path.join(containerDir, 'etc/sudoers')
        if os.path.exists(sudoers):
            try:
                file(sudoers, 'a').write("%s ALL=(ALL) NOPASSWD: ALL\n" % name)
            except:
                logger.log_exc("exception while updating /etc/sudoers")

    # customizations for the user environment - root or slice uid
    # we save the whole business in /etc/planetlab.profile
    # and source this file for both root and the slice uid's .profile
    # prompt for slice owner, + LD_PRELOAD for transparently wrap bind
    pl_profile = os.path.join(containerDir, "etc/planetlab.profile")
    ld_preload_text = """# by default, we define this setting so that calls to bind(2),
# when invoked on 0.0.0.0, get transparently redirected to the public interface of this node
# see https://svn.planet-lab.org/wiki/LxcPortForwarding"""
    usrmove_path_text = """# VM's before Features/UsrMove need /bin and /sbin in their PATH"""
    usrmove_path_code = """
pathmunge () {
        if ! echo $PATH | /bin/egrep -q "(^|:)$1($|:)" ; then
           if [ "$2" = "after" ] ; then
              PATH=$PATH:$1
           else
              PATH=$1:$PATH
           fi
        fi
}
pathmunge /bin after
pathmunge /sbin after
unset pathmunge
"""
    with open(pl_profile, 'w') as f:
        f.write("export PS1='%s@\H \$ '\n" % (name))
        f.write("%s\n" % ld_preload_text)
        f.write("export LD_PRELOAD=/etc/planetlab/lib/bind_public.so\n")
        f.write("%s\n" % usrmove_path_text)
        f.write("%s\n" % usrmove_path_code)

    # make sure this file is sourced from both root's and slice's .profile
    enforced_line = "[ -f /etc/planetlab.profile ] && source /etc/planetlab.profile\n"
    for path in ['root/.profile', 'home/%s/.profile' % name]:
        from_root = os.path.join(containerDir, path)
        # if dir is not yet existing let's forget it for now
        if not os.path.isdir(os.path.dirname(from_root)): continue
        found = False
        try:
            contents = file(from_root).readlines()
            for content in contents:
                if content == enforced_line: found = True
        except IOError:
            pass
        if not found:
            with open(from_root, "a") as user_profile:
                user_profile.write(enforced_line)
            # in case we create the slice's .profile when writing
            if from_root.find("/home") >= 0:
                command = ['chown', '%s:slices' % name, from_root]
                logger.log_call(command, timeout=5)

    # Lookup for xid and create template after the user is created so we
    # can get the correct xid based on the name of the slice
    xid = bwlimit.get_xid(name)

    # Template for libvirt sliver configuration
    template_filename_sliceimage = os.path.join('/vservers/.lvref',
                                                'lxc_template.xml')
    if os.path.isfile(template_filename_sliceimage):
        logger.log("WARNING: using compat template %s" %
                   template_filename_sliceimage)
        template_filename = template_filename_sliceimage
    else:
        logger.log("Cannot find XML template %s" %
                   template_filename_sliceimage)
        return

    interfaces = get_interfaces_xml(rec)

    try:
        with open(template_filename) as f:
            template = Template(f.read())
            xml = template.substitute(name=name,
                                      xid=xid,
                                      interfaces=interfaces,
                                      arch=arch)
    except IOError:
        logger.log('Failed to parse or use XML template file %s' %
                   template_filename)
        return

    # Lookup for the sliver before actually
    # defining it, just in case it was already defined.
    try:
        dom = conn.lookupByName(name)
    except:
        dom = conn.defineXML(xml)

    def debuginfo(dom):
        ''' Helper method to get a "nice" output of the info struct for debug'''
        [state, maxmem, mem, ncpu, cputime] = dom.info()
        return '%s is %s, maxmem = %s, mem = %s, ncpu = %s, cputime = %s' % (
            dom.name(), STATES.get(state, state), maxmem, mem, ncpu, cputime)

    logger.verbose('create_lxc: %s -> %s' % (name, debuginfo(dom)))

    # the sliver has been created
    # then configure the sliver, write keys to authorized_keys file
    configure(name, rec)

    # finally, start the sliver
    dom.create()
    logger.verbose('create_lxc:%s start' % (name))

    # After the VM is started... we can play with the virtual interface
    # Create the ebtables rule to mark the packets going out from the virtual
    # interface to the actual device so the filter canmatch against the mark
    bwlimit.ebtables("-A INPUT -i veth%d -j mark --set-mark %d" % \
        (xid, xid))