1 8a7ec697 2015-10-26 adrien package p9pnew
4 8a7ec697 2015-10-26 adrien "golang.org/x/net/context"
9 8a7ec697 2015-10-26 adrien type client struct {
10 fb37ce2a 2015-10-30 stephen.d version string
11 dce371f6 2015-11-05 stephen.d msize int
12 e6bcde66 2015-10-29 stephen.d ctx context.Context
13 e6bcde66 2015-10-29 stephen.d transport roundTripper
16 5f1e8105 2015-10-28 stephen.d // NewSession returns a session using the connection. The Context ctx provides
17 5f1e8105 2015-10-28 stephen.d // a context for out of bad messages, such as flushes, that may be sent by the
18 5f1e8105 2015-10-28 stephen.d // session. The session can effectively shutdown with this context.
19 5f1e8105 2015-10-28 stephen.d func NewSession(ctx context.Context, conn net.Conn) (Session, error) {
20 dce371f6 2015-11-05 stephen.d ch := newChannel(conn, codec9p{}, DefaultMSize) // sets msize, effectively.
22 fb37ce2a 2015-10-30 stephen.d // negotiate the protocol version
23 dce371f6 2015-11-05 stephen.d version, err := clientnegotiate(ctx, ch, DefaultVersion)
24 fb37ce2a 2015-10-30 stephen.d if err != nil {
25 fb37ce2a 2015-10-30 stephen.d return nil, err
28 8a7ec697 2015-10-26 adrien return &client{
29 dce371f6 2015-11-05 stephen.d version: version,
30 dce371f6 2015-11-05 stephen.d msize: ch.MSize(),
31 e6bcde66 2015-10-29 stephen.d ctx: ctx,
32 fb37ce2a 2015-10-30 stephen.d transport: newTransport(ctx, ch),
36 8a7ec697 2015-10-26 adrien var _ Session = &client{}
38 dce371f6 2015-11-05 stephen.d func (c *client) Version() (int, string) {
39 fb37ce2a 2015-10-30 stephen.d return c.msize, c.version
42 8a7ec697 2015-10-26 adrien func (c *client) Auth(ctx context.Context, afid Fid, uname, aname string) (Qid, error) {
43 deb98ab4 2015-11-05 stephen.d m := MessageTauth{
44 deb98ab4 2015-11-05 stephen.d Afid: afid,
45 deb98ab4 2015-11-05 stephen.d Uname: uname,
46 deb98ab4 2015-11-05 stephen.d Aname: aname,
49 deb98ab4 2015-11-05 stephen.d resp, err := c.transport.send(ctx, m)
50 deb98ab4 2015-11-05 stephen.d if err != nil {
51 deb98ab4 2015-11-05 stephen.d return Qid{}, nil
54 deb98ab4 2015-11-05 stephen.d rauth, ok := resp.(MessageRauth)
55 deb98ab4 2015-11-05 stephen.d if !ok {
56 deb98ab4 2015-11-05 stephen.d return Qid{}, ErrUnexpectedMsg
59 deb98ab4 2015-11-05 stephen.d return rauth.Qid, nil
62 8a7ec697 2015-10-26 adrien func (c *client) Attach(ctx context.Context, fid, afid Fid, uname, aname string) (Qid, error) {
63 deb98ab4 2015-11-05 stephen.d m := MessageTattach{
64 deb98ab4 2015-11-05 stephen.d Fid: fid,
65 deb98ab4 2015-11-05 stephen.d Afid: afid,
66 deb98ab4 2015-11-05 stephen.d Uname: uname,
67 deb98ab4 2015-11-05 stephen.d Aname: aname,
70 deb98ab4 2015-11-05 stephen.d resp, err := c.transport.send(ctx, m)
71 5f1e8105 2015-10-28 stephen.d if err != nil {
72 5f1e8105 2015-10-28 stephen.d return Qid{}, err
75 deb98ab4 2015-11-05 stephen.d rattach, ok := resp.(MessageRattach)
76 5f1e8105 2015-10-28 stephen.d if !ok {
77 deb98ab4 2015-11-05 stephen.d return Qid{}, ErrUnexpectedMsg
80 deb98ab4 2015-11-05 stephen.d return rattach.Qid, nil
83 8a7ec697 2015-10-26 adrien func (c *client) Clunk(ctx context.Context, fid Fid) error {
84 deb98ab4 2015-11-05 stephen.d resp, err := c.transport.send(ctx, MessageTclunk{
85 e6bcde66 2015-10-29 stephen.d Fid: fid,
87 e6bcde66 2015-10-29 stephen.d if err != nil {
88 e6bcde66 2015-10-29 stephen.d return err
91 deb98ab4 2015-11-05 stephen.d _, ok := resp.(MessageRclunk)
92 deb98ab4 2015-11-05 stephen.d if !ok {
93 deb98ab4 2015-11-05 stephen.d return ErrUnexpectedMsg
96 e6bcde66 2015-10-29 stephen.d return nil
99 8a7ec697 2015-10-26 adrien func (c *client) Remove(ctx context.Context, fid Fid) error {
100 deb98ab4 2015-11-05 stephen.d resp, err := c.transport.send(ctx, MessageTremove{
101 deb98ab4 2015-11-05 stephen.d Fid: fid,
103 deb98ab4 2015-11-05 stephen.d if err != nil {
104 deb98ab4 2015-11-05 stephen.d return err
107 deb98ab4 2015-11-05 stephen.d _, ok := resp.(MessageRremove)
108 deb98ab4 2015-11-05 stephen.d if !ok {
109 deb98ab4 2015-11-05 stephen.d return ErrUnexpectedMsg
112 deb98ab4 2015-11-05 stephen.d return nil
115 8a7ec697 2015-10-26 adrien func (c *client) Walk(ctx context.Context, fid Fid, newfid Fid, names ...string) ([]Qid, error) {
116 5f1e8105 2015-10-28 stephen.d if len(names) > 16 {
117 deb98ab4 2015-11-05 stephen.d return nil, ErrWalkLimit
120 deb98ab4 2015-11-05 stephen.d resp, err := c.transport.send(ctx, MessageTwalk{
121 deb98ab4 2015-11-05 stephen.d Fid: fid,
122 deb98ab4 2015-11-05 stephen.d Newfid: newfid,
123 deb98ab4 2015-11-05 stephen.d Wnames: names,
125 5f1e8105 2015-10-28 stephen.d if err != nil {
126 5f1e8105 2015-10-28 stephen.d return nil, err
129 deb98ab4 2015-11-05 stephen.d rwalk, ok := resp.(MessageRwalk)
130 5f1e8105 2015-10-28 stephen.d if !ok {
131 deb98ab4 2015-11-05 stephen.d return nil, ErrUnexpectedMsg
134 deb98ab4 2015-11-05 stephen.d return rwalk.Qids, nil
137 8a7ec697 2015-10-26 adrien func (c *client) Read(ctx context.Context, fid Fid, p []byte, offset int64) (n int, err error) {
138 deb98ab4 2015-11-05 stephen.d resp, err := c.transport.send(ctx, MessageTread{
139 deb98ab4 2015-11-05 stephen.d Fid: fid,
140 deb98ab4 2015-11-05 stephen.d Offset: uint64(offset),
141 deb98ab4 2015-11-05 stephen.d Count: uint32(len(p)),
143 5f1e8105 2015-10-28 stephen.d if err != nil {
144 5f1e8105 2015-10-28 stephen.d return 0, err
147 deb98ab4 2015-11-05 stephen.d rread, ok := resp.(MessageRread)
148 5f1e8105 2015-10-28 stephen.d if !ok {
149 deb98ab4 2015-11-05 stephen.d return 0, ErrUnexpectedMsg
152 deb98ab4 2015-11-05 stephen.d return copy(p, rread.Data), nil
155 8a7ec697 2015-10-26 adrien func (c *client) Write(ctx context.Context, fid Fid, p []byte, offset int64) (n int, err error) {
156 deb98ab4 2015-11-05 stephen.d resp, err := c.transport.send(ctx, MessageTwrite{
157 deb98ab4 2015-11-05 stephen.d Fid: fid,
158 deb98ab4 2015-11-05 stephen.d Offset: uint64(offset),
159 deb98ab4 2015-11-05 stephen.d Data: p,
161 5f1e8105 2015-10-28 stephen.d if err != nil {
162 5f1e8105 2015-10-28 stephen.d return 0, err
165 deb98ab4 2015-11-05 stephen.d rwrite, ok := resp.(MessageRwrite)
166 5f1e8105 2015-10-28 stephen.d if !ok {
167 deb98ab4 2015-11-05 stephen.d return 0, ErrUnexpectedMsg
170 deb98ab4 2015-11-05 stephen.d return int(rwrite.Count), nil
173 deb98ab4 2015-11-05 stephen.d func (c *client) Open(ctx context.Context, fid Fid, mode Flag) (Qid, uint32, error) {
174 deb98ab4 2015-11-05 stephen.d resp, err := c.transport.send(ctx, MessageTopen{
175 e6bcde66 2015-10-29 stephen.d Fid: fid,
176 e6bcde66 2015-10-29 stephen.d Mode: mode,
178 e6bcde66 2015-10-29 stephen.d if err != nil {
179 e6bcde66 2015-10-29 stephen.d return Qid{}, 0, err
182 deb98ab4 2015-11-05 stephen.d ropen, ok := resp.(MessageRopen)
183 e6bcde66 2015-10-29 stephen.d if !ok {
184 deb98ab4 2015-11-05 stephen.d return Qid{}, 0, ErrUnexpectedMsg
187 deb98ab4 2015-11-05 stephen.d return ropen.Qid, ropen.IOUnit, nil
190 deb98ab4 2015-11-05 stephen.d func (c *client) Create(ctx context.Context, parent Fid, name string, perm uint32, mode Flag) (Qid, uint32, error) {
191 deb98ab4 2015-11-05 stephen.d resp, err := c.transport.send(ctx, MessageTcreate{
192 deb98ab4 2015-11-05 stephen.d Fid: parent,
193 deb98ab4 2015-11-05 stephen.d Name: name,
194 deb98ab4 2015-11-05 stephen.d Perm: perm,
195 deb98ab4 2015-11-05 stephen.d Mode: mode,
197 deb98ab4 2015-11-05 stephen.d if err != nil {
198 deb98ab4 2015-11-05 stephen.d return Qid{}, 0, err
201 deb98ab4 2015-11-05 stephen.d rcreate, ok := resp.(MessageRcreate)
202 deb98ab4 2015-11-05 stephen.d if !ok {
203 deb98ab4 2015-11-05 stephen.d return Qid{}, 0, ErrUnexpectedMsg
206 deb98ab4 2015-11-05 stephen.d return rcreate.Qid, rcreate.IOUnit, nil
209 74ec7ac9 2015-10-30 stephen.d func (c *client) Stat(ctx context.Context, fid Fid) (Dir, error) {
210 deb98ab4 2015-11-05 stephen.d resp, err := c.transport.send(ctx, MessageTstat{Fid: fid})
211 74ec7ac9 2015-10-30 stephen.d if err != nil {
212 74ec7ac9 2015-10-30 stephen.d return Dir{}, err
215 deb98ab4 2015-11-05 stephen.d rstat, ok := resp.(MessageRstat)
216 74ec7ac9 2015-10-30 stephen.d if !ok {
217 deb98ab4 2015-11-05 stephen.d return Dir{}, ErrUnexpectedMsg
220 deb98ab4 2015-11-05 stephen.d return rstat.Stat, nil
223 deb98ab4 2015-11-05 stephen.d func (c *client) WStat(ctx context.Context, fid Fid, dir Dir) error {
224 deb98ab4 2015-11-05 stephen.d resp, err := c.transport.send(ctx, MessageTwstat{
225 deb98ab4 2015-11-05 stephen.d Fid: fid,
226 deb98ab4 2015-11-05 stephen.d Stat: dir,
228 deb98ab4 2015-11-05 stephen.d if err != nil {
229 deb98ab4 2015-11-05 stephen.d return err
232 deb98ab4 2015-11-05 stephen.d _, ok := resp.(MessageRwstat)
233 deb98ab4 2015-11-05 stephen.d if !ok {
234 deb98ab4 2015-11-05 stephen.d return ErrUnexpectedMsg
237 deb98ab4 2015-11-05 stephen.d return nil