3pp-mirror/pomerium
1
0
Fork 0
mirror of https://github.com/pomerium/pomerium.git synced 2025-08-03 16:59:22 +02:00
Code Issues Projects Releases Packages Wiki Activity

write asciicast recordings to the databroker for now

Browse source 
For demo purposes, store the asciicast recordings as a BytesValue in the
databroker. Eventually we'll likely want to replace this with some
S3-compatible object store.
This commit is contained in:
Kenneth Jenkins 2025-03-27 16:35:37 -07:00
parent 19b67bf32d
commit 17fedbfa6f
1 changed files with 21 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

21
authorize/ssh_grpc.go
View file

@ -35,6 +35,7 @@ import (
"github.com/pomerium/pomerium/pkg/identity"
"github.com/pomerium/pomerium/pkg/identity/manager"
"github.com/pomerium/pomerium/pkg/identity/oauth"
"github.com/pomerium/pomerium/pkg/protoutil"
"github.com/pomerium/pomerium/pkg/storage"
"github.com/spf13/cobra"
gossh "golang.org/x/crypto/ssh"
@ -138,6 +139,26 @@ READ:
switch md.Format {
case extensions_session_recording.Format_AsciicastFormat:
log.Ctx(stream.Context()).Info().Int("compressed_size", len(recording)).Msg("asciicast recording received")
uncompressed, err := io.ReadAll(r)
if err != nil {
return fmt.Errorf("couldn't extract asciicast recording: %w", err)
}
// For demo purposes, store asciicast recordings in the databroker.
_, err = a.state.Load().dataBrokerClient.Put(context.Background(), &databroker.PutRequest{
Records: []*databroker.Record{
{
Type: "ssh-session-recording",
Id: md.RecordingName,
Data: protoutil.NewAnyBytes(uncompressed),
},
},
})
if err != nil {
return fmt.Errorf("couldn't save asciicast recording: %w", err)
}
case extensions_session_recording.Format_RawFormat:
reader := bufio.NewReader(r)
var header extensions_session_recording.Header