(转)clientX 异步客户端,支持SOCKS5

整理自: aar群 感谢 ҉k҉o҉m҉(znkee)提供

以下代码未经测试!

//clientX 异步客户端,支持SOCKS5

import crypt;
import crypt.bin; 
import inet.url;
import web;
import wsock.tcp.asynClient;

namespace web.socket;

class clientX{
	ctor(){
		this.readyState = 0;
		this.secKey = getSecKey();
		this.heartbeatInterval = 30;
		this.heartbeatData = "";
		this.heartbeatType = 0xA;
		this.userAgent = "Mozilla/5.0";
		..table.gc(this,"close");
	};
	close = function(code, reason){
		if(this.socket){
			if( ( this.readyState !=3 ) && ( this.readyState !=2 ) ){
				
				if( code && reason ){
					this.sendData({
						WORD code = ..raw.swap(code,"WORD");
						BYTE reason[] = reason;
					},8);
				} 
				else{
					this.sendData("",8);
				};
				
				this.readyState = 2;
			}
			this.socket.shutdown();
		}
	};
	_upgradeRequest = function(){
		var host = this.uri.host;
		if( this.uri.port!=80 ) host =  host +":"+ this.uri.port;
		
		if(!this.originUrl ){
			this.originUrl = "http://" + host +"/"; 
		}
		
		var headers = ..web.joinHeaders({
			["Connection"] = "Upgrade";
			["Host"] = host;
			["User-Agent"] = this.userAgent;
			["Pragma"] = "no-cache";
			["Accept"] = "*/*;";
			["Upgrade"] = "websocket";
			["Connection"] = "Upgrade"; 
			["Origin"] =  this.originUrl;
			["Sec-WebSocket-Version"] = "13";
			["Sec-WebSocket-Protocol"]  = this.protocol : "chat"; 
			["Sec-WebSocket-Key"] = this.secKey; 
			["Sec-WebSocket-Origin"] = this.url;
    		["Cache-Control"] = "no-cache";
    		["Accept-Language"] = "zh-CN,zh;q=0.9";
    		["Sec-WebSocket-Extensions"] = "permessage-deflate; client_max_window_bits";
		},this.headers)
				
		var reqPath = this.url : "/";
		if( this.uri.extraInfo ) reqPath = reqPath + this.uri.extraInfo;
		
		var ret  = this.socket.write('GET '+ reqPath  +' HTTP/1.1\r\n'+ headers +'\r\n')
	
		if(!ret){
			if( this.onError ) this.onError( 'Handshake failure'); 
			this.close();
			return false;
		}
		return true; 
	}
	
	connect = function(url,proxy){
		if(proxy){
			var tmp=..string.split(proxy,':')
			if(#tmp==2){
				this.proxy = {
					host = tmp[1];
					port = tonumber(tmp[2])
				};
				this.connectStep=1;	
			}
		}
		
		if(!url) url = this.url;
		if(!url) error("请指定连接网址,例如 ws://localhost:7511");
		
		if( this.isConnected() ) {
			this.close();
			..win.delay(100);
		};
		 
		this.readyState = 0;
		this.socket = ..wsock.tcp.asynClient(); 
		this.socket.bufferSize = 1024*10;
		
		this.webSocketCloseCode = null;	
		this.webSocketCloseReason = null;
				
		this.uri = ..inet.url.split(url);
		this.port = this.port:80;
		if( ..string.cmp(this.uri.scheme,"ws") !=0 ){
			error("错误的URL协议:" + this.uri.scheme,2);
		}
		this.url = url;
		
		this.socket.onConnect = function(err){
			if(err){
				if( this.onError ) this.onError( ..wsock.err.lasterr(err) ); 
				this.close();
				return; 
			};
			var ret;
			if(this.proxy){
				var tab = {
					_struct_aligned = 1;
					BYTE ver=5;
					BYTE num=1;
					BYTE method=0;
				}
				var ret  = this.socket.write(tab)
				if(!ret){
					if( this.onError ) this.onError( 'SOCKS5 认证失败 1'); 
					this.close();
					return;
				}
			}else {
				if(!this._upgradeRequest()){
					return ; 
				}
			}
		}
		
		if(this.proxy){
			this.socket.onRead = this._onSOCKS5Data;
		}else {
			this.socket.onRead = this._onReadHttpHeader;
		}
		
		this.socket.onClose = function(err){
			if( this.socket == owner ){
				this.readyState = 2;
				
				if( this.onClose ) {
					if(err){
						this.onClose( 
							code = 1006;
							reason = ..wsock.err.lasterr(err);
						); 
					}
					else {
						this.onClose( 
							code = this.webSocketCloseCode : 1000;
							reason = this.webSocketCloseReason 
						);
					}
				}
				
				if(( this.socket == owner )&&(this.readyState == 2)){
					this.webSocketCloseCode = null;	
					this.webSocketCloseReason = null;
					this.readyState = 3; 
					this.socket = null;	
				}
			}
			
			/*
			这里不能用this.socket.close(),
			因为这个套接字关闭事件被触发时，可能已经创建新的this.socket了
			*/
			owner.close();
		}
		if(this.proxy){
			return this.socket.connect(this.proxy.host, this.proxy.port);
		}else {
			return this.socket.connect(this.uri.host, this.uri.port:80);
		}	
	};
	isClosed = function(){
		if( this.readyState == 3 ) return true;
		if( !this.hSocket ) return true;
	};
	isConnected = function(){
		return ( this.readyState == 1 ) && ( owner.hSocket );
	};
	waitForConnected = function(hwnd,timeout){ 
		..win.delay(1000);
		return ..win.wait( 
			function(){
				if( this.readyState > 1 ) return false;
				if( this.readyState == 1 ) return true;
			},hwnd,timeout,1000
		);	 
	}; 
	_beginTranslateMessage = function(msg){ 
		if( this.onMessage ) { this.onMessage(msg); };	
		if( this._translateMessage ) { this._translateMessage(msg); };
	};
	_onReadMessage = function(err){
		if(err){
			if( this.onError ) this.onError( ..wsock.err.lasterr(err) );
			this.close();
			return;	
		}; 
		
		var msg = this._beginRecvMessage();
		if(!msg){ 
			return;
		};

		//control frames
		if( msg.type & 2#1000 ){
			if( msg.type == 8 ){//close
				if(#msg.data>=2){
					this.webSocketCloseCode = ..raw.swap( ..raw.convert(msg.data,{WORD code}).code );
					this.webSocketCloseReason = ..raw.tostring(msg.data,3);
				}
				else {
					this.webSocketCloseCode = 1005;	
					this.webSocketCloseReason = null;
				}
				
            	this.close();
        	}
        	elseif( msg.type == 9){//ping
            	this.sendData(msg.data,10);
        	}
        	elseif( msg.type == 10){//pong
        	}
			return;
		}
		
		if( msg.fin ){
			if( msg.type ){
				if( msg.type == 1 ) msg.data = ..raw.tostring(msg.data);
				this._beginTranslateMessage(msg);
			}
			else {
				if( this.onFragment ){
					this.onFragment(msg);
				}
				else {
					..table.push(this.cacheFragment.data,msg.data);
					this.cacheFragment.length = this.cacheFragment.length + #(msg.data);
					
					if( this.cacheFragment.type == 1 ) this.cacheFragment.data = ..string.join(this.cacheFragment.data);
					else {
						var len = 0;
						var buffer = ..raw.buffer(this.cacheFragment.length); 
						var data = this.cacheFragment.data;
						for(i=1;#data;1){
							..raw.concat(buffer,data[i],len)
							len = len + #data[i];
						}
						this.cacheFragment.data = buffer;
					}
					this._beginTranslateMessage(this.cacheFragment);
				}
			}
		}
		else {
			if( msg.type ){
				if( this.onFragment ){
					this.onFragment(msg);
				}	
				else {
					this.cacheFragment = msg;
					this.cacheFragment.data = {msg.data}
					this.cacheFragment.length = #(msg.data);
				}
				
			}
			else {
				if( this.onFragment ){
					this.onFragment(msg);
				}
				else {
					..table.push(this.cacheFragment.data,msg.data);
					this.cacheFragment.length = this.cacheFragment.length + #(msg.data);
				}
			}
		}
	};
	_onSOCKS5Data = function(err){
		if(err){
			if( this.onError ) this.onError( ..wsock.err.lasterr(err) );
			this.close();
			return;	
		};
		select(this.connectStep) {
			case 1 {
				this.connectStep++
				var tab = this.socket.read({
					_struct_aligned = 1;
					BYTE ver;
					BYTE method;
				})
				//清空缓存
				this.socket.readAll()
				if(tab.ver==5 and tab.method==0){
					//如果是域名
					var host = this.uri.host
					var port = this.uri.port
					if(..string.match(host,"\a")){
						tab = {
							_struct_aligned = 1;
							BYTE ver=5;
							BYTE cmd=1;
							BYTE resverd=0;
							BYTE type=3;
							BYTE len=#host;
							BYTE domain[]=host;
							WORD port=..raw.swap(port,"WORD");
						}	
					}else {
					//ip
						tab = {
							_struct_aligned = 1;
							BYTE ver=5;
							BYTE cmd=1;
							BYTE resverd=0;
							BYTE type=1;
							INT ip=..raw.swap(..wsock.htonl(..wsock.inet_addr(host)));
							WORD port=..raw.swap(port,"WORD");
						}
					}
					var ret  = this.socket.write(tab)
					if(!ret){
						if( this.onError ) this.onError( 'SOCKS5 认证失败 2'); 
						this.close();
					}
				}
			}
			case 2 {
				var tab = this.socket.read({
					_struct_aligned = 1;
					BYTE ver;
					BYTE code;
				})
				//清空缓存
				this.socket.readAll()
				if(tab.code==0){
					
					//重新绑定onRead
					this.socket.onRead = this._onReadHttpHeader;
					//发送协议升级
					this._upgradeRequest()	
				}else {
					if( this.onError ) this.onError( 'SOCKS5 认证失败 3'); 
				}
			}
		}
	}
	
	_onReadHttpHeader = function(err){
		if(err){
			if( this.onError ) this.onError( ..wsock.err.lasterr(err) );
			this.close();
			return;	
		};

		var responseHead = this.socket.readTo('\r\n\r\n');//HTTP头以两个回车换行结束
		if(!responseHead){ 
			return;	
		};
	
		var httpHeaders = ..string.split(responseHead,'<\r\n>');
		var statusLine = httpHeaders[1] ? ..string.splitEx(httpHeaders[1],"\s+",3); 
		if( ..string.startWith(statusLine[1],"HTTP/",true) ){
			this.status = statusLine[2];
			if( this.status == "101" ){
				..table.shift(httpHeaders,1); 
				this.responseHeaders = {};
				for(i=#httpHeaders;1;-1){
					var h = ..string.splitEx(httpHeaders[i],"\:\s*",2); 
					h[1] = ..string.trim(h[1]); h[2] = ..string.trim(h[2]);
					this.responseHeaders[ ..string.lower(h[1])] = h[2];  
				} 

				if( this.responseHeaders["sec-websocket-accept"] != getSecAccept(this.secKey) ){
					if( this.onError ) this.onError( 'Incorrect "Sec-WebSocket-Accept" header value'); 
					this.close();
					return;
				}
				this.readyState = 1;//连接成功
				if( this.onOpen ) this.onOpen();
				this.socket.onRead = this._onReadMessage;
				
				if( this.heartbeatInterval > 0 ){
					var d,t = this.heartbeatData,this.heartbeatType;
					this.socket.heartbeatTimerId = this.socket._form.addtimer( 
						this.heartbeatInterval * 1000,function(hwnd,msg,id,tick){
							if(!this.sendData(d,t)){//单向心跳
								if( this.onError ) this.onError( "Heartbeat failed" ); 
								this.socket.shutdown();
							}
						} 
					)
				}
			}
			else {
				if( this.onError ) this.onError( 'Unexpected response code:' + this.status);
				this.close();;
			}	
		}
		else {
			if( this.onError ) this.onError( 'Invalid status line');
			this.close();;
		}
	}
	send = function(data){
		var t = type(data);
		
		if( t = type.string )
			return this.sendData(data,1);

		if( t = type.buffer )
			return this.sendData(data,2);
			
		if( t[["_struct"]] )
			return this.sendData(..raw.buffer(data),2);
	}
	sendData = function(data,opcode = 1,fin=1,mask=1,rsv1 = 0,rsv2 = 0,rsv3 = 0){ 
		if(this.readyState != 1) {
			if( this.onError ) this.onError( 'Failed to execute "send/sendData"');
			return;
		};
		
		if( data[["_struct"]] ) data = ..raw.buffer(data);
			 
		var len = #data; 
		var buf = ..raw.buffer( len + 14);
		buf[1] = (fin << 7) | (rsv1 << 6) | (rsv2 << 5) | (rsv3 << 4) | opcode;

   		var w  = 2;
    	if (len <= 125) {
        	buf[2] = (mask << 7) | len; 
    	}
    	elseif(len < (1 << 16) ) {
        	buf[2] = (mask << 7) | 126;
        	..raw.convert({WORD v = ..raw.swap(len,"WORD"); },buf,2);
        	w = 4;
    	} 
    	else {
        	buf[2] = (mask << 7) | 127;
        	..raw.convert({LONG v=..math.size64(len).swap(); },buf,2);
        	w = 10;
    	}
    	
    	if(mask){
        	var k = ..string.random(4);
        	..raw.concat(buf,k,w);
        	w = w + 4;
        
        	for(i=1;len;1) buf[w+i] = data[i]  ^ k[ i% 4 : 4 ];
    	}
    	else {
    		..raw.concat(buf,data,w);
    	}
    
    	if(!this.socket.writeBuffer(buf,w + len)){
    		if( this.onError ) this.onError( ..wsock.err.lasterr() );
    		this.socket.shutdown();
    		return false;
    	}
    	return true;
	};
	_endRecvMessage4 = function(){
		var len = this.recvMessage.dataSize;
		
		if(len){
        	var buf = ..raw.buffer(len);
       		if( ! this.socket.readBuffer(buf) ){ 
				return;
       		}  
       		
        	var maskKey = this.recvMessage.maskKey;
        	if( maskKey ){ 
            	for(i=1;len;1) buf[i] = buf[i]  ^ maskKey[ i% 4 : 4 ];
        	}
        	this.recvMessage.data = buf;
        }
        else {
        	this.recvMessage.data = "";
        }
        
        
        var msg = this.recvMessage;
        this.recvMessage = null;
        return msg;
	};
	_recvMessageMaskKey3 = function(){
		var msg = this.recvMessage;
		
		//上次的消息还没有接收完整，继续尝试
		if( msg.maskKey ) 
			return this._endRecvMessage4();
 
        if( msg.mask ){
            msg.maskKey = this.socket.read(4); 
            if(!msg.maskKey) return;
        }
        
        return this._endRecvMessage4();    
	}
	_recvMessageDataSize2 = function(){
		var msg = this.recvMessage;
		
		//上次的消息还没有接收完整，继续尝试
		if( msg.dataSize!==null ) 
			return this._recvMessageMaskKey3();
			
		var len = msg.len;
        if (len == 126) {
            len = this.socket.read({WORD v});
            if(!len)  return; 
            msg.dataSize = ..raw.swap(len.v,"WORD");
        }
        elseif(len == 127) {
            len = this.socket.read({LONG size64 = ..math.size64() });
            if(!len)  return; 
            
            msg.dataSize = tonumber( len.size64.swap() ); 
        }
        else {
        	msg.dataSize = len;
        }
         
        return this._recvMessageMaskKey3();     
	}
	_beginRecvMessage = function(){
		if(!this.socket) return;
		
		//上次的消息还没有接收完整，继续尝试
		if( this.recvMessage ) 
			return this._recvMessageDataSize2();
		
		var msg = {}; 
	    var h = this.socket.read(2);
	    if(!h) {
	    	return; //异止套接字读取数据不完整会自动退回缓冲区
	    }
	    
        var h1,h2 = h[1],h[2];
        this.recvMessage = {
        	type = h1 & 2#1111;
            fin = (h1 >> 7) & 1;
            rsv1 = (h1 >> 6) & 1;
            rsv2 = (h1 >> 5) & 1;
            rsv3 = (h1 >> 4) & 1;
            mask = (h2 >> 7) & 1;
            len = h2 & 2#1111111;
        } 	
        
        return this._recvMessageDataSize2();
	}
}

namespace clientX{
    import console
	sha1 = function(data){
		var cr = ..crypt();
		var hash = cr.createHash( 0x8004/*_CALG_SHA1*/,data ); 
		var data = hash.getValue();
		hash.destroy();
		cr.release();
		
		return ..crypt.bin.encodeBase64(data);
	}
	
	getSecAccept = function(data){
		return sha1(..string.trim(data)+"258EAFA5-E914-47DA-95CA-C5AB0DC85B11");
	}
	
	getSecKey = function(){
		return ..crypt.bin.encodeBase64(..string.random(16) );
	}
}

/**intellisense()
web.socket.clientX = 支持单线程异步的WebSocket客户端\n可直接在界面线程中使用，支持SOCKS5代理，不会阻塞界面，不需要创建多线程\n支持服务端心跳(Ping/Pong帧)，客户端单向心跳(Pong帧)机制,\n可调用close函数断线，并可调用connect函数实现重新连接服务器
web.socket.clientX() = !stdwebsocketclientX.
web.socket.clientX.sha1(__) = 使用sha1算法取哈希值，并使用Base64编码为普通文本
web.socket.clientX.getSecKey() = 获取WebSocket客户端密钥
web.socket.clientX.getSecAccept(__) = 获取WebSocket客户端配对密钥，\n参数指定服务端HTTP头中sec-websocket-accept返回的值
end intellisense**/

/**intellisense(!stdwebsocketclientX)
socket = 异步套接字对象\n在关闭连接状态下此属性的值为null\n应由对象自动打开或删除套接字对象，调用者不可改动此属性的值\n!stdtcpaclient.
connect("ws://__","ip:port") = 重新连接到WebSocket服务端\n参数指定WebSocket服务端网址，例如 "ws://localhost:7511"\n如果不指定参数,则获取上次调用此函数指定的网址参数,\n第二个参数是SOCKS5代理\n如果之前也没有指定网址则抛出异常
waitForConnected(.(关联窗口句柄,超时) = 等待连接到WebSocket服务端\n所有参数可选,超时以毫秒为单位,\n\n连接成功返回true,失败返回false或null
url = 上次成功连接的网址\n也可以用于指定下次连接的默认网址
close() = 关闭连接\n可选增加2个参数指定发送给服务器的关闭帧附加数据:\n参数@1为数值类型的错误代码,参数@2为字符串类型错误描述
send(__) = 发送数据,支持字符串或缓冲区(buffer)、结构体\n字符串作为UTF8文本类型发送,其他以二进制类型发送，\n成功返回true
sendData(.(data,opcode,fin,mask,rsv1,rsv2,rsv3) = 发送WebSocket数据包\n参数@1支持支持字符串或缓冲区(buffer)、结构体\n除参数@1以外,所以参数可选\n一般应当调用send函数，而不是调用sendData函数\n\n如果一定要使用这个函数,请阅读此函数源码,以及WebSocket协议相关说明
onOpen = @.onOpen = function(){
	??.send("已连接到WebSocket服务");
}
onClose = @.onClose = function(e){
	__/*连接被关闭\ne.code为错误代码e.reason为错误原因*/	
}
onError = @.onError = function(err){
	__/*发生错误,err为错误信息*/
}
onMessage = @.onMessage = function(msg){
    __/*收到服务端数据\nmsg.type为1时msg.data为文本,\n否则msg.data为字节数组（buffer类型）*/
	
}
onFragment = @.onFragment = function(msg){
   __/*收到分片数据\n第一个数据包使用msg.type指明类型,参考WebSocket协议规范\n后续数据包msg.type为0,最后一个数据包msg.fin为1\n\n如果不指定这个回调函数,则自动并接分片数据后触发onMessage事件*/	
}
_translateMessage = 此回调函数的参数与onMessage相同,\n如果定义了这个回调函数,\n那么此函数将在调用onMessage以后被调用,\n这个函数提供了一个机会用于自动处理服务器消息\n,为其他需要扩展web.socket.client功能的库所预留，\n一旦定义将不能修改
headers = 其他HTTP请求头\n值可以是文本或数组、或键值对组成的表\n请求时会调用 web.joinHeaders()函数拼接并转换HTTP头\n该函数支持的类型和格式这个属性都可以支持
responseHeaders = 服务端响应的HTTP头\n这是一个表对象，键名都已转为小写
readyState = 连接状态,\n0为等待连接,1为已连接并准备就绪,2为正在关闭,3为已关闭\n只有成功通过WebSocket协议握手以后readyState才会被置为1\n这与socket.readyState连接成功就会置为1是不同的
isClosed() = 套接字是否已关闭
isConnected() = 套接字是否已连接并准备就绪(已与服务器握手成功)
secKey = 连接密钥,不可改动
heartbeatInterval = 客户端主动发送单向心跳（默认发送Pong空帧）间隔,默认为30秒\n设为-1时禁用客户端心跳,一般不建议禁用\n这个值修改以后，只能在下次调用connect函数才会生效
heartbeatData = 单向心跳发送的数据,默认为空数据\n这个值修改以后，只能在下次调用connect函数才会生效
heartbeatType = 单向心跳发送的的帧类型,\n默认为0xA,也就是Pong帧\n这个值修改以后，只能在下次调用connect函数才会生效
originUrl = 浏览器启动WebSocket客户端的网址\n一些WebSocket服务器根据这个判断是不是允许连接,\n所以有时候设置这个很重要\n默认使用WebSocket网址，并把 前面的ws://改为http://
userAgent = 客户端应用程序代理头\n默认为"Mozilla/5.0"
protocol = 应用程序支持的协议列表,默认为"chat"
end intellisense**/