Browse Source

module: private_key and preshared_keys can now be set

tags/v0.9
Maximilian Eschenbacher 6 years ago
parent
commit
1ed82deade
1 changed files with 20 additions and 5 deletions
  1. +20
    -5
      _modules/wireguard.py

+ 20
- 5
_modules/wireguard.py View File

import yaml import yaml
import os import os
from tempfile import mkstemp


__virtualname__ = 'wg' __virtualname__ = 'wg'


if fwmark: if fwmark:
s = '%s fwmark %s' % (s, fwmark) s = '%s fwmark %s' % (s, fwmark)
if private_key: if private_key:
assert os.stat(private_key)
# TODO private key must be given as file
s = '%s private-key %s' % (s, private_key)
fd, filename = mkstemp(text=True)
with open(filename, 'w') as f:
f.write(private_key)
os.close(fd)
s = '%s private-key %s' % (s, filename)
if peer: if peer:
s = '%s peer %s' % (s, peer) s = '%s peer %s' % (s, peer)
if preshared_key: if preshared_key:
s = '%s preshared-key %s' % (s, preshared_key)
fd2, filename2 = mkstemp(text=True)
with open(filename2, 'w') as f:
f.write(preshared_key)
os.close(fd2)
s = '%s preshared-key %s' % (s, filename2)
if endpoint: if endpoint:
s = '%s endpoint %s' % (s, endpoint) s = '%s endpoint %s' % (s, endpoint)
if persistent_keepalive: if persistent_keepalive:
s = '%s persistent-keepalive %s' % (s, persistent_keepalive) s = '%s persistent-keepalive %s' % (s, persistent_keepalive)
if allowed_ips: if allowed_ips:
s = '%s allowed-ips %s' % (s, allowed_ips) s = '%s allowed-ips %s' % (s, allowed_ips)
return __salt__['cmd.run'](s)

r = __salt__['cmd.run'](s)

if private_key:
os.unlink(filename)
if preshared_key:
os.unlink(filename2)

return r


def remove_peer(name, peer): def remove_peer(name, peer):
return __salt__['cmd.run']( return __salt__['cmd.run'](

Loading…
Cancel
Save