TheArchive/headscale
1
0
Fork 0
mirror of https://github.com/juanfont/headscale.git synced 2025-04-03 23:05:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Work in progress for web auth flow

Browse Source
This commit is contained in:
Juan Font Alonso 2022-11-04 00:10:36 +01:00
parent b2dc480f22
commit 3a6ef6bece
1 changed files with 29 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
integration/auth_web_flow_test.go
View File

@ -9,7 +9,7 @@ import (
"strings" "strings"
"testing" "testing"
"github.com/rs/zerolog/log" "log"
) )
type AuthWebFlowScenario struct { type AuthWebFlowScenario struct {
@ -86,6 +86,7 @@ func (s *AuthWebFlowScenario) CreateHeadscaleEnv(namespaces map[string]int) erro
} }
for namespaceName, clientCount := range namespaces { for namespaceName, clientCount := range namespaces {
log.Printf("creating namespace %s with %d clients", namespaceName, clientCount)
err = s.CreateNamespace(namespaceName) err = s.CreateNamespace(namespaceName)
if err != nil { if err != nil {
return err return err
@ -108,28 +109,28 @@ func (s *AuthWebFlowScenario) CreateHeadscaleEnv(namespaces map[string]int) erro
func (s *AuthWebFlowScenario) runTailscaleUp( func (s *AuthWebFlowScenario) runTailscaleUp(
namespaceStr, loginServer string, namespaceStr, loginServer string,
) error { ) error {
log.Printf("running tailscale up's for namespace %s", namespaceStr)
if namespace, ok := s.namespaces[namespaceStr]; ok { if namespace, ok := s.namespaces[namespaceStr]; ok {
for _, client := range namespace.Clients { for _, client := range namespace.Clients {
// namespace.joinWaitGroup.Add(1) namespace.joinWaitGroup.Add(1)
// go func(c TailscaleClient) error { go func(c TailscaleClient) {
// defer namespace.joinWaitGroup.Done() defer namespace.joinWaitGroup.Done()
// TODO(juanfont): error handle this // TODO(juanfont): error handle this
loginURL, err := c.UpWithLoginURL(loginServer)
if err != nil {
log.Printf("failed to run tailscale up: %s", err)
}
loginURL, err := client.UpWithLoginURL(loginServer) err = s.runHeadscaleRegister(namespaceStr, loginURL)
if err != nil { if err != nil {
return err log.Printf("failed to register client: %s", err)
} }
err = s.runHeadscaleRegister(namespaceStr, loginURL) }(client)
if err != nil {
return err
}
// }(client)
} }
// namespace.joinWaitGroup.Wait() namespace.joinWaitGroup.Wait()
return nil return nil
} }
@ -138,7 +139,10 @@ func (s *AuthWebFlowScenario) runTailscaleUp(
} }
func (s *AuthWebFlowScenario) runHeadscaleRegister(namespaceStr string, loginURL *url.URL) error { func (s *AuthWebFlowScenario) runHeadscaleRegister(namespaceStr string, loginURL *url.URL) error {
log.Info().Msgf("loginURL: %s", loginURL) log.Printf("loginURL: %s", loginURL)
loginURL.Host = fmt.Sprintf("%s:8080", s.Headscale().GetIP())
loginURL.Scheme = "http"
insecureTransport := &http.Transport{ insecureTransport := &http.Transport{
TLSClientConfig: &tls.Config{InsecureSkipVerify: true}, TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
} }
@ -155,15 +159,20 @@ func (s *AuthWebFlowScenario) runHeadscaleRegister(namespaceStr string, loginURL
} }
// see api.go HTML template // see api.go HTML template
key := strings.Split(strings.Split(string(body), "</code>")[0], "--key ")[1] code := strings.Split(string(body), "</code>")[0]
key := strings.Split(code, "key ")[1]
if headscale, ok := s.controlServers["headscale"]; ok { if headscale, ok := s.controlServers["headscale"]; ok {
_, err = headscale.Execute([]string{ _, err = headscale.Execute([]string{
"headscale", "-n", namespaceStr, "register", "--key", key}) "headscale", "-n", namespaceStr, "nodes", "register", "--key", key})
if err != nil { if err != nil {
log.Printf("failed to register node: %s", err)
return err return err
} }
log.Printf("registered node %s", key)
return nil
} }
return fmt.Errorf("failed to register client: %w", errNoHeadscaleAvailable) return fmt.Errorf("failed to find headscale: %w", errNoHeadscaleAvailable)
} }