1 files changed, 160 insertions, 0 deletions

diff --git a/src/lib/libssl/tls13_record.c b/src/lib/libssl/tls13_record.c
new file mode 100644
index 0000000000..857d3bee49
--- /dev/null
+++ b/src/lib/libssl/tls13_record.c
@@ -0,0 +1,160 @@
+/* $OpenBSD: tls13_record.c,v 1.1 2019/01/19 02:53:54 jsing Exp $ */
+/*
+ * Copyright (c) 2018, 2019 Joel Sing <jsing@openbsd.org>
+ *
+ * Permission to use, copy, modify, and distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+ * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ */
+
+#include "ssl_locl.h"
+
+#include <openssl/curve25519.h>
+
+#include "tls13_internal.h"
+#include "tls13_record.h"
+
+struct tls13_record {
+        uint8_t content_type;
+        size_t rec_len;
+        uint8_t *data;
+        size_t data_len;
+        CBS cbs;
+
+        struct tls13_buffer *buf;
+};
+
+struct tls13_record *
+tls13_record_new(void)
+{
+        struct tls13_record *rec = NULL;
+
+        if ((rec = calloc(1, sizeof(struct tls13_record))) == NULL)
+                goto err;
+        if ((rec->buf = tls13_buffer_new(TLS13_RECORD_MAX_LEN)) == NULL)
+                goto err;
+
+        return rec;
+
+ err:
+        tls13_record_free(rec);
+
+        return NULL;
+}
+
+void
+tls13_record_free(struct tls13_record *rec)
+{
+        if (rec == NULL)
+                return;
+
+        tls13_buffer_free(rec->buf);
+
+        freezero(rec->data, rec->data_len);
+        freezero(rec, sizeof(struct tls13_record));
+}
+
+uint8_t
+tls13_record_content_type(struct tls13_record *rec)
+{
+        return rec->content_type;
+}
+
+int
+tls13_record_content(struct tls13_record *rec, CBS *cbs)
+{
+        CBS content;
+
+        tls13_record_data(rec, &content);
+
+        if (!CBS_skip(&content, TLS13_RECORD_HEADER_LEN))
+                return 0;
+
+        CBS_dup(&content, cbs);
+
+        return 1;
+}
+
+void
+tls13_record_data(struct tls13_record *rec, CBS *cbs)
+{
+        CBS_init(cbs, rec->data, rec->data_len);
+}
+
+void
+tls13_record_set_data(struct tls13_record *rec, uint8_t *data, size_t data_len)
+{
+        freezero(rec->data, rec->data_len);
+        rec->data = data;
+        rec->data_len = data_len;
+        CBS_init(&rec->cbs, rec->data, rec->data_len);
+}
+
+ssize_t
+tls13_record_recv(struct tls13_record *rec, tls13_read_cb wire_read,
+    void *wire_arg)
+{
+        uint16_t rec_len, rec_version;
+        uint8_t content_type;
+        CBS cbs;
+        int ret;
+
+        if (rec->data != NULL)
+                return TLS13_IO_FAILURE;
+
+        if (rec->content_type == 0) {
+                if ((ret = tls13_buffer_extend(rec->buf,
+                    TLS13_RECORD_HEADER_LEN, wire_read, wire_arg)) <= 0)
+                        return ret;
+
+                tls13_buffer_cbs(rec->buf, &cbs);
+
+                if (!CBS_get_u8(&cbs, &content_type))
+                        return TLS13_IO_FAILURE;
+                if (!CBS_get_u16(&cbs, &rec_version))
+                        return TLS13_IO_FAILURE;
+                if (!CBS_get_u16(&cbs, &rec_len))
+                        return TLS13_IO_FAILURE;
+
+                rec->content_type = content_type;
+                rec->rec_len = rec_len;
+        }
+
+        if ((ret = tls13_buffer_extend(rec->buf,
+            TLS13_RECORD_HEADER_LEN + rec->rec_len, wire_read, wire_arg)) <= 0)
+                return ret;
+
+        if (!tls13_buffer_finish(rec->buf, &rec->data, &rec->data_len))
+                return TLS13_IO_FAILURE;
+
+        return rec->data_len;
+}
+
+ssize_t
+tls13_record_send(struct tls13_record *rec, tls13_write_cb wire_write,
+    void *wire_arg)
+{
+        ssize_t ret;
+
+        if (rec->data == NULL)
+                return TLS13_IO_FAILURE;
+
+        while (CBS_len(&rec->cbs) > 0) {
+                if ((ret = wire_write(CBS_data(&rec->cbs),
+                    CBS_len(&rec->cbs), wire_arg)) <= 0)
+                        return ret;
+
+                if (!CBS_skip(&rec->cbs, ret))
+                        return TLS13_IO_FAILURE;
+        }
+
+        return rec->data_len;
+}