shadowsocks

### Copyright (c) 2014 clowwindy Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions: The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. ### net = require("net") fs = require("fs") path = require("path") udpRelay = require("./udprelay") utils = require('./utils') inet = require('./inet') Encryptor = require("./encrypt").Encryptor connections = 0 createServer = (serverAddr, serverPort, port, key, method, timeout, local_address='127.0.0.1') -> udpServer = udpRelay.createServer(local_address, port, serverAddr, serverPort, key, method, timeout, true) getServer = -> aPort = serverPort aServer = serverAddr if serverPort instanceof Array # support config like "server_port": [8081, 8082] aPort = serverPort[Math.floor(Math.random() * serverPort .length)] if serverAddr instanceof Array # support config like "server": ["123.123.123.1", "123.123.123.2"] aServer = serverAddr[Math.floor(Math.random() * serverAddr .length)] r = /^([^:]*)\:(\d+)$/.exec(aServer) # support config like "server": "123.123.123.1:8381" # or "server": ["123.123.123.1:8381", "123.123.123.2:8381", "123.123.123.2:8382"] if r? aServer = r[1] aPort = +r[2] return [aServer, aPort] server = net.createServer((connection) -> connections += 1 connected = true encryptor = new Encryptor(key, method) stage = 0 headerLength = 0 remote = null addrLen = 0 remoteAddr = null remotePort = null addrToSend = "" utils.debug "connections: #{connections}" clean = -> utils.debug "clean" connections -= 1 remote = null connection = null encryptor = null utils.debug "connections: #{connections}" connection.on "data", (data) -> utils.log utils.EVERYTHING, "connection on data" if stage is 5 # pipe sockets data = encryptor.encrypt data connection.pause() unless remote.write(data) return if stage is 0 tempBuf = new Buffer(2) tempBuf.write "\u0005\u0000", 0 connection.write tempBuf stage = 1 utils.debug "stage = 1" return if stage is 1 try # +----+-----+-------+------+----------+----------+ # |VER | CMD | RSV | ATYP | DST.ADDR | DST.PORT | # +----+-----+-------+------+----------+----------+ # | 1 | 1 | X'00' | 1 | Variable | 2 | # +----+-----+-------+------+----------+----------+ #cmd and addrtype cmd = data[1] addrtype = data[3] if cmd is 1 # TCP else if cmd is 3 # UDP utils.info "UDP assc request from #{connection.localAddress}:#{connection.localPort}" reply = new Buffer(10) reply.write "\u0005\u0000\u0000\u0001", 0, 4, "binary" utils.debug connection.localAddress utils.inetAton(connection.localAddress).copy reply, 4 reply.writeUInt16BE connection.localPort, 8 connection.write reply stage = 10 else utils.error "unsupported cmd: " + cmd reply = new Buffer("\u0005\u0007\u0000\u0001", "binary") connection.end reply return if addrtype is 3 addrLen = data[4] else unless addrtype in [1, 4] utils.error "unsupported addrtype: " + addrtype connection.destroy() return addrToSend = data.slice(3, 4).toString("binary") # read address and port if addrtype is 1 remoteAddr = utils.inetNtoa(data.slice(4, 8)) addrToSend += data.slice(4, 10).toString("binary") remotePort = data.readUInt16BE(8) headerLength = 10 else if addrtype is 4 remoteAddr = inet.inet_ntop(data.slice(4, 20)) addrToSend += data.slice(4, 22).toString("binary") remotePort = data.readUInt16BE(20) headerLength = 22 else remoteAddr = data.slice(5, 5 + addrLen).toString("binary") addrToSend += data.slice(4, 5 + addrLen + 2).toString("binary") remotePort = data.readUInt16BE(5 + addrLen) headerLength = 5 + addrLen + 2 if cmd is 3 utils.info "UDP assc: #{remoteAddr}:#{remotePort}" return buf = new Buffer(10) buf.write "\u0005\u0000\u0000\u0001", 0, 4, "binary" buf.write "\u0000\u0000\u0000\u0000", 4, 4, "binary" # 2222 can be any number between 1 and 65535 buf.writeInt16BE 2222, 8 connection.write buf # connect remote server [aServer, aPort] = getServer() utils.info "connecting #{aServer}:#{aPort}" remote = net.connect(aPort, aServer, -> remote.setNoDelay true if remote stage = 5 utils.debug "stage = 5" ) remote.on "data", (data) -> return if !connected # returns when connection disconnected utils.log utils.EVERYTHING, "remote on data" try if encryptor data = encryptor.decrypt data remote.pause() unless connection.write(data) else remote.destroy() catch e utils.error e remote.destroy() if remote connection.destroy() if connection remote.on "end", -> utils.debug "remote on end" connection.end() if connection remote.on "error", (e)-> utils.debug "remote on error" utils.error "remote #{remoteAddr}:#{remotePort} error: #{e}" remote.on "close", (had_error)-> utils.debug "remote on close:#{had_error}" if had_error connection.destroy() if connection else connection.end() if connection remote.on "drain", -> utils.debug "remote on drain" connection.resume() if connection remote.setTimeout timeout, -> utils.debug "remote on timeout" remote.destroy() if remote connection.destroy() if connection addrToSendBuf = new Buffer(addrToSend, "binary") addrToSendBuf = encryptor.encrypt addrToSendBuf remote.setNoDelay false remote.write addrToSendBuf if data.length > headerLength buf = new Buffer(data.length - headerLength) data.copy buf, 0, headerLength piece = encryptor.encrypt buf remote.write piece stage = 4 utils.debug "stage = 4" catch e # may encounter index out of range utils.error e connection.destroy() if connection remote.destroy() if remote clean() else if stage is 4 if not remote? connection.destroy() if connection return data = encryptor.encrypt data remote.setNoDelay true connection.pause() unless remote.write(data) connection.on "end", -> connected = false utils.debug "connection on end" remote.end() if remote connection.on "error", (e)-> utils.debug "connection on error" utils.error "local error: #{e}" connection.on "close", (had_error)-> connected = false utils.debug "connection on close:#{had_error}" if had_error remote.destroy() if remote else remote.end() if remote clean() connection.on "drain", -> # calling resume() when remote not is connected will crash node.js utils.debug "connection on drain" remote.resume() if remote and stage is 5 connection.setTimeout timeout, -> utils.debug "connection on timeout" remote.destroy() if remote connection.destroy() if connection ) if local_address? server.listen port, local_address, -> utils.info "local listening at #{server.address().address}:#{port}" else server.listen port, -> utils.info "local listening at 0.0.0.0:" + port server.on "error", (e) -> if e.code is "EADDRINUSE" utils.error "Address in use, aborting" else utils.error e server.on "close", -> udpServer.close() return server exports.createServer = createServer exports.main = -> console.log(utils.version) configFromArgs = utils.parseArgs() configPath = 'config.json' if configFromArgs.config_file configPath = configFromArgs.config_file if not fs.existsSync(configPath) configPath = path.resolve(__dirname, "config.json") if not fs.existsSync(configPath) configPath = path.resolve(__dirname, "../../config.json") if not fs.existsSync(configPath) configPath = null if configPath utils.info 'loading config from ' + configPath configContent = fs.readFileSync(configPath) try config = JSON.parse(configContent) catch e utils.error('found an error in config.json: ' + e.message) process.exit 1 else config = {} for k, v of configFromArgs config[k] = v if config.verbose utils.config(utils.DEBUG) utils.checkConfig config SERVER = config.server REMOTE_PORT = config.server_port PORT = config.local_port KEY = config.password METHOD = config.method local_address = config.local_address if not (SERVER and REMOTE_PORT and PORT and KEY) utils.warn 'config.json not found, you have to specify all config in commandline' process.exit 1 timeout = Math.floor(config.timeout * 1000) or 600000 s = createServer SERVER, REMOTE_PORT, PORT, KEY, METHOD, timeout, local_address s.on "error", (e) -> process.stdout.on 'drain', -> process.exit 1 if require.main is module exports.main()