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.
20 #include <sys/types.h>
21 #include <sys/socket.h>
22 #include <netinet/in.h>
23 #include <netinet/tcp.h>
34 static int nixio_sock_setblocking(lua_State *L) {
35 int fd = nixio__checkfd(L, 1);
37 int set = lua_toboolean(L, 2);
38 int flags = fcntl(fd, F_GETFL);
41 return nixio__perror(L);
50 return nixio__pstatus(L, !fcntl(fd, F_SETFL, flags));
53 static int nixio__gso_int(lua_State *L, int fd, int level, int opt, int set) {
55 socklen_t optlen = sizeof(value);
57 if (!getsockopt(fd, level, opt, &value, &optlen)) {
58 lua_pushinteger(L, value);
62 value = luaL_checkinteger(L, set);
63 if (!setsockopt(fd, level, opt, &value, optlen)) {
64 lua_pushboolean(L, 1);
68 return nixio__perror(L);
71 static int nixio__gso_ling(lua_State *L, int fd, int level, int opt, int set) {
73 socklen_t optlen = sizeof(value);
75 if (!getsockopt(fd, level, opt, &value, &optlen)) {
76 lua_pushinteger(L, value.l_onoff ? value.l_linger : 0);
80 value.l_linger = luaL_checkinteger(L, set);
81 value.l_onoff = value.l_linger ? 1 : 0;
82 if (!setsockopt(fd, level, opt, &value, optlen)) {
83 lua_pushboolean(L, 1);
87 return nixio__perror(L);
90 static int nixio__gso_timev(lua_State *L, int fd, int level, int opt, int set) {
92 socklen_t optlen = sizeof(value);
94 if (!getsockopt(fd, level, opt, &value, &optlen)) {
95 lua_pushinteger(L, value.tv_sec);
96 lua_pushinteger(L, value.tv_usec);
100 value.tv_sec = luaL_checkinteger(L, set);
101 value.tv_usec = luaL_optinteger(L, set + 1, 0);
102 if (!setsockopt(fd, level, opt, &value, optlen)) {
103 lua_pushboolean(L, 1);
107 return nixio__perror(L);
111 * get/setsockopt() helper
113 static int nixio__getsetsockopt(lua_State *L, int set) {
114 nixio_sock *sock = nixio__checksock(L);
115 const char *level = luaL_optlstring(L, 2, "", NULL);
116 const char *option = luaL_optlstring(L, 3, "", NULL);
119 if (!strcmp(level, "socket")) {
120 if (!strcmp(option, "keepalive")) {
121 return nixio__gso_int(L, sock->fd, SOL_SOCKET, SO_KEEPALIVE, set);
122 } else if (!strcmp(option, "reuseaddr")) {
123 return nixio__gso_int(L, sock->fd, SOL_SOCKET, SO_REUSEADDR, set);
124 } else if (!strcmp(option, "rcvbuf")) {
125 return nixio__gso_int(L, sock->fd, SOL_SOCKET, SO_RCVBUF, set);
126 } else if (!strcmp(option, "sndbuf")) {
127 return nixio__gso_int(L, sock->fd, SOL_SOCKET, SO_SNDBUF, set);
128 } else if (!strcmp(option, "priority")) {
130 return nixio__gso_int(L, sock->fd, SOL_SOCKET, SO_PRIORITY, set);
132 return nixio__pstatus(L, !(errno = ENOPROTOOPT));
134 } else if (!strcmp(option, "broadcast")) {
135 return nixio__gso_int(L, sock->fd, SOL_SOCKET, SO_BROADCAST, set);
136 } else if (!strcmp(option, "linger")) {
137 return nixio__gso_ling(L, sock->fd, SOL_SOCKET, SO_LINGER, set);
138 } else if (!strcmp(option, "sndtimeo")) {
139 return nixio__gso_timev(L, sock->fd, SOL_SOCKET, SO_SNDTIMEO, set);
140 } else if (!strcmp(option, "rcvtimeo")) {
141 return nixio__gso_timev(L, sock->fd, SOL_SOCKET, SO_RCVTIMEO, set);
143 return luaL_argerror(L, 3, "supported values: keepalive, reuseaddr,"
144 " sndbuf, rcvbuf, priority, broadcast, linger, sndtimeo, rcvtimeo"
147 } else if (!strcmp(level, "tcp")) {
148 if (sock->type != SOCK_STREAM) {
149 return luaL_error(L, "not a TCP socket");
151 if (!strcmp(option, "cork")) {
153 return nixio__gso_int(L, sock->fd, IPPROTO_TCP, TCP_CORK, set);
155 return nixio__pstatus(L, !(errno = ENOPROTOOPT));
157 } else if (!strcmp(option, "nodelay")) {
158 return nixio__gso_int(L, sock->fd, IPPROTO_TCP, TCP_NODELAY, set);
160 return luaL_argerror(L, 3, "supported values: cork, nodelay");
163 return luaL_argerror(L, 2, "supported values: socket, tcp");
170 static int nixio_sock_getsockopt(lua_State *L) {
171 return nixio__getsetsockopt(L, 0);
177 static int nixio_sock_setsockopt(lua_State *L) {
178 return nixio__getsetsockopt(L, 1);
182 static const luaL_reg M[] = {
183 {"setblocking", nixio_sock_setblocking},
184 {"getsockopt", nixio_sock_getsockopt},
185 {"setsockopt", nixio_sock_setsockopt},
189 void nixio_open_sockopt(lua_State *L) {
190 lua_pushvalue(L, -2);
191 luaL_register(L, NULL, M);
194 luaL_getmetatable(L, NIXIO_FILE_META);
195 lua_pushcfunction(L, nixio_sock_setblocking);
196 lua_setfield(L, -2, "setblocking");