1
0
mirror of https://github.com/RIOT-OS/RIOT.git synced 2024-12-29 04:50:03 +01:00
RIOT/examples/nanocoap_server/coap_handler.c

116 lines
3.4 KiB
C

/*
* Copyright (C) 2016 Kaspar Schleiser <kaspar@schleiser.de>
*
* This file is subject to the terms and conditions of the GNU Lesser
* General Public License v2.1. See the file LICENSE in the top level
* directory for more details.
*/
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include "fmt.h"
#include "net/nanocoap.h"
#include "hashes/sha256.h"
/* internal value that can be read/written via CoAP */
static uint8_t internal_value = 0;
static ssize_t _riot_board_handler(coap_pkt_t *pkt, uint8_t *buf, size_t len, void *context)
{
(void)context;
return coap_reply_simple(pkt, COAP_CODE_205, buf, len,
COAP_FORMAT_TEXT, (uint8_t*)RIOT_BOARD, strlen(RIOT_BOARD));
}
static ssize_t _riot_value_handler(coap_pkt_t *pkt, uint8_t *buf, size_t len, void *context)
{
(void) context;
ssize_t p = 0;
char rsp[16];
unsigned code = COAP_CODE_EMPTY;
/* read coap method type in packet */
unsigned method_flag = coap_method2flag(coap_get_code_detail(pkt));
switch(method_flag) {
case COAP_GET:
/* write the response buffer with the internal value */
p += fmt_u32_dec(rsp, internal_value);
code = COAP_CODE_205;
break;
case COAP_PUT:
case COAP_POST:
{
/* convert the payload to an integer and update the internal value */
char payload[16] = { 0 };
memcpy(payload, (char*)pkt->payload, pkt->payload_len);
internal_value = strtol(payload, NULL, 10);
code = COAP_CODE_CHANGED;
}
}
return coap_reply_simple(pkt, code, buf, len,
COAP_FORMAT_TEXT, (uint8_t*)rsp, p);
}
ssize_t _sha256_handler(coap_pkt_t* pkt, uint8_t *buf, size_t len, void *context)
{
(void)context;
/* using a shared sha256 context *will* break if two requests are handled
* at the same time. doing it anyways, as this is meant to showcase block1
* support, not proper synchronisation. */
static sha256_context_t sha256;
uint8_t digest[SHA256_DIGEST_LENGTH];
uint32_t result = COAP_CODE_204;
coap_block1_t block1;
int blockwise = coap_get_block1(pkt, &block1);
printf("_sha256_handler(): received data: offset=%u len=%u blockwise=%i more=%i\n", \
(unsigned)block1.offset, pkt->payload_len, blockwise, block1.more);
if (block1.offset == 0) {
puts("_sha256_handler(): init");
sha256_init(&sha256);
}
sha256_update(&sha256, pkt->payload, pkt->payload_len);
if (block1.more == 1) {
result = COAP_CODE_231;
}
size_t result_len = 0;
if (!blockwise || !block1.more) {
puts("_sha256_handler(): finish");
sha256_final(&sha256, digest);
result_len = SHA256_DIGEST_LENGTH * 2;
}
ssize_t reply_len = coap_build_reply(pkt, result, buf, len, 0);
uint8_t *pkt_pos = (uint8_t*)pkt->hdr + reply_len;
pkt_pos += coap_put_block1_ok(pkt_pos, &block1, 0);
if (result_len) {
*pkt_pos++ = 0xFF;
pkt_pos += fmt_bytes_hex((char *)pkt_pos, digest, sizeof(digest));
}
return pkt_pos - (uint8_t*)pkt->hdr;
}
/* must be sorted by path (alphabetically) */
const coap_resource_t coap_resources[] = {
COAP_WELL_KNOWN_CORE_DEFAULT_HANDLER,
{ "/sha256", COAP_POST, _sha256_handler, NULL },
{ "/riot/board", COAP_GET, _riot_board_handler, NULL },
{ "/riot/value", COAP_GET | COAP_PUT | COAP_POST, _riot_value_handler, NULL },
};
const unsigned coap_resources_numof = sizeof(coap_resources) / sizeof(coap_resources[0]);