mirror of https://github.com/tailscale/tailscale/
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
102 lines
2.4 KiB
Go
102 lines
2.4 KiB
Go
2 years ago
|
// Copyright (c) 2022 Tailscale Inc & AUTHORS All rights reserved.
|
||
|
// Use of this source code is governed by a BSD-style
|
||
|
// license that can be found in the LICENSE file.
|
||
|
|
||
|
package cli
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"errors"
|
||
|
"fmt"
|
||
|
"strconv"
|
||
|
"strings"
|
||
|
|
||
|
"github.com/peterbourgon/ff/v3/ffcli"
|
||
|
"tailscale.com/tka"
|
||
|
"tailscale.com/types/key"
|
||
|
)
|
||
|
|
||
|
var netlockCmd = &ffcli.Command{
|
||
|
Name: "lock",
|
||
|
ShortUsage: "lock <sub-command> <arguments>",
|
||
|
ShortHelp: "Manipulate the tailnet key authority",
|
||
|
Subcommands: []*ffcli.Command{nlInitCmd, nlStatusCmd},
|
||
|
Exec: runNetworkLockStatus,
|
||
|
}
|
||
|
|
||
|
var nlInitCmd = &ffcli.Command{
|
||
|
Name: "init",
|
||
|
ShortUsage: "init <public-key>...",
|
||
|
ShortHelp: "Initialize the tailnet key authority",
|
||
|
Exec: runNetworkLockInit,
|
||
|
}
|
||
|
|
||
|
func runNetworkLockInit(ctx context.Context, args []string) error {
|
||
|
st, err := localClient.NetworkLockStatus(ctx)
|
||
|
if err != nil {
|
||
|
return fixTailscaledConnectError(err)
|
||
|
}
|
||
|
if st.Enabled {
|
||
|
return errors.New("network-lock is already enabled")
|
||
|
}
|
||
|
|
||
|
// Parse the set of initially-trusted keys.
|
||
|
// Keys are specified using their key.NLPublic.MarshalText representation,
|
||
|
// with an optional '?<votes>' suffix.
|
||
|
var keys []tka.Key
|
||
|
for i, a := range args {
|
||
|
var key key.NLPublic
|
||
|
spl := strings.SplitN(a, "?", 2)
|
||
|
if err := key.UnmarshalText([]byte(spl[0])); err != nil {
|
||
|
return fmt.Errorf("parsing key %d: %v", i+1, err)
|
||
|
}
|
||
|
|
||
|
k := tka.Key{
|
||
|
Kind: tka.Key25519,
|
||
|
Public: key.Verifier(),
|
||
|
Votes: 1,
|
||
|
}
|
||
|
if len(spl) > 1 {
|
||
|
votes, err := strconv.Atoi(spl[1])
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("parsing key %d votes: %v", i+1, err)
|
||
|
}
|
||
|
k.Votes = uint(votes)
|
||
|
}
|
||
|
keys = append(keys, k)
|
||
|
}
|
||
|
|
||
|
status, err := localClient.NetworkLockInit(ctx, keys)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
fmt.Printf("Status: %+v\n\n", status)
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
var nlStatusCmd = &ffcli.Command{
|
||
|
Name: "status",
|
||
|
ShortUsage: "status",
|
||
|
ShortHelp: "Outputs the state of network lock",
|
||
|
Exec: runNetworkLockStatus,
|
||
|
}
|
||
|
|
||
|
func runNetworkLockStatus(ctx context.Context, args []string) error {
|
||
|
st, err := localClient.NetworkLockStatus(ctx)
|
||
|
if err != nil {
|
||
|
return fixTailscaledConnectError(err)
|
||
|
}
|
||
|
if st.Enabled {
|
||
|
fmt.Println("Network-lock is ENABLED.")
|
||
|
} else {
|
||
|
fmt.Println("Network-lock is NOT enabled.")
|
||
|
}
|
||
|
p, err := st.PublicKey.MarshalText()
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
fmt.Printf("our public-key: %s\n", p)
|
||
|
return nil
|
||
|
}
|