【测试平台系列】第一章 手撸压力机（4）- http证书认证的实现

// AdvancedOptions 高级选项
type AdvancedOptions struct {
  Tls Tls // 验证设置
}

// Tls tls认证结构体
type Tls struct {
  IsVerify   bool   // 是否开启验证，默认不开启，开启后需要上传密钥文件
  VerifyType int32  // 认证类型：0表示双向认证；1表示单向认证；默认为0
  CaCert     string // 密钥文件
}

// HttpClientSettings http客户端设置
type HttpClientSettings struct {
  //  客户端的名称，在header中的user-agent使用，通常我们默认就好
  Name string

  // 默认为flase，表示User-Agent使用fasthttp的默认值
  NoDefaultUserAgentHeader bool

  // 每台主机可以建立的最大连接数。如果没有设置，则使用DefaultMaxConnsPerHost。
  MaxConnsPerHost int

  // 空闲的保持连接在此持续时间之后关闭。默认情况下，在DefaultMaxIdleConnDuration之后关闭空闲连接。
  // 该连接如果空闲的话，在此时间后断开。
  MaxIdleConnDuration int64

  // Keep-alive连接在此持续时间后关闭。默认情况下，连接时间是不限制的。
  MaxConnDuration int

  // 默认情况下，响应读取超时时间是不限制的。
  ReadTimeout int64
  // 默认情况下，请求写超时时间不受限制。
  WriteTimeout int64

  // 请求头是否按标准格式传输
  DisableHeaderNamesNormalizing bool
  // url路径是按照原样输出，还是按照规范化输出。默认按照规范化输出
  DisablePathNormalizing bool
  AdvancedOptions        AdvancedOptions   // 高级选项
}

// 新建一个http客户端
func newHttpClient(httpClientSettings model.HttpClientSettings) (httpClient *fasthttp.Client) {
  // tls验证，关闭验证
  tr := &tls.Config{
    InsecureSkipVerify: true,
  }
  // 新建指针类型的客户端
  httpClient = &fasthttp.Client{}

  if httpClientSettings.Name != "" {
    httpClient.Name = httpClientSettings.Name
  }

  if httpClientSettings.NoDefaultUserAgentHeader == true {
    httpClient.NoDefaultUserAgentHeader = true
  }

  // 如果最大连接数不为0，将设置此数
  if httpClientSettings.MaxConnsPerHost != 0 {
    httpClient.MaxConnsPerHost = httpClientSettings.MaxConnsPerHost
  }

  // url不按照标准输出，按照原样输出
  if httpClientSettings.DisablePathNormalizing == true {
    httpClient.DisablePathNormalizing = true
  }
  // 请求头不按标准格式传输
  if httpClientSettings.DisableHeaderNamesNormalizing == true {
    httpClient.DisableHeaderNamesNormalizing = true
  }

  // 如果此时间不为0，那么将设置此时间。keep-alive维持此时长后将关闭。时间单位为毫秒
  if httpClientSettings.MaxConnDuration != 0 {
    httpClient.MaxConnDuration = time.Duration(httpClientSettings.MaxConnDuration) * time.Millisecond
  }

  if httpClientSettings.ReadTimeout != 0 {
    httpClient.ReadTimeout = time.Duration(httpClientSettings.ReadTimeout) * time.Millisecond
  }

  if httpClientSettings.WriteTimeout != 0 {
    httpClient.WriteTimeout = time.Duration(httpClientSettings.WriteTimeout) * time.Millisecond
  }

  // 该连接如果空闲的话，在此时间后断开。
  if httpClientSettings.MaxIdleConnDuration != 0 {
    httpClient.MaxIdleConnDuration = time.Duration(httpClientSettings.MaxIdleConnDuration) * time.Millisecond
  }
  
  // 
  httpsTls := httpClientSettings.AdvancedOptions.Tls
  
  // 如果开启认证
  if httpsTls.IsVerify {
    // switch条件选择语句，如果认证类型为0：则表示双向认证，如果是1：则表示为单向认证
    switch httpsTls.VerifyType {
    case 0: // 开启双向验证
      tr.InsecureSkipVerify = false
      // 如果密钥文件为空则跳出switch语句
      if httpsTls.CaCert == "" {
        break
      }
      // 生成一个cert对象池
      caCertPool := x509.NewCertPool()
      if caCertPool == nil {
        fmt.Println("生成CertPool失败！")
        break
      }
      
      // 读取认证文件，读出后为字节数组
      key, err := ioutil.ReadFile(httpsTls.CaCert)
      // 如果读取错误，则跳出switch语句
      if err != nil {
        fmt.Println("打开密钥文件失败：", err.Error())
        break
      }
      // 将认证文件添加到cert池中
      ok := caCertPool.AppendCertsFromPEM(key)
      // 如果添加失败则跳出switch语句
      if !ok {
        fmt.Println("密钥文件错误，生成失败！！！")
        break
      }
      // 将认证信息添加到客户端认证结构体
      tr.ClientCAs = caCertPool
    case 1: // 开启单向验证，客户端验证服务端密钥
      tr.InsecureSkipVerify = false
    }
  }

  // 客户端认证配置项
  httpClient.TLSConfig = tr
  return
}

package main

import (
  "kitchen-engine/client"
  "kitchen-engine/model"
)

func main() {

  // 一个类型中的字段，可以重置，也可以使用默认值，在go中，所有的类型的初始值，都是字段类型的0值，比如string的初始值是""空字符串，int类型的初始值是0等等
  httpClientSettings := model.HttpClientSettings{
    Name:                     "测试厨房",
    NoDefaultUserAgentHeader: true,
    MaxConnDuration:          1000,
    AdvancedOptions: model.AdvancedOptions{
      Tls: model.Tls{
        IsVerify:   false,
        VerifyType: 1,
      },
    },
  }

  headers := []model.Header{
    model.Header{
      Field: "name",
      Value: "你好",
    },
  }

  httpRequest := model.HttpRequest{
    Name:               "planet",
    Url:                "https://www.baidu.com",
    Method:             "GET",
    HttpClientSettings: httpClientSettings,
    Headers:            headers,
  }

  client.RequestHttp(httpRequest)
}