summaryrefslogtreecommitdiffstatshomepage
path: root/3rdparty/luv/src/tcp.c
diff options
context:
space:
mode:
author ImJezze <jezze@gmx.net>2016-02-21 11:48:45 +0100
committer ImJezze <jezze@gmx.net>2016-02-21 11:48:45 +0100
commitcc24a339d8c0517259084b5c178d784626ba965c (patch)
tree9868e9687b5802ae0a3733712a3bbeb3bc75c953 /3rdparty/luv/src/tcp.c
parentb5daabda5495dea5c50e17961ecfed2ea8619d76 (diff)
Merge remote-tracking branch 'refs/remotes/mamedev/master'
Second attempt
Diffstat (limited to '3rdparty/luv/src/tcp.c')
-rw-r--r--3rdparty/luv/src/tcp.c182
1 files changed, 182 insertions, 0 deletions
diff --git a/3rdparty/luv/src/tcp.c b/3rdparty/luv/src/tcp.c
new file mode 100644
index 00000000000..7ffef5ae763
--- /dev/null
+++ b/3rdparty/luv/src/tcp.c
@@ -0,0 +1,182 @@
+/*
+ * Copyright 2014 The Luvit Authors. All Rights Reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ */
+#include "luv.h"
+
+static uv_tcp_t* luv_check_tcp(lua_State* L, int index) {
+ uv_tcp_t* handle = luv_checkudata(L, index, "uv_tcp");
+ luaL_argcheck(L, handle->type == UV_TCP && handle->data, index, "Expected uv_tcp_t");
+ return handle;
+}
+
+static int luv_new_tcp(lua_State* L) {
+ uv_tcp_t* handle = luv_newuserdata(L, sizeof(*handle));
+ int ret = uv_tcp_init(luv_loop(L), handle);
+ if (ret < 0) {
+ lua_pop(L, 1);
+ return luv_error(L, ret);
+ }
+ handle->data = luv_setup_handle(L);
+ return 1;
+}
+
+static int luv_tcp_open(lua_State* L) {
+ uv_tcp_t* handle = luv_check_tcp(L, 1);
+ uv_os_sock_t sock = luaL_checkinteger(L, 2);
+ int ret = uv_tcp_open(handle, sock);
+ if (ret < 0) return luv_error(L, ret);
+ lua_pushinteger(L, ret);
+ return 1;
+}
+
+static int luv_tcp_nodelay(lua_State* L) {
+ uv_tcp_t* handle = luv_check_tcp(L, 1);
+ int ret, enable;
+ luaL_checktype(L, 2, LUA_TBOOLEAN);
+ enable = lua_toboolean(L, 2);
+ ret = uv_tcp_nodelay(handle, enable);
+ if (ret < 0) return luv_error(L, ret);
+ lua_pushinteger(L, ret);
+ return 1;
+}
+
+static int luv_tcp_keepalive(lua_State* L) {
+ uv_tcp_t* handle = luv_check_tcp(L, 1);
+ int ret, enable;
+ unsigned int delay = 0;
+ luaL_checktype(L, 2, LUA_TBOOLEAN);
+ enable = lua_toboolean(L, 2);
+ if (enable) {
+ delay = luaL_checkinteger(L, 3);
+ }
+ ret = uv_tcp_keepalive(handle, enable, delay);
+ if (ret < 0) return luv_error(L, ret);
+ lua_pushinteger(L, ret);
+ return 1;
+}
+
+static int luv_tcp_simultaneous_accepts(lua_State* L) {
+ uv_tcp_t* handle = luv_check_tcp(L, 1);
+ int ret, enable;
+ luaL_checktype(L, 2, LUA_TBOOLEAN);
+ enable = lua_toboolean(L, 2);
+ ret = uv_tcp_simultaneous_accepts(handle, enable);
+ if (ret < 0) return luv_error(L, ret);
+ lua_pushinteger(L, ret);
+ return 1;
+}
+
+static int luv_tcp_bind(lua_State* L) {
+ uv_tcp_t* handle = luv_check_tcp(L, 1);
+ const char* host = luaL_checkstring(L, 2);
+ int port = luaL_checkinteger(L, 3);
+ unsigned int flags = 0;
+ struct sockaddr_storage addr;
+ int ret;
+ if (uv_ip4_addr(host, port, (struct sockaddr_in*)&addr) &&
+ uv_ip6_addr(host, port, (struct sockaddr_in6*)&addr)) {
+ return luaL_error(L, "Invalid IP address or port [%s:%d]", host, port);
+ }
+ if (lua_type(L, 4) == LUA_TTABLE) {
+ lua_getfield(L, 4, "ipv6only");
+ if (lua_toboolean(L, -1)) flags |= UV_TCP_IPV6ONLY;
+ lua_pop(L, 1);
+ }
+ ret = uv_tcp_bind(handle, (struct sockaddr*)&addr, flags);
+ if (ret < 0) return luv_error(L, ret);
+ lua_pushinteger(L, ret);
+ return 1;
+}
+
+static void parse_sockaddr(lua_State* L, struct sockaddr_storage* address, int addrlen) {
+ char ip[INET6_ADDRSTRLEN];
+ int port = 0;
+ lua_newtable(L);
+ if (address->ss_family == AF_INET) {
+ struct sockaddr_in* addrin = (struct sockaddr_in*)address;
+ uv_inet_ntop(AF_INET, &(addrin->sin_addr), ip, addrlen);
+ port = ntohs(addrin->sin_port);
+ } else if (address->ss_family == AF_INET6) {
+ struct sockaddr_in6* addrin6 = (struct sockaddr_in6*)address;
+ uv_inet_ntop(AF_INET6, &(addrin6->sin6_addr), ip, addrlen);
+ port = ntohs(addrin6->sin6_port);
+ }
+
+ lua_pushstring(L, luv_af_num_to_string(address->ss_family));
+ lua_setfield(L, -2, "family");
+ lua_pushinteger(L, port);
+ lua_setfield(L, -2, "port");
+ lua_pushstring(L, ip);
+ lua_setfield(L, -2, "ip");
+}
+
+static int luv_tcp_getsockname(lua_State* L) {
+ uv_tcp_t* handle = luv_check_tcp(L, 1);
+ struct sockaddr_storage address;
+ int addrlen = sizeof(address);
+ int ret = uv_tcp_getsockname(handle, (struct sockaddr*)&address, &addrlen);
+ if (ret < 0) return luv_error(L, ret);
+ parse_sockaddr(L, &address, addrlen);
+ return 1;
+}
+
+static int luv_tcp_getpeername(lua_State* L) {
+ uv_tcp_t* handle = luv_check_tcp(L, 1);
+ struct sockaddr_storage address;
+ int addrlen = sizeof(address);
+ int ret = uv_tcp_getpeername(handle, (struct sockaddr*)&address, &addrlen);
+ if (ret < 0) return luv_error(L, ret);
+ parse_sockaddr(L, &address, addrlen);
+ return 1;
+}
+
+
+static void luv_connect_cb(uv_connect_t* req, int status) {
+ lua_State* L = luv_state(req->handle->loop);
+ luv_status(L, status);
+ luv_fulfill_req(L, req->data, 1);
+ luv_cleanup_req(L, req->data);
+ req->data = NULL;
+}
+
+static int luv_write_queue_size(lua_State* L) {
+ uv_tcp_t* handle = luv_check_tcp(L, 1);
+ lua_pushinteger(L, handle->write_queue_size);
+ return 1;
+}
+
+static int luv_tcp_connect(lua_State* L) {
+ uv_tcp_t* handle = luv_check_tcp(L, 1);
+ const char* host = luaL_checkstring(L, 2);
+ int port = luaL_checkinteger(L, 3);
+ struct sockaddr_storage addr;
+ uv_connect_t* req;
+ int ret, ref;
+ if (uv_ip4_addr(host, port, (struct sockaddr_in*)&addr) &&
+ uv_ip6_addr(host, port, (struct sockaddr_in6*)&addr)) {
+ return luaL_error(L, "Invalid IP address or port [%s:%d]", host, port);
+ }
+ ref = luv_check_continuation(L, 4);
+
+ req = lua_newuserdata(L, sizeof(*req));
+ req->data = luv_setup_req(L, ref);
+ ret = uv_tcp_connect(req, handle, (struct sockaddr*)&addr, luv_connect_cb);
+ if (ret < 0) {
+ lua_pop(L, 1);
+ return luv_error(L, ret);
+ }
+ return 1;
+}