1 e342de7d 2015-11-11 adrien package p9p
4 e342de7d 2015-11-11 adrien "golang.org/x/net/context"
8 e342de7d 2015-11-11 adrien type client struct {
9 e342de7d 2015-11-11 adrien version string
11 e342de7d 2015-11-11 adrien ctx context.Context
12 e342de7d 2015-11-11 adrien transport roundTripper
15 e342de7d 2015-11-11 adrien // NewSession returns a session using the connection. The Context ctx provides
16 e342de7d 2015-11-11 adrien // a context for out of bad messages, such as flushes, that may be sent by the
17 e342de7d 2015-11-11 adrien // session. The session can effectively shutdown with this context.
18 e342de7d 2015-11-11 adrien func NewSession(ctx context.Context, conn net.Conn) (Session, error) {
19 e342de7d 2015-11-11 adrien ch := newChannel(conn, codec9p{}, DefaultMSize) // sets msize, effectively.
21 e342de7d 2015-11-11 adrien // negotiate the protocol version
22 e342de7d 2015-11-11 adrien version, err := clientnegotiate(ctx, ch, DefaultVersion)
23 e342de7d 2015-11-11 adrien if err != nil {
24 e342de7d 2015-11-11 adrien return nil, err
27 e342de7d 2015-11-11 adrien return &client{
28 e342de7d 2015-11-11 adrien version: version,
29 e342de7d 2015-11-11 adrien msize: ch.MSize(),
31 e342de7d 2015-11-11 adrien transport: newTransport(ctx, ch),
35 e342de7d 2015-11-11 adrien var _ Session = &client{}
37 e342de7d 2015-11-11 adrien func (c *client) Version() (int, string) {
38 e342de7d 2015-11-11 adrien return c.msize, c.version
41 e342de7d 2015-11-11 adrien func (c *client) Auth(ctx context.Context, afid Fid, uname, aname string) (Qid, error) {
42 e342de7d 2015-11-11 adrien m := MessageTauth{
43 e342de7d 2015-11-11 adrien Afid: afid,
44 e342de7d 2015-11-11 adrien Uname: uname,
45 e342de7d 2015-11-11 adrien Aname: aname,
48 e342de7d 2015-11-11 adrien resp, err := c.transport.send(ctx, m)
49 e342de7d 2015-11-11 adrien if err != nil {
50 e342de7d 2015-11-11 adrien return Qid{}, nil
53 e342de7d 2015-11-11 adrien rauth, ok := resp.(MessageRauth)
55 e342de7d 2015-11-11 adrien return Qid{}, ErrUnexpectedMsg
58 e342de7d 2015-11-11 adrien return rauth.Qid, nil
61 e342de7d 2015-11-11 adrien func (c *client) Attach(ctx context.Context, fid, afid Fid, uname, aname string) (Qid, error) {
62 e342de7d 2015-11-11 adrien m := MessageTattach{
64 e342de7d 2015-11-11 adrien Afid: afid,
65 e342de7d 2015-11-11 adrien Uname: uname,
66 e342de7d 2015-11-11 adrien Aname: aname,
69 e342de7d 2015-11-11 adrien resp, err := c.transport.send(ctx, m)
70 e342de7d 2015-11-11 adrien if err != nil {
71 e342de7d 2015-11-11 adrien return Qid{}, err
74 e342de7d 2015-11-11 adrien rattach, ok := resp.(MessageRattach)
76 e342de7d 2015-11-11 adrien return Qid{}, ErrUnexpectedMsg
79 e342de7d 2015-11-11 adrien return rattach.Qid, nil
82 e342de7d 2015-11-11 adrien func (c *client) Clunk(ctx context.Context, fid Fid) error {
83 e342de7d 2015-11-11 adrien resp, err := c.transport.send(ctx, MessageTclunk{
86 e342de7d 2015-11-11 adrien if err != nil {
87 e342de7d 2015-11-11 adrien return err
90 e342de7d 2015-11-11 adrien _, ok := resp.(MessageRclunk)
92 e342de7d 2015-11-11 adrien return ErrUnexpectedMsg
95 e342de7d 2015-11-11 adrien return nil
98 e342de7d 2015-11-11 adrien func (c *client) Remove(ctx context.Context, fid Fid) error {
99 e342de7d 2015-11-11 adrien resp, err := c.transport.send(ctx, MessageTremove{
100 e342de7d 2015-11-11 adrien Fid: fid,
102 e342de7d 2015-11-11 adrien if err != nil {
103 e342de7d 2015-11-11 adrien return err
106 e342de7d 2015-11-11 adrien _, ok := resp.(MessageRremove)
108 e342de7d 2015-11-11 adrien return ErrUnexpectedMsg
111 e342de7d 2015-11-11 adrien return nil
114 e342de7d 2015-11-11 adrien func (c *client) Walk(ctx context.Context, fid Fid, newfid Fid, names ...string) ([]Qid, error) {
115 e342de7d 2015-11-11 adrien if len(names) > 16 {
116 e342de7d 2015-11-11 adrien return nil, ErrWalkLimit
119 e342de7d 2015-11-11 adrien resp, err := c.transport.send(ctx, MessageTwalk{
120 e342de7d 2015-11-11 adrien Fid: fid,
121 e342de7d 2015-11-11 adrien Newfid: newfid,
122 e342de7d 2015-11-11 adrien Wnames: names,
124 e342de7d 2015-11-11 adrien if err != nil {
125 e342de7d 2015-11-11 adrien return nil, err
128 e342de7d 2015-11-11 adrien rwalk, ok := resp.(MessageRwalk)
130 e342de7d 2015-11-11 adrien return nil, ErrUnexpectedMsg
133 e342de7d 2015-11-11 adrien return rwalk.Qids, nil
136 e342de7d 2015-11-11 adrien func (c *client) Read(ctx context.Context, fid Fid, p []byte, offset int64) (n int, err error) {
137 e342de7d 2015-11-11 adrien resp, err := c.transport.send(ctx, MessageTread{
138 e342de7d 2015-11-11 adrien Fid: fid,
139 e342de7d 2015-11-11 adrien Offset: uint64(offset),
140 e342de7d 2015-11-11 adrien Count: uint32(len(p)),
142 e342de7d 2015-11-11 adrien if err != nil {
143 e342de7d 2015-11-11 adrien return 0, err
146 e342de7d 2015-11-11 adrien rread, ok := resp.(MessageRread)
148 e342de7d 2015-11-11 adrien return 0, ErrUnexpectedMsg
151 e342de7d 2015-11-11 adrien return copy(p, rread.Data), nil
154 e342de7d 2015-11-11 adrien func (c *client) Write(ctx context.Context, fid Fid, p []byte, offset int64) (n int, err error) {
155 e342de7d 2015-11-11 adrien resp, err := c.transport.send(ctx, MessageTwrite{
156 e342de7d 2015-11-11 adrien Fid: fid,
157 e342de7d 2015-11-11 adrien Offset: uint64(offset),
160 e342de7d 2015-11-11 adrien if err != nil {
161 e342de7d 2015-11-11 adrien return 0, err
164 e342de7d 2015-11-11 adrien rwrite, ok := resp.(MessageRwrite)
166 e342de7d 2015-11-11 adrien return 0, ErrUnexpectedMsg
169 e342de7d 2015-11-11 adrien return int(rwrite.Count), nil
172 e342de7d 2015-11-11 adrien func (c *client) Open(ctx context.Context, fid Fid, mode Flag) (Qid, uint32, error) {
173 e342de7d 2015-11-11 adrien resp, err := c.transport.send(ctx, MessageTopen{
174 e342de7d 2015-11-11 adrien Fid: fid,
175 e342de7d 2015-11-11 adrien Mode: mode,
177 e342de7d 2015-11-11 adrien if err != nil {
178 e342de7d 2015-11-11 adrien return Qid{}, 0, err
181 e342de7d 2015-11-11 adrien ropen, ok := resp.(MessageRopen)
183 e342de7d 2015-11-11 adrien return Qid{}, 0, ErrUnexpectedMsg
186 e342de7d 2015-11-11 adrien return ropen.Qid, ropen.IOUnit, nil
189 e342de7d 2015-11-11 adrien func (c *client) Create(ctx context.Context, parent Fid, name string, perm uint32, mode Flag) (Qid, uint32, error) {
190 e342de7d 2015-11-11 adrien resp, err := c.transport.send(ctx, MessageTcreate{
191 e342de7d 2015-11-11 adrien Fid: parent,
192 e342de7d 2015-11-11 adrien Name: name,
193 e342de7d 2015-11-11 adrien Perm: perm,
194 e342de7d 2015-11-11 adrien Mode: mode,
196 e342de7d 2015-11-11 adrien if err != nil {
197 e342de7d 2015-11-11 adrien return Qid{}, 0, err
200 e342de7d 2015-11-11 adrien rcreate, ok := resp.(MessageRcreate)
202 e342de7d 2015-11-11 adrien return Qid{}, 0, ErrUnexpectedMsg
205 e342de7d 2015-11-11 adrien return rcreate.Qid, rcreate.IOUnit, nil
208 e342de7d 2015-11-11 adrien func (c *client) Stat(ctx context.Context, fid Fid) (Dir, error) {
209 e342de7d 2015-11-11 adrien resp, err := c.transport.send(ctx, MessageTstat{Fid: fid})
210 e342de7d 2015-11-11 adrien if err != nil {
211 e342de7d 2015-11-11 adrien return Dir{}, err
214 e342de7d 2015-11-11 adrien rstat, ok := resp.(MessageRstat)
216 e342de7d 2015-11-11 adrien return Dir{}, ErrUnexpectedMsg
219 e342de7d 2015-11-11 adrien return rstat.Stat, nil
222 e342de7d 2015-11-11 adrien func (c *client) WStat(ctx context.Context, fid Fid, dir Dir) error {
223 e342de7d 2015-11-11 adrien resp, err := c.transport.send(ctx, MessageTwstat{
224 e342de7d 2015-11-11 adrien Fid: fid,
225 e342de7d 2015-11-11 adrien Stat: dir,
227 e342de7d 2015-11-11 adrien if err != nil {
228 e342de7d 2015-11-11 adrien return err
231 e342de7d 2015-11-11 adrien _, ok := resp.(MessageRwstat)
233 e342de7d 2015-11-11 adrien return ErrUnexpectedMsg
236 e342de7d 2015-11-11 adrien return nil