pantry/projects/crates.io/rustls-ffi/common.h
Marc Seitz 5461125a78
+crates.io/rustls-ffi (#87)
* +crates.io/rustls-ffi

* add build step for cbindgen

* add make dependency

* implement client test

Co-authored-by: Jacob Heider <jacob@tea.xyz>
2023-01-14 19:20:35 -05:00

129 lines
3.5 KiB
C

#ifndef COMMON_H
#define COMMON_H
#ifdef _WIN32
#define sleep(s) Sleep(1000 * (s))
#define read(s, buf, n) recv(s, buf, n, 0)
#define close(s) closesocket(s)
#define bzero(buf, n) memset(buf, '\0', n)
/* Hacks for 'errno' stuff
*/
#undef EAGAIN
#define EAGAIN WSAEWOULDBLOCK
#undef EWOULDBLOCK
#define EWOULDBLOCK WSAEWOULDBLOCK
#undef errno
#define errno WSAGetLastError()
#define perror(str) fprintf(stderr, str ": %d.\n", WSAGetLastError())
const char * ws_strerror(int err);
#define strerror(e) ws_strerror(e)
#ifndef STDOUT_FILENO
#define STDOUT_FILENO 1 /* MinGW has this */
#endif /* !STDOUT_FILENO */
#endif /* _WIN32 */
enum crustls_demo_result
{
CRUSTLS_DEMO_OK,
CRUSTLS_DEMO_ERROR,
CRUSTLS_DEMO_AGAIN,
CRUSTLS_DEMO_EOF,
};
/* A growable vector of bytes. */
struct bytevec {
char *data;
size_t len;
size_t capacity;
};
struct conndata {
int fd;
const char *verify_arg;
const char *program_name;
struct bytevec data;
struct rustls_connection *rconn;
};
void
print_error(const char *program_name, const char *prefix, rustls_result result);
int
write_all(int fd, const char *buf, int n);
/* Make a socket nonblocking. */
enum crustls_demo_result
nonblock(int sockfd);
/* A callback that reads bytes from the network. */
int
read_cb(void *userdata, uint8_t *buf, uintptr_t len, uintptr_t *out_n);
/* Invoke rustls_connection_write_tls with either a vectored or unvectored
callback, depending on environment variable. */
rustls_io_result
write_tls(struct rustls_connection *rconn, struct conndata *conn, size_t *n);
/* A callback that writes bytes to the network. */
int
write_cb(void *userdata, const uint8_t *buf, uintptr_t len, uintptr_t *out_n);
#ifndef _WIN32
rustls_io_result write_vectored_cb(
void *userdata, const struct rustls_iovec *iov, size_t count, size_t *out_n);
#endif /* _WIN32 */
/* Number of bytes available for writing. */
size_t
bytevec_available(struct bytevec *vec);
/* Pointer to the writeable region. */
char *
bytevec_writeable(struct bytevec *vec);
/* Indicate that n bytes have been written, increasing len. */
void
bytevec_consume(struct bytevec *vec, size_t n);
/* Ensure there are at least n bytes available between vec->len and
* vec->capacity. If this requires reallocating, this may return
* CRUSTLS_DEMO_ERROR. */
enum crustls_demo_result
bytevec_ensure_available(struct bytevec *vec, size_t n);
/* Read all available bytes from the rustls_connection until EOF.
* Note that EOF here indicates "no more bytes until
* process_new_packets", not "stream is closed".
*
* Returns CRUSTLS_DEMO_OK for success,
* CRUSTLS_DEMO_ERROR for error,
* CRUSTLS_DEMO_EOF for "connection cleanly terminated by peer"
*/
int
copy_plaintext_to_buffer(struct conndata *conn);
/* Polyfill */
void *memmem(const void *haystack, size_t haystacklen, const void *needle, size_t needlelen);
/* If headers are done (received \r\n\r\n), return a pointer to the beginning
* of the body. Otherwise return NULL.
*/
char *
body_beginning(struct bytevec *vec);
/* If any header matching the provided name (NUL-terminated) exists, return
* a pointer to the beginning of the value for the first such occurrence
* and store the length of the header in n.
* If no such header exists, return NULL and don't modify n.
* The returned pointer will be borrowed from `headers`.
*/
const char *
get_first_header_value(const char *headers, size_t headers_len,
const char *name, size_t name_len, size_t *n);
void
log_cb(void *userdata, const struct rustls_log_params *params);
#endif /* COMMON_H */