LoadRunner简单应用之：win/websockets通信中获取服务器返回数据

通过socket进行通信的分布式应用是比较常见的，包括winsocket和websocket，LR支持这两个协议。

在使用LR回放socket协议的脚本时，可能会经常需要获取服务器返回的数据，语句格式如下：

    lrs_create_socket("socket0", "TCP", "RemoteHost=1.2.3.4:12345",  LrsLastArg);

    lrs_send("socket0", "buf0", LrsLastArg);

    lrs_receive("socket0", "buf1", LrsLastArg);

很明显，socket0是通信所用套接字，buf0是LR将要发送给服务器的内容（来自用户行为），buf1则接收服务器返回。

我们经常需要通过提取buf1中的内容，并适当的加一些断定语言，来判断这个通信操作有没有获取期望的数据。经常使用函数lrs_get_last_received_buffer，如下：

lrs_get_last_received_buffer("socket0", &ActualBuffer, &NumberOfBytes); 

这样就把lrs_receive接收到的数据保存在了内存中（指针ActualBuffer指向的一个有LR自动动态分配的内存块），并返回数据长度（保存在NumberOfBytes中）。

然后就可以像使用用户自定义的数组一样使用该缓冲区，如ActualBuffer[0]。需要注意的是，使用完成后需要手动释放ActualBuffer指向的内存，通过调用lrs_free_buffer。

下面是一个简单示例，在处理服务器返回数据时，我们将其作为char数组处理，经过验证这是没问题的。

完整代码：

vuser_init.c

#include "lrs.h"


vuser_init()
{
    lrs_startup(257);

    return 0;
}

Action.c

/*********************************************************************
 * Created by Mercury Interactive Windows Sockets Recorder
 *
 * Created on: Tue Nov 12 14:51:33
 *********************************************************************/

#include "lrs.h"


Action()
{
	int NumberOfBytes=0;
	char *ActualBuffer; 
	unsigned char num1,num2,num3,num4;
	unsigned int jifen =0;
	int dj =0 ;
	char * tmp;
	int i=0;
    lrs_create_socket("socket0", "TCP", "RemoteHost=10.15.107.12:12345",  LrsLastArg);

    lrs_send("socket0", "buf0", LrsLastArg);

    lrs_receive("socket0", "buf1", LrsLastArg);

	/* Get the last received buffer and its size. */ 
	
	lrs_get_last_received_buffer("socket0", &ActualBuffer, &NumberOfBytes); 
	//7B 9C 0B 00 00 0F 00 02 6C 00 00 00 08 00 90 5F 01 00 0D 00 00 00，从第偏移量17取数，直到14，作为积分；偏移量18为等级。
	lr_output_message("The last buffer's size is:%d/n", NumberOfBytes); 
	num1 = ActualBuffer[17];  //最高位
	num2 = ActualBuffer[16];
	num3 = ActualBuffer[15];
	num4 = ActualBuffer[14];  //最低位
	lr_output_message("num1 = %d,num2 = %d,num3 = %d,num4 = %u\n",num1,num2,num3,num4); 
	jifen = (num1 / 16 ) * 268435456 + (num1 % 16)* 16777216 + (num2 / 16)* 1048576 +(num2 % 16) * 65536 +(num3 / 16) * 4096 +  (num3 % 16 ) * 256 +(num4 / 16)* 16 + (num4 % 16) * 1; //注意00 90 5F 01是16进制格式，其中01（num1）是最高位。
	dj=(int)ActualBuffer[18];

	lrs_free_buffer(ActualBuffer); 
	lr_output_message("jifen = %d\n",jifen ); 
	lr_output_message("dj = %d\n",dj ); 

    lrs_close_socket("socket0");

    return 0;
}

 

vuser_end.c

#include "lrs.h"


vuser_end()
{
    lrs_cleanup();

    return 0;
}