def edit(request, plugin_id):
(mylar_key, mylar_secret) = utils.get_mylar_oauth_creds()
url = utils.get_rpc_url(request)
trans = OAuthTransport(url, key=mylar_key, secret=mylar_secret)
"""
Get the Mylar object
If it does not exist create a new entry
"""
try:
mylar = models.Mylar.objects.order_by('-id')[0]
except IndexError:
mylar = models.Mylar.objects.create()
try:
server = jsonrpclib.Server(url, transport=trans)
jail_path = server.plugins.jail.path(plugin_id)
jail = json.loads(server.plugins.jail.info(plugin_id))[0]['fields']
jail_ipv4 = jail['jail_ipv4'].split('/')[0]
auth = server.plugins.is_authenticated(
request.COOKIES.get("sessionid", ""))
assert auth
except Exception as e:
raise
if request.method == "GET":
form = forms.MylarForm(instance=mylar, jail_path=jail_path)
return render(request, "edit.html", {'form': form, 'ipv4': jail_ipv4})
if not request.POST:
return JsonResponse(request, error=True, message="A problem occurred.")
form = forms.MylarForm(request.POST, instance=mylar, jail_path=jail_path)
if form.is_valid():
form.save()
cmd = "%s restart" % utils.mylar_control
pipe = Popen(cmd,
stdin=PIPE,
stdout=PIPE,
stderr=PIPE,
shell=True,
close_fds=True)
return JsonResponse(request,
error=True,
message="Mylar settings successfully saved.")
return JsonResponse(request, form=form)
def start(request, plugin_id):
(mylar_key, mylar_secret) = utils.get_mylar_oauth_creds()
url = utils.get_rpc_url(request)
trans = OAuthTransport(url, key=mylar_key, secret=mylar_secret)
server = jsonrpclib.Server(url, transport=trans)
auth = server.plugins.is_authenticated(request.COOKIES.get(
"sessionid", ""))
jail_path = server.plugins.jail.path(plugin_id)
assert auth
try:
mylar = models.Mylar.objects.order_by('-id')[0]
mylar.enable = True
mylar.save()
except IndexError:
mylar = models.Mylar.objects.create(enable=True)
try:
form = forms.MylarForm(mylar.__dict__,
instance=mylar,
jail_path=jail_path)
form.is_valid()
form.save()
except ValueError:
return HttpResponse(simplejson.dumps({
'error':
True,
'message': ('Mylar data did not validate, configure '
'it first.'),
}),
content_type='application/json')
cmd = "%s onestart" % utils.mylar_control
pipe = Popen(cmd,
stdin=PIPE,
stdout=PIPE,
stderr=PIPE,
shell=True,
close_fds=True)
out = pipe.communicate()[0]
return HttpResponse(simplejson.dumps({
'error': False,
'message': out,
}),
content_type='application/json')