Minor patch

stamparm · stamparm · commit 788c9fa1349c · 2026-07-06T18:54:26.000+02:00
diff --git a/lib/core/settings.py b/lib/core/settings.py
@@ -20,7 +20,7 @@
 from thirdparty import six
 
 # sqlmap version (<major>.<minor>.<month>.<monthly commit>)
-VERSION = "1.10.7.32"
+VERSION = "1.10.7.33"
 TYPE = "dev" if VERSION.count('.') > 2 and VERSION.split('.')[-1] != '0' else "stable"
 TYPE_COLORS = {"dev": 33, "stable": 90, "pip": 34}
 VERSION_STRING = "sqlmap/%s#%s" % ('.'.join(VERSION.split('.')[:-1]) if VERSION.count('.') > 2 and VERSION.split('.')[-1] == '0' else VERSION, TYPE)
diff --git a/lib/request/connect.py b/lib/request/connect.py
@@ -524,31 +524,33 @@ def getPage(**kwargs):
 
             if webSocket:
                 ws = websocket.WebSocket()
-                ws.settimeout(WEBSOCKET_INITIAL_TIMEOUT if kb.webSocketRecvCount is None else timeout)
-                wsHeaders = tuple("%s: %s" % (getUnicode(key), getUnicode(value)) for key, value in headers.items() if getUnicode(key).upper() != HTTP_HEADER.HOST.upper())
-                ws.connect(url, header=wsHeaders, cookie=cookie)  # WebSocket will add Host field of headers automatically
-                ws.send(urldecode(post or ""))
-
-                _page = []
-
-                if kb.webSocketRecvCount is None:
-                    while True:
-                        try:
-                            _page.append(ws.recv())
-                            if sum(len(_) for _ in _page) > MAX_CONNECTION_TOTAL_SIZE:
-                                warnMsg = "too large websocket response detected. Automatically trimming it"
-                                singleTimeWarnMessage(warnMsg)
+                try:
+                    ws.settimeout(WEBSOCKET_INITIAL_TIMEOUT if kb.webSocketRecvCount is None else timeout)
+                    wsHeaders = tuple("%s: %s" % (getUnicode(key), getUnicode(value)) for key, value in headers.items() if getUnicode(key).upper() != HTTP_HEADER.HOST.upper())
+                    ws.connect(url, header=wsHeaders, cookie=cookie)  # WebSocket will add Host field of headers automatically
+                    ws.send(urldecode(post or ""))
+
+                    _page = []
+
+                    if kb.webSocketRecvCount is None:
+                        while True:
+                            try:
+                                _page.append(ws.recv())
+                                if sum(len(_) for _ in _page) > MAX_CONNECTION_TOTAL_SIZE:
+                                    warnMsg = "too large websocket response detected. Automatically trimming it"
+                                    singleTimeWarnMessage(warnMsg)
+                                    break
+                            except websocket.WebSocketTimeoutException:
+                                kb.webSocketRecvCount = len(_page)
                                 break
-                        except websocket.WebSocketTimeoutException:
-                            kb.webSocketRecvCount = len(_page)
-                            break
-                else:
-                    for i in xrange(max(1, kb.webSocketRecvCount)):
-                        _page.append(ws.recv())
+                    else:
+                        for i in xrange(max(1, kb.webSocketRecvCount)):
+                            _page.append(ws.recv())
 
-                page = "\n".join(_page)
+                    page = "\n".join(_page)
+                finally:
+                    ws.close()
 
-                ws.close()
                 code = ws.status
                 status = _http_client.responses.get(code, "")
 
