From 13fd806f4f56262c07f1fed39a17af0cd099e7fc Mon Sep 17 00:00:00 2001
From: Benjamin Valentin <benjamin.valentin@ml-pa.com>
Date: Wed, 10 Apr 2024 12:36:53 +0200
Subject: [PATCH] nanocoap_sock: add option to include token for block-wise

---
 sys/include/net/nanocoap_sock.h           | 12 ++++++++++++
 sys/net/application_layer/nanocoap/sock.c | 19 ++++++++++++++++++-
 2 files changed, 30 insertions(+), 1 deletion(-)

diff --git a/sys/include/net/nanocoap_sock.h b/sys/include/net/nanocoap_sock.h
index 55b6881c04bb..e079df79cc5c 100644
--- a/sys/include/net/nanocoap_sock.h
+++ b/sys/include/net/nanocoap_sock.h
@@ -175,6 +175,18 @@ extern "C" {
 #define CONFIG_NANOCOAP_SERVER_STACK_SIZE       THREAD_STACKSIZE_DEFAULT
 #endif
 
+/**
+ * @brief   Include a random token with block-wise transfers.
+ *
+ * This is a workaround for buggy CoPA implementations (e.g. go-coap) that expect
+ * to identify block-wise transfers based on the token.
+ *
+ * See https://github.com/plgd-dev/go-coap/issues/512
+ */
+#ifndef CONFIG_NANOCOAP_SOCK_BLOCK_TOKEN
+#define CONFIG_NANOCOAP_SOCK_BLOCK_TOKEN        (0)
+#endif
+
 /**
  * @brief   NanoCoAP socket types
  */
diff --git a/sys/net/application_layer/nanocoap/sock.c b/sys/net/application_layer/nanocoap/sock.c
index eb82c7bef174..1802ad26cdcf 100644
--- a/sys/net/application_layer/nanocoap/sock.c
+++ b/sys/net/application_layer/nanocoap/sock.c
@@ -60,6 +60,9 @@ typedef struct {
     coap_blockwise_cb_t callback;
     void *arg;
     bool more;
+#if CONFIG_NANOCOAP_SOCK_BLOCK_TOKEN
+    uint8_t token[4];
+#endif
 } _block_ctx_t;
 
 int nanocoap_sock_dtls_connect(nanocoap_sock_t *sock, sock_udp_ep_t *local,
@@ -601,7 +604,17 @@ static int _fetch_block(nanocoap_sock_t *sock, uint8_t *buf, size_t len,
     };
     uint16_t lastonum = 0;
 
-    buf += coap_build_hdr(pkt.hdr, COAP_TYPE_CON, NULL, 0, COAP_METHOD_GET,
+    void *token = NULL;
+    size_t token_len = 0;
+
+#if CONFIG_NANOCOAP_SOCK_BLOCK_TOKEN
+    /* HACK: go-coap always expects a token */
+    /* see https://github.com/plgd-dev/go-coap/issues/512 */
+    token = ctx->token;
+    token_len = sizeof(ctx->token);
+#endif
+
+    buf += coap_build_hdr(pkt.hdr, COAP_TYPE_CON, token, token_len, COAP_METHOD_GET,
                           nanocoap_sock_next_msg_id(sock));
     buf += coap_opt_put_uri_pathquery(buf, &lastonum, path);
     buf += coap_opt_put_uint(buf, lastonum, COAP_OPT_BLOCK2, (num << 4) | blksize);
@@ -674,6 +687,10 @@ int nanocoap_sock_get_blockwise(nanocoap_sock_t *sock, const char *path,
         .more = true,
     };
 
+#if CONFIG_NANOCOAP_SOCK_BLOCK_TOKEN
+    random_bytes(ctx.token, sizeof(ctx.token));
+#endif
+
     unsigned num = 0;
     while (ctx.more) {
         DEBUG("nanocoap: fetching block %u\n", num);