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