vbatts/grub
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Prevente error in broken cards by limiting data size.

Browse source
This commit is contained in:
Manoel Rebelo Abranches 2011-05-10 14:07:20 -03:00
parent 36e2782dba
commit 0696f08f92
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
grub-core/net/tftp.c
View file

@ -126,8 +126,12 @@ tftp_receive (grub_net_socket_t sock, struct grub_net_buff *nb)
if (grub_be_to_cpu16 (tftph->u.data.block) == data->block + 1) if (grub_be_to_cpu16 (tftph->u.data.block) == data->block + 1)
{ {
data->block++; data->block++;
if (nb->tail - nb->data < 1024) unsigned size = nb->tail - nb->data;
if (size < 1024)
sock->status = 2; sock->status = 2;
/* Prevent garbage in broken cards. */
if (size > 1024)
grub_netbuff_unput (nb, size - 1024);
} }
else else
grub_netbuff_clear(nb); grub_netbuff_clear(nb);