kamp/WGDashboard
1
0
Fork 0
mirror of https://github.com/donaldzou/WGDashboard.git synced 2024-11-22 07:10:09 +01:00
Code

Fixed bulk add

Browse Source
This commit is contained in:
Donald Zou 2024-11-06 20:58:53 +08:00
parent 4956b0d89d
commit 29e7a5ecdb
1 changed files with 61 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

73
src/dashboard.py
View File

@ -720,12 +720,48 @@ class WireguardConfiguration:
self.Peers.append(Peer(i, self)) self.Peers.append(Peer(i, self))
def addPeers(self, peers: list): def addPeers(self, peers: list):
try:
for i in peers:
newPeer = {
"id": i['id'],
"private_key": i['private_key'],
"DNS": i['DNS'],
"endpoint_allowed_ip": i['endpoint_allowed_ip'],
"name": i['name'],
"total_receive": 0,
"total_sent": 0,
"total_data": 0,
"endpoint": "N/A",
"status": "stopped",
"latest_handshake": "N/A",
"allowed_ip": i.get("allowed_ip", "N/A"),
"cumu_receive": 0,
"cumu_sent": 0,
"cumu_data": 0,
"traffic": [],
"mtu": i['mtu'],
"keepalive": i['keepalive'],
"remote_endpoint": DashboardConfig.GetConfig("Peers", "remote_endpoint")[1],
"preshared_key": i["preshared_key"]
}
sqlUpdate(
"""
INSERT INTO '%s'
VALUES (:id, :private_key, :DNS, :endpoint_allowed_ip, :name, :total_receive, :total_sent,
:total_data, :endpoint, :status, :latest_handshake, :allowed_ip, :cumu_receive, :cumu_sent,
:cumu_data, :mtu, :keepalive, :remote_endpoint, :preshared_key);
""" % self.Name
, newPeer)
for p in peers: for p in peers:
subprocess.check_output(f"wg set {self.Name} peer {p['id']} allowed-ips {p['allowed_ip']}", subprocess.check_output(f"wg set {self.Name} peer {p['id']} allowed-ips {p['allowed_ip']}",
shell=True, stderr=subprocess.STDOUT) shell=True, stderr=subprocess.STDOUT)
subprocess.check_output( subprocess.check_output(
f"wg-quick save {self.Name}", shell=True, stderr=subprocess.STDOUT) f"wg-quick save {self.Name}", shell=True, stderr=subprocess.STDOUT)
self.getPeersList() self.getPeersList()
return True
except Exception as e:
print(str(e))
return False
def searchPeer(self, publicKey): def searchPeer(self, publicKey):
for i in self.Peers: for i in self.Peers:
@ -2098,16 +2134,12 @@ def API_addPeers(configName):
dns_addresses = DashboardConfig.GetConfig("Peers", "peer_global_DNS")[1] dns_addresses = DashboardConfig.GetConfig("Peers", "peer_global_DNS")[1]
if len(endpoint_allowed_ip) == 0: if len(endpoint_allowed_ip) == 0:
endpoint_allowed_ip = DashboardConfig.GetConfig("Peers", "peer_endpoint_allowed_ip")[1] endpoint_allowed_ip = DashboardConfig.GetConfig("Peers", "peer_endpoint_allowed_ip")[1]
config = WireguardConfigurations.get(configName) config = WireguardConfigurations.get(configName)
if not bulkAdd and (len(public_key) == 0 or len(allowed_ips) == 0): if not bulkAdd and (len(public_key) == 0 or len(allowed_ips) == 0):
return ResponseObject(False, "Please provide at least public_key and allowed_ips") return ResponseObject(False, "Please provide at least public_key and allowed_ips")
if not config.getStatus(): if not config.getStatus():
config.toggleConfiguration() config.toggleConfiguration()
availableIps = _getWireguardConfigurationAvailableIP(configName) availableIps = _getWireguardConfigurationAvailableIP(configName)
if bulkAdd: if bulkAdd:
if type(preshared_key_bulkAdd) is not bool: if type(preshared_key_bulkAdd) is not bool:
preshared_key_bulkAdd = False preshared_key_bulkAdd = False
@ -2119,7 +2151,6 @@ def API_addPeers(configName):
if bulkAddAmount > len(availableIps[1]): if bulkAddAmount > len(availableIps[1]):
return ResponseObject(False, return ResponseObject(False,
f"The maximum number of peers can add is {len(availableIps[1])}") f"The maximum number of peers can add is {len(availableIps[1])}")
keyPairs = [] keyPairs = []
for i in range(bulkAddAmount): for i in range(bulkAddAmount):
newPrivateKey = _generatePrivateKey()[1] newPrivateKey = _generatePrivateKey()[1]
@ -2128,18 +2159,15 @@ def API_addPeers(configName):
"id": _generatePublicKey(newPrivateKey)[1], "id": _generatePublicKey(newPrivateKey)[1],
"preshared_key": (_generatePrivateKey()[1] if preshared_key_bulkAdd else ""), "preshared_key": (_generatePrivateKey()[1] if preshared_key_bulkAdd else ""),
"allowed_ip": availableIps[1][i], "allowed_ip": availableIps[1][i],
"name": f"BulkPeer #{(i + 1)}_{datetime.now().strftime('%Y%m%d_%H%M%S')}" "name": f"BulkPeer #{(i + 1)}_{datetime.now().strftime('%Y%m%d_%H%M%S')}",
"DNS": dns_addresses,
"endpoint_allowed_ip": endpoint_allowed_ip,
"mtu": mtu,
"keepalive": keep_alive
}) })
if len(keyPairs) == 0: if len(keyPairs) == 0:
return ResponseObject(False, "Generating key pairs by bulk failed") return ResponseObject(False, "Generating key pairs by bulk failed")
config.addPeers(keyPairs) config.addPeers(keyPairs)
for kp in keyPairs:
found, peer = config.searchPeer(kp['id'])
if found:
if not peer.updatePeer(kp['name'], kp['private_key'], kp['preshared_key'], dns_addresses,
kp['allowed_ip'], endpoint_allowed_ip, mtu, keep_alive):
return ResponseObject(False, "Failed to add peers in bulk")
return ResponseObject() return ResponseObject()
else: else:
@ -2152,11 +2180,20 @@ def API_addPeers(configName):
if i not in availableIps[1]: if i not in availableIps[1]:
return ResponseObject(False, f"This IP is not available: {i}") return ResponseObject(False, f"This IP is not available: {i}")
config.addPeers([{"id": public_key, "allowed_ip": ','.join(allowed_ips)}]) status = config.addPeers([
found, peer = config.searchPeer(public_key) {
if found: "name": name,
return peer.updatePeer(name, private_key, preshared_key, dns_addresses, ",".join(allowed_ips), "id": public_key,
endpoint_allowed_ip, mtu, keep_alive) "private_key": private_key,
"allowed_ip": ','.join(allowed_ips),
"preshared_key": preshared_key,
"endpoint_allowed_ip": endpoint_allowed_ip,
"DNS": dns_addresses,
"mtu": mtu,
"keepalive": keep_alive
}]
)
return ResponseObject(status)
except Exception as e: except Exception as e:
print(e) print(e)
return ResponseObject(False, "Add peers failed. Please see data for specific issue") return ResponseObject(False, "Add peers failed. Please see data for specific issue")