Sybase TDS协议分析

浏览数：32 / 时间：2015年06月12日

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{ //非最后一个包，继续读
//...
}
//....

郑重声明：本站内容如果来自互联网及其他传播媒体，其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享，并不代表本站赞同其观点和对其真实性负责，也不构成任何其他建议。

Sybase TDS协议分析