--- libaitio/inc/aitio.h 2013/03/13 14:54:39 1.29 +++ libaitio/inc/aitio.h 2013/11/22 13:49:14 1.37 @@ -3,7 +3,7 @@ * by Michael Pounov * * $Author: misho $ -* $Id: aitio.h,v 1.29 2013/03/13 14:54:39 misho Exp $ +* $Id: aitio.h,v 1.37 2013/11/22 13:49:14 misho Exp $ * ************************************************************************** The ELWIX and AITNET software is distributed under the following @@ -59,21 +59,132 @@ SUCH DAMAGE. #include #include #include +#include #include +#include +#include #ifndef STRSIZ -#define STRSIZ 256 +#define STRSIZ 256 #endif +#define IO_SOCK_ROLE_CLIENT 0 +#define IO_SOCK_ROLE_SERVER 1 +#define IO_ETHER_FILTER_PROMISC 0 +#define IO_ETHER_FILTER_NOTREAD -1 +#define IO_ETHER_FILTER_READ 1 +#define IO_ETHER_FILTER_WRITE 2 + + +typedef struct tagCliSock sock_cli_t; +typedef void *(*sock_cb_t)(sock_cli_t*); +struct tagCliSock { + void *cli_parent; + int cli_fd; + int cli_pty; + sockaddr_t cli_addr; + char cli_name[64]; + char cli_cmdline[PATH_MAX]; + pid_t cli_pid; + + sched_task_func_t cli_func; + + ait_val_t cli_buf[2]; + + TAILQ_ENTRY(tagCliSock) cli_node; +}; + +typedef struct { + int sock_role; + int sock_backq; + int sock_type; + int sock_proto; + int sock_fd; + struct timespec sock_timeout; + sockaddr_t sock_addr; + sockaddr_t sock_peer; + + ait_val_t sock_buf; + + volatile intptr_t sock_kill; + sched_root_task_t *sock_root; + + pthread_mutex_t sock_mtx; + TAILQ_HEAD(, tagCliSock) sock_cli; +} sock_t; + + // io_GetErrno() Get error code of last operation -inline int io_GetErrno(); +int io_GetErrno(); // io_GetError() Get error text of last operation -inline const char *io_GetError(); +const char *io_GetError(); /* + * ioInitSocket() - Init socket and allocate resources + * + * @role = Socket role + * @type = Socket type + * @proto = Socket protocol + * @addr = Bind to address + * @port = Bind to port + * @buflen = Socket buffer, optional if =0 == BUFSIZ + * return: NULL error or !=NULL created socket + */ +sock_t *ioInitSocket(int role, int type, int proto, + const char *addr, unsigned short port, size_t buflen); +/* + * ioCloseSocket() - Close socket and free resources + * + * @s = Socket + * return: none + */ +void ioCloseSocket(sock_t ** __restrict s); +#define ioKillSocket(x) (assert((x)), (x)->sock_kill = 1) +/* + * ioCloseClient() - Close client socket + * + * @c = Client socket + * return: 0 ok or !=0 error + */ +int ioCloseClient(sock_cli_t * __restrict c); +/* + * ioUpSocket() - Setup socket for use + * + * @s = Socket + * @arg = Server role = listen backlog queue and Client role = peer address + * @timeout = Socket timeout in ms (default -1 infinit) + * return: -1 error or 0 ok + */ +int ioUpSocket(sock_t * __restrict s, void *arg, int timeout); +/* + * ioUpdTimerSocket() - Update timeout of socket + * + * @c = Client socket + * @arg = Optional data argument + * return: none + */ +void ioUpdTimerSocket(sock_cli_t * __restrict c, void *arg); +/* + * ioLoopSocket() - Start socket scheduler + * + * @s = Socket + * @rcb = Read callback + * return: -1 error or return result from scheduler + */ +int ioLoopSocket(sock_t * __restrict s, sched_task_func_t rcb); +/* + * ioBridgeProg2Socket() - Start socket scheduler and bridge program to socket + * + * @s = Socket + * @prgname = Program name + * return: 0 ok or !=0 error + */ +int ioBridgeProg2Socket(sock_t * __restrict s, const char *prgname); + +/* * ioPromptRead() - Read data from input h[0] with prompt to output h[1] * * @h = file handles h[0] = input, h[1] = output, if NULL use stdin, stdout @@ -82,7 +193,8 @@ inline const char *io_GetError(); * @dataLen = Length of data * return: 0 EOF; -1 error:: can`t read; >0 count of readed chars */ -int ioPromptRead(int *h, const char *csPrompt, char * __restrict psData, int dataLen); +int ioPromptRead(int *h, const char *csPrompt, + char * __restrict psData, int dataLen); /* * ioPromptPassword() - Read password from input h[0] with prompt to output h[1] * @@ -93,7 +205,8 @@ int ioPromptRead(int *h, const char *csPrompt, char * * @confirm = Confirm password, 0 - get password, !=0 Ask for confirmation * return: 0 EOF; -1 error:: can`t read; >0 count of readed chars */ -int ioPromptPassword(int *h, const char *csPrompt, char * __restrict psPass, int passLen, int confirm); +int ioPromptPassword(int *h, const char *csPrompt, + char * __restrict psPass, int passLen, int confirm); /* @@ -125,7 +238,7 @@ int ioWatchDirLoop(const char *csDir, int (*callback)( * @sig = Event for completed operations, may be =NULL * return: -1 error or 0 ok */ -inline int io_aiobulk(int mode, struct aiocb ** __restrict acbs, int nacb, +int io_aiobulk(int mode, struct aiocb ** __restrict acbs, int nacb, struct sigevent *sig); #endif /* @@ -162,7 +275,7 @@ int io_rwritev(int fd, struct iovec * __restrict bufs, * @update = Update file handle position !0 * return: -1 error or !=-1 readed bytes */ -inline int io_rread(int fd, void * __restrict buf, size_t nbytes, off_t offset, +int io_rread(int fd, void * __restrict buf, size_t nbytes, off_t offset, int update); /* * io_rwrite() - Raw VFS write function @@ -174,7 +287,7 @@ inline int io_rread(int fd, void * __restrict buf, siz * @update = Update file handle position !0 * return: -1 error or !=-1 written bytes */ -inline int io_rwrite(int fd, void * __restrict buf, size_t nbytes, off_t offset, +int io_rwrite(int fd, void * __restrict buf, size_t nbytes, off_t offset, int update); /* Disk I/O helper macros */ @@ -237,7 +350,7 @@ int io_ctr_AES(unsigned char *pInput, int inLen, unsig * @winz = winsize for terminal * return: -1 error or 0 ok */ -inline int ioAllocPTY(int *ptyfd, int *ttyfd, char * __restrict name, int namesiz, +int ioAllocPTY(int *ptyfd, int *ttyfd, char * __restrict name, int namesiz, struct termios * __restrict term, struct winsize * __restrict winz); /* * ioFreePTY() - Release PTY and TTY device @@ -246,7 +359,7 @@ inline int ioAllocPTY(int *ptyfd, int *ttyfd, char * _ * @ttyname = tty filename * return: none */ -inline void ioFreePTY(int ptyfd, const char *ttyname); +void ioFreePTY(int ptyfd, const char *ttyname); /* * ioChgWinPTY() - Change window size of PTY * @@ -257,7 +370,7 @@ inline void ioFreePTY(int ptyfd, const char *ttyname); * @ypxl = y pixels * return: -1 error or 0 ok */ -inline int ioChgWinPTY(int ptyfd, unsigned short row, unsigned short col, +int ioChgWinPTY(int ptyfd, unsigned short row, unsigned short col, unsigned short xpxl, unsigned short ypxl); /* * ioSetOwnerTTY() - Set owner to TTY @@ -283,7 +396,7 @@ int ioSetSidTTY(int *ttyfd, const char *ttyname); * @otio = saved old termios for later restore if !=NULL * return: -1 error or 0 ok */ -inline int ioSetRAWMode(int fd, struct termios *otio); +int ioSetRAWMode(int fd, struct termios *otio); /* * ioRestoreMode() - Restore termios to tty fd * @@ -291,7 +404,7 @@ inline int ioSetRAWMode(int fd, struct termios *otio); * @tio = termios structure for restore * return: -1 error or 0 ok */ -inline int ioRestoreMode(int fd, struct termios tio); +int ioRestoreMode(int fd, struct termios tio); /* * ioForkPTY() - Fork new process with session leader and new TTY * @@ -313,7 +426,7 @@ pid_t ioForkPTY(int *ptyfd, char * __restrict name, in * @ifExists = !=0 if filename exists return error * return: -1 error or 0 ok */ -inline int ioCreatePIDFile(const char *csName, int ifExists); +int ioCreatePIDFile(const char *csName, int ifExists); /* * ioSendFile() - AITNET sendfile() userland implementation, not dependant from OS @@ -338,6 +451,13 @@ size_t ioSendFile(int s, const char *csFile, size_t se */ size_t ioRecvFile(int s, const char *csFile, size_t recvLen, int over, int rcvbuf); +/* + * ioRealFileName() - Get real file name + * + * @fname = filename + * return: =NULL error or !=NULL real filename, should be free with e_free() + */ +char *ioRealFileName(const char *fname); /* Buffered file access over memory block */ @@ -368,7 +488,7 @@ FILE *io_fmapopen(const char *csFile, int mode, int pe * @mode = Permissions for new buffered file I/O * return: NULL error or open buffered file */ -inline FILE *io_fd2buf(int fd, const char *mode); +FILE *io_fd2buf(int fd, const char *mode); /* * io_dumbFile() - Create empry or dumb file with fixed size * @@ -378,6 +498,70 @@ inline FILE *io_fd2buf(int fd, const char *mode); * return: -1 error or open file handle */ int io_dumbFile(const char *csFile, int mode, off_t size); + + +/* + * io_get1stiface() - Get first interface of host + * + * @szIface = interface string buffer + * @iflen = size of interface buffer + * return: -1 error or 0 ok + */ +int io_get1stiface(char *szIface, int iflen); +/* + * io_etherOpen() - Open BPF interface to device + * + * @csIface = interface name + * @flags = open flags + * @whdr = with complete headers + * @wdlt = with data link type + * @buflen = buffer length + * @zcbuf = zero copy buffer, if BPF supports it and isn't NULL + * return: -1 error or >-1 bpf handle + */ +int io_etherOpen(const char *csIface, int flags, int whdr, int wdlt, + unsigned int *buflen, void **zcbuf); +/* + * io_etherClose() - Close BPF interface + * + * @eth = bpf handle + * @zcbuf = zero copy buffer, if BPF supports it and isn't NULL + * return: none + */ +void io_etherClose(int eth, void **zcbuf); + +/* + * io_etherFilter() - BPF filter routine + * + * @eth = bpf handle + * @io = filter direction + * (IO_ETHER_FILTER_PROMISC|IO_ETHER_FILTER_NOTREAD|IO_ETHER_FILTER_READ|IO_ETHER_FILTER_WRITE) + * @insn = BPF filter instruction array + * @insnlen = Length of BPF filter instruction array + * return: -1 error or 0 ok + */ +int io_etherFilter(int eth, int io, struct bpf_insn * __restrict insn, size_t insnlen); + +/* + * io_etherSend() - Send packet to bpf + * + * @eth = bpf handle + * @buf = buffer + * @buflen = buffer length + * return: -1 error or !=-1 written bytes + */ +ssize_t io_etherSend(int eth, const void *buf, size_t buflen); +/* + * io_etherRecv() - Receive packet from bpf + * + * @eth = bpf handle + * @buf = buffer + * @buflen = buffer length + * @zcbuf = zero copy buffer, if BPF supports it and isn't NULL + * return: -1 error or !=-1 readed bytes + */ +ssize_t io_etherRecv(int eth, void * __restrict buf, + size_t buflen, void * __restrict zcbuf); #endif