之前,给大家分享了LoadRunner编写socket协议脚本的基本方法与规则,今天给大家分享下,如何从本地文件,读取内容,并作为报文,发送到服务端;该方法也是我在工作中遇到的一个难点,想通过这种方法,进行参数化几个交易的报文,通过迭代读取不同的文件,即可以发送不同的交易报文。
1. int和end与默认一致,不再做说明;
2. Action脚本编写
#include "lrs.h"
Action()
{
int rc,rv;//保存连接成功返回值
char *recvBuf;//保存接收数据的内容
int recvLen;//保存接收数据的大小
int count, total = 0;
char buffer[1000];
long file_stream;
char * filename = "E:/LR_date/ZHQZ/LR_scripts/baowen.txt";
if ((file_stream = fopen(filename, "rb")) == NULL ) {
lr_error_message ("Cannot open %s", filename);
return -1;
}
// Read until end of file
while (!feof(file_stream)) {
// Read 1000 bytes while maintaining a running count
count = fread(buffer,sizeof(char), 1000, file_stream);
lr_output_message ("%d bytes read", count);
// lr_output_message ("读取文件的buffer内容:\n %s", buffer);
if (ferror(file_stream)) { /* Check for file I/O errors */
lr_output_message ("Error reading file %s", filename);
break;
}
total += count; // Add up actual bytes read
}
// Display final total
lr_output_message ("Total number of bytes read = %d", total );
// Close the file stream
if (fclose(file_stream))
lr_error_message ("Error closing file %s", filename);
// memcpy(buffer+271,lr_eval_string("<custID>"),8);
lrs_save_param_ex("socket0", "user", buffer, 0, strlen(buffer),NULL, "paraBuf");
rc=lrs_create_socket("socket0","TCP","RemoteHost=188.123.123.233:1234", LrsLastArg);
lr_start_transaction("测试");//事务开始
//发送socket请求消息(数据包内容放在data.ws中)
lrs_send("socket0", "buf1", LrsLastArg); //取缓冲区buf0的报文并发送
rv = lrs_receive("socket0", "buf2", LrsLastArg);//接收响应报文
//获取最后收到的buffer和大小
lrs_get_last_received_buffer("socket0",&recvBuf,&recvLen);
/*设置检查点,验证返回数据是否成功,这个根据各交易具体情况进行判断,*/
if(recvLen>3)
{
lr_end_transaction("测试",PASS);
}
else
{
lr_end_transaction("测试",FAIL);
lr_error_message("测试失败!出错信息:[%s]",recvBuf);
}
lrs_free_buffer(recvBuf);//释放recvBuf内存空间,否则会引起内存泄露
lrs_close_socket("socket0");
return 0;
}
脚本上半部分是指定从某路径下,读取文件内容。有人要说那我如果想对文件中,某一字段进行参数化怎么办;于是我们就可以利用memcpy这个函数进行参数化,举例如:memcpy(buffer+271,lr_eval_string("<custID>"),8); //这是对读取报文的第272位到279位通过调用参数化池custID的值,共8位字节进行参数化。
3. data.ws编写
;WSRData 2 1
send buf1 683
"<paraBuf>"
recv buf2 350
""
-1
其中,报文在Action中已经获取到,由lrs_save_param_ex函数关联存储到paraBuf参数中;此时,我们可以直接通过该参数名调用即可,如上。
ENDing...