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