ALSA: pcm: Fix potential data race at PCM memory allocation helpers
[linux-2.6-block.git] / sound / core / pcm_memory.c
index 7bde7fb64011e0f9372f898de4e909d854ab93ae..a0b9514716995fb65c31ca0fef62a8368d359416 100644 (file)
@@ -31,15 +31,41 @@ static unsigned long max_alloc_per_card = 32UL * 1024UL * 1024UL;
 module_param(max_alloc_per_card, ulong, 0644);
 MODULE_PARM_DESC(max_alloc_per_card, "Max total allocation bytes per card.");
 
+static void __update_allocated_size(struct snd_card *card, ssize_t bytes)
+{
+       card->total_pcm_alloc_bytes += bytes;
+}
+
+static void update_allocated_size(struct snd_card *card, ssize_t bytes)
+{
+       mutex_lock(&card->memory_mutex);
+       __update_allocated_size(card, bytes);
+       mutex_unlock(&card->memory_mutex);
+}
+
+static void decrease_allocated_size(struct snd_card *card, size_t bytes)
+{
+       mutex_lock(&card->memory_mutex);
+       WARN_ON(card->total_pcm_alloc_bytes < bytes);
+       __update_allocated_size(card, -(ssize_t)bytes);
+       mutex_unlock(&card->memory_mutex);
+}
+
 static int do_alloc_pages(struct snd_card *card, int type, struct device *dev,
                          int str, size_t size, struct snd_dma_buffer *dmab)
 {
        enum dma_data_direction dir;
        int err;
 
+       /* check and reserve the requested size */
+       mutex_lock(&card->memory_mutex);
        if (max_alloc_per_card &&
-           card->total_pcm_alloc_bytes + size > max_alloc_per_card)
+           card->total_pcm_alloc_bytes + size > max_alloc_per_card) {
+               mutex_unlock(&card->memory_mutex);
                return -ENOMEM;
+       }
+       __update_allocated_size(card, size);
+       mutex_unlock(&card->memory_mutex);
 
        if (str == SNDRV_PCM_STREAM_PLAYBACK)
                dir = DMA_TO_DEVICE;
@@ -47,9 +73,14 @@ static int do_alloc_pages(struct snd_card *card, int type, struct device *dev,
                dir = DMA_FROM_DEVICE;
        err = snd_dma_alloc_dir_pages(type, dev, dir, size, dmab);
        if (!err) {
-               mutex_lock(&card->memory_mutex);
-               card->total_pcm_alloc_bytes += dmab->bytes;
-               mutex_unlock(&card->memory_mutex);
+               /* the actual allocation size might be bigger than requested,
+                * and we need to correct the account
+                */
+               if (dmab->bytes != size)
+                       update_allocated_size(card, dmab->bytes - size);
+       } else {
+               /* take back on allocation failure */
+               decrease_allocated_size(card, size);
        }
        return err;
 }
@@ -58,10 +89,7 @@ static void do_free_pages(struct snd_card *card, struct snd_dma_buffer *dmab)
 {
        if (!dmab->area)
                return;
-       mutex_lock(&card->memory_mutex);
-       WARN_ON(card->total_pcm_alloc_bytes < dmab->bytes);
-       card->total_pcm_alloc_bytes -= dmab->bytes;
-       mutex_unlock(&card->memory_mutex);
+       decrease_allocated_size(card, dmab->bytes);
        snd_dma_free_pages(dmab);
        dmab->area = NULL;
 }