Пример #1
0
        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);
        }
Пример #2
0
        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());
            }
        }
Пример #3
0
        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);
        }