2019独角兽企业重金招聘Python工程师标准>>>
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 Packetreturn 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{ //非最后一个包,继续读
//...
}
//....