Wireguard/Wireguard-tools.js
1
1
Fork 1
Code Issues 1 Pull Requests 1 Actions Projects 1 Releases Wiki Activity

Big code refactoring #10

Merged
Sirherobrine23 merged 15 commits from code_refactoring into main 2024-03-15 04:08:27 +00:00
Conversation 0 Commits 15 Files Changed 49 +163 -153
11 changed files with 163 additions and 153 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 3f33e77d8e - Show all commits

16
README.md
View File

@ -1 +1,17 @@
# Wireguard-tools for Nodejs
Manage your Wireguard interfaces directly from Node.js without any wrappers over `wg` or `wg-quick`
```js
```
## Licences
- `Wireguard-tools.js`: GPL-3.0
### Wireguard
- `Embeddable-wg-library`: LGPL-2.1+.
- `Wireguard-nt`: GPL-2.0
- `Wireguard-go`: MIT

6
addon/linux/wginterface.cpp
View File

@ -23,12 +23,10 @@ extern "C" {
#include <net/if.h>
std::string getWireguardVersion() {
return "Kernel";
return "0.0.0";
}
std::string driveLoad(std::map<std::string, std::string> load) {
throw std::string("Use userspace module");
}
std::string driveLoad(std::map<std::string, std::string> load) {}
void WireguardDevices::getInterfaces() {
size_t len; char *device_name, *devicesList = wg_list_device_names();

2
addon/main.cpp
View File

@ -19,7 +19,7 @@ Napi::Object StartAddon(const Napi::Env env, const Napi::Object exports) {
const Napi::Object Constants = Napi::Object::New(env), Userspace = Napi::Object::New(env);
if (getWireguardVersion().length() > 0) Constants.Set("driveVersion", getWireguardVersion());
Userspace.Set("driveVersion", WireguardUserspace::getVersion());
Userspace.Set("driveVersion", WireguardUserspace::getWireguardVersion());
exports.Set("deleteInterface", Napi::Function::New(env, [](const Napi::CallbackInfo &info) -> Napi::Value {
const Napi::Env env = info.Env();

2
addon/userspace/go/go.mod
View File

@ -3,11 +3,11 @@ module sirherobrine23.org/Wireguard/wireguard-tools.js/wg-tun
go 1.21.6
require (
golang.org/x/sys v0.12.0
golang.zx2c4.com/wireguard v0.0.0-20231211153847-12269c276173
)
require (
golang.org/x/sys v0.12.0 // indirect
golang.org/x/crypto v0.13.0 // indirect
golang.org/x/net v0.15.0 // indirect
golang.zx2c4.com/wintun v0.0.0-20230126152724-0fa3db229ce2 // indirect

98
addon/userspace/go/main.go
View File

@ -32,21 +32,35 @@ func wgVersion() *C.char {
return C.CString("unknown")
}
//export wgSocketDirectory
func wgSocketDirectory() *C.char {
return C.CString(socketDirectory)
}
//export deleteTun
func deleteTun(_tunName *C.char) bool {
tunName := C.GoString(_tunName)
if !existTun(_tunName) {
return false
}
//export existTun
func existTun(tunName string) bool {
Files, err := os.ReadDir(socketDirectory)
if err != nil {
return false
}
for _, file := range Files {
if file.IsDir() {
continue
}
splits := strings.Split(file.Name(), "/")
splits[len(splits)-1] = strings.TrimSuffix(splits[len(splits)-1], ".sock")
if splits[len(splits)-1] == tunName {
return true
}
}
return false
}
//export deleteTun
func deleteTun(_tunName *C.char) *C.char {
tunName := C.GoString(_tunName)
if !existTun(tunName) {
return C.CString("Tun does not exist")
}
Files, err := os.ReadDir(socketDirectory)
if err != nil {
return C.CString("Tun does not exist")
}
for _, file := range Files {
if file.IsDir() {
continue
@ -55,49 +69,21 @@ func deleteTun(_tunName *C.char) bool {
splits[len(splits)-1] = strings.TrimSuffix(splits[len(splits)-1], ".sock")
if splits[len(splits)-1] == tunName {
os.Remove(strings.Join(([]string{socketDirectory, file.Name()}), "/"))
return true
return C.CString("")
}
}
return false
return C.CString("Tun does not exist")
}
//export existTun
func existTun(_tunName *C.char) bool {
tunName := C.GoString(_tunName)
Files, err := os.ReadDir(socketDirectory)
if err != nil {
return false
}
for _, file := range Files {
if file.IsDir() {
continue
}
splits := strings.Split(file.Name(), "/")
splits[len(splits)-1] = strings.TrimSuffix(splits[len(splits)-1], ".sock")
if splits[len(splits)-1] == tunName {
return true
}
}
return false
}
//export getSocketPath
func getSocketPath(_tunName *C.char) *C.char {
tunName := C.GoString(_tunName)
uapi, err := ipc.UAPIOpen(tunName)
if err != nil {
return C.CString(fmt.Sprintf("Failed to get socket path: %v", err))
}
return C.CString(strings.Join(([]string{"/", uapi.Name()}), ""))
}
const levelLog = device.LogLevelVerbose
//export createTun
func createTun(_tunName *C.char) *C.char {
interfaceName := C.GoString(_tunName)
if existTun(_tunName) {
if existTun(interfaceName) {
return C.CString("Tun already exists")
}
logger := device.NewLogger(device.LogLevelSilent, fmt.Sprintf("(%s) ", interfaceName))
logger := device.NewLogger(levelLog, fmt.Sprintf("(%s) ", interfaceName))
// open TUN device (or use supplied fd)
tdev, err := tun.CreateTUN(interfaceName, device.DefaultMTU)
@ -133,6 +119,8 @@ func createTun(_tunName *C.char) *C.char {
uapiListened := uapi.Addr().String()
clean := func() {
logger.Verbosef("Shutting down")
uapi.Close()
dev.Close()
tdev.Close()
@ -145,17 +133,39 @@ func createTun(_tunName *C.char) *C.char {
for {
conn, err := uapi.Accept()
if err != nil {
logger.Verbosef("UAPI listener closed")
clean()
break
}
logger.Verbosef("UAPI recive message")
go dev.IpcHandle(conn)
}
}()
go func() {
logger.Verbosef("Device started")
<-dev.Wait()
logger.Verbosef("Device closing")
clean()
}()
logger.Verbosef("End start process")
return C.CString(strings.Join(([]string{"/", uapiListened}), ""))
}
// first\0second\0third\0forth\0last\0\0
//export listUapis
func listUapis() *C.char {
Files, err := os.ReadDir(socketDirectory)
if err != nil {
return C.CString("")
}
var uapis []string
for _, file := range Files {
if file.IsDir() {
continue
}
uapis = append(uapis, strings.Join(([]string{socketDirectory, file.Name()}), "/"))
}
return C.CString(strings.Join(uapis, "\x00") + "\x00")
}

43
addon/userspace/userspace.cpp Normal file
View File

@ -0,0 +1,43 @@
#include "userspace/wg-go.h"
#include "wginterface.hh"
#include <iostream>
#include <filesystem>
#include <string>
#include <vector>
// Get Wireguard-go version
std::string WireguardUserspace::getWireguardVersion() {
return wgVersion();
}
// Create tunel and return path to tunel
std::string WireguardUserspace::createWireguardTunnel(std::string wgName) {
std::string PathError = createTun((char*)wgName.c_str());
if (PathError.rfind("/", 0) != 0) throw PathError;
return PathError.substr(1);
}
// Delete tunel by name and return true if success
void WireguardUserspace::deleteWireguardTunnel(std::string wgName) {
std::string deleteStatus = deleteTun((char*)wgName.c_str());
if (!deleteStatus.empty()) throw deleteStatus;
}
std::vector<std::string> WireguardUserspace::listTunnels() {
std::vector<std::string> tunnels;
size_t len; char *device_name, *devicesList = listUapis();
// Set new devices
for (device_name = devicesList, len = 0; (len = strlen(device_name)); device_name += len + 1) tunnels.push_back(std::string(device_name));
return tunnels;
}
bool WireguardUserspace::checkIfExistTunnel(std::string wgName) {
const auto wgTunels = listTunnels();
if (wgTunels.size() == 0) return false;
for (auto tunelName : wgTunels) {
if (tunelName.find_last_of("/") != std::string::npos) tunelName = tunelName.substr(tunelName.find_last_of("/") + 1);
if (tunelName.find_last_of(".sock") != std::string::npos) tunelName = tunelName.substr(0, tunelName.find_last_of(".sock"));
if (tunelName == wgName) return true;
}
return false;
};

46
addon/userspace/wginterface.cpp
View File

@ -1,46 +0,0 @@
#include "wginterface.hh"
#include "userspace/wg-go.h"
#include <filesystem>
#include <string>
#include <vector>
// Get Wireguard-go version
std::string WireguardUserspace::getVersion() {
return wgVersion();
}
// Create tunel and return path to tunel
std::string WireguardUserspace::createTunel(std::string wgName) {
std::string PathError = createTun((char*)wgName.c_str());
if (PathError.rfind("/", 0) != 0) throw PathError;
return PathError.substr(1);
}
// Delete tunel by name and return true if success
bool WireguardUserspace::deleteTunel(std::string wgName) {
return !!deleteTun((char*)wgName.c_str());
}
// Check if tunel exist
bool WireguardUserspace::checkTunel(std::string wgName) {
return !!existTun((char*)wgName.c_str());
}
// Get uapi folder
std::string WireguardUserspace::getUapiDirectory() {
return wgSocketDirectory();
}
// List all tunels
std::vector<std::string> WireguardUserspace::listTunels() {
std::vector<std::string> tunels;
std::filesystem::path p = wgSocketDirectory();
if (std::filesystem::exists(p)) {
for (const auto & entry : std::filesystem::directory_iterator(p)) {
// Absolute path to tunel
tunels.push_back(entry.path().string());
}
}
return tunels;
}

16
addon/wg.hh
View File

@ -1,6 +1,6 @@
#ifndef __WG_NODE__
#define __WG_NODE__
#include <wginterface.hh>
#include "wginterface.hh"
#include <napi.h>
#include <functional>
#include <iostream>
@ -178,10 +178,16 @@ namespace WgUserspace {
public:
CreateWgTunnel(const Napi::Env &env, const Napi::String &name): Promised(env), tunName{name.Utf8Value()} {}
void runOk(std::function<void(Napi::Value)> callback) override {
Napi::HandleScope scope(Env());
callback(Napi::String::New(Env(), tunName));
}
void Execute() override {
try {
std::string status = WireguardUserspace::createTunel(tunName);
std::string status = WireguardUserspace::createWireguardTunnel(tunName);
if (status.rfind("/", 0) != 0) SetError(status);
else tunName = status;
} catch (std::string &err) { SetError(err); }
}
};
@ -193,7 +199,7 @@ namespace WgUserspace {
void Execute() override {
try {
WireguardUserspace::deleteTunel(tunName);
WireguardUserspace::deleteWireguardTunnel(tunName);
} catch (std::string &err) { SetError(err); }
}
};
@ -206,7 +212,7 @@ namespace WgUserspace {
void Execute() override {
try {
status = WireguardUserspace::checkTunel(tunName);
status = WireguardUserspace::checkIfExistTunnel(tunName);
} catch (std::string &err) { SetError(err); }
}
@ -223,7 +229,7 @@ namespace WgUserspace {
void Execute() override {
try {
tuns = WireguardUserspace::listTunels();
tuns = WireguardUserspace::listTunnels();
if (tuns.size() == 0) SetError("No interfaces found");
} catch (std::string &err) { SetError(err); }
}

17
addon/wginterface.hh
View File

@ -197,21 +197,18 @@ class WireguardConfig {
namespace WireguardUserspace {
// Get Wireguard-go version
std::string getVersion();
std::string getWireguardVersion();
// Create tunel and return path to tunel
std::string createTunel(std::string wgName);
std::string createWireguardTunnel(std::string wgName);
// Delete tunel by name and return true if success
bool deleteTunel(std::string wgName);
// Check if tunel exist
bool checkTunel(std::string wgName);
// Get uapi folder
std::string getUapiDirectory();
void deleteWireguardTunnel(std::string wgName);
// List all tunels
std::vector<std::string> listTunels();
std::vector<std::string> listTunnels();
// Check if tunel exist
bool checkIfExistTunnel(std::string wgName);
};
#endif

9
binding.yaml
View File

@ -1,6 +1,6 @@
name: wginterface
defines:
- "NODE_VERSION=9"
- "NODE_VERSION=8"
- "NAPI_CPP_EXCEPTIONS"
includes:
- node_modules/node-addon-api
@ -8,7 +8,7 @@ includes:
sources:
- "addon/main.cpp"
- "addon/genKey/wgkeys.cpp"
- "addon/userspace/wginterface.cpp"
- "addon/userspace/userspace.cpp"
# Set undefined functions to non supported system
- "addon/dummy/wginterface.cpp"
prebuild:
@ -16,9 +16,10 @@ prebuild:
cwd: ./addon/userspace/go
env:
CGO_ENABLED: "1"
LDFLAGS: "-w"
run: |
go build -ldflags=-w -trimpath -v -o ../wg-go.o -buildmode c-archive .
mv -vf ../wg-go.o ${BUILDDIR}/wg-go.o
go build -trimpath -v -o ../wg-go.o -buildmode c-archive .
mv -fv ../wg-go.o "${BUILDDIR}"
target:
linux:
sources:

61
src/wginterface.ts
View File

@ -70,9 +70,9 @@ const addon = (await loadAddon(path.resolve(__dirname, "../binding.yaml"))).wgin
/** Wireguard-go version */
driveVersion?: string;
createTunel(wgName: string): Promise<string>;
existTunel(wgName: string): Promise<boolean>;
deleteTunel(wgName: string): Promise<void>;
listTuns(): Promise<string[]>;
checkTunel(wgName: string): Promise<boolean>;
listTunels(): Promise<string[]>;
};
/** Kernel wireguard version */
@ -107,8 +107,8 @@ export namespace Userspace {
driveVersion,
createTunel,
deleteTunel,
listTuns,
existTunel,
checkTunel,
listTunels,
} = userspace || {};
};
@ -166,50 +166,35 @@ export class Wireguard {
if (!!Kernel.driveVersion) {
return Kernel.setConfig({ name, ...this.toJSON() });
}
if (!(await Userspace.existTunel(name))) await Userspace.createTunel(name);
const sockPath = (await Userspace.listTuns()).find((tun) => { let tt = tun.split("/").pop(); return (tt.endsWith(".sock") ? tt.slice(0, -5) : tt) === name; });
if (!(await Userspace.checkTunel(name))) await Userspace.createTunel(name);
const sockPath = (await Userspace.listTunels()).find((tun) => { let tt = tun.split("/").pop(); return (tt.endsWith(".sock") ? tt.slice(0, -5) : tt) === name; });
if (!sockPath) throw new Error("Wireguard interface not found");
const sock = net.connect(sockPath);
await new Promise<void>((resolve, reject) => sock.once("connect", resolve).once("error", reject));
const rl = readline.createInterface({ input: sock, output: sock });
let stop = [];
rl.on("line", (line): any => {
if (stop.length) {
stop.push(line);
if (line === "") sock.end();
return;
}
if (line.startsWith("errno=")) {
if (parseInt(line.slice(6)) !== 0) stop.push("");
}
});
rl.once("close", () => {
if (stop.length) throw new Error(stop.join("\n").trim());
});
rl.write(`set=1\n`);
if (this.#_privateKey.length == key.Base64Length) rl.write(`\nprivate_key=${key.keyToHex(this.#_privateKey)}\npublic_key=${key.keyToHex(this.publicKey)}`);
if (this.#_portListen >= 0) rl.write(`\nlisten_port=${this.#_portListen}`);
if (this.#_fwmark >= 0) rl.write(`\nfwmark=${this.#_fwmark}`);
if (this.address.size > 0) rl.write(`\naddress=${Array.from(this.address).join(",")}`);
if (this.#_replacePeers) rl.write(`\nreplace_peers=1`);
sock.write(`set=1\n`);
if (this.#_privateKey.length == key.Base64Length) sock.write(`\nprivate_key=${key.keyToHex(this.privateKey)}\npublic_key=${key.keyToHex(this.publicKey)}`);
if (this.#_portListen >= 0) sock.write(`\nlisten_port=${this.#_portListen}`);
if (this.#_fwmark >= 0) sock.write(`\nfwmark=${this.#_fwmark}`);
if (this.address.size > 0) sock.write(`\naddress=${Array.from(this.address).join(",")}`);
if (this.#_replacePeers) sock.write(`\nreplace_peers=1`);
for (const [key, value] of this.#_peers) {
rl.write(`\npublic_key=${key}`);
if (value.removeMe) rl.write(`\nremove=${key}`);
sock.write(`\npublic_key=${key}`);
if (value.removeMe) sock.write(`\nremove=${key}`);
else {
if (value.presharedKey) rl.write(`\npreshared_key=${value.presharedKey}`);
if (value.keepInterval) rl.write(`\npersistent_keepalive_interval=${value.keepInterval}`);
if (value.endpoint) rl.write(`\nendpoint=${value.endpoint}`);
if (value.allowedIPs) rl.write(`\nallowed_ips=${value.allowedIPs.join(",")}`);
if (value.presharedKey) sock.write(`\npreshared_key=${value.presharedKey}`);
if (value.keepInterval) sock.write(`\npersistent_keepalive_interval=${value.keepInterval}`);
if (value.endpoint) sock.write(`\nendpoint=${value.endpoint}`);
if (value.allowedIPs) sock.write(`\nallowed_ips=${value.allowedIPs.join(",")}`);
}
}
rl.write("\n\n");
return new Promise<void>((resolve, reject) => rl.once("close", () => resolve()).once("error", reject));
sock.end("\n\n");
return new Promise<void>((resolve, reject) => sock.once("close", () => resolve()).once("error", reject));
}
async getConfig(name: string) {
if (!!Kernel.driveVersion) return Kernel.getConfig(name);
const sockPath = (await Userspace.listTuns()).find((tun) => { let tt = tun.split("/").pop(); return (tt.endsWith(".sock") ? tt.slice(0, -5) : tt) === name; });
const sockPath = (await Userspace.listTunels()).find((tun) => { let tt = tun.split("/").pop(); return (tt.endsWith(".sock") ? tt.slice(0, -5) : tt) === name; });
if (!sockPath) throw new Error("Wireguard interface not found");
const sock = net.connect(sockPath);
await new Promise<void>((resolve, reject) => sock.once("connect", resolve).once("error", reject));
@ -229,13 +214,13 @@ export class Wireguard {
rl.once("close", () => {
if (stop.length) throw new Error(stop.join("\n").trim());
});
rl.write("get=1\n\n");
sock.write("get=1\n\n");
return null;
}
async deleteInterface(name: string): Promise<void> {
if (!!Kernel.driveVersion) return Kernel.deleteInterface(name);
if (await Userspace.existTunel(name)) return Userspace.deleteTunel(name);
if (await Userspace.checkTunel(name)) return Userspace.deleteTunel(name);
}
}
export default Wireguard;