1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
|
// Copyright 2009 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
// The websocket package implements Web Socket protocol server.
package websocket
// References:
// The Web Socket protocol: http://tools.ietf.org/html/draft-hixie-thewebsocketprotocol
// TODO(ukai):
// better logging.
import (
"bufio";
"io";
"net";
"os";
)
type WebSocketAddr string
func (addr WebSocketAddr) Network() string { return "websocket" }
func (addr WebSocketAddr) String() string { return string(addr) }
// Conn is an channels to communicate over Web Socket.
type Conn struct {
// An origin URI of the Web Socket.
Origin string;
// A location URI of the Web Socket.
Location string;
// A subprotocol of the Web Socket.
Protocol string;
buf *bufio.ReadWriter;
rwc io.ReadWriteCloser;
}
// newConn creates a new Web Socket.
func newConn(origin, location, protocol string, buf *bufio.ReadWriter, rwc io.ReadWriteCloser) *Conn {
if buf == nil {
br := bufio.NewReader(rwc);
bw := bufio.NewWriter(rwc);
buf = bufio.NewReadWriter(br, bw);
}
ws := &Conn{origin, location, protocol, buf, rwc};
return ws;
}
func (ws *Conn) Read(msg []byte) (n int, err os.Error) {
for {
frameByte, err := ws.buf.ReadByte();
if err != nil {
return n, err
}
if (frameByte & 0x80) == 0x80 {
length := 0;
for {
c, err := ws.buf.ReadByte();
if err != nil {
return n, err
}
if (c & 0x80) == 0x80 {
length = length*128 + int(c&0x7f)
} else {
break
}
}
for length > 0 {
_, err := ws.buf.ReadByte();
if err != nil {
return n, err
}
length--;
}
} else {
for {
c, err := ws.buf.ReadByte();
if err != nil {
return n, err
}
if c == '\xff' {
return n, err
}
if frameByte == 0 {
if n+1 <= cap(msg) {
msg = msg[0 : n+1]
}
msg[n] = c;
n++;
}
if n >= cap(msg) {
return n, os.E2BIG
}
}
}
}
panic("unreachable");
}
func (ws *Conn) Write(msg []byte) (n int, err os.Error) {
ws.buf.WriteByte(0);
ws.buf.Write(msg);
ws.buf.WriteByte(0xff);
err = ws.buf.Flush();
return len(msg), err;
}
func (ws *Conn) Close() os.Error { return ws.rwc.Close() }
func (ws *Conn) LocalAddr() net.Addr { return WebSocketAddr(ws.Origin) }
func (ws *Conn) RemoteAddr() net.Addr { return WebSocketAddr(ws.Location) }
func (ws *Conn) SetTimeout(nsec int64) os.Error {
if conn, ok := ws.rwc.(net.Conn); ok {
return conn.SetTimeout(nsec)
}
return os.EINVAL;
}
func (ws *Conn) SetReadTimeout(nsec int64) os.Error {
if conn, ok := ws.rwc.(net.Conn); ok {
return conn.SetReadTimeout(nsec)
}
return os.EINVAL;
}
func (ws *Conn) SetWriteTimeout(nsec int64) os.Error {
if conn, ok := ws.rwc.(net.Conn); ok {
return conn.SetWriteTimeout(nsec)
}
return os.EINVAL;
}
var _ net.Conn = (*Conn)(nil) // compile-time check that *Conn implements net.Conn.
|
