107 lines
2.2 KiB
C
107 lines
2.2 KiB
C
|
/*
|
||
|
|
* prng.h -- Pseudo Random Numbers
|
||
|
|
*
|
||
|
|
* Copyright (C) 2010-2011 Olaf Bergmann <bergmann@tzi.org>
|
||
|
|
*
|
||
|
|
* This file is part of the CoAP library libcoap. Please see README for terms
|
||
|
|
* of use.
|
||
|
|
*/
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @file prng.h
|
||
|
|
* @brief Pseudo Random Numbers
|
||
|
|
*/
|
||
|
|
|
||
|
|
#ifndef _COAP_PRNG_H_
|
||
|
|
#define _COAP_PRNG_H_
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @defgroup prng Pseudo Random Numbers
|
||
|
|
* @{
|
||
|
|
*/
|
||
|
|
|
||
|
|
#if defined(WITH_POSIX) || (defined(WITH_LWIP) && !defined(LWIP_RAND))
|
||
|
|
#include <stdlib.h>
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Fills \p buf with \p len random bytes. This is the default implementation for
|
||
|
|
* prng(). You might want to change prng() to use a better PRNG on your specific
|
||
|
|
* platform.
|
||
|
|
*/
|
||
|
|
static inline int
|
||
|
|
coap_prng_impl(unsigned char *buf, size_t len) {
|
||
|
|
while (len--)
|
||
|
|
*buf++ = rand() & 0xFF;
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
#endif /* WITH_POSIX */
|
||
|
|
|
||
|
|
#ifdef WITH_CONTIKI
|
||
|
|
#include <string.h>
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Fills \p buf with \p len random bytes. This is the default implementation for
|
||
|
|
* prng(). You might want to change prng() to use a better PRNG on your specific
|
||
|
|
* platform.
|
||
|
|
*/
|
||
|
|
static inline int
|
||
|
|
contiki_prng_impl(unsigned char *buf, size_t len) {
|
||
|
|
unsigned short v = random_rand();
|
||
|
|
while (len > sizeof(v)) {
|
||
|
|
memcpy(buf, &v, sizeof(v));
|
||
|
|
len -= sizeof(v);
|
||
|
|
buf += sizeof(v);
|
||
|
|
v = random_rand();
|
||
|
|
}
|
||
|
|
|
||
|
|
memcpy(buf, &v, len);
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
#define prng(Buf,Length) contiki_prng_impl((Buf), (Length))
|
||
|
|
#define prng_init(Value) random_init((unsigned short)(Value))
|
||
|
|
#endif /* WITH_CONTIKI */
|
||
|
|
|
||
|
|
#if defined(WITH_LWIP) && defined(LWIP_RAND)
|
||
|
|
static inline int
|
||
|
|
lwip_prng_impl(unsigned char *buf, size_t len) {
|
||
|
|
u32_t v = LWIP_RAND();
|
||
|
|
while (len > sizeof(v)) {
|
||
|
|
memcpy(buf, &v, sizeof(v));
|
||
|
|
len -= sizeof(v);
|
||
|
|
buf += sizeof(v);
|
||
|
|
v = LWIP_RAND();
|
||
|
|
}
|
||
|
|
|
||
|
|
memcpy(buf, &v, len);
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
#define prng(Buf,Length) lwip_prng_impl((Buf), (Length))
|
||
|
|
#define prng_init(Value)
|
||
|
|
|
||
|
|
#endif /* WITH_LWIP */
|
||
|
|
|
||
|
|
#ifndef prng
|
||
|
|
/**
|
||
|
|
* Fills \p Buf with \p Length bytes of random data.
|
||
|
|
*
|
||
|
|
* @hideinitializer
|
||
|
|
*/
|
||
|
|
#define prng(Buf,Length) coap_prng_impl((Buf), (Length))
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#ifndef prng_init
|
||
|
|
/**
|
||
|
|
* Called to set the PRNG seed. You may want to re-define this to allow for a
|
||
|
|
* better PRNG.
|
||
|
|
*
|
||
|
|
* @hideinitializer
|
||
|
|
*/
|
||
|
|
#define prng_init(Value) srand((unsigned long)(Value))
|
||
|
|
#endif
|
||
|
|
|
||
|
|
/** @} */
|
||
|
|
|
||
|
|
#endif /* _COAP_PRNG_H_ */
|