Add HMAC protection on /approval endpoint

Signed-off-by: Bob Callaway <bcallaway@google.com>
This commit is contained in:
Bob Callaway
2022-07-06 07:11:37 -04:00
parent 454122ca22
commit fcfbb1ecb0
19 changed files with 274 additions and 14 deletions

View File

@@ -45,6 +45,8 @@ const (
FieldCodeChallenge = "code_challenge"
// FieldCodeChallengeMethod holds the string denoting the code_challenge_method field in the database.
FieldCodeChallengeMethod = "code_challenge_method"
// FieldHmacKey holds the string denoting the hmac_key field in the database.
FieldHmacKey = "hmac_key"
// Table holds the table name of the authrequest in the database.
Table = "auth_requests"
)
@@ -71,6 +73,7 @@ var Columns = []string{
FieldExpiry,
FieldCodeChallenge,
FieldCodeChallengeMethod,
FieldHmacKey,
}
// ValidColumn reports if the column name is valid (part of the table columns).

View File

@@ -204,6 +204,13 @@ func CodeChallengeMethod(v string) predicate.AuthRequest {
})
}
// HmacKey applies equality check predicate on the "hmac_key" field. It's identical to HmacKeyEQ.
func HmacKey(v []byte) predicate.AuthRequest {
return predicate.AuthRequest(func(s *sql.Selector) {
s.Where(sql.EQ(s.C(FieldHmacKey), v))
})
}
// ClientIDEQ applies the EQ predicate on the "client_id" field.
func ClientIDEQ(v string) predicate.AuthRequest {
return predicate.AuthRequest(func(s *sql.Selector) {
@@ -1675,6 +1682,82 @@ func CodeChallengeMethodContainsFold(v string) predicate.AuthRequest {
})
}
// HmacKeyEQ applies the EQ predicate on the "hmac_key" field.
func HmacKeyEQ(v []byte) predicate.AuthRequest {
return predicate.AuthRequest(func(s *sql.Selector) {
s.Where(sql.EQ(s.C(FieldHmacKey), v))
})
}
// HmacKeyNEQ applies the NEQ predicate on the "hmac_key" field.
func HmacKeyNEQ(v []byte) predicate.AuthRequest {
return predicate.AuthRequest(func(s *sql.Selector) {
s.Where(sql.NEQ(s.C(FieldHmacKey), v))
})
}
// HmacKeyIn applies the In predicate on the "hmac_key" field.
func HmacKeyIn(vs ...[]byte) predicate.AuthRequest {
v := make([]interface{}, len(vs))
for i := range v {
v[i] = vs[i]
}
return predicate.AuthRequest(func(s *sql.Selector) {
// if not arguments were provided, append the FALSE constants,
// since we can't apply "IN ()". This will make this predicate falsy.
if len(v) == 0 {
s.Where(sql.False())
return
}
s.Where(sql.In(s.C(FieldHmacKey), v...))
})
}
// HmacKeyNotIn applies the NotIn predicate on the "hmac_key" field.
func HmacKeyNotIn(vs ...[]byte) predicate.AuthRequest {
v := make([]interface{}, len(vs))
for i := range v {
v[i] = vs[i]
}
return predicate.AuthRequest(func(s *sql.Selector) {
// if not arguments were provided, append the FALSE constants,
// since we can't apply "IN ()". This will make this predicate falsy.
if len(v) == 0 {
s.Where(sql.False())
return
}
s.Where(sql.NotIn(s.C(FieldHmacKey), v...))
})
}
// HmacKeyGT applies the GT predicate on the "hmac_key" field.
func HmacKeyGT(v []byte) predicate.AuthRequest {
return predicate.AuthRequest(func(s *sql.Selector) {
s.Where(sql.GT(s.C(FieldHmacKey), v))
})
}
// HmacKeyGTE applies the GTE predicate on the "hmac_key" field.
func HmacKeyGTE(v []byte) predicate.AuthRequest {
return predicate.AuthRequest(func(s *sql.Selector) {
s.Where(sql.GTE(s.C(FieldHmacKey), v))
})
}
// HmacKeyLT applies the LT predicate on the "hmac_key" field.
func HmacKeyLT(v []byte) predicate.AuthRequest {
return predicate.AuthRequest(func(s *sql.Selector) {
s.Where(sql.LT(s.C(FieldHmacKey), v))
})
}
// HmacKeyLTE applies the LTE predicate on the "hmac_key" field.
func HmacKeyLTE(v []byte) predicate.AuthRequest {
return predicate.AuthRequest(func(s *sql.Selector) {
s.Where(sql.LTE(s.C(FieldHmacKey), v))
})
}
// And groups predicates with the AND operator between them.
func And(predicates ...predicate.AuthRequest) predicate.AuthRequest {
return predicate.AuthRequest(func(s *sql.Selector) {