bedohswe/almatyd
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add PROXY protocol support.

Browse Source
This commit is contained in:
bʰedoh₂ swé 2024-12-18 22:13:52 +05:00
parent a29b0ac898
commit 7590a3ba64
3 changed files with 23 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
index.ts
View File

@ -1,3 +1,7 @@
import * as net from "net";
import * as sq from "sqlite3";
import * as pp from "proxy-protocol-js";
if (process.argv[2] == null) { if (process.argv[2] == null) {
console.log("No IP specified"); console.log("No IP specified");
process.exit(1) process.exit(1)
@ -17,7 +21,6 @@ function msgToString(time: Date, address: string, msg: string) {
} }
import * as sq from "sqlite3";
const db = new sq.Database("data.db"); const db = new sq.Database("data.db");
db.run("CREATE TABLE IF NOT EXISTS messages (time INTEGER PRIMARY KEY, address TEXT NOT NULL, msg TEXT NOT NULL) STRICT"); db.run("CREATE TABLE IF NOT EXISTS messages (time INTEGER PRIMARY KEY, address TEXT NOT NULL, msg TEXT NOT NULL) STRICT");
@ -28,20 +31,20 @@ db.each("SELECT time, address, msg FROM messages", (err: Error, m: {time: number
}); });
const textEncoder = new TextEncoder(); const textEncoder = new TextEncoder();
import * as net from "net";
let srv = net.createServer(function(socket: net.Socket) { let srv = net.createServer(function(socket: net.Socket) {
let addr = socket.address() as net.AddressInfo; let a = socket.address() as net.AddressInfo;
if (!addr.hasOwnProperty('address')) { if (!a.hasOwnProperty('address')) {
console.log("?!?!?!"); console.log("?!?!?!");
socket.end(); socket.end();
return; return;
} }
let addr = a.address;
socket.on('data', function(data: Buffer | string) { socket.on('data', function(data: Buffer | string) {
switch (data[0]) { switch (data[0]) {
case 0x30: case 0x30:
const store = db.prepare("INSERT INTO messages VALUES (?,?,?)"); const store = db.prepare("INSERT INTO messages VALUES (?,?,?)");
let t = new Date; let t = new Date;
let a = addr.address; let a = addr;
let m = String(data.slice(1)).replace("/\n/g", "\\n").replace("/\x1b/g", "\\x1b"); let m = String(data.slice(1)).replace("/\n/g", "\\n").replace("/\x1b/g", "\\x1b");
msize += textEncoder.encode(msgToString(t, a, m)).length msize += textEncoder.encode(msgToString(t, a, m)).length
store.run(Number(t),a,m); store.run(Number(t),a,m);
@ -60,8 +63,15 @@ let srv = net.createServer(function(socket: net.Socket) {
socket.write(str); socket.write(str);
}); });
return; return;
case 'P'.charCodeAt(0):
if (addr == "127.0.0.1") {
addr = pp.V1ProxyProtocol.parse(String(data)).source.ipAddress;
} else {
console.log("Address " + addr + " tried to use proxy protocol.");
} }
console.log(addr.address + "???" + data[0]); return;
}
console.log(addr + "???" + data[0]);
}); });
socket.on('error', function(e: Error) { socket.on('error', function(e: Error) {
console.log(e.stack); console.log(e.stack);

6
package-lock.json generated
View File

@ -5,6 +5,7 @@
"packages": { "packages": {
"": { "": {
"dependencies": { "dependencies": {
"proxy-protocol-js": "^4.0.6",
"sqlite3": "^5.1.7" "sqlite3": "^5.1.7"
}, },
"devDependencies": { "devDependencies": {
@ -958,6 +959,11 @@
"node": ">=10" "node": ">=10"
} }
}, },
"node_modules/proxy-protocol-js": {
"version": "4.0.6",
"resolved": "https://registry.npmjs.org/proxy-protocol-js/-/proxy-protocol-js-4.0.6.tgz",
"integrity": "sha512-SjXgyBmr0dBbKUZ0jOzp0N9urTcDOI1cd1oEeE43W1vG4OMwYYLggCRcMJ0zv0gdTA8Imb4cAiYj8Ic/PWv1mw=="
},
"node_modules/pump": { "node_modules/pump": {
"version": "3.0.2", "version": "3.0.2",
"resolved": "https://registry.npmjs.org/pump/-/pump-3.0.2.tgz", "resolved": "https://registry.npmjs.org/pump/-/pump-3.0.2.tgz",

1
package.json
View File

@ -4,6 +4,7 @@
"typescript": "^5.7.2" "typescript": "^5.7.2"
}, },
"dependencies": { "dependencies": {
"proxy-protocol-js": "^4.0.6",
"sqlite3": "^5.1.7" "sqlite3": "^5.1.7"
} }
} }