Exemplo n.º 1
0
def edit(request, plugin_id):
    (maraschino_key, maraschino_secret) = utils.get_maraschino_oauth_creds()
    url = utils.get_rpc_url(request)
    trans = OAuthTransport(url, key=maraschino_key, secret=maraschino_secret)
    """
    Get the Maraschino object
    If it does not exist create a new entry
    """
    try:
        maraschino = models.Maraschino.objects.order_by('-id')[0]
    except IndexError:
        maraschino = models.Maraschino.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.MaraschinoForm(instance=maraschino, 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.MaraschinoForm(request.POST,
                                instance=maraschino,
                                jail_path=jail_path)
    if form.is_valid():
        form.save()

        cmd = "%s restart" % utils.maraschino_control
        pipe = Popen(cmd,
                     stdin=PIPE,
                     stdout=PIPE,
                     stderr=PIPE,
                     shell=True,
                     close_fds=True)

        return JsonResponse(request,
                            error=True,
                            message="Maraschino settings successfully saved.")

    return JsonResponse(request, form=form)
Exemplo n.º 2
0
def start(request, plugin_id):
    (maraschino_key, maraschino_secret) = utils.get_maraschino_oauth_creds()

    url = utils.get_rpc_url(request)
    trans = OAuthTransport(url, key=maraschino_key, secret=maraschino_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:
        maraschino = models.Maraschino.objects.order_by('-id')[0]
        maraschino.enable = True
        maraschino.save()
    except IndexError:
        maraschino = models.Maraschino.objects.create(enable=True)

    try:
        form = forms.MaraschinoForm(maraschino.__dict__,
                                    instance=maraschino,
                                    jail_path=jail_path)
        form.is_valid()
        form.save()
    except ValueError:
        return HttpResponse(simplejson.dumps({
            'error':
            True,
            'message': ('Maraschino data did not validate, configure '
                        'it first.'),
        }),
                            content_type='application/json')

    cmd = "%s onestart" % utils.maraschino_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')