Author: pfg
Date: Thu Jan 11 15:29:29 2018
New Revision: 327829
URL: https://svnweb.freebsd.org/changeset/base/327829

Log:
  netpfil/ipfw: Make some use of mallocarray(9).
  
  Reviewed by:  kp, ae
  Differential Revision: https://reviews.freebsd.org/D13834

Modified:
  head/sys/netpfil/ipfw/dn_heap.c
  head/sys/netpfil/ipfw/dn_sched_fq_codel.c
  head/sys/netpfil/ipfw/dn_sched_fq_pie.c

Modified: head/sys/netpfil/ipfw/dn_heap.c
==============================================================================
--- head/sys/netpfil/ipfw/dn_heap.c     Thu Jan 11 15:25:26 2018        
(r327828)
+++ head/sys/netpfil/ipfw/dn_heap.c     Thu Jan 11 15:29:29 2018        
(r327829)
@@ -94,7 +94,7 @@ heap_resize(struct dn_heap *h, unsigned int new_size)
 #else
        new_size = (new_size + HEAP_INCREMENT ) & ~HEAP_INCREMENT;
 #endif
-       p = malloc(new_size * sizeof(*p), M_DN_HEAP, M_NOWAIT);
+       p = mallocarray(new_size, sizeof(*p), M_DN_HEAP, M_NOWAIT);
        if (p == NULL) {
                printf("--- %s, resize %d failed\n", __func__, new_size );
                return 1; /* error */

Modified: head/sys/netpfil/ipfw/dn_sched_fq_codel.c
==============================================================================
--- head/sys/netpfil/ipfw/dn_sched_fq_codel.c   Thu Jan 11 15:25:26 2018        
(r327828)
+++ head/sys/netpfil/ipfw/dn_sched_fq_codel.c   Thu Jan 11 15:29:29 2018        
(r327829)
@@ -453,8 +453,8 @@ fq_codel_new_sched(struct dn_sch_inst *_si)
        q->fs = _si->sched->fs;
 
        /* allocate memory for flows array */
-       si->flows = malloc(schk->cfg.flows_cnt * sizeof(struct fq_codel_flow),
-                M_DUMMYNET, M_NOWAIT | M_ZERO);
+       si->flows = mallocarray(schk->cfg.flows_cnt,
+           sizeof(struct fq_codel_flow), M_DUMMYNET, M_NOWAIT | M_ZERO);
        if (si->flows == NULL) {
                D("cannot allocate memory for fq_codel configuration 
parameters");
                return ENOMEM ; 

Modified: head/sys/netpfil/ipfw/dn_sched_fq_pie.c
==============================================================================
--- head/sys/netpfil/ipfw/dn_sched_fq_pie.c     Thu Jan 11 15:25:26 2018        
(r327828)
+++ head/sys/netpfil/ipfw/dn_sched_fq_pie.c     Thu Jan 11 15:29:29 2018        
(r327829)
@@ -1034,8 +1034,8 @@ fq_pie_new_sched(struct dn_sch_inst *_si)
                return ENOMEM ; 
        }
        /* allocate memory for flows array */
-       si->si_extra->flows = malloc(schk->cfg.flows_cnt * sizeof(struct 
fq_pie_flow),
-                M_DUMMYNET, M_NOWAIT | M_ZERO);
+       si->si_extra->flows = mallocarray(schk->cfg.flows_cnt,
+           sizeof(struct fq_pie_flow), M_DUMMYNET, M_NOWAIT | M_ZERO);
        flows = si->si_extra->flows;
        if (flows == NULL) {
                free(si->si_extra, M_DUMMYNET);
_______________________________________________
[email protected] mailing list
https://lists.freebsd.org/mailman/listinfo/svn-src-head
To unsubscribe, send any mail to "[email protected]"

Reply via email to