Mina的WebSocket问题排查[转]

笔者上个项目经历过的一个bug。

项目使用mina进行收发包,收发包的代码是直接从Apache讨论组那边拷贝过来的。在后来的开发过程中,就出现了不必现的网络包读取失败,导致websocket连接中断。

过程和下面转载的文章提到的差不多,结论是一样的,所以记录下来,方便mina使用者跳过这个坑。


转载请注明文章出处:
https://tlanyan.me/resolve-mina-websocket-problem/

项目的客户端需要重构,探讨后决定客户端与服务器的交互使用WebSocket协议。后端的网络层框架是Mina,在现有基础上增加对WebSocket的解析即可完成无缝迁移。正好Apache讨论组上有人提供Mina的WebSocket过滤器代码。下载后添加到项目,扔到服务器上一切正常。

今天前端同事反馈说请求某个接口会导致客户端立马报错然后断开连接,提示信息为:One or more reserved bits are on, reserved1 = 0, reserved2 = 1, reserved3 = 1。看到信息后,第一反应是:数据没封包就发送到客户端了?排查log,确认数据按照WebSocket的格式封装后再发送出去的。

根据错误信息上网查解决方案,基本上甩锅给杀毒软件、系统、浏览器等,没找到靠谱的原因和解决方法。没办法只好让前端开wireshark抓包,与服务端数据进行对比。一番折腾后,确认是因为接口返回大量数据,客户端没能完整接收造成的。

数据量大才会出现,难道WebSocket客户端的缓冲区或者单次传输有大小限制?上网搜寻后发现都没有限制,很有可能是服务端没能正确封包数据造成的。由于使用的是第三方非权威代码对数据封包,过一遍编码的代码是必要的:

 1  /*
 2   * To change this template, choose Tools | Templates
 3   * and open the template in the editor.
 4   */
 5  package com.shephertz.appwarp.websocket.binary;
 6  
 7  import org.apache.mina.core.buffer.IoBuffer;
 8  import org.apache.mina.core.session.IoSession;
 9  import org.apache.mina.filter.codec.ProtocolEncoderAdapter;
10  import org.apache.mina.filter.codec.ProtocolEncoderOutput;
11  
12  /**
13   * Encodes incoming buffers in a manner that makes the receiving client type transparent to the 
14   * encoders further up in the filter chain. If the receiving client is a native client then
15   * the buffer contents are simply passed through. If the receiving client is a websocket, it will encode
16   * the buffer contents in to WebSocket DataFrame before passing it along the filter chain.
17   * 
18   * Note: you must wrap the IoBuffer you want to send around a WebSocketCodecPacket instance.
19   * 
20   * @author DHRUV CHOPRA
21   */
22  public class WebSocketEncoder extends ProtocolEncoderAdapter{
23  
24      @Override
25      public void encode(IoSession session, Object message, ProtocolEncoderOutput out) throws Exception {
26          boolean isHandshakeResponse = message instanceof WebSocketHandShakeResponse;
27          boolean isDataFramePacket = message instanceof WebSocketCodecPacket;
28          boolean isRemoteWebSocket = session.containsAttribute(WebSocketUtils.SessionAttribute) && (true==(Boolean)session.getAttribute(WebSocketUtils.SessionAttribute));
29          IoBuffer resultBuffer;
30          if(isHandshakeResponse){
31              WebSocketHandShakeResponse response = (WebSocketHandShakeResponse)message;
32              resultBuffer = WebSocketEncoder.buildWSResponseBuffer(response);
33          }
34          else if(isDataFramePacket){
35              WebSocketCodecPacket packet = (WebSocketCodecPacket)message;
36              resultBuffer = isRemoteWebSocket ? WebSocketEncoder.buildWSDataFrameBuffer(packet.getPacket()) : packet.getPacket();
37          }
38          else{
39              throw (new Exception("message not a websocket type"));
40          }
41          
42          out.write(resultBuffer);
43      }
44      
45      // Web Socket handshake response go as a plain string.
46      private static IoBuffer buildWSResponseBuffer(WebSocketHandShakeResponse response) {                
47          IoBuffer buffer = IoBuffer.allocate(response.getResponse().getBytes().length, false);
48          buffer.setAutoExpand(true);
49          buffer.put(response.getResponse().getBytes());
50          buffer.flip();
51          return buffer;
52      }
53      
54      // Encode the in buffer according to the Section 5.2. RFC 6455
55      private static IoBuffer buildWSDataFrameBuffer(IoBuffer buf) {
56          
57          IoBuffer buffer = IoBuffer.allocate(buf.limit() + 2, false);
58          buffer.setAutoExpand(true);
59          buffer.put((byte) 0x82);
60          if(buffer.capacity() <= 125){
61              byte capacity = (byte) (buf.limit());
62              buffer.put(capacity);
63          }
64          else{
65              buffer.put((byte)126);
66              buffer.putShort((short)buf.limit());
67          }        
68          buffer.put(buf);
69          buffer.flip();
70          return buffer;
71      }
72  }


代码的51到71行对数据进行封包,直觉提示60行到67行的封包处理在数据量大时有问题。打开WebSocket的rfc页面,查看规范中定义的数据包结构。根据规范,数据包小于125时,只需增加两个字节的包头;大于等于126小于等于2^16时,包头有4个字节;数据长度更大时,包头有10个字节,其中8个字节用来存储数据的长度。

  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 ...                |
 +---------------------------------------------------------------+

下载的第三方代码只处理了payload len<=126的情形,数据量大时payload == 127的情形仍按照payload len == 126方式处理,显然会导致客户端接收不完整数据。将构建WebSocket数据帧的函数更改如下,顺利解决问题:

    // Encode the in buffer according to the Section 5.2. RFC 6455
    private static IoBuffer buildWSDataFrameBuffer(IoBuffer buf) {
        IoBuffer buffer = null;
        if (buf.limit() <= 125) {
            buffer = IoBuffer.allocate(buf.limit() + 2, false);
            buffer.put((byte) 0x82);
            buffer.put((byte)buf.limit());
        } else if (buf.limit() <= 0xFFFF) {
            buffer = IoBuffer.allocate(buf.limit() + 4, false);
            buffer.put((byte) 0x82);
            buffer.put((byte)126);
            buffer.putShort((short)buf.limit());
        } else {
            buffer = IoBuffer.allocate(buf.limit() + 10, false);
            buffer.put((byte) 0x82);
            buffer.put((byte)127);
            buffer.putLong((long)buf.limit());
        }

        buffer.put(buf);
        buffer.flip();
        return buffer;
    }

主要改动两点:1. 新增数据大于2^16时的处理; 2. 分配大小合适的缓冲区,避免触发重分配。

总结

根据规范,类似"One or more reserved bits are on"的错误提示,是服务端没有正确封包造成的。另外在github上搜websocket解析库,发现 cjoo/WebSocket_mina 这个库中的封包处理也是错误的。错误的地方在 https://github.com/cjoo/WebSo... 中的70行到72行。

参考

https://issues.apache.org/jira/browse/DIRMINA-907
https://tools.ietf.org/html/rfc6455#section-5.2

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 205,236评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,867评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,715评论 0 340
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,899评论 1 278
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,895评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,733评论 1 283
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,085评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,722评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,025评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,696评论 2 323
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,816评论 1 333
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,447评论 4 322
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,057评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,009评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,254评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,204评论 2 352
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,561评论 2 343

推荐阅读更多精彩内容

  • 原文地址:http://www.ibm.com/developerworks/cn/java/j-lo-WebSo...
    敢梦敢当阅读 8,882评论 0 50
  • 1 述 WebSocket是一种网络通信协议WebSocket 协议在2008年诞生,2011年成为国际标准。HT...
    凯玲之恋阅读 670评论 0 0
  • 以阳光作墨 轻水作笔 在绿苔之间 绘制 山水 斜阳
    赵着急_阅读 215评论 0 2
  • 本来昨天计划更新的,可惜,出去吃饭的时间陪伴我两年的手机丢失了,打电话关机,估计被谁拿去用了,不过所幸,我手机没有...
    言和意顺阅读 244评论 0 0
  • 传播正能量大概是一种使命,在忙碌的每一天,我们是每一个渺小的自己,在为着我们不确定的未来,努力。 芒果台新开的一档...
    木可栗阅读 464评论 0 0