mirror of
https://gitlab.torproject.org/tpo/anti-censorship/pluggable-transports/snowflake.git
synced 2025-10-13 11:11:30 -04:00
Lightly massage some of the generated JavaScript
This commit is contained in:
Arlo Breault
parent
31ad9566e6
commit
1867a3f121
19 changed files with 986 additions and 989 deletions
|
|
@ -1,4 +1,3 @@
|
|||
// Generated by CoffeeScript 2.4.1
|
||||
/*
|
||||
A Coffeescript WebRTC snowflake proxy
|
||||
|
||||
|
|
@ -9,174 +8,169 @@ this proxy must always act as the answerer.
|
|||
|
||||
TODO: More documentation
|
||||
*/
|
||||
var Snowflake;
|
||||
|
||||
Snowflake = (function() {
|
||||
// Minimum viable snowflake for now - just 1 client.
|
||||
class Snowflake {
|
||||
// Prepare the Snowflake with a Broker (to find clients) and optional UI.
|
||||
constructor(config, ui, broker) {
|
||||
// Receive an SDP offer from some client assigned by the Broker,
|
||||
// |pair| - an available ProxyPair.
|
||||
this.receiveOffer = this.receiveOffer.bind(this);
|
||||
this.config = config;
|
||||
this.ui = ui;
|
||||
this.broker = broker;
|
||||
this.state = Snowflake.MODE.INIT;
|
||||
this.proxyPairs = [];
|
||||
if (void 0 === this.config.rateLimitBytes) {
|
||||
this.rateLimit = new DummyRateLimit();
|
||||
} else {
|
||||
this.rateLimit = new BucketRateLimit(this.config.rateLimitBytes * this.config.rateLimitHistory, this.config.rateLimitHistory);
|
||||
}
|
||||
this.retries = 0;
|
||||
// Minimum viable snowflake for now - just 1 client.
|
||||
class Snowflake {
|
||||
|
||||
// Prepare the Snowflake with a Broker (to find clients) and optional UI.
|
||||
constructor(config, ui, broker) {
|
||||
// Receive an SDP offer from some client assigned by the Broker,
|
||||
// |pair| - an available ProxyPair.
|
||||
this.receiveOffer = this.receiveOffer.bind(this);
|
||||
this.config = config;
|
||||
this.ui = ui;
|
||||
this.broker = broker;
|
||||
this.state = Snowflake.MODE.INIT;
|
||||
this.proxyPairs = [];
|
||||
if (void 0 === this.config.rateLimitBytes) {
|
||||
this.rateLimit = new DummyRateLimit();
|
||||
} else {
|
||||
this.rateLimit = new BucketRateLimit(this.config.rateLimitBytes * this.config.rateLimitHistory, this.config.rateLimitHistory);
|
||||
}
|
||||
this.retries = 0;
|
||||
}
|
||||
|
||||
// Set the target relay address spec, which is expected to be websocket.
|
||||
// TODO: Should potentially fetch the target from broker later, or modify
|
||||
// entirely for the Tor-independent version.
|
||||
setRelayAddr(relayAddr) {
|
||||
this.relayAddr = relayAddr;
|
||||
log('Using ' + relayAddr.host + ':' + relayAddr.port + ' as Relay.');
|
||||
return true;
|
||||
// Set the target relay address spec, which is expected to be websocket.
|
||||
// TODO: Should potentially fetch the target from broker later, or modify
|
||||
// entirely for the Tor-independent version.
|
||||
setRelayAddr(relayAddr) {
|
||||
this.relayAddr = relayAddr;
|
||||
log('Using ' + relayAddr.host + ':' + relayAddr.port + ' as Relay.');
|
||||
return true;
|
||||
}
|
||||
|
||||
// Initialize WebRTC PeerConnection, which requires beginning the signalling
|
||||
// process. |pollBroker| automatically arranges signalling.
|
||||
beginWebRTC() {
|
||||
this.state = Snowflake.MODE.WEBRTC_CONNECTING;
|
||||
log('ProxyPair Slots: ' + this.proxyPairs.length);
|
||||
log('Snowflake IDs: ' + (this.proxyPairs.map(function(p) {
|
||||
return p.id;
|
||||
})).join(' | '));
|
||||
this.pollBroker();
|
||||
return this.pollInterval = setInterval((() => {
|
||||
return this.pollBroker();
|
||||
}), this.config.defaultBrokerPollInterval);
|
||||
}
|
||||
|
||||
// Regularly poll Broker for clients to serve until this snowflake is
|
||||
// serving at capacity, at which point stop polling.
|
||||
pollBroker() {
|
||||
var msg, pair, recv;
|
||||
// Poll broker for clients.
|
||||
pair = this.nextAvailableProxyPair();
|
||||
if (!pair) {
|
||||
log('At client capacity.');
|
||||
return;
|
||||
}
|
||||
|
||||
// Initialize WebRTC PeerConnection, which requires beginning the signalling
|
||||
// process. |pollBroker| automatically arranges signalling.
|
||||
beginWebRTC() {
|
||||
this.state = Snowflake.MODE.WEBRTC_CONNECTING;
|
||||
log('ProxyPair Slots: ' + this.proxyPairs.length);
|
||||
log('Snowflake IDs: ' + (this.proxyPairs.map(function(p) {
|
||||
return p.id;
|
||||
})).join(' | '));
|
||||
this.pollBroker();
|
||||
return this.pollInterval = setInterval((() => {
|
||||
return this.pollBroker();
|
||||
}), this.config.defaultBrokerPollInterval);
|
||||
// Do nothing until a new proxyPair is available.
|
||||
pair.active = true;
|
||||
msg = 'Polling for client ... ';
|
||||
if (this.retries > 0) {
|
||||
msg += '[retries: ' + this.retries + ']';
|
||||
}
|
||||
|
||||
// Regularly poll Broker for clients to serve until this snowflake is
|
||||
// serving at capacity, at which point stop polling.
|
||||
pollBroker() {
|
||||
var msg, pair, recv;
|
||||
// Poll broker for clients.
|
||||
pair = this.nextAvailableProxyPair();
|
||||
if (!pair) {
|
||||
log('At client capacity.');
|
||||
return;
|
||||
}
|
||||
// Do nothing until a new proxyPair is available.
|
||||
pair.active = true;
|
||||
msg = 'Polling for client ... ';
|
||||
if (this.retries > 0) {
|
||||
msg += '[retries: ' + this.retries + ']';
|
||||
}
|
||||
this.ui.setStatus(msg);
|
||||
recv = this.broker.getClientOffer(pair.id);
|
||||
recv.then((desc) => {
|
||||
if (pair.running) {
|
||||
if (!this.receiveOffer(pair, desc)) {
|
||||
return pair.active = false;
|
||||
}
|
||||
} else {
|
||||
this.ui.setStatus(msg);
|
||||
recv = this.broker.getClientOffer(pair.id);
|
||||
recv.then((desc) => {
|
||||
if (pair.running) {
|
||||
if (!this.receiveOffer(pair, desc)) {
|
||||
return pair.active = false;
|
||||
}
|
||||
}, function(err) {
|
||||
} else {
|
||||
return pair.active = false;
|
||||
});
|
||||
return this.retries++;
|
||||
}
|
||||
|
||||
// Returns the first ProxyPair that's available to connect.
|
||||
nextAvailableProxyPair() {
|
||||
if (this.proxyPairs.length < this.config.connectionsPerClient) {
|
||||
return this.makeProxyPair(this.relayAddr);
|
||||
}
|
||||
return this.proxyPairs.find(function(pp, i, arr) {
|
||||
return !pp.active;
|
||||
});
|
||||
}
|
||||
}, function(err) {
|
||||
return pair.active = false;
|
||||
});
|
||||
return this.retries++;
|
||||
}
|
||||
|
||||
receiveOffer(pair, desc) {
|
||||
var e, offer, sdp;
|
||||
try {
|
||||
offer = JSON.parse(desc);
|
||||
dbg('Received:\n\n' + offer.sdp + '\n');
|
||||
sdp = new SessionDescription(offer);
|
||||
if (pair.receiveWebRTCOffer(sdp)) {
|
||||
this.sendAnswer(pair);
|
||||
return true;
|
||||
} else {
|
||||
return false;
|
||||
}
|
||||
} catch (error) {
|
||||
e = error;
|
||||
log('ERROR: Unable to receive Offer: ' + e);
|
||||
// Returns the first ProxyPair that's available to connect.
|
||||
nextAvailableProxyPair() {
|
||||
if (this.proxyPairs.length < this.config.connectionsPerClient) {
|
||||
return this.makeProxyPair(this.relayAddr);
|
||||
}
|
||||
return this.proxyPairs.find(function(pp, i, arr) {
|
||||
return !pp.active;
|
||||
});
|
||||
}
|
||||
|
||||
receiveOffer(pair, desc) {
|
||||
var e, offer, sdp;
|
||||
try {
|
||||
offer = JSON.parse(desc);
|
||||
dbg('Received:\n\n' + offer.sdp + '\n');
|
||||
sdp = new SessionDescription(offer);
|
||||
if (pair.receiveWebRTCOffer(sdp)) {
|
||||
this.sendAnswer(pair);
|
||||
return true;
|
||||
} else {
|
||||
return false;
|
||||
}
|
||||
} catch (error) {
|
||||
e = error;
|
||||
log('ERROR: Unable to receive Offer: ' + e);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
sendAnswer(pair) {
|
||||
var fail, next;
|
||||
next = function(sdp) {
|
||||
dbg('webrtc: Answer ready.');
|
||||
return pair.pc.setLocalDescription(sdp);
|
||||
};
|
||||
fail = function() {
|
||||
return dbg('webrtc: Failed to create Answer');
|
||||
};
|
||||
return pair.pc.createAnswer().then(next).catch(fail);
|
||||
}
|
||||
sendAnswer(pair) {
|
||||
var fail, next;
|
||||
next = function(sdp) {
|
||||
dbg('webrtc: Answer ready.');
|
||||
return pair.pc.setLocalDescription(sdp);
|
||||
};
|
||||
fail = function() {
|
||||
return dbg('webrtc: Failed to create Answer');
|
||||
};
|
||||
return pair.pc.createAnswer().then(next).catch(fail);
|
||||
}
|
||||
|
||||
makeProxyPair(relay) {
|
||||
var pair;
|
||||
pair = new ProxyPair(relay, this.rateLimit, this.config.pcConfig);
|
||||
this.proxyPairs.push(pair);
|
||||
pair.onCleanup = (event) => {
|
||||
var ind;
|
||||
// Delete from the list of active proxy pairs.
|
||||
ind = this.proxyPairs.indexOf(pair);
|
||||
if (ind > -1) {
|
||||
return this.proxyPairs.splice(ind, 1);
|
||||
}
|
||||
};
|
||||
pair.begin();
|
||||
return pair;
|
||||
}
|
||||
|
||||
// Stop all proxypairs.
|
||||
disable() {
|
||||
var results;
|
||||
log('Disabling Snowflake.');
|
||||
clearInterval(this.pollInterval);
|
||||
results = [];
|
||||
while (this.proxyPairs.length > 0) {
|
||||
results.push(this.proxyPairs.pop().close());
|
||||
makeProxyPair(relay) {
|
||||
var pair;
|
||||
pair = new ProxyPair(relay, this.rateLimit, this.config.pcConfig);
|
||||
this.proxyPairs.push(pair);
|
||||
pair.onCleanup = (event) => {
|
||||
var ind;
|
||||
// Delete from the list of active proxy pairs.
|
||||
ind = this.proxyPairs.indexOf(pair);
|
||||
if (ind > -1) {
|
||||
return this.proxyPairs.splice(ind, 1);
|
||||
}
|
||||
return results;
|
||||
};
|
||||
pair.begin();
|
||||
return pair;
|
||||
}
|
||||
|
||||
// Stop all proxypairs.
|
||||
disable() {
|
||||
var results;
|
||||
log('Disabling Snowflake.');
|
||||
clearInterval(this.pollInterval);
|
||||
results = [];
|
||||
while (this.proxyPairs.length > 0) {
|
||||
results.push(this.proxyPairs.pop().close());
|
||||
}
|
||||
return results;
|
||||
}
|
||||
|
||||
};
|
||||
};
|
||||
|
||||
Snowflake.prototype.relayAddr = null;
|
||||
Snowflake.prototype.relayAddr = null;
|
||||
|
||||
Snowflake.prototype.rateLimit = null;
|
||||
Snowflake.prototype.rateLimit = null;
|
||||
|
||||
Snowflake.prototype.pollInterval = null;
|
||||
Snowflake.prototype.pollInterval = null;
|
||||
|
||||
Snowflake.prototype.retries = 0;
|
||||
Snowflake.prototype.retries = 0;
|
||||
|
||||
// Janky state machine
|
||||
Snowflake.MODE = {
|
||||
INIT: 0,
|
||||
WEBRTC_CONNECTING: 1,
|
||||
WEBRTC_READY: 2
|
||||
};
|
||||
// Janky state machine
|
||||
Snowflake.MODE = {
|
||||
INIT: 0,
|
||||
WEBRTC_CONNECTING: 1,
|
||||
WEBRTC_READY: 2
|
||||
};
|
||||
|
||||
Snowflake.MESSAGE = {
|
||||
CONFIRMATION: 'You\'re currently serving a Tor user via Snowflake.'
|
||||
};
|
||||
|
||||
return Snowflake;
|
||||
|
||||
}).call(this);
|
||||
Snowflake.MESSAGE = {
|
||||
CONFIRMATION: 'You\'re currently serving a Tor user via Snowflake.'
|
||||
};
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue