sfa/rspecs/aggregates/rspec_manager_openflow.py

   1 from sfa.util.faults import *
   2 from sfa.util.misc import *
   3 from sfa.util.rspec import Rspec
   4 from sfa.server.registry import Registries
   5 from sfa.util.config import Config
   6 from sfa.plc.nodes import *
   7 import sys
   8
   9 #The following is not essential
  10 #from soaplib.wsgi_soap import SimpleWSGISoapApp
  11 #from soaplib.serializers.primitive import *
  12 #from soaplib.serializers.clazz import *
  13
  14 import socket
  15 import struct
  16
  17 # Message IDs for all the GENI light calls
  18 # This will be used by the aggrMgr controller
  19 SFA_GET_RESOURCES = 101
  20 SFA_CREATE_SLICE = 102
  21 SFA_START_SLICE = 103
  22 SFA_STOP_SLICE = 104
  23 SFA_DELETE_SLICE = 105
  24 SFA_GET_SLICES = 106
  25 SFA_RESET_SLICES = 107
  26
  27 DEBUG = 1
  28
  29 def print_buffer(buf):
  30     for i in range(0,len(buf)):
  31         print('%x' % buf[i])
  32
  33 def extract(sock):
  34     # Shud we first obtain the message length?
  35     # msg_len = socket.ntohs(sock.recv(2))
  36     msg = ""
  37
  38     while (1):
  39         try:
  40             chunk = sock.recv(1)
  41         except socket.error, message:
  42             if 'timed out' in message:
  43                 break
  44             else:
  45                 sys.exit("Socket error: " + message)
  46
  47         if len(chunk) == 0:
  48             break
  49         msg += chunk
  50
  51     print 'Done extracting %d bytes of response from aggrMgr' % len(msg)
  52     return msg
  53
  54 def connect(server, port):
  55     '''Connect to the Aggregate Manager module'''
  56     sock = socket.socket ( socket.AF_INET, socket.SOCK_STREAM )
  57     sock.connect ( ( server, port) )
  58     sock.settimeout(1)
  59     if DEBUG: print 'Connected!'
  60     return sock
  61
  62 def connect_aggrMgr():
  63     (aggr_mgr_ip, aggr_mgr_port) = Config().get_openflow_aggrMgr_info()
  64     if DEBUG: print """Connecting to port %d of %s""" % (aggr_mgr_port, aggr_mgr_ip)
  65     return connect(aggr_mgr_ip, aggr_mgr_port)
  66
  67 def generate_slide_id(cred, hrn):
  68     if cred == None:
  69         cred = ""
  70     if hrn == None:
  71         hrn = ""
  72     #return cred + '_' + hrn
  73     return str(hrn)
  74
  75 def msg_aggrMgr(cred, hrn, msg_id):
  76     slice_id = generate_slide_id(cred, hrn)
  77
  78     msg = struct.pack('> B%ds' % len(slice_id), msg_id, slice_id)
  79     buf = struct.pack('> H', len(msg)+2) + msg
  80
  81     try:
  82         aggrMgr_sock = connect_aggrMgr()
  83         aggrMgr_sock.send(buf)
  84         aggrMgr_sock.close()
  85         return 1
  86     except socket.error, message:
  87         print "Socket error"
  88     except IOerror, message:
  89         print "IO error"
  90     return 0
  91
  92 def start_slice(cred, hrn):
  93     if DEBUG: print "Received start_slice call"
  94     return msg_aggrMgr(SFA_START_SLICE)
  95
  96 def stop_slice(cred, hrn):
  97     if DEBUG: print "Received stop_slice call"
  98     return msg_aggrMgr(SFA_STOP_SLICE)
  99
 100 def delete_slice(cred, hrn):
 101     if DEBUG: print "Received delete_slice call"
 102     return msg_aggrMgr(SFA_DELETE_SLICE)
 103
 104 def reset_slices(cred, hrn):
 105     if DEBUG: print "Received reset_slices call"
 106     return msg_aggrMgr(SFA_RESET_SLICES)
 107
 108 def create_slice(cred, hrn, rspec):
 109     if DEBUG: print "Received create_slice call"
 110     slice_id = generate_slide_id(cred, hrn)
 111
 112     msg = struct.pack('> B%ds%ds' % (len(slice_id)+1, len(rspec)), SFA_CREATE_SLICE, slice_id, rspec)
 113     buf = struct.pack('> H', len(msg)+2) + msg
 114
 115     try:
 116         aggrMgr_sock = connect_aggrMgr()
 117         aggrMgr_sock.send(buf)
 118         if DEBUG: print "Sent %d bytes and closing connection" % len(buf)
 119         aggrMgr_sock.close()
 120
 121         if DEBUG: print "----------------"
 122         return 1
 123     except socket.error, message:
 124         print "Socket error"
 125     except IOerror, message:
 126         print "IO error"
 127     return 0
 128
 129 def get_rspec(cred, hrn=None):
 130     if DEBUG: print "Received get_rspec call"
 131     slice_id = generate_slide_id(cred, hrn)
 132
 133     msg = struct.pack('> B%ds' % len(slice_id), SFA_GET_RESOURCES, slice_id)
 134     buf = struct.pack('> H', len(msg)+2) + msg
 135
 136     try:
 137         aggrMgr_sock = connect_aggrMgr()
 138         aggrMgr_sock.send(buf)
 139         resource_list = extract(aggrMgr_sock);
 140         aggrMgr_sock.close()
 141
 142         if DEBUG: print "----------------"
 143         return resource_list
 144     except socket.error, message:
 145         print "Socket error"
 146     except IOerror, message:
 147         print "IO error"
 148     return None
 149
 150 def fetch_context(slice_hrn, user_hrn, contexts):
 151     return None
 152
 153 def main():
 154     r = Rspec()
 155     r.parseFile(sys.argv[1])
 156     rspec = r.toDict()
 157     create_slice(None,'plc',rspec)
 158
 159 if __name__ == "__main__":
 160     main()