Changeset 22942 in vbox for trunk/src/VBox/Devices/Network/slirp

Timestamp:

Sep 11, 2009 6:29:10 AM (15 years ago)

Author:

vboxsync

svn:sync-xref-src-repo-rev:

52207

Message:

NAT: bsd mbuf related changeset (misc.c)

File:

• trunk/src/VBox/Devices/Network/slirp/misc.c (modified) (1 diff)

trunk/src/VBox/Devices/Network/slirp/misc.c

@@ -115 +115 @@
 }
 
+#ifdef VBOX_WITH_SLIRP_BSD_MBUF
+#define ITEM_MAGIC 0xdead0001
+struct item 
+{
+    uint32_t magic;
+    uma_zone_t zone;
+    uint32_t ref_count;
+    LIST_ENTRY(item) list;
+};
+
+#define ZONE_MAGIC 0xdead0002
+struct uma_zone
+{
+    uint32_t magic;
+    PNATState pData; /* to minimize changes in the rest of UMA emulation code */
+    const char *name;
+    size_t size; /* item size */
+    ctor_t pfCtor;
+    dtor_t pfDtor;
+    zinit_t pfInit;
+    zfini_t pfFini;
+    uma_alloc_t pfAlloc;
+    uma_free_t pfFree;
+    int max_items;
+    int cur_items;
+    LIST_HEAD(RT_NOTHING, item) used_items;
+    LIST_HEAD(RT_NOTHING, item) free_items;
+    uma_zone_t master_zone;
+};
+
+
+static void *slirp_uma_alloc(uma_zone_t zone, 
+    int size, uint8_t *pflags, int wait)
+{
+    struct item *it;
+    if (    (zone->max_items != 0 && zone->cur_items >= zone->max_items)
+        || (zone->max_items == 0 && !LIST_EMPTY(&zone->free_items))
+    )
+    {
+        /*
+         * @todo (r=vvl) here should be some 
+         * accounting of extra items in case 
+         * breakthrough barrier
+         */
+        if (LIST_EMPTY(&zone->free_items))
+            return NULL; 
+        it = LIST_FIRST(&zone->free_items);
+        LIST_REMOVE(it, list); 
+        LIST_INSERT_HEAD(&zone->used_items, it, list);
+        goto allocated;
+    }
+    /*@todo 'Z' should be depend on flag */
+    it = RTMemAllocZ(sizeof(struct item) + zone->size);
+    if (it == NULL)
+    {
+        Log(("NAT: uma no memory"));
+        return NULL;
+    }
+    it->magic = ITEM_MAGIC;
+    LIST_INSERT_HEAD(&zone->used_items, it, list);
+    zone->cur_items++;
+    it->zone = zone;
+
+    allocated:
+    if (zone->pfInit)
+        zone->pfInit(zone->pData, (void *)&it[1], zone->size, M_DONTWAIT);
+    return (void *)&it[1];
+}
+
+static void slirp_uma_free(void *item, int size, uint8_t flags)
+{
+    struct item *it;
+    uma_zone_t zone;
+    Assert(item); 
+    it = &((struct item *)item)[-1];
+    Assert(it->magic == ITEM_MAGIC);
+    zone = it->zone;
+    Assert(zone->magic == ZONE_MAGIC);
+    LIST_REMOVE(it, list); 
+    LIST_INSERT_HEAD(&zone->free_items, it, list);
+}
+
+uma_zone_t uma_zcreate(PNATState pData, char *name, size_t size, 
+    ctor_t ctor, dtor_t dtor, zinit_t init, zfini_t fini, int flags1, int flags2)
+{
+    uma_zone_t zone = RTMemAllocZ(sizeof(struct uma_zone) + size);
+    Assert((pData));
+    zone->magic = ZONE_MAGIC;
+    zone->pData = pData;
+    zone->name = name;
+    zone->size = size;
+    zone->pfCtor = ctor;
+    zone->pfDtor = dtor;
+    zone->pfInit = init;
+    zone->pfFini = fini;
+    zone->pfAlloc = slirp_uma_alloc;
+    zone->pfFree = slirp_uma_free;
+    return zone;
+    
+}
+uma_zone_t uma_zsecond_create(char *name, ctor_t ctor, 
+    dtor_t dtor, zinit_t init, zfini_t fini, uma_zone_t master)
+{
+    uma_zone_t zone;
+#if 0
+    if (master->pfAlloc != NULL) 
+        zone = (uma_zone_t)master->pfAlloc(master, sizeof(struct uma_zone), NULL, 0);
+#endif        
+    zone = RTMemAllocZ(sizeof(struct uma_zone));
+    if (zone == NULL) 
+    {
+        return NULL;
+    }
+    Assert((master && master->pData));
+    zone->magic = ZONE_MAGIC;
+    zone->pData = master->pData;
+    zone->name = name;
+    zone->pfCtor = ctor;
+    zone->pfDtor = dtor;
+    zone->pfInit = init;
+    zone->pfFini = fini;
+    zone->pfAlloc = slirp_uma_alloc;
+    zone->pfFree = slirp_uma_free;
+    return zone;
+}
+void uma_zone_set_max(uma_zone_t zone, int max)
+{
+    zone->max_items = max;
+}
+void uma_zone_set_allocf(uma_zone_t zone, uma_alloc_t pfAlloc)
+{
+   zone->pfAlloc = pfAlloc; 
+}
+void uma_zone_set_freef(uma_zone_t zone, uma_free_t pfFree)
+{
+   zone->pfFree = pfFree; 
+}
+
+uint32_t *uma_find_refcnt(uma_zone_t zone, void *mem)
+{
+    /*@todo (r-vvl) this function supposed to work with special zone storing 
+    reference counters */ 
+    struct item *it = (struct item *)mem; /* 1st element */
+    Assert(zone->magic == ZONE_MAGIC);
+    /* for returning pointer to counter we need get 0 elemnt */
+    Assert(it[-1].magic == ITEM_MAGIC);
+    return &it[-1].ref_count;
+}
+void *uma_zalloc_arg(uma_zone_t zone, void *args, int how)
+{
+    void *mem;
+    Assert(zone->magic == ZONE_MAGIC);
+    if (zone->pfAlloc == NULL)
+        return NULL;
+    mem = zone->pfAlloc(zone, zone->size, NULL, 0);
+    if (zone->pfCtor)
+        zone->pfCtor(zone->pData, mem, zone->size, args, M_DONTWAIT);
+    return mem;
+}
+
+void uma_zfree(uma_zone_t zone, void *item)
+{
+    uma_zfree_arg(zone, item, NULL);
+}
+
+void uma_zfree_arg(uma_zone_t zone, void *mem, void *flags)
+{
+    struct item *it;
+    Assert(zone->magic == ZONE_MAGIC);
+    if (zone->pfFree == NULL)
+        return;
+    Assert((mem));
+    it = &((struct item *)mem)[-1]; 
+    if (it->magic != ITEM_MAGIC)
+    {
+        Log(("NAT:UMA: %p seems to be allocated on heap ... freeing\n", mem));
+        RTMemFree(mem);
+        return;
+    }
+    Assert((zone->magic == ZONE_MAGIC && zone == it->zone));
+     
+    if (zone->pfDtor)
+        zone->pfDtor(zone->pData, mem, zone->size, flags);
+    zone->pfFree(mem,  0, 0);
+}
+int uma_zone_exhausted_nolock(uma_zone_t zone)
+{
+    return 0;
+}
+void zone_drain(uma_zone_t zone)
+{
+    
+}
+
+void slirp_null_arg_free(void *mem, void *arg)
+{
+    /*@todo (r=vvl) make it wiser*/
+    Assert(mem);
+    RTMemFree(mem);
+}
+
+void *uma_zalloc(uma_zone_t zone, int len)
+{
+    return NULL;
+}
+#endif