stv090x: handle allocation failures
Commit Message
kmalloc() can fail so check whether state->internal is NULL.
append_internal() can return NULL on allocation failures so check that.
Also if we hit the error condition later in the function then there is
a memory leak and we need to call remove_dev() to fix it.
Signed-off-by: Dan Carpenter <error27@gmail.com>
---
Compile tested only. I'm not very familiar with this code.
--
To unsubscribe from this list: send the line "unsubscribe linux-media" in
the body of a message to majordomo@vger.kernel.org
More majordomo info at http://vger.kernel.org/majordomo-info.html
Comments
On Monday 07 February 2011 17:56:50 Dan Carpenter wrote:
> kmalloc() can fail so check whether state->internal is NULL.
> append_internal() can return NULL on allocation failures so check that.
> Also if we hit the error condition later in the function then there is
> a memory leak and we need to call remove_dev() to fix it.
> ...
Thanks for the patch. See my comment below.
> --- a/drivers/media/dvb/frontends/stv090x.c
> +++ b/drivers/media/dvb/frontends/stv090x.c
> @@ -4783,7 +4783,13 @@ struct dvb_frontend *stv090x_attach(const struct stv090x_config *config,
> } else {
> state->internal = kmalloc(sizeof(struct stv090x_internal),
> GFP_KERNEL);
> + if (!state->internal)
> + goto error;
> temp_int = append_internal(state->internal);
> + if (!temp_int) {
> + kfree(state->internal);
> + goto error;
> + }
> state->internal->num_used = 1;
> state->internal->mclk = 0;
> state->internal->dev_ver = 0;
> @@ -4796,7 +4802,7 @@ struct dvb_frontend *stv090x_attach(const struct stv090x_config *config,
>
> if (stv090x_setup(&state->frontend) < 0) {
> dprintk(FE_ERROR, 1, "Error setting up device");
> - goto error;
> + goto err_remove;
> }
> }
>
> @@ -4811,6 +4817,8 @@ struct dvb_frontend *stv090x_attach(const struct stv090x_config *config,
>
> return &state->frontend;
>
> +err_remove:
> + remove_dev(state->internal);
AFAICS
kfree(state->internal);
must be added here, as we allocated state->internal, state and temp_int.
> error:
> kfree(state);
> return NULL;
CU
Oliver
@@ -4783,7 +4783,13 @@ struct dvb_frontend *stv090x_attach(const struct stv090x_config *config,
} else {
state->internal = kmalloc(sizeof(struct stv090x_internal),
GFP_KERNEL);
+ if (!state->internal)
+ goto error;
temp_int = append_internal(state->internal);
+ if (!temp_int) {
+ kfree(state->internal);
+ goto error;
+ }
state->internal->num_used = 1;
state->internal->mclk = 0;
state->internal->dev_ver = 0;
@@ -4796,7 +4802,7 @@ struct dvb_frontend *stv090x_attach(const struct stv090x_config *config,
if (stv090x_setup(&state->frontend) < 0) {
dprintk(FE_ERROR, 1, "Error setting up device");
- goto error;
+ goto err_remove;
}
}
@@ -4811,6 +4817,8 @@ struct dvb_frontend *stv090x_attach(const struct stv090x_config *config,
return &state->frontend;
+err_remove:
+ remove_dev(state->internal);
error:
kfree(state);
return NULL;