Proyecto

General

Perfil

61 manumora
##############################################################################
# -*- coding: utf-8 -*-
# Project: ControlIES
# Module: MainLoop.py
# Purpose: ControlIES web response
# Language: Python 2.5
# Date: 7-Feb-2011.
# Ver: 7-Feb-2011.
# Author: Manuel Mora Gordillo
# Copyright: 2011 - Manuel Mora Gordillo <manuito @no-spam@ gmail.com>
#
# ControlIES is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
# ControlIES is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
# You should have received a copy of the GNU General Public License
# along with ControlAula. If not, see <http://www.gnu.org/licenses/>.
#
##############################################################################

from twisted.web import server,resource,static
import twisted
import os
import simplejson as json
from Plugins.LdapConnection import LdapConnection, ILdapConnection

class ControlIESProtocol(resource.Resource):

isLeaf = True # This is a resource end point.

def __init__ (self):
resource.Resource.__init__(self)
self.PageDir=""


def render_GET(self, request):
session = request.getSession()
l = ILdapConnection(session)
connection = l.connect()

pagename=request.path[1:].lower()
62 manumora
61 manumora
if pagename=='':
request.path='/index.html'
pagename='index.html'

if pagename=='main.html' and connection==False:
request.path='/index.html'
pagename='index.html'

requestedfile = os.path.join(self.PageDir,request.path[1:])

if not os.path.isfile(requestedfile):
# Didn't find it? Return an error.
request.setResponseCode(404)
return"""
<html><head><title>404 - No Such Resource</title></head>
<body><h1>No Such Resource</h1>
<p>File not found: %s - No such file.</p></body></html>
""" % requestedfile

# Look for the requested file.
f=static.File(requestedfile)

if f.type is None:
f.type, f.encoding = static.getTypeAndEncoding(requestedfile,
f.contentTypes,
f.contentEncodings,
f.defaultType)

if f.type:
ctype=f.type.split(":")[0]
# Send the headers.
request.setHeader('content-type', ctype)

if f.encoding:
request.setHeader('content-encoding', f.encoding)
# Send the page.
if twisted.version.major >= 9:
static.NoRangeStaticProducer(request,f.openForReading()).start()
else:
static.FileTransfer(f.openForReading(), f.getFileSize(), request)
return server.NOT_DONE_YET


def render_POST(self, request):

command = request.path[1:]

session = request.getSession()
l = ILdapConnection(session)

if command == "authentication":

l.setCredentials(request.args['host'][0], request.args['user'][0], request.args['password'][0])
response = l.process()
return json.dumps({"connection" : response})

if command == "users":

from Plugins.Users import Users

if request.args['action'][0] == "list":
u = Users(l,"","","","","","","")
response = u.list(request.args)
return json.dumps(response)

elif request.args['action'][0] == "searchUsername":
u = Users(l,"",request.args['name'][0],request.args['surname'][0],"","","","")
response = u.searchNewUsername()
return json.dumps({"response" : response})

elif request.args['action'][0] == "getUserData":
u = Users(l,"","","","",request.args['user'][0],"","")
response = u.getUserData()
return json.dumps({"response" : response})

elif request.args['action'][0] == "delete":
u = Users(l,"","","","",request.args['user'][0],"","")
response = u.delete()
return json.dumps({"response" : response})

else:
u = Users(l,request.args['type'][0],request.args['name'][0],request.args['surname'][0],request.args['nif'][0],request.args['user'][0],request.args['password'][0],request.args['password2'][0])
response = u.process(request.args['action'][0])
return json.dumps({"response" : response})

# Gestion de Grupos
if command == "groups":

from Plugins.Groups import Groups

if request.args['action'][0] == "list":
g = Groups(l,"","")
response = g.list(request.args)
return json.dumps(response)
62 manumora
elif request.args['action'][0] == "getAllGroups":
g = Groups(l,"","")
response = g.getAllGroups()
return json.dumps(response)
61 manumora
else:
g = Groups(l,request.args['name'][0],request.args['gid'][0])
response = g.process(request.args['action'][0])
return json.dumps({"response" : response})

# Gestion de Grupos
if command == "hosts":

from Plugins.Hosts import Hosts

if request.args['action'][0] == "list":
h = Hosts (l,"","","","")
response = h.list(request.args)

#prueba para encender un equipo
#h = Hosts ("a21-pro","172.23.36.47","00:22:15:6e:9c:ba")
#print "Encendiendo host "+h.getName()
#h.wakeup()
return json.dumps(response)
else:
h = Hosts(l,request.args['name'][0],request.args['ip'][0],request.args['mac'][0])
print request.args['name'][0]
response = h.process(request.args['action'][0])
return json.dumps({"response" : response})

if command == "dhcp":
from Plugins.DHCP import DHCP
from Plugins.LdapConnection import LdapConnection
if request.args['action'][0] == "list":

#busqueda dhcp
search = l.search("cn=INTERNAL,cn=DHCP Config,dc=instituto,dc=extremadura,dc=es","dhcpOption=*",["dhcpOption"])
#obtengo valores de dhcpOPtion
#OJO: Hacer algo para que no haya desbordamiento de indices
dhcpOption = search[0][0][1]['dhcpOption']
subnet_mask = dhcpOption[0].replace('subnet-mask ','')
broadcast_address = dhcpOption[1].replace('broadcast-address ','')
routers = dhcpOption[2].replace('routers ','')
domain_name_servers = dhcpOption[3].replace('domain-name-servers ','')
domain_name = dhcpOption[4].replace('domain-name ','')
ntp_servers = dhcpOption[5].replace('ntp-servers ','')
log_servers = dhcpOption[6].replace('log-servers ','')
netbios_name_servers = dhcpOption[7].replace('netbios-name-servers ','')
netbios_node_type = dhcpOption[8].replace('netbios-node-type ','')

d = DHCP (self,subnet_mask,broadcast_address,routers,domain_name,ntp_servers,log_servers,netbios_name_servers,netbios_node_type)

print "Mascara de red: " + d.subnet_mask
print "Direccion de Broadcast: " + d.broadcast_address
#return json.dumps(response)
else:
h = Hosts(l,request.args['name'][0],request.args['ip'][0],request.args['mac'][0])
print request.args['name'][0]
response = h.process(request.args['action'][0])
return json.dumps({"response" : response})

if command == "logout":
request.getSession().expire()
return json.dumps({"response" : "logout"})

return

def response_fail(self, messages=None):
messages = messages or []
return json.dumps({
'status': 'fail',
'errors': messages,
'time': int(time())
})
def response_ok(self, **kwds):
kwds.update({'status': 'ok', 'time': int(time())})
return json.dumps(kwds)