1 | /*
|
2 | Copyright 2012 Timothy J Fontaine <tjfontaine@gmail.com>
|
3 |
|
4 | Permission is hereby granted, free of charge, to any person obtaining a copy
|
5 | of this software and associated documentation files (the "Software"), to deal
|
6 | in the Software without restriction, including without limitation the rights
|
7 | to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
8 | copies of the Software, and to permit persons to whom the Software is
|
9 | furnished to do so, subject to the following conditions:
|
10 |
|
11 | The above copyright notice and this permission notice shall be included in
|
12 | all copies or substantial portions of the Software.
|
13 |
|
14 | THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
15 | IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
16 | FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
17 | AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
18 | LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
19 | OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
20 |
|
21 | */
|
22 |
|
23 | var Socket = require('./socket');
|
24 |
|
25 | var TCPMessage = module.exports = function (socket, handleMessage) {
|
26 | var self = this, rest;
|
27 |
|
28 | this.socket = new Socket(null, socket);
|
29 |
|
30 | socket.on('data', function (data) {
|
31 | var len, tmp;
|
32 | if (!rest) {
|
33 | rest = data;
|
34 | } else {
|
35 | tmp = new Buffer(rest.length + data.length);
|
36 | rest.copy(tmp, 0);
|
37 | data.copy(tmp, rest.length);
|
38 | rest = tmp;
|
39 | }
|
40 | while (rest && rest.length > 2) {
|
41 | len = rest.readUInt16BE(0);
|
42 | if (rest.length >= len + 2) {
|
43 | handleMessage(rest.slice(2, len + 2), self.socket);
|
44 | rest = rest.slice(len + 2);
|
45 | } else {
|
46 | break;
|
47 | }
|
48 | }
|
49 | });
|
50 | };
|