1 #ifndef DROPBEAR_NETIO_H 2 #define DROPBEAR_NETIO_H 3 4 #include "includes.h" 5 #include "buffer.h" 6 #include "queue.h" 7 8 enum dropbear_prio { 9 DROPBEAR_PRIO_DEFAULT = 10, 10 DROPBEAR_PRIO_LOWDELAY = 11, 11 DROPBEAR_PRIO_BULK = 12, 12 }; 13 14 void set_sock_nodelay(int sock); 15 void set_sock_priority(int sock, enum dropbear_prio prio); 16 17 int get_sock_port(int sock); 18 void get_socket_address(int fd, char **local_host, char **local_port, 19 char **remote_host, char **remote_port, int host_lookup); 20 void getaddrstring(struct sockaddr_storage* addr, 21 char **ret_host, char **ret_port, int host_lookup); 22 int dropbear_listen(const char* address, const char* port, 23 int *socks, unsigned int sockcount, char **errstring, int *maxfd); 24 25 struct dropbear_progress_connection; 26 27 /* result is DROPBEAR_SUCCESS or DROPBEAR_FAILURE. 28 errstring is only set on DROPBEAR_FAILURE, returns failure message for the last attempted socket */ 29 typedef void(*connect_callback)(int result, int sock, void* data, const char* errstring); 30 31 /* Always returns a progress connection, if it fails it will call the callback at a later point */ 32 struct dropbear_progress_connection * connect_remote (const char* remotehost, const char* remoteport, 33 connect_callback cb, void *cb_data, const char* bind_address, const char* bind_port); 34 35 /* Sets up for select() */ 36 void set_connect_fds(fd_set *writefd); 37 /* Handles ready sockets after select() */ 38 void handle_connect_fds(const fd_set *writefd); 39 /* Cleanup */ 40 void remove_connect_pending(void); 41 42 /* Doesn't actually stop the connect, but adds a dummy callback instead */ 43 void cancel_connect(struct dropbear_progress_connection *c); 44 45 void connect_set_writequeue(struct dropbear_progress_connection *c, struct Queue *writequeue); 46 47 /* TODO: writev #ifdef guard */ 48 /* Fills out iov which contains iov_count slots, returning the number filled in iov_count */ 49 void packet_queue_to_iovec(const struct Queue *queue, struct iovec *iov, unsigned int *iov_count); 50 void packet_queue_consume(struct Queue *queue, ssize_t written); 51 52 #if DROPBEAR_SERVER_TCP_FAST_OPEN 53 /* Try for any Linux builds, will fall back if the kernel doesn't support it */ 54 void set_listen_fast_open(int sock); 55 /* Define values which may be supported by the kernel even if the libc is too old */ 56 #ifndef TCP_FASTOPEN 57 #define TCP_FASTOPEN 23 58 #endif 59 #ifndef MSG_FASTOPEN 60 #define MSG_FASTOPEN 0x20000000 61 #endif 62 #endif 63 64 #endif 65 66