nik/FreedomBox
1
0
Fork 0
mirror of https://github.com/freedombox/FreedomBox.git synced 2026-01-21 07:55:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

wireguard: Write pre-shared key to tempfile

Browse Source 
Signed-off-by: James Valleroy <jvalleroy@mailbox.org>
Reviewed-by: Sunil Mohan Adapa <sunil@medhas.org>
This commit is contained in:
James Valleroy 2019-09-18 06:47:20 -04:00
parent a0ea33d9b6
commit 28bc880dc5
No known key found for this signature in database
GPG Key ID: 77C0C75E7B650808
1 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
plinth/modules/wireguard/views.py
View File

@ -18,6 +18,7 @@
Views for WireGuard application.
"""
import tempfile
import urllib.parse
from django.contrib import messages
@ -164,14 +165,19 @@ class AddServerView(SuccessMessageMixin, FormView):
all_outgoing_traffic = form.cleaned_data.get('all_outgoing_traffic')
args = ['add-server', '--endpoint', endpoint, '--client-ip',
client_ip_address, '--public-key', public_key]
optional_psk_file = None
if pre_shared_key:
# TODO: pass pre-shared key through stdin
args += ['--pre-shared-key', pre_shared_key]
optional_psk_file = tempfile.NamedTemporaryFile()
optional_psk_file.write(pre_shared_key.encode())
args += ['--pre-shared-key', optional_psk_file.name]
if all_outgoing_traffic:
args.append('--all-outgoing')
actions.superuser_run('wireguard', args)
if optional_psk_file:
optional_psk_file.close()
return super().form_valid(form)
@ -236,14 +242,19 @@ class EditServerView(SuccessMessageMixin, FormView):
all_outgoing_traffic = form.cleaned_data.get('all_outgoing_traffic')
args = ['modify-server', '--endpoint', endpoint, '--client-ip',
client_ip_address, '--public-key', public_key]
optional_psk_file = None
if pre_shared_key:
# TODO: pass pre-shared key through stdin
args += ['--pre-shared-key', pre_shared_key]
optional_psk_file = tempfile.NamedTemporaryFile()
optional_psk_file.write(pre_shared_key.encode())
args += ['--pre-shared-key', optional_psk_file.name]
if all_outgoing_traffic:
args.append('--all-outgoing')
actions.superuser_run('wireguard', args)
if optional_psk_file:
optional_psk_file.close()
return super().form_valid(form)