瀏覽代碼

tls: get wss connections as well for selects and vars

- GH #4167
Daniel-Constantin Mierla 6 月之前
父節點
當前提交
3a3856c68c
共有 1 個文件被更改,包括 2 次插入2 次删除
  1. 2 2
      src/modules/tls/tls_select.c

+ 2 - 2
src/modules/tls/tls_select.c

@@ -129,14 +129,14 @@ struct tcp_connection *get_cur_connection(struct sip_msg *msg)
 	if(_tls_pv_con != 0)
 		return _tls_pv_con;
 
-	if(msg->rcv.proto != PROTO_TLS) {
+	if(msg->rcv.proto != PROTO_TLS && msg->rcv.proto != PROTO_WSS) {
 		ERR("Transport protocol is not TLS (bug in config)\n");
 		return 0;
 	}
 
 	c = tcpconn_get(msg->rcv.proto_reserved1, 0, 0, 0,
 			cfg_get(tls, tls_cfg, con_lifetime));
-	if(c && c->type != PROTO_TLS) {
+	if(c && c->type != PROTO_TLS && msg->rcv.proto != PROTO_WSS) {
 		ERR("Connection found but is not TLS\n");
 		tcpconn_put(c);
 		return 0;