mirror of https://github.com/CPunch/Laika.git
125 lines
4.3 KiB
C
125 lines
4.3 KiB
C
#ifndef LAIKA_SOCKET_H
|
|
#define LAIKA_SOCKET_H
|
|
|
|
/* clang-format will change the order of the included windows headers, this BREAKS THINGS.
|
|
for now, make clang ignore this section */
|
|
|
|
/* clang-format off */
|
|
|
|
/* socket/winsock headers */
|
|
#ifdef _WIN32
|
|
/* windows */
|
|
# ifndef NOMINMAX
|
|
# define NOMINMAX
|
|
# endif
|
|
# define _WINSOCK_DEPRECATED_NO_WARNINGS
|
|
# include <winsock2.h>
|
|
# include <windows.h>
|
|
# include <ws2tcpip.h>
|
|
# pragma comment(lib, "Ws2_32.lib")
|
|
|
|
typedef char buffer_t;
|
|
# define PollFD WSAPOLLFD
|
|
# define poll WSAPoll
|
|
# define LN_ERRNO WSAGetLastError()
|
|
# define LN_EWOULD WSAEWOULDBLOCK
|
|
# define LN_MSG_NOSIGNAL 0
|
|
# define SOCKETINVALID(x) (x == INVALID_SOCKET)
|
|
# define SOCKETERROR(x) (x == SOCKET_ERROR)
|
|
#else
|
|
/* posix platform */
|
|
# include <arpa/inet.h>
|
|
# include <netdb.h>
|
|
# include <netinet/in.h>
|
|
# include <poll.h>
|
|
# include <sys/socket.h>
|
|
# include <sys/types.h>
|
|
# ifdef __linux__
|
|
# include <sys/epoll.h>
|
|
/* max events for epoll() */
|
|
# define MAX_EPOLL_EVENTS 128
|
|
# define LAIKA_USE_EPOLL
|
|
# endif
|
|
# include <errno.h>
|
|
# include <unistd.h>
|
|
|
|
typedef int SOCKET;
|
|
typedef void buffer_t;
|
|
# define PollFD struct pollfd
|
|
# define LN_ERRNO errno
|
|
# define LN_EWOULD EWOULDBLOCK
|
|
# define LN_MSG_NOSIGNAL MSG_NOSIGNAL
|
|
# define INVALID_SOCKET -1
|
|
# define SOCKETINVALID(x) (x < 0)
|
|
# define SOCKETERROR(x) (x == -1)
|
|
#endif
|
|
#include "laika.h"
|
|
#include "lsodium.h"
|
|
#include "lmem.h"
|
|
|
|
#include <fcntl.h>
|
|
#include <stdbool.h>
|
|
|
|
/* clang-format on */
|
|
|
|
typedef enum
|
|
{
|
|
RAWSOCK_OK,
|
|
RAWSOCK_ERROR,
|
|
RAWSOCK_CLOSED,
|
|
RAWSOCK_POLL
|
|
} RAWSOCKCODE;
|
|
|
|
typedef bool (*pollEvent)(struct sLaika_socket *sock);
|
|
typedef void (*pollFailEvent)(struct sLaika_socket *sock, void *uData);
|
|
|
|
struct sLaika_socket
|
|
{
|
|
SOCKET sock; /* raw socket fd */
|
|
pollFailEvent onPollFail;
|
|
pollEvent onPollIn;
|
|
pollEvent onPollOut;
|
|
void *uData; /* passed to onPollFail */
|
|
laikaM_newVector(uint8_t, outBuf); /* raw data to be sent() */
|
|
laikaM_newVector(uint8_t, inBuf); /* raw data we recv()'d */
|
|
bool flipEndian;
|
|
bool setPollOut; /* is EPOLLOUT/POLLOUT is set on sock's pollfd ? */
|
|
};
|
|
|
|
#define laikaS_isAlive(arg) (arg->sock != INVALID_SOCKET)
|
|
|
|
bool laikaS_isBigEndian(void);
|
|
|
|
void laikaS_init(void);
|
|
void laikaS_cleanUp(void);
|
|
|
|
void laikaS_initSocket(struct sLaika_socket *sock, pollEvent onPollIn, pollEvent onPollOut,
|
|
pollFailEvent onPollFail, void *uData);
|
|
void laikaS_cleanSocket(struct sLaika_socket *sock);
|
|
void laikaS_kill(struct sLaika_socket *sock); /* kills a socket */
|
|
void laikaS_connect(struct sLaika_socket *sock, char *ip, char *port); /* connect to ip & port */
|
|
void laikaS_bind(struct sLaika_socket *sock, uint16_t port); /* bind sock to port */
|
|
void laikaS_acceptFrom(struct sLaika_socket *sock, struct sLaika_socket *from, char *ipStr);
|
|
bool laikaS_setNonBlock(struct sLaika_socket *sock);
|
|
|
|
void laikaS_consumeRead(struct sLaika_socket *sock,
|
|
size_t sz); /* throws sz bytes away from the inBuf */
|
|
void laikaS_zeroWrite(struct sLaika_socket *sock,
|
|
size_t sz); /* writes sz NULL bytes to the outBuf */
|
|
void laikaS_read(struct sLaika_socket *sock, void *buf, size_t sz); /* reads from inBuf */
|
|
void laikaS_write(struct sLaika_socket *sock, void *buf, size_t sz); /* writes to outBuf */
|
|
void laikaS_writeKeyEncrypt(struct sLaika_socket *sock, void *buf, size_t sz,
|
|
uint8_t *pub); /* encrypts & writes from buf using pub key */
|
|
void laikaS_readKeyDecrypt(struct sLaika_socket *sock, void *buf, size_t sz, uint8_t *pub,
|
|
uint8_t *priv); /* decrypts & reads to buf using pub & priv key*/
|
|
void laikaS_writeByte(struct sLaika_socket *sock, uint8_t data);
|
|
uint8_t laikaS_readByte(struct sLaika_socket *sock);
|
|
void laikaS_readInt(struct sLaika_socket *sock, void *buf,
|
|
size_t sz); /* reads INT, respecting endianness */
|
|
void laikaS_writeInt(struct sLaika_socket *sock, void *buf,
|
|
size_t sz); /* writes INT, respecting endianness */
|
|
|
|
RAWSOCKCODE laikaS_rawRecv(struct sLaika_socket *sock, size_t sz, int *processed);
|
|
RAWSOCKCODE laikaS_rawSend(struct sLaika_socket *sock, size_t sz, int *processed);
|
|
|
|
#endif |