mirror of
https://github.com/nxshock/gonx.git
synced 2025-04-19 14:51:52 +05:00
Compare commits
7 Commits
Author | SHA1 | Date | |
---|---|---|---|
ea03e2c4fb | |||
e9ad16b83e | |||
5903c6fbcd | |||
9447891759 | |||
4c779d0954 | |||
19d0db3a00 | |||
52536b2675 |
2
PKGBUILD
2
PKGBUILD
@ -1,5 +1,5 @@
|
||||
pkgname=gonx
|
||||
pkgver=0.0.1
|
||||
pkgver=0.0.4
|
||||
pkgrel=1
|
||||
pkgdesc='Simple reverse proxy server'
|
||||
arch=('x86_64' 'aarch64')
|
||||
|
77
app.go
Normal file
77
app.go
Normal file
@ -0,0 +1,77 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"crypto/tls"
|
||||
"fmt"
|
||||
"log/slog"
|
||||
"net"
|
||||
)
|
||||
|
||||
type App struct {
|
||||
config *Config
|
||||
tlsListener net.Listener
|
||||
}
|
||||
|
||||
func newApp(config *Config) (*App, error) {
|
||||
err := config.initTls()
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to load TLS keys: %v", err)
|
||||
}
|
||||
|
||||
app := &App{config: config}
|
||||
|
||||
return app, nil
|
||||
}
|
||||
|
||||
func (app *App) reloadConfig(configFilePath string) error {
|
||||
config, err := LoadConfig(configFilePath)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to load config: %v", err)
|
||||
}
|
||||
|
||||
err = config.initTls()
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to load TLS keys: %v", err)
|
||||
}
|
||||
|
||||
app.config = config
|
||||
|
||||
slog.Debug("Deactivating TLS listener", slog.String("addr", app.config.TlsListenAddr))
|
||||
err = app.tlsListener.Close()
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to close TLS listener: %v", err)
|
||||
}
|
||||
|
||||
err = app.restartTlsListener()
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to restart TLS listener: %v", err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (app *App) restartTlsListener() error {
|
||||
slog.Debug("Starting TLS listener", slog.String("addr", app.config.TlsListenAddr))
|
||||
|
||||
tlsListener, err := tls.Listen("tcp", app.config.TlsListenAddr, app.config.tlsConfig)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to open tls listener: %v", err)
|
||||
}
|
||||
|
||||
app.tlsListener = tlsListener
|
||||
|
||||
go func() {
|
||||
for {
|
||||
conn, err := tlsListener.Accept()
|
||||
if err != nil {
|
||||
slog.Error("failed to receive connection", slog.String("err", err.Error())) // TODO: drop error on closing TLS listener
|
||||
break
|
||||
}
|
||||
slog.Debug("incoming connection", slog.String("RemoteAddr", conn.RemoteAddr().String()))
|
||||
|
||||
go func() { _ = handleTlsConn(conn.(*tls.Conn), app.config.proxyRules) }()
|
||||
}
|
||||
}()
|
||||
|
||||
return nil
|
||||
}
|
@ -55,6 +55,8 @@ func LoadConfig(configFilePath string) (*Config, error) {
|
||||
}
|
||||
|
||||
func (c *Config) initTls() error {
|
||||
slog.Debug("Loading TLS keys")
|
||||
|
||||
c.tlsConfig = new(tls.Config)
|
||||
|
||||
for hostName := range c.proxyRules {
|
||||
|
@ -37,6 +37,11 @@ RestrictRealtime=true
|
||||
RestrictSUIDSGID=true
|
||||
SystemCallArchitectures=native
|
||||
UMask=0027
|
||||
ProtectProc=noaccess
|
||||
ProcSubset=pid
|
||||
SystemCallFilter=~@clock @swap @reboot @raw-io @privileged @obsolete @mount @module @debug @cpu-emulation
|
||||
RestrictAddressFamilies=~AF_UNIX AF_PACKET AF_NETLINK
|
||||
UMask=0066
|
||||
|
||||
[Install]
|
||||
WantedBy=multi-user.target
|
||||
|
43
main.go
43
main.go
@ -1,7 +1,6 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"crypto/tls"
|
||||
"log/slog"
|
||||
"net/http"
|
||||
"os"
|
||||
@ -33,40 +32,25 @@ func main() {
|
||||
}}))
|
||||
slog.SetDefault(logger)
|
||||
|
||||
err = config.initTls()
|
||||
app, err := newApp(config)
|
||||
if err != nil {
|
||||
slog.Error("init tls error", slog.String("err", err.Error()))
|
||||
slog.Error("Failed to start app", slog.String("err", err.Error()))
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
go func() {
|
||||
slog.Debug("Starting TLS listener", slog.String("addr", config.TlsListenAddr))
|
||||
|
||||
listener, err := tls.Listen("tcp", config.TlsListenAddr, config.tlsConfig)
|
||||
err = app.restartTlsListener()
|
||||
if err != nil {
|
||||
slog.Error("Failed to open tls listener", slog.String("err", err.Error()))
|
||||
slog.Error("Failed to start TLS listener", slog.String("err", err.Error()))
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
for {
|
||||
conn, err := listener.Accept()
|
||||
if err != nil {
|
||||
slog.Debug("incoming connection failed", slog.String("err", err.Error()))
|
||||
continue
|
||||
}
|
||||
slog.Debug("incoming connection", slog.String("RemoteAddr", conn.RemoteAddr().String()))
|
||||
|
||||
go func() { _ = handleTlsConn(conn.(*tls.Conn), config.proxyRules) }()
|
||||
}
|
||||
}()
|
||||
|
||||
go func() {
|
||||
slog.Debug("Starting HTTP listener", slog.String("addr", config.HttpListenAddr))
|
||||
|
||||
smux := http.NewServeMux()
|
||||
smux.Handle(defaultAcmeChallengePath, http.StripPrefix(defaultAcmeChallengePath, http.FileServer(http.Dir(config.AcmeChallengePath))))
|
||||
smux.Handle(defaultAcmeChallengePath, http.FileServer(http.Dir(config.AcmeChallengePath)))
|
||||
smux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
||||
http.Redirect(w, r, "https://"+r.Host+r.RequestURI, http.StatusMovedPermanently) // TODO: заполнить URL переадресации
|
||||
http.Redirect(w, r, "https://"+r.Host+r.RequestURI, http.StatusMovedPermanently)
|
||||
})
|
||||
httpServer := http.Server{Handler: smux}
|
||||
httpServer.Addr = config.HttpListenAddr
|
||||
@ -77,6 +61,21 @@ func main() {
|
||||
}
|
||||
}()
|
||||
|
||||
go func() {
|
||||
c := make(chan os.Signal, 1)
|
||||
signal.Notify(c, syscall.SIGUSR1)
|
||||
for {
|
||||
<-c
|
||||
slog.Debug("TLS keys reload requested")
|
||||
|
||||
err := app.reloadConfig(configFilePath)
|
||||
if err != nil {
|
||||
slog.Error("failed to reload TLS keys", slog.String("err", err.Error()))
|
||||
}
|
||||
slog.Debug("Reloading TLS keys completed")
|
||||
}
|
||||
}()
|
||||
|
||||
c := make(chan os.Signal, 1)
|
||||
signal.Notify(c, os.Interrupt, syscall.SIGTERM)
|
||||
<-c
|
||||
|
Loading…
x
Reference in New Issue
Block a user