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