From a0429ccf05f6398166047a96449e568cfed39ef1 Mon Sep 17 00:00:00 2001 From: Derek Meer Date: Wed, 19 Mar 2025 13:54:02 -0700 Subject: [PATCH] redbean: return early from LuaWSWrite if upgrade fails --- tool/net/redbean.c | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tool/net/redbean.c b/tool/net/redbean.c index 93d714bf9..a755fec96 100644 --- a/tool/net/redbean.c +++ b/tool/net/redbean.c @@ -5342,13 +5342,16 @@ close: } static int LuaWSWrite(lua_State *L) { - int type; + int type, retval; size_t size; const char *data; OnlyCallDuringRequest(L, "ws.Write"); if (!cpm.wstype) { - LuaWSUpgrade(L); + retval = LuaWSUpgrade(L); + if (retval != 0) { + return retval; + } } type = luaL_optinteger(L, 2, -1);