TDS for Sybase協議分析

參考文檔: http://www.cnblogs.com/yylqinghao/archive/2010/03/16/1687551.html

定義了一些TDS for Sybase常量

package org.sl.analysis;

/**
 * 定義SyBase TDS的各種類型<br/>
 * TDS版本:4.2/5.0<br/>
 * 格式:<br/>
 * INT8, INT8, INT16, 4 bytes <br/>
 * packet type, last packet indicator, 	packet size,unknown(always 0x00)
 * @author shanl
 *
 */
public interface SybaseTDSType {
	/**0x01 TDS 4.2 or 7.0 query*/
	byte PACKAGE_TYPE_QUERY_V42_OR_V70 = 0x01;	
	/**0x02 TDS 4.2 or 5.0 login packet*/
	byte PACKAGE_TYPE_LOGIN_V42_OR_V50 = 0x02;
	/** 0x03 RPC*/
	byte PACKAGE_TYPE_RPC = 0x03;
	/**0x04 responses from server*/
	byte PACKAGE_TYPE_RESPONSES_FROM_SERVER = 0x04;
	/**0x06 cancels*/
	byte PACKAGE_TYPE_CANCELS = 0x06;
	/**0x07 Used in Bulk Copy*/
	byte PACKAGE_TYPE_USERD_IN_BULK_COPY = 0x07;
	/**0x0F TDS 5.0 query*/
	byte PACKAGE_TYPE_QUERY_V50 = 0x0f;
	/**0x10 TDS 7.0 login packet*/
	byte PACKAGE_TYPE_LOGIN = 0x10;
	/***0x11 TDS 7.0 authentication packet*/
	byte PACKAGE_TYPE_AUTH = 0x11;
	/**0x12 TDS 8 prelogin packet*/
	byte PACKAGE_TYPE_PRELOGIN = 0x12;
	
	/**0x00 if more packets*/
	byte LAST_INDICATOR_MORE_PACKETS = 0x00;
	/**0x01 if last packet*/
	byte LAST_INDICATOR_LAST_PACKETS = 0x01;
}

TDS分析類(只能取出sybase通信時的sql)

Utility這個類在我以前的文章裏有。

package org.sl.analysis;

import java.util.Deque;
import java.util.LinkedList;

import org.sl.analysis.SybaseTDSType;
import org.sl.util.Utility;

/**
 * TDS協議分析適用於Sybase<br/>
 * TDS版本4.2應用Sybase SQL Server < 10 and Microsoft SQL Server 6.5<br/>
 * TDS版本5.0應用於Sybase SQL Server >= 10
 * @author shanl
 *
 */
public class TDSAnalyForSybase implements SybaseTDSType{
	private LinkedList<byte[]> packageBuffer = new LinkedList<byte[]>();
	
	/**
	 * 向緩存區插入一個數據庫
	 * @param bys
	 * @param offset
	 * @param len
	 * @return true:最後一個包,false:非最後一個數據庫
	 */
	public boolean addPackage(byte[] bys, int offset, int len){
		byte[] tmp = new byte[len];
		int lastFlag = bys[1];
		System.arraycopy(bys, offset, tmp, 0, len);
		this.packageBuffer.addLast(tmp);
		return LAST_INDICATOR_LAST_PACKETS==lastFlag;
	}
	
	/**
	 * 得到並從隊列中移除最後一個數據包
	 * @return
	 */
	public byte[] pollPackage(){
		return this.packageBuffer.pollFirst();
	}
	
	/**
	 * 取(並移除)最後一個數據包
	 * @param bys
	 * @param offset
	 * @return
	 */
	public int getFirstPackage(byte[] bys, int offset){
		byte[] tmp = pollPackage();
		System.arraycopy(tmp, 0, bys, offset, tmp.length);
		return tmp.length;
	}
			
//		//頭8個字節爲標誌
//		int packageType = bys[offset+0]; 	//0x0f表示TDS5.0 query; 0x02表示TDS 4.2 & 5.0 Login Packet
//		int lastFlag = bys[offset+1];		//最後一個包標誌,0x00表示還有更多包,0x01表示是最後一個包
//		int packageSize = Utility.byte2Short(bys, offset+2);	//包他長度
//		byte space0 = bys[offset+4];		//以下4(從當前位置算起)個爲0x00
//		byte space1 = bys[offset+5];
//		byte space2 = bys[offset+6];
//		byte space3 = bys[offset+7];		
//		//內容
//		byte flag0 = bys[offset+8]; 	//0x21 
//		byte flag1 = bys[offset+9]; 	//0x00
//		byte flag2 = bys[offset+10]; 	//0x00				
//		int sqlLen = Utility.byte2Short(bys, offset+11);	//sql長度
//				
//		byte[] flag = new byte[3];		
//		int packageCount = 0;
//	/**
//	 * 返回sql<br/>
//	 * 建議在測試isLastPackage()後，再執行此語句，否則你拿到的sql可能是不完整的。
//	 * @return 如果數據不完整或沒有sql則返回""
//	 */
//	public String getString(){
//		String sql = "";	
//		int packageSize = 0;
////		byte[] bys = null;
//		
////		for(int i=0,end=size(); i<end; i++){
//		for(byte[] bys: packageBuffer){
//			packageSize = Utility.byte2Short(bys, 2);
//			
//			//字節過於少的數據包將被拋棄
//			if(packageSize>=9){
//				if(0x21==bys[8] && 0x00==bys[9] && 0x00==bys[10]){
//					sql += new String(bys, 4+4+3+2, packageSize-(4+4+3+2));
////				}else if(0x01 == bys[1]){
////					sql += new String(bys, 4+4, packageSize-(4+4));
////					break;
//				}else if(0xE6==bys[8] && 0x00==bys[9] && 0x09==bys[10]){
//				}else{
//					sql += new String(bys, 4+4, packageSize-(4+4));
//				}				
//				
//				bys = null;
//			}			
//		}
//		
//		return sql;
//	}
	
	/**
	 * 返回sql語句
	 * @return
	 */
	public String getSql(){
		String sql = "";	
		int packageSize = 0;
		byte[] bys = null;
		boolean flagSql = false;
		int sysbaseHeaderSize = 4+4+4+2;
		
		for(int i=0,end=size(); i<end; i++){
//		for(byte[] bys: packageBuffer){
			bys = this.packageBuffer.get(i);
			packageSize = Utility.byte2Short(bys, 2);
			
			if(packageSize>=9){
				if(0x0f!=bys[0]){	//連接請求					
					break;
				}
				
//				if(0xe6==bys[8] && 0x00==bys[9] && 0x09==bys[10]){	//不符合sql語句規範
//					break;
//				}
				
				if(flagSql){
					sql += new String(bys, 4+4, packageSize-(4+4));
					continue;
				}
				
				if(0x21==bys[8] ){	//符合sql語句規範 //&& 0x00==bys[9] && 0x00==bys[10]		
					sql += new String(bys, sysbaseHeaderSize, packageSize-(sysbaseHeaderSize));
					flagSql = true;
				}else{	//不符合sql語句規範
					break;
				}				
			}
		}		
		reset();
		
		return sql;
	}
	
	/**
	 * 返回緩存區的數據包數
	 * @return
	 */
	public int size(){
		return this.packageBuffer.size();
	}
	
	/**
	 * 緩存區是否爲空
	 * @return
	 */
	public boolean isEmpty(){
		return this.packageBuffer.isEmpty();
	}
	
	/**
	 * 重置數據<br/>
	 * 調用此方法用清空舊數據
	 */
	public void reset(){
		this.packageBuffer.clear();
	}
	
	/**
	 * 返回最後一次插入的包類型
	 * @return 	0x0f表示TDS5.0 query;<br/> 
	 * 			0x02表示TDS 4.2 & 5.0 Login Packet
	 */
	public int getPackageType(){
		int packageType = packageBuffer.getLast()[0];; 	//0x0f表示TDS5.0 query; 0x02表示TDS 4.2 & 5.0 Login Packet
		return packageType;
	}
	
//	/**
//	 * 返回最後一次插入的包是否爲SQL的最後一個包
//	 * @return 	0:最後一個sql數據包 <br/>
//	 * 			1:非最後一個sql數據包 <br/>
//	 * 			2:非sql數據包 
//	 */
//	public int isSqlLastPackage(){
//		byte[] bys = packageBuffer.getLast();
//		int lastFlag = bys[1];		//最後一個包標誌,0x00表示還有更多包,0x01表示是最後一個包
//		int res = 2;
//		
//		if(LAST_INDICATOR_LAST_PACKETS==lastFlag){
//			if(0x21==bys[8] && 0x00==bys[9] && 0x00==bys[10]){ 			//sql標誌
//				res = 1;
//			}else if(0xE6==bys[8] && 0x00==bys[9] && 0x09==bys[10]){ 	//非sql標誌
//				res = 2;
//			}else{
//				res = 0;
//			}
//		}else{
//			res = 2;
//		}
//		
//		return res;
//	}
		
	/**
	 * 返回最後一次插入的包長度
	 * @return 長度>=0
	 */
	public int getPackageLength(){
		return Utility.byte2Short(packageBuffer.getLast(), 2);	
	}
}

測試類:

package test;

import org.sl.analysis.TDSAnalyForSybase;

public class Test5 {
	public static void main(String[] args){
		t1();
	}
	
	static void t1(){
		String hexStr = "" 
			+"0F0002000000000021000002060073656C656374204E414D453D492E6E616D652C204F"
			+"574E45523D552E6E616D652C20494E4445585F49443D492E696E6469642C20434F4C55"
			+"4D4E5F434F554E543D492E6B6579636E742C205354415455533D492E7374617475732C"
			+"20535441545553323D492E737461747573322C204D41585F524F575F5045525F504147"
			+"453D492E6D6178726F7773706572706167652C2046494C4C5F464143544F523D492E66"
			+"696C6C5F666163746F722C205245535F504147455F4741503D492E7265735F70616765"
			+"5F6761702C205345474D454E545F4E414D453D532E6E616D652C205441424C455F4E41"
			+"4D453D4F2E6E616D652066726F6D20736C5F64622E64626F2E7379736F626A65637473"
			+"204F2C20736C5F64622E64626F2E737973696E646578657320492C20736C5F64622E64"
			+"626F2E737973757365727320552C20736C5F64622E64626F2E7379737365676D656E74"
			+"73205320776865726520492E6964203D204F2E696420616E6420552E756964203D204F"
			+"2E75696420616E64204F2E69643D6F626A6563745F69642827736C5F64622E2E272920"
			+"616E6420492E7365676D656E74203D20532E7365676D656E7420616E6420492E696E64"
			+"6964206265747765656E203120616E642032353420616E642028492E73746174757332"
			+"2026203229203D203220616E6420492E6E616D65206C";
		
		byte[] bys = new byte[hexStr.length()/2];
		int hex = 0;
		int c = 0;
		int len = 0;
		TDSAnalyForSybase tds = new TDSAnalyForSybase();
		
		for(int i=0,j=0,endi=hexStr.length(); i<endi; j++){
			hex = Integer.valueOf(hexStr.substring(i, i+=2), 16);
			bys[j] = (byte)hex;			
		}		
		if(tds.addPackage(bys, 0, bys.length)){
			System.out.println(tds.getSql());
		}
			
		hexStr = "0F01001B00000000696B6520272527206F72646572206279203120";
		for(int i=0,j=0,endi=hexStr.length(); i<endi; j++){
			hex = Integer.valueOf(hexStr.substring(i, i+=2), 16);
			bys[j] = (byte)hex;			
		}
		tds.addPackage(bys, 0, hexStr.length()/2);
		if(tds.addPackage(bys, 0, bys.length)){
			System.out.println(tds.getSql());
		}
	}
}