mirror of
				https://source.quilibrium.com/quilibrium/ceremonyclient.git
				synced 2025-11-04 13:27:27 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			69 lines
		
	
	
		
			1.6 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			69 lines
		
	
	
		
			1.6 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
package tlsdiag
 | 
						|
 | 
						|
import (
 | 
						|
	"context"
 | 
						|
	"flag"
 | 
						|
	"fmt"
 | 
						|
	"net"
 | 
						|
	"time"
 | 
						|
 | 
						|
	libp2ptls "github.com/libp2p/go-libp2p/p2p/security/tls"
 | 
						|
 | 
						|
	"github.com/libp2p/go-libp2p/core/peer"
 | 
						|
)
 | 
						|
 | 
						|
func StartServer() error {
 | 
						|
	port := flag.Int("p", 5533, "port")
 | 
						|
	keyType := flag.String("key", "ecdsa", "rsa, ecdsa, ed25519 or secp256k1")
 | 
						|
	flag.Parse()
 | 
						|
 | 
						|
	priv, err := generateKey(*keyType)
 | 
						|
	if err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
 | 
						|
	id, err := peer.IDFromPrivateKey(priv)
 | 
						|
	if err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	fmt.Printf(" Peer ID: %s\n", id.Pretty())
 | 
						|
	tp, err := libp2ptls.New(libp2ptls.ID, priv, nil)
 | 
						|
	if err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
 | 
						|
	ln, err := net.Listen("tcp", fmt.Sprintf("localhost:%d", *port))
 | 
						|
	if err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	fmt.Printf("Listening for new connections on %s\n", ln.Addr())
 | 
						|
	fmt.Printf("Now run the following command in a separate terminal:\n")
 | 
						|
	fmt.Printf("\tgo run cmd/tlsdiag.go client -p %d -id %s\n", *port, id.Pretty())
 | 
						|
 | 
						|
	for {
 | 
						|
		conn, err := ln.Accept()
 | 
						|
		if err != nil {
 | 
						|
			return err
 | 
						|
		}
 | 
						|
		fmt.Printf("Accepted raw connection from %s\n", conn.RemoteAddr())
 | 
						|
		go func() {
 | 
						|
			if err := handleConn(tp, conn); err != nil {
 | 
						|
				fmt.Printf("Error handling connection from %s: %s\n", conn.RemoteAddr(), err)
 | 
						|
			}
 | 
						|
		}()
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func handleConn(tp *libp2ptls.Transport, conn net.Conn) error {
 | 
						|
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
 | 
						|
	defer cancel()
 | 
						|
	sconn, err := tp.SecureInbound(ctx, conn, "")
 | 
						|
	if err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	fmt.Printf("Authenticated client: %s\n", sconn.RemotePeer().Pretty())
 | 
						|
	fmt.Fprintf(sconn, "Hello client!")
 | 
						|
	fmt.Printf("Closing connection to %s\n", conn.RemoteAddr())
 | 
						|
	return sconn.Close()
 | 
						|
}
 |