51Testing软件测试论坛

标题: 如何使用base64验证码登录--Jmeter？ [打印本页]

作者: lsekfe 时间: 2022-9-8 13:53
标题: 如何使用base64验证码登录--Jmeter？
依赖环境/工具：java环境、jmeter、python3、pycharm(ide工具)。
　　此次主要操作步骤：
　　· 使用jmeter工具把验证码base64加密信息保存至本地。
　　· 编写python脚本(包含：解码bse64验证码图片、验证码图片背景降噪、对验证码图片文字识别、去除图片文字内空格、特殊字符并保存本地为txt文件)。
　　· jmeter内运行py文件，把打印内容变量给登录接口验证码字段。
　　1. 所有的操作、准备都基于2个接口【获取验证码、登录接口】执行。
[attach]142673[/attach]
[attach]142675[/attach]
jmeter BeanShell后置处理程序

FileWriter fstream = new FileWriter("E:/test001/code.txt");//定义本地文件路径
　　BufferedWriter out = new BufferedWriter(fstream);
　　out.write("${image}");// 验证码base64变量名
　　out.write(System.getProperty("line.separator"));//换行
　　out.close();
　　fstream.close();

jmeter BeanShell 取样器【执行py文件】

import java.io.BufferedReader;
　　import java.io.InputStreamReader;
　　String command = "cmd /c python E://D-drive-26093//pycharm//test01//venv//testA.py ";
　　Runtime rt = Runtime.getRuntime();
　　Process pr = rt.exec(command);
　　pr.waitFor();
　　BufferedReader b = new BufferedReader(new InputStreamReader(pr.getInputStream()));
　　String line = "";
　　StringBuilder response = new StringBuilder();
　　while ((line = b.readLine()) != null) {
　　 response.append(line);
　　}
　　String response_data = response.toString();
　　System.out.println(response_data);
　　b.close();
　　//System.out.println(response.toString());
　　vars.put("code",response.toString()); //把结果赋值给变量 TEST ,方便后面调用

　2.py文件

from PIL import Image
　　import re
　　import pytesseract
　　import os, base64
　　import numpy as np
　　import time
　　import matplotlib.pyplot as plt
　　import ddddocr
　　# 对base64图片解码
　　with open("E:\\test001\\code.txt", "r") as f:
　　 imgdata = base64.b64decode(f.read())
　　 filea = open('E:\\test001\\1.jpg', 'wb')
　　 filea.write(imgdata)
　　 filea.close()
　　# 验证码降噪
　　image = Image.open('E:\\test001\\1.jpg','r')
　　image = image.convert('L')
　　image = np.asarray(image)
　　image = (image > 135) * 255
　　image = Image.fromarray(image).convert('RGB')
　　image.save('E:\\test001\\12345.jpg')
　　time.sleep(2)
　　# 识别验证码图片文字
　　ocr = ddddocr.DdddOcr()
　　with open('E:\\test001\\12345.jpg', 'rb') as f:
　　 img_bytes = f.read()
　　res = ocr.classification(img_bytes)
　　file = open('E:\\test001\\001test.txt', mode='w')
　　file.writelines(res)
　　print (res)

欢迎光临 51Testing软件测试论坛 (http://bbs.51testing.com/)