[tarantool-patches] [PATCH] socket: better args handling for connect()/bind()
Roman Khabibov
roman.habibov at tarantool.org
Tue Sep 10 15:58:38 MSK 2019
Check args types in socket.connect() and socket.bind() functions.
'host' should be a string and 'port' should be a string or a
number.
Part of #4138
---
Branch: https://github.com/tarantool/tarantool/tree/romanhabibov/gh-4138-getaddrinfo
Issue: https://github.com/tarantool/tarantool/issues/4138
src/lua/socket.lua | 6 ++++--
test/app/socket.result | 25 +++++++++++++++++++++++++
test/app/socket.test.lua | 8 ++++++++
3 files changed, 37 insertions(+), 2 deletions(-)
diff --git a/src/lua/socket.lua b/src/lua/socket.lua
index e4815adbb..c4a61074e 100644
--- a/src/lua/socket.lua
+++ b/src/lua/socket.lua
@@ -1554,7 +1554,8 @@ local function lsocket_tcp()
end
local function lsocket_connect(host, port)
- if host == nil or port == nil then
+ if type(host) ~= 'string' or (type(port) ~= 'string' and
+ type(port) ~= 'number') then
error("Usage: luasocket.connect(host, port)")
end
local s, err = tcp_connect(host, port)
@@ -1569,7 +1570,8 @@ local function lsocket_connect(host, port)
end
local function lsocket_bind(host, port, backlog)
- if host == nil or port == nil then
+ if type(host) ~= 'string' or (type(port) ~= 'string' and
+ type(port) ~= 'number') then
error("Usage: luasocket.bind(host, port [, backlog])")
end
local function prepare(s) return backlog end
diff --git a/test/app/socket.result b/test/app/socket.result
index 87b1d0387..20507bafb 100644
--- a/test/app/socket.result
+++ b/test/app/socket.result
@@ -2857,6 +2857,31 @@ check_err(err)
---
- true
...
+-- gh-4138 Check the usage error in case of wrong args types.
+socket.connect('127.0.0.1', {3301})
+---
+- error: 'builtin/socket.lua: Usage: luasocket.connect(host, port)'
+...
+socket.bind('127.0.0.1', {3301})
+---
+- error: 'builtin/socket.lua: Usage: luasocket.bind(host, port [, backlog])'
+...
+socket.connect(127.0.0.1, 3301)
+---
+- error: '[string "socket.connect(127.0.0.1, 3301) "]:1: malformed number near ''127.0.0.1'''
+...
+socket.bind(127.0.0.1, 3301)
+---
+- error: '[string "socket.bind(127.0.0.1, 3301) "]:1: malformed number near ''127.0.0.1'''
+...
+socket.connect({'127.0.0.1'}, 3301)
+---
+- error: 'builtin/socket.lua: Usage: luasocket.connect(host, port)'
+...
+socket.bind({'127.0.0.1'}, 3301)
+---
+- error: 'builtin/socket.lua: Usage: luasocket.bind(host, port [, backlog])'
+...
test_run:cmd("clear filter")
---
- true
diff --git a/test/app/socket.test.lua b/test/app/socket.test.lua
index 08cb7fc9e..1974e8cea 100644
--- a/test/app/socket.test.lua
+++ b/test/app/socket.test.lua
@@ -985,6 +985,14 @@ check_err(err)
s, err = socket.connect('non_exists_hostname', 3301)
check_err(err)
+-- gh-4138 Check the usage error in case of wrong args types.
+socket.connect('127.0.0.1', {3301})
+socket.bind('127.0.0.1', {3301})
+socket.connect(127.0.0.1, 3301)
+socket.bind(127.0.0.1, 3301)
+socket.connect({'127.0.0.1'}, 3301)
+socket.bind({'127.0.0.1'}, 3301)
+
test_run:cmd("clear filter")
-- case: sicket receive inconsistent behavior
--
2.20.1 (Apple Git-117)
More information about the Tarantool-patches
mailing list