sergiotarxz/nuttx
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

net/tcp:make initial tcp port more random

Browse Source 
Signed-off-by: 田昕 <tianxin7@xiaomi.com>
This commit is contained in:
田昕 2022-04-26 19:59:07 +08:00 committed by Petro Karashchenko
parent 042640abbf
commit 670ea1e5fb
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
net/tcp/tcp_conn.c
View File

@ -49,6 +49,7 @@
#include <assert.h> #include <assert.h>
#include <errno.h> #include <errno.h>
#include <debug.h> #include <debug.h>
#include <sys/random.h>
#include <netinet/in.h> #include <netinet/in.h>
@ -201,12 +202,26 @@ static int tcp_selectport(uint8_t domain,
uint16_t portno) uint16_t portno)
{ {
static uint16_t g_last_tcp_port; static uint16_t g_last_tcp_port;
ssize_t ret;
/* Generate port base dynamically */ /* Generate port base dynamically */
if (g_last_tcp_port == 0) if (g_last_tcp_port == 0)
{ {
g_last_tcp_port = clock_systime_ticks() % 32000; ret = getrandom(&g_last_tcp_port, sizeof(uint16_t), 0);
if (ret < 0)
{
ret = getrandom(&g_last_tcp_port, sizeof(uint16_t), GRND_RANDOM);
}
if (ret != sizeof(uint16_t))
{
g_last_tcp_port = clock_systime_ticks() % 32000;
}
else
{
g_last_tcp_port = g_last_tcp_port % 32000;
}
if (g_last_tcp_port < 4096) if (g_last_tcp_port < 4096)
{ {