1 package p9p
2 
3 import (
4 	"net"
5 
6 	"context"
7 )
8 
9 type client struct {
10 	version   string
11 	msize     int
12 	ctx       context.Context
13 	transport roundTripper
14 }
15 
16 // NewSession returns a session using the connection. The Context ctx provides
17 // a context for out of bad messages, such as flushes, that may be sent by the
18 // session. The session can effectively shutdown with this context.
19 func NewSession(ctx context.Context, conn net.Conn) (Session, error) {
20 	ch := newChannel(conn, codec9p{}, DefaultMSize) // sets msize, effectively.
21 
22 	// negotiate the protocol version
23 	version, err := clientnegotiate(ctx, ch, DefaultVersion)
24 	if err != nil {
25 		return nil, err
26 	}
27 
28 	return &client{
29 		version:   version,
30 		msize:     ch.MSize(),
31 		ctx:       ctx,
32 		transport: newTransport(ctx, ch),
33 	}, nil
34 }
35 
36 var _ Session = &client{}
37 
38 func (c *client) Version() (int, string) {
39 	return c.msize, c.version
40 }
41 
42 func (c *client) Auth(ctx context.Context, afid Fid, uname, aname string) (Qid, error) {
43 	m := MessageTauth{
44 		Afid:  afid,
45 		Uname: uname,
46 		Aname: aname,
47 	}
48 
49 	resp, err := c.transport.send(ctx, m)
50 	if err != nil {
51 		return Qid{}, err
52 	}
53 
54 	rauth, ok := resp.(MessageRauth)
55 	if !ok {
56 		return Qid{}, ErrUnexpectedMsg
57 	}
58 
59 	return rauth.Qid, nil
60 }
61 
62 func (c *client) Attach(ctx context.Context, fid, afid Fid, uname, aname string) (Qid, error) {
63 	m := MessageTattach{
64 		Fid:   fid,
65 		Afid:  afid,
66 		Uname: uname,
67 		Aname: aname,
68 	}
69 
70 	resp, err := c.transport.send(ctx, m)
71 	if err != nil {
72 		return Qid{}, err
73 	}
74 
75 	rattach, ok := resp.(MessageRattach)
76 	if !ok {
77 		return Qid{}, ErrUnexpectedMsg
78 	}
79 
80 	return rattach.Qid, nil
81 }
82 
83 func (c *client) Clunk(ctx context.Context, fid Fid) error {
84 	resp, err := c.transport.send(ctx, MessageTclunk{
85 		Fid: fid,
86 	})
87 	if err != nil {
88 		return err
89 	}
90 
91 	_, ok := resp.(MessageRclunk)
92 	if !ok {
93 		return ErrUnexpectedMsg
94 	}
95 
96 	return nil
97 }
98 
99 func (c *client) Remove(ctx context.Context, fid Fid) error {
100 	resp, err := c.transport.send(ctx, MessageTremove{
101 		Fid: fid,
102 	})
103 	if err != nil {
104 		return err
105 	}
106 
107 	_, ok := resp.(MessageRremove)
108 	if !ok {
109 		return ErrUnexpectedMsg
110 	}
111 
112 	return nil
113 }
114 
115 func (c *client) Walk(ctx context.Context, fid Fid, newfid Fid, names ...string) ([]Qid, error) {
116 	if len(names) > 16 {
117 		return nil, ErrWalkLimit
118 	}
119 
120 	resp, err := c.transport.send(ctx, MessageTwalk{
121 		Fid:    fid,
122 		Newfid: newfid,
123 		Wnames: names,
124 	})
125 	if err != nil {
126 		return nil, err
127 	}
128 
129 	rwalk, ok := resp.(MessageRwalk)
130 	if !ok {
131 		return nil, ErrUnexpectedMsg
132 	}
133 
134 	return rwalk.Qids, nil
135 }
136 
137 func (c *client) Read(ctx context.Context, fid Fid, p []byte, offset int64) (n int, err error) {
138 	resp, err := c.transport.send(ctx, MessageTread{
139 		Fid:    fid,
140 		Offset: uint64(offset),
141 		Count:  uint32(len(p)),
142 	})
143 	if err != nil {
144 		return 0, err
145 	}
146 
147 	rread, ok := resp.(MessageRread)
148 	if !ok {
149 		return 0, ErrUnexpectedMsg
150 	}
151 
152 	return copy(p, rread.Data), nil
153 }
154 
155 func (c *client) Write(ctx context.Context, fid Fid, p []byte, offset int64) (n int, err error) {
156 	resp, err := c.transport.send(ctx, MessageTwrite{
157 		Fid:    fid,
158 		Offset: uint64(offset),
159 		Data:   p,
160 	})
161 	if err != nil {
162 		return 0, err
163 	}
164 
165 	rwrite, ok := resp.(MessageRwrite)
166 	if !ok {
167 		return 0, ErrUnexpectedMsg
168 	}
169 
170 	return int(rwrite.Count), nil
171 }
172 
173 func (c *client) Open(ctx context.Context, fid Fid, mode Flag) (Qid, uint32, error) {
174 	resp, err := c.transport.send(ctx, MessageTopen{
175 		Fid:  fid,
176 		Mode: mode,
177 	})
178 	if err != nil {
179 		return Qid{}, 0, err
180 	}
181 
182 	ropen, ok := resp.(MessageRopen)
183 	if !ok {
184 		return Qid{}, 0, ErrUnexpectedMsg
185 	}
186 
187 	return ropen.Qid, ropen.IOUnit, nil
188 }
189 
190 func (c *client) Create(ctx context.Context, parent Fid, name string, perm uint32, mode Flag) (Qid, uint32, error) {
191 	resp, err := c.transport.send(ctx, MessageTcreate{
192 		Fid:  parent,
193 		Name: name,
194 		Perm: perm,
195 		Mode: mode,
196 	})
197 	if err != nil {
198 		return Qid{}, 0, err
199 	}
200 
201 	rcreate, ok := resp.(MessageRcreate)
202 	if !ok {
203 		return Qid{}, 0, ErrUnexpectedMsg
204 	}
205 
206 	return rcreate.Qid, rcreate.IOUnit, nil
207 }
208 
209 func (c *client) Stat(ctx context.Context, fid Fid) (Dir, error) {
210 	resp, err := c.transport.send(ctx, MessageTstat{Fid: fid})
211 	if err != nil {
212 		return Dir{}, err
213 	}
214 
215 	rstat, ok := resp.(MessageRstat)
216 	if !ok {
217 		return Dir{}, ErrUnexpectedMsg
218 	}
219 
220 	return rstat.Stat, nil
221 }
222 
223 func (c *client) WStat(ctx context.Context, fid Fid, dir Dir) error {
224 	resp, err := c.transport.send(ctx, MessageTwstat{
225 		Fid:  fid,
226 		Stat: dir,
227 	})
228 	if err != nil {
229 		return err
230 	}
231 
232 	_, ok := resp.(MessageRwstat)
233 	if !ok {
234 		return ErrUnexpectedMsg
235 	}
236 
237 	return nil
238 }