Fedor Lyanguzov
53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
import subprocess
|
|
|
|
def user_cfg(private_key, n):
|
|
user_cfg = f'''
|
|
[Interface]
|
|
PrivateKey = {private_key}
|
|
Address = 10.0.0.{n}/32
|
|
DNS = 1.1.1.1
|
|
|
|
[Peer]
|
|
PublicKey = {public_key}
|
|
AllowedIPs = 0.0.0.0/0, ::/0
|
|
Endpoint = 89.19.211.56:51820
|
|
'''
|
|
return user_cfg
|
|
|
|
def server_cfg(peers):
|
|
server_cfg = '''
|
|
[Interface]
|
|
PrivateKey = {private_key}
|
|
Address = 10.0.0.1/32
|
|
PostUp = iptables -A FORWARD -i wg0 -j ACCEPT; iptables -t nat -A POSTROUTING -o eth0 -j MASQUERADE
|
|
PostDown = iptables -D FORWARD -i wg0 -j ACCEPT; iptables -t nat -D POSTROUTING -o eth0 -j MASQUERADE
|
|
ListenPort = 51820
|
|
|
|
[Peer]
|
|
PublicKey = {public_key}
|
|
AllowedIPs = 10.0.0.2/32
|
|
'''
|
|
return '\n\n'.join([server_cfg]+list(map(lambda x: peer_cfg(*x), enumerate(peers, 3))))
|
|
|
|
def peer_cfg(n, peer_public_key):
|
|
peer_cfg = f'''
|
|
[Peer]
|
|
PublicKey = {peer_public_key}
|
|
AllowedIPs = 10.0.0.{n}/32
|
|
'''
|
|
return peer_cfg
|
|
|
|
peers = []
|
|
for n, user in enumerate(['fedor-walker', 'fedor-phone', 'polina-notebook', 'polina-phone'], 3):
|
|
with subprocess.Popen('wg genkey | tee privatekey | wg pubkey > publickey', shell=True):
|
|
pass
|
|
with open('privatekey') as f:
|
|
private_key = f.read().strip()
|
|
with open('publickey') as f:
|
|
public_key = f.read().strip()
|
|
with open(f'{user}.conf', 'w') as f:
|
|
f.write(user_cfg(private_key, n))
|
|
peers.append(public_key)
|
|
with open(f'server.conf', 'w') as f:
|
|
f.write(server_cfg(peers))
|