ATM-ZeitNet: Improve a size determination in zatm_open()

[cascardo/linux.git] / drivers / atm / zatm.c

diff --git a/drivers/atm/zatm.c b/drivers/atm/zatm.c
index cecfb94..218c6af 100644 (file)
--- a/drivers/atm/zatm.c
+++ b/drivers/atm/zatm.c
@@ -998,8 +998,9 @@ static int start_tx(struct atm_dev *dev)
 
        DPRINTK("start_tx\n");
        zatm_dev = ZATM_DEV(dev);
-       zatm_dev->tx_map = kmalloc(sizeof(struct atm_vcc *)*
-           zatm_dev->chans,GFP_KERNEL);
+       zatm_dev->tx_map = kmalloc_array(zatm_dev->chans,
+                                        sizeof(*zatm_dev->tx_map),
+                                        GFP_KERNEL);
        if (!zatm_dev->tx_map) return -ENOMEM;
        zatm_dev->tx_bw = ATM_OC3_PCR;
        zatm_dev->free_shapers = (1 << NR_SHAPERS)-1;
@@ -1398,7 +1399,7 @@ static int zatm_open(struct atm_vcc *vcc)
        DPRINTK(DEV_LABEL "(itf %d): open %d.%d\n",vcc->dev->number,vcc->vpi,
            vcc->vci);
        if (!test_bit(ATM_VF_PARTIAL,&vcc->flags)) {
-               zatm_vcc = kmalloc(sizeof(struct zatm_vcc),GFP_KERNEL);
+               zatm_vcc = kmalloc(sizeof(*zatm_vcc), GFP_KERNEL);
                if (!zatm_vcc) {
                        clear_bit(ATM_VF_ADDR,&vcc->flags);
                        return -ENOMEM;