gtsocial-umbx

Unnamed repository; edit this file 'description' to name the repository.
Log | Files | Refs | README | LICENSE

authentication_sasl.go (2073B)

      1 package pgproto3
      2 
      3 import (
      4 	"bytes"
      5 	"encoding/binary"
      6 	"encoding/json"
      7 	"errors"
      8 
      9 	"github.com/jackc/pgx/v5/internal/pgio"
     10 )
     11 
     12 // AuthenticationSASL is a message sent from the backend indicating that SASL authentication is required.
     13 type AuthenticationSASL struct {
     14 	AuthMechanisms []string
     15 }
     16 
     17 // Backend identifies this message as sendable by the PostgreSQL backend.
     18 func (*AuthenticationSASL) Backend() {}
     19 
     20 // Backend identifies this message as an authentication response.
     21 func (*AuthenticationSASL) AuthenticationResponse() {}
     22 
     23 // Decode decodes src into dst. src must contain the complete message with the exception of the initial 1 byte message
     24 // type identifier and 4 byte message length.
     25 func (dst *AuthenticationSASL) Decode(src []byte) error {
     26 	if len(src) < 4 {
     27 		return errors.New("authentication message too short")
     28 	}
     29 
     30 	authType := binary.BigEndian.Uint32(src)
     31 
     32 	if authType != AuthTypeSASL {
     33 		return errors.New("bad auth type")
     34 	}
     35 
     36 	authMechanisms := src[4:]
     37 	for len(authMechanisms) > 1 {
     38 		idx := bytes.IndexByte(authMechanisms, 0)
     39 		if idx == -1 {
     40 			return &invalidMessageFormatErr{messageType: "AuthenticationSASL", details: "unterminated string"}
     41 		}
     42 		dst.AuthMechanisms = append(dst.AuthMechanisms, string(authMechanisms[:idx]))
     43 		authMechanisms = authMechanisms[idx+1:]
     44 	}
     45 
     46 	return nil
     47 }
     48 
     49 // Encode encodes src into dst. dst will include the 1 byte message type identifier and the 4 byte message length.
     50 func (src *AuthenticationSASL) Encode(dst []byte) []byte {
     51 	dst = append(dst, 'R')
     52 	sp := len(dst)
     53 	dst = pgio.AppendInt32(dst, -1)
     54 	dst = pgio.AppendUint32(dst, AuthTypeSASL)
     55 
     56 	for _, s := range src.AuthMechanisms {
     57 		dst = append(dst, []byte(s)...)
     58 		dst = append(dst, 0)
     59 	}
     60 	dst = append(dst, 0)
     61 
     62 	pgio.SetInt32(dst[sp:], int32(len(dst[sp:])))
     63 
     64 	return dst
     65 }
     66 
     67 // MarshalJSON implements encoding/json.Marshaler.
     68 func (src AuthenticationSASL) MarshalJSON() ([]byte, error) {
     69 	return json.Marshal(struct {
     70 		Type           string
     71 		AuthMechanisms []string
     72 	}{
     73 		Type:           "AuthenticationSASL",
     74 		AuthMechanisms: src.AuthMechanisms,
     75 	})
     76 }