mirror of https://github.com/tailscale/tailscale/
Merge branch 'main' into claude/analyze-test-coverage-01BnJsiXhLinMJyRe78R29e9
Signed-off-by: Ofer Erez <ofer43211@users.noreply.github.com>pull/17963/head
Ofer Erez
2 weeks ago
committed by
GitHub
commit
d6f61e54b8
@ -0,0 +1,17 @@
|
|||||||
|
PRs welcome! But please file bugs first and explain the problem or
|
||||||
|
motivation. For new or changed functionality, strike up a discussion
|
||||||
|
and get agreement on the design/solution before spending too much time writing
|
||||||
|
code.
|
||||||
|
|
||||||
|
Commit messages should [reference
|
||||||
|
bugs](https://docs.github.com/en/github/writing-on-github/autolinked-references-and-urls).
|
||||||
|
|
||||||
|
We require [Developer Certificate of
|
||||||
|
Origin](https://en.wikipedia.org/wiki/Developer_Certificate_of_Origin) (DCO)
|
||||||
|
`Signed-off-by` lines in commits. (`git commit -s`)
|
||||||
|
|
||||||
|
Please squash your code review edits & force push. Multiple commits in
|
||||||
|
a PR are fine, but only if they're each logically separate and all tests pass
|
||||||
|
at each stage. No fixup commits.
|
||||||
|
|
||||||
|
See [commit-messages.md](docs/commit-messages.md) (or skim `git log`) for our commit message style.
|
||||||
@ -0,0 +1,27 @@
|
|||||||
|
# Run some natlab integration tests.
|
||||||
|
# See https://github.com/tailscale/tailscale/issues/13038
|
||||||
|
name: "natlab-integrationtest"
|
||||||
|
|
||||||
|
concurrency:
|
||||||
|
group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
|
||||||
|
cancel-in-progress: true
|
||||||
|
|
||||||
|
on:
|
||||||
|
pull_request:
|
||||||
|
paths:
|
||||||
|
- "tstest/integration/nat/nat_test.go"
|
||||||
|
jobs:
|
||||||
|
natlab-integrationtest:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
- name: Check out code
|
||||||
|
uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
|
||||||
|
- name: Install qemu
|
||||||
|
run: |
|
||||||
|
sudo rm /var/lib/man-db/auto-update
|
||||||
|
sudo apt-get -y update
|
||||||
|
sudo apt-get -y remove man-db
|
||||||
|
sudo apt-get install -y qemu-system-x86 qemu-utils
|
||||||
|
- name: Run natlab integration tests
|
||||||
|
run: |
|
||||||
|
./tool/go test -v -run=^TestEasyEasy$ -timeout=3m -count=1 ./tstest/integration/nat --run-vm-tests
|
||||||
@ -0,0 +1,29 @@
|
|||||||
|
# Pin images used in github actions to a hash instead of a version tag.
|
||||||
|
name: pin-github-actions
|
||||||
|
on:
|
||||||
|
pull_request:
|
||||||
|
branches:
|
||||||
|
- main
|
||||||
|
paths:
|
||||||
|
- ".github/workflows/**"
|
||||||
|
|
||||||
|
workflow_dispatch:
|
||||||
|
|
||||||
|
permissions:
|
||||||
|
contents: read
|
||||||
|
pull-requests: read
|
||||||
|
|
||||||
|
concurrency:
|
||||||
|
group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
|
||||||
|
cancel-in-progress: true
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
run:
|
||||||
|
name: pin-github-actions
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
|
||||||
|
- name: pin
|
||||||
|
run: make pin-github-actions
|
||||||
|
- name: check for changed workflow files
|
||||||
|
run: git diff --no-ext-diff --exit-code .github/workflows || (echo "Some github actions versions need pinning, run make pin-github-actions."; exit 1)
|
||||||
@ -0,0 +1,30 @@
|
|||||||
|
name: request-dataplane-review
|
||||||
|
|
||||||
|
on:
|
||||||
|
pull_request:
|
||||||
|
paths:
|
||||||
|
- ".github/workflows/request-dataplane-review.yml"
|
||||||
|
- "**/*derp*"
|
||||||
|
- "**/derp*/**"
|
||||||
|
- "!**/depaware.txt"
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
request-dataplane-review:
|
||||||
|
name: Request Dataplane Review
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
- name: Check out code
|
||||||
|
uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
|
||||||
|
- name: Get access token
|
||||||
|
uses: actions/create-github-app-token@df432ceedc7162793a195dd1713ff69aefc7379e # v2.0.6
|
||||||
|
id: generate-token
|
||||||
|
with:
|
||||||
|
# Get token for app: https://github.com/apps/change-visibility-bot
|
||||||
|
app-id: ${{ secrets.VISIBILITY_BOT_APP_ID }}
|
||||||
|
private-key: ${{ secrets.VISIBILITY_BOT_APP_PRIVATE_KEY }}
|
||||||
|
- name: Add reviewers
|
||||||
|
env:
|
||||||
|
GH_TOKEN: ${{ steps.generate-token.outputs.token }}
|
||||||
|
url: ${{ github.event.pull_request.html_url }}
|
||||||
|
run: |
|
||||||
|
gh pr edit "$url" --add-reviewer tailscale/dataplane
|
||||||
@ -0,0 +1,38 @@
|
|||||||
|
name: tailscale.com/cmd/vet
|
||||||
|
|
||||||
|
env:
|
||||||
|
HOME: ${{ github.workspace }}
|
||||||
|
# GOMODCACHE is the same definition on all OSes. Within the workspace, we use
|
||||||
|
# toplevel directories "src" (for the checked out source code), and "gomodcache"
|
||||||
|
# and other caches as siblings to follow.
|
||||||
|
GOMODCACHE: ${{ github.workspace }}/gomodcache
|
||||||
|
|
||||||
|
on:
|
||||||
|
push:
|
||||||
|
branches:
|
||||||
|
- main
|
||||||
|
- "release-branch/*"
|
||||||
|
paths:
|
||||||
|
- "**.go"
|
||||||
|
pull_request:
|
||||||
|
paths:
|
||||||
|
- "**.go"
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
vet:
|
||||||
|
runs-on: [ self-hosted, linux ]
|
||||||
|
timeout-minutes: 5
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- name: Check out code
|
||||||
|
uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
|
||||||
|
with:
|
||||||
|
path: src
|
||||||
|
|
||||||
|
- name: Build 'go vet' tool
|
||||||
|
working-directory: src
|
||||||
|
run: ./tool/go build -o /tmp/vettool tailscale.com/cmd/vet
|
||||||
|
|
||||||
|
- name: Run 'go vet'
|
||||||
|
working-directory: src
|
||||||
|
run: ./tool/go vet -vettool=/tmp/vettool tailscale.com/...
|
||||||
@ -1 +1 @@
|
|||||||
3.18
|
3.22
|
||||||
@ -1,135 +1,103 @@
|
|||||||
# Contributor Covenant Code of Conduct
|
# Tailscale Community Code of Conduct
|
||||||
|
|
||||||
## Our Pledge
|
## Our Pledge
|
||||||
|
|
||||||
We as members, contributors, and leaders pledge to make participation
|
We are committed to creating an open, welcoming, diverse, inclusive, healthy and respectful community.
|
||||||
in our community a harassment-free experience for everyone, regardless
|
Unacceptable, harmful and inappropriate behavior will not be tolerated.
|
||||||
of age, body size, visible or invisible disability, ethnicity, sex
|
|
||||||
characteristics, gender identity and expression, level of experience,
|
|
||||||
education, socio-economic status, nationality, personal appearance,
|
|
||||||
race, religion, or sexual identity and orientation.
|
|
||||||
|
|
||||||
We pledge to act and interact in ways that contribute to an open,
|
|
||||||
welcoming, diverse, inclusive, and healthy community.
|
|
||||||
|
|
||||||
## Our Standards
|
## Our Standards
|
||||||
|
|
||||||
Examples of behavior that contributes to a positive environment for
|
Examples of behavior that contributes to a positive environment for our community include:
|
||||||
our community include:
|
|
||||||
|
|
||||||
* Demonstrating empathy and kindness toward other people
|
|
||||||
* Being respectful of differing opinions, viewpoints, and experiences
|
|
||||||
* Giving and gracefully accepting constructive feedback
|
|
||||||
* Accepting responsibility and apologizing to those affected by our
|
|
||||||
mistakes, and learning from the experience
|
|
||||||
* Focusing on what is best not just for us as individuals, but for the
|
|
||||||
overall community
|
|
||||||
|
|
||||||
Examples of unacceptable behavior include:
|
|
||||||
|
|
||||||
* The use of sexualized language or imagery, and sexual attention or
|
- Demonstrating empathy and kindness toward other people.
|
||||||
advances of any kind
|
- Being respectful of differing opinions, viewpoints, and experiences.
|
||||||
* Trolling, insulting or derogatory comments, and personal or
|
- Giving and gracefully accepting constructive feedback.
|
||||||
political attacks
|
- Accepting responsibility and apologizing to those affected by our mistakes, and learning from the experience.
|
||||||
* Public or private harassment
|
- Focusing on what is best not just for us as individuals, but for the overall community.
|
||||||
* Publishing others' private information, such as a physical or email
|
|
||||||
address, without their explicit permission
|
|
||||||
* Other conduct which could reasonably be considered inappropriate in
|
|
||||||
a professional setting
|
|
||||||
|
|
||||||
## Enforcement Responsibilities
|
Examples of unacceptable behavior include without limitation:
|
||||||
|
|
||||||
Community leaders are responsible for clarifying and enforcing our
|
- The use of language, imagery or emojis (collectively "content") that is racist, sexist, homophobic, transphobic, or otherwise harassing or discriminatory based on any protected characteristic.
|
||||||
standards of acceptable behavior and will take appropriate and fair
|
- The use of sexualized content and sexual attention or advances of any kind.
|
||||||
corrective action in response to any behavior that they deem
|
- The use of violent, intimidating or bullying content.
|
||||||
inappropriate, threatening, offensive, or harmful.
|
- Trolling, concern trolling, insulting or derogatory comments, and personal or political attacks.
|
||||||
|
- Public or private harassment.
|
||||||
|
- Publishing others' personal information, such as a photo, physical address, email address, online profile information, or other personal information, without their explicit permission or with the intent to bully or harass the other person.
|
||||||
|
- Posting deep fake or other AI generated content about or involving another person without the explicit permission.
|
||||||
|
- Spamming community channels and members, such as sending repeat messages, low-effort content, or automated messages.
|
||||||
|
- Phishing or any similar activity.
|
||||||
|
- Distributing or promoting malware.
|
||||||
|
- The use of any coded or suggestive content to hide or provoke otherwise unacceptable behavior.
|
||||||
|
- Other conduct which could reasonably be considered harmful, illegal, or inappropriate in a professional setting.
|
||||||
|
|
||||||
Community leaders have the right and responsibility to remove, edit,
|
Please also see the Tailscale Acceptable Use Policy, available at [tailscale.com/tailscale-aup](https://tailscale.com/tailscale-aup).
|
||||||
or reject comments, commits, code, wiki edits, issues, and other
|
|
||||||
contributions that are not aligned to this Code of Conduct, and will
|
|
||||||
communicate reasons for moderation decisions when appropriate.
|
|
||||||
|
|
||||||
## Scope
|
## Reporting Incidents
|
||||||
|
|
||||||
This Code of Conduct applies within all community spaces, and also
|
Instances of abusive, harassing, or otherwise unacceptable behavior may be reported to Tailscale directly via <info@tailscale.com>, or to the community leaders or moderators via DM or similar.
|
||||||
applies when an individual is officially representing the community in
|
All complaints will be reviewed and investigated promptly and fairly.
|
||||||
public spaces. Examples of representing our community include using an
|
We will respect the privacy and safety of the reporter of any issues.
|
||||||
official e-mail address, posting via an official social media account,
|
|
||||||
or acting as an appointed representative at an online or offline
|
|
||||||
event.
|
|
||||||
|
|
||||||
## Enforcement
|
Please note that this community is not moderated by staff 24/7, and we do not have, and do not undertake, any obligation to prescreen, monitor, edit, or remove any content or data, or to actively seek facts or circumstances indicating illegal activity.
|
||||||
|
While we strive to keep the community safe and welcoming, moderation may not be immediate at all hours.
|
||||||
|
If you encounter any issues, report them using the appropriate channels.
|
||||||
|
|
||||||
Instances of abusive, harassing, or otherwise unacceptable behavior
|
## Enforcement Guidelines
|
||||||
may be reported to the community leaders responsible for enforcement
|
|
||||||
at [info@tailscale.com](mailto:info@tailscale.com). All complaints
|
|
||||||
will be reviewed and investigated promptly and fairly.
|
|
||||||
|
|
||||||
All community leaders are obligated to respect the privacy and
|
Community leaders and moderators are responsible for clarifying and enforcing our standards of acceptable behavior and will take appropriate and fair corrective action in response to any behavior that they deem inappropriate, threatening, offensive, or harmful.
|
||||||
security of the reporter of any incident.
|
|
||||||
|
|
||||||
## Enforcement Guidelines
|
Community leaders and moderators have the right and responsibility to remove, edit, or reject comments, commits, code, wiki edits, issues, and other contributions that are not aligned to this Community Code of Conduct.
|
||||||
|
Tailscale retains full discretion to take action (or not) in response to a violation of these guidelines with or without notice or liability to you.
|
||||||
|
We will interpret our policies and resolve disputes in favor of protecting users, customers, the public, our community and our company, as a whole.
|
||||||
|
|
||||||
Community leaders will follow these Community Impact Guidelines in
|
Community leaders will follow these community enforcement guidelines in determining the consequences for any action they deem in violation of this Code of Conduct,
|
||||||
determining the consequences for any action they deem in violation of
|
and retain full discretion to apply the enforcement guidelines as necessary depending on the circumstances:
|
||||||
this Code of Conduct:
|
|
||||||
|
|
||||||
### 1. Correction
|
### 1. Correction
|
||||||
|
|
||||||
**Community Impact**: Use of inappropriate language or other behavior
|
Community Impact: Use of inappropriate language or other behavior deemed unprofessional or unwelcome in the community.
|
||||||
deemed unprofessional or unwelcome in the community.
|
|
||||||
|
|
||||||
**Consequence**: A private, written warning from community leaders,
|
Consequence: A private, written warning from community leaders, providing clarity around the nature of the violation and an explanation of why the behavior was inappropriate.
|
||||||
providing clarity around the nature of the violation and an
|
A public apology may be requested.
|
||||||
explanation of why the behavior was inappropriate. A public apology
|
|
||||||
may be requested.
|
|
||||||
|
|
||||||
### 2. Warning
|
### 2. Warning
|
||||||
|
|
||||||
**Community Impact**: A violation through a single incident or series
|
Community Impact: A violation through a single incident or series of actions.
|
||||||
of actions.
|
|
||||||
|
|
||||||
**Consequence**: A warning with consequences for continued
|
Consequence: A warning with consequences for continued behavior.
|
||||||
behavior. No interaction with the people involved, including
|
No interaction with the people involved, including unsolicited interaction with those enforcing this Community Code of Conduct, for a specified period of time.
|
||||||
unsolicited interaction with those enforcing the Code of Conduct, for
|
This includes avoiding interactions in community spaces as well as external channels like social media.
|
||||||
a specified period of time. This includes avoiding interactions in
|
Violating these terms may lead to a temporary or permanent ban.
|
||||||
community spaces as well as external channels like social
|
|
||||||
media. Violating these terms may lead to a temporary or permanent ban.
|
|
||||||
|
|
||||||
### 3. Temporary Ban
|
### 3. Temporary Ban
|
||||||
|
|
||||||
**Community Impact**: A serious violation of community standards,
|
Community Impact: A serious violation of community standards, including sustained inappropriate behavior.
|
||||||
including sustained inappropriate behavior.
|
|
||||||
|
|
||||||
**Consequence**: A temporary ban from any sort of interaction or
|
Consequence: A temporary ban from any sort of interaction or public communication with the community for a specified period of time.
|
||||||
public communication with the community for a specified period of
|
No public or private interaction with the people involved, including unsolicited interaction with those enforcing the Code of Conduct, is allowed during this period. Violating these terms may lead to a permanent ban.
|
||||||
time. No public or private interaction with the people involved,
|
|
||||||
including unsolicited interaction with those enforcing the Code of
|
|
||||||
Conduct, is allowed during this period. Violating these terms may lead
|
|
||||||
to a permanent ban.
|
|
||||||
|
|
||||||
### 4. Permanent Ban
|
### 4. Permanent Ban
|
||||||
|
|
||||||
**Community Impact**: Demonstrating a pattern of violation of
|
Community Impact: Demonstrating a pattern of violation of community standards, including sustained inappropriate behavior, harassment of an individual, or aggression toward or disparagement of classes of individuals.
|
||||||
community standards, including sustained inappropriate behavior,
|
|
||||||
harassment of an individual, or aggression toward or disparagement of
|
|
||||||
classes of individuals.
|
|
||||||
|
|
||||||
**Consequence**: A permanent ban from any sort of public interaction
|
Consequence: A permanent ban from any sort of public interaction within the community.
|
||||||
within the community.
|
|
||||||
|
## Acceptable Use Policy
|
||||||
|
|
||||||
|
Violation of this Community Code of Conduct may also violate the Tailscale Acceptable Use Policy, which may result in suspension or termination of your Tailscale account.
|
||||||
|
For more information, please see the Tailscale Acceptable Use Policy, available at [tailscale.com/tailscale-aup](https://tailscale.com/tailscale-aup).
|
||||||
|
|
||||||
|
## Privacy
|
||||||
|
|
||||||
|
Please see the Tailscale [Privacy Policy](https://tailscale.com/privacy-policy) for more information about how Tailscale collects, uses, discloses and protects information.
|
||||||
|
|
||||||
## Attribution
|
## Attribution
|
||||||
|
|
||||||
This Code of Conduct is adapted from the [Contributor
|
This Code of Conduct is adapted from the [Contributor Covenant][homepage], version 2.0, available at <https://www.contributor-covenant.org/version/2/0/code_of_conduct.html>.
|
||||||
Covenant][homepage], version 2.0, available at
|
|
||||||
https://www.contributor-covenant.org/version/2/0/code_of_conduct.html.
|
|
||||||
|
|
||||||
Community Impact Guidelines were inspired by [Mozilla's code of
|
Community Impact Guidelines were inspired by [Mozilla's code of conduct enforcement ladder](https://github.com/mozilla/diversity).
|
||||||
conduct enforcement ladder](https://github.com/mozilla/diversity).
|
|
||||||
|
|
||||||
[homepage]: https://www.contributor-covenant.org
|
[homepage]: https://www.contributor-covenant.org
|
||||||
|
|
||||||
For answers to common questions about this code of conduct, see the
|
For answers to common questions about this code of conduct, see the FAQ at <https://www.contributor-covenant.org/faq>.
|
||||||
FAQ at https://www.contributor-covenant.org/faq. Translations are
|
Translations are available at <https://www.contributor-covenant.org/translations>.
|
||||||
available at https://www.contributor-covenant.org/translations.
|
|
||||||
|
|
||||||
|
|||||||
@ -1,5 +1,12 @@
|
|||||||
# Copyright (c) Tailscale Inc & AUTHORS
|
# Copyright (c) Tailscale Inc & AUTHORS
|
||||||
# SPDX-License-Identifier: BSD-3-Clause
|
# SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
FROM alpine:3.18
|
FROM alpine:3.22
|
||||||
RUN apk add --no-cache ca-certificates iptables iproute2 ip6tables iputils
|
RUN apk add --no-cache ca-certificates iptables iptables-legacy iproute2 ip6tables iputils
|
||||||
|
# Alpine 3.19 replaced legacy iptables with nftables based implementation. We
|
||||||
|
# can't be certain that all hosts that run Tailscale containers currently
|
||||||
|
# suppport nftables, so link back to legacy for backwards compatibility reasons.
|
||||||
|
# TODO(irbekrm): add some way how to determine if we still run on nodes that
|
||||||
|
# don't support nftables, so that we can eventually remove these symlinks.
|
||||||
|
RUN ln -s /sbin/iptables-legacy /sbin/iptables
|
||||||
|
RUN ln -s /sbin/ip6tables-legacy /sbin/ip6tables
|
||||||
|
|||||||
@ -1 +1 @@
|
|||||||
1.75.0
|
1.91.0
|
||||||
|
|||||||
@ -0,0 +1,61 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
package appc
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"net/netip"
|
||||||
|
|
||||||
|
"go4.org/netipx"
|
||||||
|
)
|
||||||
|
|
||||||
|
// errPoolExhausted is returned when there are no more addresses to iterate over.
|
||||||
|
var errPoolExhausted = errors.New("ip pool exhausted")
|
||||||
|
|
||||||
|
// ippool allows for iteration over all the addresses within a netipx.IPSet.
|
||||||
|
// netipx.IPSet has a Ranges call that returns the "minimum and sorted set of IP ranges that covers [the set]".
|
||||||
|
// netipx.IPRange is "an inclusive range of IP addresses from the same address family.". So we can iterate over
|
||||||
|
// all the addresses in the set by keeping a track of the last address we returned, calling Next on the last address
|
||||||
|
// to get the new one, and if we run off the edge of the current range, starting on the next one.
|
||||||
|
type ippool struct {
|
||||||
|
// ranges defines the addresses in the pool
|
||||||
|
ranges []netipx.IPRange
|
||||||
|
// last is internal tracking of which the last address provided was.
|
||||||
|
last netip.Addr
|
||||||
|
// rangeIdx is internal tracking of which netipx.IPRange from the IPSet we are currently on.
|
||||||
|
rangeIdx int
|
||||||
|
}
|
||||||
|
|
||||||
|
func newIPPool(ipset *netipx.IPSet) *ippool {
|
||||||
|
if ipset == nil {
|
||||||
|
return &ippool{}
|
||||||
|
}
|
||||||
|
return &ippool{ranges: ipset.Ranges()}
|
||||||
|
}
|
||||||
|
|
||||||
|
// next returns the next address from the set, or errPoolExhausted if we have
|
||||||
|
// iterated over the whole set.
|
||||||
|
func (ipp *ippool) next() (netip.Addr, error) {
|
||||||
|
if ipp.rangeIdx >= len(ipp.ranges) {
|
||||||
|
// ipset is empty or we have iterated off the end
|
||||||
|
return netip.Addr{}, errPoolExhausted
|
||||||
|
}
|
||||||
|
if !ipp.last.IsValid() {
|
||||||
|
// not initialized yet
|
||||||
|
ipp.last = ipp.ranges[0].From()
|
||||||
|
return ipp.last, nil
|
||||||
|
}
|
||||||
|
currRange := ipp.ranges[ipp.rangeIdx]
|
||||||
|
if ipp.last == currRange.To() {
|
||||||
|
// then we need to move to the next range
|
||||||
|
ipp.rangeIdx++
|
||||||
|
if ipp.rangeIdx >= len(ipp.ranges) {
|
||||||
|
return netip.Addr{}, errPoolExhausted
|
||||||
|
}
|
||||||
|
ipp.last = ipp.ranges[ipp.rangeIdx].From()
|
||||||
|
return ipp.last, nil
|
||||||
|
}
|
||||||
|
ipp.last = ipp.last.Next()
|
||||||
|
return ipp.last, nil
|
||||||
|
}
|
||||||
@ -0,0 +1,60 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
package appc
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"net/netip"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"go4.org/netipx"
|
||||||
|
"tailscale.com/util/must"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestNext(t *testing.T) {
|
||||||
|
a := ippool{}
|
||||||
|
_, err := a.next()
|
||||||
|
if !errors.Is(err, errPoolExhausted) {
|
||||||
|
t.Fatalf("expected errPoolExhausted, got %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var isb netipx.IPSetBuilder
|
||||||
|
ipset := must.Get(isb.IPSet())
|
||||||
|
b := newIPPool(ipset)
|
||||||
|
_, err = b.next()
|
||||||
|
if !errors.Is(err, errPoolExhausted) {
|
||||||
|
t.Fatalf("expected errPoolExhausted, got %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
isb.AddRange(netipx.IPRangeFrom(netip.MustParseAddr("192.168.0.0"), netip.MustParseAddr("192.168.0.2")))
|
||||||
|
isb.AddRange(netipx.IPRangeFrom(netip.MustParseAddr("200.0.0.0"), netip.MustParseAddr("200.0.0.0")))
|
||||||
|
isb.AddRange(netipx.IPRangeFrom(netip.MustParseAddr("201.0.0.0"), netip.MustParseAddr("201.0.0.1")))
|
||||||
|
ipset = must.Get(isb.IPSet())
|
||||||
|
c := newIPPool(ipset)
|
||||||
|
expected := []string{
|
||||||
|
"192.168.0.0",
|
||||||
|
"192.168.0.1",
|
||||||
|
"192.168.0.2",
|
||||||
|
"200.0.0.0",
|
||||||
|
"201.0.0.0",
|
||||||
|
"201.0.0.1",
|
||||||
|
}
|
||||||
|
for i, want := range expected {
|
||||||
|
addr, err := c.next()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if addr != netip.MustParseAddr(want) {
|
||||||
|
t.Fatalf("next call %d want: %s, got: %v", i, want, addr)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_, err = c.next()
|
||||||
|
if !errors.Is(err, errPoolExhausted) {
|
||||||
|
t.Fatalf("expected errPoolExhausted, got %v", err)
|
||||||
|
}
|
||||||
|
_, err = c.next()
|
||||||
|
if !errors.Is(err, errPoolExhausted) {
|
||||||
|
t.Fatalf("expected errPoolExhausted, got %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,132 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
//go:build !ts_omit_appconnectors
|
||||||
|
|
||||||
|
package appc
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/netip"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"golang.org/x/net/dns/dnsmessage"
|
||||||
|
"tailscale.com/util/mak"
|
||||||
|
)
|
||||||
|
|
||||||
|
// ObserveDNSResponse is a callback invoked by the DNS resolver when a DNS
|
||||||
|
// response is being returned over the PeerAPI. The response is parsed and
|
||||||
|
// matched against the configured domains, if matched the routeAdvertiser is
|
||||||
|
// advised to advertise the discovered route.
|
||||||
|
func (e *AppConnector) ObserveDNSResponse(res []byte) error {
|
||||||
|
var p dnsmessage.Parser
|
||||||
|
if _, err := p.Start(res); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := p.SkipAllQuestions(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// cnameChain tracks a chain of CNAMEs for a given query in order to reverse
|
||||||
|
// a CNAME chain back to the original query for flattening. The keys are
|
||||||
|
// CNAME record targets, and the value is the name the record answers, so
|
||||||
|
// for www.example.com CNAME example.com, the map would contain
|
||||||
|
// ["example.com"] = "www.example.com".
|
||||||
|
var cnameChain map[string]string
|
||||||
|
|
||||||
|
// addressRecords is a list of address records found in the response.
|
||||||
|
var addressRecords map[string][]netip.Addr
|
||||||
|
|
||||||
|
for {
|
||||||
|
h, err := p.AnswerHeader()
|
||||||
|
if err == dnsmessage.ErrSectionDone {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if h.Class != dnsmessage.ClassINET {
|
||||||
|
if err := p.SkipAnswer(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
switch h.Type {
|
||||||
|
case dnsmessage.TypeCNAME, dnsmessage.TypeA, dnsmessage.TypeAAAA:
|
||||||
|
default:
|
||||||
|
if err := p.SkipAnswer(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
domain := strings.TrimSuffix(strings.ToLower(h.Name.String()), ".")
|
||||||
|
if len(domain) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if h.Type == dnsmessage.TypeCNAME {
|
||||||
|
res, err := p.CNAMEResource()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
cname := strings.TrimSuffix(strings.ToLower(res.CNAME.String()), ".")
|
||||||
|
if len(cname) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
mak.Set(&cnameChain, cname, domain)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
switch h.Type {
|
||||||
|
case dnsmessage.TypeA:
|
||||||
|
r, err := p.AResource()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
addr := netip.AddrFrom4(r.A)
|
||||||
|
mak.Set(&addressRecords, domain, append(addressRecords[domain], addr))
|
||||||
|
case dnsmessage.TypeAAAA:
|
||||||
|
r, err := p.AAAAResource()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
addr := netip.AddrFrom16(r.AAAA)
|
||||||
|
mak.Set(&addressRecords, domain, append(addressRecords[domain], addr))
|
||||||
|
default:
|
||||||
|
if err := p.SkipAnswer(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
e.mu.Lock()
|
||||||
|
defer e.mu.Unlock()
|
||||||
|
|
||||||
|
for domain, addrs := range addressRecords {
|
||||||
|
domain, isRouted := e.findRoutedDomainLocked(domain, cnameChain)
|
||||||
|
|
||||||
|
// domain and none of the CNAMEs in the chain are routed
|
||||||
|
if !isRouted {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// advertise each address we have learned for the routed domain, that
|
||||||
|
// was not already known.
|
||||||
|
var toAdvertise []netip.Prefix
|
||||||
|
for _, addr := range addrs {
|
||||||
|
if !e.isAddrKnownLocked(domain, addr) {
|
||||||
|
toAdvertise = append(toAdvertise, netip.PrefixFrom(addr, addr.BitLen()))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(toAdvertise) > 0 {
|
||||||
|
e.logf("[v2] observed new routes for %s: %s", domain, toAdvertise)
|
||||||
|
e.scheduleAdvertisement(domain, toAdvertise...)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
@ -0,0 +1,8 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
//go:build ts_omit_appconnectors
|
||||||
|
|
||||||
|
package appc
|
||||||
|
|
||||||
|
func (e *AppConnector) ObserveDNSResponse(res []byte) error { return nil }
|
||||||
@ -0,0 +1,27 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
//go:build tailscale_go
|
||||||
|
|
||||||
|
package tailscaleroot
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
tsRev, ok := tailscaleToolchainRev()
|
||||||
|
if !ok {
|
||||||
|
panic("binary built with tailscale_go build tag but failed to read build info or find tailscale.toolchain.rev in build info")
|
||||||
|
}
|
||||||
|
want := strings.TrimSpace(GoToolchainRev)
|
||||||
|
if tsRev != want {
|
||||||
|
if os.Getenv("TS_PERMIT_TOOLCHAIN_MISMATCH") == "1" {
|
||||||
|
fmt.Fprintf(os.Stderr, "tailscale.toolchain.rev = %q, want %q; but ignoring due to TS_PERMIT_TOOLCHAIN_MISMATCH=1\n", tsRev, want)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
panic(fmt.Sprintf("binary built with tailscale_go build tag but Go toolchain %q doesn't match github.com/tailscale/tailscale expected value %q; override this failure with TS_PERMIT_TOOLCHAIN_MISMATCH=1", tsRev, want))
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,14 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
//go:build !windows
|
||||||
|
|
||||||
|
package atomicfile
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
)
|
||||||
|
|
||||||
|
func rename(srcFile, destFile string) error {
|
||||||
|
return os.Rename(srcFile, destFile)
|
||||||
|
}
|
||||||
@ -0,0 +1,33 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
package atomicfile
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"golang.org/x/sys/windows"
|
||||||
|
)
|
||||||
|
|
||||||
|
func rename(srcFile, destFile string) error {
|
||||||
|
// Use replaceFile when possible to preserve the original file's attributes and ACLs.
|
||||||
|
if err := replaceFile(destFile, srcFile); err == nil || err != windows.ERROR_FILE_NOT_FOUND {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
// destFile doesn't exist. Just do a normal rename.
|
||||||
|
return os.Rename(srcFile, destFile)
|
||||||
|
}
|
||||||
|
|
||||||
|
func replaceFile(destFile, srcFile string) error {
|
||||||
|
destFile16, err := windows.UTF16PtrFromString(destFile)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
srcFile16, err := windows.UTF16PtrFromString(srcFile)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return replaceFileW(destFile16, srcFile16, nil, 0, nil, nil)
|
||||||
|
}
|
||||||
@ -0,0 +1,146 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
package atomicfile
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"testing"
|
||||||
|
"unsafe"
|
||||||
|
|
||||||
|
"golang.org/x/sys/windows"
|
||||||
|
)
|
||||||
|
|
||||||
|
var _SECURITY_RESOURCE_MANAGER_AUTHORITY = windows.SidIdentifierAuthority{[6]byte{0, 0, 0, 0, 0, 9}}
|
||||||
|
|
||||||
|
// makeRandomSID generates a SID derived from a v4 GUID.
|
||||||
|
// This is basically the same algorithm used by browser sandboxes for generating
|
||||||
|
// random SIDs.
|
||||||
|
func makeRandomSID() (*windows.SID, error) {
|
||||||
|
guid, err := windows.GenerateGUID()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
rids := *((*[4]uint32)(unsafe.Pointer(&guid)))
|
||||||
|
|
||||||
|
var pSID *windows.SID
|
||||||
|
if err := windows.AllocateAndInitializeSid(&_SECURITY_RESOURCE_MANAGER_AUTHORITY, 4, rids[0], rids[1], rids[2], rids[3], 0, 0, 0, 0, &pSID); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer windows.FreeSid(pSID)
|
||||||
|
|
||||||
|
// Make a copy that lives on the Go heap
|
||||||
|
return pSID.Copy()
|
||||||
|
}
|
||||||
|
|
||||||
|
func getExistingFileSD(name string) (*windows.SECURITY_DESCRIPTOR, error) {
|
||||||
|
const infoFlags = windows.DACL_SECURITY_INFORMATION
|
||||||
|
return windows.GetNamedSecurityInfo(name, windows.SE_FILE_OBJECT, infoFlags)
|
||||||
|
}
|
||||||
|
|
||||||
|
func getExistingFileDACL(name string) (*windows.ACL, error) {
|
||||||
|
sd, err := getExistingFileSD(name)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
dacl, _, err := sd.DACL()
|
||||||
|
return dacl, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func addDenyACEForRandomSID(dacl *windows.ACL) (*windows.ACL, error) {
|
||||||
|
randomSID, err := makeRandomSID()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
randomSIDTrustee := windows.TRUSTEE{nil, windows.NO_MULTIPLE_TRUSTEE,
|
||||||
|
windows.TRUSTEE_IS_SID, windows.TRUSTEE_IS_UNKNOWN,
|
||||||
|
windows.TrusteeValueFromSID(randomSID)}
|
||||||
|
|
||||||
|
entries := []windows.EXPLICIT_ACCESS{
|
||||||
|
{
|
||||||
|
windows.GENERIC_ALL,
|
||||||
|
windows.DENY_ACCESS,
|
||||||
|
windows.NO_INHERITANCE,
|
||||||
|
randomSIDTrustee,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
return windows.ACLFromEntries(entries, dacl)
|
||||||
|
}
|
||||||
|
|
||||||
|
func setExistingFileDACL(name string, dacl *windows.ACL) error {
|
||||||
|
return windows.SetNamedSecurityInfo(name, windows.SE_FILE_OBJECT,
|
||||||
|
windows.DACL_SECURITY_INFORMATION, nil, nil, dacl, nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
// makeOrigFileWithCustomDACL creates a new, temporary file with a custom
|
||||||
|
// DACL that we can check for later. It returns the name of the temporary
|
||||||
|
// file and the security descriptor for the file in SDDL format.
|
||||||
|
func makeOrigFileWithCustomDACL() (name, sddl string, err error) {
|
||||||
|
f, err := os.CreateTemp("", "foo*.tmp")
|
||||||
|
if err != nil {
|
||||||
|
return "", "", err
|
||||||
|
}
|
||||||
|
name = f.Name()
|
||||||
|
if err := f.Close(); err != nil {
|
||||||
|
return "", "", err
|
||||||
|
}
|
||||||
|
f = nil
|
||||||
|
defer func() {
|
||||||
|
if err != nil {
|
||||||
|
os.Remove(name)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
dacl, err := getExistingFileDACL(name)
|
||||||
|
if err != nil {
|
||||||
|
return "", "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add a harmless, deny-only ACE for a random SID that isn't used for anything
|
||||||
|
// (but that we can check for later).
|
||||||
|
dacl, err = addDenyACEForRandomSID(dacl)
|
||||||
|
if err != nil {
|
||||||
|
return "", "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := setExistingFileDACL(name, dacl); err != nil {
|
||||||
|
return "", "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
sd, err := getExistingFileSD(name)
|
||||||
|
if err != nil {
|
||||||
|
return "", "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
return name, sd.String(), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestPreserveSecurityInfo(t *testing.T) {
|
||||||
|
// Make a test file with a custom ACL.
|
||||||
|
origFileName, want, err := makeOrigFileWithCustomDACL()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("makeOrigFileWithCustomDACL returned %v", err)
|
||||||
|
}
|
||||||
|
t.Cleanup(func() {
|
||||||
|
os.Remove(origFileName)
|
||||||
|
})
|
||||||
|
|
||||||
|
if err := WriteFile(origFileName, []byte{}, 0); err != nil {
|
||||||
|
t.Fatalf("WriteFile returned %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// We expect origFileName's security descriptor to be unchanged despite
|
||||||
|
// the WriteFile call.
|
||||||
|
sd, err := getExistingFileSD(origFileName)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("getExistingFileSD(%q) returned %v", origFileName, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if got := sd.String(); got != want {
|
||||||
|
t.Errorf("security descriptor comparison failed: got %q, want %q", got, want)
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,8 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
package atomicfile
|
||||||
|
|
||||||
|
//go:generate go run golang.org/x/sys/windows/mkwinsyscall -output zsyscall_windows.go mksyscall.go
|
||||||
|
|
||||||
|
//sys replaceFileW(replaced *uint16, replacement *uint16, backup *uint16, flags uint32, exclude unsafe.Pointer, reserved unsafe.Pointer) (err error) [int32(failretval)==0] = kernel32.ReplaceFileW
|
||||||
@ -0,0 +1,52 @@
|
|||||||
|
// Code generated by 'go generate'; DO NOT EDIT.
|
||||||
|
|
||||||
|
package atomicfile
|
||||||
|
|
||||||
|
import (
|
||||||
|
"syscall"
|
||||||
|
"unsafe"
|
||||||
|
|
||||||
|
"golang.org/x/sys/windows"
|
||||||
|
)
|
||||||
|
|
||||||
|
var _ unsafe.Pointer
|
||||||
|
|
||||||
|
// Do the interface allocations only once for common
|
||||||
|
// Errno values.
|
||||||
|
const (
|
||||||
|
errnoERROR_IO_PENDING = 997
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
errERROR_IO_PENDING error = syscall.Errno(errnoERROR_IO_PENDING)
|
||||||
|
errERROR_EINVAL error = syscall.EINVAL
|
||||||
|
)
|
||||||
|
|
||||||
|
// errnoErr returns common boxed Errno values, to prevent
|
||||||
|
// allocations at runtime.
|
||||||
|
func errnoErr(e syscall.Errno) error {
|
||||||
|
switch e {
|
||||||
|
case 0:
|
||||||
|
return errERROR_EINVAL
|
||||||
|
case errnoERROR_IO_PENDING:
|
||||||
|
return errERROR_IO_PENDING
|
||||||
|
}
|
||||||
|
// TODO: add more here, after collecting data on the common
|
||||||
|
// error values see on Windows. (perhaps when running
|
||||||
|
// all.bat?)
|
||||||
|
return e
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
modkernel32 = windows.NewLazySystemDLL("kernel32.dll")
|
||||||
|
|
||||||
|
procReplaceFileW = modkernel32.NewProc("ReplaceFileW")
|
||||||
|
)
|
||||||
|
|
||||||
|
func replaceFileW(replaced *uint16, replacement *uint16, backup *uint16, flags uint32, exclude unsafe.Pointer, reserved unsafe.Pointer) (err error) {
|
||||||
|
r1, _, e1 := syscall.SyscallN(procReplaceFileW.Addr(), uintptr(unsafe.Pointer(replaced)), uintptr(unsafe.Pointer(replacement)), uintptr(unsafe.Pointer(backup)), uintptr(flags), uintptr(exclude), uintptr(reserved))
|
||||||
|
if int32(r1) == 0 {
|
||||||
|
err = errnoErr(e1)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
@ -0,0 +1,151 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
//go:build !js && !ts_omit_acme
|
||||||
|
|
||||||
|
package local
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"crypto/tls"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"net/url"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"go4.org/mem"
|
||||||
|
)
|
||||||
|
|
||||||
|
// SetDNS adds a DNS TXT record for the given domain name, containing
|
||||||
|
// the provided TXT value. The intended use case is answering
|
||||||
|
// LetsEncrypt/ACME dns-01 challenges.
|
||||||
|
//
|
||||||
|
// The control plane will only permit SetDNS requests with very
|
||||||
|
// specific names and values. The name should be
|
||||||
|
// "_acme-challenge." + your node's MagicDNS name. It's expected that
|
||||||
|
// clients cache the certs from LetsEncrypt (or whichever CA is
|
||||||
|
// providing them) and only request new ones as needed; the control plane
|
||||||
|
// rate limits SetDNS requests.
|
||||||
|
//
|
||||||
|
// This is a low-level interface; it's expected that most Tailscale
|
||||||
|
// users use a higher level interface to getting/using TLS
|
||||||
|
// certificates.
|
||||||
|
func (lc *Client) SetDNS(ctx context.Context, name, value string) error {
|
||||||
|
v := url.Values{}
|
||||||
|
v.Set("name", name)
|
||||||
|
v.Set("value", value)
|
||||||
|
_, err := lc.send(ctx, "POST", "/localapi/v0/set-dns?"+v.Encode(), 200, nil)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// CertPair returns a cert and private key for the provided DNS domain.
|
||||||
|
//
|
||||||
|
// It returns a cached certificate from disk if it's still valid.
|
||||||
|
//
|
||||||
|
// Deprecated: use [Client.CertPair].
|
||||||
|
func CertPair(ctx context.Context, domain string) (certPEM, keyPEM []byte, err error) {
|
||||||
|
return defaultClient.CertPair(ctx, domain)
|
||||||
|
}
|
||||||
|
|
||||||
|
// CertPair returns a cert and private key for the provided DNS domain.
|
||||||
|
//
|
||||||
|
// It returns a cached certificate from disk if it's still valid.
|
||||||
|
//
|
||||||
|
// API maturity: this is considered a stable API.
|
||||||
|
func (lc *Client) CertPair(ctx context.Context, domain string) (certPEM, keyPEM []byte, err error) {
|
||||||
|
return lc.CertPairWithValidity(ctx, domain, 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
// CertPairWithValidity returns a cert and private key for the provided DNS
|
||||||
|
// domain.
|
||||||
|
//
|
||||||
|
// It returns a cached certificate from disk if it's still valid.
|
||||||
|
// When minValidity is non-zero, the returned certificate will be valid for at
|
||||||
|
// least the given duration, if permitted by the CA. If the certificate is
|
||||||
|
// valid, but for less than minValidity, it will be synchronously renewed.
|
||||||
|
//
|
||||||
|
// API maturity: this is considered a stable API.
|
||||||
|
func (lc *Client) CertPairWithValidity(ctx context.Context, domain string, minValidity time.Duration) (certPEM, keyPEM []byte, err error) {
|
||||||
|
res, err := lc.send(ctx, "GET", fmt.Sprintf("/localapi/v0/cert/%s?type=pair&min_validity=%s", domain, minValidity), 200, nil)
|
||||||
|
if err != nil {
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
// with ?type=pair, the response PEM is first the one private
|
||||||
|
// key PEM block, then the cert PEM blocks.
|
||||||
|
i := mem.Index(mem.B(res), mem.S("--\n--"))
|
||||||
|
if i == -1 {
|
||||||
|
return nil, nil, fmt.Errorf("unexpected output: no delimiter")
|
||||||
|
}
|
||||||
|
i += len("--\n")
|
||||||
|
keyPEM, certPEM = res[:i], res[i:]
|
||||||
|
if mem.Contains(mem.B(certPEM), mem.S(" PRIVATE KEY-----")) {
|
||||||
|
return nil, nil, fmt.Errorf("unexpected output: key in cert")
|
||||||
|
}
|
||||||
|
return certPEM, keyPEM, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetCertificate fetches a TLS certificate for the TLS ClientHello in hi.
|
||||||
|
//
|
||||||
|
// It returns a cached certificate from disk if it's still valid.
|
||||||
|
//
|
||||||
|
// It's the right signature to use as the value of
|
||||||
|
// [tls.Config.GetCertificate].
|
||||||
|
//
|
||||||
|
// Deprecated: use [Client.GetCertificate].
|
||||||
|
func GetCertificate(hi *tls.ClientHelloInfo) (*tls.Certificate, error) {
|
||||||
|
return defaultClient.GetCertificate(hi)
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetCertificate fetches a TLS certificate for the TLS ClientHello in hi.
|
||||||
|
//
|
||||||
|
// It returns a cached certificate from disk if it's still valid.
|
||||||
|
//
|
||||||
|
// It's the right signature to use as the value of
|
||||||
|
// [tls.Config.GetCertificate].
|
||||||
|
//
|
||||||
|
// API maturity: this is considered a stable API.
|
||||||
|
func (lc *Client) GetCertificate(hi *tls.ClientHelloInfo) (*tls.Certificate, error) {
|
||||||
|
if hi == nil || hi.ServerName == "" {
|
||||||
|
return nil, errors.New("no SNI ServerName")
|
||||||
|
}
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), time.Minute)
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
name := hi.ServerName
|
||||||
|
if !strings.Contains(name, ".") {
|
||||||
|
if v, ok := lc.ExpandSNIName(ctx, name); ok {
|
||||||
|
name = v
|
||||||
|
}
|
||||||
|
}
|
||||||
|
certPEM, keyPEM, err := lc.CertPair(ctx, name)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
cert, err := tls.X509KeyPair(certPEM, keyPEM)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &cert, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ExpandSNIName expands bare label name into the most likely actual TLS cert name.
|
||||||
|
//
|
||||||
|
// Deprecated: use [Client.ExpandSNIName].
|
||||||
|
func ExpandSNIName(ctx context.Context, name string) (fqdn string, ok bool) {
|
||||||
|
return defaultClient.ExpandSNIName(ctx, name)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ExpandSNIName expands bare label name into the most likely actual TLS cert name.
|
||||||
|
func (lc *Client) ExpandSNIName(ctx context.Context, name string) (fqdn string, ok bool) {
|
||||||
|
st, err := lc.StatusWithoutPeers(ctx)
|
||||||
|
if err != nil {
|
||||||
|
return "", false
|
||||||
|
}
|
||||||
|
for _, d := range st.CertDomains {
|
||||||
|
if len(d) > len(name)+1 && strings.HasPrefix(d, name) && d[len(name)] == '.' {
|
||||||
|
return d, true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return "", false
|
||||||
|
}
|
||||||
@ -0,0 +1,84 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
//go:build !ts_omit_debugportmapper
|
||||||
|
|
||||||
|
package local
|
||||||
|
|
||||||
|
import (
|
||||||
|
"cmp"
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"net/netip"
|
||||||
|
"net/url"
|
||||||
|
"strconv"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"tailscale.com/client/tailscale/apitype"
|
||||||
|
)
|
||||||
|
|
||||||
|
// DebugPortmapOpts contains options for the [Client.DebugPortmap] command.
|
||||||
|
type DebugPortmapOpts struct {
|
||||||
|
// Duration is how long the mapping should be created for. It defaults
|
||||||
|
// to 5 seconds if not set.
|
||||||
|
Duration time.Duration
|
||||||
|
|
||||||
|
// Type is the kind of portmap to debug. The empty string instructs the
|
||||||
|
// portmap client to perform all known types. Other valid options are
|
||||||
|
// "pmp", "pcp", and "upnp".
|
||||||
|
Type string
|
||||||
|
|
||||||
|
// GatewayAddr specifies the gateway address used during portmapping.
|
||||||
|
// If set, SelfAddr must also be set. If unset, it will be
|
||||||
|
// autodetected.
|
||||||
|
GatewayAddr netip.Addr
|
||||||
|
|
||||||
|
// SelfAddr specifies the gateway address used during portmapping. If
|
||||||
|
// set, GatewayAddr must also be set. If unset, it will be
|
||||||
|
// autodetected.
|
||||||
|
SelfAddr netip.Addr
|
||||||
|
|
||||||
|
// LogHTTP instructs the debug-portmap endpoint to print all HTTP
|
||||||
|
// requests and responses made to the logs.
|
||||||
|
LogHTTP bool
|
||||||
|
}
|
||||||
|
|
||||||
|
// DebugPortmap invokes the debug-portmap endpoint, and returns an
|
||||||
|
// io.ReadCloser that can be used to read the logs that are printed during this
|
||||||
|
// process.
|
||||||
|
//
|
||||||
|
// opts can be nil; if so, default values will be used.
|
||||||
|
func (lc *Client) DebugPortmap(ctx context.Context, opts *DebugPortmapOpts) (io.ReadCloser, error) {
|
||||||
|
vals := make(url.Values)
|
||||||
|
if opts == nil {
|
||||||
|
opts = &DebugPortmapOpts{}
|
||||||
|
}
|
||||||
|
|
||||||
|
vals.Set("duration", cmp.Or(opts.Duration, 5*time.Second).String())
|
||||||
|
vals.Set("type", opts.Type)
|
||||||
|
vals.Set("log_http", strconv.FormatBool(opts.LogHTTP))
|
||||||
|
|
||||||
|
if opts.GatewayAddr.IsValid() != opts.SelfAddr.IsValid() {
|
||||||
|
return nil, fmt.Errorf("both GatewayAddr and SelfAddr must be provided if one is")
|
||||||
|
} else if opts.GatewayAddr.IsValid() {
|
||||||
|
vals.Set("gateway_and_self", fmt.Sprintf("%s/%s", opts.GatewayAddr, opts.SelfAddr))
|
||||||
|
}
|
||||||
|
|
||||||
|
req, err := http.NewRequestWithContext(ctx, "GET", "http://"+apitype.LocalAPIHost+"/localapi/v0/debug-portmap?"+vals.Encode(), nil)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
res, err := lc.doLocalRequestNiceError(req)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if res.StatusCode != 200 {
|
||||||
|
body, _ := io.ReadAll(res.Body)
|
||||||
|
res.Body.Close()
|
||||||
|
return nil, fmt.Errorf("HTTP %s: %s", res.Status, body)
|
||||||
|
}
|
||||||
|
|
||||||
|
return res.Body, nil
|
||||||
|
}
|
||||||
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,55 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
//go:build !ts_omit_serve
|
||||||
|
|
||||||
|
package local
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"tailscale.com/ipn"
|
||||||
|
)
|
||||||
|
|
||||||
|
// GetServeConfig return the current serve config.
|
||||||
|
//
|
||||||
|
// If the serve config is empty, it returns (nil, nil).
|
||||||
|
func (lc *Client) GetServeConfig(ctx context.Context) (*ipn.ServeConfig, error) {
|
||||||
|
body, h, err := lc.sendWithHeaders(ctx, "GET", "/localapi/v0/serve-config", 200, nil, nil)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("getting serve config: %w", err)
|
||||||
|
}
|
||||||
|
sc, err := getServeConfigFromJSON(body)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if sc == nil {
|
||||||
|
sc = new(ipn.ServeConfig)
|
||||||
|
}
|
||||||
|
sc.ETag = h.Get("Etag")
|
||||||
|
return sc, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func getServeConfigFromJSON(body []byte) (sc *ipn.ServeConfig, err error) {
|
||||||
|
if err := json.Unmarshal(body, &sc); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return sc, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetServeConfig sets or replaces the serving settings.
|
||||||
|
// If config is nil, settings are cleared and serving is disabled.
|
||||||
|
func (lc *Client) SetServeConfig(ctx context.Context, config *ipn.ServeConfig) error {
|
||||||
|
h := make(http.Header)
|
||||||
|
if config != nil {
|
||||||
|
h.Set("If-Match", config.ETag)
|
||||||
|
}
|
||||||
|
_, _, err := lc.sendWithHeaders(ctx, "POST", "/localapi/v0/serve-config", 200, jsonBody(config), h)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("sending serve config: %w", err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
@ -0,0 +1,40 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
//go:build !ts_omit_syspolicy
|
||||||
|
|
||||||
|
package local
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"tailscale.com/util/syspolicy/setting"
|
||||||
|
)
|
||||||
|
|
||||||
|
// GetEffectivePolicy returns the effective policy for the specified scope.
|
||||||
|
func (lc *Client) GetEffectivePolicy(ctx context.Context, scope setting.PolicyScope) (*setting.Snapshot, error) {
|
||||||
|
scopeID, err := scope.MarshalText()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
body, err := lc.get200(ctx, "/localapi/v0/policy/"+string(scopeID))
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return decodeJSON[*setting.Snapshot](body)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ReloadEffectivePolicy reloads the effective policy for the specified scope
|
||||||
|
// by reading and merging policy settings from all applicable policy sources.
|
||||||
|
func (lc *Client) ReloadEffectivePolicy(ctx context.Context, scope setting.PolicyScope) (*setting.Snapshot, error) {
|
||||||
|
scopeID, err := scope.MarshalText()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
body, err := lc.send(ctx, "POST", "/localapi/v0/policy/"+string(scopeID), 200, http.NoBody)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return decodeJSON[*setting.Snapshot](body)
|
||||||
|
}
|
||||||
@ -0,0 +1,204 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
//go:build !ts_omit_tailnetlock
|
||||||
|
|
||||||
|
package local
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"net/url"
|
||||||
|
|
||||||
|
"tailscale.com/ipn/ipnstate"
|
||||||
|
"tailscale.com/tka"
|
||||||
|
"tailscale.com/types/key"
|
||||||
|
"tailscale.com/types/tkatype"
|
||||||
|
)
|
||||||
|
|
||||||
|
// NetworkLockStatus fetches information about the tailnet key authority, if one is configured.
|
||||||
|
func (lc *Client) NetworkLockStatus(ctx context.Context) (*ipnstate.NetworkLockStatus, error) {
|
||||||
|
body, err := lc.send(ctx, "GET", "/localapi/v0/tka/status", 200, nil)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("error: %w", err)
|
||||||
|
}
|
||||||
|
return decodeJSON[*ipnstate.NetworkLockStatus](body)
|
||||||
|
}
|
||||||
|
|
||||||
|
// NetworkLockInit initializes the tailnet key authority.
|
||||||
|
//
|
||||||
|
// TODO(tom): Plumb through disablement secrets.
|
||||||
|
func (lc *Client) NetworkLockInit(ctx context.Context, keys []tka.Key, disablementValues [][]byte, supportDisablement []byte) (*ipnstate.NetworkLockStatus, error) {
|
||||||
|
var b bytes.Buffer
|
||||||
|
type initRequest struct {
|
||||||
|
Keys []tka.Key
|
||||||
|
DisablementValues [][]byte
|
||||||
|
SupportDisablement []byte
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := json.NewEncoder(&b).Encode(initRequest{Keys: keys, DisablementValues: disablementValues, SupportDisablement: supportDisablement}); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
body, err := lc.send(ctx, "POST", "/localapi/v0/tka/init", 200, &b)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("error: %w", err)
|
||||||
|
}
|
||||||
|
return decodeJSON[*ipnstate.NetworkLockStatus](body)
|
||||||
|
}
|
||||||
|
|
||||||
|
// NetworkLockWrapPreauthKey wraps a pre-auth key with information to
|
||||||
|
// enable unattended bringup in the locked tailnet.
|
||||||
|
func (lc *Client) NetworkLockWrapPreauthKey(ctx context.Context, preauthKey string, tkaKey key.NLPrivate) (string, error) {
|
||||||
|
encodedPrivate, err := tkaKey.MarshalText()
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
var b bytes.Buffer
|
||||||
|
type wrapRequest struct {
|
||||||
|
TSKey string
|
||||||
|
TKAKey string // key.NLPrivate.MarshalText
|
||||||
|
}
|
||||||
|
if err := json.NewEncoder(&b).Encode(wrapRequest{TSKey: preauthKey, TKAKey: string(encodedPrivate)}); err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
body, err := lc.send(ctx, "POST", "/localapi/v0/tka/wrap-preauth-key", 200, &b)
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("error: %w", err)
|
||||||
|
}
|
||||||
|
return string(body), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// NetworkLockModify adds and/or removes key(s) to the tailnet key authority.
|
||||||
|
func (lc *Client) NetworkLockModify(ctx context.Context, addKeys, removeKeys []tka.Key) error {
|
||||||
|
var b bytes.Buffer
|
||||||
|
type modifyRequest struct {
|
||||||
|
AddKeys []tka.Key
|
||||||
|
RemoveKeys []tka.Key
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := json.NewEncoder(&b).Encode(modifyRequest{AddKeys: addKeys, RemoveKeys: removeKeys}); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err := lc.send(ctx, "POST", "/localapi/v0/tka/modify", 204, &b); err != nil {
|
||||||
|
return fmt.Errorf("error: %w", err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// NetworkLockSign signs the specified node-key and transmits that signature to the control plane.
|
||||||
|
// rotationPublic, if specified, must be an ed25519 public key.
|
||||||
|
func (lc *Client) NetworkLockSign(ctx context.Context, nodeKey key.NodePublic, rotationPublic []byte) error {
|
||||||
|
var b bytes.Buffer
|
||||||
|
type signRequest struct {
|
||||||
|
NodeKey key.NodePublic
|
||||||
|
RotationPublic []byte
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := json.NewEncoder(&b).Encode(signRequest{NodeKey: nodeKey, RotationPublic: rotationPublic}); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err := lc.send(ctx, "POST", "/localapi/v0/tka/sign", 200, &b); err != nil {
|
||||||
|
return fmt.Errorf("error: %w", err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// NetworkLockAffectedSigs returns all signatures signed by the specified keyID.
|
||||||
|
func (lc *Client) NetworkLockAffectedSigs(ctx context.Context, keyID tkatype.KeyID) ([]tkatype.MarshaledSignature, error) {
|
||||||
|
body, err := lc.send(ctx, "POST", "/localapi/v0/tka/affected-sigs", 200, bytes.NewReader(keyID))
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("error: %w", err)
|
||||||
|
}
|
||||||
|
return decodeJSON[[]tkatype.MarshaledSignature](body)
|
||||||
|
}
|
||||||
|
|
||||||
|
// NetworkLockLog returns up to maxEntries number of changes to network-lock state.
|
||||||
|
func (lc *Client) NetworkLockLog(ctx context.Context, maxEntries int) ([]ipnstate.NetworkLockUpdate, error) {
|
||||||
|
v := url.Values{}
|
||||||
|
v.Set("limit", fmt.Sprint(maxEntries))
|
||||||
|
body, err := lc.send(ctx, "GET", "/localapi/v0/tka/log?"+v.Encode(), 200, nil)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("error %w: %s", err, body)
|
||||||
|
}
|
||||||
|
return decodeJSON[[]ipnstate.NetworkLockUpdate](body)
|
||||||
|
}
|
||||||
|
|
||||||
|
// NetworkLockForceLocalDisable forcibly shuts down network lock on this node.
|
||||||
|
func (lc *Client) NetworkLockForceLocalDisable(ctx context.Context) error {
|
||||||
|
// This endpoint expects an empty JSON stanza as the payload.
|
||||||
|
var b bytes.Buffer
|
||||||
|
if err := json.NewEncoder(&b).Encode(struct{}{}); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err := lc.send(ctx, "POST", "/localapi/v0/tka/force-local-disable", 200, &b); err != nil {
|
||||||
|
return fmt.Errorf("error: %w", err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// NetworkLockVerifySigningDeeplink verifies the network lock deeplink contained
|
||||||
|
// in url and returns information extracted from it.
|
||||||
|
func (lc *Client) NetworkLockVerifySigningDeeplink(ctx context.Context, url string) (*tka.DeeplinkValidationResult, error) {
|
||||||
|
vr := struct {
|
||||||
|
URL string
|
||||||
|
}{url}
|
||||||
|
|
||||||
|
body, err := lc.send(ctx, "POST", "/localapi/v0/tka/verify-deeplink", 200, jsonBody(vr))
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("sending verify-deeplink: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return decodeJSON[*tka.DeeplinkValidationResult](body)
|
||||||
|
}
|
||||||
|
|
||||||
|
// NetworkLockGenRecoveryAUM generates an AUM for recovering from a tailnet-lock key compromise.
|
||||||
|
func (lc *Client) NetworkLockGenRecoveryAUM(ctx context.Context, removeKeys []tkatype.KeyID, forkFrom tka.AUMHash) ([]byte, error) {
|
||||||
|
vr := struct {
|
||||||
|
Keys []tkatype.KeyID
|
||||||
|
ForkFrom string
|
||||||
|
}{removeKeys, forkFrom.String()}
|
||||||
|
|
||||||
|
body, err := lc.send(ctx, "POST", "/localapi/v0/tka/generate-recovery-aum", 200, jsonBody(vr))
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("sending generate-recovery-aum: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return body, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// NetworkLockCosignRecoveryAUM co-signs a recovery AUM using the node's tailnet lock key.
|
||||||
|
func (lc *Client) NetworkLockCosignRecoveryAUM(ctx context.Context, aum tka.AUM) ([]byte, error) {
|
||||||
|
r := bytes.NewReader(aum.Serialize())
|
||||||
|
body, err := lc.send(ctx, "POST", "/localapi/v0/tka/cosign-recovery-aum", 200, r)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("sending cosign-recovery-aum: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return body, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// NetworkLockSubmitRecoveryAUM submits a recovery AUM to the control plane.
|
||||||
|
func (lc *Client) NetworkLockSubmitRecoveryAUM(ctx context.Context, aum tka.AUM) error {
|
||||||
|
r := bytes.NewReader(aum.Serialize())
|
||||||
|
_, err := lc.send(ctx, "POST", "/localapi/v0/tka/submit-recovery-aum", 200, r)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("sending cosign-recovery-aum: %w", err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// NetworkLockDisable shuts down network-lock across the tailnet.
|
||||||
|
func (lc *Client) NetworkLockDisable(ctx context.Context, secret []byte) error {
|
||||||
|
if _, err := lc.send(ctx, "POST", "/localapi/v0/tka/disable", 200, bytes.NewReader(secret)); err != nil {
|
||||||
|
return fmt.Errorf("error: %w", err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
@ -0,0 +1,327 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
//go:build cgo || !darwin
|
||||||
|
|
||||||
|
package systray
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"context"
|
||||||
|
"image"
|
||||||
|
"image/color"
|
||||||
|
"image/png"
|
||||||
|
"runtime"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"fyne.io/systray"
|
||||||
|
ico "github.com/Kodeworks/golang-image-ico"
|
||||||
|
"github.com/fogleman/gg"
|
||||||
|
)
|
||||||
|
|
||||||
|
// tsLogo represents the Tailscale logo displayed as the systray icon.
|
||||||
|
type tsLogo struct {
|
||||||
|
// dots represents the state of the 3x3 dot grid in the logo.
|
||||||
|
// A 0 represents a gray dot, any other value is a white dot.
|
||||||
|
dots [9]byte
|
||||||
|
|
||||||
|
// dotMask returns an image mask to be used when rendering the logo dots.
|
||||||
|
dotMask func(dc *gg.Context, borderUnits int, radius int) *image.Alpha
|
||||||
|
|
||||||
|
// overlay is called after the dots are rendered to draw an additional overlay.
|
||||||
|
overlay func(dc *gg.Context, borderUnits int, radius int)
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
// disconnected is all gray dots
|
||||||
|
disconnected = tsLogo{dots: [9]byte{
|
||||||
|
0, 0, 0,
|
||||||
|
0, 0, 0,
|
||||||
|
0, 0, 0,
|
||||||
|
}}
|
||||||
|
|
||||||
|
// connected is the normal Tailscale logo
|
||||||
|
connected = tsLogo{dots: [9]byte{
|
||||||
|
0, 0, 0,
|
||||||
|
1, 1, 1,
|
||||||
|
0, 1, 0,
|
||||||
|
}}
|
||||||
|
|
||||||
|
// loading is a special tsLogo value that is not meant to be rendered directly,
|
||||||
|
// but indicates that the loading animation should be shown.
|
||||||
|
loading = tsLogo{dots: [9]byte{'l', 'o', 'a', 'd', 'i', 'n', 'g'}}
|
||||||
|
|
||||||
|
// loadingIcons are shown in sequence as an animated loading icon.
|
||||||
|
loadingLogos = []tsLogo{
|
||||||
|
{dots: [9]byte{
|
||||||
|
0, 1, 1,
|
||||||
|
1, 0, 1,
|
||||||
|
0, 0, 1,
|
||||||
|
}},
|
||||||
|
{dots: [9]byte{
|
||||||
|
0, 1, 1,
|
||||||
|
0, 0, 1,
|
||||||
|
0, 1, 0,
|
||||||
|
}},
|
||||||
|
{dots: [9]byte{
|
||||||
|
0, 1, 1,
|
||||||
|
0, 0, 0,
|
||||||
|
0, 0, 1,
|
||||||
|
}},
|
||||||
|
{dots: [9]byte{
|
||||||
|
0, 0, 1,
|
||||||
|
0, 1, 0,
|
||||||
|
0, 0, 0,
|
||||||
|
}},
|
||||||
|
{dots: [9]byte{
|
||||||
|
0, 1, 0,
|
||||||
|
0, 0, 0,
|
||||||
|
0, 0, 0,
|
||||||
|
}},
|
||||||
|
{dots: [9]byte{
|
||||||
|
0, 0, 0,
|
||||||
|
0, 0, 1,
|
||||||
|
0, 0, 0,
|
||||||
|
}},
|
||||||
|
{dots: [9]byte{
|
||||||
|
0, 0, 0,
|
||||||
|
0, 0, 0,
|
||||||
|
0, 0, 0,
|
||||||
|
}},
|
||||||
|
{dots: [9]byte{
|
||||||
|
0, 0, 1,
|
||||||
|
0, 0, 0,
|
||||||
|
0, 0, 0,
|
||||||
|
}},
|
||||||
|
{dots: [9]byte{
|
||||||
|
0, 0, 0,
|
||||||
|
0, 0, 0,
|
||||||
|
1, 0, 0,
|
||||||
|
}},
|
||||||
|
{dots: [9]byte{
|
||||||
|
0, 0, 0,
|
||||||
|
0, 0, 0,
|
||||||
|
1, 1, 0,
|
||||||
|
}},
|
||||||
|
{dots: [9]byte{
|
||||||
|
0, 0, 0,
|
||||||
|
1, 0, 0,
|
||||||
|
1, 1, 0,
|
||||||
|
}},
|
||||||
|
{dots: [9]byte{
|
||||||
|
0, 0, 0,
|
||||||
|
1, 1, 0,
|
||||||
|
0, 1, 0,
|
||||||
|
}},
|
||||||
|
{dots: [9]byte{
|
||||||
|
0, 0, 0,
|
||||||
|
1, 1, 0,
|
||||||
|
0, 1, 1,
|
||||||
|
}},
|
||||||
|
{dots: [9]byte{
|
||||||
|
0, 0, 0,
|
||||||
|
1, 1, 1,
|
||||||
|
0, 0, 1,
|
||||||
|
}},
|
||||||
|
{dots: [9]byte{
|
||||||
|
0, 1, 0,
|
||||||
|
0, 1, 1,
|
||||||
|
1, 0, 1,
|
||||||
|
}},
|
||||||
|
}
|
||||||
|
|
||||||
|
// exitNodeOnline is the Tailscale logo with an additional arrow overlay in the corner.
|
||||||
|
exitNodeOnline = tsLogo{
|
||||||
|
dots: [9]byte{
|
||||||
|
0, 0, 0,
|
||||||
|
1, 1, 1,
|
||||||
|
0, 1, 0,
|
||||||
|
},
|
||||||
|
// draw an arrow mask in the bottom right corner with a reasonably thick line width.
|
||||||
|
dotMask: func(dc *gg.Context, borderUnits int, radius int) *image.Alpha {
|
||||||
|
bu, r := float64(borderUnits), float64(radius)
|
||||||
|
|
||||||
|
x1 := r * (bu + 3.5)
|
||||||
|
y := r * (bu + 7)
|
||||||
|
x2 := x1 + (r * 5)
|
||||||
|
|
||||||
|
mc := gg.NewContext(dc.Width(), dc.Height())
|
||||||
|
mc.DrawLine(x1, y, x2, y) // arrow center line
|
||||||
|
mc.DrawLine(x2-(1.5*r), y-(1.5*r), x2, y) // top of arrow tip
|
||||||
|
mc.DrawLine(x2-(1.5*r), y+(1.5*r), x2, y) // bottom of arrow tip
|
||||||
|
mc.SetLineWidth(r * 3)
|
||||||
|
mc.Stroke()
|
||||||
|
return mc.AsMask()
|
||||||
|
},
|
||||||
|
// draw an arrow in the bottom right corner over the masked area.
|
||||||
|
overlay: func(dc *gg.Context, borderUnits int, radius int) {
|
||||||
|
bu, r := float64(borderUnits), float64(radius)
|
||||||
|
|
||||||
|
x1 := r * (bu + 3.5)
|
||||||
|
y := r * (bu + 7)
|
||||||
|
x2 := x1 + (r * 5)
|
||||||
|
|
||||||
|
dc.DrawLine(x1, y, x2, y) // arrow center line
|
||||||
|
dc.DrawLine(x2-(1.5*r), y-(1.5*r), x2, y) // top of arrow tip
|
||||||
|
dc.DrawLine(x2-(1.5*r), y+(1.5*r), x2, y) // bottom of arrow tip
|
||||||
|
dc.SetColor(fg)
|
||||||
|
dc.SetLineWidth(r)
|
||||||
|
dc.Stroke()
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
// exitNodeOffline is the Tailscale logo with a red "x" in the corner.
|
||||||
|
exitNodeOffline = tsLogo{
|
||||||
|
dots: [9]byte{
|
||||||
|
0, 0, 0,
|
||||||
|
1, 1, 1,
|
||||||
|
0, 1, 0,
|
||||||
|
},
|
||||||
|
// Draw a square that hides the four dots in the bottom right corner,
|
||||||
|
dotMask: func(dc *gg.Context, borderUnits int, radius int) *image.Alpha {
|
||||||
|
bu, r := float64(borderUnits), float64(radius)
|
||||||
|
x := r * (bu + 3)
|
||||||
|
|
||||||
|
mc := gg.NewContext(dc.Width(), dc.Height())
|
||||||
|
mc.DrawRectangle(x, x, r*6, r*6)
|
||||||
|
mc.Fill()
|
||||||
|
return mc.AsMask()
|
||||||
|
},
|
||||||
|
// draw a red "x" over the bottom right corner.
|
||||||
|
overlay: func(dc *gg.Context, borderUnits int, radius int) {
|
||||||
|
bu, r := float64(borderUnits), float64(radius)
|
||||||
|
|
||||||
|
x1 := r * (bu + 4)
|
||||||
|
x2 := x1 + (r * 3.5)
|
||||||
|
dc.DrawLine(x1, x1, x2, x2) // top-left to bottom-right stroke
|
||||||
|
dc.DrawLine(x1, x2, x2, x1) // bottom-left to top-right stroke
|
||||||
|
dc.SetColor(red)
|
||||||
|
dc.SetLineWidth(r)
|
||||||
|
dc.Stroke()
|
||||||
|
},
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
bg = color.NRGBA{0, 0, 0, 255}
|
||||||
|
fg = color.NRGBA{255, 255, 255, 255}
|
||||||
|
gray = color.NRGBA{255, 255, 255, 102}
|
||||||
|
red = color.NRGBA{229, 111, 74, 255}
|
||||||
|
)
|
||||||
|
|
||||||
|
// render returns a PNG image of the logo.
|
||||||
|
func (logo tsLogo) render() *bytes.Buffer {
|
||||||
|
const borderUnits = 1
|
||||||
|
return logo.renderWithBorder(borderUnits)
|
||||||
|
}
|
||||||
|
|
||||||
|
// renderWithBorder returns a PNG image of the logo with the specified border width.
|
||||||
|
// One border unit is equal to the radius of a tailscale logo dot.
|
||||||
|
func (logo tsLogo) renderWithBorder(borderUnits int) *bytes.Buffer {
|
||||||
|
const radius = 25
|
||||||
|
dim := radius * (8 + borderUnits*2)
|
||||||
|
|
||||||
|
dc := gg.NewContext(dim, dim)
|
||||||
|
dc.DrawRectangle(0, 0, float64(dim), float64(dim))
|
||||||
|
dc.SetColor(bg)
|
||||||
|
dc.Fill()
|
||||||
|
|
||||||
|
if logo.dotMask != nil {
|
||||||
|
mask := logo.dotMask(dc, borderUnits, radius)
|
||||||
|
dc.SetMask(mask)
|
||||||
|
dc.InvertMask()
|
||||||
|
}
|
||||||
|
|
||||||
|
for y := 0; y < 3; y++ {
|
||||||
|
for x := 0; x < 3; x++ {
|
||||||
|
px := (borderUnits + 1 + 3*x) * radius
|
||||||
|
py := (borderUnits + 1 + 3*y) * radius
|
||||||
|
col := fg
|
||||||
|
if logo.dots[y*3+x] == 0 {
|
||||||
|
col = gray
|
||||||
|
}
|
||||||
|
dc.DrawCircle(float64(px), float64(py), radius)
|
||||||
|
dc.SetColor(col)
|
||||||
|
dc.Fill()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if logo.overlay != nil {
|
||||||
|
dc.ResetClip()
|
||||||
|
logo.overlay(dc, borderUnits, radius)
|
||||||
|
}
|
||||||
|
|
||||||
|
b := bytes.NewBuffer(nil)
|
||||||
|
|
||||||
|
// Encode as ICO format on Windows, PNG on all other platforms.
|
||||||
|
if runtime.GOOS == "windows" {
|
||||||
|
_ = ico.Encode(b, dc.Image())
|
||||||
|
} else {
|
||||||
|
_ = png.Encode(b, dc.Image())
|
||||||
|
}
|
||||||
|
return b
|
||||||
|
}
|
||||||
|
|
||||||
|
// setAppIcon renders logo and sets it as the systray icon.
|
||||||
|
func setAppIcon(icon tsLogo) {
|
||||||
|
if icon.dots == loading.dots {
|
||||||
|
startLoadingAnimation()
|
||||||
|
} else {
|
||||||
|
stopLoadingAnimation()
|
||||||
|
systray.SetIcon(icon.render().Bytes())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
loadingMu sync.Mutex // protects loadingCancel
|
||||||
|
|
||||||
|
// loadingCancel stops the loading animation in the systray icon.
|
||||||
|
// This is nil if the animation is not currently active.
|
||||||
|
loadingCancel func()
|
||||||
|
)
|
||||||
|
|
||||||
|
// startLoadingAnimation starts the animated loading icon in the system tray.
|
||||||
|
// The animation continues until [stopLoadingAnimation] is called.
|
||||||
|
// If the loading animation is already active, this func does nothing.
|
||||||
|
func startLoadingAnimation() {
|
||||||
|
loadingMu.Lock()
|
||||||
|
defer loadingMu.Unlock()
|
||||||
|
|
||||||
|
if loadingCancel != nil {
|
||||||
|
// loading icon already displayed
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
ctx, loadingCancel = context.WithCancel(ctx)
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
t := time.NewTicker(500 * time.Millisecond)
|
||||||
|
var i int
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
return
|
||||||
|
case <-t.C:
|
||||||
|
systray.SetIcon(loadingLogos[i].render().Bytes())
|
||||||
|
i++
|
||||||
|
if i >= len(loadingLogos) {
|
||||||
|
i = 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
|
||||||
|
// stopLoadingAnimation stops the animated loading icon in the system tray.
|
||||||
|
// If the loading animation is not currently active, this func does nothing.
|
||||||
|
func stopLoadingAnimation() {
|
||||||
|
loadingMu.Lock()
|
||||||
|
defer loadingMu.Unlock()
|
||||||
|
|
||||||
|
if loadingCancel != nil {
|
||||||
|
loadingCancel()
|
||||||
|
loadingCancel = nil
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,76 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
//go:build cgo || !darwin
|
||||||
|
|
||||||
|
// Package systray provides a minimal Tailscale systray application.
|
||||||
|
package systray
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"bytes"
|
||||||
|
_ "embed"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"os/exec"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
//go:embed tailscale-systray.service
|
||||||
|
var embedSystemd string
|
||||||
|
|
||||||
|
func InstallStartupScript(initSystem string) error {
|
||||||
|
switch initSystem {
|
||||||
|
case "systemd":
|
||||||
|
return installSystemd()
|
||||||
|
default:
|
||||||
|
return fmt.Errorf("unsupported init system '%s'", initSystem)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func installSystemd() error {
|
||||||
|
// Find the path to tailscale, just in case it's not where the example file
|
||||||
|
// has it placed, and replace that before writing the file.
|
||||||
|
tailscaleBin, err := exec.LookPath("tailscale")
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to find tailscale binary %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var output bytes.Buffer
|
||||||
|
scanner := bufio.NewScanner(strings.NewReader(embedSystemd))
|
||||||
|
for scanner.Scan() {
|
||||||
|
line := scanner.Text()
|
||||||
|
if strings.HasPrefix(line, "ExecStart=") {
|
||||||
|
line = fmt.Sprintf("ExecStart=%s systray", tailscaleBin)
|
||||||
|
}
|
||||||
|
output.WriteString(line + "\n")
|
||||||
|
}
|
||||||
|
|
||||||
|
configDir, err := os.UserConfigDir()
|
||||||
|
if err != nil {
|
||||||
|
homeDir, err := os.UserHomeDir()
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("unable to locate user home: %w", err)
|
||||||
|
}
|
||||||
|
configDir = filepath.Join(homeDir, ".config")
|
||||||
|
}
|
||||||
|
|
||||||
|
systemdDir := filepath.Join(configDir, "systemd", "user")
|
||||||
|
if err := os.MkdirAll(systemdDir, 0o755); err != nil {
|
||||||
|
return fmt.Errorf("failed creating systemd uuser dir: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
serviceFile := filepath.Join(systemdDir, "tailscale-systray.service")
|
||||||
|
|
||||||
|
if err := os.WriteFile(serviceFile, output.Bytes(), 0o755); err != nil {
|
||||||
|
return fmt.Errorf("failed writing systemd user service: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Printf("Successfully installed systemd service to: %s\n", serviceFile)
|
||||||
|
fmt.Println("To enable and start the service, run:")
|
||||||
|
fmt.Println(" systemctl --user daemon-reload")
|
||||||
|
fmt.Println(" systemctl --user enable --now tailscale-systray")
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
@ -0,0 +1,801 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
//go:build cgo || !darwin
|
||||||
|
|
||||||
|
// Package systray provides a minimal Tailscale systray application.
|
||||||
|
package systray
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"image"
|
||||||
|
"io"
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
"os/signal"
|
||||||
|
"runtime"
|
||||||
|
"slices"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
"syscall"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"fyne.io/systray"
|
||||||
|
ico "github.com/Kodeworks/golang-image-ico"
|
||||||
|
"github.com/atotto/clipboard"
|
||||||
|
dbus "github.com/godbus/dbus/v5"
|
||||||
|
"github.com/toqueteos/webbrowser"
|
||||||
|
"tailscale.com/client/local"
|
||||||
|
"tailscale.com/ipn"
|
||||||
|
"tailscale.com/ipn/ipnstate"
|
||||||
|
"tailscale.com/tailcfg"
|
||||||
|
"tailscale.com/util/slicesx"
|
||||||
|
"tailscale.com/util/stringsx"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
// newMenuDelay is the amount of time to sleep after creating a new menu,
|
||||||
|
// but before adding items to it. This works around a bug in some dbus implementations.
|
||||||
|
newMenuDelay time.Duration
|
||||||
|
|
||||||
|
// if true, treat all mullvad exit node countries as single-city.
|
||||||
|
// Instead of rendering a submenu with cities, just select the highest-priority peer.
|
||||||
|
hideMullvadCities bool
|
||||||
|
)
|
||||||
|
|
||||||
|
// Run starts the systray menu and blocks until the menu exits.
|
||||||
|
// If client is nil, a default local.Client is used.
|
||||||
|
func (menu *Menu) Run(client *local.Client) {
|
||||||
|
if client == nil {
|
||||||
|
client = &local.Client{}
|
||||||
|
}
|
||||||
|
menu.lc = client
|
||||||
|
menu.updateState()
|
||||||
|
|
||||||
|
// exit cleanly on SIGINT and SIGTERM
|
||||||
|
go func() {
|
||||||
|
interrupt := make(chan os.Signal, 1)
|
||||||
|
signal.Notify(interrupt, syscall.SIGINT, syscall.SIGTERM)
|
||||||
|
select {
|
||||||
|
case <-interrupt:
|
||||||
|
menu.onExit()
|
||||||
|
case <-menu.bgCtx.Done():
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
go menu.lc.IncrementGauge(menu.bgCtx, "systray_running", 1)
|
||||||
|
defer menu.lc.IncrementGauge(menu.bgCtx, "systray_running", -1)
|
||||||
|
|
||||||
|
systray.Run(menu.onReady, menu.onExit)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Menu represents the systray menu, its items, and the current Tailscale state.
|
||||||
|
type Menu struct {
|
||||||
|
mu sync.Mutex // protects the entire Menu
|
||||||
|
|
||||||
|
lc *local.Client
|
||||||
|
status *ipnstate.Status
|
||||||
|
curProfile ipn.LoginProfile
|
||||||
|
allProfiles []ipn.LoginProfile
|
||||||
|
|
||||||
|
// readonly is whether the systray app is running in read-only mode.
|
||||||
|
// This is set if LocalAPI returns a permission error,
|
||||||
|
// typically because the user needs to run `tailscale set --operator=$USER`.
|
||||||
|
readonly bool
|
||||||
|
|
||||||
|
bgCtx context.Context // ctx for background tasks not involving menu item clicks
|
||||||
|
bgCancel context.CancelFunc
|
||||||
|
|
||||||
|
// Top-level menu items
|
||||||
|
connect *systray.MenuItem
|
||||||
|
disconnect *systray.MenuItem
|
||||||
|
self *systray.MenuItem
|
||||||
|
exitNodes *systray.MenuItem
|
||||||
|
more *systray.MenuItem
|
||||||
|
rebuildMenu *systray.MenuItem
|
||||||
|
quit *systray.MenuItem
|
||||||
|
|
||||||
|
rebuildCh chan struct{} // triggers a menu rebuild
|
||||||
|
accountsCh chan ipn.ProfileID
|
||||||
|
exitNodeCh chan tailcfg.StableNodeID // ID of selected exit node
|
||||||
|
|
||||||
|
eventCancel context.CancelFunc // cancel eventLoop
|
||||||
|
|
||||||
|
notificationIcon *os.File // icon used for desktop notifications
|
||||||
|
}
|
||||||
|
|
||||||
|
func (menu *Menu) init() {
|
||||||
|
if menu.bgCtx != nil {
|
||||||
|
// already initialized
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
menu.rebuildCh = make(chan struct{}, 1)
|
||||||
|
menu.accountsCh = make(chan ipn.ProfileID)
|
||||||
|
menu.exitNodeCh = make(chan tailcfg.StableNodeID)
|
||||||
|
|
||||||
|
// dbus wants a file path for notification icons, so copy to a temp file.
|
||||||
|
menu.notificationIcon, _ = os.CreateTemp("", "tailscale-systray.png")
|
||||||
|
io.Copy(menu.notificationIcon, connected.renderWithBorder(3))
|
||||||
|
|
||||||
|
menu.bgCtx, menu.bgCancel = context.WithCancel(context.Background())
|
||||||
|
go menu.watchIPNBus()
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
if runtime.GOOS != "linux" {
|
||||||
|
// so far, these tweaks are only needed on Linux
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
desktop := strings.ToLower(os.Getenv("XDG_CURRENT_DESKTOP"))
|
||||||
|
switch desktop {
|
||||||
|
case "gnome", "ubuntu:gnome":
|
||||||
|
// GNOME expands submenus downward in the main menu, rather than flyouts to the side.
|
||||||
|
// Either as a result of that or another limitation, there seems to be a maximum depth of submenus.
|
||||||
|
// Mullvad countries that have a city submenu are not being rendered, and so can't be selected.
|
||||||
|
// Handle this by simply treating all mullvad countries as single-city and select the best peer.
|
||||||
|
hideMullvadCities = true
|
||||||
|
case "kde":
|
||||||
|
// KDE doesn't need a delay, and actually won't render submenus
|
||||||
|
// if we delay for more than about 400µs.
|
||||||
|
newMenuDelay = 0
|
||||||
|
default:
|
||||||
|
// Add a slight delay to ensure the menu is created before adding items.
|
||||||
|
//
|
||||||
|
// Systray implementations that use libdbusmenu sometimes process messages out of order,
|
||||||
|
// resulting in errors such as:
|
||||||
|
// (waybar:153009): LIBDBUSMENU-GTK-WARNING **: 18:07:11.551: Children but no menu, someone's been naughty with their 'children-display' property: 'submenu'
|
||||||
|
//
|
||||||
|
// See also: https://github.com/fyne-io/systray/issues/12
|
||||||
|
newMenuDelay = 10 * time.Millisecond
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// onReady is called by the systray package when the menu is ready to be built.
|
||||||
|
func (menu *Menu) onReady() {
|
||||||
|
log.Printf("starting")
|
||||||
|
if os.Getuid() == 0 || os.Getuid() != os.Geteuid() || os.Getenv("SUDO_USER") != "" || os.Getenv("DOAS_USER") != "" {
|
||||||
|
fmt.Fprintln(os.Stderr, `
|
||||||
|
It appears that you might be running the systray with sudo/doas.
|
||||||
|
This can lead to issues with D-Bus, and should be avoided.
|
||||||
|
|
||||||
|
The systray application should be run with the same user as your desktop session.
|
||||||
|
This usually means that you should run the application like:
|
||||||
|
|
||||||
|
tailscale systray
|
||||||
|
|
||||||
|
See https://tailscale.com/kb/1597/linux-systray for more information.`)
|
||||||
|
}
|
||||||
|
setAppIcon(disconnected)
|
||||||
|
menu.rebuild()
|
||||||
|
|
||||||
|
menu.mu.Lock()
|
||||||
|
if menu.readonly {
|
||||||
|
fmt.Fprintln(os.Stderr, `
|
||||||
|
No permission to manage Tailscale. Set operator by running:
|
||||||
|
|
||||||
|
sudo tailscale set --operator=$USER
|
||||||
|
|
||||||
|
See https://tailscale.com/s/cli-operator for more information.`)
|
||||||
|
}
|
||||||
|
menu.mu.Unlock()
|
||||||
|
}
|
||||||
|
|
||||||
|
// updateState updates the Menu state from the Tailscale local client.
|
||||||
|
func (menu *Menu) updateState() {
|
||||||
|
menu.mu.Lock()
|
||||||
|
defer menu.mu.Unlock()
|
||||||
|
menu.init()
|
||||||
|
|
||||||
|
menu.readonly = false
|
||||||
|
|
||||||
|
var err error
|
||||||
|
menu.status, err = menu.lc.Status(menu.bgCtx)
|
||||||
|
if err != nil {
|
||||||
|
log.Print(err)
|
||||||
|
}
|
||||||
|
menu.curProfile, menu.allProfiles, err = menu.lc.ProfileStatus(menu.bgCtx)
|
||||||
|
if err != nil {
|
||||||
|
if local.IsAccessDeniedError(err) {
|
||||||
|
menu.readonly = true
|
||||||
|
}
|
||||||
|
log.Print(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// rebuild the systray menu based on the current Tailscale state.
|
||||||
|
//
|
||||||
|
// We currently rebuild the entire menu because it is not easy to update the existing menu.
|
||||||
|
// You cannot iterate over the items in a menu, nor can you remove some items like separators.
|
||||||
|
// So for now we rebuild the whole thing, and can optimize this later if needed.
|
||||||
|
func (menu *Menu) rebuild() {
|
||||||
|
menu.mu.Lock()
|
||||||
|
defer menu.mu.Unlock()
|
||||||
|
menu.init()
|
||||||
|
|
||||||
|
if menu.eventCancel != nil {
|
||||||
|
menu.eventCancel()
|
||||||
|
}
|
||||||
|
ctx := context.Background()
|
||||||
|
ctx, menu.eventCancel = context.WithCancel(ctx)
|
||||||
|
|
||||||
|
systray.ResetMenu()
|
||||||
|
|
||||||
|
if menu.readonly {
|
||||||
|
const readonlyMsg = "No permission to manage Tailscale.\nSee tailscale.com/s/cli-operator"
|
||||||
|
m := systray.AddMenuItem(readonlyMsg, "")
|
||||||
|
onClick(ctx, m, func(_ context.Context) {
|
||||||
|
webbrowser.Open("https://tailscale.com/s/cli-operator")
|
||||||
|
})
|
||||||
|
systray.AddSeparator()
|
||||||
|
}
|
||||||
|
|
||||||
|
menu.connect = systray.AddMenuItem("Connect", "")
|
||||||
|
menu.disconnect = systray.AddMenuItem("Disconnect", "")
|
||||||
|
menu.disconnect.Hide()
|
||||||
|
systray.AddSeparator()
|
||||||
|
|
||||||
|
// delay to prevent race setting icon on first start
|
||||||
|
time.Sleep(newMenuDelay)
|
||||||
|
|
||||||
|
// Set systray menu icon and title.
|
||||||
|
// Also adjust connect/disconnect menu items if needed.
|
||||||
|
var backendState string
|
||||||
|
if menu.status != nil {
|
||||||
|
backendState = menu.status.BackendState
|
||||||
|
}
|
||||||
|
switch backendState {
|
||||||
|
case ipn.Running.String():
|
||||||
|
if menu.status.ExitNodeStatus != nil && !menu.status.ExitNodeStatus.ID.IsZero() {
|
||||||
|
if menu.status.ExitNodeStatus.Online {
|
||||||
|
setTooltip("Using exit node")
|
||||||
|
setAppIcon(exitNodeOnline)
|
||||||
|
} else {
|
||||||
|
setTooltip("Exit node offline")
|
||||||
|
setAppIcon(exitNodeOffline)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
setTooltip(fmt.Sprintf("Connected to %s", menu.status.CurrentTailnet.Name))
|
||||||
|
setAppIcon(connected)
|
||||||
|
}
|
||||||
|
menu.connect.SetTitle("Connected")
|
||||||
|
menu.connect.Disable()
|
||||||
|
menu.disconnect.Show()
|
||||||
|
menu.disconnect.Enable()
|
||||||
|
case ipn.Starting.String():
|
||||||
|
setTooltip("Connecting")
|
||||||
|
setAppIcon(loading)
|
||||||
|
default:
|
||||||
|
setTooltip("Disconnected")
|
||||||
|
setAppIcon(disconnected)
|
||||||
|
}
|
||||||
|
|
||||||
|
if menu.readonly {
|
||||||
|
menu.connect.Disable()
|
||||||
|
menu.disconnect.Disable()
|
||||||
|
}
|
||||||
|
|
||||||
|
account := "Account"
|
||||||
|
if pt := profileTitle(menu.curProfile); pt != "" {
|
||||||
|
account = pt
|
||||||
|
}
|
||||||
|
if !menu.readonly {
|
||||||
|
accounts := systray.AddMenuItem(account, "")
|
||||||
|
setRemoteIcon(accounts, menu.curProfile.UserProfile.ProfilePicURL)
|
||||||
|
time.Sleep(newMenuDelay)
|
||||||
|
for _, profile := range menu.allProfiles {
|
||||||
|
title := profileTitle(profile)
|
||||||
|
var item *systray.MenuItem
|
||||||
|
if profile.ID == menu.curProfile.ID {
|
||||||
|
item = accounts.AddSubMenuItemCheckbox(title, "", true)
|
||||||
|
} else {
|
||||||
|
item = accounts.AddSubMenuItem(title, "")
|
||||||
|
}
|
||||||
|
setRemoteIcon(item, profile.UserProfile.ProfilePicURL)
|
||||||
|
onClick(ctx, item, func(ctx context.Context) {
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
case menu.accountsCh <- profile.ID:
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if menu.status != nil && menu.status.Self != nil && len(menu.status.Self.TailscaleIPs) > 0 {
|
||||||
|
title := fmt.Sprintf("This Device: %s (%s)", menu.status.Self.HostName, menu.status.Self.TailscaleIPs[0])
|
||||||
|
menu.self = systray.AddMenuItem(title, "")
|
||||||
|
} else {
|
||||||
|
menu.self = systray.AddMenuItem("This Device: not connected", "")
|
||||||
|
menu.self.Disable()
|
||||||
|
}
|
||||||
|
systray.AddSeparator()
|
||||||
|
|
||||||
|
if !menu.readonly {
|
||||||
|
menu.rebuildExitNodeMenu(ctx)
|
||||||
|
}
|
||||||
|
|
||||||
|
menu.more = systray.AddMenuItem("More settings", "")
|
||||||
|
if menu.status != nil && menu.status.BackendState == "Running" {
|
||||||
|
// web client is only available if backend is running
|
||||||
|
onClick(ctx, menu.more, func(_ context.Context) {
|
||||||
|
webbrowser.Open("http://100.100.100.100/")
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
menu.more.Disable()
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO(#15528): this menu item shouldn't be necessary at all,
|
||||||
|
// but is at least more discoverable than having users switch profiles or exit nodes.
|
||||||
|
menu.rebuildMenu = systray.AddMenuItem("Rebuild menu", "Fix missing menu items")
|
||||||
|
onClick(ctx, menu.rebuildMenu, func(ctx context.Context) {
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
case menu.rebuildCh <- struct{}{}:
|
||||||
|
}
|
||||||
|
})
|
||||||
|
menu.rebuildMenu.Enable()
|
||||||
|
|
||||||
|
menu.quit = systray.AddMenuItem("Quit", "Quit the app")
|
||||||
|
menu.quit.Enable()
|
||||||
|
|
||||||
|
go menu.eventLoop(ctx)
|
||||||
|
}
|
||||||
|
|
||||||
|
// profileTitle returns the title string for a profile menu item.
|
||||||
|
func profileTitle(profile ipn.LoginProfile) string {
|
||||||
|
title := profile.Name
|
||||||
|
if profile.NetworkProfile.DomainName != "" {
|
||||||
|
if runtime.GOOS == "windows" || runtime.GOOS == "darwin" {
|
||||||
|
// windows and mac don't support multi-line menu
|
||||||
|
title += " (" + profile.NetworkProfile.DisplayNameOrDefault() + ")"
|
||||||
|
} else {
|
||||||
|
title += "\n" + profile.NetworkProfile.DisplayNameOrDefault()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return title
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
cacheMu sync.Mutex
|
||||||
|
httpCache = map[string][]byte{} // URL => response body
|
||||||
|
)
|
||||||
|
|
||||||
|
// setRemoteIcon sets the icon for menu to the specified remote image.
|
||||||
|
// Remote images are fetched as needed and cached.
|
||||||
|
func setRemoteIcon(menu *systray.MenuItem, urlStr string) {
|
||||||
|
if menu == nil || urlStr == "" {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
cacheMu.Lock()
|
||||||
|
b, ok := httpCache[urlStr]
|
||||||
|
if !ok {
|
||||||
|
resp, err := http.Get(urlStr)
|
||||||
|
if err == nil && resp.StatusCode == http.StatusOK {
|
||||||
|
b, _ = io.ReadAll(resp.Body)
|
||||||
|
|
||||||
|
// Convert image to ICO format on Windows
|
||||||
|
if runtime.GOOS == "windows" {
|
||||||
|
im, _, err := image.Decode(bytes.NewReader(b))
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
buf := bytes.NewBuffer(nil)
|
||||||
|
if err := ico.Encode(buf, im); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
b = buf.Bytes()
|
||||||
|
}
|
||||||
|
|
||||||
|
httpCache[urlStr] = b
|
||||||
|
resp.Body.Close()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
cacheMu.Unlock()
|
||||||
|
|
||||||
|
if len(b) > 0 {
|
||||||
|
menu.SetIcon(b)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// setTooltip sets the tooltip text for the systray icon.
|
||||||
|
func setTooltip(text string) {
|
||||||
|
if runtime.GOOS == "darwin" || runtime.GOOS == "windows" {
|
||||||
|
systray.SetTooltip(text)
|
||||||
|
} else {
|
||||||
|
// on Linux, SetTitle actually sets the tooltip
|
||||||
|
systray.SetTitle(text)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// eventLoop is the main event loop for handling click events on menu items
|
||||||
|
// and responding to Tailscale state changes.
|
||||||
|
// This method does not return until ctx.Done is closed.
|
||||||
|
func (menu *Menu) eventLoop(ctx context.Context) {
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
return
|
||||||
|
case <-menu.rebuildCh:
|
||||||
|
menu.updateState()
|
||||||
|
menu.rebuild()
|
||||||
|
case <-menu.connect.ClickedCh:
|
||||||
|
_, err := menu.lc.EditPrefs(ctx, &ipn.MaskedPrefs{
|
||||||
|
Prefs: ipn.Prefs{
|
||||||
|
WantRunning: true,
|
||||||
|
},
|
||||||
|
WantRunningSet: true,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("error connecting: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
case <-menu.disconnect.ClickedCh:
|
||||||
|
_, err := menu.lc.EditPrefs(ctx, &ipn.MaskedPrefs{
|
||||||
|
Prefs: ipn.Prefs{
|
||||||
|
WantRunning: false,
|
||||||
|
},
|
||||||
|
WantRunningSet: true,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("error disconnecting: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
case <-menu.self.ClickedCh:
|
||||||
|
menu.copyTailscaleIP(menu.status.Self)
|
||||||
|
|
||||||
|
case id := <-menu.accountsCh:
|
||||||
|
if err := menu.lc.SwitchProfile(ctx, id); err != nil {
|
||||||
|
log.Printf("error switching to profile ID %v: %v", id, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
case exitNode := <-menu.exitNodeCh:
|
||||||
|
if exitNode.IsZero() {
|
||||||
|
log.Print("disable exit node")
|
||||||
|
if err := menu.lc.SetUseExitNode(ctx, false); err != nil {
|
||||||
|
log.Printf("error disabling exit node: %v", err)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
log.Printf("enable exit node: %v", exitNode)
|
||||||
|
mp := &ipn.MaskedPrefs{
|
||||||
|
Prefs: ipn.Prefs{
|
||||||
|
ExitNodeID: exitNode,
|
||||||
|
},
|
||||||
|
ExitNodeIDSet: true,
|
||||||
|
}
|
||||||
|
if _, err := menu.lc.EditPrefs(ctx, mp); err != nil {
|
||||||
|
log.Printf("error setting exit node: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
case <-menu.quit.ClickedCh:
|
||||||
|
systray.Quit()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// onClick registers a click handler for a menu item.
|
||||||
|
func onClick(ctx context.Context, item *systray.MenuItem, fn func(ctx context.Context)) {
|
||||||
|
go func() {
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
return
|
||||||
|
case <-item.ClickedCh:
|
||||||
|
fn(ctx)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
|
||||||
|
// watchIPNBus subscribes to the tailscale event bus and sends state updates to chState.
|
||||||
|
// This method does not return.
|
||||||
|
func (menu *Menu) watchIPNBus() {
|
||||||
|
for {
|
||||||
|
if err := menu.watchIPNBusInner(); err != nil {
|
||||||
|
log.Println(err)
|
||||||
|
if errors.Is(err, context.Canceled) {
|
||||||
|
// If the context got canceled, we will never be able to
|
||||||
|
// reconnect to IPN bus, so exit the process.
|
||||||
|
log.Fatalf("watchIPNBus: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// If our watch connection breaks, wait a bit before reconnecting. No
|
||||||
|
// reason to spam the logs if e.g. tailscaled is restarting or goes
|
||||||
|
// down.
|
||||||
|
time.Sleep(3 * time.Second)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (menu *Menu) watchIPNBusInner() error {
|
||||||
|
watcher, err := menu.lc.WatchIPNBus(menu.bgCtx, 0)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("watching ipn bus: %w", err)
|
||||||
|
}
|
||||||
|
defer watcher.Close()
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-menu.bgCtx.Done():
|
||||||
|
return nil
|
||||||
|
default:
|
||||||
|
n, err := watcher.Next()
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("ipnbus error: %w", err)
|
||||||
|
}
|
||||||
|
var rebuild bool
|
||||||
|
if n.State != nil {
|
||||||
|
log.Printf("new state: %v", n.State)
|
||||||
|
rebuild = true
|
||||||
|
}
|
||||||
|
if n.Prefs != nil {
|
||||||
|
rebuild = true
|
||||||
|
}
|
||||||
|
if rebuild {
|
||||||
|
menu.rebuildCh <- struct{}{}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// copyTailscaleIP copies the first Tailscale IP of the given device to the clipboard
|
||||||
|
// and sends a notification with the copied value.
|
||||||
|
func (menu *Menu) copyTailscaleIP(device *ipnstate.PeerStatus) {
|
||||||
|
if device == nil || len(device.TailscaleIPs) == 0 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
name := strings.Split(device.DNSName, ".")[0]
|
||||||
|
ip := device.TailscaleIPs[0].String()
|
||||||
|
err := clipboard.WriteAll(ip)
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("clipboard error: %v", err)
|
||||||
|
} else {
|
||||||
|
menu.sendNotification(fmt.Sprintf("Copied Address for %v", name), ip)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// sendNotification sends a desktop notification with the given title and content.
|
||||||
|
func (menu *Menu) sendNotification(title, content string) {
|
||||||
|
conn, err := dbus.SessionBus()
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("dbus: %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
timeout := 3 * time.Second
|
||||||
|
obj := conn.Object("org.freedesktop.Notifications", "/org/freedesktop/Notifications")
|
||||||
|
call := obj.Call("org.freedesktop.Notifications.Notify", 0, "Tailscale", uint32(0),
|
||||||
|
menu.notificationIcon.Name(), title, content, []string{}, map[string]dbus.Variant{}, int32(timeout.Milliseconds()))
|
||||||
|
if call.Err != nil {
|
||||||
|
log.Printf("dbus: %v", call.Err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (menu *Menu) rebuildExitNodeMenu(ctx context.Context) {
|
||||||
|
if menu.status == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
status := menu.status
|
||||||
|
menu.exitNodes = systray.AddMenuItem("Exit Nodes", "")
|
||||||
|
time.Sleep(newMenuDelay)
|
||||||
|
|
||||||
|
// register a click handler for a menu item to set nodeID as the exit node.
|
||||||
|
setExitNodeOnClick := func(item *systray.MenuItem, nodeID tailcfg.StableNodeID) {
|
||||||
|
onClick(ctx, item, func(ctx context.Context) {
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
case menu.exitNodeCh <- nodeID:
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
noExitNodeMenu := menu.exitNodes.AddSubMenuItemCheckbox("None", "", status.ExitNodeStatus == nil)
|
||||||
|
setExitNodeOnClick(noExitNodeMenu, "")
|
||||||
|
|
||||||
|
// Show recommended exit node if available.
|
||||||
|
if status.Self.CapMap.Contains(tailcfg.NodeAttrSuggestExitNodeUI) {
|
||||||
|
sugg, err := menu.lc.SuggestExitNode(ctx)
|
||||||
|
if err == nil {
|
||||||
|
title := "Recommended: "
|
||||||
|
if loc := sugg.Location; loc.Valid() && loc.Country() != "" {
|
||||||
|
flag := countryFlag(loc.CountryCode())
|
||||||
|
title += fmt.Sprintf("%s %s: %s", flag, loc.Country(), loc.City())
|
||||||
|
} else {
|
||||||
|
title += strings.Split(sugg.Name, ".")[0]
|
||||||
|
}
|
||||||
|
menu.exitNodes.AddSeparator()
|
||||||
|
rm := menu.exitNodes.AddSubMenuItemCheckbox(title, "", false)
|
||||||
|
setExitNodeOnClick(rm, sugg.ID)
|
||||||
|
if status.ExitNodeStatus != nil && sugg.ID == status.ExitNodeStatus.ID {
|
||||||
|
rm.Check()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add tailnet exit nodes if present.
|
||||||
|
var tailnetExitNodes []*ipnstate.PeerStatus
|
||||||
|
for _, ps := range status.Peer {
|
||||||
|
if ps.ExitNodeOption && ps.Location == nil {
|
||||||
|
tailnetExitNodes = append(tailnetExitNodes, ps)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if len(tailnetExitNodes) > 0 {
|
||||||
|
menu.exitNodes.AddSeparator()
|
||||||
|
menu.exitNodes.AddSubMenuItem("Tailnet Exit Nodes", "").Disable()
|
||||||
|
for _, ps := range status.Peer {
|
||||||
|
if !ps.ExitNodeOption || ps.Location != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
name := strings.Split(ps.DNSName, ".")[0]
|
||||||
|
if !ps.Online {
|
||||||
|
name += " (offline)"
|
||||||
|
}
|
||||||
|
sm := menu.exitNodes.AddSubMenuItemCheckbox(name, "", false)
|
||||||
|
if !ps.Online {
|
||||||
|
sm.Disable()
|
||||||
|
}
|
||||||
|
if status.ExitNodeStatus != nil && ps.ID == status.ExitNodeStatus.ID {
|
||||||
|
sm.Check()
|
||||||
|
}
|
||||||
|
setExitNodeOnClick(sm, ps.ID)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add mullvad exit nodes if present.
|
||||||
|
var mullvadExitNodes mullvadPeers
|
||||||
|
if status.Self.CapMap.Contains("mullvad") {
|
||||||
|
mullvadExitNodes = newMullvadPeers(status)
|
||||||
|
}
|
||||||
|
if len(mullvadExitNodes.countries) > 0 {
|
||||||
|
menu.exitNodes.AddSeparator()
|
||||||
|
menu.exitNodes.AddSubMenuItem("Location-based Exit Nodes", "").Disable()
|
||||||
|
mullvadMenu := menu.exitNodes.AddSubMenuItemCheckbox("Mullvad VPN", "", false)
|
||||||
|
|
||||||
|
for _, country := range mullvadExitNodes.sortedCountries() {
|
||||||
|
flag := countryFlag(country.code)
|
||||||
|
countryMenu := mullvadMenu.AddSubMenuItemCheckbox(flag+" "+country.name, "", false)
|
||||||
|
|
||||||
|
// single-city country, no submenu
|
||||||
|
if len(country.cities) == 1 || hideMullvadCities {
|
||||||
|
setExitNodeOnClick(countryMenu, country.best.ID)
|
||||||
|
if status.ExitNodeStatus != nil {
|
||||||
|
for _, city := range country.cities {
|
||||||
|
for _, ps := range city.peers {
|
||||||
|
if status.ExitNodeStatus.ID == ps.ID {
|
||||||
|
mullvadMenu.Check()
|
||||||
|
countryMenu.Check()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// multi-city country, build submenu with "best available" option and cities.
|
||||||
|
time.Sleep(newMenuDelay)
|
||||||
|
bm := countryMenu.AddSubMenuItemCheckbox("Best Available", "", false)
|
||||||
|
setExitNodeOnClick(bm, country.best.ID)
|
||||||
|
countryMenu.AddSeparator()
|
||||||
|
|
||||||
|
for _, city := range country.sortedCities() {
|
||||||
|
cityMenu := countryMenu.AddSubMenuItemCheckbox(city.name, "", false)
|
||||||
|
setExitNodeOnClick(cityMenu, city.best.ID)
|
||||||
|
if status.ExitNodeStatus != nil {
|
||||||
|
for _, ps := range city.peers {
|
||||||
|
if status.ExitNodeStatus.ID == ps.ID {
|
||||||
|
mullvadMenu.Check()
|
||||||
|
countryMenu.Check()
|
||||||
|
cityMenu.Check()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: "Allow Local Network Access" and "Run Exit Node" menu items
|
||||||
|
}
|
||||||
|
|
||||||
|
// mullvadPeers contains all mullvad peer nodes, sorted by country and city.
|
||||||
|
type mullvadPeers struct {
|
||||||
|
countries map[string]*mvCountry // country code (uppercase) => country
|
||||||
|
}
|
||||||
|
|
||||||
|
// sortedCountries returns countries containing mullvad nodes, sorted by name.
|
||||||
|
func (mp mullvadPeers) sortedCountries() []*mvCountry {
|
||||||
|
countries := slicesx.MapValues(mp.countries)
|
||||||
|
slices.SortFunc(countries, func(a, b *mvCountry) int {
|
||||||
|
return stringsx.CompareFold(a.name, b.name)
|
||||||
|
})
|
||||||
|
return countries
|
||||||
|
}
|
||||||
|
|
||||||
|
type mvCountry struct {
|
||||||
|
code string
|
||||||
|
name string
|
||||||
|
best *ipnstate.PeerStatus // highest priority peer in the country
|
||||||
|
cities map[string]*mvCity // city code => city
|
||||||
|
}
|
||||||
|
|
||||||
|
// sortedCities returns cities containing mullvad nodes, sorted by name.
|
||||||
|
func (mc *mvCountry) sortedCities() []*mvCity {
|
||||||
|
cities := slicesx.MapValues(mc.cities)
|
||||||
|
slices.SortFunc(cities, func(a, b *mvCity) int {
|
||||||
|
return stringsx.CompareFold(a.name, b.name)
|
||||||
|
})
|
||||||
|
return cities
|
||||||
|
}
|
||||||
|
|
||||||
|
// countryFlag takes a 2-character ASCII string and returns the corresponding emoji flag.
|
||||||
|
// It returns the empty string on error.
|
||||||
|
func countryFlag(code string) string {
|
||||||
|
if len(code) != 2 {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
runes := make([]rune, 0, 2)
|
||||||
|
for i := range 2 {
|
||||||
|
b := code[i] | 32 // lowercase
|
||||||
|
if b < 'a' || b > 'z' {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
// https://en.wikipedia.org/wiki/Regional_indicator_symbol
|
||||||
|
runes = append(runes, 0x1F1E6+rune(b-'a'))
|
||||||
|
}
|
||||||
|
return string(runes)
|
||||||
|
}
|
||||||
|
|
||||||
|
type mvCity struct {
|
||||||
|
name string
|
||||||
|
best *ipnstate.PeerStatus // highest priority peer in the city
|
||||||
|
peers []*ipnstate.PeerStatus
|
||||||
|
}
|
||||||
|
|
||||||
|
func newMullvadPeers(status *ipnstate.Status) mullvadPeers {
|
||||||
|
countries := make(map[string]*mvCountry)
|
||||||
|
for _, ps := range status.Peer {
|
||||||
|
if !ps.ExitNodeOption || ps.Location == nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
loc := ps.Location
|
||||||
|
country, ok := countries[loc.CountryCode]
|
||||||
|
if !ok {
|
||||||
|
country = &mvCountry{
|
||||||
|
code: loc.CountryCode,
|
||||||
|
name: loc.Country,
|
||||||
|
cities: make(map[string]*mvCity),
|
||||||
|
}
|
||||||
|
countries[loc.CountryCode] = country
|
||||||
|
}
|
||||||
|
city, ok := countries[loc.CountryCode].cities[loc.CityCode]
|
||||||
|
if !ok {
|
||||||
|
city = &mvCity{
|
||||||
|
name: loc.City,
|
||||||
|
}
|
||||||
|
countries[loc.CountryCode].cities[loc.CityCode] = city
|
||||||
|
}
|
||||||
|
city.peers = append(city.peers, ps)
|
||||||
|
if city.best == nil || ps.Location.Priority > city.best.Location.Priority {
|
||||||
|
city.best = ps
|
||||||
|
}
|
||||||
|
if country.best == nil || ps.Location.Priority > country.best.Location.Priority {
|
||||||
|
country.best = ps
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return mullvadPeers{countries}
|
||||||
|
}
|
||||||
|
|
||||||
|
// onExit is called by the systray package when the menu is exiting.
|
||||||
|
func (menu *Menu) onExit() {
|
||||||
|
log.Printf("exiting")
|
||||||
|
if menu.bgCancel != nil {
|
||||||
|
menu.bgCancel()
|
||||||
|
}
|
||||||
|
if menu.eventCancel != nil {
|
||||||
|
menu.eventCancel()
|
||||||
|
}
|
||||||
|
|
||||||
|
os.Remove(menu.notificationIcon.Name())
|
||||||
|
}
|
||||||
@ -0,0 +1,10 @@
|
|||||||
|
[Unit]
|
||||||
|
Description=Tailscale System Tray
|
||||||
|
After=systemd.service
|
||||||
|
|
||||||
|
[Service]
|
||||||
|
Type=simple
|
||||||
|
ExecStart=/usr/bin/tailscale systray
|
||||||
|
|
||||||
|
[Install]
|
||||||
|
WantedBy=default.target
|
||||||
@ -0,0 +1,34 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
//go:build !js && !ts_omit_acme
|
||||||
|
|
||||||
|
package tailscale
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"crypto/tls"
|
||||||
|
|
||||||
|
"tailscale.com/client/local"
|
||||||
|
)
|
||||||
|
|
||||||
|
// GetCertificate is an alias for [tailscale.com/client/local.GetCertificate].
|
||||||
|
//
|
||||||
|
// Deprecated: import [tailscale.com/client/local] instead and use [local.Client.GetCertificate].
|
||||||
|
func GetCertificate(hi *tls.ClientHelloInfo) (*tls.Certificate, error) {
|
||||||
|
return local.GetCertificate(hi)
|
||||||
|
}
|
||||||
|
|
||||||
|
// CertPair is an alias for [tailscale.com/client/local.CertPair].
|
||||||
|
//
|
||||||
|
// Deprecated: import [tailscale.com/client/local] instead and use [local.Client.CertPair].
|
||||||
|
func CertPair(ctx context.Context, domain string) (certPEM, keyPEM []byte, err error) {
|
||||||
|
return local.CertPair(ctx, domain)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ExpandSNIName is an alias for [tailscale.com/client/local.ExpandSNIName].
|
||||||
|
//
|
||||||
|
// Deprecated: import [tailscale.com/client/local] instead and use [local.Client.ExpandSNIName].
|
||||||
|
func ExpandSNIName(ctx context.Context, name string) (fqdn string, ok bool) {
|
||||||
|
return local.ExpandSNIName(ctx, name)
|
||||||
|
}
|
||||||
@ -0,0 +1,79 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
package tailscale
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
|
||||||
|
"tailscale.com/client/local"
|
||||||
|
"tailscale.com/client/tailscale/apitype"
|
||||||
|
"tailscale.com/ipn/ipnstate"
|
||||||
|
)
|
||||||
|
|
||||||
|
// ErrPeerNotFound is an alias for [tailscale.com/client/local.ErrPeerNotFound].
|
||||||
|
//
|
||||||
|
// Deprecated: import [tailscale.com/client/local] instead.
|
||||||
|
var ErrPeerNotFound = local.ErrPeerNotFound
|
||||||
|
|
||||||
|
// LocalClient is an alias for [tailscale.com/client/local.Client].
|
||||||
|
//
|
||||||
|
// Deprecated: import [tailscale.com/client/local] instead.
|
||||||
|
type LocalClient = local.Client
|
||||||
|
|
||||||
|
// IPNBusWatcher is an alias for [tailscale.com/client/local.IPNBusWatcher].
|
||||||
|
//
|
||||||
|
// Deprecated: import [tailscale.com/client/local] instead.
|
||||||
|
type IPNBusWatcher = local.IPNBusWatcher
|
||||||
|
|
||||||
|
// BugReportOpts is an alias for [tailscale.com/client/local.BugReportOpts].
|
||||||
|
//
|
||||||
|
// Deprecated: import [tailscale.com/client/local] instead.
|
||||||
|
type BugReportOpts = local.BugReportOpts
|
||||||
|
|
||||||
|
// PingOpts is an alias for [tailscale.com/client/local.PingOpts].
|
||||||
|
//
|
||||||
|
// Deprecated: import [tailscale.com/client/local] instead.
|
||||||
|
type PingOpts = local.PingOpts
|
||||||
|
|
||||||
|
// SetVersionMismatchHandler is an alias for [tailscale.com/client/local.SetVersionMismatchHandler].
|
||||||
|
//
|
||||||
|
// Deprecated: import [tailscale.com/client/local] instead.
|
||||||
|
func SetVersionMismatchHandler(f func(clientVer, serverVer string)) {
|
||||||
|
local.SetVersionMismatchHandler(f)
|
||||||
|
}
|
||||||
|
|
||||||
|
// IsAccessDeniedError is an alias for [tailscale.com/client/local.IsAccessDeniedError].
|
||||||
|
//
|
||||||
|
// Deprecated: import [tailscale.com/client/local] instead.
|
||||||
|
func IsAccessDeniedError(err error) bool {
|
||||||
|
return local.IsAccessDeniedError(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// IsPreconditionsFailedError is an alias for [tailscale.com/client/local.IsPreconditionsFailedError].
|
||||||
|
//
|
||||||
|
// Deprecated: import [tailscale.com/client/local] instead.
|
||||||
|
func IsPreconditionsFailedError(err error) bool {
|
||||||
|
return local.IsPreconditionsFailedError(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// WhoIs is an alias for [tailscale.com/client/local.WhoIs].
|
||||||
|
//
|
||||||
|
// Deprecated: import [tailscale.com/client/local] instead and use [local.Client.WhoIs].
|
||||||
|
func WhoIs(ctx context.Context, remoteAddr string) (*apitype.WhoIsResponse, error) {
|
||||||
|
return local.WhoIs(ctx, remoteAddr)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Status is an alias for [tailscale.com/client/local.Status].
|
||||||
|
//
|
||||||
|
// Deprecated: import [tailscale.com/client/local] instead.
|
||||||
|
func Status(ctx context.Context) (*ipnstate.Status, error) {
|
||||||
|
return local.Status(ctx)
|
||||||
|
}
|
||||||
|
|
||||||
|
// StatusWithoutPeers is an alias for [tailscale.com/client/local.StatusWithoutPeers].
|
||||||
|
//
|
||||||
|
// Deprecated: import [tailscale.com/client/local] instead.
|
||||||
|
func StatusWithoutPeers(ctx context.Context) (*ipnstate.Status, error) {
|
||||||
|
return local.StatusWithoutPeers(ctx)
|
||||||
|
}
|
||||||
@ -0,0 +1,86 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
package tailscale
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/url"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestClientBuildURL(t *testing.T) {
|
||||||
|
c := Client{BaseURL: "http://127.0.0.1:1234"}
|
||||||
|
for _, tt := range []struct {
|
||||||
|
desc string
|
||||||
|
elements []any
|
||||||
|
want string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
desc: "single-element",
|
||||||
|
elements: []any{"devices"},
|
||||||
|
want: "http://127.0.0.1:1234/api/v2/devices",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
desc: "multiple-elements",
|
||||||
|
elements: []any{"tailnet", "example.com"},
|
||||||
|
want: "http://127.0.0.1:1234/api/v2/tailnet/example.com",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
desc: "escape-element",
|
||||||
|
elements: []any{"tailnet", "example dot com?foo=bar"},
|
||||||
|
want: `http://127.0.0.1:1234/api/v2/tailnet/example%20dot%20com%3Ffoo=bar`,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
desc: "url.Values",
|
||||||
|
elements: []any{"tailnet", "example.com", "acl", url.Values{"details": {"1"}}},
|
||||||
|
want: `http://127.0.0.1:1234/api/v2/tailnet/example.com/acl?details=1`,
|
||||||
|
},
|
||||||
|
} {
|
||||||
|
t.Run(tt.desc, func(t *testing.T) {
|
||||||
|
got := c.BuildURL(tt.elements...)
|
||||||
|
if got != tt.want {
|
||||||
|
t.Errorf("got %q, want %q", got, tt.want)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestClientBuildTailnetURL(t *testing.T) {
|
||||||
|
c := Client{
|
||||||
|
BaseURL: "http://127.0.0.1:1234",
|
||||||
|
tailnet: "example.com",
|
||||||
|
}
|
||||||
|
for _, tt := range []struct {
|
||||||
|
desc string
|
||||||
|
elements []any
|
||||||
|
want string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
desc: "single-element",
|
||||||
|
elements: []any{"devices"},
|
||||||
|
want: "http://127.0.0.1:1234/api/v2/tailnet/example.com/devices",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
desc: "multiple-elements",
|
||||||
|
elements: []any{"devices", 123},
|
||||||
|
want: "http://127.0.0.1:1234/api/v2/tailnet/example.com/devices/123",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
desc: "escape-element",
|
||||||
|
elements: []any{"foo bar?baz=qux"},
|
||||||
|
want: `http://127.0.0.1:1234/api/v2/tailnet/example.com/foo%20bar%3Fbaz=qux`,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
desc: "url.Values",
|
||||||
|
elements: []any{"acl", url.Values{"details": {"1"}}},
|
||||||
|
want: `http://127.0.0.1:1234/api/v2/tailnet/example.com/acl?details=1`,
|
||||||
|
},
|
||||||
|
} {
|
||||||
|
t.Run(tt.desc, func(t *testing.T) {
|
||||||
|
got := c.BuildTailnetURL(tt.elements...)
|
||||||
|
if got != tt.want {
|
||||||
|
t.Errorf("got %q, want %q", got, tt.want)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,20 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
//go:build (linux && !android) || windows
|
||||||
|
|
||||||
|
package clientupdate
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
|
||||||
|
"tailscale.com/clientupdate/distsign"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (up *Updater) downloadURLToFile(pathSrc, fileDst string) (ret error) {
|
||||||
|
c, err := distsign.NewClient(up.Logf, up.PkgsAddr)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return c.Download(context.Background(), pathSrc, fileDst)
|
||||||
|
}
|
||||||
@ -0,0 +1,10 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
//go:build !((linux && !android) || windows)
|
||||||
|
|
||||||
|
package clientupdate
|
||||||
|
|
||||||
|
func (up *Updater) downloadURLToFile(pathSrc, fileDst string) (ret error) {
|
||||||
|
panic("unreachable")
|
||||||
|
}
|
||||||
@ -0,0 +1,10 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
//go:build !windows
|
||||||
|
|
||||||
|
package clientupdate
|
||||||
|
|
||||||
|
func (up *Updater) updateWindows() error {
|
||||||
|
panic("unreachable")
|
||||||
|
}
|
||||||
@ -0,0 +1,131 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
// checkmetrics validates that all metrics in the tailscale client-metrics
|
||||||
|
// are documented in a given path or URL.
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"flag"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
"net/http/httptest"
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"tailscale.com/ipn/store/mem"
|
||||||
|
"tailscale.com/tsnet"
|
||||||
|
"tailscale.com/tstest/integration/testcontrol"
|
||||||
|
"tailscale.com/util/httpm"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
kbPath = flag.String("kb-path", "", "filepath to the client-metrics knowledge base")
|
||||||
|
kbUrl = flag.String("kb-url", "", "URL to the client-metrics knowledge base page")
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
flag.Parse()
|
||||||
|
if *kbPath == "" && *kbUrl == "" {
|
||||||
|
log.Fatalf("either -kb-path or -kb-url must be set")
|
||||||
|
}
|
||||||
|
|
||||||
|
var control testcontrol.Server
|
||||||
|
ts := httptest.NewServer(&control)
|
||||||
|
defer ts.Close()
|
||||||
|
|
||||||
|
td, err := os.MkdirTemp("", "testcontrol")
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
defer os.RemoveAll(td)
|
||||||
|
|
||||||
|
// tsnet is used not used as a Tailscale client, but as a way to
|
||||||
|
// boot up Tailscale, have all the metrics registered, and then
|
||||||
|
// verifiy that all the metrics are documented.
|
||||||
|
tsn := &tsnet.Server{
|
||||||
|
Dir: td,
|
||||||
|
Store: new(mem.Store),
|
||||||
|
UserLogf: log.Printf,
|
||||||
|
Ephemeral: true,
|
||||||
|
ControlURL: ts.URL,
|
||||||
|
}
|
||||||
|
if err := tsn.Start(); err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
defer tsn.Close()
|
||||||
|
|
||||||
|
log.Printf("checking that all metrics are documented, looking for: %s", tsn.Sys().UserMetricsRegistry().MetricNames())
|
||||||
|
|
||||||
|
if *kbPath != "" {
|
||||||
|
kb, err := readKB(*kbPath)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("reading kb: %v", err)
|
||||||
|
}
|
||||||
|
missing := undocumentedMetrics(kb, tsn.Sys().UserMetricsRegistry().MetricNames())
|
||||||
|
|
||||||
|
if len(missing) > 0 {
|
||||||
|
log.Fatalf("found undocumented metrics in %q: %v", *kbPath, missing)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if *kbUrl != "" {
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
kb, err := getKB(ctx, *kbUrl)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("getting kb: %v", err)
|
||||||
|
}
|
||||||
|
missing := undocumentedMetrics(kb, tsn.Sys().UserMetricsRegistry().MetricNames())
|
||||||
|
|
||||||
|
if len(missing) > 0 {
|
||||||
|
log.Fatalf("found undocumented metrics in %q: %v", *kbUrl, missing)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func readKB(path string) (string, error) {
|
||||||
|
b, err := os.ReadFile(path)
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("reading file: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return string(b), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func getKB(ctx context.Context, url string) (string, error) {
|
||||||
|
req, err := http.NewRequestWithContext(ctx, httpm.GET, url, nil)
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("creating request: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
resp, err := http.DefaultClient.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("getting kb page: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if resp.StatusCode != http.StatusOK {
|
||||||
|
return "", fmt.Errorf("unexpected status code: %d", resp.StatusCode)
|
||||||
|
}
|
||||||
|
|
||||||
|
b, err := io.ReadAll(resp.Body)
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("reading body: %w", err)
|
||||||
|
}
|
||||||
|
return string(b), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func undocumentedMetrics(b string, metrics []string) []string {
|
||||||
|
var missing []string
|
||||||
|
for _, metric := range metrics {
|
||||||
|
if !strings.Contains(b, metric) {
|
||||||
|
missing = append(missing, metric)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return missing
|
||||||
|
}
|
||||||
@ -1,51 +0,0 @@
|
|||||||
// Copyright (c) Tailscale Inc & AUTHORS
|
|
||||||
// SPDX-License-Identifier: BSD-3-Clause
|
|
||||||
|
|
||||||
//go:build linux
|
|
||||||
|
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"log"
|
|
||||||
"net"
|
|
||||||
"net/http"
|
|
||||||
"sync"
|
|
||||||
)
|
|
||||||
|
|
||||||
// healthz is a simple health check server, if enabled it returns 200 OK if
|
|
||||||
// this tailscale node currently has at least one tailnet IP address else
|
|
||||||
// returns 503.
|
|
||||||
type healthz struct {
|
|
||||||
sync.Mutex
|
|
||||||
hasAddrs bool
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h *healthz) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
||||||
h.Lock()
|
|
||||||
defer h.Unlock()
|
|
||||||
if h.hasAddrs {
|
|
||||||
w.Write([]byte("ok"))
|
|
||||||
} else {
|
|
||||||
http.Error(w, "node currently has no tailscale IPs", http.StatusInternalServerError)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// runHealthz runs a simple HTTP health endpoint on /healthz, listening on the
|
|
||||||
// provided address. A containerized tailscale instance is considered healthy if
|
|
||||||
// it has at least one tailnet IP address.
|
|
||||||
func runHealthz(addr string, h *healthz) {
|
|
||||||
lis, err := net.Listen("tcp", addr)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatalf("error listening on the provided health endpoint address %q: %v", addr, err)
|
|
||||||
}
|
|
||||||
mux := http.NewServeMux()
|
|
||||||
mux.Handle("/healthz", h)
|
|
||||||
log.Printf("Running healthcheck endpoint at %s/healthz", addr)
|
|
||||||
hs := &http.Server{Handler: mux}
|
|
||||||
|
|
||||||
go func() {
|
|
||||||
if err := hs.Serve(lis); err != nil {
|
|
||||||
log.Fatalf("failed running health endpoint: %v", err)
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
}
|
|
||||||
@ -0,0 +1,331 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
//go:build linux
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"net/netip"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"reflect"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/fsnotify/fsnotify"
|
||||||
|
"tailscale.com/kube/ingressservices"
|
||||||
|
"tailscale.com/kube/kubeclient"
|
||||||
|
"tailscale.com/util/linuxfw"
|
||||||
|
"tailscale.com/util/mak"
|
||||||
|
)
|
||||||
|
|
||||||
|
// ingressProxy corresponds to a Kubernetes Operator's network layer ingress
|
||||||
|
// proxy. It configures firewall rules (iptables or nftables) to proxy tailnet
|
||||||
|
// traffic to Kubernetes Services. Currently this is only used for network
|
||||||
|
// layer proxies in HA mode.
|
||||||
|
type ingressProxy struct {
|
||||||
|
cfgPath string // path to ingress configfile.
|
||||||
|
|
||||||
|
// nfr is the netfilter runner used to configure firewall rules.
|
||||||
|
// This is going to be either iptables or nftables based runner.
|
||||||
|
// Never nil.
|
||||||
|
nfr linuxfw.NetfilterRunner
|
||||||
|
|
||||||
|
kc kubeclient.Client // never nil
|
||||||
|
stateSecret string // Secret that holds Tailscale state
|
||||||
|
|
||||||
|
// Pod's IP addresses are used as an identifier of this particular Pod.
|
||||||
|
podIPv4 string // empty if Pod does not have IPv4 address
|
||||||
|
podIPv6 string // empty if Pod does not have IPv6 address
|
||||||
|
}
|
||||||
|
|
||||||
|
// run starts the ingress proxy and ensures that firewall rules are set on start
|
||||||
|
// and refreshed as ingress config changes.
|
||||||
|
func (p *ingressProxy) run(ctx context.Context, opts ingressProxyOpts) error {
|
||||||
|
log.Printf("starting ingress proxy...")
|
||||||
|
p.configure(opts)
|
||||||
|
var tickChan <-chan time.Time
|
||||||
|
var eventChan <-chan fsnotify.Event
|
||||||
|
if w, err := fsnotify.NewWatcher(); err != nil {
|
||||||
|
log.Printf("failed to create fsnotify watcher, timer-only mode: %v", err)
|
||||||
|
ticker := time.NewTicker(5 * time.Second)
|
||||||
|
defer ticker.Stop()
|
||||||
|
tickChan = ticker.C
|
||||||
|
} else {
|
||||||
|
defer w.Close()
|
||||||
|
dir := filepath.Dir(p.cfgPath)
|
||||||
|
if err := w.Add(dir); err != nil {
|
||||||
|
return fmt.Errorf("failed to add fsnotify watch for %v: %w", dir, err)
|
||||||
|
}
|
||||||
|
eventChan = w.Events
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := p.sync(ctx); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
return nil
|
||||||
|
case <-tickChan:
|
||||||
|
log.Printf("periodic sync, ensuring firewall config is up to date...")
|
||||||
|
case <-eventChan:
|
||||||
|
log.Printf("config file change detected, ensuring firewall config is up to date...")
|
||||||
|
}
|
||||||
|
if err := p.sync(ctx); err != nil {
|
||||||
|
return fmt.Errorf("error syncing ingress service config: %w", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// sync reconciles proxy's firewall rules (iptables or nftables) on ingress config changes:
|
||||||
|
// - ensures that new firewall rules are added
|
||||||
|
// - ensures that old firewall rules are deleted
|
||||||
|
// - updates ingress proxy's status in the state Secret
|
||||||
|
func (p *ingressProxy) sync(ctx context.Context) error {
|
||||||
|
// 1. Get the desired firewall configuration
|
||||||
|
cfgs, err := p.getConfigs()
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("ingress proxy: error retrieving configs: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 2. Get the recorded firewall status
|
||||||
|
status, err := p.getStatus(ctx)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("ingress proxy: error retrieving current status: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 3. Ensure that firewall configuration is up to date
|
||||||
|
if err := p.syncIngressConfigs(cfgs, status); err != nil {
|
||||||
|
return fmt.Errorf("ingress proxy: error syncing configs: %w", err)
|
||||||
|
}
|
||||||
|
var existingConfigs *ingressservices.Configs
|
||||||
|
if status != nil {
|
||||||
|
existingConfigs = &status.Configs
|
||||||
|
}
|
||||||
|
|
||||||
|
// 4. Update the recorded firewall status
|
||||||
|
if !(ingressServicesStatusIsEqual(cfgs, existingConfigs) && p.isCurrentStatus(status)) {
|
||||||
|
if err := p.recordStatus(ctx, cfgs); err != nil {
|
||||||
|
return fmt.Errorf("ingress proxy: error setting status: %w", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// getConfigs returns the desired ingress service configuration from the mounted
|
||||||
|
// configfile.
|
||||||
|
func (p *ingressProxy) getConfigs() (*ingressservices.Configs, error) {
|
||||||
|
j, err := os.ReadFile(p.cfgPath)
|
||||||
|
if os.IsNotExist(err) {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if len(j) == 0 || string(j) == "" {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
cfg := &ingressservices.Configs{}
|
||||||
|
if err := json.Unmarshal(j, &cfg); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return cfg, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// getStatus gets the recorded status of the configured firewall. The status is
|
||||||
|
// stored in the proxy's state Secret. Note that the recorded status might not
|
||||||
|
// be the current status of the firewall if it belongs to a previous Pod- we
|
||||||
|
// take that into account further down the line when determining if the desired
|
||||||
|
// rules are actually present.
|
||||||
|
func (p *ingressProxy) getStatus(ctx context.Context) (*ingressservices.Status, error) {
|
||||||
|
secret, err := p.kc.GetSecret(ctx, p.stateSecret)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("error retrieving state Secret: %w", err)
|
||||||
|
}
|
||||||
|
status := &ingressservices.Status{}
|
||||||
|
raw, ok := secret.Data[ingressservices.IngressConfigKey]
|
||||||
|
if !ok {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
if err := json.Unmarshal([]byte(raw), status); err != nil {
|
||||||
|
return nil, fmt.Errorf("error unmarshalling previous config: %w", err)
|
||||||
|
}
|
||||||
|
return status, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// syncIngressConfigs takes the desired firewall configuration and the recorded
|
||||||
|
// status and ensures that any missing rules are added and no longer needed
|
||||||
|
// rules are deleted.
|
||||||
|
func (p *ingressProxy) syncIngressConfigs(cfgs *ingressservices.Configs, status *ingressservices.Status) error {
|
||||||
|
rulesToAdd := p.getRulesToAdd(cfgs, status)
|
||||||
|
rulesToDelete := p.getRulesToDelete(cfgs, status)
|
||||||
|
|
||||||
|
if err := ensureIngressRulesDeleted(rulesToDelete, p.nfr); err != nil {
|
||||||
|
return fmt.Errorf("error deleting ingress rules: %w", err)
|
||||||
|
}
|
||||||
|
if err := ensureIngressRulesAdded(rulesToAdd, p.nfr); err != nil {
|
||||||
|
return fmt.Errorf("error adding ingress rules: %w", err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// recordStatus writes the configured firewall status to the proxy's state
|
||||||
|
// Secret. This allows the Kubernetes Operator to determine whether this proxy
|
||||||
|
// Pod has setup firewall rules to route traffic for an ingress service.
|
||||||
|
func (p *ingressProxy) recordStatus(ctx context.Context, newCfg *ingressservices.Configs) error {
|
||||||
|
status := &ingressservices.Status{}
|
||||||
|
if newCfg != nil {
|
||||||
|
status.Configs = *newCfg
|
||||||
|
}
|
||||||
|
// Pod IPs are used to determine if recorded status applies to THIS proxy Pod.
|
||||||
|
status.PodIPv4 = p.podIPv4
|
||||||
|
status.PodIPv6 = p.podIPv6
|
||||||
|
secret, err := p.kc.GetSecret(ctx, p.stateSecret)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("error retrieving state Secret: %w", err)
|
||||||
|
}
|
||||||
|
bs, err := json.Marshal(status)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("error marshalling status: %w", err)
|
||||||
|
}
|
||||||
|
secret.Data[ingressservices.IngressConfigKey] = bs
|
||||||
|
patch := kubeclient.JSONPatch{
|
||||||
|
Op: "replace",
|
||||||
|
Path: fmt.Sprintf("/data/%s", ingressservices.IngressConfigKey),
|
||||||
|
Value: bs,
|
||||||
|
}
|
||||||
|
if err := p.kc.JSONPatchResource(ctx, p.stateSecret, kubeclient.TypeSecrets, []kubeclient.JSONPatch{patch}); err != nil {
|
||||||
|
return fmt.Errorf("error patching state Secret: %w", err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// getRulesToAdd takes the desired firewall configuration and the recorded
|
||||||
|
// firewall status and returns a map of missing Tailscale Services and rules.
|
||||||
|
func (p *ingressProxy) getRulesToAdd(cfgs *ingressservices.Configs, status *ingressservices.Status) map[string]ingressservices.Config {
|
||||||
|
if cfgs == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
var rulesToAdd map[string]ingressservices.Config
|
||||||
|
for tsSvc, wantsCfg := range *cfgs {
|
||||||
|
if status == nil || !p.isCurrentStatus(status) {
|
||||||
|
mak.Set(&rulesToAdd, tsSvc, wantsCfg)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
gotCfg := status.Configs.GetConfig(tsSvc)
|
||||||
|
if gotCfg == nil || !reflect.DeepEqual(wantsCfg, *gotCfg) {
|
||||||
|
mak.Set(&rulesToAdd, tsSvc, wantsCfg)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return rulesToAdd
|
||||||
|
}
|
||||||
|
|
||||||
|
// getRulesToDelete takes the desired firewall configuration and the recorded
|
||||||
|
// status and returns a map of Tailscale Services and rules that need to be deleted.
|
||||||
|
func (p *ingressProxy) getRulesToDelete(cfgs *ingressservices.Configs, status *ingressservices.Status) map[string]ingressservices.Config {
|
||||||
|
if status == nil || !p.isCurrentStatus(status) {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
var rulesToDelete map[string]ingressservices.Config
|
||||||
|
for tsSvc, gotCfg := range status.Configs {
|
||||||
|
if cfgs == nil {
|
||||||
|
mak.Set(&rulesToDelete, tsSvc, gotCfg)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
wantsCfg := cfgs.GetConfig(tsSvc)
|
||||||
|
if wantsCfg != nil && reflect.DeepEqual(*wantsCfg, gotCfg) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
mak.Set(&rulesToDelete, tsSvc, gotCfg)
|
||||||
|
}
|
||||||
|
return rulesToDelete
|
||||||
|
}
|
||||||
|
|
||||||
|
// ensureIngressRulesAdded takes a map of Tailscale Services and rules and ensures that the firewall rules are added.
|
||||||
|
func ensureIngressRulesAdded(cfgs map[string]ingressservices.Config, nfr linuxfw.NetfilterRunner) error {
|
||||||
|
for serviceName, cfg := range cfgs {
|
||||||
|
if cfg.IPv4Mapping != nil {
|
||||||
|
if err := addDNATRuleForSvc(nfr, serviceName, cfg.IPv4Mapping.TailscaleServiceIP, cfg.IPv4Mapping.ClusterIP); err != nil {
|
||||||
|
return fmt.Errorf("error adding ingress rule for %s: %w", serviceName, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if cfg.IPv6Mapping != nil {
|
||||||
|
if err := addDNATRuleForSvc(nfr, serviceName, cfg.IPv6Mapping.TailscaleServiceIP, cfg.IPv6Mapping.ClusterIP); err != nil {
|
||||||
|
return fmt.Errorf("error adding ingress rule for %s: %w", serviceName, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func addDNATRuleForSvc(nfr linuxfw.NetfilterRunner, serviceName string, tsIP, clusterIP netip.Addr) error {
|
||||||
|
log.Printf("adding DNAT rule for Tailscale Service %s with IP %s to Kubernetes Service IP %s", serviceName, tsIP, clusterIP)
|
||||||
|
return nfr.EnsureDNATRuleForSvc(serviceName, tsIP, clusterIP)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ensureIngressRulesDeleted takes a map of Tailscale Services and rules and ensures that the firewall rules are deleted.
|
||||||
|
func ensureIngressRulesDeleted(cfgs map[string]ingressservices.Config, nfr linuxfw.NetfilterRunner) error {
|
||||||
|
for serviceName, cfg := range cfgs {
|
||||||
|
if cfg.IPv4Mapping != nil {
|
||||||
|
if err := deleteDNATRuleForSvc(nfr, serviceName, cfg.IPv4Mapping.TailscaleServiceIP, cfg.IPv4Mapping.ClusterIP); err != nil {
|
||||||
|
return fmt.Errorf("error deleting ingress rule for %s: %w", serviceName, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if cfg.IPv6Mapping != nil {
|
||||||
|
if err := deleteDNATRuleForSvc(nfr, serviceName, cfg.IPv6Mapping.TailscaleServiceIP, cfg.IPv6Mapping.ClusterIP); err != nil {
|
||||||
|
return fmt.Errorf("error deleting ingress rule for %s: %w", serviceName, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func deleteDNATRuleForSvc(nfr linuxfw.NetfilterRunner, serviceName string, tsIP, clusterIP netip.Addr) error {
|
||||||
|
log.Printf("deleting DNAT rule for Tailscale Service %s with IP %s to Kubernetes Service IP %s", serviceName, tsIP, clusterIP)
|
||||||
|
return nfr.DeleteDNATRuleForSvc(serviceName, tsIP, clusterIP)
|
||||||
|
}
|
||||||
|
|
||||||
|
// isCurrentStatus returns true if the status of an ingress proxy as read from
|
||||||
|
// the proxy's state Secret is the status of the current proxy Pod. We use
|
||||||
|
// Pod's IP addresses to determine that the status is for this Pod.
|
||||||
|
func (p *ingressProxy) isCurrentStatus(status *ingressservices.Status) bool {
|
||||||
|
if status == nil {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return status.PodIPv4 == p.podIPv4 && status.PodIPv6 == p.podIPv6
|
||||||
|
}
|
||||||
|
|
||||||
|
type ingressProxyOpts struct {
|
||||||
|
cfgPath string
|
||||||
|
nfr linuxfw.NetfilterRunner // never nil
|
||||||
|
kc kubeclient.Client // never nil
|
||||||
|
stateSecret string
|
||||||
|
podIPv4 string
|
||||||
|
podIPv6 string
|
||||||
|
}
|
||||||
|
|
||||||
|
// configure sets the ingress proxy's configuration. It is called once on start
|
||||||
|
// so we don't care about concurrent access to fields.
|
||||||
|
func (p *ingressProxy) configure(opts ingressProxyOpts) {
|
||||||
|
p.cfgPath = opts.cfgPath
|
||||||
|
p.nfr = opts.nfr
|
||||||
|
p.kc = opts.kc
|
||||||
|
p.stateSecret = opts.stateSecret
|
||||||
|
p.podIPv4 = opts.podIPv4
|
||||||
|
p.podIPv6 = opts.podIPv6
|
||||||
|
}
|
||||||
|
|
||||||
|
func ingressServicesStatusIsEqual(st, st1 *ingressservices.Configs) bool {
|
||||||
|
if st == nil && st1 == nil {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
if st == nil || st1 == nil {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return reflect.DeepEqual(*st, *st1)
|
||||||
|
}
|
||||||
@ -0,0 +1,223 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
//go:build linux
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/netip"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"tailscale.com/kube/ingressservices"
|
||||||
|
"tailscale.com/util/linuxfw"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestSyncIngressConfigs(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
currentConfigs *ingressservices.Configs
|
||||||
|
currentStatus *ingressservices.Status
|
||||||
|
wantServices map[string]struct {
|
||||||
|
TailscaleServiceIP netip.Addr
|
||||||
|
ClusterIP netip.Addr
|
||||||
|
}
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "add_new_rules_when_no_existing_config",
|
||||||
|
currentConfigs: &ingressservices.Configs{
|
||||||
|
"svc:foo": makeServiceConfig("100.64.0.1", "10.0.0.1", "", ""),
|
||||||
|
},
|
||||||
|
currentStatus: nil,
|
||||||
|
wantServices: map[string]struct {
|
||||||
|
TailscaleServiceIP netip.Addr
|
||||||
|
ClusterIP netip.Addr
|
||||||
|
}{
|
||||||
|
"svc:foo": makeWantService("100.64.0.1", "10.0.0.1"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "add_multiple_services",
|
||||||
|
currentConfigs: &ingressservices.Configs{
|
||||||
|
"svc:foo": makeServiceConfig("100.64.0.1", "10.0.0.1", "", ""),
|
||||||
|
"svc:bar": makeServiceConfig("100.64.0.2", "10.0.0.2", "", ""),
|
||||||
|
"svc:baz": makeServiceConfig("100.64.0.3", "10.0.0.3", "", ""),
|
||||||
|
},
|
||||||
|
currentStatus: nil,
|
||||||
|
wantServices: map[string]struct {
|
||||||
|
TailscaleServiceIP netip.Addr
|
||||||
|
ClusterIP netip.Addr
|
||||||
|
}{
|
||||||
|
"svc:foo": makeWantService("100.64.0.1", "10.0.0.1"),
|
||||||
|
"svc:bar": makeWantService("100.64.0.2", "10.0.0.2"),
|
||||||
|
"svc:baz": makeWantService("100.64.0.3", "10.0.0.3"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "add_both_ipv4_and_ipv6_rules",
|
||||||
|
currentConfigs: &ingressservices.Configs{
|
||||||
|
"svc:foo": makeServiceConfig("100.64.0.1", "10.0.0.1", "2001:db8::1", "2001:db8::2"),
|
||||||
|
},
|
||||||
|
currentStatus: nil,
|
||||||
|
wantServices: map[string]struct {
|
||||||
|
TailscaleServiceIP netip.Addr
|
||||||
|
ClusterIP netip.Addr
|
||||||
|
}{
|
||||||
|
"svc:foo": makeWantService("2001:db8::1", "2001:db8::2"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "add_ipv6_only_rules",
|
||||||
|
currentConfigs: &ingressservices.Configs{
|
||||||
|
"svc:ipv6": makeServiceConfig("", "", "2001:db8::10", "2001:db8::20"),
|
||||||
|
},
|
||||||
|
currentStatus: nil,
|
||||||
|
wantServices: map[string]struct {
|
||||||
|
TailscaleServiceIP netip.Addr
|
||||||
|
ClusterIP netip.Addr
|
||||||
|
}{
|
||||||
|
"svc:ipv6": makeWantService("2001:db8::10", "2001:db8::20"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "delete_all_rules_when_config_removed",
|
||||||
|
currentConfigs: nil,
|
||||||
|
currentStatus: &ingressservices.Status{
|
||||||
|
Configs: ingressservices.Configs{
|
||||||
|
"svc:foo": makeServiceConfig("100.64.0.1", "10.0.0.1", "", ""),
|
||||||
|
"svc:bar": makeServiceConfig("100.64.0.2", "10.0.0.2", "", ""),
|
||||||
|
},
|
||||||
|
PodIPv4: "10.0.0.2", // Current pod IPv4
|
||||||
|
PodIPv6: "2001:db8::2", // Current pod IPv6
|
||||||
|
},
|
||||||
|
wantServices: map[string]struct {
|
||||||
|
TailscaleServiceIP netip.Addr
|
||||||
|
ClusterIP netip.Addr
|
||||||
|
}{},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "add_remove_modify",
|
||||||
|
currentConfigs: &ingressservices.Configs{
|
||||||
|
"svc:foo": makeServiceConfig("100.64.0.1", "10.0.0.2", "", ""), // Changed cluster IP
|
||||||
|
"svc:new": makeServiceConfig("100.64.0.4", "10.0.0.4", "", ""),
|
||||||
|
},
|
||||||
|
currentStatus: &ingressservices.Status{
|
||||||
|
Configs: ingressservices.Configs{
|
||||||
|
"svc:foo": makeServiceConfig("100.64.0.1", "10.0.0.1", "", ""),
|
||||||
|
"svc:bar": makeServiceConfig("100.64.0.2", "10.0.0.2", "", ""),
|
||||||
|
"svc:baz": makeServiceConfig("100.64.0.3", "10.0.0.3", "", ""),
|
||||||
|
},
|
||||||
|
PodIPv4: "10.0.0.2", // Current pod IPv4
|
||||||
|
PodIPv6: "2001:db8::2", // Current pod IPv6
|
||||||
|
},
|
||||||
|
wantServices: map[string]struct {
|
||||||
|
TailscaleServiceIP netip.Addr
|
||||||
|
ClusterIP netip.Addr
|
||||||
|
}{
|
||||||
|
"svc:foo": makeWantService("100.64.0.1", "10.0.0.2"),
|
||||||
|
"svc:new": makeWantService("100.64.0.4", "10.0.0.4"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "update_with_outdated_status",
|
||||||
|
currentConfigs: &ingressservices.Configs{
|
||||||
|
"svc:web": makeServiceConfig("100.64.0.10", "10.0.0.10", "", ""),
|
||||||
|
"svc:web-ipv6": {
|
||||||
|
IPv6Mapping: &ingressservices.Mapping{
|
||||||
|
TailscaleServiceIP: netip.MustParseAddr("2001:db8::10"),
|
||||||
|
ClusterIP: netip.MustParseAddr("2001:db8::20"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
"svc:api": makeServiceConfig("100.64.0.20", "10.0.0.20", "", ""),
|
||||||
|
},
|
||||||
|
currentStatus: &ingressservices.Status{
|
||||||
|
Configs: ingressservices.Configs{
|
||||||
|
"svc:web": makeServiceConfig("100.64.0.10", "10.0.0.10", "", ""),
|
||||||
|
"svc:web-ipv6": {
|
||||||
|
IPv6Mapping: &ingressservices.Mapping{
|
||||||
|
TailscaleServiceIP: netip.MustParseAddr("2001:db8::10"),
|
||||||
|
ClusterIP: netip.MustParseAddr("2001:db8::20"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
"svc:old": makeServiceConfig("100.64.0.30", "10.0.0.30", "", ""),
|
||||||
|
},
|
||||||
|
PodIPv4: "10.0.0.1", // Outdated pod IP
|
||||||
|
PodIPv6: "2001:db8::1", // Outdated pod IP
|
||||||
|
},
|
||||||
|
wantServices: map[string]struct {
|
||||||
|
TailscaleServiceIP netip.Addr
|
||||||
|
ClusterIP netip.Addr
|
||||||
|
}{
|
||||||
|
"svc:web": makeWantService("100.64.0.10", "10.0.0.10"),
|
||||||
|
"svc:web-ipv6": makeWantService("2001:db8::10", "2001:db8::20"),
|
||||||
|
"svc:api": makeWantService("100.64.0.20", "10.0.0.20"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
var nfr linuxfw.NetfilterRunner = linuxfw.NewFakeNetfilterRunner()
|
||||||
|
|
||||||
|
ep := &ingressProxy{
|
||||||
|
nfr: nfr,
|
||||||
|
podIPv4: "10.0.0.2", // Current pod IPv4
|
||||||
|
podIPv6: "2001:db8::2", // Current pod IPv6
|
||||||
|
}
|
||||||
|
|
||||||
|
err := ep.syncIngressConfigs(tt.currentConfigs, tt.currentStatus)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("syncIngressConfigs failed: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
fake := nfr.(*linuxfw.FakeNetfilterRunner)
|
||||||
|
gotServices := fake.GetServiceState()
|
||||||
|
if len(gotServices) != len(tt.wantServices) {
|
||||||
|
t.Errorf("got %d services, want %d", len(gotServices), len(tt.wantServices))
|
||||||
|
}
|
||||||
|
for svc, want := range tt.wantServices {
|
||||||
|
got, ok := gotServices[svc]
|
||||||
|
if !ok {
|
||||||
|
t.Errorf("service %s not found", svc)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if got.TailscaleServiceIP != want.TailscaleServiceIP {
|
||||||
|
t.Errorf("service %s: got TailscaleServiceIP %v, want %v", svc, got.TailscaleServiceIP, want.TailscaleServiceIP)
|
||||||
|
}
|
||||||
|
if got.ClusterIP != want.ClusterIP {
|
||||||
|
t.Errorf("service %s: got ClusterIP %v, want %v", svc, got.ClusterIP, want.ClusterIP)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func makeServiceConfig(tsIP, clusterIP string, tsIP6, clusterIP6 string) ingressservices.Config {
|
||||||
|
cfg := ingressservices.Config{}
|
||||||
|
if tsIP != "" && clusterIP != "" {
|
||||||
|
cfg.IPv4Mapping = &ingressservices.Mapping{
|
||||||
|
TailscaleServiceIP: netip.MustParseAddr(tsIP),
|
||||||
|
ClusterIP: netip.MustParseAddr(clusterIP),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if tsIP6 != "" && clusterIP6 != "" {
|
||||||
|
cfg.IPv6Mapping = &ingressservices.Mapping{
|
||||||
|
TailscaleServiceIP: netip.MustParseAddr(tsIP6),
|
||||||
|
ClusterIP: netip.MustParseAddr(clusterIP6),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return cfg
|
||||||
|
}
|
||||||
|
|
||||||
|
func makeWantService(tsIP, clusterIP string) struct {
|
||||||
|
TailscaleServiceIP netip.Addr
|
||||||
|
ClusterIP netip.Addr
|
||||||
|
} {
|
||||||
|
return struct {
|
||||||
|
TailscaleServiceIP netip.Addr
|
||||||
|
ClusterIP netip.Addr
|
||||||
|
}{
|
||||||
|
TailscaleServiceIP: netip.MustParseAddr(tsIP),
|
||||||
|
ClusterIP: netip.MustParseAddr(clusterIP),
|
||||||
|
}
|
||||||
|
}
|
||||||
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,271 @@
|
|||||||
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||||||
|
// SPDX-License-Identifier: BSD-3-Clause
|
||||||
|
|
||||||
|
//go:build linux
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/google/go-cmp/cmp"
|
||||||
|
"tailscale.com/client/local"
|
||||||
|
"tailscale.com/ipn"
|
||||||
|
"tailscale.com/kube/kubetypes"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestUpdateServeConfig(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
sc *ipn.ServeConfig
|
||||||
|
certDomain string
|
||||||
|
wantCall bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "no_https_no_cert_domain",
|
||||||
|
sc: &ipn.ServeConfig{
|
||||||
|
TCP: map[uint16]*ipn.TCPPortHandler{
|
||||||
|
80: {HTTP: true},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
certDomain: kubetypes.ValueNoHTTPS, // tailnet has HTTPS disabled
|
||||||
|
wantCall: true, // should set serve config as it doesn't have HTTPS endpoints
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "https_with_cert_domain",
|
||||||
|
sc: &ipn.ServeConfig{
|
||||||
|
TCP: map[uint16]*ipn.TCPPortHandler{
|
||||||
|
443: {HTTPS: true},
|
||||||
|
},
|
||||||
|
Web: map[ipn.HostPort]*ipn.WebServerConfig{
|
||||||
|
"${TS_CERT_DOMAIN}:443": {
|
||||||
|
Handlers: map[string]*ipn.HTTPHandler{
|
||||||
|
"/": {Proxy: "http://10.0.1.100:8080"},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
certDomain: "test-node.tailnet.ts.net",
|
||||||
|
wantCall: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "https_without_cert_domain",
|
||||||
|
sc: &ipn.ServeConfig{
|
||||||
|
TCP: map[uint16]*ipn.TCPPortHandler{
|
||||||
|
443: {HTTPS: true},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
certDomain: kubetypes.ValueNoHTTPS,
|
||||||
|
wantCall: false, // incorrect configuration- should not set serve config
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
fakeLC := &fakeLocalClient{}
|
||||||
|
err := updateServeConfig(context.Background(), tt.sc, tt.certDomain, fakeLC)
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("updateServeConfig() error = %v", err)
|
||||||
|
}
|
||||||
|
if fakeLC.setServeCalled != tt.wantCall {
|
||||||
|
t.Errorf("SetServeConfig() called = %v, want %v", fakeLC.setServeCalled, tt.wantCall)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestReadServeConfig(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
gotSC string
|
||||||
|
certDomain string
|
||||||
|
wantSC *ipn.ServeConfig
|
||||||
|
wantErr bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "empty_file",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "valid_config_with_cert_domain_placeholder",
|
||||||
|
gotSC: `{
|
||||||
|
"TCP": {
|
||||||
|
"443": {
|
||||||
|
"HTTPS": true
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"Web": {
|
||||||
|
"${TS_CERT_DOMAIN}:443": {
|
||||||
|
"Handlers": {
|
||||||
|
"/api": {
|
||||||
|
"Proxy": "https://10.2.3.4/api"
|
||||||
|
}}}}}`,
|
||||||
|
certDomain: "example.com",
|
||||||
|
wantSC: &ipn.ServeConfig{
|
||||||
|
TCP: map[uint16]*ipn.TCPPortHandler{
|
||||||
|
443: {
|
||||||
|
HTTPS: true,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
Web: map[ipn.HostPort]*ipn.WebServerConfig{
|
||||||
|
ipn.HostPort("example.com:443"): {
|
||||||
|
Handlers: map[string]*ipn.HTTPHandler{
|
||||||
|
"/api": {
|
||||||
|
Proxy: "https://10.2.3.4/api",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "valid_config_for_http_proxy",
|
||||||
|
gotSC: `{
|
||||||
|
"TCP": {
|
||||||
|
"80": {
|
||||||
|
"HTTP": true
|
||||||
|
}
|
||||||
|
}}`,
|
||||||
|
wantSC: &ipn.ServeConfig{
|
||||||
|
TCP: map[uint16]*ipn.TCPPortHandler{
|
||||||
|
80: {
|
||||||
|
HTTP: true,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "config_without_cert_domain",
|
||||||
|
gotSC: `{
|
||||||
|
"TCP": {
|
||||||
|
"443": {
|
||||||
|
"HTTPS": true
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"Web": {
|
||||||
|
"localhost:443": {
|
||||||
|
"Handlers": {
|
||||||
|
"/api": {
|
||||||
|
"Proxy": "https://10.2.3.4/api"
|
||||||
|
}}}}}`,
|
||||||
|
certDomain: "",
|
||||||
|
wantErr: false,
|
||||||
|
wantSC: &ipn.ServeConfig{
|
||||||
|
TCP: map[uint16]*ipn.TCPPortHandler{
|
||||||
|
443: {
|
||||||
|
HTTPS: true,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
Web: map[ipn.HostPort]*ipn.WebServerConfig{
|
||||||
|
ipn.HostPort("localhost:443"): {
|
||||||
|
Handlers: map[string]*ipn.HTTPHandler{
|
||||||
|
"/api": {
|
||||||
|
Proxy: "https://10.2.3.4/api",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "invalid_json",
|
||||||
|
gotSC: "invalid json",
|
||||||
|
wantErr: true,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
dir := t.TempDir()
|
||||||
|
path := filepath.Join(dir, "serve-config.json")
|
||||||
|
if err := os.WriteFile(path, []byte(tt.gotSC), 0644); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
got, err := readServeConfig(path, tt.certDomain)
|
||||||
|
if (err != nil) != tt.wantErr {
|
||||||
|
t.Errorf("readServeConfig() error = %v, wantErr %v", err, tt.wantErr)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if !cmp.Equal(got, tt.wantSC) {
|
||||||
|
t.Errorf("readServeConfig() diff (-got +want):\n%s", cmp.Diff(got, tt.wantSC))
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type fakeLocalClient struct {
|
||||||
|
*local.Client
|
||||||
|
setServeCalled bool
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *fakeLocalClient) SetServeConfig(ctx context.Context, cfg *ipn.ServeConfig) error {
|
||||||
|
m.setServeCalled = true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *fakeLocalClient) CertPair(ctx context.Context, domain string) (certPEM, keyPEM []byte, err error) {
|
||||||
|
return nil, nil, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestHasHTTPSEndpoint(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
cfg *ipn.ServeConfig
|
||||||
|
want bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "nil_config",
|
||||||
|
cfg: nil,
|
||||||
|
want: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "empty_config",
|
||||||
|
cfg: &ipn.ServeConfig{},
|
||||||
|
want: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "no_https_endpoints",
|
||||||
|
cfg: &ipn.ServeConfig{
|
||||||
|
TCP: map[uint16]*ipn.TCPPortHandler{
|
||||||
|
80: {
|
||||||
|
HTTPS: false,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
want: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "has_https_endpoint",
|
||||||
|
cfg: &ipn.ServeConfig{
|
||||||
|
TCP: map[uint16]*ipn.TCPPortHandler{
|
||||||
|
443: {
|
||||||
|
HTTPS: true,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
want: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "mixed_endpoints",
|
||||||
|
cfg: &ipn.ServeConfig{
|
||||||
|
TCP: map[uint16]*ipn.TCPPortHandler{
|
||||||
|
80: {HTTPS: false},
|
||||||
|
443: {HTTPS: true},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
want: true,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
got := hasHTTPSEndpoint(tt.cfg)
|
||||||
|
if got != tt.want {
|
||||||
|
t.Errorf("hasHTTPSEndpoint() = %v, want %v", got, tt.want)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue