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

* grub-core/net/ip.c (reassemble): Make asm_buffer into asm_netbuff.

Browse source 
All users updated.
	(free_rsm): Free header as well.
	(free_old_fragments): Fix memory leak.
	* grub-core/net/netbuff.c (grub_netbuff_free): Make return void.
	* grub-core/net/tftp.c (tftp_receive): Fix memory leak.
	(destroy_pq): Likewise.
	* include/grub/net/netbuff.h (grub_netbuff_free): Make return void.
This commit is contained in:
Bean 2012-05-08 16:03:02 +02:00 committed by Vladimir 'phcoder' Serbinenko
parent 668ea6d26d
commit 5efb817d64
5 changed files with 51 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
ChangeLog
View file

@ -1,3 +1,14 @@
2012-05-08 Bean <bean123ch@gmail.com>
* grub-core/net/ip.c (reassemble): Make asm_buffer into asm_netbuff.
All users updated.
(free_rsm): Free header as well.
(free_old_fragments): Fix memory leak.
* grub-core/net/netbuff.c (grub_netbuff_free): Make return void.
* grub-core/net/tftp.c (tftp_receive): Fix memory leak.
(destroy_pq): Likewise.
* include/grub/net/netbuff.h (grub_netbuff_free): Make return void.
2012-05-08 Vladimir Serbinenko <phcoder@gmail.com> 2012-05-08 Vladimir Serbinenko <phcoder@gmail.com>
* grub-core/commands/hashsum.c (grub_cmd_hashsum): Align space for * grub-core/commands/hashsum.c (grub_cmd_hashsum): Align space for

49
grub-core/net/ip.c
View file

@ -84,7 +84,7 @@ struct reassemble
grub_uint8_t proto; grub_uint8_t proto;
grub_uint64_t last_time; grub_uint64_t last_time;
grub_priority_queue_t pq; grub_priority_queue_t pq;
grub_uint8_t *asm_buffer; struct grub_net_buff *asm_netbuff;
grub_size_t total_len; grub_size_t total_len;
grub_size_t cur_ptr; grub_size_t cur_ptr;
grub_uint8_t ttl; grub_uint8_t ttl;
@ -351,8 +351,9 @@ free_rsm (struct reassemble *rsm)
grub_netbuff_free (*nb); grub_netbuff_free (*nb);
grub_priority_queue_pop (rsm->pq); grub_priority_queue_pop (rsm->pq);
} }
grub_free (rsm->asm_buffer); grub_netbuff_free (rsm->asm_netbuff);
grub_priority_queue_destroy (rsm->pq); grub_priority_queue_destroy (rsm->pq);
grub_free (rsm);
} }
static void static void
@ -361,12 +362,16 @@ free_old_fragments (void)
struct reassemble *rsm, **prev; struct reassemble *rsm, **prev;
grub_uint64_t limit_time = grub_get_time_ms () - 90000; grub_uint64_t limit_time = grub_get_time_ms () - 90000;
for (prev = &reassembles, rsm = *prev; rsm; prev = &rsm->next, rsm = *prev) for (prev = &reassembles, rsm = *prev; rsm; rsm = *prev)
if (rsm->last_time < limit_time) if (rsm->last_time < limit_time)
{ {
*prev = rsm->next; *prev = rsm->next;
free_rsm (rsm); free_rsm (rsm);
} }
else
{
prev = &rsm->next;
}
} }
static grub_err_t static grub_err_t
@ -473,7 +478,7 @@ grub_net_recv_ip4_packets (struct grub_net_buff *nb,
grub_free (rsm); grub_free (rsm);
return grub_errno; return grub_errno;
} }
rsm->asm_buffer = 0; rsm->asm_netbuff = 0;
rsm->total_len = 0; rsm->total_len = 0;
rsm->cur_ptr = 0; rsm->cur_ptr = 0;
rsm->ttl = 0xff; rsm->ttl = 0xff;
@ -492,22 +497,21 @@ grub_net_recv_ip4_packets (struct grub_net_buff *nb,
rsm->total_len = (8 * (grub_be_to_cpu16 (iph->frags) & OFFSET_MASK) rsm->total_len = (8 * (grub_be_to_cpu16 (iph->frags) & OFFSET_MASK)
+ (nb->tail - nb->data)); + (nb->tail - nb->data));
rsm->total_len -= ((iph->verhdrlen & 0xf) * sizeof (grub_uint32_t)); rsm->total_len -= ((iph->verhdrlen & 0xf) * sizeof (grub_uint32_t));
rsm->asm_buffer = grub_zalloc (rsm->total_len); rsm->asm_netbuff = grub_netbuff_alloc (rsm->total_len);
if (!rsm->asm_buffer) if (!rsm->asm_netbuff)
{ {
*prev = rsm->next; *prev = rsm->next;
free_rsm (rsm); free_rsm (rsm);
return grub_errno; return grub_errno;
} }
} }
if (!rsm->asm_buffer) if (!rsm->asm_netbuff)
return GRUB_ERR_NONE; return GRUB_ERR_NONE;
while (1) while (1)
{ {
struct grub_net_buff **nb_top_p, *nb_top; struct grub_net_buff **nb_top_p, *nb_top;
grub_size_t copy; grub_size_t copy;
grub_uint8_t *res;
grub_size_t res_len; grub_size_t res_len;
struct grub_net_buff *ret; struct grub_net_buff *ret;
grub_net_ip_protocol_t proto; grub_net_ip_protocol_t proto;
@ -532,7 +536,10 @@ grub_net_recv_ip4_packets (struct grub_net_buff *nb,
} }
if (rsm->cur_ptr < (grub_size_t) 8 * (grub_be_to_cpu16 (iph->frags) if (rsm->cur_ptr < (grub_size_t) 8 * (grub_be_to_cpu16 (iph->frags)
& OFFSET_MASK)) & OFFSET_MASK))
return GRUB_ERR_NONE; {
grub_netbuff_free (nb_top);
return GRUB_ERR_NONE;
}
rsm->cur_ptr = (8 * (grub_be_to_cpu16 (iph->frags) & OFFSET_MASK) rsm->cur_ptr = (8 * (grub_be_to_cpu16 (iph->frags) & OFFSET_MASK)
+ (nb_top->tail - nb_top->head)); + (nb_top->tail - nb_top->head));
@ -547,31 +554,33 @@ grub_net_recv_ip4_packets (struct grub_net_buff *nb,
< copy) < copy)
copy = rsm->total_len - 8 * (grub_be_to_cpu16 (iph->frags) copy = rsm->total_len - 8 * (grub_be_to_cpu16 (iph->frags)
& OFFSET_MASK); & OFFSET_MASK);
grub_memcpy (&rsm->asm_buffer[8 * (grub_be_to_cpu16 (iph->frags) grub_memcpy (&rsm->asm_netbuff->data[8 * (grub_be_to_cpu16 (iph->frags)
& OFFSET_MASK)], & OFFSET_MASK)],
nb_top->data, copy); nb_top->data, copy);
if ((grub_be_to_cpu16 (iph->frags) & MORE_FRAGMENTS)) if ((grub_be_to_cpu16 (iph->frags) & MORE_FRAGMENTS))
continue; {
grub_netbuff_free (nb_top);
continue;
}
grub_netbuff_free (nb_top);
res = rsm->asm_buffer; ret = rsm->asm_netbuff;
proto = rsm->proto; proto = rsm->proto;
src = rsm->source; src = rsm->source;
dst = rsm->dest; dst = rsm->dest;
ttl = rsm->ttl; ttl = rsm->ttl;
rsm->asm_buffer = 0; rsm->asm_netbuff = 0;
res_len = rsm->total_len; res_len = rsm->total_len;
*prev = rsm->next; *prev = rsm->next;
free_rsm (rsm); free_rsm (rsm);
ret = grub_malloc (sizeof (*ret));
if (!ret) if (grub_netbuff_put (ret, res_len))
{ {
grub_free (res); grub_netbuff_free (ret);
return grub_errno; return GRUB_ERR_NONE;
} }
ret->data = ret->head = res;
ret->tail = ret->end = res + res_len;
source.type = GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV4; source.type = GRUB_NET_NETWORK_LEVEL_PROTOCOL_IPV4;
source.ipv4 = src; source.ipv4 = src;

5
grub-core/net/netbuff.c
View file

@ -93,11 +93,12 @@ grub_netbuff_alloc (grub_size_t len)
return nb; return nb;
} }
grub_err_t void
grub_netbuff_free (struct grub_net_buff *nb) grub_netbuff_free (struct grub_net_buff *nb)
{ {
if (!nb)
return;
grub_free (nb->head); grub_free (nb->head);
return GRUB_ERR_NONE;
} }
grub_err_t grub_err_t

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

@ -214,6 +214,7 @@ tftp_receive (grub_net_udp_socket_t sock __attribute__ ((unused)),
tftph = (struct tftphdr *) nb_top->data; tftph = (struct tftphdr *) nb_top->data;
if (grub_be_to_cpu16 (tftph->u.data.block) >= data->block + 1) if (grub_be_to_cpu16 (tftph->u.data.block) >= data->block + 1)
break; break;
grub_netbuff_free (nb_top);
grub_priority_queue_pop (data->pq); grub_priority_queue_pop (data->pq);
} }
if (grub_be_to_cpu16 (tftph->u.data.block) == data->block + 1) if (grub_be_to_cpu16 (tftph->u.data.block) == data->block + 1)
@ -248,7 +249,7 @@ tftp_receive (grub_net_udp_socket_t sock __attribute__ ((unused)),
if ((nb_top->tail - nb_top->data) > 0) if ((nb_top->tail - nb_top->data) > 0)
grub_net_put_packet (&file->device->net->packs, nb_top); grub_net_put_packet (&file->device->net->packs, nb_top);
else else
grub_netbuff_free (nb); grub_netbuff_free (nb_top);
} }
} }
return GRUB_ERR_NONE; return GRUB_ERR_NONE;
@ -269,7 +270,10 @@ destroy_pq (tftp_data_t data)
{ {
struct grub_net_buff **nb_p; struct grub_net_buff **nb_p;
while ((nb_p = grub_priority_queue_top (data->pq))) while ((nb_p = grub_priority_queue_top (data->pq)))
grub_netbuff_free (*nb_p); {
grub_netbuff_free (*nb_p);
grub_priority_queue_pop (data->pq);
}
grub_priority_queue_destroy (data->pq); grub_priority_queue_destroy (data->pq);
} }

2
include/grub/net/netbuff.h
View file

@ -25,6 +25,6 @@ grub_err_t grub_netbuff_pull (struct grub_net_buff *net_buff, grub_size_t len);
grub_err_t grub_netbuff_reserve (struct grub_net_buff *net_buff, grub_size_t len); grub_err_t grub_netbuff_reserve (struct grub_net_buff *net_buff, grub_size_t len);
grub_err_t grub_netbuff_clear (struct grub_net_buff *net_buff); grub_err_t grub_netbuff_clear (struct grub_net_buff *net_buff);
struct grub_net_buff * grub_netbuff_alloc (grub_size_t len); struct grub_net_buff * grub_netbuff_alloc (grub_size_t len);
grub_err_t grub_netbuff_free (struct grub_net_buff *net_buff); void grub_netbuff_free (struct grub_net_buff *net_buff);
#endif #endif