mirror of https://git.sr.ht/~cadence/bibliogram
87 lines
2.2 KiB
JavaScript
87 lines
2.2 KiB
JavaScript
const SocksProxyAgent = require("socks-proxy-agent")
|
|
const {connect} = require("net");
|
|
const constants = require("../constants")
|
|
const {request} = require("./request")
|
|
const {RequestCache} = require("../cache")
|
|
|
|
class TorManager {
|
|
/**
|
|
* @param {import("@deadcanaries/granax/lib/controller")} tor
|
|
* @param {number} port
|
|
*/
|
|
constructor(tor, port) {
|
|
this.tor = tor
|
|
this.port = port
|
|
this.agent = new SocksProxyAgent("socks5://localhost:"+this.port)
|
|
this.circuitManager = new RequestCache()
|
|
}
|
|
|
|
async request(url, test) {
|
|
let done = false
|
|
let g
|
|
while (!done) {
|
|
g = await request(url, {agent: this.agent}, {log: true, statusLine: "TOR"})
|
|
try {
|
|
await g.check(test)
|
|
break
|
|
} catch (e) {
|
|
await this.newCircuit()
|
|
}
|
|
}
|
|
return g
|
|
}
|
|
|
|
newCircuit() {
|
|
return this.circuitManager.getOrFetchPromise("circuit", () => {
|
|
console.log(" <> [TOR-CIR] Finding a new circuit...")
|
|
return new Promise(resolve => {
|
|
this.tor.cleanCircuits(() => resolve())
|
|
})
|
|
}).then(x => x.result)
|
|
}
|
|
}
|
|
|
|
try {
|
|
var granax = require("@deadcanaries/granax")
|
|
} catch (e) {}
|
|
|
|
/** @type {Promise<TorManager>} */
|
|
module.exports = new Promise(resolve => {
|
|
if (granax) {
|
|
/** @type {import("@deadcanaries/granax/lib/controller")} */
|
|
// @ts-ignore
|
|
let tor
|
|
if (constants.tor.password == null || constants.tor.port == null) {
|
|
// @ts-ignore
|
|
tor = new granax()
|
|
} else {
|
|
tor = new granax.TorController(connect(constants.tor.port), {authOnConnect: false})
|
|
tor.authenticate(`"${constants.tor.password}"`, err => {
|
|
if (err) console.log("Tor auth error:", err)
|
|
})
|
|
}
|
|
|
|
console.log("Starting Tor...")
|
|
|
|
tor.once("ready", () => {
|
|
tor.getInfo("net/listeners/socks", (err, result) => {
|
|
if (err) throw err
|
|
// result is string containing something like "127.0.0.1:36977"
|
|
// yes, the string contains double quotes!
|
|
const port = +result.match(/:(\d+)/)[1]
|
|
const torManager = new TorManager(tor, port)
|
|
console.log("Tor is ready, using SOCKS port "+port)
|
|
resolve(torManager)
|
|
})
|
|
})
|
|
|
|
tor.on("error", function() {
|
|
console.log("Tor error!")
|
|
console.log(...arguments)
|
|
})
|
|
} else {
|
|
console.log("Note: Tor functionality not installed. You may wish to run `npm install`. (78+ MB download required.)")
|
|
resolve(null)
|
|
}
|
|
})
|