diff options
Diffstat (limited to 'drivers/base/devres.c')
-rw-r--r-- | drivers/base/devres.c | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/drivers/base/devres.c b/drivers/base/devres.c index 95b67281cd2a..f98a097e73f2 100644 --- a/drivers/base/devres.c +++ b/drivers/base/devres.c | |||
@@ -84,9 +84,14 @@ static struct devres_group * node_to_group(struct devres_node *node) | |||
84 | static __always_inline struct devres * alloc_dr(dr_release_t release, | 84 | static __always_inline struct devres * alloc_dr(dr_release_t release, |
85 | size_t size, gfp_t gfp, int nid) | 85 | size_t size, gfp_t gfp, int nid) |
86 | { | 86 | { |
87 | size_t tot_size = sizeof(struct devres) + size; | 87 | size_t tot_size; |
88 | struct devres *dr; | 88 | struct devres *dr; |
89 | 89 | ||
90 | /* We must catch any near-SIZE_MAX cases that could overflow. */ | ||
91 | if (unlikely(check_add_overflow(sizeof(struct devres), size, | ||
92 | &tot_size))) | ||
93 | return NULL; | ||
94 | |||
90 | dr = kmalloc_node_track_caller(tot_size, gfp, nid); | 95 | dr = kmalloc_node_track_caller(tot_size, gfp, nid); |
91 | if (unlikely(!dr)) | 96 | if (unlikely(!dr)) |
92 | return NULL; | 97 | return NULL; |