mirror of
https://github.com/ctruLua/ctruLua.git
synced 2025-10-27 16:39:29 +00:00
Added UDP sockets functions (untested, like 80% of my code)
I didn't find a server/protocol to test the UDP sockets, so I didn't test. Also, there's no built-in "pseudo-connected" mode, you have to manage it in your code.
This commit is contained in:
parent
8e94add9a6
commit
81f8edb25d
1 changed files with 132 additions and 25 deletions
143
source/socket.c
143
source/socket.c
|
|
@ -1,5 +1,6 @@
|
|||
/***
|
||||
The `socket` module. Almost like luasocket, but for TCP only.
|
||||
The `socket` module. Almost like luasocket, but for the TCP part only.
|
||||
The UDP part is only without connection.
|
||||
@module ctr.socket
|
||||
@usage local socket = require("ctr.socket")
|
||||
*/
|
||||
|
|
@ -79,34 +80,34 @@ static int socket_tcp(lua_State *L) {
|
|||
}
|
||||
|
||||
/***
|
||||
TCP Sockets
|
||||
@section TCP
|
||||
Return an UDP socket.
|
||||
@function udp
|
||||
@treturn UDPMaster UDP socket
|
||||
*/
|
||||
|
||||
/***
|
||||
Accept a connection on a server.
|
||||
@function :accept
|
||||
@treturn TCPClient tcp client object, or nil.
|
||||
*/
|
||||
static int socket_accept(lua_State *L) {
|
||||
socket_userdata *userdata = luaL_checkudata(L, 1, "LSocket");
|
||||
|
||||
socket_userdata *client = lua_newuserdata(L, sizeof(*client));
|
||||
static int socket_udp(lua_State *L) {
|
||||
socket_userdata *userdata = lua_newuserdata(L, sizeof(*userdata));
|
||||
luaL_getmetatable(L, "LSocket");
|
||||
lua_setmetatable(L, -2);
|
||||
|
||||
socklen_t addrSize = sizeof(client->addr);
|
||||
client->socket = accept(userdata->socket, (struct sockaddr*)&client->addr, &addrSize);
|
||||
if (client->socket < 0) {
|
||||
userdata->socket = socket(AF_INET, SOCK_DGRAM, 0);
|
||||
if (userdata->socket < 0) {
|
||||
lua_pushnil(L);
|
||||
return 1;
|
||||
lua_pushstring(L, "Failed to create a TCP socket");
|
||||
return 2;
|
||||
}
|
||||
|
||||
userdata->addr.sin_family = AF_INET;
|
||||
|
||||
return 1;
|
||||
}
|
||||
|
||||
/***
|
||||
Bind a socket. The TCP object become a TCPServer object.
|
||||
All sockets
|
||||
@section sockets
|
||||
*/
|
||||
|
||||
/***
|
||||
Bind a socket. The socket object become a socketServer object.
|
||||
@function :bind
|
||||
@tparam number port the port to bind the socket on.
|
||||
*/
|
||||
|
|
@ -134,6 +135,33 @@ static int socket_close(lua_State *L) {
|
|||
return 0;
|
||||
}
|
||||
|
||||
/***
|
||||
TCP Sockets
|
||||
@section TCP
|
||||
*/
|
||||
|
||||
/***
|
||||
Accept a connection on a server.
|
||||
@function :accept
|
||||
@treturn TCPClient tcp client object, or nil.
|
||||
*/
|
||||
static int socket_accept(lua_State *L) {
|
||||
socket_userdata *userdata = luaL_checkudata(L, 1, "LSocket");
|
||||
|
||||
socket_userdata *client = lua_newuserdata(L, sizeof(*client));
|
||||
luaL_getmetatable(L, "LSocket");
|
||||
lua_setmetatable(L, -2);
|
||||
|
||||
socklen_t addrSize = sizeof(client->addr);
|
||||
client->socket = accept(userdata->socket, (struct sockaddr*)&client->addr, &addrSize);
|
||||
if (client->socket < 0) {
|
||||
lua_pushnil(L);
|
||||
return 1;
|
||||
}
|
||||
|
||||
return 1;
|
||||
}
|
||||
|
||||
/***
|
||||
Connect a socket to a server. The TCP object becomes a TCPClient object.
|
||||
@function :connect
|
||||
|
|
@ -254,11 +282,88 @@ static int socket_send(lua_State *L) {
|
|||
return 1;
|
||||
}
|
||||
|
||||
/***
|
||||
UDP sockets
|
||||
@section UDP
|
||||
*/
|
||||
|
||||
/***
|
||||
Receive some data from a server.
|
||||
@function :receivefrom
|
||||
@tparam number count amount of data to receive
|
||||
@tparam string host host name
|
||||
@tparam number port port
|
||||
@treturn string data
|
||||
*/
|
||||
static int socket_receivefrom(lua_State *L) {
|
||||
socket_userdata *userdata = luaL_checkudata(L, 1, "LSocket");
|
||||
int count = luaL_checkinteger(L, 2);
|
||||
size_t namesize = 0;
|
||||
char *hostname = (char*)luaL_optlstring(L, 3, NULL, &namesize);
|
||||
int port = luaL_optinteger(L, 4, 0);
|
||||
|
||||
struct sockaddr_in from = {0};
|
||||
if (hostname != NULL) { // For a server
|
||||
struct hostent *hostinfo = gethostbyname(hostname);
|
||||
if (hostinfo == NULL) {
|
||||
lua_pushnil(L);
|
||||
return 1;
|
||||
}
|
||||
from.sin_addr = *(struct in_addr*)hostinfo->h_addr;
|
||||
from.sin_port = htons(port);
|
||||
from.sin_family = AF_INET;
|
||||
}
|
||||
|
||||
char* buffer = malloc(count+1);
|
||||
int n = recvfrom(userdata->socket, buffer, count, 0, (struct sockaddr*)&from, NULL);
|
||||
*(buffer+n) = 0x0;
|
||||
|
||||
lua_pushstring(L, buffer);
|
||||
if (hostname != NULL) {
|
||||
return 1;
|
||||
} else {
|
||||
lua_pushstring(L, inet_ntoa(from.sin_addr));
|
||||
lua_pushinteger(L, ntohs(from.sin_port));
|
||||
return 3;
|
||||
}
|
||||
}
|
||||
|
||||
/***
|
||||
Send some data to a server.
|
||||
@function :sendto
|
||||
@tparam string data data to send
|
||||
@tparam string host host name
|
||||
@tparam number port port
|
||||
*/
|
||||
static int socket_sendto(lua_State *L) {
|
||||
socket_userdata *userdata = luaL_checkudata(L, 1, "LSocket");
|
||||
size_t datasize = 0;
|
||||
char *data = (char*)luaL_checklstring(L, 2, &datasize);
|
||||
size_t namesize = 0;
|
||||
char *hostname = (char*)luaL_checklstring(L, 3, &namesize);
|
||||
int port = luaL_checkinteger(L, 4);
|
||||
|
||||
struct hostent *hostinfo = gethostbyname(hostname);
|
||||
if (hostinfo == NULL) {
|
||||
lua_pushnil(L);
|
||||
return 1;
|
||||
}
|
||||
struct sockaddr_in to = {0};
|
||||
to.sin_addr = *(struct in_addr*)hostinfo->h_addr;
|
||||
to.sin_port = htons(port);
|
||||
to.sin_family = AF_INET;
|
||||
|
||||
sendto(userdata->socket, data, datasize, 0, (struct sockaddr*)&to, sizeof(to));
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
// module functions
|
||||
static const struct luaL_Reg socket_functions[] = {
|
||||
{"init", socket_init },
|
||||
{"shutdown", socket_shutdown},
|
||||
{"tcp", socket_tcp },
|
||||
{"udp", socket_udp },
|
||||
{NULL, NULL}
|
||||
};
|
||||
|
||||
|
|
@ -270,7 +375,9 @@ static const struct luaL_Reg socket_methods[] = {
|
|||
{"connect", socket_connect },
|
||||
{"listen", socket_listen },
|
||||
{"receive", socket_receive },
|
||||
{"receivefrom", socket_receivefrom},
|
||||
{"send", socket_send },
|
||||
{"sendto", socket_sendto },
|
||||
{NULL, NULL}
|
||||
};
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue