1 files changed, 75 insertions, 0 deletions
diff --git a/src/socket.h b/src/socket.h
new file mode 100644
index 0000000..b70e188
--- /dev/null
+++ b/src/socket.h
@@ -0,0 +1,75 @@
+#ifndef GNUTLS_SRC_SOCKET_H
+#define GNUTLS_SRC_SOCKET_H
+
+#include <gnutls/gnutls.h>
+#include <gnutls/socket.h>
+
+#define SOCKET_FLAG_UDP 1
+#define SOCKET_FLAG_FASTOPEN (1<<1)
+#define SOCKET_FLAG_STARTTLS (1<<2)
+#define SOCKET_FLAG_RAW (1<<3) /* unencrypted */
+#define SOCKET_FLAG_VERBOSE (1<<4)
+#define SOCKET_FLAG_SKIP_INIT (1<<5)
+#define SOCKET_FLAG_DONT_PRINT_ERRORS (1<<6)
+
+
+typedef struct {
+	int fd;
+	gnutls_session_t session;
+	int secure;
+	char *hostname;
+	const char *app_proto;
+	char *ip;
+	char *service;
+	struct addrinfo *ptr;
+	struct addrinfo *addr_info;
+	int verbose;
+
+	/* Needed for TCP Fast Open */
+	struct sockaddr_storage connect_addr;
+	socklen_t connect_addrlen;
+
+	FILE *server_trace;
+	FILE *client_trace;
+
+	/* resumption data */
+	gnutls_datum_t rdata;
+	/* early data */
+	gnutls_datum_t edata;
+} socket_st;
+
+/* calling program must provide that */
+extern gnutls_session_t init_tls_session(const char *host);
+int do_handshake(socket_st * socket);
+
+ssize_t socket_recv(const socket_st * socket, void *buffer,
+		    int buffer_size);
+ssize_t socket_recv_timeout(const socket_st * socket, void *buffer,
+		    int buffer_size, unsigned ms);
+ssize_t socket_send(const socket_st * socket, const void *buffer,
+		    int buffer_size);
+ssize_t socket_send_range(const socket_st * socket, const void *buffer,
+			  int buffer_size, gnutls_range_st * range);
+void
+socket_open2(socket_st * hd, const char *hostname, const char *service,
+	    const char *app_proto, int flags, const char *msg, gnutls_datum_t *rdata, gnutls_datum_t *edata,
+	    FILE *server_trace, FILE *client_trace);
+
+#define socket_open(hd, host, service, app_proto, flags, msg, rdata) \
+	socket_open2(hd, host, service, app_proto, flags, msg, rdata, NULL, NULL, NULL)
+
+#define socket_open3(hd, host, service, app_proto, flags, msg, rdata, edata)	\
+	socket_open2(hd, host, service, app_proto, flags, msg, rdata, edata, NULL, NULL)
+
+void socket_bye(socket_st * socket, unsigned polite);
+
+int service_to_port(const char *service, const char *proto);
+const char *port_to_service(const char *sport, const char *proto);
+int starttls_proto_to_port(const char *app_proto);
+const char *starttls_proto_to_service(const char *app_proto);
+
+void canonicalize_host(char *hostname, char *service, unsigned service_size);
+
+#define CONNECT_MSG "Connecting to"
+
+#endif /* GNUTLS_SRC_SOCKET_H */