node.jsでhttpsも扱えるProxyを作った

動機

話せば長いのですがｗ、iPhone4Sで合法テザリングをnode.jsベースで
作り出したのがきっかけです。

iPhone4SへはWebSocketでリクエストを投げ、3G回線へWebSocketを
転送、インターネットの受信側ではWebSocketを受け取りHTTPに変換
して目的のサイトへアクセスといった流れ。

3G回線からWebSocketで受けた後の動作がモロHTTP Proxy

3G回線の受け側でWebSocketで受け、これをhttpやhttpsで
通信を行う必要があったので、せっかくならnode.jsベースで実装してみるか！
ということで作ったのでした。

ソース

proxy.js

// 例外が発生してもサービスを停止しないようにする
process.on('uncaughtException', function(err) {
	console.log(err.stack);
});

var sys = require('util');
var net = require('net');
var url = require('url');

var server = net.createServer(function(c) {
	// 'connection' listener
	console.log('server connected');
	var socket = "";
	var isClientClosed = false;

	c.on('data', function(data) {
		c.pause();
		//if (socket == "") {
		// ソケットが未作成の場合、HTTP要求とみなし解析する。

		var parseheader = data.toString().split(/\n/);
		if (parseheader[0].match(/^GET/) || parseheader[0].match(/^CONNECT/)
				|| parseheader[0].match(/^POST/)) {
			var hthead = parseheader[0].split(/ /);
			// console.log(hthead[1]);
			var destPort = "";
			var destHost = "";
			if (parseheader[0].match(/^CONNECT/)) {
				// httpsの場合の処理
				// hostname:port
				var parseheader = data.toString().split(/\n/);
				var hthead = parseheader[0].split(/ /);

				var tmp = hthead[1].split(/:/);
				destHost = tmp[0];
				destPort = tmp[1];

			} else {
				// GET,POSTの場合の処理
				destPort = url.parse(hthead[1]).port || 80;
				destHost = url.parse(hthead[1]).host;
			}
			console.log("destPort = " + destPort);
			console.log("destHost = " + destHost);

			console.log("client send : " + parseheader[0].toString());
			//if(socket == "") {
			socket = net.createConnection(destPort, destHost, function() {
				console.log('server connected');

				if (parseheader[0].match(/^CONNECT/)) {
					c.pause();
					//socket.write(data);
					c.write("HTTP/1.0 200 OK\r\nConnection: close\r\n\r\n");
					c.resume();
				} else {
					socket.write(data);
				}

				// c.resume();
			});
			socket.on('data', function(data) {
				// socket.pause();
				if (isClientClosed) {
					socket.end();
					return;
				}
				c.pause();
				socket.pause();
				console.log('server send data');
				c.write(data);
				// socket.resume();
				c.resume();
				socket.resume();
			});

			socket.on('end', function() {
				console.log('server is disconnected');
				c.end();
				socket = "";
			});

			socket.on('close', function() {
				console.log('server has been disconnected');
				c.end();
				socket = "";
			});

			socket.on('error', function(err) {
				console.log("socket error! " + err);
				socket = "";
				c.end();
			});
			//} else {
				// 既に接続済みのソケットに対してのHTTP要求の場合
			//	console.log("socket has made! ");
			//}
		} else {
			if (socket != "") {
				socket.pause();
				c.pause();
				socket.write(data);
				c.resume();
				socket.resume();
			}
		}

		//	} else {
		//		console.log("client send ");
		//	if (socket != "") {
		//		socket.pause();
		//			c.pause();
		//			socket.write(data);
		//			c.resume();
		//			socket.resume();
		//		} else {
		//			console.log("Oops! no socket.");
		//			dumpResponse(data);
		//			c.end();
		//		}
		// socket.write(data);

		//	}

	});
	c.on('end', function() {
		console.log('client disconnected');
		if (socket != "") {
			socket.end();
			socket = "";
		}
		isClientClosed = true;
	});

	c.on('close', function() {
		console.log('client is closed');
		if (socket != "") {
			socket.end();
			socket = "";
		}
		isClientClosed = true;
	});

	c.on('error', function(err) {
		console.log('client error! ' + err);
		if (socket != "") {
			socket.end();
			socket = "";
		}
		isClientClosed = true;
	});
});

// https接続
function doConnect(client, data) {
	var parseheader = data.toString().split(/\n/);
	var hthead = parseheader[0].split(/ /);

	var tmp = hthead[1].split(/:/);
	var destHost = tmp[0];
	var destPort = tmp[1];
	var isClientClosed = false;

	console.log("destPort = " + destPort);
	console.log("destHost = " + destHost);

	var socket = net.createConnection(destPort, destHost, function() {
		console.log('server connected');
		client.write("HTTP/1.0 200 OK\r\nConnection: close\r\n\r\n");
		client.resume();
	});
	client.on('data', function(dt) {
		client.pause();
		socket.pause();
		console.log('client send');
		socket.write(dt);
		// client.resume();
		socket.resume();
		client.resume();
	});
	client.on('err', function(err) {
		console.log("client error! " + err);
		socket.end();
		isClientClosed = true;
	});
	client.on('close', function() {
		console.log("client has been closed.");
		socket.end();
		isClientClosed = true;
	});
	client.on('end', function() {
		console.log("client is closed.");
		socket.end();
		isClientClosed = true;
	});
	socket.on('data', function(dt) {
		socket.pause();

		if (isClientClosed) {
			console.log("client socket has been closed.");
			socket.end();
		}
		client.pause();
		// socket.pause();
		console.log('sever send');
		client.write(dt);
		client.resume();
		socket.resume();
	});
	socket.on('end', function() {
		client.end();
		console.log('sever close1');
	});
	socket.on('close', function() {
		client.end();
		console.log('sever close1');
	});
	socket.on('error', function(err) {
		console("socket error! " + err);
		client.end();
	});
	// socket.write(data);
	// client.pipe(socket);
	// socket.pipe(client);

}
var port = process.env.PORT || 8080;
server.listen(port, function() { // 'listening' listener
	console.log('server bound');
});

sys.puts('Server listening on port ' + port);

function dumpResponse(buf) {
	var tmp = "";
	// 表示できる文字は表示する
	for ( var i = 0; i < buf.length; i++) {
		var c = buf.readUInt8(i);
		if ((c > 31 && c < 127) || c == 13 || c == 10) {
			tmp = tmp + String.fromCharCode(c);
		} else {
			tmp = tmp + c + ":";
		}
	}
	console.log(tmp);
}

今年書いたブログ記事

• node.jsでhttpsも使えるHttp proxyを書いている

iPhone4Sで合法テザリングするソースのありか

• https://github.com/kjunichi/wsproxy