666 lines
		
	
	
		
			20 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			666 lines
		
	
	
		
			20 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| #pragma once
 | |
| 
 | |
| #include <kinc/global.h>
 | |
| 
 | |
| /*! \file socket.h
 | |
|     \brief Provides low-level network-communication via UDP or TCP-sockets.
 | |
| */
 | |
| 
 | |
| #ifdef __cplusplus
 | |
| extern "C" {
 | |
| #endif
 | |
| 
 | |
| typedef enum kinc_socket_protocol { KINC_SOCKET_PROTOCOL_UDP, KINC_SOCKET_PROTOCOL_TCP } kinc_socket_protocol_t;
 | |
| 
 | |
| typedef enum kinc_socket_family { KINC_SOCKET_FAMILY_IP4, KINC_SOCKET_FAMILY_IP6 } kinc_socket_family_t;
 | |
| 
 | |
| #ifdef KINC_MICROSOFT
 | |
| #if defined(_WIN64)
 | |
| typedef unsigned __int64 UINT_PTR, *PUINT_PTR;
 | |
| #else
 | |
| #if !defined _W64
 | |
| #define _W64
 | |
| #endif
 | |
| typedef _W64 unsigned int UINT_PTR, *PUINT_PTR;
 | |
| #endif
 | |
| typedef UINT_PTR SOCKET;
 | |
| #endif
 | |
| 
 | |
| typedef struct kinc_socket {
 | |
| #ifdef KINC_MICROSOFT
 | |
| 	SOCKET handle;
 | |
| #else
 | |
| 	int handle;
 | |
| #endif
 | |
| 	uint32_t host;
 | |
| 	uint32_t port;
 | |
| 	kinc_socket_protocol_t protocol;
 | |
| 	kinc_socket_family_t family;
 | |
| 	bool connected;
 | |
| } kinc_socket_t;
 | |
| 
 | |
| typedef struct kinc_socket_options {
 | |
| 	bool non_blocking;
 | |
| 	bool broadcast;
 | |
| 	bool tcp_no_delay;
 | |
| } kinc_socket_options_t;
 | |
| 
 | |
| /// <summary>
 | |
| /// Initializes a socket-options-object to the default options
 | |
| /// </summary>
 | |
| /// <param name="options">The new default options</param>
 | |
| KINC_FUNC void kinc_socket_options_set_defaults(kinc_socket_options_t *options);
 | |
| 
 | |
| /// <summary>
 | |
| /// Initializes a socket-object. To set the host and port use kinc_socket_set.
 | |
| /// Host will be localhost
 | |
| /// Port will be 8080
 | |
| /// Family will be IPv4
 | |
| /// Protocol will be TCP
 | |
| /// </summary>
 | |
| /// <param name="socket">The socket to initialize</param>
 | |
| KINC_FUNC void kinc_socket_init(kinc_socket_t *socket);
 | |
| 
 | |
| /// <summary>
 | |
| /// Sets the sockets properties.
 | |
| /// </summary>
 | |
| /// <param name="socket">The socket-object to use</param>
 | |
| /// <param name="host">The host to use as IP or URL</param>
 | |
| /// <param name="port">The port to use</param>
 | |
| /// <param name="family">The IP-family to use</param>
 | |
| /// <param name="protocol">The protocol to use</param>
 | |
| /// <returns>Whether the socket was set correctly</returns>
 | |
| KINC_FUNC bool kinc_socket_set(kinc_socket_t *socket, const char *host, int port, kinc_socket_family_t family, kinc_socket_protocol_t protocol);
 | |
| 
 | |
| /// <summary>
 | |
| /// Destroys a socket-object.
 | |
| /// </summary>
 | |
| /// <param name="socket">The socket to destroy</param>
 | |
| KINC_FUNC void kinc_socket_destroy(kinc_socket_t *socket);
 | |
| 
 | |
| /// <summary>
 | |
| /// Opens a socket-connection.
 | |
| /// </summary>
 | |
| /// <param name="socket">The socket-object to use</param>
 | |
| /// <param name="protocol">The protocol to use</param>
 | |
| /// <param name="port">The port to use</param>
 | |
| /// <param name="options">The options to use</param>
 | |
| /// <returns>Whether the socket-connection could be opened</returns>
 | |
| KINC_FUNC bool kinc_socket_open(kinc_socket_t *socket, kinc_socket_options_t *options);
 | |
| 
 | |
| /// <summary>
 | |
| /// For use with non-blocking sockets to try to see if we are connected.
 | |
| /// </summary>
 | |
| /// <param name="socket">The socket-object to use</param>
 | |
| /// <param name="waittime">The amount of time in seconds the select function will timeout.</param>
 | |
| /// <param name="read">Check if the socket is ready to be read from.</param>
 | |
| /// <param name="write">Check if the socket is ready to be written to.</param>
 | |
| /// <returns>Whether the socket-connection can read or write or checks both.</returns>
 | |
| KINC_FUNC bool kinc_socket_select(kinc_socket_t *socket, uint32_t waittime, bool read, bool write);
 | |
| 
 | |
| /*Typically these are server actions.*/
 | |
| KINC_FUNC bool kinc_socket_bind(kinc_socket_t *socket);
 | |
| KINC_FUNC bool kinc_socket_listen(kinc_socket_t *socket, int backlog);
 | |
| KINC_FUNC bool kinc_socket_accept(kinc_socket_t *socket, kinc_socket_t *new_socket, unsigned *remote_address, unsigned *remote_port);
 | |
| 
 | |
| /*Typically this is a client action.*/
 | |
| KINC_FUNC bool kinc_socket_connect(kinc_socket_t *socket);
 | |
| 
 | |
| KINC_FUNC int kinc_socket_send(kinc_socket_t *socket, const uint8_t *data, int size);
 | |
| KINC_FUNC int kinc_socket_send_address(kinc_socket_t *socket, unsigned address, int port, const uint8_t *data, int size);
 | |
| KINC_FUNC int kinc_socket_send_url(kinc_socket_t *socket, const char *url, int port, const uint8_t *data, int size);
 | |
| KINC_FUNC int kinc_socket_receive(kinc_socket_t *socket, uint8_t *data, int maxSize, unsigned *from_address, unsigned *from_port);
 | |
| 
 | |
| /// <summary>
 | |
| /// Resolves a DNS-entry to an IP and returns its integer representation.
 | |
| /// </summary>
 | |
| /// <param name="url"></param>
 | |
| /// <param name="port"></param>
 | |
| /// <returns></returns>
 | |
| KINC_FUNC unsigned kinc_url_to_int(const char *url, int port);
 | |
| 
 | |
| #ifdef KINC_IMPLEMENTATION_NETWORK
 | |
| #define KINC_IMPLEMENTATION
 | |
| #endif
 | |
| 
 | |
| #ifdef KINC_IMPLEMENTATION
 | |
| 
 | |
| #undef KINC_IMPLEMENTATION
 | |
| #include <kinc/libs/stb_sprintf.h>
 | |
| #include <kinc/log.h>
 | |
| #define KINC_IMPLEMENTATION
 | |
| 
 | |
| #include <errno.h>
 | |
| #include <stdbool.h>
 | |
| #include <stdio.h>
 | |
| #include <string.h>
 | |
| 
 | |
| #if defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP)
 | |
| 
 | |
| // Windows 7
 | |
| #define WINVER 0x0601
 | |
| #define _WIN32_WINNT 0x0601
 | |
| 
 | |
| #define NOATOM
 | |
| #define NOCLIPBOARD
 | |
| #define NOCOLOR
 | |
| #define NOCOMM
 | |
| #define NOCTLMGR
 | |
| #define NODEFERWINDOWPOS
 | |
| #define NODRAWTEXT
 | |
| #define NOGDI
 | |
| #define NOGDICAPMASKS
 | |
| #define NOHELP
 | |
| #define NOICONS
 | |
| #define NOKANJI
 | |
| #define NOKEYSTATES
 | |
| #define NOMB
 | |
| #define NOMCX
 | |
| #define NOMEMMGR
 | |
| #define NOMENUS
 | |
| #define NOMETAFILE
 | |
| #define NOMINMAX
 | |
| #define NOMSG
 | |
| #define NONLS
 | |
| #define NOOPENFILE
 | |
| #define NOPROFILER
 | |
| #define NORASTEROPS
 | |
| #define NOSCROLL
 | |
| #define NOSERVICE
 | |
| #define NOSHOWWINDOW
 | |
| #define NOSOUND
 | |
| #define NOSYSCOMMANDS
 | |
| #define NOSYSMETRICS
 | |
| #define NOTEXTMETRIC
 | |
| #define NOUSER
 | |
| #define NOVIRTUALKEYCODES
 | |
| #define NOWH
 | |
| #define NOWINMESSAGES
 | |
| #define NOWINOFFSETS
 | |
| #define NOWINSTYLES
 | |
| #define WIN32_LEAN_AND_MEAN
 | |
| 
 | |
| #include <Ws2tcpip.h>
 | |
| #include <winsock2.h>
 | |
| #elif defined(KINC_POSIX) || defined(KINC_EMSCRIPTEN)
 | |
| #include <arpa/inet.h> // for inet_addr()
 | |
| #include <ctype.h>
 | |
| #include <fcntl.h>
 | |
| #include <netdb.h>
 | |
| #include <netinet/in.h>
 | |
| #include <netinet/tcp.h>
 | |
| #include <string.h>
 | |
| #include <sys/socket.h>
 | |
| #include <unistd.h>
 | |
| #endif
 | |
| 
 | |
| #if defined(KINC_EMSCRIPTEN)
 | |
| #include <emscripten.h>
 | |
| #include <emscripten/posix_socket.h>
 | |
| #include <emscripten/threading.h>
 | |
| #include <emscripten/websocket.h>
 | |
| 
 | |
| static EMSCRIPTEN_WEBSOCKET_T bridgeSocket = 0;
 | |
| #elif defined(KINC_POSIX)
 | |
| #include <sys/select.h>
 | |
| #endif
 | |
| 
 | |
| static int counter = 0;
 | |
| 
 | |
| #if defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP) || defined(KINC_POSIX)
 | |
| // Important: Must be cleaned with freeaddrinfo(address) later if the result is 0 in order to prevent memory leaks
 | |
| static int resolveAddress(const char *url, int port, struct addrinfo **result) {
 | |
| 	struct addrinfo hints = {0};
 | |
| 	hints.ai_family = AF_INET;
 | |
| 	hints.ai_socktype = SOCK_DGRAM;
 | |
| 	hints.ai_protocol = IPPROTO_UDP;
 | |
| 
 | |
| 	char serv[6];
 | |
| 	sprintf(serv, "%u", port);
 | |
| 
 | |
| 	return getaddrinfo(url, serv, &hints, result);
 | |
| }
 | |
| #endif
 | |
| 
 | |
| KINC_FUNC bool kinc_socket_bind(kinc_socket_t *sock) {
 | |
| #if defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP) || defined(KINC_POSIX)
 | |
| 	struct sockaddr_in address;
 | |
| 	address.sin_family = sock->family == KINC_SOCKET_FAMILY_IP4 ? AF_INET : AF_INET6;
 | |
| 	address.sin_addr.s_addr = sock->host;
 | |
| 	address.sin_port = sock->port;
 | |
| 	if (bind(sock->handle, (const struct sockaddr *)&address, sizeof(struct sockaddr_in)) < 0) {
 | |
| 		kinc_log(KINC_LOG_LEVEL_ERROR, "Could not bind socket: %s", strerror(errno));
 | |
| 		return false;
 | |
| 	}
 | |
| 	return true;
 | |
| #else
 | |
| 	return false;
 | |
| #endif
 | |
| }
 | |
| 
 | |
| KINC_FUNC void kinc_socket_options_set_defaults(kinc_socket_options_t *options) {
 | |
| 	options->non_blocking = true;
 | |
| 	options->broadcast = false;
 | |
| 	options->tcp_no_delay = false;
 | |
| }
 | |
| 
 | |
| void kinc_socket_init(kinc_socket_t *sock) {
 | |
| 	sock->handle = 0;
 | |
| 
 | |
| #if defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP) || defined(KINC_POSIX)
 | |
| 	sock->host = INADDR_ANY;
 | |
| 	sock->port = htons((unsigned short)8080);
 | |
| 	sock->protocol = KINC_SOCKET_PROTOCOL_TCP;
 | |
| 	sock->family = KINC_SOCKET_FAMILY_IP4;
 | |
| #endif
 | |
| 	sock->connected = false;
 | |
| 
 | |
| #if defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP)
 | |
| 	if (counter == 0) {
 | |
| 		WSADATA WsaData;
 | |
| 		WSAStartup(MAKEWORD(2, 2), &WsaData);
 | |
| 	}
 | |
| #if defined(KINC_EMSCRIPTEN)
 | |
| 	if (!bridgeSocket) {
 | |
| 		bridgeSocket = emscripten_init_websocket_to_posix_socket_bridge("ws://localhost:8080");
 | |
| 		// Synchronously wait until connection has been established.
 | |
| 		uint16_t readyState = 0;
 | |
| 		do {
 | |
| 			emscripten_websocket_get_ready_state(bridgeSocket, &readyState);
 | |
| 			emscripten_thread_sleep(100);
 | |
| 		} while (readyState == 0);
 | |
| 	}
 | |
| #endif
 | |
| #endif
 | |
| 	++counter;
 | |
| }
 | |
| 
 | |
| bool kinc_socket_open(kinc_socket_t *sock, struct kinc_socket_options *options) {
 | |
| #if defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP) || defined(KINC_POSIX)
 | |
| 	switch (sock->protocol) {
 | |
| 	case KINC_SOCKET_PROTOCOL_UDP:
 | |
| 		sock->handle = socket(sock->family == KINC_SOCKET_FAMILY_IP4 ? AF_INET : AF_INET6, SOCK_DGRAM, IPPROTO_UDP);
 | |
| 		break;
 | |
| 	case KINC_SOCKET_PROTOCOL_TCP:
 | |
| 		sock->handle = socket(sock->family == KINC_SOCKET_FAMILY_IP4 ? AF_INET : AF_INET6, SOCK_STREAM, IPPROTO_TCP);
 | |
| 		break;
 | |
| 	default:
 | |
| 		kinc_log(KINC_LOG_LEVEL_ERROR, "Unsupported socket protocol.");
 | |
| 		return false;
 | |
| 	}
 | |
| 
 | |
| 	if (sock->handle <= 0) {
 | |
| 		kinc_log(KINC_LOG_LEVEL_ERROR, "Could not create socket.");
 | |
| #if defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP)
 | |
| 		int errorCode = WSAGetLastError();
 | |
| 		switch (errorCode) {
 | |
| 		case (WSANOTINITIALISED):
 | |
| 			kinc_log(KINC_LOG_LEVEL_ERROR, "A successful WSAStartup call must occur before using this function.");
 | |
| 			break;
 | |
| 		case (WSAENETDOWN):
 | |
| 			kinc_log(KINC_LOG_LEVEL_ERROR, "The network subsystem or the associated service provider has failed.");
 | |
| 			break;
 | |
| 		case (WSAEAFNOSUPPORT):
 | |
| 			kinc_log(KINC_LOG_LEVEL_ERROR,
 | |
| 			         "The specified address family is not supported.For example, an application tried to create a socket for the AF_IRDA address "
 | |
| 			         "family but an infrared adapter and device driver is not installed on the local computer.");
 | |
| 			break;
 | |
| 		case (WSAEINPROGRESS):
 | |
| 			kinc_log(KINC_LOG_LEVEL_ERROR,
 | |
| 			         "A blocking Windows Sockets 1.1 call is in progress, or the service provider is still processing a callback function.");
 | |
| 			break;
 | |
| 		case (WSAEMFILE):
 | |
| 			kinc_log(KINC_LOG_LEVEL_ERROR, "No more socket descriptors are available.");
 | |
| 			break;
 | |
| 		case (WSAEINVAL):
 | |
| 			kinc_log(KINC_LOG_LEVEL_ERROR,
 | |
| 			         "An invalid argument was supplied.This error is returned if the af parameter is set to AF_UNSPEC and the type and protocol "
 | |
| 			         "parameter are unspecified.");
 | |
| 			break;
 | |
| 		case (WSAENOBUFS):
 | |
| 			kinc_log(KINC_LOG_LEVEL_ERROR, "No buffer space is available.The socket cannot be created.");
 | |
| 			break;
 | |
| 		case (WSAEPROTONOSUPPORT):
 | |
| 			kinc_log(KINC_LOG_LEVEL_ERROR, "The specified protocol is not supported.");
 | |
| 			break;
 | |
| 		case (WSAEPROTOTYPE):
 | |
| 			kinc_log(KINC_LOG_LEVEL_ERROR, "The specified protocol is the wrong type for this socket.");
 | |
| 			break;
 | |
| 		case (WSAEPROVIDERFAILEDINIT):
 | |
| 			kinc_log(KINC_LOG_LEVEL_ERROR,
 | |
| 			         "The service provider failed to initialize.This error is returned if a layered service provider(LSP) or namespace provider was "
 | |
| 			         "improperly installed or the provider fails to operate correctly.");
 | |
| 			break;
 | |
| 		case (WSAESOCKTNOSUPPORT):
 | |
| 			kinc_log(KINC_LOG_LEVEL_ERROR, "The specified socket type is not supported in this address family.");
 | |
| 			break;
 | |
| 		case (WSAEINVALIDPROVIDER):
 | |
| 			kinc_log(KINC_LOG_LEVEL_ERROR, "The service provider returned a version other than 2.2.");
 | |
| 			break;
 | |
| 		case (WSAEINVALIDPROCTABLE):
 | |
| 			kinc_log(KINC_LOG_LEVEL_ERROR, "The service provider returned an invalid or incomplete procedure table to the WSPStartup.");
 | |
| 			break;
 | |
| 		default:
 | |
| 			kinc_log(KINC_LOG_LEVEL_ERROR, "Unknown error.");
 | |
| 		}
 | |
| #elif defined(KINC_POSIX) && !defined(KINC_EMSCRIPTEN)
 | |
| 		kinc_log(KINC_LOG_LEVEL_ERROR, "%s", strerror(errno));
 | |
| #endif
 | |
| 		return false;
 | |
| 	}
 | |
| #endif
 | |
| 
 | |
| 	if (options) {
 | |
| 		if (options->non_blocking) {
 | |
| #if defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP)
 | |
| 			DWORD value = 1;
 | |
| 			if (ioctlsocket(sock->handle, FIONBIO, &value) != 0) {
 | |
| 				kinc_log(KINC_LOG_LEVEL_ERROR, "Could not set non-blocking mode.");
 | |
| 				return false;
 | |
| 			}
 | |
| #elif defined(KINC_POSIX)
 | |
| 			int value = 1;
 | |
| 			if (fcntl(sock->handle, F_SETFL, O_NONBLOCK, value) == -1) {
 | |
| 				kinc_log(KINC_LOG_LEVEL_ERROR, "Could not set non-blocking mode.");
 | |
| 				return false;
 | |
| 			}
 | |
| #endif
 | |
| 		}
 | |
| 
 | |
| 		if (options->broadcast) {
 | |
| #if defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP) || defined(KINC_POSIX)
 | |
| 			int value = 1;
 | |
| 			if (setsockopt(sock->handle, SOL_SOCKET, SO_BROADCAST, (const char *)&value, sizeof(value)) < 0) {
 | |
| 				kinc_log(KINC_LOG_LEVEL_ERROR, "Could not set broadcast mode.");
 | |
| 				return false;
 | |
| 			}
 | |
| #endif
 | |
| 		}
 | |
| 
 | |
| 		if (options->tcp_no_delay) {
 | |
| #if defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP) || defined(KINC_POSIX)
 | |
| 			int value = 1;
 | |
| 			if (setsockopt(sock->handle, IPPROTO_TCP, TCP_NODELAY, (const char *)&value, sizeof(value)) != 0) {
 | |
| 				kinc_log(KINC_LOG_LEVEL_ERROR, "Could not set no-delay mode.");
 | |
| 				return false;
 | |
| 			}
 | |
| #endif
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	return true;
 | |
| }
 | |
| 
 | |
| void kinc_socket_destroy(kinc_socket_t *sock) {
 | |
| #if defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP)
 | |
| 	closesocket(sock->handle);
 | |
| #elif defined(KINC_POSIX)
 | |
| 	close(sock->handle);
 | |
| #endif
 | |
| 
 | |
| 	memset(sock, 0, sizeof(kinc_socket_t));
 | |
| 
 | |
| 	--counter;
 | |
| #if defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP)
 | |
| 	if (counter == 0) {
 | |
| 		WSACleanup();
 | |
| 	}
 | |
| #endif
 | |
| }
 | |
| 
 | |
| bool kinc_socket_select(kinc_socket_t *sock, uint32_t waittime, bool read, bool write) {
 | |
| #if !defined(KINC_EMSCRIPTEN) && (defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP) || defined(KINC_POSIX))
 | |
| 	fd_set r_fds, w_fds;
 | |
| 	struct timeval timeout;
 | |
| 
 | |
| 	FD_ZERO(&r_fds);
 | |
| 	FD_ZERO(&w_fds);
 | |
| 
 | |
| 	FD_SET(sock->handle, &r_fds);
 | |
| 	FD_SET(sock->handle, &w_fds);
 | |
| 
 | |
| 	timeout.tv_sec = waittime;
 | |
| 	timeout.tv_usec = 0;
 | |
| 
 | |
| 	if (select(0, &r_fds, &w_fds, NULL, &timeout) < 0) {
 | |
| 		kinc_log(KINC_LOG_LEVEL_ERROR, "kinc_socket_select didn't work: %s", strerror(errno));
 | |
| 		return false;
 | |
| 	}
 | |
| 
 | |
| 	if (read && write) {
 | |
| 		return FD_ISSET(sock->handle, &w_fds) && FD_ISSET(sock->handle, &r_fds);
 | |
| 	}
 | |
| 	else if (read) {
 | |
| 		return FD_ISSET(sock->handle, &r_fds);
 | |
| 	}
 | |
| 	else if (write) {
 | |
| 		return FD_ISSET(sock->handle, &w_fds);
 | |
| 	}
 | |
| 	else {
 | |
| 		kinc_log(KINC_LOG_LEVEL_ERROR, "Calling kinc_socket_select with both read and write set to false is useless.");
 | |
| 		return false;
 | |
| 	}
 | |
| #else
 | |
| 	return false;
 | |
| #endif
 | |
| }
 | |
| 
 | |
| bool kinc_socket_set(kinc_socket_t *sock, const char *host, int port, kinc_socket_family_t family, kinc_socket_protocol_t protocol) {
 | |
| #if defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP) || defined(KINC_POSIX)
 | |
| 
 | |
| 	sock->family = family;
 | |
| 	sock->protocol = protocol;
 | |
| 	sock->port = htons((unsigned short)port);
 | |
| 
 | |
| 	if (host == NULL)
 | |
| 		return true;
 | |
| 
 | |
| 	if (isdigit(host[0]) || (family == KINC_SOCKET_FAMILY_IP6 && host[4] == ':')) { // Is IPv4 or IPv6 string
 | |
| 		struct in_addr addr;
 | |
| 
 | |
| 		if (inet_pton(sock->family == KINC_SOCKET_FAMILY_IP4 ? AF_INET : AF_INET6, host, &addr) == 0) {
 | |
| 			kinc_log(KINC_LOG_LEVEL_ERROR, "Invalid %s address: %s\n", sock->family == KINC_SOCKET_FAMILY_IP4 ? "IPv4" : "IPv6", host);
 | |
| 			return false;
 | |
| 		}
 | |
| 		sock->host = addr.s_addr;
 | |
| 		return true;
 | |
| 	}
 | |
| 	else {
 | |
| 		struct addrinfo *address = NULL;
 | |
| 		int res = resolveAddress(host, port, &address);
 | |
| 		if (res != 0) {
 | |
| 			kinc_log(KINC_LOG_LEVEL_ERROR, "Could not resolve address.");
 | |
| 			return false;
 | |
| 		}
 | |
| #if defined(KINC_POSIX)
 | |
| 		sock->host = ((struct sockaddr_in *)address->ai_addr)->sin_addr.s_addr;
 | |
| #else
 | |
| 		sock->host = ((struct sockaddr_in *)address->ai_addr)->sin_addr.S_un.S_addr;
 | |
| #endif
 | |
| 		freeaddrinfo(address);
 | |
| 
 | |
| 		return true;
 | |
| 	}
 | |
| #else
 | |
| 	return false;
 | |
| #endif
 | |
| }
 | |
| 
 | |
| bool kinc_socket_listen(kinc_socket_t *socket, int backlog) {
 | |
| #if defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP) || defined(KINC_POSIX)
 | |
| 	int res = listen(socket->handle, backlog);
 | |
| 	return (res == 0);
 | |
| #else
 | |
| 	return false;
 | |
| #endif
 | |
| }
 | |
| 
 | |
| bool kinc_socket_accept(kinc_socket_t *sock, kinc_socket_t *newSocket, unsigned *remoteAddress, unsigned *remotePort) {
 | |
| #if defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP)
 | |
| 	typedef int socklen_t;
 | |
| #endif
 | |
| #if defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP) || defined(KINC_POSIX)
 | |
| 	struct sockaddr_in addr;
 | |
| 	socklen_t addrLength = sizeof(addr);
 | |
| 	newSocket->handle = accept(sock->handle, (struct sockaddr *)&addr, &addrLength);
 | |
| 	if (newSocket->handle <= 0) {
 | |
| 		return false;
 | |
| 	}
 | |
| 
 | |
| 	newSocket->connected = sock->connected = true;
 | |
| 	newSocket->host = addr.sin_addr.s_addr;
 | |
| 	newSocket->port = addr.sin_port;
 | |
| 	newSocket->family = sock->family;
 | |
| 	newSocket->protocol = sock->protocol;
 | |
| 	*remoteAddress = ntohl(addr.sin_addr.s_addr);
 | |
| 	*remotePort = ntohs(addr.sin_port);
 | |
| 	return true;
 | |
| #else
 | |
| 	return false;
 | |
| #endif
 | |
| }
 | |
| 
 | |
| bool kinc_socket_connect(kinc_socket_t *sock) {
 | |
| #if defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP) || defined(KINC_POSIX)
 | |
| 	struct sockaddr_in addr;
 | |
| 	addr.sin_family = sock->family == KINC_SOCKET_FAMILY_IP4 ? AF_INET : AF_INET6;
 | |
| 	addr.sin_addr.s_addr = sock->host;
 | |
| 	addr.sin_port = sock->port;
 | |
| 
 | |
| 	int res = connect(sock->handle, (struct sockaddr *)&addr, sizeof(struct sockaddr_in));
 | |
| 	return sock->connected = (res == 0);
 | |
| #else
 | |
| 	return false;
 | |
| #endif
 | |
| }
 | |
| 
 | |
| int kinc_socket_send(kinc_socket_t *sock, const uint8_t *data, int size) {
 | |
| #if defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP)
 | |
| 	typedef int socklen_t;
 | |
| #endif
 | |
| #if defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP) || defined(KINC_POSIX)
 | |
| 	if (sock->protocol == KINC_SOCKET_PROTOCOL_UDP) {
 | |
| 		struct sockaddr_in addr;
 | |
| 		addr.sin_family = sock->family == KINC_SOCKET_FAMILY_IP4 ? AF_INET : AF_INET6;
 | |
| 		addr.sin_addr.s_addr = sock->host;
 | |
| 		addr.sin_port = sock->port;
 | |
| 
 | |
| 		size_t sent = sendto(sock->handle, (const char *)data, size, 0, (struct sockaddr *)&addr, sizeof(struct sockaddr_in));
 | |
| 		if (sent != size) {
 | |
| 			kinc_log(KINC_LOG_LEVEL_ERROR, "Could not send packet.");
 | |
| 			return -1;
 | |
| 		}
 | |
| 		return (int)sent;
 | |
| 	}
 | |
| 	else {
 | |
| 		if (!sock->connected) {
 | |
| 			kinc_log(KINC_LOG_LEVEL_ERROR, "Call kinc_sockect_connect/bind before send/recv can be called for TCP sockets.");
 | |
| 			return -1;
 | |
| 		}
 | |
| 
 | |
| 		size_t sent = send(sock->handle, (const char *)data, size, 0);
 | |
| 		if (sent != size) {
 | |
| 			kinc_log(KINC_LOG_LEVEL_ERROR, "Could not send packet.");
 | |
| 		}
 | |
| 		return (int)sent;
 | |
| 	}
 | |
| #else
 | |
| 	return 0;
 | |
| #endif
 | |
| }
 | |
| 
 | |
| int kinc_socket_send_address(kinc_socket_t *sock, unsigned address, int port, const uint8_t *data, int size) {
 | |
| #if defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP) || defined(KINC_POSIX)
 | |
| 	struct sockaddr_in addr;
 | |
| 	addr.sin_family = AF_INET;
 | |
| 	addr.sin_addr.s_addr = htonl(address);
 | |
| 	addr.sin_port = htons(port);
 | |
| 
 | |
| 	size_t sent = sendto(sock->handle, (const char *)data, size, 0, (struct sockaddr *)&addr, sizeof(struct sockaddr_in));
 | |
| 	if (sent != size) {
 | |
| 		kinc_log(KINC_LOG_LEVEL_ERROR, "Could not send packet.");
 | |
| 	}
 | |
| 	return (int)sent;
 | |
| #else
 | |
| 	return 0;
 | |
| #endif
 | |
| }
 | |
| 
 | |
| int kinc_socket_send_url(kinc_socket_t *sock, const char *url, int port, const uint8_t *data, int size) {
 | |
| #if defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP) || defined(KINC_POSIX)
 | |
| 	struct addrinfo *address = NULL;
 | |
| 	int res = resolveAddress(url, port, &address);
 | |
| 	if (res != 0) {
 | |
| 		kinc_log(KINC_LOG_LEVEL_ERROR, "Could not resolve address.");
 | |
| 		return 0;
 | |
| 	}
 | |
| 
 | |
| 	size_t sent = sendto(sock->handle, (const char *)data, size, 0, address->ai_addr, sizeof(struct sockaddr_in));
 | |
| 	if (sent != size) {
 | |
| 		kinc_log(KINC_LOG_LEVEL_ERROR, "Could not send packet.");
 | |
| 	}
 | |
| 	freeaddrinfo(address);
 | |
| 	return (int)sent;
 | |
| #else
 | |
| 	return 0;
 | |
| #endif
 | |
| }
 | |
| 
 | |
| int kinc_socket_receive(kinc_socket_t *sock, uint8_t *data, int maxSize, unsigned *fromAddress, unsigned *fromPort) {
 | |
| #if defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP)
 | |
| 	typedef int socklen_t;
 | |
| 	typedef int ssize_t;
 | |
| #endif
 | |
| #if defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP) || defined(KINC_POSIX)
 | |
| 
 | |
| 	if (sock->protocol == KINC_SOCKET_PROTOCOL_UDP) {
 | |
| 		struct sockaddr_in from;
 | |
| 		socklen_t fromLength = sizeof(from);
 | |
| 		ssize_t bytes = recvfrom(sock->handle, (char *)data, maxSize, 0, (struct sockaddr *)&from, &fromLength);
 | |
| 		if (bytes <= 0) {
 | |
| 			return (int)bytes;
 | |
| 		}
 | |
| 		*fromAddress = ntohl(from.sin_addr.s_addr);
 | |
| 		*fromPort = ntohs(from.sin_port);
 | |
| 		return (int)bytes;
 | |
| 	}
 | |
| 	else {
 | |
| 
 | |
| 		if (!sock->connected) {
 | |
| 			kinc_log(KINC_LOG_LEVEL_ERROR, "Call kinc_sockect_connect/bind before send/recv can be called for TCP sockets.");
 | |
| 			return -1;
 | |
| 		}
 | |
| 		ssize_t bytes = recv(sock->handle, (char *)data, maxSize, 0);
 | |
| 		*fromAddress = ntohl(sock->host);
 | |
| 		*fromPort = ntohs(sock->port);
 | |
| 		return (int)bytes;
 | |
| 	}
 | |
| #else
 | |
| 	return 0;
 | |
| #endif
 | |
| }
 | |
| 
 | |
| unsigned kinc_url_to_int(const char *url, int port) {
 | |
| #if defined(KINC_WINDOWS) || defined(KINC_WINDOWSAPP)
 | |
| 	struct addrinfo *address = NULL;
 | |
| 	int res = resolveAddress(url, port, &address);
 | |
| 	if (res != 0) {
 | |
| 		kinc_log(KINC_LOG_LEVEL_ERROR, "Could not resolve address.");
 | |
| 		return -1;
 | |
| 	}
 | |
| 
 | |
| 	unsigned fromAddress = ntohl(((struct sockaddr_in *)address->ai_addr)->sin_addr.S_un.S_addr);
 | |
| 	freeaddrinfo(address);
 | |
| 
 | |
| 	return fromAddress;
 | |
| #else
 | |
| 	return 0;
 | |
| #endif
 | |
| }
 | |
| 
 | |
| #endif
 | |
| 
 | |
| #ifdef __cplusplus
 | |
| }
 | |
| #endif
 |