diff options
-rw-r--r-- | net/netrom/nr_route.c | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/net/netrom/nr_route.c b/net/netrom/nr_route.c index c11737f472d6..0096105bcd47 100644 --- a/net/netrom/nr_route.c +++ b/net/netrom/nr_route.c | |||
@@ -155,14 +155,15 @@ static int nr_add_node(ax25_address *nr, const char *mnemonic, ax25_address *ax2 | |||
155 | atomic_set(&nr_neigh->refcount, 1); | 155 | atomic_set(&nr_neigh->refcount, 1); |
156 | 156 | ||
157 | if (ax25_digi != NULL && ax25_digi->ndigi > 0) { | 157 | if (ax25_digi != NULL && ax25_digi->ndigi > 0) { |
158 | if ((nr_neigh->digipeat = kmalloc(sizeof(*ax25_digi), GFP_KERNEL)) == NULL) { | 158 | nr_neigh->digipeat = kmemdup(ax25_digi, |
159 | sizeof(*ax25_digi), | ||
160 | GFP_KERNEL); | ||
161 | if (nr_neigh->digipeat == NULL) { | ||
159 | kfree(nr_neigh); | 162 | kfree(nr_neigh); |
160 | if (nr_node) | 163 | if (nr_node) |
161 | nr_node_put(nr_node); | 164 | nr_node_put(nr_node); |
162 | return -ENOMEM; | 165 | return -ENOMEM; |
163 | } | 166 | } |
164 | memcpy(nr_neigh->digipeat, ax25_digi, | ||
165 | sizeof(*ax25_digi)); | ||
166 | } | 167 | } |
167 | 168 | ||
168 | spin_lock_bh(&nr_neigh_list_lock); | 169 | spin_lock_bh(&nr_neigh_list_lock); |
@@ -432,11 +433,12 @@ static int nr_add_neigh(ax25_address *callsign, ax25_digi *ax25_digi, struct net | |||
432 | atomic_set(&nr_neigh->refcount, 1); | 433 | atomic_set(&nr_neigh->refcount, 1); |
433 | 434 | ||
434 | if (ax25_digi != NULL && ax25_digi->ndigi > 0) { | 435 | if (ax25_digi != NULL && ax25_digi->ndigi > 0) { |
435 | if ((nr_neigh->digipeat = kmalloc(sizeof(*ax25_digi), GFP_KERNEL)) == NULL) { | 436 | nr_neigh->digipeat = kmemdup(ax25_digi, sizeof(*ax25_digi), |
437 | GFP_KERNEL); | ||
438 | if (nr_neigh->digipeat == NULL) { | ||
436 | kfree(nr_neigh); | 439 | kfree(nr_neigh); |
437 | return -ENOMEM; | 440 | return -ENOMEM; |
438 | } | 441 | } |
439 | memcpy(nr_neigh->digipeat, ax25_digi, sizeof(*ax25_digi)); | ||
440 | } | 442 | } |
441 | 443 | ||
442 | spin_lock_bh(&nr_neigh_list_lock); | 444 | spin_lock_bh(&nr_neigh_list_lock); |