54 lines
1.1 KiB
Go
54 lines
1.1 KiB
Go
package server
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"fmt"
|
|
"golang.org/x/crypto/ssh"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
httpServer "tunnel_pls/http"
|
|
"tunnel_pls/utils"
|
|
)
|
|
|
|
type Server struct {
|
|
Conn *net.Listener
|
|
Config *ssh.ServerConfig
|
|
HttpServer *http.Server
|
|
}
|
|
|
|
func NewServer(config ssh.ServerConfig) *Server {
|
|
listener, err := net.Listen("tcp", ":2200")
|
|
if err != nil {
|
|
log.Fatalf("failed to listen on port 2200: %v", err)
|
|
return nil
|
|
}
|
|
go httpServer.Listen()
|
|
if utils.Getenv("tls_enabled") == "true" {
|
|
cert, err := tls.LoadX509KeyPair(utils.Getenv("cert_loc"), utils.Getenv("key_loc"))
|
|
if err != nil {
|
|
log.Fatal("Failed to load key pair:", err)
|
|
}
|
|
|
|
tlsConfig := &tls.Config{Certificates: []tls.Certificate{cert}, NextProtos: []string{"h2"}}
|
|
go httpServer.ListenTLS(tlsConfig)
|
|
}
|
|
return &Server{
|
|
Conn: &listener,
|
|
Config: &config,
|
|
}
|
|
}
|
|
|
|
func (s *Server) Start() {
|
|
fmt.Println("SSH server is starting on port 2200...")
|
|
for {
|
|
conn, err := (*s.Conn).Accept()
|
|
if err != nil {
|
|
log.Printf("failed to accept connection: %v", err)
|
|
continue
|
|
}
|
|
|
|
go s.handleConnection(conn)
|
|
}
|
|
}
|