2 * nixio - Linux I/O library for lua
4 * Copyright (C) 2009 Steven Barth <steven@midlink.org>
6 * Licensed under the Apache License, Version 2.0 (the "License");
7 * you may not use this file except in compliance with the License.
8 * You may obtain a copy of the License at
10 * http://www.apache.org/licenses/LICENSE-2.0
12 * Unless required by applicable law or agreed to in writing, software
13 * distributed under the License is distributed on an "AS IS" BASIS,
14 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 * See the License for the specific language governing permissions and
16 * limitations under the License.
22 #include <sys/socket.h>
23 #include <netinet/in.h>
24 #include <netinet/tcp.h>
33 static int nixio_sock_setblocking(lua_State *L) {
34 int fd = nixio__checkfd(L, 1);
35 int set = lua_toboolean(L, 2);
36 int flags = fcntl(fd, F_GETFL);
39 return nixio__perror(L);
48 return nixio__pstatus(L, !fcntl(fd, F_SETFL, flags));
51 static int nixio__gso_int(lua_State *L, int fd, int level, int opt, int set) {
53 socklen_t optlen = sizeof(value);
55 if (!getsockopt(fd, level, opt, &value, &optlen)) {
56 lua_pushinteger(L, value);
60 value = luaL_checkinteger(L, set);
61 if (!setsockopt(fd, level, opt, &value, optlen)) {
62 lua_pushboolean(L, 1);
66 return nixio__perror(L);
69 static int nixio__gso_ling(lua_State *L, int fd, int level, int opt, int set) {
71 socklen_t optlen = sizeof(value);
73 if (!getsockopt(fd, level, opt, &value, &optlen)) {
74 lua_pushinteger(L, value.l_onoff ? value.l_linger : 0);
78 value.l_linger = luaL_checkinteger(L, set);
79 value.l_onoff = value.l_linger ? 1 : 0;
80 if (!setsockopt(fd, level, opt, &value, optlen)) {
81 lua_pushboolean(L, 1);
85 return nixio__perror(L);
88 static int nixio__gso_timev(lua_State *L, int fd, int level, int opt, int set) {
90 socklen_t optlen = sizeof(value);
92 if (!getsockopt(fd, level, opt, &value, &optlen)) {
93 lua_pushinteger(L, value.tv_sec);
94 lua_pushinteger(L, value.tv_usec);
98 value.tv_sec = luaL_checkinteger(L, set);
99 value.tv_usec = luaL_optinteger(L, set + 1, 0);
100 if (!setsockopt(fd, level, opt, &value, optlen)) {
101 lua_pushboolean(L, 1);
105 return nixio__perror(L);
109 * get/setsockopt() helper
111 static int nixio__getsetsockopt(lua_State *L, int set) {
112 nixio_sock *sock = nixio__checksock(L);
113 const char *level = luaL_optlstring(L, 2, "", NULL);
114 const char *option = luaL_optlstring(L, 3, "", NULL);
117 if (!strcmp(level, "socket")) {
118 if (!strcmp(option, "keepalive")) {
119 return nixio__gso_int(L, sock->fd, SOL_SOCKET, SO_KEEPALIVE, set);
120 } else if (!strcmp(option, "reuseaddr")) {
121 return nixio__gso_int(L, sock->fd, SOL_SOCKET, SO_REUSEADDR, set);
122 } else if (!strcmp(option, "rcvbuf")) {
123 return nixio__gso_int(L, sock->fd, SOL_SOCKET, SO_RCVBUF, set);
124 } else if (!strcmp(option, "sndbuf")) {
125 return nixio__gso_int(L, sock->fd, SOL_SOCKET, SO_SNDBUF, set);
126 } else if (!strcmp(option, "priority")) {
127 return nixio__gso_int(L, sock->fd, SOL_SOCKET, SO_PRIORITY, set);
128 } else if (!strcmp(option, "broadcast")) {
129 return nixio__gso_int(L, sock->fd, SOL_SOCKET, SO_BROADCAST, set);
130 } else if (!strcmp(option, "linger")) {
131 return nixio__gso_ling(L, sock->fd, SOL_SOCKET, SO_LINGER, set);
132 } else if (!strcmp(option, "sndtimeo")) {
133 return nixio__gso_timev(L, sock->fd, SOL_SOCKET, SO_SNDTIMEO, set);
134 } else if (!strcmp(option, "rcvtimeo")) {
135 return nixio__gso_timev(L, sock->fd, SOL_SOCKET, SO_RCVTIMEO, set);
137 return luaL_argerror(L, 3, "supported values: keepalive, reuseaddr,"
138 " sndbuf, rcvbuf, priority, broadcast, linger, sndtimeo, rcvtimeo"
141 } else if (!strcmp(level, "tcp")) {
142 if (sock->type != SOCK_STREAM) {
143 return luaL_error(L, "not a TCP socket");
145 if (!strcmp(option, "cork")) {
146 return nixio__gso_int(L, sock->fd, SOL_TCP, TCP_CORK, set);
147 } else if (!strcmp(option, "nodelay")) {
148 return nixio__gso_int(L, sock->fd, SOL_TCP, TCP_NODELAY, set);
150 return luaL_argerror(L, 3, "supported values: cork, nodelay");
153 return luaL_argerror(L, 2, "supported values: socket, tcp");
160 static int nixio_sock_getsockopt(lua_State *L) {
161 return nixio__getsetsockopt(L, 0);
167 static int nixio_sock_setsockopt(lua_State *L) {
168 return nixio__getsetsockopt(L, 1);
172 static const luaL_reg M[] = {
173 {"setblocking", nixio_sock_setblocking},
174 {"getsockopt", nixio_sock_getsockopt},
175 {"setsockopt", nixio_sock_setsockopt},
179 void nixio_open_sockopt(lua_State *L) {
180 lua_pushvalue(L, -2);
181 luaL_register(L, NULL, M);
184 luaL_getmetatable(L, LUA_FILEHANDLE);
185 lua_pushcfunction(L, nixio_sock_setblocking);
186 lua_setfield(L, -2, "setblocking");