fix
This commit is contained in:
Smile Rex
183
main.go
183
main.go
@@ -1,164 +1,65 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"log"
|
||||
"crypto/tls"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"strings"
|
||||
"sync"
|
||||
|
||||
"github.com/caddyserver/certmagic"
|
||||
"github.com/quic-go/quic-go/http3"
|
||||
"github.com/quic-go/webtransport-go"
|
||||
"golang.org/x/crypto/acme/autocert"
|
||||
)
|
||||
|
||||
type Room struct {
|
||||
conns map[*webtransport.Session]bool
|
||||
mu sync.Mutex
|
||||
}
|
||||
|
||||
type Server struct {
|
||||
rooms map[string]*Room
|
||||
mu sync.Mutex
|
||||
}
|
||||
|
||||
func NewServer() *Server {
|
||||
return &Server{rooms: make(map[string]*Room)}
|
||||
}
|
||||
|
||||
func (s *Server) getRoom(name string) *Room {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
if room, ok := s.rooms[name]; ok {
|
||||
return room
|
||||
}
|
||||
room := &Room{conns: make(map[*webtransport.Session]bool)}
|
||||
s.rooms[name] = room
|
||||
return room
|
||||
}
|
||||
|
||||
func (s *Server) handleSession(roomName string, sess *webtransport.Session) {
|
||||
room := s.getRoom(roomName)
|
||||
room.mu.Lock()
|
||||
room.conns[sess] = true
|
||||
room.mu.Unlock()
|
||||
|
||||
log.Printf("Joined room: %s | Total: %d", roomName, len(room.conns))
|
||||
|
||||
defer func() {
|
||||
room.mu.Lock()
|
||||
delete(room.conns, sess)
|
||||
room.mu.Unlock()
|
||||
log.Printf("Left room: %s", roomName)
|
||||
}()
|
||||
|
||||
for {
|
||||
stream, err := sess.AcceptStream(context.Background())
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
go s.handleStream(room, sess, stream)
|
||||
}
|
||||
}
|
||||
|
||||
func (s *Server) handleStream(room *Room, sender *webtransport.Session, stream *webtransport.Stream) {
|
||||
defer stream.Close()
|
||||
buf := make([]byte, 4096)
|
||||
for {
|
||||
n, err := stream.Read(buf)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
data := append([]byte(nil), buf[:n]...)
|
||||
|
||||
room.mu.Lock()
|
||||
for conn := range room.conns {
|
||||
if conn == sender {
|
||||
continue
|
||||
}
|
||||
// Отправляем данные другим участникам
|
||||
go func(c *webtransport.Session, d []byte) {
|
||||
out, err := c.OpenStream()
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
defer out.Close()
|
||||
out.Write(d)
|
||||
}(conn, data)
|
||||
}
|
||||
room.mu.Unlock()
|
||||
}
|
||||
}
|
||||
|
||||
func main() {
|
||||
domain := "qgo-server.quizer.space"
|
||||
email := "serverussnap@outlook.com"
|
||||
|
||||
// 1. Настройка CertMagic
|
||||
certmagic.DefaultACME.Email = email
|
||||
certmagic.DefaultACME.Agreed = true
|
||||
|
||||
cfg, err := certmagic.TLS([]string{domain})
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
// Важно для QUIC: устанавливаем NextProtos
|
||||
cfg.NextProtos = []string{"h3", "h2", "http/1.1"}
|
||||
|
||||
server := NewServer()
|
||||
|
||||
// 2. Инициализация WebTransport сервера
|
||||
wt := &webtransport.Server{
|
||||
CheckOrigin: func(r *http.Request) bool { return true },
|
||||
// 1. Настройка менеджера сертификатов
|
||||
certManager := autocert.Manager{
|
||||
Prompt: autocert.AcceptTOS,
|
||||
HostPolicy: autocert.HostWhitelist("qgo-server.quizer.space"), // ЗАМЕНИТЕ НА ВАШ ДОМЕН
|
||||
Cache: autocert.DirCache("certs"), // Папка для хранения ключей
|
||||
}
|
||||
|
||||
mux := http.NewServeMux()
|
||||
mux.HandleFunc("/room/", func(w http.ResponseWriter, r *http.Request) {
|
||||
// Сообщаем браузеру о наличии HTTP/3
|
||||
w.Header().Set("Alt-Svc", `h3=":443"; ma=86400`)
|
||||
mux.HandleFunc("/api/hello", func(w http.ResponseWriter, r *http.Request) {
|
||||
// Добавляем CORS для вашего фронтенда
|
||||
w.Header().Set("Access-Control-Allow-Origin", "https://your-svelte-app.com")
|
||||
w.Header().Set("Access-Control-Allow-Methods", "GET, POST, OPTIONS")
|
||||
w.Header().Set("Alt-Svc", `h3=":443"; ma=86400`) // Порт 443 стандартный для HTTPS
|
||||
|
||||
if r.Method == http.MethodConnect {
|
||||
sess, err := wt.Upgrade(w, r)
|
||||
if err != nil {
|
||||
log.Printf("Upgrade error: %v", err)
|
||||
return
|
||||
}
|
||||
roomName := strings.TrimPrefix(r.URL.Path, "/room/")
|
||||
go server.handleSession(roomName, sess)
|
||||
return
|
||||
}
|
||||
|
||||
w.WriteHeader(http.StatusOK)
|
||||
w.Write([]byte("Waiting for WebTransport connection..."))
|
||||
fmt.Printf("Запрос по протоколу: %s\n", r.Proto)
|
||||
w.Write([]byte(`{"message": "Работает через Let's Encrypt и QUIC!"}`))
|
||||
})
|
||||
|
||||
// 3. HTTP/3 Сервер (UDP)
|
||||
h3Server := &http3.Server{
|
||||
Addr: ":443",
|
||||
Handler: mux,
|
||||
TLSConfig: cfg,
|
||||
EnableDatagrams: true,
|
||||
}
|
||||
wt.H3 = h3Server
|
||||
|
||||
// 4. Обычный HTTPS Сервер (TCP) — обязателен для первого запроса и Alt-Svc
|
||||
tcpServer := &http.Server{
|
||||
Addr: ":443",
|
||||
Handler: mux,
|
||||
TLSConfig: cfg,
|
||||
// Настройка TLS конфигурации для QUIC
|
||||
tlsConfig := &tls.Config{
|
||||
GetCertificate: certManager.GetCertificate,
|
||||
NextProtos: []string{"h3", "h3-29"}, // Поддержка разных версий HTTP/3
|
||||
}
|
||||
|
||||
log.Printf("Starting Server on %s (TCP/UDP 443)", domain)
|
||||
|
||||
// Запускаем TCP сервер в фоне
|
||||
// 2. Запуск HTTP/3 (UDP) на порту 443
|
||||
go func() {
|
||||
if err := tcpServer.ListenAndServeTLS("", ""); err != nil {
|
||||
log.Printf("TCP Server error: %v", err)
|
||||
server := &http3.Server{
|
||||
Addr: ":443",
|
||||
Handler: mux,
|
||||
TLSConfig: tlsConfig,
|
||||
}
|
||||
fmt.Println("Запуск QUIC на :443 UDP...")
|
||||
if err := server.ListenAndServe(); err != nil {
|
||||
fmt.Printf("Ошибка QUIC: %v\n", err)
|
||||
}
|
||||
}()
|
||||
|
||||
// Запускаем HTTP/3 сервер (основной поток)
|
||||
if err := h3Server.ListenAndServe(); err != nil {
|
||||
log.Fatalf("H3 Server failed: %v", err)
|
||||
// 3. Запуск стандартного HTTPS (TCP) на порту 443
|
||||
// Это также необходимо для выдачи сертификатов через ACME
|
||||
fmt.Println("Запуск HTTPS на :443 TCP...")
|
||||
httpsServer := &http.Server{
|
||||
Addr: ":443",
|
||||
Handler: mux,
|
||||
TLSConfig: tlsConfig,
|
||||
}
|
||||
|
||||
// Важно: для Let's Encrypt также нужен сервер на порту 80 (HTTP)
|
||||
go http.ListenAndServe(":80", certManager.HTTPHandler(nil))
|
||||
|
||||
if err := httpsServer.ListenAndServeTLS("", ""); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user