Yuli Khodorkovskiy
7c9e840726
The pgprotocol overloads 'p' messages with PasswordMessage, SASLInitialResponse, SASLResponse, and GSSResponse. This patch allows contextual identification of the message by setting the authType in the frontend and then setting this value in the backend when a AuthenticationResponseMessage is received.
66 lines
1.6 KiB
Go
66 lines
1.6 KiB
Go
package pgproto3
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"errors"
|
|
"fmt"
|
|
)
|
|
|
|
// Message is the interface implemented by an object that can decode and encode
|
|
// a particular PostgreSQL message.
|
|
type Message interface {
|
|
// Decode is allowed and expected to retain a reference to data after
|
|
// returning (unlike encoding.BinaryUnmarshaler).
|
|
Decode(data []byte) error
|
|
|
|
// Encode appends itself to dst and returns the new buffer.
|
|
Encode(dst []byte) []byte
|
|
}
|
|
|
|
type FrontendMessage interface {
|
|
Message
|
|
Frontend() // no-op method to distinguish frontend from backend methods
|
|
}
|
|
|
|
type BackendMessage interface {
|
|
Message
|
|
Backend() // no-op method to distinguish frontend from backend methods
|
|
}
|
|
|
|
type AuthenticationResponseMessage interface {
|
|
BackendMessage
|
|
AuthenticationResponse() // no-op method to distinguish authentication responses
|
|
}
|
|
|
|
type invalidMessageLenErr struct {
|
|
messageType string
|
|
expectedLen int
|
|
actualLen int
|
|
}
|
|
|
|
func (e *invalidMessageLenErr) Error() string {
|
|
return fmt.Sprintf("%s body must have length of %d, but it is %d", e.messageType, e.expectedLen, e.actualLen)
|
|
}
|
|
|
|
type invalidMessageFormatErr struct {
|
|
messageType string
|
|
}
|
|
|
|
func (e *invalidMessageFormatErr) Error() string {
|
|
return fmt.Sprintf("%s body is invalid", e.messageType)
|
|
}
|
|
|
|
// getValueFromJSON gets the value from a protocol message representation in JSON.
|
|
func getValueFromJSON(v map[string]string) ([]byte, error) {
|
|
if v == nil {
|
|
return nil, nil
|
|
}
|
|
if text, ok := v["text"]; ok {
|
|
return []byte(text), nil
|
|
}
|
|
if binary, ok := v["binary"]; ok {
|
|
return hex.DecodeString(binary)
|
|
}
|
|
return nil, errors.New("unknown protocol representation")
|
|
}
|