source: uKadecot/trunk/uip/apps/webserver/websocket.c@ 148

Last change on this file since 148 was 148, checked in by coas-nagasima, 8 years ago

メッセージバッファ領域がなくなったときにWebSocketパケットを誤って処理してしまうのを修正。
Ethernet送信バッファが取れないときにパケットが送信できなかったのを修正。
その他、処理の安定性の向上。
  • Property svn:eol-style set to native
  • Property svn:keywords set to Id
  • Property svn:mime-type set to text/x-chdr; charset=SHIFT_JIS
File size: 7.0 KB
Line 
1/*
2 * TOPPERS ECHONET Lite Communication Middleware
3 *
4 * Copyright (C) 2015 Cores Co., Ltd. Japan
5 *
6 * 上記著作権者は，以下の(1)〜(4)の条件を満たす場合に限り，本ソフトウェ
7 * ア（本ソフトウェアを改変したものを含む．以下同じ）を使用・複製・改
8 * 変・再配布（以下，利用と呼ぶ）することを無償で許諾する．
9 * (1) 本ソフトウェアをソースコードの形で利用する場合には，上記の著作
10 * 権表示，この利用条件および下記の無保証規定が，そのままの形でソー
11 * スコード中に含まれていること．
12 * (2) 本ソフトウェアを，ライブラリ形式など，他のソフトウェア開発に使
13 * 用できる形で再配布する場合には，再配布に伴うドキュメント（利用
14 * 者マニュアルなど）に，上記の著作権表示，この利用条件および下記
15 * の無保証規定を掲載すること．
16 * (3) 本ソフトウェアを，機器に組み込むなど，他のソフトウェア開発に使
17 * 用できない形で再配布する場合には，次のいずれかの条件を満たすこ
18 * と．
19 * (a) 再配布に伴うドキュメント（利用者マニュアルなど）に，上記の著
20 * 作権表示，この利用条件および下記の無保証規定を掲載すること．
21 * (b) 再配布の形態を，別に定める方法によって，TOPPERSプロジェクトに
22 * 報告すること．
23 * (4) 本ソフトウェアの利用により直接的または間接的に生じるいかなる損
24 * 害からも，上記著作権者およびTOPPERSプロジェクトを免責すること．
25 * また，本ソフトウェアのユーザまたはエンドユーザからのいかなる理
26 * 由に基づく請求からも，上記著作権者およびTOPPERSプロジェクトを
27 * 免責すること．
28 *
29 * 本ソフトウェアは，無保証で提供されているものである．上記著作権者お
30 * よびTOPPERSプロジェクトは，本ソフトウェアに関して，特定の使用目的
31 * に対する適合性も含めて，いかなる保証も行わない．また，本ソフトウェ
32 * アの利用により直接的または間接的に生じたいかなる損害に関しても，そ
33 * の責任を負わない．
34 *
35 * @(#) $Id: websocket.c 148 2016-01-08 05:35:52Z coas-nagasima $
36 */
37
38#include <string.h>
39#include "websocket.h"
40#include "uip.h"
41#include "kernel.h"
42#include "kernel_cfg.h"
43
44static ECN_FBS_ID cur_out_msg;
45
46void websocket_init(struct websocket *ws, ID wbsid)
47{
48 memset(ws, 0, sizeof(*ws));
49 ws->wbsid = wbsid;
50}
51
52void websocket_destroy(struct websocket *ws)
53{
54 if ((cur_out_msg.ptr != NULL) && (((ID *)cur_out_msg.ptr->_gap)[0] == ws->wbsid))
55 {
56 _ecn_fbs_del(cur_out_msg);
57 memset(&cur_out_msg, 0, sizeof(cur_out_msg));
58 }
59
60 if (ws->cur_in_msg.ptr != NULL) {
61 _ecn_fbs_del(ws->cur_in_msg);
62 memset(&ws->cur_in_msg, 0, sizeof(ws->cur_in_msg));
63 }
64
65 ws->wbsid = 0;
66}
67
68int websocket_input(struct websocket *ws, const void *data, int len)
69{
70 ws_state_t *s = &ws->rstate;
71 const u8_t *pos = (const u8_t *)data, *end = &((const u8_t *)data)[len];
72
73 for(; pos < end; pos++){
74 switch (s->state)
75 {
76 // FIN RSV1-3 OPECODE
77 case 0:
78 s->fin = (*pos & 0x80) != 0;
79 s->opecode = (enum opecode_t)(*pos & 0x0F);
80 s->state = 1;
81 break;
82 // MASK Payload len
83 case 1:
84 // MASK=0はNG
85 if((*pos & 0x80) == 0)
86 return -1/*MASK_ERROR*/;
87
88 switch(*pos & 0x7F){
89 case 127:
90 s->payload_len = 0;
91 s->state = 2/*7+64bit*/;
92 break;
93 case 126:
94 s->payload_len = 0;
95 s->state = 8/*7+16bit*/;
96 break;
97 default:
98 s->payload_len = *pos & 0x7F;
99 s->state = 10/*7bit*/;
100 break;
101 }
102 break;
103 // 64bit Extended payload length
104 case 2: case 3: case 4: case 5: case 6: case 7:
105 // 16bit Extended payload length
106 case 8: case 9:
107 s->payload_len = (s->payload_len << 8) + *pos;
108 s->state++;
109 break;
110 // Masking-key
111 case 10: case 11: case 12:
112 s->masking_key[s->state - 10] = *pos;
113 s->state++;
114 break;
115 case 13:
116 s->masking_key[3] = *pos;
117
118 s->data_pos = 0;
119 if (s->payload_len <= 0) {
120 s->state = 0;
121 }
122 else {
123 _ecn_fbs_cre(1, &ws->cur_in_msg);
124 s->state++;
125 }
126 break;
127 // Payload buffer
128 default:
129 if(ws->cur_in_msg.ptr != NULL)
130 _ecn_fbs_poke(ws->cur_in_msg, s->data_pos, *pos ^ s->masking_key[s->data_pos % 4]);
131
132 s->data_pos++;
133 if (s->data_pos >= s->payload_len) {
134 ER ret;
135 s->state = 0;
136
137 if (ws->cur_in_msg.ptr != NULL) {
138 ((ID *)ws->cur_in_msg.ptr->_gap)[0] = ws->wbsid;
139 ret = psnd_dtq(ecn_api_mailboxid, (intptr_t)ws->cur_in_msg.ptr);
140 if (ret != E_OK) {
141 syslog(LOG_WARNING, "websocket_input() : psnd_dtq(%d) result = %d", ecn_api_mailboxid, ret);
142 _ecn_fbs_del(ws->cur_in_msg);
143 }
144 }
145 memset(&ws->cur_in_msg, 0, sizeof(ws->cur_in_msg));
146 }
147 break;
148 }
149 }
150
151 return 0;
152}
153
154bool_t websocket_newdata(struct websocket *ws)
155{
156 ws_state_t *s = &ws->wstate;
157 struct websocket *mws;
158 ER ret;
159 ECN_FBS_ID data;
160 ID wbsid;
161
162 if(s->state != 0)
163 return true;
164
165 if (cur_out_msg.ptr != NULL)
166 return (((ID *)cur_out_msg.ptr->_gap)[0] == ws->wbsid);
167
168 for (;;) {
169 ret = prcv_dtq(WEBSOCKET_MBXID, (intptr_t *)&data.ptr);
170 if (ret == E_TMOUT)
171 return false;
172
173 if (ret != E_OK) {
174 TOPPERS_assert_abort();
175 return false;
176 }
177
178 wbsid = ((ID *)data.ptr->_gap)[0];
179 mws = websocket_getws(wbsid);
180 if ((mws != NULL) && (mws->wbsid != 0))
181 break;
182
183 _ecn_fbs_del(cur_out_msg);
184 }
185
186 cur_out_msg.ptr = data.ptr;
187
188 return (wbsid == ws->wbsid);
189}
190
191int websocket_output(struct websocket *ws, void *data, int len)
192{
193 ws_state_t *s = &ws->wstate;
194 u8_t *pos = (u8_t *)data, *end = &((u8_t *)data)[len];
195
196 for(; pos < end; pos++){
197 switch (s->state)
198 {
199 // FIN RSV1-3 OPECODE
200 case 0:
201 if ((cur_out_msg.ptr == NULL) || (((ID *)cur_out_msg.ptr->_gap)[0] != ws->wbsid))
202 return (intptr_t)pos - (intptr_t)data;
203
204 s->payload_len = _ecn_fbs_get_datalen(cur_out_msg);
205 if(s->payload_len == 0){
206 goto next;
207 }
208
209 s->fin = 1;
210 s->opecode = text_frame;
211 s->masked = 0;//
212 s->masking_key[0] = 0;//0x01;
213 s->masking_key[1] = 0;//0x23;
214 s->masking_key[2] = 0;//0x45;
215 s->masking_key[3] = 0;//0x67;
216
217 *pos = ((s->fin != 0) ? 0x80 : 0)
218 | ((char)s->opecode & 0x0F);
219 s->state = 1;
220 break;
221 // MASK Payload len
222 case 1:
223 if(s->payload_len < 126){
224 *pos = ((s->masked != 0) ? 0x80 : 0) | s->payload_len;
225 if (s->masked != 0) {
226 s->state = 10/*7bit*/;
227 }
228 else{
229 goto payload_check;
230 }
231 }
232 else if(s->payload_len < 0x10000){
233 *pos = ((s->masked != 0) ? 0x80 : 0) | 126;
234 s->state = 8/*7+16bit*/;
235 }
236 else{
237 *pos = ((s->masked != 0) ? 0x80 : 0) | 127;
238 s->state = 2/*7+64bit*/;
239 }
240 break;
241 // 64bit Extended payload length
242 case 2: case 3: case 4: case 5: case 6: case 7:
243 // 16bit Extended payload length
244 case 8: case 9:
245 *pos = (s->payload_len >> (8 * (9 - s->state)));
246 s->state++;
247 if((s->state == 10) && (s->masked == 0)){
248 goto payload_check;
249 }
250 break;
251 // Masking-key
252 case 10: case 11: case 12:
253 *pos = s->masking_key[s->state - 10];
254 s->state++;
255 break;
256 case 13:
257 *pos = s->masking_key[3];
258 s->state++;
259 payload_check:
260 s->data_pos = 0;
261 if (s->data_pos < s->payload_len)
262 s->state = 14;
263 else
264 goto next;
265 break;
266 // Payload buffer
267 default:
268 *pos = _ecn_fbs_peek(cur_out_msg, s->data_pos) ^ s->masking_key[s->data_pos % 4];
269
270 s->data_pos++;
271 if(s->data_pos >= s->payload_len){
272 goto next;
273 }
274 break;
275 next:
276 _ecn_fbs_del(cur_out_msg);
277 memset(&cur_out_msg, 0, sizeof(cur_out_msg));
278 s->state = 0;
279 websocket_newdata(ws);
280 break;
281 }
282 }
283
284 return (intptr_t)pos - (intptr_t)data;
285}
Note: See TracBrowser for help on using the repository browser.