source: fedd/fedd_split.py @ 2c6128f

#!/usr/local/bin/python

import os,sys

from ZSI import *
from M2Crypto import SSL
from M2Crypto.SSL.SSLServer import SSLServer
import M2Crypto.httpslib

import xml.parsers.expat

import re
import random
import string
import subprocess
import tempfile
import copy
import pickle

import traceback

from threading import *

from subprocess import *

from fedd_services import *
from fedd_internal_services import *
from fedd_util import *
import parse_detail
from service_error import *

import logging

class nullHandler(logging.Handler):
    def emit(self, record): pass

fl = logging.getLogger("fedd.splitter")
fl.addHandler(nullHandler())

class fedd_split_local:
    def __init__(self, config=None):
        """
        Intialize the various attributes, most from the config object
        """
        if config.has_section("splitter"):
            self.debug = config.getboolean("splitter", "debug")
            self.muxmax = config.getint("splitter", "muxmax", 3)
            self.tclsh = config.get("splitter", "tclsh", 
                    "/usr/local/bin/otclsh")
            self.tcl_splitter = config.get("splitter", "tcl_splitter",
                    "/usr/testbed/lib/ns2ir/parse.tcl")
        else:
            self.debug = False
            self.muxmax = 3
            self.tclsh = "/usr/local/bin/otclsh"
            self.tcl_splitter = "/usr/testbed/lib/ns2ir/parse.tcl"

        self.log = logging.getLogger("fedd.splitter")
        self.trace_file = sys.stderr

        set_log_level(config, "splitter", self.log)
        # Dispatch tables
        self.soap_services = {\
                'Ns2Split': make_soap_handler(\
                        Ns2SplitRequestMessage.typecode,
                        getattr(self, "run_splitter"), 
                        Ns2SplitResponseMessage,
                        "Ns2SplitResponseBody"),
        }

        self.xmlrpc_services = {\
                'Ns2Split': make_xmlrpc_handler(\
                        getattr(self, "run_splitter"), 
                        "Ns2SplitResponseBody"),
        }

    def run_splitter(self, req, fid):
        """
        The external interface to experiment creation called from the
        dispatcher.

        Creates a working directory, splits the incoming description using the
        splitter script and parses out the avrious subsections using the
        lcasses above.  Once each sub-experiment is created, use pooled threads
        to instantiate them and start it all up.
        """
        try:
            tmpdir = tempfile.mkdtemp(prefix="split-")
        except IOError:
            raise service_error(service_error.internal, "Cannot create tmp dir")

        tclfile = tmpdir + "/experiment.tcl"
        pid = "dummy"
        gid = "dummy"
        eid = "dummy"

        req = req.get('Ns2SplitRequestBody', None)
        if not req:
            raise service_error(service_error.req,
                    "Bad request format (no Ns2SplitRequestBody)")
        # The tcl parser needs to read a file so put the content into that file
        descr=req.get('description', None)
        if descr:
            file_content=descr.get('ns2description', None)
            if file_content:
                try:
                    f = open(tclfile, 'w')
                    f.write(file_content)
                    f.close()
                except IOError:
                    raise service_error(service_error.internal,
                            "Cannot write temp experiment description")
            else:
                raise service_error(service_error.req, 
                        "Only ns2descriptions supported")
        else:
            raise service_error(service_error.req, "No description")

        master = req.get('master', None)
        if master == None:
            raise service_error(service_error.req, "No master testbed label")
        
        include_fedkit = req.get('include_fedkit', None)
        if include_fedkit == None:
            raise service_error(service_error.req, 
                    "No federation kit inclusion directive")
        
        
        tclcmd = [self.tclsh, self.tcl_splitter, '-s', '-x', 
            str(self.muxmax), '-m', master]
        if include_fedkit: tclcmd.append('-k')
        tclcmd.extend([pid, gid, eid, tclfile])
        self.log.debug("Calling splitter %s" % " ".join(tclcmd))
        tclparser = Popen(tclcmd, stdout=PIPE)

        out = ""
        # Package up the split data
        for line in tclparser.stdout:
            out += line

        # Walk up tmpdir, deleting as we go
        for path, dirs, files in os.walk(tmpdir, topdown=False):
            for f in files:
                os.remove(os.path.join(path, f))
            for d in dirs:
                os.rmdir(os.path.join(path, d))
        os.rmdir(tmpdir)

        resp = { 'output' : out }

        return resp