oidc.*fakeOidcProvider.SignIDToken - Code Metrics - Inspection of "Merge pull request #1306 from Permify/next" - Permify/permify - Measure and Improve Code Quality continuously with Scrutinizer

Passed

Push — master ( 04cfac...845541 )

by Tolga

created 2024-06-11 12:09 UTC

oidc.*fakeOidcProvider.SignIDToken A

↳ Parent: internal/authn/oidc/fakes_test.go

Complexity

Conditions

Size

Total Lines	16
Code Lines	11

Duplication

Lines	0
Ratio	0 %

Importance

Changes

Metric	Value
cc	4
eloc	11
nop	1
dl	0
loc	16
rs	9.85
c	0
b	0
f	0

package oidc

import (
	"crypto/ecdsa"
	"crypto/elliptic"
	"crypto/rand"
	"crypto/rsa"
	"encoding/json"
	"fmt"
	"net"
	"net/http"
	"net/http/httptest"
	"sync"

	"github.com/go-jose/go-jose/v3"
	"github.com/golang-jwt/jwt/v4"
)

type fakeOidcProvider struct {
	issuerURL    string
	authPath     string
	tokenPath    string
	userInfoPath string
	JWKSPath     string

	algorithms         []string
	keyIds             map[jwt.SigningMethod]string
	jwks               []jose.JSONWebKey
	rsaPrivateKey      *rsa.PrivateKey
	rsaPrivateKeyForPS *rsa.PrivateKey
	ecdsaPrivateKey    *ecdsa.PrivateKey
	hmacKey            []byte

	mu sync.RWMutex
}

type ProviderConfig struct {
	IssuerURL    string
	AuthPath     string
	TokenPath    string
	UserInfoPath string
	JWKSPath     string
	Algorithms   []string
}

func newFakeOidcProvider(config ProviderConfig) (*fakeOidcProvider, error) {
	rsaPrivateKey, err := rsa.GenerateKey(rand.Reader, 2048)
	if err != nil {
		return nil, fmt.Errorf("failed to generate RSA key: %w", err)

	}
	rsaPrivateKeyForPS, err := rsa.GenerateKey(rand.Reader, 2048)
	if err != nil {
		return nil, fmt.Errorf("failed to generate RSA key for PS: %w", err)

	}
	ecdsaPrivateKey, err := ecdsa.GenerateKey(elliptic.P256(), rand.Reader)
	if err != nil {
		return nil, fmt.Errorf("failed to generate ECDSA key: %w", err)

	}
	hmacKey := []byte("hmackeysecret")

	keyIds := map[jwt.SigningMethod]string{
		jwt.SigningMethodRS256: "rs256keyid",
	}
	jwks := []jose.JSONWebKey{
		{
			Key:       rsaPrivateKey.Public(),
			KeyID:     keyIds[jwt.SigningMethodRS256],
			Algorithm: "RS256",
			Use:       "sig",
		},
	}

	return &fakeOidcProvider{
		issuerURL:          config.IssuerURL,
		authPath:           config.AuthPath,
		tokenPath:          config.TokenPath,
		userInfoPath:       config.UserInfoPath,
		JWKSPath:           config.JWKSPath,
		algorithms:         config.Algorithms,
		rsaPrivateKey:      rsaPrivateKey,
		rsaPrivateKeyForPS: rsaPrivateKeyForPS,
		hmacKey:            hmacKey,
		jwks:               jwks,
		ecdsaPrivateKey:    ecdsaPrivateKey,
		keyIds:             keyIds,
		mu:                 sync.RWMutex{},
	}, nil
}

func (s *fakeOidcProvider) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	s.mu.RLock()
	defer s.mu.RUnlock()

	switch r.URL.Path {
	case "/.well-known/openid-configuration":
		s.responseWellKnown(w)
	case s.JWKSPath:
		s.responseJWKS(w)
	case s.authPath, s.tokenPath, s.userInfoPath:
		httpError(w, http.StatusNotFound)
	default:
		httpError(w, http.StatusNotFound)
	}
}

type providerJSON struct {
	Issuer      string   `json:"issuer"`
	AuthURL     string   `json:"authorization_endpoint"`
	TokenURL    string   `json:"token_endpoint"`
	JWKSURL     string   `json:"jwks_uri"`
	UserInfoURL string   `json:"userinfo_endpoint"`
	Algorithms  []string `json:"id_token_signing_alg_values_supported"`
}

func (s *fakeOidcProvider) responseWellKnown(w http.ResponseWriter) {
	jso := providerJSON{
		Issuer:      s.issuerURL,
		AuthURL:     s.issuerURL + s.authPath,
		TokenURL:    s.issuerURL + s.tokenPath,
		JWKSURL:     s.issuerURL + s.JWKSPath,
		UserInfoURL: s.issuerURL + s.userInfoPath,
		Algorithms:  s.algorithms,
	}
	httpJSON(w, jso)
}

func (s *fakeOidcProvider) responseJWKS(w http.ResponseWriter) {
	jwks := &jose.JSONWebKeySet{
		Keys: s.jwks,
	}
	httpJSON(w, jwks)
}

func httpJSON(w http.ResponseWriter, v interface{}) {
	w.Header().Set("Content-Type", "application/json")
	e := json.NewEncoder(w)
	e.SetIndent("", "  ")
	if err := e.Encode(v); err != nil {
		httpError(w, http.StatusInternalServerError)
	}
}

func httpError(w http.ResponseWriter, code int) {
	http.Error(w, http.StatusText(code), code)
}

func (s *fakeOidcProvider) UpdateKeyID(method jwt.SigningMethod, newKeyID string) {
	s.mu.Lock()
	defer s.mu.Unlock()

	s.keyIds[method] = newKeyID
	for i, key := range s.jwks {
		if key.Algorithm == method.Alg() {
			s.jwks[i].KeyID = newKeyID
		}
	}
}

func (s *fakeOidcProvider) SignIDToken(unsignedToken *jwt.Token) (string, error) {
	var signedToken string
	var err error

	switch unsignedToken.Method {
	case jwt.SigningMethodRS256:
		signedToken, err = unsignedToken.SignedString(s.rsaPrivateKey)
	default:
		return "", fmt.Errorf("incorrect signing method type, supported algorithms: HS256, RS256, ES256, PS256")
	}

	if err != nil {
		return "", err
	}

	return signedToken, nil
}

func createUnsignedToken(regClaims jwt.RegisteredClaims, method jwt.SigningMethod) *jwt.Token {
	claims := struct {
		jwt.RegisteredClaims
	}{
		RegisteredClaims: regClaims,
	}
	return jwt.NewWithClaims(method, claims)
}

func fakeHttpServer(url string, handler http.HandlerFunc) (*httptest.Server, error) {
	l, err := net.Listen("tcp", url)
	if err != nil {
		return nil, fmt.Errorf("failed to start listener: %w", err)

	}
	ts := httptest.NewUnstartedServer(handler)
	_ = ts.Listener.Close()
	ts.Listener = l
	ts.Start()
	return ts, nil
}


1			package oidc
2
3			import (
4			"crypto/ecdsa"
5			"crypto/elliptic"
6			"crypto/rand"
7			"crypto/rsa"
8			"encoding/json"
9			"fmt"
10			"net"
11			"net/http"
12			"net/http/httptest"
13			"sync"
14
15			"github.com/go-jose/go-jose/v3"
16			"github.com/golang-jwt/jwt/v4"
17			)
18
19			type fakeOidcProvider struct {
20			issuerURL string
21			authPath string
22			tokenPath string
23			userInfoPath string
24			JWKSPath string
25
26			algorithms []string
27			keyIds map[jwt.SigningMethod]string
28			jwks []jose.JSONWebKey
29			rsaPrivateKey *rsa.PrivateKey
30			rsaPrivateKeyForPS *rsa.PrivateKey
31			ecdsaPrivateKey *ecdsa.PrivateKey
32			hmacKey []byte
33
34			mu sync.RWMutex
35			}
36
37			type ProviderConfig struct {
38			IssuerURL string
39			AuthPath string
40			TokenPath string
41			UserInfoPath string
42			JWKSPath string
43			Algorithms []string
44			}
45
46			func newFakeOidcProvider(config ProviderConfig) (*fakeOidcProvider, error) {
47			rsaPrivateKey, err := rsa.GenerateKey(rand.Reader, 2048)
48			if err != nil {
49			return nil, fmt.Errorf("failed to generate RSA key: %w", err)
			0 ignored issues – show introduced 2024-06-11 12:04 UTC by Report Bug Copy Issue Report unrecognized printf verb 'w' Loading history...
50			}
51			rsaPrivateKeyForPS, err := rsa.GenerateKey(rand.Reader, 2048)
52			if err != nil {
53			return nil, fmt.Errorf("failed to generate RSA key for PS: %w", err)
			0 ignored issues – show introduced 2024-06-11 12:04 UTC by Report Bug Copy Issue Report unrecognized printf verb 'w' Loading history...
54			}
55			ecdsaPrivateKey, err := ecdsa.GenerateKey(elliptic.P256(), rand.Reader)
56			if err != nil {
57			return nil, fmt.Errorf("failed to generate ECDSA key: %w", err)
			0 ignored issues – show introduced 2024-06-11 12:04 UTC by Report Bug Copy Issue Report unrecognized printf verb 'w' Loading history...
58			}
59			hmacKey := []byte("hmackeysecret")
60
61			keyIds := map[jwt.SigningMethod]string{
62			jwt.SigningMethodRS256: "rs256keyid",
63			}
64			jwks := []jose.JSONWebKey{
65			{
66			Key: rsaPrivateKey.Public(),
67			KeyID: keyIds[jwt.SigningMethodRS256],
68			Algorithm: "RS256",
69			Use: "sig",
70			},
71			}
72
73			return &fakeOidcProvider{
74			issuerURL: config.IssuerURL,
75			authPath: config.AuthPath,
76			tokenPath: config.TokenPath,
77			userInfoPath: config.UserInfoPath,
78			JWKSPath: config.JWKSPath,
79			algorithms: config.Algorithms,
80			rsaPrivateKey: rsaPrivateKey,
81			rsaPrivateKeyForPS: rsaPrivateKeyForPS,
82			hmacKey: hmacKey,
83			jwks: jwks,
84			ecdsaPrivateKey: ecdsaPrivateKey,
85			keyIds: keyIds,
86			mu: sync.RWMutex{},
87			}, nil
88			}
89
90			func (s fakeOidcProvider) ServeHTTP(w http.ResponseWriter, r http.Request) {
91			s.mu.RLock()
92			defer s.mu.RUnlock()
93
94			switch r.URL.Path {
95			case "/.well-known/openid-configuration":
96			s.responseWellKnown(w)
97			case s.JWKSPath:
98			s.responseJWKS(w)
99			case s.authPath, s.tokenPath, s.userInfoPath:
100			httpError(w, http.StatusNotFound)
101			default:
102			httpError(w, http.StatusNotFound)
103			}
104			}
105
106			type providerJSON struct {
107			Issuer string `json:"issuer"`
108			AuthURL string `json:"authorization_endpoint"`
109			TokenURL string `json:"token_endpoint"`
110			JWKSURL string `json:"jwks_uri"`
111			UserInfoURL string `json:"userinfo_endpoint"`
112			Algorithms []string `json:"id_token_signing_alg_values_supported"`
113			}
114
115			func (s *fakeOidcProvider) responseWellKnown(w http.ResponseWriter) {
116			jso := providerJSON{
117			Issuer: s.issuerURL,
118			AuthURL: s.issuerURL + s.authPath,
119			TokenURL: s.issuerURL + s.tokenPath,
120			JWKSURL: s.issuerURL + s.JWKSPath,
121			UserInfoURL: s.issuerURL + s.userInfoPath,
122			Algorithms: s.algorithms,
123			}
124			httpJSON(w, jso)
125			}
126
127			func (s *fakeOidcProvider) responseJWKS(w http.ResponseWriter) {
128			jwks := &jose.JSONWebKeySet{
129			Keys: s.jwks,
130			}
131			httpJSON(w, jwks)
132			}
133
134			func httpJSON(w http.ResponseWriter, v interface{}) {
135			w.Header().Set("Content-Type", "application/json")
136			e := json.NewEncoder(w)
137			e.SetIndent("", " ")
138			if err := e.Encode(v); err != nil {
139			httpError(w, http.StatusInternalServerError)
140			}
141			}
142
143			func httpError(w http.ResponseWriter, code int) {
144			http.Error(w, http.StatusText(code), code)
145			}
146
147			func (s *fakeOidcProvider) UpdateKeyID(method jwt.SigningMethod, newKeyID string) {
148			s.mu.Lock()
149			defer s.mu.Unlock()
150
151			s.keyIds[method] = newKeyID
152			for i, key := range s.jwks {
153			if key.Algorithm == method.Alg() {
154			s.jwks[i].KeyID = newKeyID
155			}
156			}
157			}
158
159			func (s fakeOidcProvider) SignIDToken(unsignedToken jwt.Token) (string, error) {
160			var signedToken string
161			var err error
162
163			switch unsignedToken.Method {
164			case jwt.SigningMethodRS256:
165			signedToken, err = unsignedToken.SignedString(s.rsaPrivateKey)
166			default:
167			return "", fmt.Errorf("incorrect signing method type, supported algorithms: HS256, RS256, ES256, PS256")
168			}
169
170			if err != nil {
171			return "", err
172			}
173
174			return signedToken, nil
175			}
176
177			func createUnsignedToken(regClaims jwt.RegisteredClaims, method jwt.SigningMethod) *jwt.Token {
178			claims := struct {
179			jwt.RegisteredClaims
180			}{
181			RegisteredClaims: regClaims,
182			}
183			return jwt.NewWithClaims(method, claims)
184			}
185
186			func fakeHttpServer(url string, handler http.HandlerFunc) (*httptest.Server, error) {
187			l, err := net.Listen("tcp", url)
188			if err != nil {
189			return nil, fmt.Errorf("failed to start listener: %w", err)
			0 ignored issues – show introduced 2024-06-11 12:04 UTC by Report Bug Copy Issue Report unrecognized printf verb 'w' Loading history...
190			}
191			ts := httptest.NewUnstartedServer(handler)
192			_ = ts.Listener.Close()
193			ts.Listener = l
194			ts.Start()
195			return ts, nil
196			}
197

Permify / permify

Push — master ( 04cfac...845541 )

oidc.*fakeOidcProvider.SignIDToken A

Complexity

Size

Duplication

Importance

Duplication Side-by-Side

Filter issues like