src/tests/memblock-test.c

   1 /* $Id$ */
   2
   3 /***
   4   This file is part of PulseAudio.
   5
   6   PulseAudio is free software; you can redistribute it and/or modify
   7   it under the terms of the GNU Lesser General Public License as published
   8   by the Free Software Foundation; either version 2 of the License,
   9   or (at your option) any later version.
  10
  11   PulseAudio is distributed in the hope that it will be useful, but
  12   WITHOUT ANY WARRANTY; without even the implied warranty of
  13   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14   General Public License for more details.
  15
  16   You should have received a copy of the GNU Lesser General Public License
  17   along with PulseAudio; if not, write to the Free Software
  18   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  19   USA.
  20 ***/
  21
  22 #ifdef HAVE_CONFIG_H
  23 #include <config.h>
  24 #endif
  25
  26 #include <assert.h>
  27 #include <stdio.h>
  28 #include <unistd.h>
  29
  30 #include <pulsecore/memblock.h>
  31 #include <pulse/xmalloc.h>
  32
  33 static void release_cb(pa_memimport *i, uint32_t block_id, void *userdata) {
  34     printf("%s: Imported block %u is released.\n", (char*) userdata, block_id);
  35 }
  36
  37 static void revoke_cb(pa_memexport *e, uint32_t block_id, void *userdata) {
  38     printf("%s: Exported block %u is revoked.\n", (char*) userdata, block_id);
  39 }
  40
  41 static void print_stats(pa_mempool *p, const char *text) {
  42     const pa_mempool_stat*s = pa_mempool_get_stat(p);
  43
  44     printf("%s = {\n"
  45            "n_allocated = %u\n"
  46            "n_accumulated = %u\n"
  47            "n_imported = %u\n"
  48            "n_exported = %u\n"
  49            "allocated_size = %u\n"
  50            "accumulated_size = %u\n"
  51            "imported_size = %u\n"
  52            "exported_size = %u\n"
  53            "n_too_large_for_pool = %u\n"
  54            "n_pool_full = %u\n"
  55            "}\n",
  56            text,
  57            (unsigned) pa_atomic_load(&s->n_allocated),
  58            (unsigned) pa_atomic_load(&s->n_accumulated),
  59            (unsigned) pa_atomic_load(&s->n_imported),
  60            (unsigned) pa_atomic_load(&s->n_exported),
  61            (unsigned) pa_atomic_load(&s->allocated_size),
  62            (unsigned) pa_atomic_load(&s->accumulated_size),
  63            (unsigned) pa_atomic_load(&s->imported_size),
  64            (unsigned) pa_atomic_load(&s->exported_size),
  65            (unsigned) pa_atomic_load(&s->n_too_large_for_pool),
  66            (unsigned) pa_atomic_load(&s->n_pool_full));
  67 }
  68
  69 int main(int argc, char *argv[]) {
  70     pa_mempool *pool_a, *pool_b, *pool_c;
  71     unsigned id_a, id_b, id_c;
  72     pa_memexport *export_a, *export_b;
  73     pa_memimport *import_b, *import_c;
  74     pa_memblock *mb_a, *mb_b, *mb_c;
  75     int r, i;
  76     pa_memblock* blocks[5];
  77     uint32_t id, shm_id;
  78     size_t offset, size;
  79
  80     const char txt[] = "This is a test!";
  81
  82     pool_a = pa_mempool_new(1);
  83     pool_b = pa_mempool_new(1);
  84     pool_c = pa_mempool_new(1);
  85
  86     pa_mempool_get_shm_id(pool_a, &id_a);
  87     pa_mempool_get_shm_id(pool_b, &id_b);
  88     pa_mempool_get_shm_id(pool_c, &id_c);
  89
  90     assert(pool_a && pool_b && pool_c);
  91
  92     blocks[0] = pa_memblock_new_fixed(pool_a, (void*) txt, sizeof(txt), 1);
  93     blocks[1] = pa_memblock_new(pool_a, sizeof(txt));
  94     snprintf(blocks[1]->data, blocks[1]->length, "%s", txt);
  95     blocks[2] = pa_memblock_new_pool(pool_a, sizeof(txt));
  96     snprintf(blocks[2]->data, blocks[2]->length, "%s", txt);
  97     blocks[3] = pa_memblock_new_malloced(pool_a, pa_xstrdup(txt), sizeof(txt));
  98     blocks[4] = NULL;
  99
 100     for (i = 0; blocks[i]; i++) {
 101         printf("Memory block %u\n", i);
 102
 103         mb_a = blocks[i];
 104         assert(mb_a);
 105
 106         export_a = pa_memexport_new(pool_a, revoke_cb, (void*) "A");
 107         export_b = pa_memexport_new(pool_b, revoke_cb, (void*) "B");
 108
 109         assert(export_a && export_b);
 110
 111         import_b = pa_memimport_new(pool_b, release_cb, (void*) "B");
 112         import_c = pa_memimport_new(pool_c, release_cb, (void*) "C");
 113
 114         assert(import_b && import_c);
 115
 116         r = pa_memexport_put(export_a, mb_a, &id, &shm_id, &offset, &size);
 117         assert(r >= 0);
 118         assert(shm_id == id_a);
 119
 120         printf("A: Memory block exported as %u\n", id);
 121
 122         mb_b = pa_memimport_get(import_b, id, shm_id, offset, size);
 123         assert(mb_b);
 124         r = pa_memexport_put(export_b, mb_b, &id, &shm_id, &offset, &size);
 125         assert(r >= 0);
 126         assert(shm_id == id_a || shm_id == id_b);
 127         pa_memblock_unref(mb_b);
 128
 129         printf("B: Memory block exported as %u\n", id);
 130
 131         mb_c = pa_memimport_get(import_c, id, shm_id, offset, size);
 132         assert(mb_c);
 133         printf("1 data=%s\n", (char*) mb_c->data);
 134
 135         print_stats(pool_a, "A");
 136         print_stats(pool_b, "B");
 137         print_stats(pool_c, "C");
 138
 139         pa_memexport_free(export_b);
 140         printf("2 data=%s\n", (char*) mb_c->data);
 141         pa_memblock_unref(mb_c);
 142
 143         pa_memimport_free(import_b);
 144
 145         pa_memblock_unref(mb_a);
 146
 147         pa_memimport_free(import_c);
 148         pa_memexport_free(export_a);
 149     }
 150
 151     printf("vaccuuming...\n");
 152
 153     pa_mempool_vacuum(pool_a);
 154     pa_mempool_vacuum(pool_b);
 155     pa_mempool_vacuum(pool_c);
 156
 157     printf("vaccuuming done...\n");
 158
 159     pa_mempool_free(pool_a);
 160     pa_mempool_free(pool_b);
 161     pa_mempool_free(pool_c);
 162
 163     return 0;
 164 }