LoadRunner_socket脚本

Action：

 

#include "lrs.h"

Action()
{
    lrs_create_socket("socket0","TCP","RemoteHost=127.0.0.1:1212",  LrsLastArg);

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

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

    lrs_close_socket("socket0");
}

 

data.ws：

 

;WSRData 2 1

send buf1 11
    "<ab><param1></ab>"

recv buf2 1024
    "111s"

-1

 

服务器端：：

#pragma once
#include <iostream>
#include <Winsock2.h>

using namespace std;

#pragma comment(lib, "ws2_32.lib")

int main()
{
 WORD wVersionRequested;
 WSADATA wsaData;
 int err;
    wVersionRequested=MAKEWORD(1,1);
 
    err=WSAStartup(wVersionRequested,&wsaData);
    if (err!=0)
        return -1;
 
    if (LOBYTE(wsaData.wVersion)!=1||HIBYTE(wsaData.wVersion)!=1)
 {
        WSACleanup();
        return -1;
    }
 
    //创建监听的套接字
    SOCKET sockSrv=socket(AF_INET,SOCK_STREAM,0);
    
    SOCKADDR_IN addrSrv;
    addrSrv.sin_addr.S_un.S_addr=htonl(INADDR_ANY);
    addrSrv.sin_family=AF_INET;
    addrSrv.sin_port=htons(6000);
    
    //绑定套接字
    bind(sockSrv,(SOCKADDR*)&addrSrv,sizeof(SOCKADDR));
    //将套接字设置为监听模式，准备接受用户请求
    listen(sockSrv,5);
    
    SOCKADDR_IN addrClient;
    int len=sizeof(SOCKADDR);
    
    cout <<"欢迎您，服务端已经建立。\n";

 SOCKET sockConn=accept(sockSrv,(SOCKADDR*)&addrClient,&len);

。。。。。。接下来就是recv() 和 send()函数的应用了。 


客户端：：


int main()   
{   
    //加载套接字   
    WORD wVersionRequested;   
    WSADATA wsaData;
    int err;   
       
    wVersionRequested=MAKEWORD(1,1);   
       
    err=WSAStartup(wVersionRequested,&wsaData);   
    if (err!=0)   
    {   
        return -1;   
    }   
       
    if (LOBYTE(wsaData.wVersion)!=1||   
        HIBYTE(wsaData.wVersion)!=1)   
    {   
        WSACleanup();   
        return -1;   
    }
 //创建套接字   
    SOCKET sockClient=socket(AF_INET,SOCK_STREAM,0);   
    SOCKADDR_IN addrSrv;
  cout <<"是否默认本地服务器？1/0      ?";
  cin >>todo;
  if (todo)
   addrSrv.sin_addr.S_un.S_addr=inet_addr("127.0.0.1");
  else{
   cout <<"请输入服务器IPv4地址：\n";
   char addr[16];
   gets(addr);
   addrSrv.sin_addr.S_un.S_addr=inet_addr(addr);
  } 
     addrSrv.sin_family=AF_INET;   
     addrSrv.sin_port=htons(6000);   
   
     //向服务器发送请求   
     if (connect(sockClient,(SOCKADDR*)&addrSrv,sizeof(SOCKADDR)))
         cout <<"连接失败！\n";
  else
   cout <<"连接成功！\n"; 




1.你不需要就不看得了
2.头文件只需要一个<WinSock2.h>已经给出
3.你不会的都去掉，就变成了C语言。

写了一个winsocket的脚本，没有通过录制的方式，是直接手写的。下面贴出来和大家分享：

脚本的写法很简单，大体说来，就像把大象放进冰箱一样，总共分三步：

第一步：把冰箱门打开。

//建立到服务端的连接
    rc =     lrs_create_socket("socket0", "TCP", "LocalHost=0", "RemoteHost=128.64.64.23:8988", LrsLastArg);
if (rc==0)

       lr_output_message("Socket  was successfully created ");

else

       lr_output_message("An error occurred while creating the socket, Error Code: %d", rc);

第二步：把大象装进去。
 
   lrs_send("socket0", "buf0", LrsLastArg);   //往"socket0"发送"buf0"中的数据


   lrs_receive("socket0", "buf1", LrsLastArg);//将"socke0"中返回的数据存放到"buf1"中


第三步：把冰箱门带上。
   
    //关闭连接
    lrs_close_socket("socket0");

大家肯定已经看出来了，整个关键是在第二步，要把这么一头大象装到冰箱里可不是件容易的事情，我们要对传送的数据做一些处理才行。如果大家有进行过
winsocket协议的脚本录制就会知道，LR会把你发送的数据包内容写到data.ws这个文件中，那么我们在此也同样应该把数据写到data.ws
中去。假设我要发送的是“00100312303456”这一串字符，那么我就直接把它写到data.ws中，脚本如下：


;WSRData 2 1

send buf0 10

"00100312303456"  //注意要加""

recv buf1 128


-1

运行脚本，可以看到执行成功。在日志信息中可以打印出发送的BUFFER和接收到的BUFFER内容。

接下来，我们要对发送的字符串进行参数化，让脚本每次发送的字符串都不一样，怎么做呢？
方法有两种：
1、直接参数化。在data.ws中是可以直接进行参数化的，具体操作我就不多说了。要注意的是默认的参数名称符号是尖括号（<>），和
HTTP协议的大括号（{}）不同（搞不懂为什么LR要把不同协议的默认参数名称符号设成不一样，这不是折腾人嘛）。脚本如下：

send buf0 106
"<string>"  //string是自定义的参数名

这种方法最简单，也容易理解。

2、第二种方法则麻烦一点，下面重点做个介绍。（不是我故意要找麻烦，而是因为我每次遇到的问题总是要比一般的情况麻烦一点，5555~~）

事情是这样的：话说如果我要发送的数据是很通过简单的方法拼接起来，举个简单的例子，比如是“用户名（假设是123）+密码（假设是456）”，在这种情
况下，使用第一种方法就够用了，我可以设置两个参数<username>和<password>，写成

send buf0 106
"<username><password>"

的方式就可以了。

但是这边的情况稍微复杂一些，发送的数据格式还需要加上字符串的长度，比如在上面的例子中，需要这样表示：00100312303456。
前四位0010是表示后面发送的字符串总长度是10，后面的03则表示用户名有3位，再后面的123才是真正的用户名，再后面的03456也是同样的道
理。而每次发送的用户名长度不同，字符串也就不同，比如用户名如果是1234，那么我的报文就应该是这样“001104123403456”这样一来，我
就不能通过简单地拼接的方式来发送了，而需要再对它进行一些处理。

我的方法是在action中进行字符串的拼接和处理：
 char data[200];
 char length[20];

 len = strlen(lr_eval_string("{usermame}"));
 sprintf(length,"%d",len);
 strcat(data,length);
 strcat(data,"{username}");

然后，再通过lrs_save_param_ex函数把该数据保存到data_param参数中。

lrs_save_param_ex("socket0", "user", data, 0, strlen(data),NULL, "data_param");

最后，在data.ws文件中使用data_param参数发送数据包：

send buf0 "<data_param>"