diff options
| author | Bernard Xiong <[email protected]> | 2019-12-18 21:38:05 +0800 |
|---|---|---|
| committer | Bernard Xiong <[email protected]> | 2019-12-18 21:38:05 +0800 |
| commit | 055061a6397bcaafab36723ca1598a8b9402059d (patch) | |
| tree | bb0fbd82d85bc6ce9299357ea9ca39e9430af6f6 /components/drivers | |
| parent | 78d42efd223275b93776d7653a5405ecd29feb8f (diff) | |
[DeviceDrivers] Fix pipe memory issue.
Diffstat (limited to 'components/drivers')
| -rw-r--r-- | components/drivers/src/pipe.c | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/components/drivers/src/pipe.c b/components/drivers/src/pipe.c index b682c351ad..74784240dc 100644 --- a/components/drivers/src/pipe.c +++ b/components/drivers/src/pipe.c @@ -19,6 +19,7 @@ static int pipe_fops_open(struct dfs_fd *fd) { + int rc = 0; rt_device_t device; rt_pipe_t *pipe; @@ -31,6 +32,11 @@ static int pipe_fops_open(struct dfs_fd *fd) if (device->ref_count == 0) { pipe->fifo = rt_ringbuffer_create(pipe->bufsz); + if (pipe->fifo == RT_NULL) + { + rc = -RT_ENOMEM; + goto __exit; + } } switch (fd->flags & O_ACCMODE) @@ -48,9 +54,10 @@ static int pipe_fops_open(struct dfs_fd *fd) } device->ref_count ++; +__exit: rt_mutex_release(&(pipe->lock)); - return 0; + return rc; } static int pipe_fops_close(struct dfs_fd *fd) @@ -90,7 +97,8 @@ static int pipe_fops_close(struct dfs_fd *fd) if (device->ref_count == 1) { - rt_ringbuffer_destroy(pipe->fifo); + if (pipe->fifo != RT_NULL) + rt_ringbuffer_destroy(pipe->fifo); pipe->fifo = RT_NULL; } device->ref_count --; |
