You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
111 lines
2.6 KiB
111 lines
2.6 KiB
// Copyright 2018 The Chromium Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file.
|
|
|
|
// TODO(pthatcher): Use proper testing framework
|
|
|
|
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"io"
|
|
"fmt"
|
|
"log"
|
|
|
|
"osp"
|
|
)
|
|
|
|
func testMdns() {
|
|
// TODO(pthatcher): log error if it fails
|
|
ctx := context.Background()
|
|
instance := "TV"
|
|
port := 10000
|
|
go osp.RunMdnsServer(ctx, instance, port)
|
|
entries, err := osp.BrowseMdns(ctx)
|
|
if (err != nil) {
|
|
log.Fatalf("Failed to browse mDNS: %v\n", err)
|
|
}
|
|
for entry := range entries {
|
|
log.Println(fmt.Sprintf("%s at %s(%s or %s):%d", entry.Instance, entry.HostName, entry.AddrIPv4, entry.AddrIPv6, entry.Port))
|
|
}
|
|
log.Println("No more entries.")
|
|
}
|
|
|
|
func dialAndwriteQuicStream(ctx context.Context, hostname string, port int, msg []byte) {
|
|
clientSession, err := osp.DialAsQuicClient(ctx, hostname, port)
|
|
if err != nil {
|
|
log.Fatalln("Failed to dial QUIC:", err.Error())
|
|
}
|
|
clientStream, err := clientSession.OpenStreamSync()
|
|
if err != nil {
|
|
log.Fatalln("Failed to open QUIC stream:", err.Error())
|
|
}
|
|
_, err = clientStream.Write(msg)
|
|
if err != nil {
|
|
log.Fatalln("Failed to write to a QUIC stream:", err.Error())
|
|
}
|
|
}
|
|
|
|
func testQuic() {
|
|
const port = 4242
|
|
msg := []byte("Hello, World")
|
|
tlsCert, err := osp.GenerateTlsCert()
|
|
if err != nil {
|
|
log.Fatalln("Failed to generate TLS cert:", err.Error())
|
|
}
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
go dialAndwriteQuicStream(ctx, "localhost", port, msg)
|
|
|
|
serverStreams := make(chan io.ReadWriteCloser)
|
|
go osp.RunQuicServer(ctx, port, *tlsCert, serverStreams)
|
|
serverStream := <-serverStreams
|
|
b := make([]byte, len(msg))
|
|
_, err = io.ReadFull(serverStream, b)
|
|
if err != nil {
|
|
log.Fatalln("Failed to read stream:", err.Error())
|
|
}
|
|
log.Println(string(b))
|
|
}
|
|
|
|
func testMessages() {
|
|
msg := osp.PresentationStartRequest{
|
|
RequestID: 1,
|
|
PresentationID: "This is a Presentation ID. It must be very long.",
|
|
URL: "https://github.com/webscreens/openscreenprotocol",
|
|
}
|
|
|
|
var rw bytes.Buffer
|
|
osp.WriteMessage(msg, &rw)
|
|
msg2, err := osp.ReadMessage(&rw)
|
|
if err != nil {
|
|
log.Fatalln("Failed to read message.");
|
|
}
|
|
log.Println(msg2)
|
|
}
|
|
|
|
func testVarint() {
|
|
n1 := uint64(12345)
|
|
rw := bytes.NewBuffer([]byte{})
|
|
log.Printf("Write: %d\n", n1)
|
|
err := osp.WriteVaruint(n1, rw)
|
|
if err != nil {
|
|
log.Fatalln("Failed to write varint", err.Error())
|
|
}
|
|
n2, err := osp.ReadVaruint(rw)
|
|
if err != nil {
|
|
log.Fatalln("Failed to read varint", err.Error())
|
|
}
|
|
log.Printf("Read: %d\n", n2)
|
|
}
|
|
|
|
func main() {
|
|
testVarint()
|
|
testMessages()
|
|
testQuic()
|
|
testMdns()
|
|
}
|