1、造数据,使响应的数据长度在每次不同请求中都一样。但实际上,方法1是有局限性的,也就是说有些请求通过造数据也不能使响应的数据长度一致,那么我们可以采用方法2。
2、自定义函数,动态解析并接受不定长响应数据流。
以下详细介绍下方法2,以举例讲解的方式来介绍:
【业务场景】:用户进行登录操作,每次登录的响应数据由于被加密压缩后才返回的缘故,导致长度不一致。
【协议简介】:用户登录操作采取的协议是自定义协议,协议头中第5,6个byte保存的是整个响应流的长度。
【自定义函数的思路】:先接受响应数据中的前6个bytes,然后取5,6位上的字节转换成int类型,得到整个响应流的长度,从而计算出剩下未被接受的数据长度,再接受剩下的数据。
【代码实现】:
第一部分,录制后未经修改的脚本如下:
Action()
{
lr_start_transaction("login");
lrs_create_socket("socket6", "TCP", "RemoteHost=172.16.4.16:1122", LrsLastArg);
// 登录请求
lr_think_time(5);
lrs_send("socket6", "buf1", LrsLastArg);
lrs_receive("socket6", "buf2", LrsLastArg);
lrs_close_socket("socket6");
lr_end_transaction("login", LR_AUTO);
return 0;
}
send buf1 49
"\x00\x00"
"¾¸"
"\x00"
"1"
"\x00"
" "
"\xc2"
"\t"
"\x00\xa7\xff\xd3\x00\x00\x00"
"\n"
"<fname>"
"\x00\xa7\xff\xd4\x00\x00\x00"
"\r"
"\x00\x00"
"1"
"\x00"
"2"
"\x00"
"3"
"\x00"
"4"
"\x00"
"5"
"\x00"
"6"
recv buf2 291 //每次请求的响应数据长度不一定是291个字节
【上述脚本的问题】:
recv buf2 291 后面的这个数字表示收到的buffer长度,这个长度在这里就固定死了,也就说每次执行这个脚本的时候都会按这个长度来接受解析响应数据,如果实际的响应数据长度与这个长度不一致会报以下错误:
Action.c(xx): Mismatch in buffer's length (expected 291 bytes, 295 bytes actually received, difference in 4 bytes)
第二部分,自定义函数和录制后修改的脚本如下:
为了能够动态接收响应数据,我们自定义了一个接收函数,如下:
#include "lrs.h"
/********************************************************************************自定义函数,用于动态接受返回的buffer
*0-6个字节是协议头的一部分,5-6个字节是整个响应包的长度
* 1.首先接受6个字节,并解析出第5,6个字节;
* 2. 计算出整个响应包的长度:length,并计算出剩下的长度:length-6;
* 3. 接受剩下的数据流;
********************************************************************************/int custom_lrs_receive(char *sock_desc, char *buf_desc,void *dummy)
{
int rc;
int buf_len = 6;
char szBytesLength[30], *buf = NULL, *pszError, *pszLastChar;
/*
* Get package header 0-6个bytes, [5..6] bytes is package length
*/
rc = lrs_receive_ex(sock_desc, buf_desc, "NumberOfBytesToRecv=6", LrsLastArg);
if (rc != 0) //正常情况下函数返回为0,非0表示函数有错误
{
lr_error_message("Receive 6 bytes failed. The error code = %d", rc);
return -1;
} /* Receive failed */
//判断前6个字节是否接受成功
lrs_get_last_received_buffer(sock_desc, &buf, &buf_len);if (buf == NULL || buf_len != 6)
{
lr_error_message("receive of %s failed", buf_desc);
return -1;
}
/* Compute buffer length */
sprintf (szBytesLength, "NumberOfBytesToRecv=%d", fiFromHexBinToInt(buf) - 6); //调用另一个自定义函数:计算总长度的函数
lr_debug_message(LR_MSG_CLASS_FULL_TRACE, "!!!! Bytes length = %s", szBytesLength);
rc = lrs_receive_ex(sock_desc, buf_desc, szBytesLength, LrsLastArg);
if (rc != 0) /* Receive failed */
return -1;
return 0;
}
/*
* 解析szBuffer中的5-6个字节,并转换成int类型
*/
int fiFromHexBinToInt(char *szBuffer)
{
int i, j, iIntValue = 0, iExp = 1;
/*lr_output_message("the szBuffer is %d %d",
szBuffer[5] & 0x000000ff,((szBuffer[6] & 0x000000ff));*/
for( i = 1; i >= 0; i--) //一个字节一个字节的取值,循环2次,分别取第6位,第5位上的字节
{
iExp = 1;
for (j = 2; j > i*2; j--) //从16进制字节流转换成int类型:2个byte4个bit,每个字节的低位分别需要乘以16的0次方和16的2次方;
iExp *= 16;iIntValue += (szBuffer[i+4] & 0x0000000f) * iExp + ((szBuffer[i+4] & 0x000000f0) >> 4) * iExp * 16;
}
lr_output_message("the length is %d", iIntValue);return iIntValue;
}
/*********************************************************************
*修改后的 测试脚本
*********************************************************************/
Action()
{
lr_start_transaction("login");
lrs_create_socket("socket6", "TCP", "RemoteHost=172.16.4.16:1122", LrsLastArg);
// 登录请求
lr_think_time(5);
lrs_send("socket6", "buf1", LrsLastArg);
custom_lrs_receive("socket6", "buf2", LrsLastArg); //自定义函数接受不定长数据流
lrs_close_socket("socket6");
lr_end_transaction("login", LR_AUTO);
return 0;
}
这样就ok了。对于使用这套协议的所有接口,该自定义函数可通用。