cache ip lookup. add try/cache to socket writing

axiak · Feb 4, 2012 · fd4cea1 · fd4cea1
1 parent 8ad7d48
commit fd4cea1
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/example/yaluandmike.js b/example/yaluandmike.js
@@ -17,10 +17,17 @@ myProxy.on('enabledCheck', function (callback) {
 });
 
 /* Reject for Russian spies */
+var ipCache = {};
 myProxy.on('shouldReject', function (request, callback) {
     var client_ip = request.connection.remoteAddress || request.connection.socket.remoteAddress;
+    var cached = ipCache[client_ip];
+    if (cached !== undefined) {
+        callback(cached);
+        return;
+    }
     var result = geoip.lookup(client_ip);
     var should_reject = result && result['country'] != 'US' && result['country'] != 'CA';
+    ipCache[client_ip] = (should_reject ? true : false);
     callback(should_reject);
 });
 

diff --git a/lib/proxy.js b/lib/proxy.js
@@ -213,7 +213,14 @@ module.exports.createProxyServer = function (opts) {
                 socket.write('HTTP/1.0 200 Connection established\r\n\r\n');
             });
 
-            clientSocket.on('data', function (data) { socket.write(data); });
+            clientSocket.on('data', function (data) {
+                try {
+                    socket.write(data);
+                } catch (error) {
+                    emitter.emit('error', error);
+                }
+            });
+
             clientSocket.on('end', function () { socket.end(); });
 
             socket.on('data', function (data) {