forked from ms/transocks
Добавлен build workflow.
This commit is contained in:
parent
9371733501
commit
a59ad95f63
108
.gitea/workflows/build.yaml
Normal file
108
.gitea/workflows/build.yaml
Normal file
@ -0,0 +1,108 @@
|
|||||||
|
name: build
|
||||||
|
|
||||||
|
run-name: ${{ gitea.actor }} build Transocks
|
||||||
|
|
||||||
|
on:
|
||||||
|
push:
|
||||||
|
branches:
|
||||||
|
- "**"
|
||||||
|
tags-ignore:
|
||||||
|
- "v*"
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
build:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
- name: check-out
|
||||||
|
uses: https://gitea.com/actions/checkout@v4
|
||||||
|
with:
|
||||||
|
fetch-depth: 0
|
||||||
|
fetch-tags: true
|
||||||
|
|
||||||
|
- name: set-up go
|
||||||
|
uses: https://gitea.com/actions/setup-go@v3
|
||||||
|
with:
|
||||||
|
go-version: ">=1.22"
|
||||||
|
|
||||||
|
- name: set-up dependencies
|
||||||
|
run: |
|
||||||
|
go install github.com/kisielk/errcheck@latest
|
||||||
|
go install honnef.co/go/tools/cmd/staticcheck@latest
|
||||||
|
go install github.com/sashamelentyev/usestdlibvars@latest
|
||||||
|
go install golang.org/x/tools/go/analysis/passes/shadow/cmd/shadow@latest
|
||||||
|
make vendor
|
||||||
|
|
||||||
|
- name: lint
|
||||||
|
run: make lint
|
||||||
|
|
||||||
|
- name: golangci-lint
|
||||||
|
uses: https://github.com/golangci/golangci-lint-action@v6
|
||||||
|
with:
|
||||||
|
version: v1.60
|
||||||
|
|
||||||
|
- name: build amd64
|
||||||
|
id: build-amd
|
||||||
|
run: |
|
||||||
|
echo "ARTIFACT=transocks-$(make version)_$(go env GOOS)-amd64" >> $GITHUB_OUTPUT
|
||||||
|
GOARCH=amd64 make clean build
|
||||||
|
|
||||||
|
- name: upload amd64
|
||||||
|
uses: https://gitea.com/actions/upload-artifact@v3
|
||||||
|
with:
|
||||||
|
name: ${{ steps.build-amd.outputs.ARTIFACT }}
|
||||||
|
path: out/bin/*
|
||||||
|
overwrite: true
|
||||||
|
|
||||||
|
- name: build arm64
|
||||||
|
id: build-arm
|
||||||
|
run: |
|
||||||
|
echo "ARTIFACT=transocks-$(make version)_$(go env GOOS)-arm64" >> $GITHUB_OUTPUT
|
||||||
|
GOARCH=arm64 make clean build
|
||||||
|
|
||||||
|
- name: upload arm64
|
||||||
|
uses: https://gitea.com/actions/upload-artifact@v3
|
||||||
|
with:
|
||||||
|
name: ${{ steps.build-arm.outputs.ARTIFACT }}
|
||||||
|
path: out/bin/*
|
||||||
|
overwrite: true
|
||||||
|
|
||||||
|
build_windows:
|
||||||
|
runs-on: windows
|
||||||
|
defaults:
|
||||||
|
run:
|
||||||
|
shell: bash
|
||||||
|
steps:
|
||||||
|
- name: check-out
|
||||||
|
uses: https://gitea.com/actions/checkout@v4
|
||||||
|
with:
|
||||||
|
fetch-depth: 0
|
||||||
|
fetch-tags: true
|
||||||
|
|
||||||
|
- name: set-up dependencies
|
||||||
|
run: make vendor
|
||||||
|
|
||||||
|
- name: build amd
|
||||||
|
id: build-amd
|
||||||
|
run: |
|
||||||
|
echo "ARTIFACT=transocks-$(make version)_$(go env GOOS)-amd64" >> $GITHUB_OUTPUT
|
||||||
|
GOARCH=amd64 make build
|
||||||
|
|
||||||
|
- name: upload amd
|
||||||
|
uses: https://gitea.com/actions/upload-artifact@v3
|
||||||
|
with:
|
||||||
|
name: ${{ steps.build-amd.outputs.ARTIFACT }}
|
||||||
|
path: out/bin/*
|
||||||
|
overwrite: true
|
||||||
|
|
||||||
|
- name: build arm
|
||||||
|
id: build-arm
|
||||||
|
run: |
|
||||||
|
echo "ARTIFACT=transocks-$(make version)_$(go env GOOS)-arm64" >> $GITHUB_OUTPUT
|
||||||
|
GOARCH=arm64 make build
|
||||||
|
|
||||||
|
- name: upload amd
|
||||||
|
uses: https://gitea.com/actions/upload-artifact@v3
|
||||||
|
with:
|
||||||
|
name: ${{ steps.build-arm.outputs.ARTIFACT }}
|
||||||
|
path: out/bin/*
|
||||||
|
overwrite: true
|
@ -10,7 +10,10 @@ jobs:
|
|||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- name: check-out
|
- name: check-out
|
||||||
uses: actions/checkout@v4
|
uses: https://gitea.com/actions/checkout@v4
|
||||||
|
with:
|
||||||
|
fetch-depth: 0
|
||||||
|
fetch-tags: true
|
||||||
|
|
||||||
- name: set-up qemu
|
- name: set-up qemu
|
||||||
uses: https://git.mousesoft.ru/ms/gitea-action-setup-qemu@v3
|
uses: https://git.mousesoft.ru/ms/gitea-action-setup-qemu@v3
|
||||||
|
@ -3,6 +3,5 @@ linters:
|
|||||||
- dupl
|
- dupl
|
||||||
- goconst
|
- goconst
|
||||||
- gofmt
|
- gofmt
|
||||||
- golint
|
|
||||||
- typecheck
|
- typecheck
|
||||||
- unparam
|
- unparam
|
||||||
|
@ -75,10 +75,10 @@ func NewConfig() *Config {
|
|||||||
// It returns non-nil error if the configuration is not valid.
|
// It returns non-nil error if the configuration is not valid.
|
||||||
func (c *Config) validate() error {
|
func (c *Config) validate() error {
|
||||||
if c.ProxyURL == nil {
|
if c.ProxyURL == nil {
|
||||||
return errors.New("ProxyURL is nil")
|
return errors.New("proxy URL is nil")
|
||||||
}
|
}
|
||||||
if c.Mode != ModeNAT {
|
if c.Mode != ModeNAT {
|
||||||
return fmt.Errorf("Unknown mode: %s", c.Mode)
|
return fmt.Errorf("unknown mode: %s", c.Mode)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
@ -12,7 +12,6 @@ import (
|
|||||||
"bufio"
|
"bufio"
|
||||||
"bytes"
|
"bytes"
|
||||||
"encoding/base64"
|
"encoding/base64"
|
||||||
"errors"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"net"
|
"net"
|
||||||
"net/http"
|
"net/http"
|
||||||
@ -51,7 +50,7 @@ func httpDialType(u *url.URL, forward proxy.Dialer) (proxy.Dialer, error) {
|
|||||||
|
|
||||||
func (d *httpDialer) Dial(network, addr string) (c net.Conn, err error) {
|
func (d *httpDialer) Dial(network, addr string) (c net.Conn, err error) {
|
||||||
req := &http.Request{
|
req := &http.Request{
|
||||||
Method: "CONNECT",
|
Method: http.MethodConnect,
|
||||||
URL: &url.URL{Opaque: addr},
|
URL: &url.URL{Opaque: addr},
|
||||||
Host: addr,
|
Host: addr,
|
||||||
Header: d.header,
|
Header: d.header,
|
||||||
@ -60,20 +59,24 @@ func (d *httpDialer) Dial(network, addr string) (c net.Conn, err error) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
req.Write(c)
|
if err = req.Write(c); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
// Read response until "\r\n\r\n".
|
// Read response until "\r\n\r\n".
|
||||||
// bufio cannot be used as the connected server may not be
|
// bufio cannot be used as the connected server may not be
|
||||||
// a HTTP(S) server.
|
// a HTTP(S) server.
|
||||||
c.SetReadDeadline(time.Now().Add(10 * time.Second))
|
if err = c.SetReadDeadline(time.Now().Add(10 * time.Second)); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
buf := make([]byte, 0, 4096)
|
buf := make([]byte, 0, 4096)
|
||||||
b := make([]byte, 1)
|
b := make([]byte, 1)
|
||||||
state := 0
|
state := 0
|
||||||
for {
|
for {
|
||||||
_, e := c.Read(b)
|
if _, err = c.Read(b); err != nil {
|
||||||
if e != nil {
|
_ = c.Close()
|
||||||
c.Close()
|
return nil, fmt.Errorf("reset proxy connection: %w", err)
|
||||||
return nil, errors.New("reset proxy connection")
|
|
||||||
}
|
}
|
||||||
buf = append(buf, b[0])
|
buf = append(buf, b[0])
|
||||||
switch state {
|
switch state {
|
||||||
@ -107,15 +110,21 @@ func (d *httpDialer) Dial(network, addr string) (c net.Conn, err error) {
|
|||||||
|
|
||||||
PARSE:
|
PARSE:
|
||||||
var zero time.Time
|
var zero time.Time
|
||||||
c.SetReadDeadline(zero)
|
|
||||||
resp, e := http.ReadResponse(bufio.NewReader(bytes.NewBuffer(buf)), req)
|
if err = c.SetReadDeadline(zero); err != nil {
|
||||||
if e != nil {
|
return nil, err
|
||||||
c.Close()
|
|
||||||
return nil, e
|
|
||||||
}
|
}
|
||||||
resp.Body.Close()
|
|
||||||
if resp.StatusCode != 200 {
|
resp, err := http.ReadResponse(bufio.NewReader(bytes.NewBuffer(buf)), req)
|
||||||
c.Close()
|
if err != nil {
|
||||||
|
_ = c.Close()
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
_ = resp.Body.Close()
|
||||||
|
|
||||||
|
if resp.StatusCode != http.StatusOK {
|
||||||
|
_ = c.Close()
|
||||||
return nil, fmt.Errorf("proxy returns %s", resp.Status)
|
return nil, fmt.Errorf("proxy returns %s", resp.Status)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -23,8 +23,9 @@ func TestHTTPDialer(t *testing.T) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
defer conn.Close()
|
|
||||||
|
|
||||||
conn.Write([]byte("GET / HTTP/1.1\r\nHost: www.yahoo.com:80\r\nConnection: close\r\n\r\n"))
|
defer func() { _ = conn.Close() }()
|
||||||
io.Copy(os.Stdout, conn)
|
|
||||||
|
_, _ = conn.Write([]byte("GET / HTTP/1.1\r\nHost: www.yahoo.com:80\r\nConnection: close\r\n\r\n"))
|
||||||
|
_, _ = io.Copy(os.Stdout, conn)
|
||||||
}
|
}
|
||||||
|
138
makefile
Normal file
138
makefile
Normal file
@ -0,0 +1,138 @@
|
|||||||
|
# transocks makefile
|
||||||
|
# ==================
|
||||||
|
|
||||||
|
SHELL := /usr/bin/env bash
|
||||||
|
|
||||||
|
PROJECT_ID := transocks
|
||||||
|
PROJECT_NAME ?= Transocks
|
||||||
|
BIN_SUFFIX :=
|
||||||
|
|
||||||
|
TMPDIR ?= $(CURDIR)/tmp
|
||||||
|
OUTDIR ?= $(CURDIR)/out
|
||||||
|
BINDIR ?= $(OUTDIR)/bin
|
||||||
|
|
||||||
|
VERSION ?= $(strip $(shell ./scripts/version.sh))
|
||||||
|
VERSION_NUMBER := $(strip $(shell ./scripts/version.sh number))
|
||||||
|
|
||||||
|
BUILD_OS ?= $(shell go env GOOS)
|
||||||
|
BUILD_ARCH ?= $(shell go env GOARCH)
|
||||||
|
|
||||||
|
GO_LDFLAGS ?=
|
||||||
|
GO_OPT_V := -X "main.version=$(VERSION)"
|
||||||
|
GO_OPT_APP := -X "main.appname=$(PROJECT_NAME)"
|
||||||
|
GO_OPT_BASE := -ldflags '$(GO_OPT_V) $(GO_OPT_APP) $(GO_LDFLAGS)'
|
||||||
|
GO_OPT ?=
|
||||||
|
EXPORT_RESULT ?= false # for CI please set EXPORT_RESULT to true
|
||||||
|
|
||||||
|
GOCMD := go
|
||||||
|
GOTEST := $(GOCMD) test
|
||||||
|
GOVET := $(GOCMD) vet
|
||||||
|
ECHO_CMD := echo -e
|
||||||
|
|
||||||
|
ifeq ($(OS),Windows_NT)
|
||||||
|
MSI_FILE ?= $(PROJECT_ID)_$(VERSION)_$(BUILD_ARCH).msi
|
||||||
|
DIST_EXT := .zip
|
||||||
|
DIST_OPTS := -a -cf
|
||||||
|
MSI_VERSION := $(shell echo $(VERSION_NUMBER) | sed -e 's/-.*//')
|
||||||
|
BIN_SUFFIX := .exe
|
||||||
|
else
|
||||||
|
PLATFORM := $(shell uname -s | tr '[:upper:]' '[:lower:]')
|
||||||
|
DIST_EXT := .tar.gz
|
||||||
|
DIST_OPTS := -czf
|
||||||
|
endif
|
||||||
|
|
||||||
|
PKG_NAME := $(PROJECT_ID)_$(VERSION)_$(BUILD_OS)-$(BUILD_ARCH)
|
||||||
|
DIST_FILE := $(PKG_NAME)$(DIST_EXT)
|
||||||
|
|
||||||
|
GREEN := $(shell tput -Txterm setaf 2)
|
||||||
|
YELLOW := $(shell tput -Txterm setaf 3)
|
||||||
|
WHITE := $(shell tput -Txterm setaf 7)
|
||||||
|
CYAN := $(shell tput -Txterm setaf 6)
|
||||||
|
RESET := $(shell tput -Txterm sgr0)
|
||||||
|
|
||||||
|
.DEFAULT_GOAL := all
|
||||||
|
|
||||||
|
version: ## Version of the project to be built
|
||||||
|
@echo $(VERSION)
|
||||||
|
.PHONY:version
|
||||||
|
|
||||||
|
version-number: ## Version number of the project to be built
|
||||||
|
@echo $(VERSION_NUMBER)
|
||||||
|
.PHONY:version-number
|
||||||
|
|
||||||
|
## Build
|
||||||
|
|
||||||
|
all: clean vendor build ## Build binary
|
||||||
|
.PHONY:all
|
||||||
|
|
||||||
|
APPS := $(patsubst cmd/%/, %, $(dir $(wildcard cmd/*/main.go)))
|
||||||
|
|
||||||
|
build: $(addprefix $(BINDIR)/, $(APPS)) ## Build your project and put the output binary in out/bin/
|
||||||
|
@mkdir -p $(BINDIR)
|
||||||
|
@$(ECHO_CMD) "Build\t\t${GREEN}[OK]${RESET}"
|
||||||
|
.PHONY:build
|
||||||
|
|
||||||
|
$(BINDIR)/%: cmd/%/main.go $(patsubst cmd/%/main.go,cmd/%/*.go,$<)
|
||||||
|
@rm -f "$(BINDIR)/$(BIN_PREFIX)$(patsubst cmd/%/main.go,%,$<)$(BIN_SUFFIX)"
|
||||||
|
$(GOCMD) build $(GO_OPT_BASE) $(GO_OPT) \
|
||||||
|
-o "$(BINDIR)/$(BIN_PREFIX)$(patsubst cmd/%/main.go,%,$<)$(BIN_SUFFIX)" \
|
||||||
|
$(patsubst %/main.go,./%,$<)
|
||||||
|
|
||||||
|
dist: ## Create all distro packages
|
||||||
|
@rm -f "$(OUTDIR)/$(DIST_FILE)"
|
||||||
|
tar $(DIST_OPTS) "$(OUTDIR)/$(DIST_FILE)" -C "$(OUTDIR)" bin
|
||||||
|
@$(ECHO_CMD) "Dist\t\t${GREEN}[OK]${RESET}"
|
||||||
|
.PHONY:dist
|
||||||
|
|
||||||
|
vendor: ## Copy of all packages needed to support builds and tests in the vendor directory
|
||||||
|
$(GOCMD) mod vendor
|
||||||
|
@$(ECHO_CMD) "Vendor\t\t${GREEN}[OK]${RESET}"
|
||||||
|
.PHONY:vendor
|
||||||
|
|
||||||
|
clean: ## Remove build related files
|
||||||
|
@rm -fr $(TMPDIR)
|
||||||
|
@rm -fr $(OUTDIR)
|
||||||
|
@$(ECHO_CMD) "Clean\t\t${GREEN}[OK]${RESET}"
|
||||||
|
.PHONY:clean
|
||||||
|
|
||||||
|
## Test
|
||||||
|
|
||||||
|
test: ## Run the tests of the project
|
||||||
|
ifeq ($(EXPORT_RESULT), true)
|
||||||
|
@mkdir -p $(OUTDIR)
|
||||||
|
$(eval OUTPUT_OPTIONS = | go-junit-report -set-exit-code > $(OUTDIR)/junit-report.xml)
|
||||||
|
endif
|
||||||
|
$(GOTEST) -v $(GO_OPT) ./... $(OUTPUT_OPTIONS)
|
||||||
|
@$(ECHO_CMD) "Test\t\t${GREEN}[OK]${RESET}"
|
||||||
|
.PHONY:test
|
||||||
|
|
||||||
|
|
||||||
|
## Lint
|
||||||
|
|
||||||
|
lint: ## Run all available linters.
|
||||||
|
go vet ./...
|
||||||
|
errcheck ./...
|
||||||
|
staticcheck ./...
|
||||||
|
usestdlibvars ./...
|
||||||
|
shadow ./...
|
||||||
|
@$(ECHO_CMD) "Lint\t\t${GREEN}[OK]${RESET}"
|
||||||
|
.PHONY:lint
|
||||||
|
|
||||||
|
golangci-lint: ## Run golangci-lint linter
|
||||||
|
@golangci-lint run
|
||||||
|
@$(ECHO_CMD) "GolangCI Lint\t${GREEN}[OK]${RESET}"
|
||||||
|
.PHONY:golangci-lint
|
||||||
|
|
||||||
|
## Help
|
||||||
|
|
||||||
|
help: ## Show this help.
|
||||||
|
@$(ECHO_CMD) ''
|
||||||
|
@$(ECHO_CMD) 'Usage:'
|
||||||
|
@$(ECHO_CMD) ' ${YELLOW}make${RESET} ${GREEN}<target>${RESET}'
|
||||||
|
@$(ECHO_CMD) ''
|
||||||
|
@$(ECHO_CMD) 'Targets:'
|
||||||
|
@awk 'BEGIN {FS = ":.*?## "} { \
|
||||||
|
if (/^[a-zA-Z_-]+:.*?##.*$$/) {printf " ${YELLOW}%-20s${GREEN}%s${RESET}\n", $$1, $$2} \
|
||||||
|
else if (/^## .*$$/) {printf " ${CYAN}%s${RESET}\n", substr($$1,4)} \
|
||||||
|
}' $(MAKEFILE_LIST)
|
||||||
|
.PHONY:help
|
@ -1,12 +1,14 @@
|
|||||||
|
//go:build linux
|
||||||
// +build linux
|
// +build linux
|
||||||
|
|
||||||
package transocks
|
package transocks
|
||||||
|
|
||||||
import (
|
import (
|
||||||
syscall "golang.org/x/sys/unix"
|
|
||||||
"net"
|
"net"
|
||||||
"os"
|
"os"
|
||||||
"unsafe"
|
"unsafe"
|
||||||
|
|
||||||
|
syscall "golang.org/x/sys/unix"
|
||||||
)
|
)
|
||||||
|
|
||||||
func getsockopt(s int, level int, optname int, optval unsafe.Pointer, optlen *uint32) (err error) {
|
func getsockopt(s int, level int, optname int, optval unsafe.Pointer, optlen *uint32) (err error) {
|
||||||
@ -31,7 +33,7 @@ func GetOriginalDST(conn *net.TCPConn) (*net.TCPAddr, error) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
defer f.Close()
|
defer func() { _ = f.Close() }()
|
||||||
|
|
||||||
fd := int(f.Fd())
|
fd := int(f.Fd())
|
||||||
// revert to non-blocking mode.
|
// revert to non-blocking mode.
|
||||||
@ -43,17 +45,14 @@ func GetOriginalDST(conn *net.TCPConn) (*net.TCPAddr, error) {
|
|||||||
v6 := conn.LocalAddr().(*net.TCPAddr).IP.To4() == nil
|
v6 := conn.LocalAddr().(*net.TCPAddr).IP.To4() == nil
|
||||||
if v6 {
|
if v6 {
|
||||||
var addr syscall.RawSockaddrInet6
|
var addr syscall.RawSockaddrInet6
|
||||||
var len uint32
|
len := uint32(unsafe.Sizeof(addr))
|
||||||
len = uint32(unsafe.Sizeof(addr))
|
|
||||||
err = getsockopt(fd, syscall.IPPROTO_IPV6, IP6T_SO_ORIGINAL_DST,
|
err = getsockopt(fd, syscall.IPPROTO_IPV6, IP6T_SO_ORIGINAL_DST,
|
||||||
unsafe.Pointer(&addr), &len)
|
unsafe.Pointer(&addr), &len)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, os.NewSyscallError("getsockopt", err)
|
return nil, os.NewSyscallError("getsockopt", err)
|
||||||
}
|
}
|
||||||
ip := make([]byte, 16)
|
ip := make([]byte, 16)
|
||||||
for i, b := range addr.Addr {
|
copy(ip, addr.Addr[:])
|
||||||
ip[i] = b
|
|
||||||
}
|
|
||||||
pb := *(*[2]byte)(unsafe.Pointer(&addr.Port))
|
pb := *(*[2]byte)(unsafe.Pointer(&addr.Port))
|
||||||
return &net.TCPAddr{
|
return &net.TCPAddr{
|
||||||
IP: ip,
|
IP: ip,
|
||||||
@ -63,17 +62,14 @@ func GetOriginalDST(conn *net.TCPConn) (*net.TCPAddr, error) {
|
|||||||
|
|
||||||
// IPv4
|
// IPv4
|
||||||
var addr syscall.RawSockaddrInet4
|
var addr syscall.RawSockaddrInet4
|
||||||
var len uint32
|
len := uint32(unsafe.Sizeof(addr))
|
||||||
len = uint32(unsafe.Sizeof(addr))
|
|
||||||
err = getsockopt(fd, syscall.IPPROTO_IP, SO_ORIGINAL_DST,
|
err = getsockopt(fd, syscall.IPPROTO_IP, SO_ORIGINAL_DST,
|
||||||
unsafe.Pointer(&addr), &len)
|
unsafe.Pointer(&addr), &len)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, os.NewSyscallError("getsockopt", err)
|
return nil, os.NewSyscallError("getsockopt", err)
|
||||||
}
|
}
|
||||||
ip := make([]byte, 4)
|
ip := make([]byte, 4)
|
||||||
for i, b := range addr.Addr {
|
copy(ip, addr.Addr[:])
|
||||||
ip[i] = b
|
|
||||||
}
|
|
||||||
pb := *(*[2]byte)(unsafe.Pointer(&addr.Port))
|
pb := *(*[2]byte)(unsafe.Pointer(&addr.Port))
|
||||||
return &net.TCPAddr{
|
return &net.TCPAddr{
|
||||||
IP: ip,
|
IP: ip,
|
||||||
|
@ -1,3 +1,4 @@
|
|||||||
|
//go:build linux
|
||||||
// +build linux
|
// +build linux
|
||||||
|
|
||||||
package transocks
|
package transocks
|
||||||
@ -14,13 +15,14 @@ func TestGetOriginalDST(t *testing.T) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
defer l.Close()
|
|
||||||
|
defer func() { _ = l.Close() }()
|
||||||
|
|
||||||
c, err := l.Accept()
|
c, err := l.Accept()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
defer c.Close()
|
defer func() { _ = c.Close() }()
|
||||||
|
|
||||||
origAddr, err := GetOriginalDST(c.(*net.TCPConn))
|
origAddr, err := GetOriginalDST(c.(*net.TCPConn))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
20
scripts/changes.awk
Normal file
20
scripts/changes.awk
Normal file
@ -0,0 +1,20 @@
|
|||||||
|
# Get changes of given version number.
|
||||||
|
{
|
||||||
|
while (index($0, "## [" version "]") <= 0) {
|
||||||
|
if (getline <= 0) {
|
||||||
|
exit
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (getline <= 0 ) {
|
||||||
|
exit
|
||||||
|
}
|
||||||
|
if (getline <= 0 ) {
|
||||||
|
exit
|
||||||
|
}
|
||||||
|
while (index($0, "## [") <= 0) {
|
||||||
|
print $0
|
||||||
|
if (getline <= 0) {
|
||||||
|
exit
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
22
scripts/version.sh
Executable file
22
scripts/version.sh
Executable file
@ -0,0 +1,22 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
if [ -z ${TAG_NAME+x} ]; then
|
||||||
|
if [ -z ${BRANCH_NAME+x} ]; then
|
||||||
|
BRANCH_NAME=$(echo $(git branch --show-current) || \
|
||||||
|
echo $(git name-rev --name-only HEAD))
|
||||||
|
fi
|
||||||
|
GIT_VERSION=$(echo ${BRANCH_NAME} | grep -q 'release/' \
|
||||||
|
&& echo ${BRANCH_NAME} | sed -e 's|release/|v|' -e 's/$/-RC/' || \
|
||||||
|
echo $(git describe --always --tags --dirty 2>/dev/null) || echo v0)
|
||||||
|
else
|
||||||
|
GIT_VERSION=${TAG_NAME}
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ -z ${VERSION+x} ]; then
|
||||||
|
VERSION=$(echo ${GIT_VERSION} | sed -e 's|^origin/||')
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ -z $1 ]; then
|
||||||
|
echo "${VERSION}"
|
||||||
|
else
|
||||||
|
echo ${VERSION} | sed -e 's/^v//'
|
||||||
|
fi
|
48
server.go
48
server.go
@ -85,7 +85,7 @@ func NewServer(c *Config) (*Server, error) {
|
|||||||
func (s *Server) handleConnection(ctx context.Context, conn net.Conn) {
|
func (s *Server) handleConnection(ctx context.Context, conn net.Conn) {
|
||||||
tc, ok := conn.(*net.TCPConn)
|
tc, ok := conn.(*net.TCPConn)
|
||||||
if !ok {
|
if !ok {
|
||||||
s.logger.Error("non-TCP connection", map[string]interface{}{
|
_ = s.logger.Error("non-TCP connection", map[string]interface{}{
|
||||||
"conn": conn,
|
"conn": conn,
|
||||||
})
|
})
|
||||||
return
|
return
|
||||||
@ -101,7 +101,7 @@ func (s *Server) handleConnection(ctx context.Context, conn net.Conn) {
|
|||||||
origAddr, err := GetOriginalDST(tc)
|
origAddr, err := GetOriginalDST(tc)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fields[log.FnError] = err.Error()
|
fields[log.FnError] = err.Error()
|
||||||
s.logger.Error("GetOriginalDST failed", fields)
|
_ = s.logger.Error("GetOriginalDST failed", fields)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
addr = origAddr.String()
|
addr = origAddr.String()
|
||||||
@ -109,14 +109,20 @@ func (s *Server) handleConnection(ctx context.Context, conn net.Conn) {
|
|||||||
addr = tc.LocalAddr().String()
|
addr = tc.LocalAddr().String()
|
||||||
}
|
}
|
||||||
|
|
||||||
var reader io.Reader = tc
|
var (
|
||||||
|
reader io.Reader = tc
|
||||||
|
isTLS bool
|
||||||
|
reader_n io.Reader
|
||||||
|
)
|
||||||
|
|
||||||
// Check if TLS
|
// Check if TLS
|
||||||
isTLS, reader_n, err := peekSSL(tc)
|
if isTLSloc, reader_nloc, err := peekSSL(tc); err != nil {
|
||||||
if err != nil {
|
|
||||||
fields[log.FnError] = err.Error()
|
fields[log.FnError] = err.Error()
|
||||||
s.logger.Error("peekSSL failed", fields)
|
_ = s.logger.Error("peekSSL failed", fields)
|
||||||
return
|
return
|
||||||
|
} else {
|
||||||
|
isTLS = isTLSloc
|
||||||
|
reader_n = reader_nloc
|
||||||
}
|
}
|
||||||
reader = reader_n
|
reader = reader_n
|
||||||
fields["is_tls"] = isTLS
|
fields["is_tls"] = isTLS
|
||||||
@ -126,7 +132,7 @@ func (s *Server) handleConnection(ctx context.Context, conn net.Conn) {
|
|||||||
hello, reader_n2, err := peekClientHello(reader)
|
hello, reader_n2, err := peekClientHello(reader)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fields[log.FnError] = err.Error()
|
fields[log.FnError] = err.Error()
|
||||||
s.logger.Warn("peekClientHello failed", fields)
|
_ = s.logger.Warn("peekClientHello failed", fields)
|
||||||
}
|
}
|
||||||
if err == nil && hello.ServerName != "" {
|
if err == nil && hello.ServerName != "" {
|
||||||
addr = hello.ServerName + addr[strings.Index(addr, ":"):]
|
addr = hello.ServerName + addr[strings.Index(addr, ":"):]
|
||||||
@ -137,7 +143,7 @@ func (s *Server) handleConnection(ctx context.Context, conn net.Conn) {
|
|||||||
host, reader_n3, err := peekHTTP(reader)
|
host, reader_n3, err := peekHTTP(reader)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fields[log.FnError] = err.Error()
|
fields[log.FnError] = err.Error()
|
||||||
s.logger.Warn("peekHTTP failed", fields)
|
_ = s.logger.Warn("peekHTTP failed", fields)
|
||||||
} else {
|
} else {
|
||||||
if err == nil && host != "" {
|
if err == nil && host != "" {
|
||||||
addr = host + addr[strings.Index(addr, ":"):]
|
addr = host + addr[strings.Index(addr, ":"):]
|
||||||
@ -151,33 +157,33 @@ func (s *Server) handleConnection(ctx context.Context, conn net.Conn) {
|
|||||||
destConn, err := s.dialer.Dial("tcp", addr)
|
destConn, err := s.dialer.Dial("tcp", addr)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fields[log.FnError] = err.Error()
|
fields[log.FnError] = err.Error()
|
||||||
s.logger.Error("failed to connect to proxy server", fields)
|
_ = s.logger.Error("failed to connect to proxy server", fields)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer destConn.Close()
|
defer func() { _ = destConn.Close() }()
|
||||||
|
|
||||||
s.logger.Info("proxy starts", fields)
|
_ = s.logger.Info("proxy starts", fields)
|
||||||
|
|
||||||
// do proxy
|
// do proxy
|
||||||
st := time.Now()
|
st := time.Now()
|
||||||
env := well.NewEnvironment(ctx)
|
env := well.NewEnvironment(ctx)
|
||||||
env.Go(func(ctx context.Context) error {
|
env.Go(func(ctx context.Context) error {
|
||||||
buf := s.pool.Get().([]byte)
|
buf := s.pool.Get().([]byte)
|
||||||
_, err := io.CopyBuffer(destConn, reader, buf)
|
_, err = io.CopyBuffer(destConn, reader, buf)
|
||||||
s.pool.Put(buf)
|
s.pool.Put(&buf)
|
||||||
if hc, ok := destConn.(netutil.HalfCloser); ok {
|
if hc, ok := destConn.(netutil.HalfCloser); ok {
|
||||||
hc.CloseWrite()
|
_ = hc.CloseWrite()
|
||||||
}
|
}
|
||||||
tc.CloseRead()
|
_ = tc.CloseRead()
|
||||||
return err
|
return err
|
||||||
})
|
})
|
||||||
env.Go(func(ctx context.Context) error {
|
env.Go(func(ctx context.Context) error {
|
||||||
buf := s.pool.Get().([]byte)
|
buf := s.pool.Get().([]byte)
|
||||||
_, err := io.CopyBuffer(tc, destConn, buf)
|
_, err = io.CopyBuffer(tc, destConn, buf)
|
||||||
s.pool.Put(buf)
|
s.pool.Put(&buf)
|
||||||
tc.CloseWrite()
|
_ = tc.CloseWrite()
|
||||||
if hc, ok := destConn.(netutil.HalfCloser); ok {
|
if hc, ok := destConn.(netutil.HalfCloser); ok {
|
||||||
hc.CloseRead()
|
_ = hc.CloseRead()
|
||||||
}
|
}
|
||||||
return err
|
return err
|
||||||
})
|
})
|
||||||
@ -188,10 +194,10 @@ func (s *Server) handleConnection(ctx context.Context, conn net.Conn) {
|
|||||||
fields["elapsed"] = time.Since(st).Seconds()
|
fields["elapsed"] = time.Since(st).Seconds()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fields[log.FnError] = err.Error()
|
fields[log.FnError] = err.Error()
|
||||||
s.logger.Error("proxy ends with an error", fields)
|
_ = s.logger.Error("proxy ends with an error", fields)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
s.logger.Info("proxy ends", fields)
|
_ = s.logger.Info("proxy ends", fields)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Peek ClientHello message from conn and returns SNI.
|
// Peek ClientHello message from conn and returns SNI.
|
||||||
|
Loading…
Reference in New Issue
Block a user