我们可以将请求表服务服务器发送到客户端并通过QUIC/HTTP3获取响应吗?

发布于 2025-02-01 16:23:43 字数 4008 浏览 4 评论 0原文

我正在使用quic-go来实现我的想法,我需要服务器将请求转发给客户端以获取响应,就像我们这样做一样,该客户端通常将请求发送到Web服务器。但是,使用quic-go,在设置了连接后,服务器可以初始化流以将请求发送到客户端并获取响应吗?我尝试过,但没有使它起作用。以下代码来自示例dir的echo.go,我添加了评论行之间的两个部分。

package main

import (
    "context"
    "crypto/rand"
    "crypto/rsa"
    "crypto/tls"
    "crypto/x509"
    "encoding/pem"
    "fmt"
    "io"
    "log"
    "math/big"

    "github.com/lucas-clemente/quic-go"
)

const addr = "localhost:4242"

const message = "foobar"

// We start a server echoing data on the first stream the client opens,
// then connect with a client, send the message, and wait for its receipt.
func main() {
    go func() { log.Fatal(echoServer()) }()

    err := clientMain()
    if err != nil {
        panic(err)
    }
}

// Start a server that echos all data on the first stream opened by the client
func echoServer() error {
    listener, err := quic.ListenAddr(addr, generateTLSConfig(), nil)
    if err != nil {
        return err
    }
    conn, err := listener.Accept(context.Background())
    if err != nil {
        return err
    }
    stream, err := conn.AcceptStream(context.Background())
    if err != nil {
        panic(err)
    }
    // Echo through the loggingWriter
    _, err = io.Copy(loggingWriter{stream}, stream)
    if err != nil {
        panic(err)
    }

    //------------------------------
    stream1, err := conn.OpenStream()
    if err != nil {
        panic(err)
    }

    message := "aaaaa"

    fmt.Printf("2-Server: Sending '%s'\n", message)

    _, err = stream1.Write([]byte(message))
    //------------------------------
    return err
}

func clientMain() error {
    tlsConf := &tls.Config{
        InsecureSkipVerify: true,
        NextProtos:         []string{"quic-echo-example"},
    }
    conn, err := quic.DialAddr(addr, tlsConf, nil)
    if err != nil {
        return err
    }

    stream, err := conn.OpenStreamSync(context.Background())
    if err != nil {
        return err
    }

    fmt.Printf("Client: Sending '%s'\n", message)
    _, err = stream.Write([]byte(message))
    if err != nil {
        return err
    }

    buf := make([]byte, len(message))
    _, err = io.ReadFull(stream, buf)
    if err != nil {
        return err
    }
    fmt.Printf("Client: Got '%s'\n", buf)
    err = stream.Close()
    if err != nil {
        return err
    }

    //-------------------------------
    for {
        stream1, err := conn.AcceptStream(context.Background())
        if err != nil {
            panic(err)
        }

        buf1 := make([]byte, len(message))
        _, err = io.ReadFull(stream1, buf1)
        if err != nil {
            panic(err)
        }
        fmt.Printf("2-Client: Got '%s'\n", buf1)

        err = stream1.Close()
        if err != nil {
            panic(err)
        }
    }
    //-------------------------------

    return nil
}

// A wrapper for io.Writer that also logs the message.
type loggingWriter struct{ io.Writer }

func (w loggingWriter) Write(b []byte) (int, error) {
    fmt.Printf("Server: Got '%s'\n", string(b))
    return w.Writer.Write(b)
}

// Setup a bare-bones TLS config for the server
func generateTLSConfig() *tls.Config {
    key, err := rsa.GenerateKey(rand.Reader, 1024)
    if err != nil {
        panic(err)
    }
    template := x509.Certificate{SerialNumber: big.NewInt(1)}
    certDER, err := x509.CreateCertificate(rand.Reader, &template, &template, &key.PublicKey, key)
    if err != nil {
        panic(err)
    }
    keyPEM := pem.EncodeToMemory(&pem.Block{Type: "RSA PRIVATE KEY", Bytes: x509.MarshalPKCS1PrivateKey(key)})
    certPEM := pem.EncodeToMemory(&pem.Block{Type: "CERTIFICATE", Bytes: certDER})

    tlsCert, err := tls.X509KeyPair(certPEM, keyPEM)
    if err != nil {
        panic(err)
    }
    return &tls.Config{
        Certificates: []tls.Certificate{tlsCert},
        NextProtos:   []string{"quic-echo-example"},
    }
}

I'm using quic-go to implement my thought, I need the server to forwardly send request to client to get response, just like we do that client sends request to web server commonly. But with quic-go, after connection is setup, can server initialize streams to send request to client and get responses? I did a trying but haven't made it work. The code below is from the echo.go of example dir, the two parts between comment lines are added by me.

package main

import (
    "context"
    "crypto/rand"
    "crypto/rsa"
    "crypto/tls"
    "crypto/x509"
    "encoding/pem"
    "fmt"
    "io"
    "log"
    "math/big"

    "github.com/lucas-clemente/quic-go"
)

const addr = "localhost:4242"

const message = "foobar"

// We start a server echoing data on the first stream the client opens,
// then connect with a client, send the message, and wait for its receipt.
func main() {
    go func() { log.Fatal(echoServer()) }()

    err := clientMain()
    if err != nil {
        panic(err)
    }
}

// Start a server that echos all data on the first stream opened by the client
func echoServer() error {
    listener, err := quic.ListenAddr(addr, generateTLSConfig(), nil)
    if err != nil {
        return err
    }
    conn, err := listener.Accept(context.Background())
    if err != nil {
        return err
    }
    stream, err := conn.AcceptStream(context.Background())
    if err != nil {
        panic(err)
    }
    // Echo through the loggingWriter
    _, err = io.Copy(loggingWriter{stream}, stream)
    if err != nil {
        panic(err)
    }

    //------------------------------
    stream1, err := conn.OpenStream()
    if err != nil {
        panic(err)
    }

    message := "aaaaa"

    fmt.Printf("2-Server: Sending '%s'\n", message)

    _, err = stream1.Write([]byte(message))
    //------------------------------
    return err
}

func clientMain() error {
    tlsConf := &tls.Config{
        InsecureSkipVerify: true,
        NextProtos:         []string{"quic-echo-example"},
    }
    conn, err := quic.DialAddr(addr, tlsConf, nil)
    if err != nil {
        return err
    }

    stream, err := conn.OpenStreamSync(context.Background())
    if err != nil {
        return err
    }

    fmt.Printf("Client: Sending '%s'\n", message)
    _, err = stream.Write([]byte(message))
    if err != nil {
        return err
    }

    buf := make([]byte, len(message))
    _, err = io.ReadFull(stream, buf)
    if err != nil {
        return err
    }
    fmt.Printf("Client: Got '%s'\n", buf)
    err = stream.Close()
    if err != nil {
        return err
    }

    //-------------------------------
    for {
        stream1, err := conn.AcceptStream(context.Background())
        if err != nil {
            panic(err)
        }

        buf1 := make([]byte, len(message))
        _, err = io.ReadFull(stream1, buf1)
        if err != nil {
            panic(err)
        }
        fmt.Printf("2-Client: Got '%s'\n", buf1)

        err = stream1.Close()
        if err != nil {
            panic(err)
        }
    }
    //-------------------------------

    return nil
}

// A wrapper for io.Writer that also logs the message.
type loggingWriter struct{ io.Writer }

func (w loggingWriter) Write(b []byte) (int, error) {
    fmt.Printf("Server: Got '%s'\n", string(b))
    return w.Writer.Write(b)
}

// Setup a bare-bones TLS config for the server
func generateTLSConfig() *tls.Config {
    key, err := rsa.GenerateKey(rand.Reader, 1024)
    if err != nil {
        panic(err)
    }
    template := x509.Certificate{SerialNumber: big.NewInt(1)}
    certDER, err := x509.CreateCertificate(rand.Reader, &template, &template, &key.PublicKey, key)
    if err != nil {
        panic(err)
    }
    keyPEM := pem.EncodeToMemory(&pem.Block{Type: "RSA PRIVATE KEY", Bytes: x509.MarshalPKCS1PrivateKey(key)})
    certPEM := pem.EncodeToMemory(&pem.Block{Type: "CERTIFICATE", Bytes: certDER})

    tlsCert, err := tls.X509KeyPair(certPEM, keyPEM)
    if err != nil {
        panic(err)
    }
    return &tls.Config{
        Certificates: []tls.Certificate{tlsCert},
        NextProtos:   []string{"quic-echo-example"},
    }
}

如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。

评论(1

五里雾 2025-02-08 16:23:43

你很近。

请注意,在客户端有机会阅读和处理数据之前,请注意服务器的写入方式,然后返回,然后访问log.fatal

注意消息变量,一个长度长6,另一个长度只有5个长度。

正确关闭您的流,服务器在移交之前没有结束。

package main

import (
    "context"
    "crypto/rand"
    "crypto/rsa"
    "crypto/tls"
    "crypto/x509"
    "encoding/pem"
    "fmt"
    "io"
    "log"
    "math/big"

    "github.com/lucas-clemente/quic-go"
)

const addr = "localhost:4242"

const message = "foobar"

// We start a server echoing data on the first stream the client opens,
// then connect with a client, send the message, and wait for its receipt.
func main() {
    go func() {
        err := echoServer()
        if err != nil {
            log.Println(err)
        }
    }()

    err := clientMain()
    if err != nil {
        panic(err)
    }
}

// Start a server that echos all data on the first stream opened by the client
func echoServer() error {
    listener, err := quic.ListenAddr(addr, generateTLSConfig(), nil)
    if err != nil {
        return err
    }
    conn, err := listener.Accept(context.Background())
    if err != nil {
        return err
    }
    stream, err := conn.AcceptStream(context.Background())
    if err != nil {
        panic(err)
    }
    // Echo through the loggingWriter
    _, err = io.Copy(loggingWriter{stream}, stream)
    if err != nil {
        log.Println(err)
        return err
    }

    //------------------------------
    fmt.Printf("Server: open stream\n")
    stream1, err := conn.OpenStream()
    if err != nil {
        return err
    }

    fmt.Printf("2-Server: Sending '%s'\n", message)

    var n int
    n, err = stream1.Write([]byte(message))
    fmt.Printf("Server: write %v %v\n", n, err)
    stream1.Close()
    //------------------------------
    return err
}

func clientMain() error {
    tlsConf := &tls.Config{
        InsecureSkipVerify: true,
        NextProtos:         []string{"quic-echo-example"},
    }
    conn, err := quic.DialAddr(addr, tlsConf, nil)
    if err != nil {
        return err
    }

    stream, err := conn.OpenStreamSync(context.Background())
    if err != nil {
        return err
    }

    fmt.Printf("Client: Sending '%s'\n", message)
    _, err = stream.Write([]byte(message))
    if err != nil {
        return err
    }

    buf := make([]byte, len(message))
    _, err = io.ReadFull(stream, buf)
    if err != nil {
        return err
    }
    fmt.Printf("Client: Got '%s'\n", buf)
    err = stream.Close()
    if err != nil {
        return err
    }

    //-------------------------------
    for {
        fmt.Printf("Client: accept stream\n")
        stream1, err := conn.AcceptStream(context.Background())
        if err != nil {
            return err
        }

        fmt.Printf("Client: got stream %v\n", err)
        buf1 := make([]byte, len(message))
        _, err = io.ReadFull(stream1, buf1)
        if err != nil {
            return err
        }
        fmt.Printf("2-Client: Got '%s'\n", buf1)

        err = stream1.Close()
        if err != nil {
            return err
        }
        break
    }
    //-------------------------------

    return nil
}

// A wrapper for io.Writer that also logs the message.
type loggingWriter struct{ io.Writer }

func (w loggingWriter) Write(b []byte) (int, error) {
    fmt.Printf("Server: Got '%s'\n", string(b))
    return w.Writer.Write(b)
}

// Setup a bare-bones TLS config for the server
func generateTLSConfig() *tls.Config {
    key, err := rsa.GenerateKey(rand.Reader, 1024)
    if err != nil {
        panic(err)
    }
    template := x509.Certificate{SerialNumber: big.NewInt(1)}
    certDER, err := x509.CreateCertificate(rand.Reader, &template, &template, &key.PublicKey, key)
    if err != nil {
        panic(err)
    }
    keyPEM := pem.EncodeToMemory(&pem.Block{Type: "RSA PRIVATE KEY", Bytes: x509.MarshalPKCS1PrivateKey(key)})
    certPEM := pem.EncodeToMemory(&pem.Block{Type: "CERTIFICATE", Bytes: certDER})

    tlsCert, err := tls.X509KeyPair(certPEM, keyPEM)
    if err != nil {
        panic(err)
    }
    return &tls.Config{
        Certificates: []tls.Certificate{tlsCert},
        NextProtos:   []string{"quic-echo-example"},
    }
}

You are close.

Notice how the server writes, then returns, it reaches out to log.Fatal before the client had a chance to read and handle the data.

Take care to the message variable, one is 6 length long, the other one is only 5 length long.

Properly close your stream, the server was not ending it ending before handing over;

package main

import (
    "context"
    "crypto/rand"
    "crypto/rsa"
    "crypto/tls"
    "crypto/x509"
    "encoding/pem"
    "fmt"
    "io"
    "log"
    "math/big"

    "github.com/lucas-clemente/quic-go"
)

const addr = "localhost:4242"

const message = "foobar"

// We start a server echoing data on the first stream the client opens,
// then connect with a client, send the message, and wait for its receipt.
func main() {
    go func() {
        err := echoServer()
        if err != nil {
            log.Println(err)
        }
    }()

    err := clientMain()
    if err != nil {
        panic(err)
    }
}

// Start a server that echos all data on the first stream opened by the client
func echoServer() error {
    listener, err := quic.ListenAddr(addr, generateTLSConfig(), nil)
    if err != nil {
        return err
    }
    conn, err := listener.Accept(context.Background())
    if err != nil {
        return err
    }
    stream, err := conn.AcceptStream(context.Background())
    if err != nil {
        panic(err)
    }
    // Echo through the loggingWriter
    _, err = io.Copy(loggingWriter{stream}, stream)
    if err != nil {
        log.Println(err)
        return err
    }

    //------------------------------
    fmt.Printf("Server: open stream\n")
    stream1, err := conn.OpenStream()
    if err != nil {
        return err
    }

    fmt.Printf("2-Server: Sending '%s'\n", message)

    var n int
    n, err = stream1.Write([]byte(message))
    fmt.Printf("Server: write %v %v\n", n, err)
    stream1.Close()
    //------------------------------
    return err
}

func clientMain() error {
    tlsConf := &tls.Config{
        InsecureSkipVerify: true,
        NextProtos:         []string{"quic-echo-example"},
    }
    conn, err := quic.DialAddr(addr, tlsConf, nil)
    if err != nil {
        return err
    }

    stream, err := conn.OpenStreamSync(context.Background())
    if err != nil {
        return err
    }

    fmt.Printf("Client: Sending '%s'\n", message)
    _, err = stream.Write([]byte(message))
    if err != nil {
        return err
    }

    buf := make([]byte, len(message))
    _, err = io.ReadFull(stream, buf)
    if err != nil {
        return err
    }
    fmt.Printf("Client: Got '%s'\n", buf)
    err = stream.Close()
    if err != nil {
        return err
    }

    //-------------------------------
    for {
        fmt.Printf("Client: accept stream\n")
        stream1, err := conn.AcceptStream(context.Background())
        if err != nil {
            return err
        }

        fmt.Printf("Client: got stream %v\n", err)
        buf1 := make([]byte, len(message))
        _, err = io.ReadFull(stream1, buf1)
        if err != nil {
            return err
        }
        fmt.Printf("2-Client: Got '%s'\n", buf1)

        err = stream1.Close()
        if err != nil {
            return err
        }
        break
    }
    //-------------------------------

    return nil
}

// A wrapper for io.Writer that also logs the message.
type loggingWriter struct{ io.Writer }

func (w loggingWriter) Write(b []byte) (int, error) {
    fmt.Printf("Server: Got '%s'\n", string(b))
    return w.Writer.Write(b)
}

// Setup a bare-bones TLS config for the server
func generateTLSConfig() *tls.Config {
    key, err := rsa.GenerateKey(rand.Reader, 1024)
    if err != nil {
        panic(err)
    }
    template := x509.Certificate{SerialNumber: big.NewInt(1)}
    certDER, err := x509.CreateCertificate(rand.Reader, &template, &template, &key.PublicKey, key)
    if err != nil {
        panic(err)
    }
    keyPEM := pem.EncodeToMemory(&pem.Block{Type: "RSA PRIVATE KEY", Bytes: x509.MarshalPKCS1PrivateKey(key)})
    certPEM := pem.EncodeToMemory(&pem.Block{Type: "CERTIFICATE", Bytes: certDER})

    tlsCert, err := tls.X509KeyPair(certPEM, keyPEM)
    if err != nil {
        panic(err)
    }
    return &tls.Config{
        Certificates: []tls.Certificate{tlsCert},
        NextProtos:   []string{"quic-echo-example"},
    }
}
~没有更多了~
我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
原文