1 package main
2 
3 import (
4 	"flag"
5 	"log"
6 	"net"
7 	"strings"
8 
9 	"github.com/docker/go-p9p"
10 	"golang.org/x/net/context"
11 )
12 
13 var (
14 	root string
15 	addr string
16 )
17 
18 func init() {
19 	flag.StringVar(&root, "root", "~/", "root of filesystem to serve over 9p")
20 	flag.StringVar(&addr, "addr", ":5640", "bind addr for 9p server, prefix with unix: for unix socket")
21 }
22 
23 func main() {
24 	ctx := context.Background()
25 	log.SetFlags(0)
26 	flag.Parse()
27 
28 	proto := "tcp"
29 	if strings.HasPrefix(addr, "unix:") {
30 		proto = "unix"
31 		addr = addr[5:]
32 	}
33 
34 	listener, err := net.Listen(proto, addr)
35 	if err != nil {
36 		log.Fatalln("error listening:", err)
37 	}
38 	defer listener.Close()
39 
40 	for {
41 		c, err := listener.Accept()
42 		if err != nil {
43 			log.Fatalln("error accepting:", err)
44 			continue
45 		}
46 
47 		go func(conn net.Conn) {
48 			defer conn.Close()
49 
50 			ctx := context.WithValue(ctx, "conn", conn)
51 			log.Println("connected", conn.RemoteAddr())
52 			session, err := newLocalSession(ctx, root)
53 			if err != nil {
54 				log.Println("error creating session")
55 				return
56 			}
57 
58 			if err := p9p.ServeConn(ctx, conn, p9p.Dispatch(session)); err != nil {
59 				log.Printf("serving conn: %v", err)
60 			}
61 		}(c)
62 	}
63 }
64 
65 // newLocalSession returns a session to serve the local filesystem, restricted
66 // to the provided root.
67 func newLocalSession(ctx context.Context, root string) (p9p.Session, error) {
68 	// silly, just connect to ufs for now! replace this with real code later!
69 	log.Println("dialing", ":5640", "for", ctx.Value("conn"))
70 	conn, err := net.Dial("tcp", ":5640")
71 	if err != nil {
72 		return nil, err
73 	}
74 
75 	session, err := p9p.NewSession(ctx, conn)
76 	if err != nil {
77 		return nil, err
78 	}
79 
80 	return session, nil
81 }