private byte[] MakeRequestPacket(uint reqMethod, string sql)
{
int bodyLen = 0;
int totalLen = 0;
if (sql.Length == 0)
{
throw new PDBException(PDBErrorCode.PdbE_SQL_ERROR, "SQL语句错误:执行的SQL语句不能为空");
}
byte[] tmpSql = Encoding.UTF8.GetBytes(sql);
bodyLen = tmpSql.Count();
totalLen = ProtoHeader.kHeadLen + bodyLen;
byte[] packetBuf = new byte[totalLen];
Array.Clear(packetBuf, 0, totalLen);
Array.Copy(tmpSql, 0, packetBuf, ProtoHeader.kHeadLen, tmpSql.Count());
uint bodyCrc = CRC.Crc32(packetBuf, ProtoHeader.kHeadLen, bodyLen);
ProtoHeader proHdr = new ProtoHeader(packetBuf);
proHdr.SetVersion(ProtoHeader.kProtoVersion);
proHdr.SetMethod(reqMethod);
proHdr.SetBodyLen((uint)bodyLen);
proHdr.SetReturnVal(0);
proHdr.SetRecordCnt(1);
proHdr.SetBodyCrc(bodyCrc);
proHdr.SetHeadCrc();
return(packetBuf);
}
private void Login()
{
byte[] loginPacket = new byte[ProtoHeader.kHeadLen + kUserNameLen + kPwdLen];
ProtoHeader proHdr = new ProtoHeader(loginPacket);
Array.Clear(loginPacket, 0, (loginPacket.Length));
byte[] nameArray = System.Text.Encoding.Default.GetBytes(UserName);
byte[] pwdArray = System.Text.Encoding.Default.GetBytes(Password);
if (nameArray.Length >= kUserNameLen)
{
throw new PDBException(PDBErrorCode.PdbE_INVALID_USER_NAME, "非法的用户名：用户名不合法");
}
for (int i = 0; i < nameArray.Length; i++)
{
loginPacket[ProtoHeader.kHeadLen + i] = nameArray[i];
}
UInt64 pwd64 = CRC.Crc64(pwdArray);
UInt32 pwd32 = (UInt32)(pwd64 & 0xFFFFFFFF);
IntTool.PutUint32(loginPacket, (ProtoHeader.kHeadLen + kUserNameLen), pwd32);
UInt32 bodyCrc = CRC.Crc32(loginPacket, ProtoHeader.kHeadLen, (kUserNameLen + kPwdLen));
proHdr.SetVersion(ProtoHeader.kProtoVersion);
proHdr.SetMethod(ProtoHeader.MethodCmdLoginReq);
proHdr.SetBodyLen((uint)(kUserNameLen + kPwdLen));
proHdr.SetReturnVal(0);
proHdr.SetRecordCnt((uint)0);
proHdr.SetBodyCrc(bodyCrc);
proHdr.SetHeadCrc();
Request(loginPacket);
byte[] resBuf = Recv();
ProtoHeader resHdr = new ProtoHeader(resBuf);
if (resHdr.GetMethod() != ProtoHeader.MethodCmdLoginRep)
{
throw new PDBException(PDBErrorCode.PdbE_PACKET_ERROR);
}
if (resHdr.GetReturnVal() != 0)
{
throw new PDBException(PDBErrorCode.PdbE_NET_ERROR, "网络错误：登录失败," + resHdr.GetReturnVal().ToString());
}
}
private byte[] MakeInsertTableRequestPacket(string tabName, DataTable dataTable)
{
int bodyLen = 0;
int totalLen = 0;
int fieldCnt = dataTable.Columns.Count;
if (string.IsNullOrEmpty(tabName))
{
throw new PDBException(PDBErrorCode.PdbE_INVALID_TABLE_NAME, "非法的表名");
}
if (Encoding.UTF8.GetByteCount(tabName) >= kTableNameLen)
{
throw new PDBException(PDBErrorCode.PdbE_INVALID_TABLE_NAME, "表名过长");
}
ByteStream byteStream = new ByteStream();
IntTool.WriteString(byteStream, tabName);
foreach (DataColumn colInfo in dataTable.Columns)
{
IntTool.WriteString(byteStream, colInfo.ColumnName);
}
foreach (DataRow dr in dataTable.Rows)
{
for (int fieldIdx = 0; fieldIdx < fieldCnt; fieldIdx++)
{
if (dr[fieldIdx] is DBNull)
{
IntTool.WriteNull(byteStream);
}
else if (dr[fieldIdx] is bool)
{
IntTool.WriteBool(byteStream, Convert.ToBoolean(dr[fieldIdx]));
}
else if (dr[fieldIdx] is sbyte)
{
IntTool.WriteVarint64(byteStream, Convert.ToSByte(dr[fieldIdx]));
}
else if (dr[fieldIdx] is short)
{
IntTool.WriteVarint16(byteStream, Convert.ToInt16(dr[fieldIdx]));
}
else if (dr[fieldIdx] is int)
{
IntTool.WriteVarint32(byteStream, Convert.ToInt32(dr[fieldIdx]));
}
else if (dr[fieldIdx] is long)
{
IntTool.WriteVarint64(byteStream, Convert.ToInt64(dr[fieldIdx]));
}
else if (dr[fieldIdx] is float)
{
IntTool.WriteFloat(byteStream, Convert.ToSingle(dr[fieldIdx]));
}
else if (dr[fieldIdx] is double)
{
IntTool.WriteDouble(byteStream, Convert.ToDouble(dr[fieldIdx]));
}
else if (dr[fieldIdx] is DateTime)
{
IntTool.WriteDatetime(byteStream, Convert.ToDateTime(dr[fieldIdx]));
}
else if (dr[fieldIdx] is string)
{
IntTool.WriteString(byteStream, Convert.ToString(dr[fieldIdx]));
}
else if (dr[fieldIdx] is byte[])
{
IntTool.WriteBlob(byteStream, (byte[])dr[fieldIdx]);
}
else
{
throw new Exception("不支持的值");
}
}
}
bodyLen = byteStream.GetTotalLen();
totalLen = ProtoHeader.kHeadLen + bodyLen;
byte[] packetBuf = new byte[totalLen];
Array.Clear(packetBuf, 0, totalLen);
byteStream.Read(packetBuf, ProtoHeader.kHeadLen);
uint bodyCrc = CRC.Crc32(packetBuf, ProtoHeader.kHeadLen, bodyLen);
ProtoHeader proHdr = new ProtoHeader(packetBuf);
proHdr.SetVersion(ProtoHeader.kProtoVersion);
proHdr.SetMethod(ProtoHeader.MethodCmdInsertTableReq);
proHdr.SetBodyLen((uint)bodyLen);
proHdr.SetFieldCnt((uint)fieldCnt);
proHdr.SetRecordCnt((uint)dataTable.Rows.Count);
proHdr.SetBodyCrc(bodyCrc);
proHdr.SetHeadCrc();
return(packetBuf);
}
