程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
 程式師世界 >> 數據庫知識 >> SyBase數據庫 >> SyBase綜合文章 >> Sybase TDS協議分析

Sybase TDS協議分析

編輯:SyBase綜合文章

TDS版本比較老,可能與當前版本有所不同。

Sybase與SQL Server部份版本 都采用TDS協議,但包大小等有所不同。


import Java.util.LinkedList;

/**
 * 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);	
	}
}

/**
 * 定義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;
}

package codedepot.hkproject.Net.ACSA.db.Sybase.util;

/**
 * 各基礎類型與byte之間的轉換
 * @author shanl
 *
 */
public class Utility {
	
	/**
	 * 將short轉成byte[2]
	 * @param a
	 * @return
	 */
	public static byte[] short2Byte(short a){
		byte[] b = new byte[2];
		
		b[0] = (byte) (a >> 8);
		b[1] = (byte) (a);
		
		return b;
	}
	
	/**
	 * 將short轉成byte[2]
	 * @param a
	 * @param b
	 * @param offset b中的偏移量
	 */
	public static void short2Byte(short a, byte[] b, int offset){
		b[offset] = (byte) (a >> 8);
		b[offset+1] = (byte) (a);
	}
	
	/**
	 * 將byte[2]轉換成short
	 * @param b
	 * @return
	 */
	public static short byte2Short(byte[] b){
		return (short) (((b[0] & 0xff) << 8) | (b[1] & 0xff));
	}
	
	/**
	 * 將byte[2]轉換成short
	 * @param b
	 * @param offset
	 * @return 
	 */
	public static short byte2Short(byte[] b, int offset){
		return (short) (((b[offset] & 0xff) << 8) | (b[offset+1] & 0xff));
	}

	/**
	 * long轉byte[8]
	 * 
	 * @param a
	 * @param b
	 * @param offset
	 *            b的偏移量
	 */
	public static void long2Byte(long a, byte[] b, int offset) {		
		b[offset + 0] = (byte) (a >> 56);
		b[offset + 1] = (byte) (a >> 48);
		b[offset + 2] = (byte) (a >> 40);
		b[offset + 3] = (byte) (a >> 32);

		b[offset + 4] = (byte) (a >> 24);
		b[offset + 5] = (byte) (a >> 16);
		b[offset + 6] = (byte) (a >> 8);
		b[offset + 7] = (byte) (a);
	}

	/**
	 * byte[8]轉long
	 * 
	 * @param b
	 * @param offset
	 *            b的偏移量
	 * @return
	 */
	public static long byte2Long(byte[] b, int offset) {
		 return ((((long) b[offset + 0] & 0xff) << 56)
		 | (((long) b[offset + 1] & 0xff) << 48)
		 | (((long) b[offset + 2] & 0xff) << 40)
		 | (((long) b[offset + 3] & 0xff) << 32)
		 
		 | (((long) b[offset + 4] & 0xff) << 24)
		 | (((long) b[offset + 5] & 0xff) << 16)
		 | (((long) b[offset + 6] & 0xff) << 8)
		 | (((long) b[offset + 7] & 0xff) << 0));
	}

	/**
	 * byte[8]轉long
	 * 
	 * @param b
	 * @return
	 */
	public static long byte2Long(byte[] b) {
		 return
		 ((b[0]&0xff)<<56)|
		 ((b[1]&0xff)<<48)|
		 ((b[2]&0xff)<<40)|
		 ((b[3]&0xff)<<32)|
		
		 ((b[4]&0xff)<<24)|
		 ((b[5]&0xff)<<16)|
		 ((b[6]&0xff)<<8)|
		 (b[7]&0xff);
	}

	/**
	 * long轉byte[8]
	 * 
	 * @param a
	 * @return
	 */
	public static byte[] long2Byte(long a) {
		byte[] b = new byte[4 * 2];

		b[0] = (byte) (a >> 56);
		b[1] = (byte) (a >> 48);
		b[2] = (byte) (a >> 40);
		b[3] = (byte) (a >> 32);
		
		b[4] = (byte) (a >> 24);
		b[5] = (byte) (a >> 16);
		b[6] = (byte) (a >> 8);
		b[7] = (byte) (a >> 0);

		return b;
	}

	/**
	 * byte數組轉int
	 * 
	 * @param b
	 * @return
	 */
	public static int byte2Int(byte[] b) {
		return ((b[0] & 0xff) << 24) | ((b[1] & 0xff) << 16)
				| ((b[2] & 0xff) << 8) | (b[3] & 0xff);
	}

	/**
	 * byte數組轉int
	 * 
	 * @param b
	 * @param offset
	 * @return
	 */
	public static int byte2Int(byte[] b, int offset) {
		return ((b[offset++] & 0xff) << 24) | ((b[offset++] & 0xff) << 16)
				| ((b[offset++] & 0xff) << 8) | (b[offset++] & 0xff);
	}

	/**
	 * int轉byte數組
	 * 
	 * @param a
	 * @return
	 */
	public static byte[] int2Byte(int a) {
		byte[] b = new byte[4];
		b[0] = (byte) (a >> 24);
		b[1] = (byte) (a >> 16);
		b[2] = (byte) (a >> 8);
		b[3] = (byte) (a);

		return b;
	}

	/**
	 * int轉byte數組
	 * 
	 * @param a
	 * @param b
	 * @param offset
	 * @return
	 */
	public static void int2Byte(int a, byte[] b, int offset) {		
		b[offset++] = (byte) (a >> 24);
		b[offset++] = (byte) (a >> 16);
		b[offset++] = (byte) (a >> 8);
		b[offset++] = (byte) (a);
	}
}


測試代碼:

TDSAnalyForSybase sySybase = new TDSAnalyForSybase();
byte[] buff = new byte[1024 * 4];
int ret_read = 0;

//....
ret_read = sybaseIn.read(buff);

if (sySybase.addPackage(buff, 0, ret_read)) {  //最後一個包
	System.out.println(sySybase.getSql());
}else{ //非最後一個包,繼續讀
//...
}
//....
  1. 上一頁:
  2. 下一頁:
Copyright © 程式師世界 All Rights Reserved