Commit 2821a4b2 authored by Yannik Dällenbach's avatar Yannik Dällenbach 🤼 Committed by Yannik Dällenbach
Browse files

Generate and store SSH key

parent 40463955

go.mod

+3 −2
Original line number Diff line number Diff line
@@ -6,6 +6,7 @@ require ( 
	github.com/cloudscale-ch/cloudscale-go-sdk/v6 v6.0.0

	github.com/hashicorp/go-hclog v1.6.3

	gitlab.com/gitlab-org/fleeting/fleeting v0.0.0-20250425145049-7f673e7c5598

	golang.org/x/crypto v0.38.0

)



require (
@@ -20,8 +21,8 @@ require ( 
	github.com/oklog/run v1.1.0 // indirect

	golang.org/x/mod v0.17.0 // indirect

	golang.org/x/net v0.25.0 // indirect

	golang.org/x/sys v0.21.0 // indirect

	golang.org/x/text v0.16.0 // indirect

	golang.org/x/sys v0.33.0 // indirect

	golang.org/x/text v0.25.0 // indirect

	google.golang.org/genproto/googleapis/rpc v0.0.0-20240528184218-531527333157 // indirect

	google.golang.org/grpc v1.64.0 // indirect

	google.golang.org/protobuf v1.34.1 // indirect

go.sum

+8 −6
Original line number Diff line number Diff line
@@ -77,8 +77,8 @@ github.com/tidwall/transform v0.0.0-20201103190739-32f242e2dbde h1:AMNpJRc7P+GTw 
github.com/tidwall/transform v0.0.0-20201103190739-32f242e2dbde/go.mod h1:MvrEmduDUz4ST5pGZ7CABCnOU5f3ZiOAZzT6b1A6nX8=

gitlab.com/gitlab-org/fleeting/fleeting v0.0.0-20250425145049-7f673e7c5598 h1:9c2gbWnmHhYu7cK5M3MRnn9Loi/KpqxrW+8L1JJLpy4=

gitlab.com/gitlab-org/fleeting/fleeting v0.0.0-20250425145049-7f673e7c5598/go.mod h1:OsXzbzavwzLlPVwNEhPHtMnd4qCMABf2jDH3JGVQifA=

golang.org/x/crypto v0.24.0 h1:mnl8DM0o513X8fdIkmyFE/5hTYxbwYOjDS/+rK6qpRI=

golang.org/x/crypto v0.24.0/go.mod h1:Z1PMYSOR5nyMcyAVAIQSKCDwalqy85Aqn1x3Ws4L5DM=

golang.org/x/crypto v0.38.0 h1:jt+WWG8IZlBnVbomuhg2Mdq0+BBQaHbtqHEFEigjUV8=

golang.org/x/crypto v0.38.0/go.mod h1:MvrbAqul58NNYPKnOra203SB9vpuZW0e+RRZV+Ggqjw=

golang.org/x/mod v0.17.0 h1:zY54UmvipHiNd+pm+m0x9KhZ9hl1/7QNMyxXbc6ICqA=

golang.org/x/mod v0.17.0/go.mod h1:hTbmBsO62+eylJbnUtE2MGJUyE7QWk4xUqPFrRgJ+7c=

golang.org/x/net v0.25.0 h1:d/OCCoBEUq33pjydKrGQhw7IlUPI2Oylr+8qLx49kac=
@@ -90,10 +90,12 @@ golang.org/x/sys v0.0.0-20210927094055-39ccf1dd6fa6/go.mod h1:oPkhp1MJrh7nUepCBc 
golang.org/x/sys v0.0.0-20220503163025-988cb79eb6c6/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=

golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=

golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=

golang.org/x/sys v0.21.0 h1:rF+pYz3DAGSQAxAu1CbC7catZg4ebC4UIeIhKxBZvws=

golang.org/x/sys v0.21.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=

golang.org/x/text v0.16.0 h1:a94ExnEXNtEwYLGJSIUxnWoxoRz/ZcCsV63ROupILh4=

golang.org/x/text v0.16.0/go.mod h1:GhwF1Be+LQoKShO3cGOHzqOgRrGaYc9AvblQOmPVHnI=

golang.org/x/sys v0.33.0 h1:q3i8TbbEz+JRD9ywIRlyRAQbM0qF7hu24q3teo2hbuw=

golang.org/x/sys v0.33.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=

golang.org/x/term v0.32.0 h1:DR4lr0TjUs3epypdhTOkMmuF5CDFJ/8pOnbzMZPQ7bg=

golang.org/x/term v0.32.0/go.mod h1:uZG1FhGx848Sqfsq4/DlJr3xGGsYMu/L5GW4abiaEPQ=

golang.org/x/text v0.25.0 h1:qVyWApTSYLk/drJRO5mDlNYskwQznZmkpV2c8q9zls4=

golang.org/x/text v0.25.0/go.mod h1:WEdwpYrmk1qmdHvhkSTNPm3app7v4rsT8F2UD6+VHIA=

google.golang.org/genproto/googleapis/rpc v0.0.0-20240528184218-531527333157 h1:Zy9XzmMEflZ/MAaA7vNcoebnRAld7FsPW1EeBB7V0m8=

google.golang.org/genproto/googleapis/rpc v0.0.0-20240528184218-531527333157/go.mod h1:EfXuqaE1J41VCDicxHzUDm+8rk+7ZdXzHV0IhO/I6s0=

google.golang.org/grpc v1.64.0 h1:KH3VH9y/MgNQg1dE7b3XfVK0GsPSIzJwdF617gUSbvY=

main.go

+24 −1
Original line number Diff line number Diff line
@@ -2,6 +2,8 @@ package main 


import (

	"context"

	"crypto/ed25519"

	"encoding/pem"

	"fmt"

	"math"

	"net/http"
@@ -10,6 +12,7 @@ import ( 
	hclog "github.com/hashicorp/go-hclog"

	"gitlab.com/gitlab-org/fleeting/fleeting/plugin"

	"gitlab.com/gitlab-org/fleeting/fleeting/provider"

	"golang.org/x/crypto/ssh"

)



type instanceGroup struct {
@@ -17,6 +20,7 @@ type instanceGroup struct { 
	client   *cloudscale.Client



	settings      provider.Settings

	authorizedKey []byte

}



// Check if interface `InstanceGroup` is implemented.
@@ -38,6 +42,25 @@ func (g *instanceGroup) Init(ctx context.Context, logger hclog.Logger, settings 
		return info, fmt.Errorf("failed to create cloudscale API client: %w", err)

	}



	pub, priv, err := ed25519.GenerateKey(nil)

	if err != nil {

		return info, fmt.Errorf("failed to generate SSH private key: %w", err)

	}



	privPem, err := ssh.MarshalPrivateKey(priv, "")

	if err != nil {

		return info,

			fmt.Errorf("failed to marshal SSH private key: %w", err)

	}

	g.settings.Key = pem.EncodeToMemory(privPem)



	pubKey, err := ssh.NewPublicKey(pub)

	if err != nil {

		return info,

			fmt.Errorf("failed to convert SSH public key: %w", err)

	}

	g.authorizedKey = ssh.MarshalAuthorizedKey(pubKey)



	return info, nil

}