Class: Protocol::WebSocket::Frame
- Inherits:
-
Object
- Object
- Protocol::WebSocket::Frame
- Includes:
- Comparable
- Defined in:
- lib/protocol/websocket/frame.rb
Direct Known Subclasses
BinaryFrame, CloseFrame, ContinuationFrame, PingFrame, PongFrame, TextFrame
Constant Summary collapse
Instance Attribute Summary collapse
-
#finished ⇒ Object
The generic frame header uses the following binary representation:.
-
#flags ⇒ Object
Returns the value of attribute flags.
-
#length ⇒ Object
Returns the value of attribute length.
-
#mask ⇒ Object
Returns the value of attribute mask.
-
#opcode ⇒ Object
Returns the value of attribute opcode.
-
#payload ⇒ Object
Returns the value of attribute payload.
Class Method Summary collapse
Instance Method Summary collapse
- #<=>(other) ⇒ Object
- #apply(connection) ⇒ Object
- #continued? ⇒ Boolean
- #control? ⇒ Boolean
- #data? ⇒ Boolean
- #finished? ⇒ Boolean
-
#initialize(finished = true, payload = nil, flags: 0, opcode: self.class::OPCODE, mask: false) ⇒ Frame
constructor
A new instance of Frame.
- #pack(data = "") ⇒ Object
- #to_ary ⇒ Object
- #unpack ⇒ Object
- #write(stream) ⇒ Object
Constructor Details
#initialize(finished = true, payload = nil, flags: 0, opcode: self.class::OPCODE, mask: false) ⇒ Frame
Returns a new instance of Frame.
24 25 26 27 28 29 30 31 32 33 34 35 |
# File 'lib/protocol/websocket/frame.rb', line 24 def initialize(finished = true, payload = nil, flags: 0, opcode: self.class::OPCODE, mask: false) if mask == true mask = SecureRandom.bytes(4) end @finished = finished @flags = flags @opcode = opcode @mask = mask @length = payload&.bytesize @payload = payload end |
Instance Attribute Details
#finished ⇒ Object
The generic frame header uses the following binary representation:
0 1 2 3
0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
-
--
--------
--------------
——————————-+ |F|R|R|R| opcode|M| Payload len | Extended payload length | |I|S|S|S| (4) |A| (7) | (16/64) | |N|V|V|V| |S| | (if payload len==126/127) | | |1|2|3| |K| | | -
--
--------
--------------
- - - - - - - - - - - - - - - + | Extended payload length continued, if payload len == 127 | + - - - - - - - - - - - - - - - -------------------------------
| |Masking-key, if MASK set to 1 | -------------------------------
——————————-+ | Masking-key (continued) | Payload Data | ——————————– - - - - - - - - - - - - - - - : Payload Data continued … : + - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - + | Payload Data continued … | ---------------------------------------------------------------
82 83 84 |
# File 'lib/protocol/websocket/frame.rb', line 82 def finished @finished end |
#flags ⇒ Object
Returns the value of attribute flags.
83 84 85 |
# File 'lib/protocol/websocket/frame.rb', line 83 def flags @flags end |
#length ⇒ Object
Returns the value of attribute length.
86 87 88 |
# File 'lib/protocol/websocket/frame.rb', line 86 def length @length end |
#mask ⇒ Object
Returns the value of attribute mask.
85 86 87 |
# File 'lib/protocol/websocket/frame.rb', line 85 def mask @mask end |
#opcode ⇒ Object
Returns the value of attribute opcode.
84 85 86 |
# File 'lib/protocol/websocket/frame.rb', line 84 def opcode @opcode end |
#payload ⇒ Object
Returns the value of attribute payload.
87 88 89 |
# File 'lib/protocol/websocket/frame.rb', line 87 def payload @payload end |
Class Method Details
.parse_header(buffer) ⇒ Object
143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 |
# File 'lib/protocol/websocket/frame.rb', line 143 def self.parse_header(buffer) byte = buffer.unpack("C").first finished = (byte & 0b1000_0000 != 0) flags = (byte & 0b0111_0000) >> 4 opcode = byte & 0b0000_1111 if (0x3 .. 0x7).include?(opcode) raise ProtocolError, "Non-control opcode = #{opcode} is reserved!" elsif (0xB .. 0xF).include?(opcode) raise ProtocolError, "Control opcode = #{opcode} is reserved!" end return finished, flags, opcode end |
.read(finished, flags, opcode, stream, maximum_frame_size) ⇒ Object
159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 |
# File 'lib/protocol/websocket/frame.rb', line 159 def self.read(finished, flags, opcode, stream, maximum_frame_size) buffer = stream.read(1) or raise EOFError, "Could not read header!" byte = buffer.unpack("C").first mask = (byte & 0b1000_0000 != 0) length = byte & 0b0111_1111 if opcode & 0x8 != 0 if length > 125 raise ProtocolError, "Invalid control frame payload length: #{length} > 125!" elsif !finished raise ProtocolError, "Fragmented control frame!" end end if length == 126 buffer = stream.read(2) or raise EOFError, "Could not read length!" length = buffer.unpack('n').first elsif length == 127 buffer = stream.read(8) or raise EOFError, "Could not read length!" length = buffer.unpack('Q>').first end if length > maximum_frame_size raise ProtocolError, "Invalid payload length: #{@length} > #{maximum_frame_size}!" end if mask mask = stream.read(4) or raise EOFError, "Could not read mask!" end payload = stream.read(length) or raise EOFError, "Could not read payload!" if payload.bytesize != length raise EOFError, "Incorrect payload length: #{@length} != #{payload.bytesize}!" end return self.new(finished, payload, flags: flags, opcode: opcode, mask: mask) end |
Instance Method Details
#<=>(other) ⇒ Object
37 38 39 |
# File 'lib/protocol/websocket/frame.rb', line 37 def <=> other to_ary <=> other.to_ary end |
#apply(connection) ⇒ Object
139 140 141 |
# File 'lib/protocol/websocket/frame.rb', line 139 def apply(connection) connection.receive_frame(self) end |
#continued? ⇒ Boolean
57 58 59 |
# File 'lib/protocol/websocket/frame.rb', line 57 def continued? @finished == false end |
#control? ⇒ Boolean
45 46 47 |
# File 'lib/protocol/websocket/frame.rb', line 45 def control? @opcode & 0x8 != 0 end |
#data? ⇒ Boolean
49 50 51 |
# File 'lib/protocol/websocket/frame.rb', line 49 def data? false end |
#finished? ⇒ Boolean
53 54 55 |
# File 'lib/protocol/websocket/frame.rb', line 53 def finished? @finished == true end |
#pack(data = "") ⇒ Object
113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 |
# File 'lib/protocol/websocket/frame.rb', line 113 def pack(data = "") length = data.bytesize if length.bit_length > 63 raise ProtocolError, "Frame length #{@length} bigger than allowed maximum!" end if @mask @payload = mask_xor(data, mask) @length = length else @payload = data @length = length end return self end |
#to_ary ⇒ Object
41 42 43 |
# File 'lib/protocol/websocket/frame.rb', line 41 def to_ary [@finished, @flags, @opcode, @mask, @length, @payload] end |
#unpack ⇒ Object
131 132 133 134 135 136 137 |
# File 'lib/protocol/websocket/frame.rb', line 131 def unpack if @mask and !@payload.empty? return mask_xor(@payload, @mask) else return @payload end end |
#write(stream) ⇒ Object
199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 |
# File 'lib/protocol/websocket/frame.rb', line 199 def write(stream) buffer = String.new(encoding: Encoding::BINARY) if @payload&.bytesize != @length raise ProtocolError, "Invalid payload length: #{@length} != #{@payload.bytesize} for #{self}!" end if @mask and @mask.bytesize != 4 raise ProtocolError, "Invalid mask length!" end if length <= 125 short_length = length elsif length.bit_length <= 16 short_length = 126 else short_length = 127 end buffer << [ (@finished ? 0b1000_0000 : 0) | (@flags << 4) | @opcode, (@mask ? 0b1000_0000 : 0) | short_length, ].pack('CC') if short_length == 126 buffer << [@length].pack('n') elsif short_length == 127 buffer << [@length].pack('Q>') end buffer << @mask if @mask stream.write(buffer) stream.write(@payload) end |