Compare commits
22 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| b049045e38 | |||
| 76df8d9f75 | |||
| b87c1f32c8 | |||
| c4e0fe3279 | |||
| 723758959e | |||
| 4daaac079f | |||
| bda3d7863d | |||
| 666f5842ae | |||
| 16185aa1c3 | |||
| a1abe08469 | |||
| 82f9b4dc0c | |||
| aa069f333d | |||
| a3d9d6f954 | |||
| da3d5cc9d9 | |||
| 4ac0f6ed4d | |||
| df8e746fc4 | |||
| 51e8073c7e | |||
| 1c175100b9 | |||
| 24838172c7 | |||
| 6d4403012f | |||
| e69abd85b6 | |||
| 54179a7112 |
@@ -40,10 +40,10 @@ jobs:
|
|||||||
uses: actions/checkout@v6
|
uses: actions/checkout@v6
|
||||||
|
|
||||||
- name: Set up Docker Buildx
|
- name: Set up Docker Buildx
|
||||||
uses: docker/setup-buildx-action@v3
|
uses: docker/setup-buildx-action@v4
|
||||||
|
|
||||||
- name: Log in to Docker Registry
|
- name: Log in to Docker Registry
|
||||||
uses: docker/login-action@v3
|
uses: docker/login-action@v4
|
||||||
with:
|
with:
|
||||||
registry: git.fossy.my.id
|
registry: git.fossy.my.id
|
||||||
username: ${{ secrets.DOCKER_USERNAME }}
|
username: ${{ secrets.DOCKER_USERNAME }}
|
||||||
@@ -79,7 +79,7 @@ jobs:
|
|||||||
fi
|
fi
|
||||||
|
|
||||||
- name: Build and push Docker image (release)
|
- name: Build and push Docker image (release)
|
||||||
uses: docker/build-push-action@v6
|
uses: docker/build-push-action@v7
|
||||||
with:
|
with:
|
||||||
context: .
|
context: .
|
||||||
push: true
|
push: true
|
||||||
@@ -97,7 +97,7 @@ jobs:
|
|||||||
if: steps.version.outputs.IS_PRERELEASE == 'false'
|
if: steps.version.outputs.IS_PRERELEASE == 'false'
|
||||||
|
|
||||||
- name: Build and push Docker image (pre-release)
|
- name: Build and push Docker image (pre-release)
|
||||||
uses: docker/build-push-action@v6
|
uses: docker/build-push-action@v7
|
||||||
with:
|
with:
|
||||||
context: .
|
context: .
|
||||||
push: true
|
push: true
|
||||||
|
|||||||
@@ -43,7 +43,7 @@ jobs:
|
|||||||
--output.checkstyle.path=golangci-lint-report.xml
|
--output.checkstyle.path=golangci-lint-report.xml
|
||||||
|
|
||||||
- name: SonarQube Scan
|
- name: SonarQube Scan
|
||||||
uses: SonarSource/sonarqube-scan-action@v7.0.0
|
uses: SonarSource/sonarqube-scan-action@v8.2.1
|
||||||
env:
|
env:
|
||||||
SONAR_HOST_URL: ${{ secrets.SONARQUBE_HOST }}
|
SONAR_HOST_URL: ${{ secrets.SONARQUBE_HOST }}
|
||||||
SONAR_TOKEN: ${{ secrets.SONARQUBE_TOKEN }}
|
SONAR_TOKEN: ${{ secrets.SONARQUBE_TOKEN }}
|
||||||
|
|||||||
+1
-1
@@ -1,4 +1,4 @@
|
|||||||
FROM golang:1.26.0-alpine AS go_builder
|
FROM golang:1.26.2-alpine AS go_builder
|
||||||
|
|
||||||
ARG VERSION=dev
|
ARG VERSION=dev
|
||||||
ARG BUILD_DATE=unknown
|
ARG BUILD_DATE=unknown
|
||||||
|
|||||||
@@ -36,7 +36,6 @@ The following environment variables can be configured in the `.env` file:
|
|||||||
| Variable | Description | Default | Required |
|
| Variable | Description | Default | Required |
|
||||||
|---------------------|-----------------------------------------------------------------------------|-------------------------|---------------------|
|
|---------------------|-----------------------------------------------------------------------------|-------------------------|---------------------|
|
||||||
| `DOMAIN` | Domain name for subdomain routing | `localhost` | No |
|
| `DOMAIN` | Domain name for subdomain routing | `localhost` | No |
|
||||||
| `FRONTEND_URL` | URL for the frontend dashboard/landing page | `https://<DOMAIN>` | No |
|
|
||||||
| `PORT` | SSH server port | `2200` | No |
|
| `PORT` | SSH server port | `2200` | No |
|
||||||
| `HTTP_PORT` | HTTP server port | `8080` | No |
|
| `HTTP_PORT` | HTTP server port | `8080` | No |
|
||||||
| `HTTPS_PORT` | HTTPS server port | `8443` | No |
|
| `HTTPS_PORT` | HTTPS server port | `8443` | No |
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ go 1.26.0
|
|||||||
|
|
||||||
require (
|
require (
|
||||||
git.fossy.my.id/bagas/tunnel-please-grpc v1.5.0
|
git.fossy.my.id/bagas/tunnel-please-grpc v1.5.0
|
||||||
github.com/caddyserver/certmagic v0.25.1
|
github.com/caddyserver/certmagic v0.25.3
|
||||||
github.com/charmbracelet/bubbles v1.0.0
|
github.com/charmbracelet/bubbles v1.0.0
|
||||||
github.com/charmbracelet/bubbletea v1.3.10
|
github.com/charmbracelet/bubbletea v1.3.10
|
||||||
github.com/charmbracelet/lipgloss v1.1.0
|
github.com/charmbracelet/lipgloss v1.1.0
|
||||||
@@ -12,15 +12,15 @@ require (
|
|||||||
github.com/libdns/cloudflare v0.2.2
|
github.com/libdns/cloudflare v0.2.2
|
||||||
github.com/muesli/termenv v0.16.0
|
github.com/muesli/termenv v0.16.0
|
||||||
github.com/stretchr/testify v1.11.1
|
github.com/stretchr/testify v1.11.1
|
||||||
golang.org/x/crypto v0.48.0
|
golang.org/x/crypto v0.50.0
|
||||||
google.golang.org/grpc v1.78.0
|
google.golang.org/grpc v1.80.0
|
||||||
google.golang.org/protobuf v1.36.11
|
google.golang.org/protobuf v1.36.11
|
||||||
)
|
)
|
||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/atotto/clipboard v0.1.4 // indirect
|
github.com/atotto/clipboard v0.1.4 // indirect
|
||||||
github.com/aymanbagabas/go-osc52/v2 v2.0.1 // indirect
|
github.com/aymanbagabas/go-osc52/v2 v2.0.1 // indirect
|
||||||
github.com/caddyserver/zerossl v0.1.4 // indirect
|
github.com/caddyserver/zerossl v0.1.5 // indirect
|
||||||
github.com/charmbracelet/colorprofile v0.4.1 // indirect
|
github.com/charmbracelet/colorprofile v0.4.1 // indirect
|
||||||
github.com/charmbracelet/x/ansi v0.11.6 // indirect
|
github.com/charmbracelet/x/ansi v0.11.6 // indirect
|
||||||
github.com/charmbracelet/x/cellbuf v0.0.15 // indirect
|
github.com/charmbracelet/x/cellbuf v0.0.15 // indirect
|
||||||
@@ -36,8 +36,8 @@ require (
|
|||||||
github.com/mattn/go-isatty v0.0.20 // indirect
|
github.com/mattn/go-isatty v0.0.20 // indirect
|
||||||
github.com/mattn/go-localereader v0.0.1 // indirect
|
github.com/mattn/go-localereader v0.0.1 // indirect
|
||||||
github.com/mattn/go-runewidth v0.0.19 // indirect
|
github.com/mattn/go-runewidth v0.0.19 // indirect
|
||||||
github.com/mholt/acmez/v3 v3.1.4 // indirect
|
github.com/mholt/acmez/v3 v3.1.6 // indirect
|
||||||
github.com/miekg/dns v1.1.69 // indirect
|
github.com/miekg/dns v1.1.72 // indirect
|
||||||
github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 // indirect
|
github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 // indirect
|
||||||
github.com/muesli/cancelreader v0.2.2 // indirect
|
github.com/muesli/cancelreader v0.2.2 // indirect
|
||||||
github.com/pmezard/go-difflib v1.0.0 // indirect
|
github.com/pmezard/go-difflib v1.0.0 // indirect
|
||||||
@@ -49,12 +49,12 @@ require (
|
|||||||
go.uber.org/multierr v1.11.0 // indirect
|
go.uber.org/multierr v1.11.0 // indirect
|
||||||
go.uber.org/zap v1.27.1 // indirect
|
go.uber.org/zap v1.27.1 // indirect
|
||||||
go.uber.org/zap/exp v0.3.0 // indirect
|
go.uber.org/zap/exp v0.3.0 // indirect
|
||||||
golang.org/x/mod v0.32.0 // indirect
|
golang.org/x/mod v0.35.0 // indirect
|
||||||
golang.org/x/net v0.49.0 // indirect
|
golang.org/x/net v0.53.0 // indirect
|
||||||
golang.org/x/sync v0.19.0 // indirect
|
golang.org/x/sync v0.20.0 // indirect
|
||||||
golang.org/x/sys v0.41.0 // indirect
|
golang.org/x/sys v0.43.0 // indirect
|
||||||
golang.org/x/text v0.34.0 // indirect
|
golang.org/x/text v0.36.0 // indirect
|
||||||
golang.org/x/tools v0.41.0 // indirect
|
golang.org/x/tools v0.44.0 // indirect
|
||||||
google.golang.org/genproto/googleapis/rpc v0.0.0-20251222181119-0a764e51fe1b // indirect
|
google.golang.org/genproto/googleapis/rpc v0.0.0-20260120221211-b8f7ae30c516 // indirect
|
||||||
gopkg.in/yaml.v3 v3.0.1 // indirect
|
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -8,8 +8,14 @@ github.com/aymanbagabas/go-udiff v0.3.1 h1:LV+qyBQ2pqe0u42ZsUEtPiCaUoqgA9gYRDs3v
|
|||||||
github.com/aymanbagabas/go-udiff v0.3.1/go.mod h1:G0fsKmG+P6ylD0r6N/KgQD/nWzgfnl8ZBcNLgcbrw8E=
|
github.com/aymanbagabas/go-udiff v0.3.1/go.mod h1:G0fsKmG+P6ylD0r6N/KgQD/nWzgfnl8ZBcNLgcbrw8E=
|
||||||
github.com/caddyserver/certmagic v0.25.1 h1:4sIKKbOt5pg6+sL7tEwymE1x2bj6CHr80da1CRRIPbY=
|
github.com/caddyserver/certmagic v0.25.1 h1:4sIKKbOt5pg6+sL7tEwymE1x2bj6CHr80da1CRRIPbY=
|
||||||
github.com/caddyserver/certmagic v0.25.1/go.mod h1:VhyvndxtVton/Fo/wKhRoC46Rbw1fmjvQ3GjHYSQTEY=
|
github.com/caddyserver/certmagic v0.25.1/go.mod h1:VhyvndxtVton/Fo/wKhRoC46Rbw1fmjvQ3GjHYSQTEY=
|
||||||
|
github.com/caddyserver/certmagic v0.25.2 h1:D7xcS7ggX/WEY54x0czj7ioTkmDWKIgxtIi2OcQclUc=
|
||||||
|
github.com/caddyserver/certmagic v0.25.2/go.mod h1:llW/CvsNmza8S6hmsuggsZeiX+uS27dkqY27wDIuBWg=
|
||||||
|
github.com/caddyserver/certmagic v0.25.3 h1:mGf5ba8F7xA4c5jfDZZbK2buY1VEkbnwpMDixaju94A=
|
||||||
|
github.com/caddyserver/certmagic v0.25.3/go.mod h1:YVs43D5+H/Dckt4bTga1KSO/xYfFBfVZainGDywYPAA=
|
||||||
github.com/caddyserver/zerossl v0.1.4 h1:CVJOE3MZeFisCERZjkxIcsqIH4fnFdlYWnPYeFtBHRw=
|
github.com/caddyserver/zerossl v0.1.4 h1:CVJOE3MZeFisCERZjkxIcsqIH4fnFdlYWnPYeFtBHRw=
|
||||||
github.com/caddyserver/zerossl v0.1.4/go.mod h1:CxA0acn7oEGO6//4rtrRjYgEoa4MFw/XofZnrYwGqG4=
|
github.com/caddyserver/zerossl v0.1.4/go.mod h1:CxA0acn7oEGO6//4rtrRjYgEoa4MFw/XofZnrYwGqG4=
|
||||||
|
github.com/caddyserver/zerossl v0.1.5 h1:dkvOjBAEEtY6LIGAHei7sw2UgqSD6TrWweXpV7lvEvE=
|
||||||
|
github.com/caddyserver/zerossl v0.1.5/go.mod h1:CxA0acn7oEGO6//4rtrRjYgEoa4MFw/XofZnrYwGqG4=
|
||||||
github.com/charmbracelet/bubbles v1.0.0 h1:12J8/ak/uCZEMQ6KU7pcfwceyjLlWsDLAxB5fXonfvc=
|
github.com/charmbracelet/bubbles v1.0.0 h1:12J8/ak/uCZEMQ6KU7pcfwceyjLlWsDLAxB5fXonfvc=
|
||||||
github.com/charmbracelet/bubbles v1.0.0/go.mod h1:9d/Zd5GdnauMI5ivUIVisuEm3ave1XwXtD1ckyV6r3E=
|
github.com/charmbracelet/bubbles v1.0.0/go.mod h1:9d/Zd5GdnauMI5ivUIVisuEm3ave1XwXtD1ckyV6r3E=
|
||||||
github.com/charmbracelet/bubbletea v1.3.10 h1:otUDHWMMzQSB0Pkc87rm691KZ3SWa4KUlvF9nRvCICw=
|
github.com/charmbracelet/bubbletea v1.3.10 h1:otUDHWMMzQSB0Pkc87rm691KZ3SWa4KUlvF9nRvCICw=
|
||||||
@@ -66,8 +72,12 @@ github.com/mattn/go-runewidth v0.0.19 h1:v++JhqYnZuu5jSKrk9RbgF5v4CGUjqRfBm05byF
|
|||||||
github.com/mattn/go-runewidth v0.0.19/go.mod h1:XBkDxAl56ILZc9knddidhrOlY5R/pDhgLpndooCuJAs=
|
github.com/mattn/go-runewidth v0.0.19/go.mod h1:XBkDxAl56ILZc9knddidhrOlY5R/pDhgLpndooCuJAs=
|
||||||
github.com/mholt/acmez/v3 v3.1.4 h1:DyzZe/RnAzT3rpZj/2Ii5xZpiEvvYk3cQEN/RmqxwFQ=
|
github.com/mholt/acmez/v3 v3.1.4 h1:DyzZe/RnAzT3rpZj/2Ii5xZpiEvvYk3cQEN/RmqxwFQ=
|
||||||
github.com/mholt/acmez/v3 v3.1.4/go.mod h1:L1wOU06KKvq7tswuMDwKdcHeKpFFgkppZy/y0DFxagQ=
|
github.com/mholt/acmez/v3 v3.1.4/go.mod h1:L1wOU06KKvq7tswuMDwKdcHeKpFFgkppZy/y0DFxagQ=
|
||||||
|
github.com/mholt/acmez/v3 v3.1.6 h1:eGVQNObP0pBN4sxqrXeg7MYqTOWyoiYpQqITVWlrevk=
|
||||||
|
github.com/mholt/acmez/v3 v3.1.6/go.mod h1:5nTPosTGosLxF3+LU4ygbgMRFDhbAVpqMI4+a4aHLBY=
|
||||||
github.com/miekg/dns v1.1.69 h1:Kb7Y/1Jo+SG+a2GtfoFUfDkG//csdRPwRLkCsxDG9Sc=
|
github.com/miekg/dns v1.1.69 h1:Kb7Y/1Jo+SG+a2GtfoFUfDkG//csdRPwRLkCsxDG9Sc=
|
||||||
github.com/miekg/dns v1.1.69/go.mod h1:7OyjD9nEba5OkqQ/hB4fy3PIoxafSZJtducccIelz3g=
|
github.com/miekg/dns v1.1.69/go.mod h1:7OyjD9nEba5OkqQ/hB4fy3PIoxafSZJtducccIelz3g=
|
||||||
|
github.com/miekg/dns v1.1.72 h1:vhmr+TF2A3tuoGNkLDFK9zi36F2LS+hKTRW0Uf8kbzI=
|
||||||
|
github.com/miekg/dns v1.1.72/go.mod h1:+EuEPhdHOsfk6Wk5TT2CzssZdqkmFhf8r+aVyDEToIs=
|
||||||
github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 h1:ZK8zHtRHOkbHy6Mmr5D264iyp3TiX5OmNcI5cIARiQI=
|
github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 h1:ZK8zHtRHOkbHy6Mmr5D264iyp3TiX5OmNcI5cIARiQI=
|
||||||
github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6/go.mod h1:CJlz5H+gyd6CUWT45Oy4q24RdLyn7Md9Vj2/ldJBSIo=
|
github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6/go.mod h1:CJlz5H+gyd6CUWT45Oy4q24RdLyn7Md9Vj2/ldJBSIo=
|
||||||
github.com/muesli/cancelreader v0.2.2 h1:3I4Kt4BQjOR54NavqnDogx/MIoWBFa0StPA8ELUXHmA=
|
github.com/muesli/cancelreader v0.2.2 h1:3I4Kt4BQjOR54NavqnDogx/MIoWBFa0StPA8ELUXHmA=
|
||||||
@@ -114,30 +124,66 @@ go.uber.org/zap/exp v0.3.0 h1:6JYzdifzYkGmTdRR59oYH+Ng7k49H9qVpWwNSsGJj3U=
|
|||||||
go.uber.org/zap/exp v0.3.0/go.mod h1:5I384qq7XGxYyByIhHm6jg5CHkGY0nsTfbDLgDDlgJQ=
|
go.uber.org/zap/exp v0.3.0/go.mod h1:5I384qq7XGxYyByIhHm6jg5CHkGY0nsTfbDLgDDlgJQ=
|
||||||
golang.org/x/crypto v0.48.0 h1:/VRzVqiRSggnhY7gNRxPauEQ5Drw9haKdM0jqfcCFts=
|
golang.org/x/crypto v0.48.0 h1:/VRzVqiRSggnhY7gNRxPauEQ5Drw9haKdM0jqfcCFts=
|
||||||
golang.org/x/crypto v0.48.0/go.mod h1:r0kV5h3qnFPlQnBSrULhlsRfryS2pmewsg+XfMgkVos=
|
golang.org/x/crypto v0.48.0/go.mod h1:r0kV5h3qnFPlQnBSrULhlsRfryS2pmewsg+XfMgkVos=
|
||||||
|
golang.org/x/crypto v0.49.0 h1:+Ng2ULVvLHnJ/ZFEq4KdcDd/cfjrrjjNSXNzxg0Y4U4=
|
||||||
|
golang.org/x/crypto v0.49.0/go.mod h1:ErX4dUh2UM+CFYiXZRTcMpEcN8b/1gxEuv3nODoYtCA=
|
||||||
|
golang.org/x/crypto v0.50.0 h1:zO47/JPrL6vsNkINmLoo/PH1gcxpls50DNogFvB5ZGI=
|
||||||
|
golang.org/x/crypto v0.50.0/go.mod h1:3muZ7vA7PBCE6xgPX7nkzzjiUq87kRItoJQM1Yo8S+Q=
|
||||||
golang.org/x/exp v0.0.0-20231006140011-7918f672742d h1:jtJma62tbqLibJ5sFQz8bKtEM8rJBtfilJ2qTU199MI=
|
golang.org/x/exp v0.0.0-20231006140011-7918f672742d h1:jtJma62tbqLibJ5sFQz8bKtEM8rJBtfilJ2qTU199MI=
|
||||||
golang.org/x/exp v0.0.0-20231006140011-7918f672742d/go.mod h1:ldy0pHrwJyGW56pPQzzkH36rKxoZW1tw7ZJpeKx+hdo=
|
golang.org/x/exp v0.0.0-20231006140011-7918f672742d/go.mod h1:ldy0pHrwJyGW56pPQzzkH36rKxoZW1tw7ZJpeKx+hdo=
|
||||||
golang.org/x/mod v0.32.0 h1:9F4d3PHLljb6x//jOyokMv3eX+YDeepZSEo3mFJy93c=
|
golang.org/x/mod v0.32.0 h1:9F4d3PHLljb6x//jOyokMv3eX+YDeepZSEo3mFJy93c=
|
||||||
golang.org/x/mod v0.32.0/go.mod h1:SgipZ/3h2Ci89DlEtEXWUk/HteuRin+HHhN+WbNhguU=
|
golang.org/x/mod v0.32.0/go.mod h1:SgipZ/3h2Ci89DlEtEXWUk/HteuRin+HHhN+WbNhguU=
|
||||||
|
golang.org/x/mod v0.33.0 h1:tHFzIWbBifEmbwtGz65eaWyGiGZatSrT9prnU8DbVL8=
|
||||||
|
golang.org/x/mod v0.33.0/go.mod h1:swjeQEj+6r7fODbD2cqrnje9PnziFuw4bmLbBZFrQ5w=
|
||||||
|
golang.org/x/mod v0.34.0 h1:xIHgNUUnW6sYkcM5Jleh05DvLOtwc6RitGHbDk4akRI=
|
||||||
|
golang.org/x/mod v0.34.0/go.mod h1:ykgH52iCZe79kzLLMhyCUzhMci+nQj+0XkbXpNYtVjY=
|
||||||
|
golang.org/x/mod v0.35.0 h1:Ww1D637e6Pg+Zb2KrWfHQUnH2dQRLBQyAtpr/haaJeM=
|
||||||
|
golang.org/x/mod v0.35.0/go.mod h1:+GwiRhIInF8wPm+4AoT6L0FA1QWAad3OMdTRx4tFYlU=
|
||||||
golang.org/x/net v0.49.0 h1:eeHFmOGUTtaaPSGNmjBKpbng9MulQsJURQUAfUwY++o=
|
golang.org/x/net v0.49.0 h1:eeHFmOGUTtaaPSGNmjBKpbng9MulQsJURQUAfUwY++o=
|
||||||
golang.org/x/net v0.49.0/go.mod h1:/ysNB2EvaqvesRkuLAyjI1ycPZlQHM3q01F02UY/MV8=
|
golang.org/x/net v0.49.0/go.mod h1:/ysNB2EvaqvesRkuLAyjI1ycPZlQHM3q01F02UY/MV8=
|
||||||
|
golang.org/x/net v0.51.0 h1:94R/GTO7mt3/4wIKpcR5gkGmRLOuE/2hNGeWq/GBIFo=
|
||||||
|
golang.org/x/net v0.51.0/go.mod h1:aamm+2QF5ogm02fjy5Bb7CQ0WMt1/WVM7FtyaTLlA9Y=
|
||||||
|
golang.org/x/net v0.52.0 h1:He/TN1l0e4mmR3QqHMT2Xab3Aj3L9qjbhRm78/6jrW0=
|
||||||
|
golang.org/x/net v0.52.0/go.mod h1:R1MAz7uMZxVMualyPXb+VaqGSa3LIaUqk0eEt3w36Sw=
|
||||||
|
golang.org/x/net v0.53.0 h1:d+qAbo5L0orcWAr0a9JweQpjXF19LMXJE8Ey7hwOdUA=
|
||||||
|
golang.org/x/net v0.53.0/go.mod h1:JvMuJH7rrdiCfbeHoo3fCQU24Lf5JJwT9W3sJFulfgs=
|
||||||
golang.org/x/sync v0.19.0 h1:vV+1eWNmZ5geRlYjzm2adRgW2/mcpevXNg50YZtPCE4=
|
golang.org/x/sync v0.19.0 h1:vV+1eWNmZ5geRlYjzm2adRgW2/mcpevXNg50YZtPCE4=
|
||||||
golang.org/x/sync v0.19.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI=
|
golang.org/x/sync v0.19.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI=
|
||||||
|
golang.org/x/sync v0.20.0 h1:e0PTpb7pjO8GAtTs2dQ6jYa5BWYlMuX047Dco/pItO4=
|
||||||
|
golang.org/x/sync v0.20.0/go.mod h1:9xrNwdLfx4jkKbNva9FpL6vEN7evnE43NNNJQ2LF3+0=
|
||||||
golang.org/x/sys v0.0.0-20210809222454-d867a43fc93e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
golang.org/x/sys v0.0.0-20210809222454-d867a43fc93e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
golang.org/x/sys v0.41.0 h1:Ivj+2Cp/ylzLiEU89QhWblYnOE9zerudt9Ftecq2C6k=
|
golang.org/x/sys v0.41.0 h1:Ivj+2Cp/ylzLiEU89QhWblYnOE9zerudt9Ftecq2C6k=
|
||||||
golang.org/x/sys v0.41.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
|
golang.org/x/sys v0.41.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
|
||||||
|
golang.org/x/sys v0.42.0 h1:omrd2nAlyT5ESRdCLYdm3+fMfNFE/+Rf4bDIQImRJeo=
|
||||||
|
golang.org/x/sys v0.42.0/go.mod h1:4GL1E5IUh+htKOUEOaiffhrAeqysfVGipDYzABqnCmw=
|
||||||
|
golang.org/x/sys v0.43.0 h1:Rlag2XtaFTxp19wS8MXlJwTvoh8ArU6ezoyFsMyCTNI=
|
||||||
|
golang.org/x/sys v0.43.0/go.mod h1:4GL1E5IUh+htKOUEOaiffhrAeqysfVGipDYzABqnCmw=
|
||||||
golang.org/x/term v0.40.0 h1:36e4zGLqU4yhjlmxEaagx2KuYbJq3EwY8K943ZsHcvg=
|
golang.org/x/term v0.40.0 h1:36e4zGLqU4yhjlmxEaagx2KuYbJq3EwY8K943ZsHcvg=
|
||||||
golang.org/x/term v0.40.0/go.mod h1:w2P8uVp06p2iyKKuvXIm7N/y0UCRt3UfJTfZ7oOpglM=
|
golang.org/x/term v0.40.0/go.mod h1:w2P8uVp06p2iyKKuvXIm7N/y0UCRt3UfJTfZ7oOpglM=
|
||||||
golang.org/x/text v0.34.0 h1:oL/Qq0Kdaqxa1KbNeMKwQq0reLCCaFtqu2eNuSeNHbk=
|
golang.org/x/text v0.34.0 h1:oL/Qq0Kdaqxa1KbNeMKwQq0reLCCaFtqu2eNuSeNHbk=
|
||||||
golang.org/x/text v0.34.0/go.mod h1:homfLqTYRFyVYemLBFl5GgL/DWEiH5wcsQ5gSh1yziA=
|
golang.org/x/text v0.34.0/go.mod h1:homfLqTYRFyVYemLBFl5GgL/DWEiH5wcsQ5gSh1yziA=
|
||||||
|
golang.org/x/text v0.35.0 h1:JOVx6vVDFokkpaq1AEptVzLTpDe9KGpj5tR4/X+ybL8=
|
||||||
|
golang.org/x/text v0.35.0/go.mod h1:khi/HExzZJ2pGnjenulevKNX1W67CUy0AsXcNubPGCA=
|
||||||
|
golang.org/x/text v0.36.0 h1:JfKh3XmcRPqZPKevfXVpI1wXPTqbkE5f7JA92a55Yxg=
|
||||||
|
golang.org/x/text v0.36.0/go.mod h1:NIdBknypM8iqVmPiuco0Dh6P5Jcdk8lJL0CUebqK164=
|
||||||
golang.org/x/tools v0.41.0 h1:a9b8iMweWG+S0OBnlU36rzLp20z1Rp10w+IY2czHTQc=
|
golang.org/x/tools v0.41.0 h1:a9b8iMweWG+S0OBnlU36rzLp20z1Rp10w+IY2czHTQc=
|
||||||
golang.org/x/tools v0.41.0/go.mod h1:XSY6eDqxVNiYgezAVqqCeihT4j1U2CCsqvH3WhQpnlg=
|
golang.org/x/tools v0.41.0/go.mod h1:XSY6eDqxVNiYgezAVqqCeihT4j1U2CCsqvH3WhQpnlg=
|
||||||
|
golang.org/x/tools v0.42.0 h1:uNgphsn75Tdz5Ji2q36v/nsFSfR/9BRFvqhGBaJGd5k=
|
||||||
|
golang.org/x/tools v0.42.0/go.mod h1:Ma6lCIwGZvHK6XtgbswSoWroEkhugApmsXyrUmBhfr0=
|
||||||
|
golang.org/x/tools v0.43.0 h1:12BdW9CeB3Z+J/I/wj34VMl8X+fEXBxVR90JeMX5E7s=
|
||||||
|
golang.org/x/tools v0.43.0/go.mod h1:uHkMso649BX2cZK6+RpuIPXS3ho2hZo4FVwfoy1vIk0=
|
||||||
|
golang.org/x/tools v0.44.0 h1:UP4ajHPIcuMjT1GqzDWRlalUEoY+uzoZKnhOjbIPD2c=
|
||||||
|
golang.org/x/tools v0.44.0/go.mod h1:KA0AfVErSdxRZIsOVipbv3rQhVXTnlU6UhKxHd1seDI=
|
||||||
gonum.org/v1/gonum v0.16.0 h1:5+ul4Swaf3ESvrOnidPp4GZbzf0mxVQpDCYUQE7OJfk=
|
gonum.org/v1/gonum v0.16.0 h1:5+ul4Swaf3ESvrOnidPp4GZbzf0mxVQpDCYUQE7OJfk=
|
||||||
gonum.org/v1/gonum v0.16.0/go.mod h1:fef3am4MQ93R2HHpKnLk4/Tbh/s0+wqD5nfa6Pnwy4E=
|
gonum.org/v1/gonum v0.16.0/go.mod h1:fef3am4MQ93R2HHpKnLk4/Tbh/s0+wqD5nfa6Pnwy4E=
|
||||||
google.golang.org/genproto/googleapis/rpc v0.0.0-20251222181119-0a764e51fe1b h1:Mv8VFug0MP9e5vUxfBcE3vUkV6CImK3cMNMIDFjmzxU=
|
google.golang.org/genproto/googleapis/rpc v0.0.0-20251222181119-0a764e51fe1b h1:Mv8VFug0MP9e5vUxfBcE3vUkV6CImK3cMNMIDFjmzxU=
|
||||||
google.golang.org/genproto/googleapis/rpc v0.0.0-20251222181119-0a764e51fe1b/go.mod h1:j9x/tPzZkyxcgEFkiKEEGxfvyumM01BEtsW8xzOahRQ=
|
google.golang.org/genproto/googleapis/rpc v0.0.0-20251222181119-0a764e51fe1b/go.mod h1:j9x/tPzZkyxcgEFkiKEEGxfvyumM01BEtsW8xzOahRQ=
|
||||||
|
google.golang.org/genproto/googleapis/rpc v0.0.0-20260120221211-b8f7ae30c516 h1:sNrWoksmOyF5bvJUcnmbeAmQi8baNhqg5IWaI3llQqU=
|
||||||
|
google.golang.org/genproto/googleapis/rpc v0.0.0-20260120221211-b8f7ae30c516/go.mod h1:j9x/tPzZkyxcgEFkiKEEGxfvyumM01BEtsW8xzOahRQ=
|
||||||
google.golang.org/grpc v1.78.0 h1:K1XZG/yGDJnzMdd/uZHAkVqJE+xIDOcmdSFZkBUicNc=
|
google.golang.org/grpc v1.78.0 h1:K1XZG/yGDJnzMdd/uZHAkVqJE+xIDOcmdSFZkBUicNc=
|
||||||
google.golang.org/grpc v1.78.0/go.mod h1:I47qjTo4OKbMkjA/aOOwxDIiPSBofUtQUI5EfpWvW7U=
|
google.golang.org/grpc v1.78.0/go.mod h1:I47qjTo4OKbMkjA/aOOwxDIiPSBofUtQUI5EfpWvW7U=
|
||||||
|
google.golang.org/grpc v1.80.0 h1:Xr6m2WmWZLETvUNvIUmeD5OAagMw3FiKmMlTdViWsHM=
|
||||||
|
google.golang.org/grpc v1.80.0/go.mod h1:ho/dLnxwi3EDJA4Zghp7k2Ec1+c2jqup0bFkw07bwF4=
|
||||||
google.golang.org/protobuf v1.36.11 h1:fV6ZwhNocDyBLK0dj+fg8ektcVegBBuEolpbTQyBNVE=
|
google.golang.org/protobuf v1.36.11 h1:fV6ZwhNocDyBLK0dj+fg8ektcVegBBuEolpbTQyBNVE=
|
||||||
google.golang.org/protobuf v1.36.11/go.mod h1:HTf+CrKn2C3g5S8VImy6tdcUvCska2kB7j23XfzDpco=
|
google.golang.org/protobuf v1.36.11/go.mod h1:HTf+CrKn2C3g5S8VImy6tdcUvCska2kB7j23XfzDpco=
|
||||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
|
||||||
|
|||||||
@@ -80,7 +80,6 @@ type MockConfig struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (m *MockConfig) Domain() string { return m.Called().String(0) }
|
func (m *MockConfig) Domain() string { return m.Called().String(0) }
|
||||||
func (m *MockConfig) FrontendURL() string { return m.Called().String(0) }
|
|
||||||
func (m *MockConfig) SSHPort() string { return m.Called().String(0) }
|
func (m *MockConfig) SSHPort() string { return m.Called().String(0) }
|
||||||
func (m *MockConfig) HTTPPort() string { return m.Called().String(0) }
|
func (m *MockConfig) HTTPPort() string { return m.Called().String(0) }
|
||||||
func (m *MockConfig) HTTPSPort() string { return m.Called().String(0) }
|
func (m *MockConfig) HTTPSPort() string { return m.Called().String(0) }
|
||||||
|
|||||||
@@ -6,7 +6,6 @@ import (
|
|||||||
|
|
||||||
type Config interface {
|
type Config interface {
|
||||||
Domain() string
|
Domain() string
|
||||||
FrontendURL() string
|
|
||||||
SSHPort() string
|
SSHPort() string
|
||||||
|
|
||||||
HTTPPort() string
|
HTTPPort() string
|
||||||
@@ -51,7 +50,6 @@ func MustLoad() (Config, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (c *config) Domain() string { return c.domain }
|
func (c *config) Domain() string { return c.domain }
|
||||||
func (c *config) FrontendURL() string { return c.frontendURL }
|
|
||||||
func (c *config) SSHPort() string { return c.sshPort }
|
func (c *config) SSHPort() string { return c.sshPort }
|
||||||
func (c *config) HTTPPort() string { return c.httpPort }
|
func (c *config) HTTPPort() string { return c.httpPort }
|
||||||
func (c *config) HTTPSPort() string { return c.httpsPort }
|
func (c *config) HTTPSPort() string { return c.httpsPort }
|
||||||
|
|||||||
@@ -12,9 +12,8 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type config struct {
|
type config struct {
|
||||||
domain string
|
domain string
|
||||||
frontendURL string
|
sshPort string
|
||||||
sshPort string
|
|
||||||
|
|
||||||
httpPort string
|
httpPort string
|
||||||
httpsPort string
|
httpsPort string
|
||||||
@@ -50,7 +49,6 @@ func parse() (*config, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
domain := getenv("DOMAIN", "localhost")
|
domain := getenv("DOMAIN", "localhost")
|
||||||
frontendURL := getenv("FRONTEND_URL", "https://"+domain)
|
|
||||||
sshPort := getenv("PORT", "2200")
|
sshPort := getenv("PORT", "2200")
|
||||||
|
|
||||||
httpPort := getenv("HTTP_PORT", "8080")
|
httpPort := getenv("HTTP_PORT", "8080")
|
||||||
@@ -91,7 +89,6 @@ func parse() (*config, error) {
|
|||||||
|
|
||||||
return &config{
|
return &config{
|
||||||
domain: domain,
|
domain: domain,
|
||||||
frontendURL: frontendURL,
|
|
||||||
sshPort: sshPort,
|
sshPort: sshPort,
|
||||||
httpPort: httpPort,
|
httpPort: httpPort,
|
||||||
httpsPort: httpsPort,
|
httpsPort: httpsPort,
|
||||||
|
|||||||
@@ -13,6 +13,7 @@ import (
|
|||||||
"tunnel_pls/internal/session/slug"
|
"tunnel_pls/internal/session/slug"
|
||||||
"tunnel_pls/internal/types"
|
"tunnel_pls/internal/types"
|
||||||
|
|
||||||
|
"tunnel_pls/internal/port"
|
||||||
"tunnel_pls/internal/registry"
|
"tunnel_pls/internal/registry"
|
||||||
|
|
||||||
proto "git.fossy.my.id/bagas/tunnel-please-grpc/gen"
|
proto "git.fossy.my.id/bagas/tunnel-please-grpc/gen"
|
||||||
@@ -753,7 +754,6 @@ type MockConfig struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (m *MockConfig) Domain() string { return m.Called().String(0) }
|
func (m *MockConfig) Domain() string { return m.Called().String(0) }
|
||||||
func (m *MockConfig) FrontendURL() string { return m.Called().String(0) }
|
|
||||||
func (m *MockConfig) SSHPort() string { return m.Called().String(0) }
|
func (m *MockConfig) SSHPort() string { return m.Called().String(0) }
|
||||||
func (m *MockConfig) HTTPPort() string { return m.Called().String(0) }
|
func (m *MockConfig) HTTPPort() string { return m.Called().String(0) }
|
||||||
func (m *MockConfig) HTTPSPort() string { return m.Called().String(0) }
|
func (m *MockConfig) HTTPSPort() string { return m.Called().String(0) }
|
||||||
@@ -885,16 +885,16 @@ func (m *mockLifecycle) Connection() ssh.Conn {
|
|||||||
return args.Get(0).(ssh.Conn)
|
return args.Get(0).(ssh.Conn)
|
||||||
}
|
}
|
||||||
func (m *mockLifecycle) User() string { return m.Called().String(0) }
|
func (m *mockLifecycle) User() string { return m.Called().String(0) }
|
||||||
func (m *mockLifecycle) SetChannel(channel ssh.Channel) error { return m.Called(channel).Error(0) }
|
func (m *mockLifecycle) SetChannel(channel ssh.Channel) { m.Called(channel) }
|
||||||
func (m *mockLifecycle) SetStatus(status types.SessionStatus) { m.Called(status) }
|
func (m *mockLifecycle) SetStatus(status types.SessionStatus) { m.Called(status) }
|
||||||
func (m *mockLifecycle) IsActive() bool { return m.Called().Bool(0) }
|
func (m *mockLifecycle) IsActive() bool { return m.Called().Bool(0) }
|
||||||
func (m *mockLifecycle) StartedAt() time.Time { return m.Called().Get(0).(time.Time) }
|
func (m *mockLifecycle) StartedAt() time.Time { return m.Called().Get(0).(time.Time) }
|
||||||
func (m *mockLifecycle) PortRegistry() lifecycle.PortRegistry {
|
func (m *mockLifecycle) PortRegistry() port.Port {
|
||||||
args := m.Called()
|
args := m.Called()
|
||||||
if args.Get(0) == nil {
|
if args.Get(0) == nil {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
return args.Get(0).(lifecycle.PortRegistry)
|
return args.Get(0).(port.Port)
|
||||||
}
|
}
|
||||||
|
|
||||||
type mockEventServiceClient struct {
|
type mockEventServiceClient struct {
|
||||||
|
|||||||
+11
-14
@@ -33,15 +33,10 @@ func (pm *port) AddRange(startPort, endPort uint16) error {
|
|||||||
if startPort > endPort {
|
if startPort > endPort {
|
||||||
return fmt.Errorf("start port cannot be greater than end port")
|
return fmt.Errorf("start port cannot be greater than end port")
|
||||||
}
|
}
|
||||||
for index := startPort; ; index++ {
|
for index := startPort; index <= endPort; index++ {
|
||||||
if index != 0 {
|
if _, exists := pm.ports[index]; !exists {
|
||||||
if _, exists := pm.ports[index]; !exists {
|
pm.ports[index] = false
|
||||||
pm.ports[index] = false
|
pm.sortedPorts = append(pm.sortedPorts, index)
|
||||||
pm.sortedPorts = append(pm.sortedPorts, index)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if index == endPort {
|
|
||||||
break
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
sort.Slice(pm.sortedPorts, func(i, j int) bool {
|
sort.Slice(pm.sortedPorts, func(i, j int) bool {
|
||||||
@@ -56,7 +51,6 @@ func (pm *port) Unassigned() (uint16, bool) {
|
|||||||
|
|
||||||
for _, index := range pm.sortedPorts {
|
for _, index := range pm.sortedPorts {
|
||||||
if !pm.ports[index] {
|
if !pm.ports[index] {
|
||||||
pm.ports[index] = true
|
|
||||||
return index, true
|
return index, true
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -67,9 +61,6 @@ func (pm *port) SetStatus(port uint16, assigned bool) error {
|
|||||||
pm.mu.Lock()
|
pm.mu.Lock()
|
||||||
defer pm.mu.Unlock()
|
defer pm.mu.Unlock()
|
||||||
|
|
||||||
if _, exists := pm.ports[port]; !exists {
|
|
||||||
return fmt.Errorf("port %d is not in the allowed range", port)
|
|
||||||
}
|
|
||||||
pm.ports[port] = assigned
|
pm.ports[port] = assigned
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
@@ -79,10 +70,16 @@ func (pm *port) Claim(port uint16) (claimed bool) {
|
|||||||
defer pm.mu.Unlock()
|
defer pm.mu.Unlock()
|
||||||
|
|
||||||
status, exists := pm.ports[port]
|
status, exists := pm.ports[port]
|
||||||
if !exists || status {
|
|
||||||
|
if exists && status {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if !exists {
|
||||||
|
pm.ports[port] = true
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
pm.ports[port] = true
|
pm.ports[port] = true
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -16,8 +16,6 @@ func TestAddRange(t *testing.T) {
|
|||||||
{"normal range", 1000, 1002, false},
|
{"normal range", 1000, 1002, false},
|
||||||
{"invalid range", 2000, 1999, true},
|
{"invalid range", 2000, 1999, true},
|
||||||
{"single port range", 3000, 3000, false},
|
{"single port range", 3000, 3000, false},
|
||||||
{"range ending at max uint16", 65533, 65535, false},
|
|
||||||
{"range including port zero", 0, 2, false},
|
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, tt := range tests {
|
for _, tt := range tests {
|
||||||
@@ -33,22 +31,6 @@ func TestAddRange(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestAddRangeBoundaries(t *testing.T) {
|
|
||||||
pm := New()
|
|
||||||
err := pm.AddRange(0, 3)
|
|
||||||
assert.NoError(t, err)
|
|
||||||
_, hasZero := pm.(*port).ports[0]
|
|
||||||
assert.False(t, hasZero, "port 0 must be skipped")
|
|
||||||
assert.Len(t, pm.(*port).ports, 3)
|
|
||||||
|
|
||||||
pm2 := New()
|
|
||||||
err = pm2.AddRange(65533, 65535)
|
|
||||||
assert.NoError(t, err)
|
|
||||||
assert.Len(t, pm2.(*port).ports, 3)
|
|
||||||
_, hasMax := pm2.(*port).ports[65535]
|
|
||||||
assert.True(t, hasMax)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestUnassigned(t *testing.T) {
|
func TestUnassigned(t *testing.T) {
|
||||||
pm := New()
|
pm := New()
|
||||||
_ = pm.AddRange(1000, 1002)
|
_ = pm.AddRange(1000, 1002)
|
||||||
@@ -76,21 +58,6 @@ func TestUnassigned(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestUnassignedReservesPort(t *testing.T) {
|
|
||||||
pm := New()
|
|
||||||
_ = pm.AddRange(1000, 1002)
|
|
||||||
|
|
||||||
p1, ok1 := pm.Unassigned()
|
|
||||||
assert.True(t, ok1)
|
|
||||||
assert.Equal(t, uint16(1000), p1)
|
|
||||||
|
|
||||||
p2, ok2 := pm.Unassigned()
|
|
||||||
assert.True(t, ok2)
|
|
||||||
assert.Equal(t, uint16(1001), p2)
|
|
||||||
|
|
||||||
assert.True(t, pm.(*port).ports[1000], "Unassigned must reserve the port")
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestSetStatus(t *testing.T) {
|
func TestSetStatus(t *testing.T) {
|
||||||
pm := New()
|
pm := New()
|
||||||
_ = pm.AddRange(1000, 1002)
|
_ = pm.AddRange(1000, 1002)
|
||||||
@@ -116,17 +83,6 @@ func TestSetStatus(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestSetStatusUnknownPort(t *testing.T) {
|
|
||||||
pm := New()
|
|
||||||
_ = pm.AddRange(1000, 1002)
|
|
||||||
|
|
||||||
err := pm.SetStatus(5000, true)
|
|
||||||
assert.Error(t, err)
|
|
||||||
|
|
||||||
_, exists := pm.(*port).ports[5000]
|
|
||||||
assert.False(t, exists, "SetStatus must not create entries for unknown ports")
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestClaim(t *testing.T) {
|
func TestClaim(t *testing.T) {
|
||||||
pm := New()
|
pm := New()
|
||||||
_ = pm.AddRange(1000, 1002)
|
_ = pm.AddRange(1000, 1002)
|
||||||
@@ -139,7 +95,7 @@ func TestClaim(t *testing.T) {
|
|||||||
}{
|
}{
|
||||||
{"claim unassigned port", 1000, false, true},
|
{"claim unassigned port", 1000, false, true},
|
||||||
{"claim already assigned port", 1001, true, false},
|
{"claim already assigned port", 1001, true, false},
|
||||||
{"claim non-existent port", 5000, false, false},
|
{"claim non-existent port", 5000, false, true},
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, tt := range tests {
|
for _, tt := range tests {
|
||||||
@@ -151,13 +107,8 @@ func TestClaim(t *testing.T) {
|
|||||||
got := pm.Claim(tt.port)
|
got := pm.Claim(tt.port)
|
||||||
assert.Equal(t, tt.want, got)
|
assert.Equal(t, tt.want, got)
|
||||||
|
|
||||||
finalState, exists := pm.(*port).ports[tt.port]
|
finalState := pm.(*port).ports[tt.port]
|
||||||
if !tt.want && tt.port == 5000 {
|
assert.True(t, finalState)
|
||||||
assert.False(t, exists, "out-of-range port must not be added to the registry")
|
|
||||||
} else {
|
|
||||||
assert.True(t, exists)
|
|
||||||
assert.True(t, finalState)
|
|
||||||
}
|
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -94,13 +94,13 @@ func (r *registry) Update(user string, oldKey, newKey Key) error {
|
|||||||
return ErrInvalidSlug
|
return ErrInvalidSlug
|
||||||
}
|
}
|
||||||
|
|
||||||
r.mu.Lock()
|
|
||||||
defer r.mu.Unlock()
|
|
||||||
|
|
||||||
if _, exists := r.slugIndex[newKey]; exists && newKey != oldKey {
|
if _, exists := r.slugIndex[newKey]; exists && newKey != oldKey {
|
||||||
return ErrSlugInUse
|
return ErrSlugInUse
|
||||||
}
|
}
|
||||||
|
|
||||||
|
r.mu.Lock()
|
||||||
|
defer r.mu.Unlock()
|
||||||
|
|
||||||
client, ok := r.byUser[user][oldKey]
|
client, ok := r.byUser[user][oldKey]
|
||||||
if !ok {
|
if !ok {
|
||||||
return ErrSessionNotFound
|
return ErrSessionNotFound
|
||||||
|
|||||||
@@ -4,6 +4,7 @@ import (
|
|||||||
"sync"
|
"sync"
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
"tunnel_pls/internal/port"
|
||||||
"tunnel_pls/internal/session/forwarder"
|
"tunnel_pls/internal/session/forwarder"
|
||||||
"tunnel_pls/internal/session/interaction"
|
"tunnel_pls/internal/session/interaction"
|
||||||
"tunnel_pls/internal/session/lifecycle"
|
"tunnel_pls/internal/session/lifecycle"
|
||||||
@@ -77,15 +78,15 @@ func (ml *mockLifecycle) Connection() ssh.Conn {
|
|||||||
return args.Get(0).(ssh.Conn)
|
return args.Get(0).(ssh.Conn)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ml *mockLifecycle) PortRegistry() lifecycle.PortRegistry {
|
func (ml *mockLifecycle) PortRegistry() port.Port {
|
||||||
args := ml.Called()
|
args := ml.Called()
|
||||||
if args.Get(0) == nil {
|
if args.Get(0) == nil {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
return args.Get(0).(lifecycle.PortRegistry)
|
return args.Get(0).(port.Port)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ml *mockLifecycle) SetChannel(channel ssh.Channel) error { return ml.Called(channel).Error(0) }
|
func (ml *mockLifecycle) SetChannel(channel ssh.Channel) { ml.Called(channel) }
|
||||||
func (ml *mockLifecycle) SetStatus(status types.SessionStatus) { ml.Called(status) }
|
func (ml *mockLifecycle) SetStatus(status types.SessionStatus) { ml.Called(status) }
|
||||||
func (ml *mockLifecycle) IsActive() bool { return ml.Called().Bool(0) }
|
func (ml *mockLifecycle) IsActive() bool { return ml.Called().Bool(0) }
|
||||||
func (ml *mockLifecycle) StartedAt() time.Time { return ml.Called().Get(0).(time.Time) }
|
func (ml *mockLifecycle) StartedAt() time.Time { return ml.Called().Get(0).(time.Time) }
|
||||||
|
|||||||
@@ -33,7 +33,6 @@ type MockConfig struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (m *MockConfig) Domain() string { return m.Called().String(0) }
|
func (m *MockConfig) Domain() string { return m.Called().String(0) }
|
||||||
func (m *MockConfig) FrontendURL() string { return m.Called().String(0) }
|
|
||||||
func (m *MockConfig) SSHPort() string { return m.Called().String(0) }
|
func (m *MockConfig) SSHPort() string { return m.Called().String(0) }
|
||||||
func (m *MockConfig) HTTPPort() string { return m.Called().String(0) }
|
func (m *MockConfig) HTTPPort() string { return m.Called().String(0) }
|
||||||
func (m *MockConfig) HTTPSPort() string { return m.Called().String(0) }
|
func (m *MockConfig) HTTPSPort() string { return m.Called().String(0) }
|
||||||
|
|||||||
@@ -28,7 +28,6 @@ type Forwarder interface {
|
|||||||
Close() error
|
Close() error
|
||||||
}
|
}
|
||||||
type forwarder struct {
|
type forwarder struct {
|
||||||
mu sync.RWMutex
|
|
||||||
listener net.Listener
|
listener net.Listener
|
||||||
tunnelType types.TunnelType
|
tunnelType types.TunnelType
|
||||||
forwardedPort uint16
|
forwardedPort uint16
|
||||||
@@ -61,7 +60,7 @@ func (f *forwarder) copyWithBuffer(dst io.Writer, src io.Reader) (written int64,
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (f *forwarder) OpenForwardedChannel(ctx context.Context, origin net.Addr) (ssh.Channel, <-chan *ssh.Request, error) {
|
func (f *forwarder) OpenForwardedChannel(ctx context.Context, origin net.Addr) (ssh.Channel, <-chan *ssh.Request, error) {
|
||||||
payload := createForwardedTCPIPPayload(origin, f.ForwardedPort())
|
payload := createForwardedTCPIPPayload(origin, f.forwardedPort)
|
||||||
type channelResult struct {
|
type channelResult struct {
|
||||||
channel ssh.Channel
|
channel ssh.Channel
|
||||||
reqs <-chan *ssh.Request
|
reqs <-chan *ssh.Request
|
||||||
@@ -142,44 +141,32 @@ func (f *forwarder) HandleConnection(dst io.ReadWriter, src ssh.Channel) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (f *forwarder) SetType(tunnelType types.TunnelType) {
|
func (f *forwarder) SetType(tunnelType types.TunnelType) {
|
||||||
f.mu.Lock()
|
|
||||||
defer f.mu.Unlock()
|
|
||||||
f.tunnelType = tunnelType
|
f.tunnelType = tunnelType
|
||||||
}
|
}
|
||||||
|
|
||||||
func (f *forwarder) TunnelType() types.TunnelType {
|
func (f *forwarder) TunnelType() types.TunnelType {
|
||||||
f.mu.RLock()
|
|
||||||
defer f.mu.RUnlock()
|
|
||||||
return f.tunnelType
|
return f.tunnelType
|
||||||
}
|
}
|
||||||
|
|
||||||
func (f *forwarder) ForwardedPort() uint16 {
|
func (f *forwarder) ForwardedPort() uint16 {
|
||||||
f.mu.RLock()
|
|
||||||
defer f.mu.RUnlock()
|
|
||||||
return f.forwardedPort
|
return f.forwardedPort
|
||||||
}
|
}
|
||||||
|
|
||||||
func (f *forwarder) SetForwardedPort(port uint16) {
|
func (f *forwarder) SetForwardedPort(port uint16) {
|
||||||
f.mu.Lock()
|
|
||||||
defer f.mu.Unlock()
|
|
||||||
f.forwardedPort = port
|
f.forwardedPort = port
|
||||||
}
|
}
|
||||||
|
|
||||||
func (f *forwarder) SetListener(listener net.Listener) {
|
func (f *forwarder) SetListener(listener net.Listener) {
|
||||||
f.mu.Lock()
|
|
||||||
defer f.mu.Unlock()
|
|
||||||
f.listener = listener
|
f.listener = listener
|
||||||
}
|
}
|
||||||
|
|
||||||
func (f *forwarder) Listener() net.Listener {
|
func (f *forwarder) Listener() net.Listener {
|
||||||
f.mu.RLock()
|
|
||||||
defer f.mu.RUnlock()
|
|
||||||
return f.listener
|
return f.listener
|
||||||
}
|
}
|
||||||
|
|
||||||
func (f *forwarder) Close() error {
|
func (f *forwarder) Close() error {
|
||||||
if listener := f.Listener(); listener != nil {
|
if f.Listener() != nil {
|
||||||
return listener.Close()
|
return f.listener.Close()
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -24,7 +24,6 @@ type mockConfig struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (m *mockConfig) Domain() string { return m.Called().String(0) }
|
func (m *mockConfig) Domain() string { return m.Called().String(0) }
|
||||||
func (m *mockConfig) FrontendURL() string { return m.Called().String(0) }
|
|
||||||
func (m *mockConfig) SSHPort() string { return m.Called().String(0) }
|
func (m *mockConfig) SSHPort() string { return m.Called().String(0) }
|
||||||
func (m *mockConfig) HTTPPort() string { return m.Called().String(0) }
|
func (m *mockConfig) HTTPPort() string { return m.Called().String(0) }
|
||||||
func (m *mockConfig) HTTPSPort() string { return m.Called().String(0) }
|
func (m *mockConfig) HTTPSPort() string { return m.Called().String(0) }
|
||||||
|
|||||||
@@ -32,7 +32,6 @@ type MockConfig struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (m *MockConfig) Domain() string { return m.Called().String(0) }
|
func (m *MockConfig) Domain() string { return m.Called().String(0) }
|
||||||
func (m *MockConfig) FrontendURL() string { return m.Called().String(0) }
|
|
||||||
func (m *MockConfig) SSHPort() string { return m.Called().String(0) }
|
func (m *MockConfig) SSHPort() string { return m.Called().String(0) }
|
||||||
func (m *MockConfig) HTTPPort() string { return m.Called().String(0) }
|
func (m *MockConfig) HTTPPort() string { return m.Called().String(0) }
|
||||||
func (m *MockConfig) HTTPSPort() string { return m.Called().String(0) }
|
func (m *MockConfig) HTTPSPort() string { return m.Called().String(0) }
|
||||||
@@ -1923,6 +1922,10 @@ func TestInteraction_Start_ProtocolSelection(t *testing.T) {
|
|||||||
time.Sleep(50 * time.Millisecond)
|
time.Sleep(50 * time.Millisecond)
|
||||||
|
|
||||||
i := mockInteraction.(*interaction)
|
i := mockInteraction.(*interaction)
|
||||||
|
if i.program != nil {
|
||||||
|
assert.NotNil(t, i.program, "program should be initialized")
|
||||||
|
}
|
||||||
|
|
||||||
i.Stop()
|
i.Stop()
|
||||||
|
|
||||||
mockConfig.AssertExpectations(t)
|
mockConfig.AssertExpectations(t)
|
||||||
|
|||||||
@@ -2,7 +2,6 @@ package lifecycle
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
|
||||||
"io"
|
"io"
|
||||||
"net"
|
"net"
|
||||||
"sync"
|
"sync"
|
||||||
@@ -10,6 +9,8 @@ import (
|
|||||||
"tunnel_pls/internal/session/slug"
|
"tunnel_pls/internal/session/slug"
|
||||||
"tunnel_pls/internal/types"
|
"tunnel_pls/internal/types"
|
||||||
|
|
||||||
|
portUtil "tunnel_pls/internal/port"
|
||||||
|
|
||||||
"golang.org/x/crypto/ssh"
|
"golang.org/x/crypto/ssh"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -23,12 +24,6 @@ type SessionRegistry interface {
|
|||||||
Remove(key types.SessionKey)
|
Remove(key types.SessionKey)
|
||||||
}
|
}
|
||||||
|
|
||||||
type PortRegistry interface {
|
|
||||||
Unassigned() (uint16, bool)
|
|
||||||
Claim(port uint16) bool
|
|
||||||
SetStatus(port uint16, assigned bool) error
|
|
||||||
}
|
|
||||||
|
|
||||||
type lifecycle struct {
|
type lifecycle struct {
|
||||||
mu sync.Mutex
|
mu sync.Mutex
|
||||||
status types.SessionStatus
|
status types.SessionStatus
|
||||||
@@ -39,18 +34,18 @@ type lifecycle struct {
|
|||||||
slug slug.Slug
|
slug slug.Slug
|
||||||
startedAt time.Time
|
startedAt time.Time
|
||||||
sessionRegistry SessionRegistry
|
sessionRegistry SessionRegistry
|
||||||
portRegistry PortRegistry
|
portRegistry portUtil.Port
|
||||||
user string
|
user string
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(conn ssh.Conn, forwarder Forwarder, slugManager slug.Slug, port PortRegistry, sessionRegistry SessionRegistry, user string) Lifecycle {
|
func New(conn ssh.Conn, forwarder Forwarder, slugManager slug.Slug, port portUtil.Port, sessionRegistry SessionRegistry, user string) Lifecycle {
|
||||||
return &lifecycle{
|
return &lifecycle{
|
||||||
status: types.SessionStatusINITIALIZING,
|
status: types.SessionStatusINITIALIZING,
|
||||||
conn: conn,
|
conn: conn,
|
||||||
channel: nil,
|
channel: nil,
|
||||||
forwarder: forwarder,
|
forwarder: forwarder,
|
||||||
slug: slugManager,
|
slug: slugManager,
|
||||||
startedAt: time.Time{},
|
startedAt: time.Now(),
|
||||||
sessionRegistry: sessionRegistry,
|
sessionRegistry: sessionRegistry,
|
||||||
portRegistry: port,
|
portRegistry: port,
|
||||||
user: user,
|
user: user,
|
||||||
@@ -60,16 +55,16 @@ func New(conn ssh.Conn, forwarder Forwarder, slugManager slug.Slug, port PortReg
|
|||||||
type Lifecycle interface {
|
type Lifecycle interface {
|
||||||
Connection() ssh.Conn
|
Connection() ssh.Conn
|
||||||
Channel() ssh.Channel
|
Channel() ssh.Channel
|
||||||
PortRegistry() PortRegistry
|
PortRegistry() portUtil.Port
|
||||||
User() string
|
User() string
|
||||||
SetChannel(channel ssh.Channel) error
|
SetChannel(channel ssh.Channel)
|
||||||
SetStatus(status types.SessionStatus)
|
SetStatus(status types.SessionStatus)
|
||||||
IsActive() bool
|
IsActive() bool
|
||||||
StartedAt() time.Time
|
StartedAt() time.Time
|
||||||
Close() error
|
Close() error
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *lifecycle) PortRegistry() PortRegistry {
|
func (l *lifecycle) PortRegistry() portUtil.Port {
|
||||||
return l.portRegistry
|
return l.portRegistry
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -77,25 +72,11 @@ func (l *lifecycle) User() string {
|
|||||||
return l.user
|
return l.user
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *lifecycle) SetChannel(channel ssh.Channel) error {
|
func (l *lifecycle) SetChannel(channel ssh.Channel) {
|
||||||
l.mu.Lock()
|
|
||||||
defer l.mu.Unlock()
|
|
||||||
if l.status == types.SessionStatusCLOSED {
|
|
||||||
return fmt.Errorf("lifecycle is closed")
|
|
||||||
}
|
|
||||||
if channel == nil {
|
|
||||||
return fmt.Errorf("channel cannot be nil")
|
|
||||||
}
|
|
||||||
if l.channel != nil {
|
|
||||||
return fmt.Errorf("channel already set")
|
|
||||||
}
|
|
||||||
l.channel = channel
|
l.channel = channel
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *lifecycle) Channel() ssh.Channel {
|
func (l *lifecycle) Channel() ssh.Channel {
|
||||||
l.mu.Lock()
|
|
||||||
defer l.mu.Unlock()
|
|
||||||
return l.channel
|
return l.channel
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -106,13 +87,7 @@ func (l *lifecycle) Connection() ssh.Conn {
|
|||||||
func (l *lifecycle) SetStatus(status types.SessionStatus) {
|
func (l *lifecycle) SetStatus(status types.SessionStatus) {
|
||||||
l.mu.Lock()
|
l.mu.Lock()
|
||||||
defer l.mu.Unlock()
|
defer l.mu.Unlock()
|
||||||
if l.status == types.SessionStatusCLOSED {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
l.status = status
|
l.status = status
|
||||||
if status == types.SessionStatusRUNNING && l.startedAt.IsZero() {
|
|
||||||
l.startedAt = time.Now()
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *lifecycle) IsActive() bool {
|
func (l *lifecycle) IsActive() bool {
|
||||||
@@ -123,74 +98,50 @@ func (l *lifecycle) IsActive() bool {
|
|||||||
|
|
||||||
func (l *lifecycle) Close() error {
|
func (l *lifecycle) Close() error {
|
||||||
l.mu.Lock()
|
l.mu.Lock()
|
||||||
|
defer l.mu.Unlock()
|
||||||
if l.status == types.SessionStatusCLOSED {
|
if l.status == types.SessionStatusCLOSED {
|
||||||
closeErr := l.closeErr
|
return l.closeErr
|
||||||
l.mu.Unlock()
|
|
||||||
return closeErr
|
|
||||||
}
|
}
|
||||||
l.status = types.SessionStatusCLOSED
|
l.status = types.SessionStatusCLOSED
|
||||||
|
|
||||||
channel := l.channel
|
|
||||||
conn := l.conn
|
|
||||||
l.mu.Unlock()
|
|
||||||
|
|
||||||
var errs []error
|
var errs []error
|
||||||
if channel != nil {
|
tunnelType := l.forwarder.TunnelType()
|
||||||
if err := channel.Close(); err != nil && !isClosedError(err) {
|
|
||||||
errs = append(errs, err)
|
if l.channel != nil {
|
||||||
}
|
if err := l.channel.Close(); err != nil && !isClosedError(err) {
|
||||||
}
|
|
||||||
if conn != nil {
|
|
||||||
if err := conn.Close(); err != nil && !isClosedError(err) {
|
|
||||||
errs = append(errs, err)
|
errs = append(errs, err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
l.cleanupRegistry()
|
if l.conn != nil {
|
||||||
if err := l.cleanupForwarder(); err != nil {
|
if err := l.conn.Close(); err != nil && !isClosedError(err) {
|
||||||
errs = append(errs, err)
|
errs = append(errs, err)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
closeErr := errors.Join(errs...)
|
clientSlug := l.slug.String()
|
||||||
|
|
||||||
l.mu.Lock()
|
|
||||||
l.closeErr = closeErr
|
|
||||||
l.mu.Unlock()
|
|
||||||
|
|
||||||
return closeErr
|
|
||||||
}
|
|
||||||
|
|
||||||
func (l *lifecycle) cleanupRegistry() {
|
|
||||||
slugStr := l.slug.String()
|
|
||||||
if slugStr == "" {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
key := types.SessionKey{
|
key := types.SessionKey{
|
||||||
Id: slugStr,
|
Id: clientSlug,
|
||||||
Type: l.forwarder.TunnelType(),
|
Type: tunnelType,
|
||||||
}
|
}
|
||||||
l.sessionRegistry.Remove(key)
|
l.sessionRegistry.Remove(key)
|
||||||
}
|
|
||||||
|
|
||||||
func (l *lifecycle) cleanupForwarder() error {
|
if tunnelType == types.TunnelTypeTCP {
|
||||||
if l.forwarder.TunnelType() != types.TunnelTypeTCP {
|
errs = append(errs, l.PortRegistry().SetStatus(l.forwarder.ForwardedPort(), false))
|
||||||
return nil
|
errs = append(errs, l.forwarder.Close())
|
||||||
}
|
}
|
||||||
var errs []error
|
|
||||||
errs = append(errs, l.portRegistry.SetStatus(l.forwarder.ForwardedPort(), false))
|
l.closeErr = errors.Join(errs...)
|
||||||
errs = append(errs, l.forwarder.Close())
|
return l.closeErr
|
||||||
return errors.Join(errs...)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func isClosedError(err error) bool {
|
func isClosedError(err error) bool {
|
||||||
if err == nil {
|
if err == nil {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
return errors.Is(err, io.EOF) || errors.Is(err, net.ErrClosed)
|
return errors.Is(err, io.EOF) || errors.Is(err, net.ErrClosed) || err.Error() == "EOF"
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *lifecycle) StartedAt() time.Time {
|
func (l *lifecycle) StartedAt() time.Time {
|
||||||
l.mu.Lock()
|
|
||||||
defer l.mu.Unlock()
|
|
||||||
return l.startedAt
|
return l.startedAt
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -5,7 +5,6 @@ import (
|
|||||||
"errors"
|
"errors"
|
||||||
"io"
|
"io"
|
||||||
"net"
|
"net"
|
||||||
"sync"
|
|
||||||
"testing"
|
"testing"
|
||||||
"tunnel_pls/internal/types"
|
"tunnel_pls/internal/types"
|
||||||
|
|
||||||
@@ -178,14 +177,8 @@ func TestLifecycle_SetChannel(t *testing.T) {
|
|||||||
|
|
||||||
mockSSHChannel := &MockSSHChannel{}
|
mockSSHChannel := &MockSSHChannel{}
|
||||||
|
|
||||||
err := mockLifecycle.SetChannel(mockSSHChannel)
|
mockLifecycle.SetChannel(mockSSHChannel)
|
||||||
assert.NoError(t, err)
|
|
||||||
assert.Equal(t, mockSSHChannel, mockLifecycle.Channel())
|
|
||||||
|
|
||||||
anotherChannel := &MockSSHChannel{}
|
|
||||||
err = mockLifecycle.SetChannel(anotherChannel)
|
|
||||||
assert.Error(t, err)
|
|
||||||
assert.Contains(t, err.Error(), "channel already set")
|
|
||||||
assert.Equal(t, mockSSHChannel, mockLifecycle.Channel())
|
assert.Equal(t, mockSSHChannel, mockLifecycle.Channel())
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -283,15 +276,14 @@ func TestLifecycle_Close(t *testing.T) {
|
|||||||
mockLifecycle := New(mockSSHConn, mockForwarder, mockSlug, mockPort, mockSessionRegistry, "mas-fuad")
|
mockLifecycle := New(mockSSHConn, mockForwarder, mockSlug, mockPort, mockSessionRegistry, "mas-fuad")
|
||||||
|
|
||||||
mockLifecycle.SetStatus(types.SessionStatusRUNNING)
|
mockLifecycle.SetStatus(types.SessionStatusRUNNING)
|
||||||
err := mockLifecycle.SetChannel(mockSSHChannel)
|
mockLifecycle.SetChannel(mockSSHChannel)
|
||||||
assert.NoError(t, err)
|
|
||||||
|
|
||||||
if tt.alreadyClosed {
|
if tt.alreadyClosed {
|
||||||
err = mockLifecycle.Close()
|
err := mockLifecycle.Close()
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
err = mockLifecycle.Close()
|
err := mockLifecycle.Close()
|
||||||
|
|
||||||
if tt.expectErr {
|
if tt.expectErr {
|
||||||
assert.Error(t, err)
|
assert.Error(t, err)
|
||||||
@@ -309,116 +301,3 @@ func TestLifecycle_Close(t *testing.T) {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestLifecycle_ConcurrentClose(t *testing.T) {
|
|
||||||
mockSSHConn := &MockSSHConn{}
|
|
||||||
mockSSHConn.On("Close").Return(nil)
|
|
||||||
|
|
||||||
mockForwarder := &MockForwarder{}
|
|
||||||
mockForwarder.On("TunnelType").Return(types.TunnelTypeHTTP)
|
|
||||||
|
|
||||||
mockSlug := &MockSlug{}
|
|
||||||
mockSlug.On("String").Return("test-slug")
|
|
||||||
|
|
||||||
mockPort := &MockPort{}
|
|
||||||
|
|
||||||
mockSessionRegistry := &MockSessionRegistry{}
|
|
||||||
mockSessionRegistry.On("Remove", mock.Anything).Return()
|
|
||||||
|
|
||||||
mockSSHChannel := &MockSSHChannel{}
|
|
||||||
mockSSHChannel.On("Close").Return(nil)
|
|
||||||
|
|
||||||
mockLifecycle := New(mockSSHConn, mockForwarder, mockSlug, mockPort, mockSessionRegistry, "mas-fuad")
|
|
||||||
mockLifecycle.SetStatus(types.SessionStatusRUNNING)
|
|
||||||
err := mockLifecycle.SetChannel(mockSSHChannel)
|
|
||||||
assert.NoError(t, err)
|
|
||||||
|
|
||||||
const numGoroutines = 10
|
|
||||||
var wg sync.WaitGroup
|
|
||||||
errChan := make(chan error, numGoroutines)
|
|
||||||
|
|
||||||
for i := 0; i < numGoroutines; i++ {
|
|
||||||
wg.Add(1)
|
|
||||||
go func() {
|
|
||||||
defer wg.Done()
|
|
||||||
err := mockLifecycle.Close()
|
|
||||||
errChan <- err
|
|
||||||
}()
|
|
||||||
}
|
|
||||||
|
|
||||||
wg.Wait()
|
|
||||||
close(errChan)
|
|
||||||
|
|
||||||
for err := range errChan {
|
|
||||||
assert.NoError(t, err)
|
|
||||||
}
|
|
||||||
|
|
||||||
assert.False(t, mockLifecycle.IsActive())
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLifecycle_SetChannel_AfterClose(t *testing.T) {
|
|
||||||
mockSSHConn := new(MockSSHConn)
|
|
||||||
mockSSHConn.On("Close").Return(nil)
|
|
||||||
mockForwarder := &MockForwarder{}
|
|
||||||
mockForwarder.On("TunnelType").Return(types.TunnelTypeHTTP)
|
|
||||||
mockSlug := &MockSlug{}
|
|
||||||
mockSlug.On("String").Return("test-slug")
|
|
||||||
mockPort := &MockPort{}
|
|
||||||
mockSessionRegistry := &MockSessionRegistry{}
|
|
||||||
mockSessionRegistry.On("Remove", mock.Anything).Return()
|
|
||||||
mockSSHChannel := &MockSSHChannel{}
|
|
||||||
mockSSHChannel.On("Close").Return(nil)
|
|
||||||
|
|
||||||
mockLifecycle := New(mockSSHConn, mockForwarder, mockSlug, mockPort, mockSessionRegistry, "mas-fuad")
|
|
||||||
mockLifecycle.SetStatus(types.SessionStatusRUNNING)
|
|
||||||
err := mockLifecycle.SetChannel(mockSSHChannel)
|
|
||||||
assert.NoError(t, err)
|
|
||||||
|
|
||||||
err = mockLifecycle.Close()
|
|
||||||
assert.NoError(t, err)
|
|
||||||
|
|
||||||
anotherChannel := &MockSSHChannel{}
|
|
||||||
err = mockLifecycle.SetChannel(anotherChannel)
|
|
||||||
assert.Error(t, err)
|
|
||||||
assert.Contains(t, err.Error(), "lifecycle is closed")
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLifecycle_SetChannel_Nil(t *testing.T) {
|
|
||||||
mockSSHConn := new(MockSSHConn)
|
|
||||||
mockForwarder := &MockForwarder{}
|
|
||||||
mockSlug := &MockSlug{}
|
|
||||||
mockPort := &MockPort{}
|
|
||||||
mockSessionRegistry := &MockSessionRegistry{}
|
|
||||||
|
|
||||||
mockLifecycle := New(mockSSHConn, mockForwarder, mockSlug, mockPort, mockSessionRegistry, "mas-fuad")
|
|
||||||
|
|
||||||
err := mockLifecycle.SetChannel(nil)
|
|
||||||
assert.Error(t, err)
|
|
||||||
assert.Contains(t, err.Error(), "channel cannot be nil")
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLifecycle_SetStatus_AfterClose(t *testing.T) {
|
|
||||||
mockSSHConn := new(MockSSHConn)
|
|
||||||
mockSSHConn.On("Close").Return(nil)
|
|
||||||
mockForwarder := &MockForwarder{}
|
|
||||||
mockForwarder.On("TunnelType").Return(types.TunnelTypeHTTP)
|
|
||||||
mockSlug := &MockSlug{}
|
|
||||||
mockSlug.On("String").Return("test-slug")
|
|
||||||
mockPort := &MockPort{}
|
|
||||||
mockSessionRegistry := &MockSessionRegistry{}
|
|
||||||
mockSessionRegistry.On("Remove", mock.Anything).Return()
|
|
||||||
mockSSHChannel := &MockSSHChannel{}
|
|
||||||
mockSSHChannel.On("Close").Return(nil)
|
|
||||||
|
|
||||||
mockLifecycle := New(mockSSHConn, mockForwarder, mockSlug, mockPort, mockSessionRegistry, "mas-fuad")
|
|
||||||
mockLifecycle.SetStatus(types.SessionStatusRUNNING)
|
|
||||||
err := mockLifecycle.SetChannel(mockSSHChannel)
|
|
||||||
assert.NoError(t, err)
|
|
||||||
|
|
||||||
err = mockLifecycle.Close()
|
|
||||||
assert.NoError(t, err)
|
|
||||||
assert.False(t, mockLifecycle.IsActive())
|
|
||||||
|
|
||||||
mockLifecycle.SetStatus(types.SessionStatusRUNNING)
|
|
||||||
assert.False(t, mockLifecycle.IsActive(), "SetStatus should be ignored after Close")
|
|
||||||
}
|
|
||||||
|
|||||||
+32
-43
@@ -26,7 +26,7 @@ type Session interface {
|
|||||||
HandleGlobalRequest(ch <-chan *ssh.Request) error
|
HandleGlobalRequest(ch <-chan *ssh.Request) error
|
||||||
HandleTCPIPForward(req *ssh.Request) error
|
HandleTCPIPForward(req *ssh.Request) error
|
||||||
HandleHTTPForward(req *ssh.Request, port uint16) error
|
HandleHTTPForward(req *ssh.Request, port uint16) error
|
||||||
HandleTCPForward(req *ssh.Request, addr string, port uint16, reserved bool) error
|
HandleTCPForward(req *ssh.Request, addr string, port uint16) error
|
||||||
Lifecycle() lifecycle.Lifecycle
|
Lifecycle() lifecycle.Lifecycle
|
||||||
Interaction() interaction.Interaction
|
Interaction() interaction.Interaction
|
||||||
Forwarder() forwarder.Forwarder
|
Forwarder() forwarder.Forwarder
|
||||||
@@ -158,15 +158,13 @@ func (s *session) setupInteractiveMode(channel ssh.NewChannel) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
go func() {
|
go func() {
|
||||||
err := s.HandleGlobalRequest(reqs)
|
err = s.HandleGlobalRequest(reqs)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Printf("global request handler error: %v", err)
|
log.Printf("global request handler error: %v", err)
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
|
||||||
if err = s.lifecycle.SetChannel(ch); err != nil {
|
s.lifecycle.SetChannel(ch)
|
||||||
return err
|
|
||||||
}
|
|
||||||
s.interaction.SetChannel(ch)
|
s.interaction.SetChannel(ch)
|
||||||
s.interaction.SetMode(types.InteractiveModeINTERACTIVE)
|
s.interaction.SetMode(types.InteractiveModeINTERACTIVE)
|
||||||
|
|
||||||
@@ -200,22 +198,23 @@ func (s *session) waitForSessionEnd() error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (s *session) waitForTCPIPForward() *ssh.Request {
|
func (s *session) waitForTCPIPForward() *ssh.Request {
|
||||||
for {
|
select {
|
||||||
select {
|
case req, ok := <-s.initialReq:
|
||||||
case req, ok := <-s.initialReq:
|
if !ok {
|
||||||
if !ok {
|
log.Println("Forwarding request channel closed")
|
||||||
log.Println("Forwarding request channel closed")
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
if req.Type == "tcpip-forward" {
|
|
||||||
return req
|
|
||||||
}
|
|
||||||
log.Printf("Ignoring unexpected global request: %s", req.Type)
|
|
||||||
_ = req.Reply(false, nil)
|
|
||||||
case <-time.After(500 * time.Millisecond):
|
|
||||||
log.Println("No tcpip-forward request received within timeout")
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
if req.Type == "tcpip-forward" {
|
||||||
|
return req
|
||||||
|
}
|
||||||
|
if err := req.Reply(false, nil); err != nil {
|
||||||
|
log.Printf("Failed to reply to request: %v", err)
|
||||||
|
}
|
||||||
|
log.Printf("Expected tcpip-forward request, got: %s", req.Type)
|
||||||
|
return nil
|
||||||
|
case <-time.After(500 * time.Millisecond):
|
||||||
|
log.Println("No forwarding request received")
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -254,35 +253,35 @@ func (s *session) HandleGlobalRequest(GlobalRequest <-chan *ssh.Request) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *session) parseForwardPayload(payload []byte) (address string, port uint16, reserved bool, err error) {
|
func (s *session) parseForwardPayload(payload []byte) (address string, port uint16, err error) {
|
||||||
var forwardPayload struct {
|
var forwardPayload struct {
|
||||||
BindAddr string
|
BindAddr string
|
||||||
BindPort uint32
|
BindPort uint32
|
||||||
}
|
}
|
||||||
|
|
||||||
if err = ssh.Unmarshal(payload, &forwardPayload); err != nil {
|
if err = ssh.Unmarshal(payload, &forwardPayload); err != nil {
|
||||||
return "", 0, false, fmt.Errorf("failed to unmarshal forward payload: %w", err)
|
return "", 0, fmt.Errorf("failed to unmarshal forward payload: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if forwardPayload.BindPort > 65535 {
|
if forwardPayload.BindPort > 65535 {
|
||||||
return "", 0, false, fmt.Errorf("port is larger than allowed port of 65535")
|
return "", 0, fmt.Errorf("port is larger than allowed port of 65535")
|
||||||
}
|
}
|
||||||
|
|
||||||
port = uint16(forwardPayload.BindPort)
|
port = uint16(forwardPayload.BindPort)
|
||||||
|
|
||||||
if isBlockedPort(port) {
|
if isBlockedPort(port) {
|
||||||
return "", 0, false, fmt.Errorf("port is blocked")
|
return "", 0, fmt.Errorf("port is blocked")
|
||||||
}
|
}
|
||||||
|
|
||||||
if port == 0 {
|
if port == 0 {
|
||||||
unassigned, ok := s.lifecycle.PortRegistry().Unassigned()
|
unassigned, ok := s.lifecycle.PortRegistry().Unassigned()
|
||||||
if !ok {
|
if !ok {
|
||||||
return "", 0, false, fmt.Errorf("no available port")
|
return "", 0, fmt.Errorf("no available port")
|
||||||
}
|
}
|
||||||
return forwardPayload.BindAddr, unassigned, true, nil
|
return forwardPayload.BindAddr, unassigned, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
return forwardPayload.BindAddr, port, false, nil
|
return forwardPayload.BindAddr, port, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *session) denyForwardingRequest(req *ssh.Request, key *types.SessionKey, listener io.Closer, msg string) error {
|
func (s *session) denyForwardingRequest(req *ssh.Request, key *types.SessionKey, listener io.Closer, msg string) error {
|
||||||
@@ -326,7 +325,7 @@ func (s *session) finalizeForwarding(req *ssh.Request, portToBind uint16, listen
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (s *session) HandleTCPIPForward(req *ssh.Request) error {
|
func (s *session) HandleTCPIPForward(req *ssh.Request) error {
|
||||||
address, port, reserved, err := s.parseForwardPayload(req.Payload)
|
address, port, err := s.parseForwardPayload(req.Payload)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return s.denyForwardingRequest(req, nil, nil, fmt.Sprintf("cannot parse forwarded payload: %s", err.Error()))
|
return s.denyForwardingRequest(req, nil, nil, fmt.Sprintf("cannot parse forwarded payload: %s", err.Error()))
|
||||||
}
|
}
|
||||||
@@ -335,7 +334,7 @@ func (s *session) HandleTCPIPForward(req *ssh.Request) error {
|
|||||||
case 80, 443:
|
case 80, 443:
|
||||||
return s.HandleHTTPForward(req, port)
|
return s.HandleHTTPForward(req, port)
|
||||||
default:
|
default:
|
||||||
return s.HandleTCPForward(req, address, port, reserved)
|
return s.HandleTCPForward(req, address, port)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -356,40 +355,30 @@ func (s *session) HandleHTTPForward(req *ssh.Request, portToBind uint16) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *session) HandleTCPForward(req *ssh.Request, addr string, portToBind uint16, reserved bool) error {
|
func (s *session) HandleTCPForward(req *ssh.Request, addr string, portToBind uint16) error {
|
||||||
if !reserved {
|
if claimed := s.lifecycle.PortRegistry().Claim(portToBind); !claimed {
|
||||||
if claimed := s.lifecycle.PortRegistry().Claim(portToBind); !claimed {
|
return s.denyForwardingRequest(req, nil, nil, fmt.Sprintf("Port %d is already in use or restricted", portToBind))
|
||||||
return s.denyForwardingRequest(req, nil, nil, fmt.Sprintf("Port %d is already in use or restricted", portToBind))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
releasePort := func() {
|
|
||||||
if err := s.lifecycle.PortRegistry().SetStatus(portToBind, false); err != nil {
|
|
||||||
log.Printf("failed to release port %d: %v", portToBind, err)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
tcpServer := transport.NewTCPServer(portToBind, s.forwarder)
|
tcpServer := transport.NewTCPServer(portToBind, s.forwarder)
|
||||||
listener, err := tcpServer.Listen()
|
listener, err := tcpServer.Listen()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
releasePort()
|
|
||||||
return s.denyForwardingRequest(req, nil, listener, fmt.Sprintf("Port %d is already in use or restricted", portToBind))
|
return s.denyForwardingRequest(req, nil, listener, fmt.Sprintf("Port %d is already in use or restricted", portToBind))
|
||||||
}
|
}
|
||||||
|
|
||||||
key := types.SessionKey{Id: fmt.Sprintf("%d", portToBind), Type: types.TunnelTypeTCP}
|
key := types.SessionKey{Id: fmt.Sprintf("%d", portToBind), Type: types.TunnelTypeTCP}
|
||||||
if !s.registry.Register(key, s) {
|
if !s.registry.Register(key, s) {
|
||||||
releasePort()
|
|
||||||
return s.denyForwardingRequest(req, nil, listener, fmt.Sprintf("Failed to register TunnelTypeTCP client with id: %s", key.Id))
|
return s.denyForwardingRequest(req, nil, listener, fmt.Sprintf("Failed to register TunnelTypeTCP client with id: %s", key.Id))
|
||||||
}
|
}
|
||||||
|
|
||||||
err = s.finalizeForwarding(req, portToBind, listener, types.TunnelTypeTCP, key.Id)
|
err = s.finalizeForwarding(req, portToBind, listener, types.TunnelTypeTCP, key.Id)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
releasePort()
|
|
||||||
return s.denyForwardingRequest(req, &key, listener, fmt.Sprintf("Failed to finalize forwarding: %s", err))
|
return s.denyForwardingRequest(req, &key, listener, fmt.Sprintf("Failed to finalize forwarding: %s", err))
|
||||||
}
|
}
|
||||||
|
|
||||||
go func() {
|
go func() {
|
||||||
if err := tcpServer.Serve(listener); err != nil {
|
err = tcpServer.Serve(listener)
|
||||||
|
if err != nil {
|
||||||
log.Printf("Failed serving tcp server: %s\n", err)
|
log.Printf("Failed serving tcp server: %s\n", err)
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
|||||||
@@ -38,9 +38,8 @@ type mockConfig struct {
|
|||||||
config.Config
|
config.Config
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *mockConfig) Domain() string { return m.Called().String(0) }
|
func (m *mockConfig) Domain() string { return m.Called().String(0) }
|
||||||
func (m *mockConfig) FrontendURL() string { return m.Called().String(0) }
|
func (m *mockConfig) SSHPort() string { return m.Called().String(0) }
|
||||||
func (m *mockConfig) SSHPort() string { return m.Called().String(0) }
|
|
||||||
func (m *mockConfig) Mode() types.ServerMode {
|
func (m *mockConfig) Mode() types.ServerMode {
|
||||||
args := m.Called()
|
args := m.Called()
|
||||||
if args.Get(0) == nil {
|
if args.Get(0) == nil {
|
||||||
@@ -397,12 +396,6 @@ func TestHandleTCPIPForward_Table(t *testing.T) {
|
|||||||
err := s.HandleTCPIPForward(req)
|
err := s.HandleTCPIPForward(req)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.Equal(t, uint16(12345), s.forwarder.ForwardedPort())
|
assert.Equal(t, uint16(12345), s.forwarder.ForwardedPort())
|
||||||
|
|
||||||
defer func() {
|
|
||||||
if l := s.forwarder.Listener(); l != nil {
|
|
||||||
_ = l.Close()
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Invalid Payload", func(t *testing.T) {
|
t.Run("Invalid Payload", func(t *testing.T) {
|
||||||
@@ -706,7 +699,6 @@ func TestForwardingFailures(t *testing.T) {
|
|||||||
s, mRegistry, mPort, _, _, sReqs, cConn, cleanup := setup(t)
|
s, mRegistry, mPort, _, _, sReqs, cConn, cleanup := setup(t)
|
||||||
defer cleanup()
|
defer cleanup()
|
||||||
mPort.On("Claim", mock.Anything).Return(true)
|
mPort.On("Claim", mock.Anything).Return(true)
|
||||||
mPort.On("SetStatus", uint16(1234), false).Return(nil)
|
|
||||||
mRegistry.On("Register", mock.Anything, mock.Anything).Return(false)
|
mRegistry.On("Register", mock.Anything, mock.Anything).Return(false)
|
||||||
|
|
||||||
payload := make([]byte, 4+9+4)
|
payload := make([]byte, 4+9+4)
|
||||||
@@ -725,7 +717,7 @@ func TestForwardingFailures(t *testing.T) {
|
|||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Finalize Forwarding Failure", func(t *testing.T) {
|
t.Run("Finalize Forwarding Failure", func(t *testing.T) {
|
||||||
s, mRegistry, _, mRandom, sConn, sReqs, cConn, cleanup := setup(t)
|
s, mRegistry, _, mRandom, _, sReqs, cConn, cleanup := setup(t)
|
||||||
defer cleanup()
|
defer cleanup()
|
||||||
mRandom.On("String", 20).Return("test-slug", nil)
|
mRandom.On("String", 20).Return("test-slug", nil)
|
||||||
mRegistry.On("Register", mock.Anything, mock.Anything).Return(true)
|
mRegistry.On("Register", mock.Anything, mock.Anything).Return(true)
|
||||||
@@ -744,7 +736,7 @@ func TestForwardingFailures(t *testing.T) {
|
|||||||
err := cConn.Close()
|
err := cConn.Close()
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
|
|
||||||
_ = sConn.Wait()
|
time.Sleep(50 * time.Millisecond)
|
||||||
|
|
||||||
err = s.HandleTCPIPForward(req)
|
err = s.HandleTCPIPForward(req)
|
||||||
assert.Error(t, err)
|
assert.Error(t, err)
|
||||||
@@ -766,7 +758,6 @@ func TestForwardingFailures(t *testing.T) {
|
|||||||
}(l)
|
}(l)
|
||||||
_, portStr, _ := net.SplitHostPort(l.Addr().String())
|
_, portStr, _ := net.SplitHostPort(l.Addr().String())
|
||||||
port, _ := strconv.Atoi(portStr)
|
port, _ := strconv.Atoi(portStr)
|
||||||
mPort.On("SetStatus", uint16(port), false).Return(nil)
|
|
||||||
|
|
||||||
payload := make([]byte, 4+9+4)
|
payload := make([]byte, 4+9+4)
|
||||||
binary.BigEndian.PutUint32(payload[0:4], 9)
|
binary.BigEndian.PutUint32(payload[0:4], 9)
|
||||||
@@ -816,7 +807,7 @@ func (m *mockNewChanFail) Accept() (ssh.Channel, <-chan *ssh.Request, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func TestWaitForTCPIPForward_EdgeCases(t *testing.T) {
|
func TestWaitForTCPIPForward_EdgeCases(t *testing.T) {
|
||||||
t.Run("Wrong Request Type Then Timeout", func(t *testing.T) {
|
t.Run("Wrong Request Type", func(t *testing.T) {
|
||||||
_, sReqs, _, cConn, cleanup := setupSSH(t)
|
_, sReqs, _, cConn, cleanup := setupSSH(t)
|
||||||
defer cleanup()
|
defer cleanup()
|
||||||
|
|
||||||
@@ -826,65 +817,10 @@ func TestWaitForTCPIPForward_EdgeCases(t *testing.T) {
|
|||||||
_, _, _ = cConn.SendRequest("not-tcpip-forward", true, nil)
|
_, _, _ = cConn.SendRequest("not-tcpip-forward", true, nil)
|
||||||
}()
|
}()
|
||||||
|
|
||||||
start := time.Now()
|
|
||||||
req := s.waitForTCPIPForward()
|
req := s.waitForTCPIPForward()
|
||||||
elapsed := time.Since(start)
|
|
||||||
|
|
||||||
if req != nil {
|
if req != nil {
|
||||||
t.Error("expected nil request")
|
t.Error("expected nil request")
|
||||||
}
|
}
|
||||||
if elapsed < 400*time.Millisecond {
|
|
||||||
t.Errorf("expected timeout ~500ms, got %v", elapsed)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("Multiple Non-Forward Requests Then Success", func(t *testing.T) {
|
|
||||||
_, sReqs, _, cConn, cleanup := setupSSH(t)
|
|
||||||
defer cleanup()
|
|
||||||
|
|
||||||
s := &session{initialReq: sReqs}
|
|
||||||
|
|
||||||
go func() {
|
|
||||||
time.Sleep(100 * time.Millisecond)
|
|
||||||
_, _, _ = cConn.SendRequest("keepalive@openssh.com", false, nil)
|
|
||||||
time.Sleep(100 * time.Millisecond)
|
|
||||||
_, _, _ = cConn.SendRequest("hostkeys-00@openssh.com", false, nil)
|
|
||||||
time.Sleep(100 * time.Millisecond)
|
|
||||||
_, _, _ = cConn.SendRequest("tcpip-forward", true, nil)
|
|
||||||
}()
|
|
||||||
|
|
||||||
req := s.waitForTCPIPForward()
|
|
||||||
if req == nil {
|
|
||||||
t.Error("expected tcpip-forward request, got nil")
|
|
||||||
}
|
|
||||||
if req != nil && req.Type != "tcpip-forward" {
|
|
||||||
t.Errorf("expected tcpip-forward, got %s", req.Type)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("Timeout After Non-Forward Requests", func(t *testing.T) {
|
|
||||||
_, sReqs, _, cConn, cleanup := setupSSH(t)
|
|
||||||
defer cleanup()
|
|
||||||
|
|
||||||
s := &session{initialReq: sReqs}
|
|
||||||
|
|
||||||
go func() {
|
|
||||||
time.Sleep(100 * time.Millisecond)
|
|
||||||
_, _, _ = cConn.SendRequest("keepalive@openssh.com", false, nil)
|
|
||||||
time.Sleep(100 * time.Millisecond)
|
|
||||||
_, _, _ = cConn.SendRequest("hostkeys-00@openssh.com", false, nil)
|
|
||||||
}()
|
|
||||||
|
|
||||||
start := time.Now()
|
|
||||||
req := s.waitForTCPIPForward()
|
|
||||||
elapsed := time.Since(start)
|
|
||||||
|
|
||||||
if req != nil {
|
|
||||||
t.Error("expected nil request after timeout")
|
|
||||||
}
|
|
||||||
if elapsed < 400*time.Millisecond {
|
|
||||||
t.Errorf("expected timeout ~500ms after last request, got %v", elapsed)
|
|
||||||
}
|
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Channel Closed", func(t *testing.T) {
|
t.Run("Channel Closed", func(t *testing.T) {
|
||||||
@@ -1046,7 +982,7 @@ func TestParseForwardPayload_Errors(t *testing.T) {
|
|||||||
s := &session{}
|
s := &session{}
|
||||||
|
|
||||||
t.Run("Short Address", func(t *testing.T) {
|
t.Run("Short Address", func(t *testing.T) {
|
||||||
_, _, _, err := s.parseForwardPayload([]byte{0, 0, 0, 4})
|
_, _, err := s.parseForwardPayload([]byte{0, 0, 0, 4})
|
||||||
if err == nil {
|
if err == nil {
|
||||||
t.Error("expected error, got nil")
|
t.Error("expected error, got nil")
|
||||||
}
|
}
|
||||||
@@ -1054,7 +990,7 @@ func TestParseForwardPayload_Errors(t *testing.T) {
|
|||||||
|
|
||||||
t.Run("Short Port", func(t *testing.T) {
|
t.Run("Short Port", func(t *testing.T) {
|
||||||
payload := append([]byte{0, 0, 0, 4}, []byte("addr")...)
|
payload := append([]byte{0, 0, 0, 4}, []byte("addr")...)
|
||||||
_, _, _, err := s.parseForwardPayload(payload)
|
_, _, err := s.parseForwardPayload(payload)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
t.Error("expected error, got nil")
|
t.Error("expected error, got nil")
|
||||||
}
|
}
|
||||||
@@ -1065,7 +1001,7 @@ func TestParseForwardPayload_Errors(t *testing.T) {
|
|||||||
portBuf := make([]byte, 4)
|
portBuf := make([]byte, 4)
|
||||||
binary.BigEndian.PutUint32(portBuf, 22)
|
binary.BigEndian.PutUint32(portBuf, 22)
|
||||||
payload = append(payload, portBuf...)
|
payload = append(payload, portBuf...)
|
||||||
_, _, _, err := s.parseForwardPayload(payload)
|
_, _, err := s.parseForwardPayload(payload)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
t.Error("expected error, got nil")
|
t.Error("expected error, got nil")
|
||||||
} else if !strings.Contains(err.Error(), "port is block") {
|
} else if !strings.Contains(err.Error(), "port is block") {
|
||||||
@@ -1229,7 +1165,7 @@ func TestHandleTCPForward_Failures(t *testing.T) {
|
|||||||
s, _, mPort, _, sReqs, cConn, cleanup := setup(t)
|
s, _, mPort, _, sReqs, cConn, cleanup := setup(t)
|
||||||
defer cleanup()
|
defer cleanup()
|
||||||
mPort.On("Claim", mock.Anything).Return(false)
|
mPort.On("Claim", mock.Anything).Return(false)
|
||||||
err := s.HandleTCPForward(getReq(t, cConn, sReqs), "localhost", 1234, false)
|
err := s.HandleTCPForward(getReq(t, cConn, sReqs), "localhost", 1234)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
t.Error("expected error, got nil")
|
t.Error("expected error, got nil")
|
||||||
} else if !strings.Contains(err.Error(), "already in use") {
|
} else if !strings.Contains(err.Error(), "already in use") {
|
||||||
@@ -1250,50 +1186,44 @@ func TestHandleTCPForward_Failures(t *testing.T) {
|
|||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
}(l)
|
}(l)
|
||||||
port := uint16(l.Addr().(*net.TCPAddr).Port)
|
port := uint16(l.Addr().(*net.TCPAddr).Port)
|
||||||
mPort.On("SetStatus", port, false).Return(nil)
|
|
||||||
|
|
||||||
err = s.HandleTCPForward(getReq(t, cConn, sReqs), "localhost", port, false)
|
err = s.HandleTCPForward(getReq(t, cConn, sReqs), "localhost", port)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
t.Error("expected error, got nil")
|
t.Error("expected error, got nil")
|
||||||
} else if !strings.Contains(err.Error(), "already in use") {
|
} else if !strings.Contains(err.Error(), "already in use") {
|
||||||
t.Errorf("expected error to contain %q, got %q", "already in use", err.Error())
|
t.Errorf("expected error to contain %q, got %q", "already in use", err.Error())
|
||||||
}
|
}
|
||||||
mPort.AssertExpectations(t)
|
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Registry Register fail", func(t *testing.T) {
|
t.Run("Registry Register fail", func(t *testing.T) {
|
||||||
s, mRegistry, mPort, _, sReqs, cConn, cleanup := setup(t)
|
s, mRegistry, mPort, _, sReqs, cConn, cleanup := setup(t)
|
||||||
defer cleanup()
|
defer cleanup()
|
||||||
mPort.On("Claim", mock.Anything).Return(true)
|
mPort.On("Claim", mock.Anything).Return(true)
|
||||||
mPort.On("SetStatus", mock.AnythingOfType("uint16"), false).Return(nil)
|
|
||||||
mRegistry.On("Register", mock.Anything, mock.Anything).Return(false)
|
mRegistry.On("Register", mock.Anything, mock.Anything).Return(false)
|
||||||
err := s.HandleTCPForward(getReq(t, cConn, sReqs), "localhost", 0, false)
|
err := s.HandleTCPForward(getReq(t, cConn, sReqs), "localhost", 0)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
t.Error("expected error, got nil")
|
t.Error("expected error, got nil")
|
||||||
} else if !strings.Contains(err.Error(), "Failed to register") {
|
} else if !strings.Contains(err.Error(), "Failed to register") {
|
||||||
t.Errorf("expected error to contain %q, got %q", "Failed to register", err.Error())
|
t.Errorf("expected error to contain %q, got %q", "Failed to register", err.Error())
|
||||||
}
|
}
|
||||||
mPort.AssertExpectations(t)
|
|
||||||
})
|
})
|
||||||
|
|
||||||
t.Run("Finalize fail (Reply fail)", func(t *testing.T) {
|
t.Run("Finalize fail (Reply fail)", func(t *testing.T) {
|
||||||
s, mRegistry, mPort, sConn, sReqs, cConn, cleanup := setup(t)
|
s, mRegistry, mPort, _, sReqs, cConn, cleanup := setup(t)
|
||||||
defer cleanup()
|
defer cleanup()
|
||||||
mPort.On("Claim", mock.Anything).Return(true)
|
mPort.On("Claim", mock.Anything).Return(true)
|
||||||
mPort.On("SetStatus", mock.AnythingOfType("uint16"), false).Return(nil)
|
|
||||||
mRegistry.On("Register", mock.Anything, mock.Anything).Return(true)
|
mRegistry.On("Register", mock.Anything, mock.Anything).Return(true)
|
||||||
req := getReq(t, cConn, sReqs)
|
req := getReq(t, cConn, sReqs)
|
||||||
err := cConn.Close()
|
err := cConn.Close()
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
_ = sConn.Wait()
|
time.Sleep(100 * time.Millisecond)
|
||||||
|
|
||||||
err = s.HandleTCPForward(req, "localhost", 0, false)
|
err = s.HandleTCPForward(req, "localhost", 0)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
t.Error("expected error, got nil")
|
t.Error("expected error, got nil")
|
||||||
} else if !strings.Contains(err.Error(), "Failed to finalize forwarding") {
|
} else if !strings.Contains(err.Error(), "Failed to finalize forwarding") {
|
||||||
t.Errorf("expected error to contain %q, got %q", "Failed to finalize forwarding", err.Error())
|
t.Errorf("expected error to contain %q, got %q", "Failed to finalize forwarding", err.Error())
|
||||||
}
|
}
|
||||||
mPort.AssertExpectations(t)
|
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,14 +1,11 @@
|
|||||||
package slug
|
package slug
|
||||||
|
|
||||||
import "sync"
|
|
||||||
|
|
||||||
type Slug interface {
|
type Slug interface {
|
||||||
String() string
|
String() string
|
||||||
Set(slug string)
|
Set(slug string)
|
||||||
}
|
}
|
||||||
|
|
||||||
type slug struct {
|
type slug struct {
|
||||||
mu sync.RWMutex
|
|
||||||
slug string
|
slug string
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -19,13 +16,9 @@ func New() Slug {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (s *slug) String() string {
|
func (s *slug) String() string {
|
||||||
s.mu.RLock()
|
|
||||||
defer s.mu.RUnlock()
|
|
||||||
return s.slug
|
return s.slug
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *slug) Set(slug string) {
|
func (s *slug) Set(slug string) {
|
||||||
s.mu.Lock()
|
|
||||||
defer s.mu.Unlock()
|
|
||||||
s.slug = slug
|
s.slug = slug
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -55,7 +55,8 @@ func TestHTTPServer_Serve(t *testing.T) {
|
|||||||
|
|
||||||
go func() {
|
go func() {
|
||||||
time.Sleep(100 * time.Millisecond)
|
time.Sleep(100 * time.Millisecond)
|
||||||
_ = listener.Close()
|
err = listener.Close()
|
||||||
|
assert.NoError(t, err)
|
||||||
}()
|
}()
|
||||||
|
|
||||||
err = srv.Serve(listener)
|
err = srv.Serve(listener)
|
||||||
|
|||||||
@@ -1,7 +1,6 @@
|
|||||||
package transport
|
package transport
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bufio"
|
|
||||||
"bytes"
|
"bytes"
|
||||||
"context"
|
"context"
|
||||||
"errors"
|
"errors"
|
||||||
@@ -53,43 +52,25 @@ func (hh *httpHandler) badRequest(conn net.Conn) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func readHTTPHeader(br *bufio.Reader, limit int) ([]byte, error) {
|
|
||||||
var headerBuf []byte
|
|
||||||
for {
|
|
||||||
line, err := br.ReadSlice('\n')
|
|
||||||
headerBuf = append(headerBuf, line...)
|
|
||||||
if errors.Is(err, bufio.ErrBufferFull) {
|
|
||||||
if len(headerBuf) > limit {
|
|
||||||
return nil, fmt.Errorf("headers too large")
|
|
||||||
}
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
if bytes.HasSuffix(headerBuf, []byte("\r\n\r\n")) {
|
|
||||||
return headerBuf, nil
|
|
||||||
}
|
|
||||||
if len(headerBuf) > limit {
|
|
||||||
return nil, fmt.Errorf("headers too large")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (hh *httpHandler) Handler(conn net.Conn, isTLS bool) {
|
func (hh *httpHandler) Handler(conn net.Conn, isTLS bool) {
|
||||||
defer hh.closeConnection(conn)
|
defer hh.closeConnection(conn)
|
||||||
|
|
||||||
_ = conn.SetReadDeadline(time.Now().Add(10 * time.Second))
|
_ = conn.SetReadDeadline(time.Now().Add(10 * time.Second))
|
||||||
br := bufio.NewReaderSize(conn, hh.config.HeaderSize())
|
buf := make([]byte, hh.config.HeaderSize())
|
||||||
headerBuf, err := readHTTPHeader(br, hh.config.HeaderSize())
|
n, err := conn.Read(buf)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
_ = hh.badRequest(conn)
|
_ = hh.badRequest(conn)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if idx := bytes.Index(buf[:n], []byte("\r\n\r\n")); idx == -1 {
|
||||||
|
_ = hh.badRequest(conn)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
_ = conn.SetReadDeadline(time.Time{})
|
_ = conn.SetReadDeadline(time.Time{})
|
||||||
|
|
||||||
reqhf, err := header.NewRequest(headerBuf)
|
reqhf, err := header.NewRequest(buf[:n])
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Printf("Error creating request header: %v", err)
|
log.Printf("Error creating request header: %v", err)
|
||||||
_ = hh.badRequest(conn)
|
_ = hh.badRequest(conn)
|
||||||
@@ -116,11 +97,11 @@ func (hh *httpHandler) Handler(conn net.Conn, isTLS bool) {
|
|||||||
Type: types.TunnelTypeHTTP,
|
Type: types.TunnelTypeHTTP,
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
_ = hh.redirect(conn, http.StatusMovedPermanently, fmt.Sprintf("%s/tunnel-not-found?slug=%s\r\n", hh.config.FrontendURL(), slug))
|
_ = hh.redirect(conn, http.StatusMovedPermanently, fmt.Sprintf("https://tunnl.live/tunnel-not-found?slug=%s\r\n", slug))
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
hw := stream.New(conn, br, conn.RemoteAddr())
|
hw := stream.New(conn, conn, conn.RemoteAddr())
|
||||||
defer func(hw stream.HTTP) {
|
defer func(hw stream.HTTP) {
|
||||||
err = hw.Close()
|
err = hw.Close()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|||||||
@@ -223,7 +223,6 @@ func TestNewHTTPHandler(t *testing.T) {
|
|||||||
msr := new(MockSessionRegistry)
|
msr := new(MockSessionRegistry)
|
||||||
mockConfig := &MockConfig{}
|
mockConfig := &MockConfig{}
|
||||||
mockConfig.On("Domain").Return("domain")
|
mockConfig.On("Domain").Return("domain")
|
||||||
mockConfig.On("FrontendURL").Return("https://domain")
|
|
||||||
mockConfig.On("TLSRedirect").Return(false)
|
mockConfig.On("TLSRedirect").Return(false)
|
||||||
hh := newHTTPHandler(mockConfig, msr)
|
hh := newHTTPHandler(mockConfig, msr)
|
||||||
assert.NotNil(t, hh)
|
assert.NotNil(t, hh)
|
||||||
@@ -291,7 +290,7 @@ func TestHandler(t *testing.T) {
|
|||||||
isTLS: true,
|
isTLS: true,
|
||||||
redirectTLS: false,
|
redirectTLS: false,
|
||||||
request: []byte("GET / HTTP/1.1\r\nHost: test.domain\r\n\r\n"),
|
request: []byte("GET / HTTP/1.1\r\nHost: test.domain\r\n\r\n"),
|
||||||
expected: []byte("HTTP/1.1 301 Moved Permanently\r\nLocation: https://example.com/tunnel-not-found?slug=test\r\nContent-Length: 0\r\nConnection: close\r\n\r\n"),
|
expected: []byte("HTTP/1.1 301 Moved Permanently\r\nLocation: https://tunnl.live/tunnel-not-found?slug=test\r\nContent-Length: 0\r\nConnection: close\r\n\r\n"),
|
||||||
setupMocks: func(msr *MockSessionRegistry) {
|
setupMocks: func(msr *MockSessionRegistry) {
|
||||||
msr.On("Get", types.SessionKey{
|
msr.On("Get", types.SessionKey{
|
||||||
Id: "test",
|
Id: "test",
|
||||||
@@ -322,14 +321,8 @@ func TestHandler(t *testing.T) {
|
|||||||
isTLS: false,
|
isTLS: false,
|
||||||
redirectTLS: false,
|
redirectTLS: false,
|
||||||
request: []byte(""),
|
request: []byte(""),
|
||||||
expected: []byte(""),
|
expected: []byte("HTTP/1.1 400 Bad Request\r\n\r\n"),
|
||||||
setupConn: func() (net.Conn, net.Conn) {
|
setupMocks: func(msr *MockSessionRegistry) {
|
||||||
mc := new(MockConn)
|
|
||||||
mc.ReadBuffer = bytes.NewBuffer(nil)
|
|
||||||
mc.On("SetReadDeadline", mock.Anything).Return(nil)
|
|
||||||
mc.On("Write", []byte("HTTP/1.1 400 Bad Request\r\n\r\n")).Return(0, nil)
|
|
||||||
mc.On("Close").Return(nil)
|
|
||||||
return mc, nil
|
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
@@ -617,7 +610,6 @@ func TestHandler(t *testing.T) {
|
|||||||
mockConfig := &MockConfig{}
|
mockConfig := &MockConfig{}
|
||||||
port := "0"
|
port := "0"
|
||||||
mockConfig.On("Domain").Return("example.com")
|
mockConfig.On("Domain").Return("example.com")
|
||||||
mockConfig.On("FrontendURL").Return("https://example.com")
|
|
||||||
mockConfig.On("HTTPPort").Return(port)
|
mockConfig.On("HTTPPort").Return(port)
|
||||||
mockConfig.On("HeaderSize").Return(4096)
|
mockConfig.On("HeaderSize").Return(4096)
|
||||||
mockConfig.On("TLSRedirect").Return(true)
|
mockConfig.On("TLSRedirect").Return(true)
|
||||||
@@ -723,114 +715,3 @@ func TestHandler(t *testing.T) {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestHandlerForwardsPostBody(t *testing.T) {
|
|
||||||
mockSessionRegistry := new(MockSessionRegistry)
|
|
||||||
mockConfig := &MockConfig{}
|
|
||||||
mockConfig.On("Domain").Return("example.com")
|
|
||||||
mockConfig.On("FrontendURL").Return("https://example.com")
|
|
||||||
mockConfig.On("HTTPPort").Return("0")
|
|
||||||
mockConfig.On("HeaderSize").Return(4096)
|
|
||||||
mockConfig.On("TLSRedirect").Return(true)
|
|
||||||
hh := &httpHandler{
|
|
||||||
sessionRegistry: mockSessionRegistry,
|
|
||||||
config: mockConfig,
|
|
||||||
}
|
|
||||||
|
|
||||||
mockSession := new(MockSession)
|
|
||||||
mockForwarder := new(MockForwarder)
|
|
||||||
mockSSHChannel := new(MockSSHChannel)
|
|
||||||
|
|
||||||
mockSessionRegistry.On("Get", types.SessionKey{
|
|
||||||
Id: "test",
|
|
||||||
Type: types.TunnelTypeHTTP,
|
|
||||||
}).Return(mockSession, nil)
|
|
||||||
mockSession.On("Forwarder").Return(mockForwarder)
|
|
||||||
|
|
||||||
reqCh := make(chan *ssh.Request)
|
|
||||||
mockForwarder.On("OpenForwardedChannel", mock.Anything, mock.Anything).Return(mockSSHChannel, (<-chan *ssh.Request)(reqCh), nil)
|
|
||||||
|
|
||||||
var mu sync.Mutex
|
|
||||||
var capturedHeaders []byte
|
|
||||||
mockSSHChannel.On("Write", mock.Anything).Run(func(args mock.Arguments) {
|
|
||||||
mu.Lock()
|
|
||||||
capturedHeaders = append(capturedHeaders, args.Get(0).([]byte)...)
|
|
||||||
mu.Unlock()
|
|
||||||
}).Return(0, nil)
|
|
||||||
mockSSHChannel.On("Close").Return(nil)
|
|
||||||
|
|
||||||
bodyChan := make(chan string, 1)
|
|
||||||
mockForwarder.On("HandleConnection", mock.Anything, mockSSHChannel).Run(func(args mock.Arguments) {
|
|
||||||
w := args.Get(0).(io.ReadWriter)
|
|
||||||
buf := make([]byte, len("hello=world"))
|
|
||||||
if _, err := io.ReadFull(w, buf); err != nil {
|
|
||||||
bodyChan <- ""
|
|
||||||
} else {
|
|
||||||
bodyChan <- string(buf)
|
|
||||||
}
|
|
||||||
_, _ = w.Write([]byte("HTTP/1.1 200 OK\r\nContent-Length: 2\r\n\r\nok"))
|
|
||||||
})
|
|
||||||
|
|
||||||
go func() {
|
|
||||||
for range reqCh {
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
serverConn, clientConn := net.Pipe()
|
|
||||||
defer func() {
|
|
||||||
_ = clientConn.Close()
|
|
||||||
}()
|
|
||||||
|
|
||||||
remoteAddr, _ := net.ResolveTCPAddr("tcp", "127.0.0.1:12345")
|
|
||||||
wrappedServerConn := &wrappedConn{Conn: serverConn, remoteAddr: remoteAddr}
|
|
||||||
|
|
||||||
go hh.Handler(wrappedServerConn, true)
|
|
||||||
|
|
||||||
request := []byte("POST / HTTP/1.1\r\nHost: test.domain\r\nContent-Type: application/x-www-form-urlencoded\r\nContent-Length: 11\r\n\r\nhello=world")
|
|
||||||
go func() {
|
|
||||||
_, _ = clientConn.Write(request)
|
|
||||||
}()
|
|
||||||
|
|
||||||
var response []byte
|
|
||||||
respDone := make(chan struct{})
|
|
||||||
go func() {
|
|
||||||
defer close(respDone)
|
|
||||||
buf := make([]byte, 4096)
|
|
||||||
for {
|
|
||||||
n, err := clientConn.Read(buf)
|
|
||||||
if err != nil {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
response = append(response, buf[:n]...)
|
|
||||||
if bytes.Contains(response, []byte("\r\n\r\nok")) {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
select {
|
|
||||||
case body := <-bodyChan:
|
|
||||||
assert.Equal(t, "hello=world", body)
|
|
||||||
case <-time.After(5 * time.Second):
|
|
||||||
t.Fatal("timed out waiting for forwarded body")
|
|
||||||
}
|
|
||||||
|
|
||||||
select {
|
|
||||||
case <-respDone:
|
|
||||||
resStr := string(response)
|
|
||||||
assert.True(t, strings.HasPrefix(resStr, "HTTP/1.1 200 OK\r\n"))
|
|
||||||
assert.Contains(t, resStr, "Server: Tunnel Please\r\n")
|
|
||||||
assert.True(t, strings.HasSuffix(resStr, "\r\n\r\nok"))
|
|
||||||
case <-time.After(5 * time.Second):
|
|
||||||
t.Fatal("timed out waiting for response")
|
|
||||||
}
|
|
||||||
|
|
||||||
mu.Lock()
|
|
||||||
hdrStr := string(capturedHeaders)
|
|
||||||
mu.Unlock()
|
|
||||||
assert.Contains(t, hdrStr, "POST / HTTP/1.1\r\n")
|
|
||||||
assert.Contains(t, hdrStr, "Content-Length: 11\r\n")
|
|
||||||
assert.Contains(t, hdrStr, "X-Forwarded-For: 127.0.0.1\r\n")
|
|
||||||
|
|
||||||
mockSessionRegistry.AssertExpectations(t)
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -63,7 +63,8 @@ func TestHTTPSServer_Serve(t *testing.T) {
|
|||||||
|
|
||||||
go func() {
|
go func() {
|
||||||
time.Sleep(100 * time.Millisecond)
|
time.Sleep(100 * time.Millisecond)
|
||||||
_ = listener.Close()
|
err = listener.Close()
|
||||||
|
assert.NoError(t, err)
|
||||||
}()
|
}()
|
||||||
|
|
||||||
err = srv.Serve(listener)
|
err = srv.Serve(listener)
|
||||||
|
|||||||
@@ -45,7 +45,8 @@ func TestTCPServer_Serve(t *testing.T) {
|
|||||||
|
|
||||||
go func() {
|
go func() {
|
||||||
time.Sleep(100 * time.Millisecond)
|
time.Sleep(100 * time.Millisecond)
|
||||||
_ = listener.Close()
|
err = listener.Close()
|
||||||
|
assert.NoError(t, err)
|
||||||
}()
|
}()
|
||||||
|
|
||||||
err = srv.Serve(listener)
|
err = srv.Serve(listener)
|
||||||
|
|||||||
@@ -25,7 +25,6 @@ type MockConfig struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (m *MockConfig) Domain() string { return m.Called().String(0) }
|
func (m *MockConfig) Domain() string { return m.Called().String(0) }
|
||||||
func (m *MockConfig) FrontendURL() string { return m.Called().String(0) }
|
|
||||||
func (m *MockConfig) SSHPort() string { return m.Called().String(0) }
|
func (m *MockConfig) SSHPort() string { return m.Called().String(0) }
|
||||||
func (m *MockConfig) HTTPPort() string { return m.Called().String(0) }
|
func (m *MockConfig) HTTPPort() string { return m.Called().String(0) }
|
||||||
func (m *MockConfig) HTTPSPort() string { return m.Called().String(0) }
|
func (m *MockConfig) HTTPSPort() string { return m.Called().String(0) }
|
||||||
|
|||||||
+6
-3
@@ -2,6 +2,8 @@
|
|||||||
"extends": [
|
"extends": [
|
||||||
"config:recommended"
|
"config:recommended"
|
||||||
],
|
],
|
||||||
|
"prConcurrentLimit": 1,
|
||||||
|
"prHourlyLimit": 1,
|
||||||
"packageRules": [
|
"packageRules": [
|
||||||
{
|
{
|
||||||
"matchUpdateTypes": [
|
"matchUpdateTypes": [
|
||||||
@@ -10,10 +12,11 @@
|
|||||||
"pin",
|
"pin",
|
||||||
"digest"
|
"digest"
|
||||||
],
|
],
|
||||||
|
"groupName": "all-dependencies",
|
||||||
"automerge": true,
|
"automerge": true,
|
||||||
"baseBranchPatterns": [
|
"matchPackageNames": [
|
||||||
"staging"
|
"*"
|
||||||
]
|
]
|
||||||
}
|
}
|
||||||
]
|
]
|
||||||
}
|
}
|
||||||
Reference in New Issue
Block a user