srdusr
aboutsummaryrefslogtreecommitdiffstats
path: root/internal/clientcert/clientcert.go
blob: 4bba835112a4613cd7c9a26eb1901b268951b4a7 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
// Package clientcert manages client (mutual-TLS) certificates: which
// certificate mitmux presents to an upstream server that requires one,
// selected by matching the request's hostname the same way scope rules
// do (see internal/scope) - substring match by default, or a regex - so
// the "which rule applies to this host" mental model stays identical
// throughout the tool.
package clientcert

import (
	"crypto/tls"
	"fmt"
	"regexp"
	"strings"
)

// Cert is one client certificate, scoped to hosts matching Pattern.
type Cert struct {
	ID      int64
	Enabled bool
	Name    string
	Pattern string
	IsRegex bool
	CertPEM []byte
	KeyPEM  []byte
}

func (c Cert) matches(host string) bool {
	if c.IsRegex {
		re, err := regexp.Compile(c.Pattern)
		if err != nil {
			return false
		}
		return re.MatchString(host)
	}
	return strings.Contains(strings.ToLower(host), strings.ToLower(c.Pattern))
}

// FindFor returns the first enabled cert whose pattern matches host, or
// nil if none applies - mitmux then just doesn't present a client
// certificate for that connection, same as if mutual TLS weren't
// configured at all. First-match-wins on ID order, same convention as
// match-and-replace rules' Position ordering, minus the extra field:
// client certs are keyed by host, not layered edits, so insertion order
// is a reasonable enough tiebreaker without adding one.
func FindFor(certs []Cert, host string) *Cert {
	for i := range certs {
		if certs[i].Enabled && certs[i].matches(host) {
			return &certs[i]
		}
	}
	return nil
}

// TLSCertificate parses c's PEM-encoded cert/key pair into the form
// crypto/tls needs to present it during a handshake.
func (c Cert) TLSCertificate() (tls.Certificate, error) {
	cert, err := tls.X509KeyPair(c.CertPEM, c.KeyPEM)
	if err != nil {
		return tls.Certificate{}, fmt.Errorf("parse client certificate %q: %w", c.Name, err)
	}
	return cert, nil
}