try{
GLOBAL.util = require('./lib/util');
}catch(e){}
var http = require('http'),
https = require('https'),
fs = require('fs'),
async = require("async"),
url = require('url'),
program = require('commander'),
color = require('colorful'),
certMgr = require("./lib/certMgr"),
getPort = require("./lib/getPort"),
Recorder = require("./lib/recorder"),
logUtil = require("./lib/log"),
wsServer = require("./lib/wsServer"),
webInterface = require("./lib/webInterface"),
SystemProxyMgr = require('./lib/systemProxyMgr'),
inherits = require("util").inherits,
util = require("./lib/util"),
path = require("path"),
juicer = require('juicer'),
events = require("events"),
express = require("express"),
ip = require("ip"),
ThrottleGroup = require("stream-throttle").ThrottleGroup,
iconv = require('iconv-lite'),
Buffer = require('buffer').Buffer;
var T_TYPE_HTTP = 0,
T_TYPE_HTTPS = 1,
DEFAULT_PORT = 8001,
DEFAULT_WEB_PORT = 8002, // port for web interface
DEFAULT_WEBSOCKET_PORT = 8003, // internal web socket for web interface, not for end users
DEFAULT_CONFIG_PORT = 8088,
DEFAULT_HOST = "localhost",
DEFAULT_TYPE = T_TYPE_HTTP;
var default_rule = util.freshRequire('./rule_default');
var requestHandler = util.freshRequire('./requestHandler');
//option
//option.type : 'http'(default) or 'https'
//option.port : 8001(default)
//option.hostname : localhost(default)
//option.rule : ruleModule
//option.webPort : 8002(default)
//option.socketPort : 8003(default)
//option.webConfigPort : 8088(default)
//option.dbFile : null(default)
//option.throttle : null(default)
//option.disableWebInterface
//option.silent : false(default)
//option.interceptHttps ,internal param for https
function proxyServer(option){
option = option || {};
var self = this,
proxyType = /https/i.test(option.type || DEFAULT_TYPE) ? T_TYPE_HTTPS : T_TYPE_HTTP ,
proxyPort = option.port || DEFAULT_PORT,
proxyHost = option.hostname || DEFAULT_HOST,
proxyRules = option.rule || default_rule,
proxyWebPort = option.webPort || DEFAULT_WEB_PORT, //port for web interface
socketPort = option.socketPort || DEFAULT_WEBSOCKET_PORT, //port for websocket
proxyConfigPort = option.webConfigPort || DEFAULT_CONFIG_PORT, //port to ui config server
disableWebInterface = !!option.disableWebInterface,
ifSilent = !!option.silent;
if(ifSilent){
logUtil.setPrintStatus(false);
}
// copy the rule to keep the original proxyRules independent
proxyRules = Object.assign({}, proxyRules);
var currentRule = requestHandler.setRules(proxyRules); //TODO : optimize calling for set rule
if(!!option.interceptHttps){
if (!certMgr.isRootCAFileExists()) {
util.showRootInstallTip();
process.exit(0);
return;
}
currentRule.setInterceptFlag(true);
//print a tip when using https features in Node < v0.12
var nodeVersion = Number(process.version.match(/^v(\d+\.\d+)/)[1]);
if(nodeVersion < 0.12){
logUtil.printLog(color.red("node >= v0.12 is required when trying to intercept HTTPS requests :("), logUtil.T_ERR);
}
logUtil.printLog(color.blue("The WebSocket will not work properly in the https intercept mode :("), logUtil.T_TIP);
}
if(option.throttle){
logUtil.printLog("throttle :" + option.throttle + "kb/s");
const rate = parseInt(option.throttle);
if (rate < 1) {
logUtil.printLog(color.red('Invalid throttle rate value, should be positive integer\n'), logUtil.T_ERR);
process.exit(0);
}
GLOBAL._throttle = new ThrottleGroup({rate: 1024 * parseFloat(option.throttle) }); // rate - byte/sec
}
self.httpProxyServer = null;
async.series(
[
//clear cache dir, prepare recorder
function(callback){
util.clearCacheDir(function(){
if(option.dbFile){
GLOBAL.recorder = new Recorder({filename: option.dbFile});
}else{
GLOBAL.recorder = new Recorder();
}
callback();
});
},
//creat proxy server
function(callback){
if(proxyType == T_TYPE_HTTPS){
certMgr.getCertificate(proxyHost,function(err,keyContent,crtContent){
if(err){
callback(err);
}else{
self.httpProxyServer = https.createServer({
key : keyContent,
cert: crtContent
},requestHandler.userRequestHandler);
callback(null);
}
});
}else{
self.httpProxyServer = http.createServer(requestHandler.userRequestHandler);
callback(null);
}
},
//handle CONNECT request for https over http
function(callback){
self.httpProxyServer.on('connect',requestHandler.connectReqHandler);
callback(null);
},
//start proxy server
function(callback){
self.httpProxyServer.listen(proxyPort);
callback(null);
},
//start web socket service
function(callback){
self.ws = new wsServer({port : socketPort});
callback(null);
},
//start web interface
function(callback){
if(disableWebInterface){
logUtil.printLog('web interface is disabled');
}else{
var config = {
port : proxyWebPort,
wsPort : socketPort,
userRule : proxyRules,
ip : ip.address()
};
self.webServerInstance = new webInterface(config);
}
callback(null);
},
//set global proxy
function(callback) {
if (option.setAsGlobalProxy) {
console.log('setting global proxy for you...');
if(!/^win/.test(process.platform) && !process.env.SUDO_UID){
console.log('sudo password may be required.');
}
var result = SystemProxyMgr.enableGlobalProxy(ip.address(), proxyPort, proxyType == T_TYPE_HTTP ? "Http" : "Https");
if (result.status) {
callback(result.stdout);
} else {
if(/^win/.test(process.platform)){
console.log('AnyProxy is now the default proxy for your system. It may take up to 1min to take effect.');
} else{
console.log('AnyProxy is now the default proxy for your system.');
}
callback(null);
}
} else {
callback(null);
}
},
//server status manager
function(callback){
process.on("exit",function(code){
logUtil.printLog('AnyProxy is about to exit with code: ' + code, logUtil.T_ERR);
if (option.setAsGlobalProxy) {
console.log('resigning global proxy...');
var result = SystemProxyMgr.disableGlobalProxy(proxyType == T_TYPE_HTTP ? "Http" : "Https");
if (result.status) {
console.log(color.red(result.stdout));
} else{
console.log('global proxy resigned.');
}
}
process.exit();
});
//exit cause ctrl+c
process.on("SIGINT", function() {
process.exit();
});
process.on("uncaughtException",function(err){
logUtil.printLog('Caught exception: ' + (err.stack || err), logUtil.T_ERR);
process.exit();
});
callback(null);
}
],
//final callback
function(err,result){
if(!err){
var webTip,webUrl;
webUrl = "http://" + ip.address() + ":" + proxyWebPort +"/";
webTip = "GUI interface started at : " + webUrl;
logUtil.printLog(color.green(webTip));
var tipText = (proxyType == T_TYPE_HTTP ? "Http" : "Https") + " proxy started at " + color.bold(ip.address() + ":" + proxyPort);
logUtil.printLog(color.green(tipText));
}else{
var tipText = "err when start proxy server :(";
logUtil.printLog(color.red(tipText), logUtil.T_ERR);
logUtil.printLog(err, logUtil.T_ERR);
}
}
);
self.close = function(){
self.httpProxyServer && self.httpProxyServer.close();
self.ws && self.ws.closeAll();
self.webServerInstance && self.webServerInstance.server && self.webServerInstance.server.close();
logUtil.printLog("server closed :" + proxyHost + ":" + proxyPort);
}
}
module.exports.proxyServer = proxyServer;
module.exports.generateRootCA = certMgr.generateRootCA;
module.exports.isRootCAFileExists = certMgr.isRootCAFileExists;
module.exports.setRules = requestHandler.setRules;