refactor: move things into related files
This commit is contained in:
Jon Lundy
parent
09ab41650d
commit
536483b73f
391
main.go
391
main.go
|
|
@ -2,26 +2,18 @@ package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"crypto/sha256"
|
|
||||||
"fmt"
|
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"log"
|
"log"
|
||||||
"net"
|
"net"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/http/httputil"
|
|
||||||
"os"
|
"os"
|
||||||
"os/signal"
|
"os/signal"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"regexp"
|
"regexp"
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
"sync"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/gliderlabs/ssh"
|
"github.com/gliderlabs/ssh"
|
||||||
"github.com/soheilhy/cmux"
|
|
||||||
"github.com/wolfeidau/humanhash"
|
|
||||||
"go.uber.org/multierr"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
|
|
@ -120,386 +112,3 @@ func envDefault(s, d string) string {
|
||||||
log.Println("env", s, "==", v)
|
log.Println("env", s, "==", v)
|
||||||
return v
|
return v
|
||||||
}
|
}
|
||||||
|
|
||||||
func (srv *server) newSession(ctx context.Context) func(ssh.Session) {
|
|
||||||
return func(s ssh.Session) {
|
|
||||||
if _, err := fmt.Fprintf(s, "Hello %s\n", s.User()); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if u, ok := srv.GetUserByName(s.User()); ok {
|
|
||||||
host := fmt.Sprintf("%v:%v", u.bindHost, u.bindPort)
|
|
||||||
director := func(req *http.Request) {
|
|
||||||
if h := req.Header.Get("X-Forwarded-Host"); h == "" {
|
|
||||||
req.Header.Set("X-Forwarded-Host", req.Host)
|
|
||||||
}
|
|
||||||
req.Header.Set("X-Origin-Host", host)
|
|
||||||
req.URL.Scheme = "http"
|
|
||||||
req.URL.Host = host
|
|
||||||
|
|
||||||
requestDump, err := httputil.DumpRequest(req, req.Method == http.MethodPost || req.Method == http.MethodPut)
|
|
||||||
if err != nil {
|
|
||||||
fmt.Println(err)
|
|
||||||
}
|
|
||||||
fmt.Fprintln(s, string(requestDump))
|
|
||||||
}
|
|
||||||
u.proxy = &httputil.ReverseProxy{Director: director}
|
|
||||||
fmt.Fprintf(s, "Created HTTP listener at: %v%v\n\n", u.name, srv.domainSuffix)
|
|
||||||
}
|
|
||||||
|
|
||||||
select {
|
|
||||||
case <-ctx.Done():
|
|
||||||
log.Println("server shutting down")
|
|
||||||
case <-s.Context().Done():
|
|
||||||
log.Println("user", s.User(), "disconnected")
|
|
||||||
}
|
|
||||||
|
|
||||||
if u, ok := srv.GetUserByName(s.User()); ok {
|
|
||||||
u.ctx = nil
|
|
||||||
u.proxy = nil
|
|
||||||
srv.ports.Delete(u.bindPort)
|
|
||||||
}
|
|
||||||
if _, err := fmt.Fprintf(s, "Goodbye! %s\n", s.User()); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
type server struct {
|
|
||||||
listenPort uint32
|
|
||||||
domainName string
|
|
||||||
domainSuffix string
|
|
||||||
bindHost string
|
|
||||||
|
|
||||||
portStart uint32
|
|
||||||
portEnd uint32
|
|
||||||
portNext uint32
|
|
||||||
|
|
||||||
ports sync.Map
|
|
||||||
users sync.Map
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *server) String() string {
|
|
||||||
var b strings.Builder
|
|
||||||
fmt.Fprintln(&b, "Server: ", s.domainName)
|
|
||||||
fmt.Fprintln(&b, " Port: ", s.listenPort)
|
|
||||||
fmt.Fprintln(&b, " Suffix: ", s.domainSuffix)
|
|
||||||
fmt.Fprintln(&b, " BindHost: ", s.bindHost)
|
|
||||||
fmt.Fprintf(&b, " PortRange: %d-%d\n", s.portStart, s.portEnd)
|
|
||||||
fmt.Fprintln(&b, " NextPort: ", s.portNext)
|
|
||||||
return b.String()
|
|
||||||
}
|
|
||||||
|
|
||||||
type user struct {
|
|
||||||
name string
|
|
||||||
pubkey ssh.PublicKey
|
|
||||||
bindHost string
|
|
||||||
bindPort uint32
|
|
||||||
ctx ssh.Context
|
|
||||||
proxy http.Handler
|
|
||||||
lastLogin time.Time
|
|
||||||
}
|
|
||||||
|
|
||||||
func (u *user) String() string {
|
|
||||||
var b strings.Builder
|
|
||||||
fmt.Fprintln(&b, "User: ", u.name)
|
|
||||||
fmt.Fprintf(&b, " Ptr: %p\n", u)
|
|
||||||
fmt.Fprintf(&b, " Pubkey: %x\n", u.pubkey)
|
|
||||||
fmt.Fprintln(&b, " Host: ", u.bindHost)
|
|
||||||
fmt.Fprintln(&b, " Port: ", u.bindPort)
|
|
||||||
fmt.Fprintf(&b, " Active: %t\n", u.ctx != nil)
|
|
||||||
fmt.Fprintln(&b, " LastLog:", u.lastLogin)
|
|
||||||
return b.String()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (srv *server) AddUser(pubkey ssh.PublicKey) *user {
|
|
||||||
u := &user{}
|
|
||||||
|
|
||||||
u.lastLogin = time.Now()
|
|
||||||
u.name = fingerprintHuman(pubkey)
|
|
||||||
u.name = strings.ToLower(u.name)
|
|
||||||
u.name = filterName.ReplaceAllString(u.name, "")
|
|
||||||
|
|
||||||
if g, ok := srv.users.LoadOrStore(u.name, u); ok {
|
|
||||||
u = g.(*user)
|
|
||||||
return u
|
|
||||||
}
|
|
||||||
|
|
||||||
u.pubkey = pubkey
|
|
||||||
u.bindPort = srv.nextPort()
|
|
||||||
u.bindHost = srv.bindHost
|
|
||||||
|
|
||||||
return u
|
|
||||||
}
|
|
||||||
func (srv *server) nextPort() uint32 {
|
|
||||||
if srv.portNext < srv.portStart || srv.portNext > srv.portEnd {
|
|
||||||
srv.portNext = srv.portStart
|
|
||||||
}
|
|
||||||
|
|
||||||
defer func() { srv.portNext++ }()
|
|
||||||
|
|
||||||
return srv.portNext
|
|
||||||
}
|
|
||||||
|
|
||||||
func (srv *server) GetUserByPort(port uint32) (*user, bool) {
|
|
||||||
if u, ok := srv.ports.Load(port); ok {
|
|
||||||
if u, ok := u.(*user); ok {
|
|
||||||
return u, true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return nil, false
|
|
||||||
}
|
|
||||||
func (srv *server) GetUserByName(name string) (*user, bool) {
|
|
||||||
if u, ok := srv.users.Load(name); ok {
|
|
||||||
if u, ok := u.(*user); ok {
|
|
||||||
return u, true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return nil, false
|
|
||||||
}
|
|
||||||
func (srv *server) ListUsers() []*user {
|
|
||||||
var lis []*user
|
|
||||||
srv.users.Range(func(key, value interface{}) bool {
|
|
||||||
if u, ok := value.(*user); ok {
|
|
||||||
lis = append(lis, u)
|
|
||||||
return true
|
|
||||||
} else {
|
|
||||||
fmt.Println(key, value)
|
|
||||||
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
})
|
|
||||||
|
|
||||||
return lis
|
|
||||||
}
|
|
||||||
func (srv *server) ListConnectedUsers() []*user {
|
|
||||||
var lis []*user
|
|
||||||
srv.ports.Range(func(key, value interface{}) bool {
|
|
||||||
if u, ok := value.(*user); ok {
|
|
||||||
lis = append(lis, u)
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
})
|
|
||||||
|
|
||||||
return lis
|
|
||||||
}
|
|
||||||
func (srv *server) optAuthUser() []ssh.Option {
|
|
||||||
return []ssh.Option{
|
|
||||||
ssh.PublicKeyAuth(func(ctx ssh.Context, key ssh.PublicKey) bool {
|
|
||||||
u, ok := srv.GetUserByName(ctx.User())
|
|
||||||
if !ok {
|
|
||||||
log.Println("user not found", ctx.User())
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
if ssh.KeysEqual(key, u.pubkey) {
|
|
||||||
log.Println("User:", ctx.User(), "Authorized:", u.bindHost, u.bindPort, ctx.ClientVersion(), ctx.SessionID(), ctx.LocalAddr(), ctx.RemoteAddr())
|
|
||||||
u.ctx = ctx
|
|
||||||
u.lastLogin = time.Now()
|
|
||||||
if _, loaded := srv.ports.LoadOrStore(u.bindPort, u); loaded {
|
|
||||||
log.Println("User:", ctx.User(), "already connected!")
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
return false
|
|
||||||
}),
|
|
||||||
func(cfg *ssh.Server) error {
|
|
||||||
hdlr := ssh.ForwardedTCPHandler{}
|
|
||||||
|
|
||||||
if cfg.RequestHandlers == nil {
|
|
||||||
cfg.RequestHandlers = make(map[string]ssh.RequestHandler, 2)
|
|
||||||
}
|
|
||||||
|
|
||||||
cfg.RequestHandlers["tcpip-forward"] = hdlr.HandleSSHRequest
|
|
||||||
cfg.RequestHandlers["cancel-tcpip-forward"] = hdlr.HandleSSHRequest
|
|
||||||
|
|
||||||
cfg.ReversePortForwardingCallback = func(ctx ssh.Context, bindHost string, bindPort uint32) bool {
|
|
||||||
u, ok := srv.GetUserByPort(bindPort)
|
|
||||||
if !ok {
|
|
||||||
log.Println("User port", bindPort, "not authorized.")
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
if u.ctx.SessionID() != ctx.SessionID() {
|
|
||||||
log.Println("Port", bindPort, "in use by", u.name, u.ctx.SessionID())
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
if bindHost != strings.Trim(u.bindHost, "[]") || bindPort != u.bindPort {
|
|
||||||
log.Println("User", ctx.User(), "Not Allowed: ", bindHost, bindPort, ctx.ClientVersion(), ctx.SessionID(), ctx.LocalAddr(), ctx.RemoteAddr())
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
log.Println("User", ctx.User(), "Allow Remote:", bindHost, bindPort, ctx.ClientVersion(), ctx.SessionID(), ctx.LocalAddr(), ctx.RemoteAddr())
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (srv *server) serveSSH(ctx context.Context, opts ...ssh.Option) func(l net.Listener) error {
|
|
||||||
return func(l net.Listener) error {
|
|
||||||
return ssh.Serve(
|
|
||||||
l,
|
|
||||||
srv.newSession(ctx),
|
|
||||||
opts...,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
func (srv *server) serveHTTP(ctx context.Context) func(net.Listener) error {
|
|
||||||
s := &http.Server{
|
|
||||||
ReadTimeout: 2500 * time.Millisecond,
|
|
||||||
WriteTimeout: 5 * time.Second,
|
|
||||||
Handler: http.DefaultServeMux,
|
|
||||||
BaseContext: func(net.Listener) context.Context { return ctx },
|
|
||||||
}
|
|
||||||
|
|
||||||
go func(ctx context.Context) {
|
|
||||||
<-ctx.Done()
|
|
||||||
s.Shutdown(context.Background())
|
|
||||||
}(ctx)
|
|
||||||
|
|
||||||
return s.Serve
|
|
||||||
}
|
|
||||||
|
|
||||||
func (srv *server) handleHTTP(rw http.ResponseWriter, r *http.Request) {
|
|
||||||
if strings.HasSuffix(r.Host, srv.domainSuffix) {
|
|
||||||
name := strings.TrimSuffix(r.Host, srv.domainSuffix)
|
|
||||||
u, ok := srv.GetUserByName(name)
|
|
||||||
if !ok || u.proxy == nil {
|
|
||||||
fmt.Fprintln(rw, "NOT FOUND", name)
|
|
||||||
}
|
|
||||||
|
|
||||||
u.proxy.ServeHTTP(rw, r)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if r.Method == http.MethodPost {
|
|
||||||
pubkey, _, _, _, err := ssh.ParseAuthorizedKey([]byte(r.FormValue("pub")))
|
|
||||||
if err != nil {
|
|
||||||
rw.WriteHeader(400)
|
|
||||||
fmt.Fprintln(rw, "ERR READING KEY")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
u := srv.AddUser(pubkey)
|
|
||||||
rw.WriteHeader(201)
|
|
||||||
fmt.Fprintf(rw, `ssh -T -p %v %v@%v -R "%v:%v:localhost:$LOCAL_PORT" -i $PRIV_KEY`+"\n", srv.listenPort, u.name, srv.domainName, u.bindHost, u.bindPort)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
fmt.Fprintln(rw, "Hello!")
|
|
||||||
fmt.Fprintln(rw, srv)
|
|
||||||
fmt.Fprintln(rw, "Registered Users")
|
|
||||||
for _, u := range srv.ListUsers() {
|
|
||||||
fmt.Fprintln(rw, u)
|
|
||||||
}
|
|
||||||
|
|
||||||
fmt.Fprintln(rw, "Connected Users")
|
|
||||||
for _, u := range srv.ListConnectedUsers() {
|
|
||||||
fmt.Fprintln(rw, u)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// serverMux is mux server which will multiplex a listener to serve an http
|
|
||||||
// server using the http.DefaultServeMux handler, as well as a grpc server
|
|
||||||
// to serve a protobuf generated grpc.serverMux
|
|
||||||
type serverMux struct {
|
|
||||||
Listener net.Listener
|
|
||||||
HTTP func(net.Listener) error
|
|
||||||
SSH func(net.Listener) error
|
|
||||||
ServeMux *http.ServeMux
|
|
||||||
}
|
|
||||||
|
|
||||||
func New(lis net.Listener, http, ssh func(net.Listener) error) *serverMux {
|
|
||||||
return &serverMux{
|
|
||||||
Listener: lis,
|
|
||||||
HTTP: http,
|
|
||||||
SSH: ssh,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Serve begins serving a multiplexed server. Any errors returned before the stop
|
|
||||||
// signal is given indicate a failure of the server to start or an unexpected shutdown.
|
|
||||||
// Serve closes the listener once Shutdown has been triggered
|
|
||||||
func (m *serverMux) Serve(ctx context.Context) error {
|
|
||||||
errChanSSH := make(chan error)
|
|
||||||
errChanHTTP := make(chan error)
|
|
||||||
|
|
||||||
defer func() {
|
|
||||||
err := multierr.Combine(m.Listener.Close(), <-errChanSSH, <-errChanHTTP)
|
|
||||||
if err != nil {
|
|
||||||
log.Println(err)
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
mux := cmux.New(m.Listener)
|
|
||||||
httpL := mux.Match(cmux.HTTP1Fast())
|
|
||||||
sshL := mux.Match(cmux.Any())
|
|
||||||
|
|
||||||
go func() {
|
|
||||||
defer close(errChanSSH)
|
|
||||||
if err := m.SSH(sshL); err != nil {
|
|
||||||
switch err {
|
|
||||||
case cmux.ErrServerClosed:
|
|
||||||
log.Println("shutting down SSH Server")
|
|
||||||
default:
|
|
||||||
errChanSSH <- fmt.Errorf("failed to start SSH: %w", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
go func() {
|
|
||||||
defer close(errChanHTTP)
|
|
||||||
if err := m.HTTP(httpL); err != nil {
|
|
||||||
switch err {
|
|
||||||
case cmux.ErrServerClosed:
|
|
||||||
log.Println("shutting down HTTP Server")
|
|
||||||
default:
|
|
||||||
errChanHTTP <- fmt.Errorf("failed to start HTTP: %w", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
errChan := make(chan error)
|
|
||||||
go func() {
|
|
||||||
defer close(errChan)
|
|
||||||
err := mux.Serve()
|
|
||||||
if err != nil {
|
|
||||||
if strings.Contains(err.Error(), "use of closed network connection") {
|
|
||||||
log.Println("shutting down mux server")
|
|
||||||
} else {
|
|
||||||
errChan <- fmt.Errorf("failed to start server multiplexing: %w", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
log.Println("server started: multiplexed http/1, http/2",
|
|
||||||
"address", m.Listener.Addr().String(),
|
|
||||||
"multiplexed", "true",
|
|
||||||
)
|
|
||||||
|
|
||||||
defer mux.Close()
|
|
||||||
|
|
||||||
select {
|
|
||||||
case <-ctx.Done():
|
|
||||||
log.Println("stopping multiplexed server gracefully")
|
|
||||||
return nil
|
|
||||||
case err := <-errChanSSH:
|
|
||||||
return err
|
|
||||||
case err := <-errChanHTTP:
|
|
||||||
return err
|
|
||||||
case err := <-errChan:
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func fingerprintHuman(pubKey ssh.PublicKey) string {
|
|
||||||
sha256sum := sha256.Sum256(pubKey.Marshal())
|
|
||||||
h, _ := humanhash.Humanize(sha256sum[:], 3)
|
|
||||||
return h
|
|
||||||
}
|
|
||||||
|
|
|
||||||
106
mux.go
Normal file
106
mux.go
Normal file
|
|
@ -0,0 +1,106 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"net"
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/soheilhy/cmux"
|
||||||
|
"go.uber.org/multierr"
|
||||||
|
)
|
||||||
|
|
||||||
|
// serverMux is mux server which will multiplex a listener to serve an http
|
||||||
|
// server using the http.DefaultServeMux handler, as well as a grpc server
|
||||||
|
// to serve a protobuf generated grpc.serverMux
|
||||||
|
type serverMux struct {
|
||||||
|
Listener net.Listener
|
||||||
|
HTTP func(net.Listener) error
|
||||||
|
SSH func(net.Listener) error
|
||||||
|
ServeMux *http.ServeMux
|
||||||
|
}
|
||||||
|
|
||||||
|
func New(lis net.Listener, http, ssh func(net.Listener) error) *serverMux {
|
||||||
|
return &serverMux{
|
||||||
|
Listener: lis,
|
||||||
|
HTTP: http,
|
||||||
|
SSH: ssh,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Serve begins serving a multiplexed server. Any errors returned before the stop
|
||||||
|
// signal is given indicate a failure of the server to start or an unexpected shutdown.
|
||||||
|
// Serve closes the listener once Shutdown has been triggered
|
||||||
|
func (m *serverMux) Serve(ctx context.Context) error {
|
||||||
|
errChanSSH := make(chan error)
|
||||||
|
errChanHTTP := make(chan error)
|
||||||
|
|
||||||
|
defer func() {
|
||||||
|
err := multierr.Combine(m.Listener.Close(), <-errChanSSH, <-errChanHTTP)
|
||||||
|
if err != nil {
|
||||||
|
log.Println(err)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
mux := cmux.New(m.Listener)
|
||||||
|
httpL := mux.Match(cmux.HTTP1Fast())
|
||||||
|
sshL := mux.Match(cmux.Any())
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
defer close(errChanSSH)
|
||||||
|
if err := m.SSH(sshL); err != nil {
|
||||||
|
switch err {
|
||||||
|
case cmux.ErrServerClosed:
|
||||||
|
log.Println("shutting down SSH Server")
|
||||||
|
default:
|
||||||
|
errChanSSH <- fmt.Errorf("failed to start SSH: %w", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
defer close(errChanHTTP)
|
||||||
|
if err := m.HTTP(httpL); err != nil {
|
||||||
|
switch err {
|
||||||
|
case cmux.ErrServerClosed:
|
||||||
|
log.Println("shutting down HTTP Server")
|
||||||
|
default:
|
||||||
|
errChanHTTP <- fmt.Errorf("failed to start HTTP: %w", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
errChan := make(chan error)
|
||||||
|
go func() {
|
||||||
|
defer close(errChan)
|
||||||
|
err := mux.Serve()
|
||||||
|
if err != nil {
|
||||||
|
if strings.Contains(err.Error(), "use of closed network connection") {
|
||||||
|
log.Println("shutting down mux server")
|
||||||
|
} else {
|
||||||
|
errChan <- fmt.Errorf("failed to start server multiplexing: %w", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
log.Println("server started: multiplexed http/1, http/2",
|
||||||
|
"address", m.Listener.Addr().String(),
|
||||||
|
"multiplexed", "true",
|
||||||
|
)
|
||||||
|
|
||||||
|
defer mux.Close()
|
||||||
|
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
log.Println("stopping multiplexed server gracefully")
|
||||||
|
return nil
|
||||||
|
case err := <-errChanSSH:
|
||||||
|
return err
|
||||||
|
case err := <-errChanHTTP:
|
||||||
|
return err
|
||||||
|
case err := <-errChan:
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
311
server.go
Normal file
311
server.go
Normal file
|
|
@ -0,0 +1,311 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"crypto/sha256"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"net"
|
||||||
|
"net/http"
|
||||||
|
"net/http/httputil"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/gliderlabs/ssh"
|
||||||
|
"github.com/wolfeidau/humanhash"
|
||||||
|
)
|
||||||
|
|
||||||
|
type user struct {
|
||||||
|
name string
|
||||||
|
pubkey ssh.PublicKey
|
||||||
|
bindHost string
|
||||||
|
bindPort uint32
|
||||||
|
ctx ssh.Context
|
||||||
|
proxy http.Handler
|
||||||
|
lastLogin time.Time
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u *user) String() string {
|
||||||
|
var b strings.Builder
|
||||||
|
fmt.Fprintln(&b, "User: ", u.name)
|
||||||
|
fmt.Fprintf(&b, " Ptr: %p\n", u)
|
||||||
|
fmt.Fprintf(&b, " Pubkey: %x\n", u.pubkey)
|
||||||
|
fmt.Fprintln(&b, " Host: ", u.bindHost)
|
||||||
|
fmt.Fprintln(&b, " Port: ", u.bindPort)
|
||||||
|
fmt.Fprintf(&b, " Active: %t\n", u.ctx != nil)
|
||||||
|
fmt.Fprintln(&b, " LastLog:", u.lastLogin)
|
||||||
|
return b.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
type server struct {
|
||||||
|
listenPort uint32
|
||||||
|
domainName string
|
||||||
|
domainSuffix string
|
||||||
|
bindHost string
|
||||||
|
|
||||||
|
portStart uint32
|
||||||
|
portEnd uint32
|
||||||
|
portNext uint32
|
||||||
|
|
||||||
|
ports sync.Map
|
||||||
|
users sync.Map
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *server) String() string {
|
||||||
|
var b strings.Builder
|
||||||
|
fmt.Fprintln(&b, "Server: ", s.domainName)
|
||||||
|
fmt.Fprintln(&b, " Port: ", s.listenPort)
|
||||||
|
fmt.Fprintln(&b, " Suffix: ", s.domainSuffix)
|
||||||
|
fmt.Fprintln(&b, " BindHost: ", s.bindHost)
|
||||||
|
fmt.Fprintf(&b, " PortRange: %d-%d\n", s.portStart, s.portEnd)
|
||||||
|
fmt.Fprintln(&b, " NextPort: ", s.portNext)
|
||||||
|
return b.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
// User Operations
|
||||||
|
func (srv *server) addUser(pubkey ssh.PublicKey) *user {
|
||||||
|
u := &user{}
|
||||||
|
|
||||||
|
u.lastLogin = time.Now()
|
||||||
|
u.name = fingerprintHuman(pubkey)
|
||||||
|
u.name = strings.ToLower(u.name)
|
||||||
|
u.name = filterName.ReplaceAllString(u.name, "")
|
||||||
|
|
||||||
|
if g, ok := srv.users.LoadOrStore(u.name, u); ok {
|
||||||
|
u = g.(*user)
|
||||||
|
return u
|
||||||
|
}
|
||||||
|
|
||||||
|
u.pubkey = pubkey
|
||||||
|
u.bindPort = srv.nextPort()
|
||||||
|
u.bindHost = srv.bindHost
|
||||||
|
|
||||||
|
return u
|
||||||
|
}
|
||||||
|
func (srv *server) disconnectUser(name string) {
|
||||||
|
if u, ok := srv.getUserByName(name); ok {
|
||||||
|
u.ctx = nil
|
||||||
|
u.proxy = nil
|
||||||
|
srv.ports.Delete(u.bindPort)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
func (srv *server) getUserByPort(port uint32) (*user, bool) {
|
||||||
|
if u, ok := srv.ports.Load(port); ok {
|
||||||
|
if u, ok := u.(*user); ok {
|
||||||
|
return u, true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil, false
|
||||||
|
}
|
||||||
|
func (srv *server) getUserByName(name string) (*user, bool) {
|
||||||
|
if u, ok := srv.users.Load(name); ok {
|
||||||
|
if u, ok := u.(*user); ok {
|
||||||
|
return u, true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil, false
|
||||||
|
}
|
||||||
|
func (srv *server) listUsers() []*user {
|
||||||
|
var lis []*user
|
||||||
|
srv.users.Range(func(key, value interface{}) bool {
|
||||||
|
if u, ok := value.(*user); ok {
|
||||||
|
lis = append(lis, u)
|
||||||
|
return true
|
||||||
|
} else {
|
||||||
|
fmt.Println(key, value)
|
||||||
|
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
})
|
||||||
|
|
||||||
|
return lis
|
||||||
|
}
|
||||||
|
func (srv *server) listConnectedUsers() []*user {
|
||||||
|
var lis []*user
|
||||||
|
srv.ports.Range(func(key, value interface{}) bool {
|
||||||
|
if u, ok := value.(*user); ok {
|
||||||
|
lis = append(lis, u)
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
})
|
||||||
|
|
||||||
|
return lis
|
||||||
|
}
|
||||||
|
func (srv *server) nextPort() uint32 {
|
||||||
|
if srv.portNext < srv.portStart || srv.portNext > srv.portEnd {
|
||||||
|
srv.portNext = srv.portStart
|
||||||
|
}
|
||||||
|
|
||||||
|
defer func() { srv.portNext++ }()
|
||||||
|
|
||||||
|
return srv.portNext
|
||||||
|
}
|
||||||
|
|
||||||
|
// SSH Operations
|
||||||
|
func (srv *server) serveSSH(ctx context.Context, opts ...ssh.Option) func(l net.Listener) error {
|
||||||
|
return func(l net.Listener) error {
|
||||||
|
return ssh.Serve(
|
||||||
|
l,
|
||||||
|
srv.newSession(ctx),
|
||||||
|
opts...,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
func (srv *server) newSession(ctx context.Context) func(ssh.Session) {
|
||||||
|
return func(s ssh.Session) {
|
||||||
|
if _, err := fmt.Fprintf(s, "Hello %s\n", s.User()); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if u, ok := srv.getUserByName(s.User()); ok {
|
||||||
|
host := fmt.Sprintf("%v:%v", u.bindHost, u.bindPort)
|
||||||
|
director := func(req *http.Request) {
|
||||||
|
if h := req.Header.Get("X-Forwarded-Host"); h == "" {
|
||||||
|
req.Header.Set("X-Forwarded-Host", req.Host)
|
||||||
|
}
|
||||||
|
req.Header.Set("X-Origin-Host", host)
|
||||||
|
req.URL.Scheme = "http"
|
||||||
|
req.URL.Host = host
|
||||||
|
|
||||||
|
requestDump, err := httputil.DumpRequest(req, req.Method == http.MethodPost || req.Method == http.MethodPut)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(err)
|
||||||
|
}
|
||||||
|
fmt.Fprintln(s, string(requestDump))
|
||||||
|
}
|
||||||
|
u.proxy = &httputil.ReverseProxy{Director: director}
|
||||||
|
fmt.Fprintf(s, "Created HTTP listener at: %v%v\n\n", u.name, srv.domainSuffix)
|
||||||
|
}
|
||||||
|
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
log.Println("server shutting down")
|
||||||
|
case <-s.Context().Done():
|
||||||
|
log.Println("user", s.User(), "disconnected")
|
||||||
|
}
|
||||||
|
|
||||||
|
srv.disconnectUser(s.User())
|
||||||
|
if _, err := fmt.Fprintf(s, "Goodbye! %s\n", s.User()); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
func (srv *server) optAuthUser() []ssh.Option {
|
||||||
|
return []ssh.Option{
|
||||||
|
ssh.PublicKeyAuth(func(ctx ssh.Context, key ssh.PublicKey) bool {
|
||||||
|
u, ok := srv.getUserByName(ctx.User())
|
||||||
|
if !ok {
|
||||||
|
log.Println("user not found", ctx.User())
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
if ssh.KeysEqual(key, u.pubkey) {
|
||||||
|
log.Println("User:", ctx.User(), "Authorized:", u.bindHost, u.bindPort, ctx.ClientVersion(), ctx.SessionID(), ctx.LocalAddr(), ctx.RemoteAddr())
|
||||||
|
u.ctx = ctx
|
||||||
|
u.lastLogin = time.Now()
|
||||||
|
if _, loaded := srv.ports.LoadOrStore(u.bindPort, u); loaded {
|
||||||
|
log.Println("User:", ctx.User(), "already connected!")
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
return false
|
||||||
|
}),
|
||||||
|
func(cfg *ssh.Server) error {
|
||||||
|
hdlr := ssh.ForwardedTCPHandler{}
|
||||||
|
|
||||||
|
if cfg.RequestHandlers == nil {
|
||||||
|
cfg.RequestHandlers = make(map[string]ssh.RequestHandler, 2)
|
||||||
|
}
|
||||||
|
|
||||||
|
cfg.RequestHandlers["tcpip-forward"] = hdlr.HandleSSHRequest
|
||||||
|
cfg.RequestHandlers["cancel-tcpip-forward"] = hdlr.HandleSSHRequest
|
||||||
|
|
||||||
|
cfg.ReversePortForwardingCallback = func(ctx ssh.Context, bindHost string, bindPort uint32) bool {
|
||||||
|
u, ok := srv.getUserByPort(bindPort)
|
||||||
|
if !ok {
|
||||||
|
log.Println("User port", bindPort, "not authorized.")
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
if u.ctx.SessionID() != ctx.SessionID() {
|
||||||
|
log.Println("Port", bindPort, "in use by", u.name, u.ctx.SessionID())
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
if bindHost != strings.Trim(u.bindHost, "[]") || bindPort != u.bindPort {
|
||||||
|
log.Println("User", ctx.User(), "Not Allowed: ", bindHost, bindPort, ctx.ClientVersion(), ctx.SessionID(), ctx.LocalAddr(), ctx.RemoteAddr())
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Println("User", ctx.User(), "Allow Remote:", bindHost, bindPort, ctx.ClientVersion(), ctx.SessionID(), ctx.LocalAddr(), ctx.RemoteAddr())
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// HTTP Operations
|
||||||
|
func (srv *server) serveHTTP(ctx context.Context) func(net.Listener) error {
|
||||||
|
s := &http.Server{
|
||||||
|
ReadTimeout: 2500 * time.Millisecond,
|
||||||
|
WriteTimeout: 5 * time.Second,
|
||||||
|
Handler: http.DefaultServeMux,
|
||||||
|
BaseContext: func(net.Listener) context.Context { return ctx },
|
||||||
|
}
|
||||||
|
|
||||||
|
go func(ctx context.Context) {
|
||||||
|
<-ctx.Done()
|
||||||
|
s.Shutdown(context.Background())
|
||||||
|
}(ctx)
|
||||||
|
|
||||||
|
return s.Serve
|
||||||
|
}
|
||||||
|
func (srv *server) handleHTTP(rw http.ResponseWriter, r *http.Request) {
|
||||||
|
if strings.HasSuffix(r.Host, srv.domainSuffix) {
|
||||||
|
name := strings.TrimSuffix(r.Host, srv.domainSuffix)
|
||||||
|
u, ok := srv.getUserByName(name)
|
||||||
|
if !ok || u.proxy == nil {
|
||||||
|
fmt.Fprintln(rw, "NOT FOUND", name)
|
||||||
|
}
|
||||||
|
|
||||||
|
u.proxy.ServeHTTP(rw, r)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if r.Method == http.MethodPost {
|
||||||
|
pubkey, _, _, _, err := ssh.ParseAuthorizedKey([]byte(r.FormValue("pub")))
|
||||||
|
if err != nil {
|
||||||
|
rw.WriteHeader(400)
|
||||||
|
fmt.Fprintln(rw, "ERR READING KEY")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
u := srv.addUser(pubkey)
|
||||||
|
rw.WriteHeader(201)
|
||||||
|
fmt.Fprintf(rw, `ssh -T -p %v %v@%v -R "%v:%v:localhost:$LOCAL_PORT" -i $PRIV_KEY`+"\n", srv.listenPort, u.name, srv.domainName, u.bindHost, u.bindPort)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Fprintln(rw, "Hello!")
|
||||||
|
fmt.Fprintln(rw, srv)
|
||||||
|
fmt.Fprintln(rw, "Registered Users")
|
||||||
|
for _, u := range srv.listUsers() {
|
||||||
|
fmt.Fprintln(rw, u)
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Fprintln(rw, "Connected Users")
|
||||||
|
for _, u := range srv.listConnectedUsers() {
|
||||||
|
fmt.Fprintln(rw, u)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func fingerprintHuman(pubKey ssh.PublicKey) string {
|
||||||
|
sha256sum := sha256.Sum256(pubKey.Marshal())
|
||||||
|
h, _ := humanhash.Humanize(sha256sum[:], 3)
|
||||||
|
return h
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user