hash.c 27.3 KB
Newer Older
D
Daniel Veillard 已提交
1
/*
2
 * hash.c: chained hash tables for domain and domain/connection deallocations
D
Daniel Veillard 已提交
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
 *
 * Reference: Your favorite introductory book on algorithms
 *
 * Copyright (C) 2000 Bjorn Reese and Daniel Veillard.
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND WITHOUT ANY EXPRESS OR IMPLIED
 * WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED WARRANTIES OF
 * MERCHANTIBILITY AND FITNESS FOR A PARTICULAR PURPOSE. THE AUTHORS AND
 * CONTRIBUTORS ACCEPT NO RESPONSIBILITY IN ANY CONCEIVABLE MANNER.
 *
 * Author: breese@users.sourceforge.net
18
 *         Daniel Veillard <veillard@redhat.com>
D
Daniel Veillard 已提交
19 20
 */

J
Jim Meyering 已提交
21 22
#include "config.h"

D
Daniel Veillard 已提交
23
#include <string.h>
24
#include <stdlib.h>
25 26
#include <libxml/threads.h>
#include "internal.h"
D
Daniel Veillard 已提交
27
#include "hash.h"
28
#include <pthread.h>
D
Daniel Veillard 已提交
29 30 31

#define MAX_HASH_LEN 8

32 33 34
#define DEBUG(fmt,...) VIR_DEBUG(__FILE__, fmt, __VA_ARGS__)
#define DEBUG0(msg) VIR_DEBUG(__FILE__, "%s", msg)

D
Daniel Veillard 已提交
35 36 37 38 39
/* #define DEBUG_GROW */

/*
 * A single entry in the hash table
 */
40 41 42 43
typedef struct _virHashEntry virHashEntry;
typedef virHashEntry *virHashEntryPtr;
struct _virHashEntry {
    struct _virHashEntry *next;
D
Daniel Veillard 已提交
44 45 46 47 48 49 50 51
    char *name;
    void *payload;
    int valid;
};

/*
 * The entire hash table
 */
52 53
struct _virHashTable {
    struct _virHashEntry *table;
D
Daniel Veillard 已提交
54 55 56 57 58
    int size;
    int nbElems;
};

/*
59
 * virHashComputeKey:
D
Daniel Veillard 已提交
60 61 62
 * Calculate the hash key
 */
static unsigned long
63 64
virHashComputeKey(virHashTablePtr table, const char *name)
{
D
Daniel Veillard 已提交
65 66
    unsigned long value = 0L;
    char ch;
67

D
Daniel Veillard 已提交
68
    if (name != NULL) {
69 70 71 72 73
        value += 30 * (*name);
        while ((ch = *name++) != 0) {
            value =
                value ^ ((value << 5) + (value >> 3) + (unsigned long) ch);
        }
D
Daniel Veillard 已提交
74 75 76 77 78
    }
    return (value % table->size);
}

/**
79
 * virHashCreate:
D
Daniel Veillard 已提交
80 81
 * @size: the size of the hash table
 *
82
 * Create a new virHashTablePtr.
D
Daniel Veillard 已提交
83 84 85
 *
 * Returns the newly created object, or NULL if an error occured.
 */
86
virHashTablePtr
87 88
virHashCreate(int size)
{
89
    virHashTablePtr table;
90

D
Daniel Veillard 已提交
91 92
    if (size <= 0)
        size = 256;
93

94
    table = malloc(sizeof(*table));
D
Daniel Veillard 已提交
95 96
    if (table) {
        table->size = size;
97
        table->nbElems = 0;
98
        table->table = calloc(1, size * sizeof(*(table->table)));
D
Daniel Veillard 已提交
99
        if (table->table) {
100
            return (table);
D
Daniel Veillard 已提交
101 102 103
        }
        free(table);
    }
104
    return (NULL);
D
Daniel Veillard 已提交
105 106 107
}

/**
108
 * virHashGrow:
D
Daniel Veillard 已提交
109 110 111 112 113 114 115 116
 * @table: the hash table
 * @size: the new size of the hash table
 *
 * resize the hash table
 *
 * Returns 0 in case of success, -1 in case of failure
 */
static int
117 118
virHashGrow(virHashTablePtr table, int size)
{
D
Daniel Veillard 已提交
119 120
    unsigned long key;
    int oldsize, i;
121 122
    virHashEntryPtr iter, next;
    struct _virHashEntry *oldtable;
123

D
Daniel Veillard 已提交
124 125 126
#ifdef DEBUG_GROW
    unsigned long nbElem = 0;
#endif
127

D
Daniel Veillard 已提交
128
    if (table == NULL)
129
        return (-1);
D
Daniel Veillard 已提交
130
    if (size < 8)
131
        return (-1);
D
Daniel Veillard 已提交
132
    if (size > 8 * 2048)
133
        return (-1);
D
Daniel Veillard 已提交
134 135 136 137

    oldsize = table->size;
    oldtable = table->table;
    if (oldtable == NULL)
138 139
        return (-1);

140
    table->table = calloc(1, size * sizeof(*(table->table)));
D
Daniel Veillard 已提交
141
    if (table->table == NULL) {
142 143
        table->table = oldtable;
        return (-1);
D
Daniel Veillard 已提交
144 145 146
    }
    table->size = size;

147 148 149 150 151 152
    /*  If the two loops are merged, there would be situations where
     * a new entry needs to allocated and data copied into it from 
     * the main table. So instead, we run through the array twice, first
     * copying all the elements in the main array (where we can't get
     * conflicts) and then the rest, so we only free (and don't allocate)
     */
D
Daniel Veillard 已提交
153
    for (i = 0; i < oldsize; i++) {
154 155 156 157 158
        if (oldtable[i].valid == 0)
            continue;
        key = virHashComputeKey(table, oldtable[i].name);
        memcpy(&(table->table[key]), &(oldtable[i]), sizeof(virHashEntry));
        table->table[key].next = NULL;
D
Daniel Veillard 已提交
159 160 161
    }

    for (i = 0; i < oldsize; i++) {
162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178
        iter = oldtable[i].next;
        while (iter) {
            next = iter->next;

            /*
             * put back the entry in the new table
             */

            key = virHashComputeKey(table, iter->name);
            if (table->table[key].valid == 0) {
                memcpy(&(table->table[key]), iter, sizeof(virHashEntry));
                table->table[key].next = NULL;
                free(iter);
            } else {
                iter->next = table->table[key].next;
                table->table[key].next = iter;
            }
D
Daniel Veillard 已提交
179 180

#ifdef DEBUG_GROW
181
            nbElem++;
D
Daniel Veillard 已提交
182 183
#endif

184 185
            iter = next;
        }
D
Daniel Veillard 已提交
186 187 188 189 190 191
    }

    free(oldtable);

#ifdef DEBUG_GROW
    xmlGenericError(xmlGenericErrorContext,
192 193
                    "virHashGrow : from %d to %d, %d elems\n", oldsize,
                    size, nbElem);
D
Daniel Veillard 已提交
194 195
#endif

196
    return (0);
D
Daniel Veillard 已提交
197 198 199
}

/**
200
 * virHashFree:
D
Daniel Veillard 已提交
201 202 203 204 205 206 207
 * @table: the hash table
 * @f:  the deallocator function for items in the hash
 *
 * Free the hash @table and its contents. The userdata is
 * deallocated with @f if provided.
 */
void
208 209
virHashFree(virHashTablePtr table, virHashDeallocator f)
{
D
Daniel Veillard 已提交
210
    int i;
211 212
    virHashEntryPtr iter;
    virHashEntryPtr next;
D
Daniel Veillard 已提交
213 214 215 216
    int inside_table = 0;
    int nbElems;

    if (table == NULL)
217
        return;
D
Daniel Veillard 已提交
218
    if (table->table) {
219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240
        nbElems = table->nbElems;
        for (i = 0; (i < table->size) && (nbElems > 0); i++) {
            iter = &(table->table[i]);
            if (iter->valid == 0)
                continue;
            inside_table = 1;
            while (iter) {
                next = iter->next;
                if ((f != NULL) && (iter->payload != NULL))
                    f(iter->payload, iter->name);
                if (iter->name)
                    free(iter->name);
                iter->payload = NULL;
                if (!inside_table)
                    free(iter);
                nbElems--;
                inside_table = 0;
                iter = next;
            }
            inside_table = 0;
        }
        free(table->table);
D
Daniel Veillard 已提交
241 242 243 244 245
    }
    free(table);
}

/**
246
 * virHashAddEntry3:
D
Daniel Veillard 已提交
247 248 249 250 251 252 253 254 255 256
 * @table: the hash table
 * @name: the name of the userdata
 * @userdata: a pointer to the userdata
 *
 * Add the @userdata to the hash @table. This can later be retrieved
 * by using @name. Duplicate entries generate errors.
 *
 * Returns 0 the addition succeeded and -1 in case of error.
 */
int
257 258
virHashAddEntry(virHashTablePtr table, const char *name, void *userdata)
{
D
Daniel Veillard 已提交
259
    unsigned long key, len = 0;
260 261
    virHashEntryPtr entry;
    virHashEntryPtr insert;
D
Daniel Veillard 已提交
262 263

    if ((table == NULL) || (name == NULL))
264
        return (-1);
D
Daniel Veillard 已提交
265 266 267 268

    /*
     * Check for duplicate and insertion location.
     */
269
    key = virHashComputeKey(table, name);
D
Daniel Veillard 已提交
270
    if (table->table[key].valid == 0) {
271
        insert = NULL;
D
Daniel Veillard 已提交
272
    } else {
273 274 275 276 277 278 279 280
        for (insert = &(table->table[key]); insert->next != NULL;
             insert = insert->next) {
            if (!strcmp(insert->name, name))
                return (-1);
            len++;
        }
        if (!strcmp(insert->name, name))
            return (-1);
D
Daniel Veillard 已提交
281 282 283
    }

    if (insert == NULL) {
284
        entry = &(table->table[key]);
D
Daniel Veillard 已提交
285
    } else {
286
        entry = malloc(sizeof(*entry));
287 288
        if (entry == NULL)
            return (-1);
D
Daniel Veillard 已提交
289 290 291 292 293 294 295 296
    }

    entry->name = strdup(name);
    entry->payload = userdata;
    entry->next = NULL;
    entry->valid = 1;


297 298
    if (insert != NULL)
        insert->next = entry;
D
Daniel Veillard 已提交
299 300 301 302

    table->nbElems++;

    if (len > MAX_HASH_LEN)
303
        virHashGrow(table, MAX_HASH_LEN * table->size);
D
Daniel Veillard 已提交
304

305
    return (0);
D
Daniel Veillard 已提交
306 307 308
}

/**
309
 * virHashUpdateEntry:
D
Daniel Veillard 已提交
310 311 312 313 314 315 316 317 318 319 320 321
 * @table: the hash table
 * @name: the name of the userdata
 * @userdata: a pointer to the userdata
 * @f: the deallocator function for replaced item (if any)
 *
 * Add the @userdata to the hash @table. This can later be retrieved
 * by using @name. Existing entry for this tuple
 * will be removed and freed with @f if found.
 *
 * Returns 0 the addition succeeded and -1 in case of error.
 */
int
322
virHashUpdateEntry(virHashTablePtr table, const char *name,
323 324
                   void *userdata, virHashDeallocator f)
{
D
Daniel Veillard 已提交
325
    unsigned long key;
326 327
    virHashEntryPtr entry;
    virHashEntryPtr insert;
D
Daniel Veillard 已提交
328 329

    if ((table == NULL) || name == NULL)
330
        return (-1);
D
Daniel Veillard 已提交
331 332 333 334

    /*
     * Check for duplicate and insertion location.
     */
335
    key = virHashComputeKey(table, name);
D
Daniel Veillard 已提交
336
    if (table->table[key].valid == 0) {
337
        insert = NULL;
D
Daniel Veillard 已提交
338
    } else {
339 340 341 342 343 344 345 346 347 348 349 350 351 352 353
        for (insert = &(table->table[key]); insert->next != NULL;
             insert = insert->next) {
            if (!strcmp(insert->name, name)) {
                if (f)
                    f(insert->payload, insert->name);
                insert->payload = userdata;
                return (0);
            }
        }
        if (!strcmp(insert->name, name)) {
            if (f)
                f(insert->payload, insert->name);
            insert->payload = userdata;
            return (0);
        }
D
Daniel Veillard 已提交
354 355 356
    }

    if (insert == NULL) {
357
        entry = &(table->table[key]);
D
Daniel Veillard 已提交
358
    } else {
359
        entry = malloc(sizeof(*entry));
360 361
        if (entry == NULL)
            return (-1);
D
Daniel Veillard 已提交
362 363 364 365 366 367 368 369 370 371
    }

    entry->name = strdup(name);
    entry->payload = userdata;
    entry->next = NULL;
    entry->valid = 1;
    table->nbElems++;


    if (insert != NULL) {
372
        insert->next = entry;
D
Daniel Veillard 已提交
373
    }
374
    return (0);
D
Daniel Veillard 已提交
375 376 377
}

/**
378
 * virHashLookup:
D
Daniel Veillard 已提交
379 380 381 382 383 384 385 386
 * @table: the hash table
 * @name: the name of the userdata
 *
 * Find the userdata specified by the (@name, @name2, @name3) tuple.
 *
 * Returns the a pointer to the userdata
 */
void *
387 388
virHashLookup(virHashTablePtr table, const char *name)
{
D
Daniel Veillard 已提交
389
    unsigned long key;
390
    virHashEntryPtr entry;
D
Daniel Veillard 已提交
391 392

    if (table == NULL)
393
        return (NULL);
D
Daniel Veillard 已提交
394
    if (name == NULL)
395
        return (NULL);
396
    key = virHashComputeKey(table, name);
D
Daniel Veillard 已提交
397
    if (table->table[key].valid == 0)
398
        return (NULL);
D
Daniel Veillard 已提交
399
    for (entry = &(table->table[key]); entry != NULL; entry = entry->next) {
400 401
        if (!strcmp(entry->name, name))
            return (entry->payload);
D
Daniel Veillard 已提交
402
    }
403
    return (NULL);
D
Daniel Veillard 已提交
404 405 406
}

/**
407
 * virHashSize:
D
Daniel Veillard 已提交
408 409 410 411 412 413 414 415
 * @table: the hash table
 *
 * Query the number of elements installed in the hash @table.
 *
 * Returns the number of elements in the hash table or
 * -1 in case of error
 */
int
416 417
virHashSize(virHashTablePtr table)
{
D
Daniel Veillard 已提交
418
    if (table == NULL)
419 420
        return (-1);
    return (table->nbElems);
D
Daniel Veillard 已提交
421 422 423
}

/**
424
 * virHashRemoveEntry:
D
Daniel Veillard 已提交
425 426 427 428 429 430 431 432 433 434 435
 * @table: the hash table
 * @name: the name of the userdata
 * @f: the deallocator function for removed item (if any)
 *
 * Find the userdata specified by the @name and remove
 * it from the hash @table. Existing userdata for this tuple will be removed
 * and freed with @f.
 *
 * Returns 0 if the removal succeeded and -1 in case of error or not found.
 */
int
436
virHashRemoveEntry(virHashTablePtr table, const char *name,
437 438
                   virHashDeallocator f)
{
D
Daniel Veillard 已提交
439
    unsigned long key;
440 441
    virHashEntryPtr entry;
    virHashEntryPtr prev = NULL;
D
Daniel Veillard 已提交
442 443

    if (table == NULL || name == NULL)
444
        return (-1);
D
Daniel Veillard 已提交
445

446
    key = virHashComputeKey(table, name);
D
Daniel Veillard 已提交
447
    if (table->table[key].valid == 0) {
448
        return (-1);
D
Daniel Veillard 已提交
449
    } else {
450 451
        for (entry = &(table->table[key]); entry != NULL;
             entry = entry->next) {
D
Daniel Veillard 已提交
452 453 454 455
            if (!strcmp(entry->name, name)) {
                if ((f != NULL) && (entry->payload != NULL))
                    f(entry->payload, entry->name);
                entry->payload = NULL;
456 457 458
                if (entry->name)
                    free(entry->name);
                if (prev) {
D
Daniel Veillard 已提交
459
                    prev->next = entry->next;
460 461 462 463 464 465 466 467 468 469 470
                    free(entry);
                } else {
                    if (entry->next == NULL) {
                        entry->valid = 0;
                    } else {
                        entry = entry->next;
                        memcpy(&(table->table[key]), entry,
                               sizeof(virHashEntry));
                        free(entry);
                    }
                }
D
Daniel Veillard 已提交
471
                table->nbElems--;
472
                return (0);
D
Daniel Veillard 已提交
473 474 475
            }
            prev = entry;
        }
476
        return (-1);
D
Daniel Veillard 已提交
477 478
    }
}
479

480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600

/**
 * virHashForEach
 * @table: the hash table to process
 * @iter: callback to process each element
 * @data: opaque data to pass to the iterator
 *
 * Iterates over every element in the hash table, invoking the
 * 'iter' callback. The callback must not call any other virHash*
 * functions, and in particular must not attempt to remove the
 * element.
 *
 * Returns number of items iterated over upon completion, -1 on failure
 */
int virHashForEach(virHashTablePtr table, virHashIterator iter, const void *data) {
    int i, count = 0;

    if (table == NULL || iter == NULL)
        return (-1);

    for (i = 0 ; i < table->size ; i++) {
        virHashEntryPtr entry = table->table + i;
        while (entry) {
            if (entry->valid) {
                iter(entry->payload, entry->name, data);
                count++;
            }
            entry = entry->next;
        }
    }
    return (count);
}

/**
 * virHashRemoveSet
 * @table: the hash table to process
 * @iter: callback to identify elements for removal
 * @f: callback to free memory from element payload
 * @data: opaque data to pass to the iterator
 *
 * Iterates over all elements in the hash table, invoking the 'iter'
 * callback. If the callback returns a non-zero value, the element
 * will be removed from the hash table & its payload passed to the
 * callback 'f' for de-allocation.
 *
 * Returns number of items removed on success, -1 on failure
 */
int virHashRemoveSet(virHashTablePtr table, virHashSearcher iter, virHashDeallocator f, const void *data) {
    int i, count = 0;

    if (table == NULL || iter == NULL)
        return (-1);

    for (i = 0 ; i < table->size ; i++) {
        virHashEntryPtr prev = NULL;
        virHashEntryPtr entry = &(table->table[i]);

        while (entry && entry->valid) {
            if (iter(entry->payload, entry->name, data)) {
                count++;
                f(entry->payload, entry->name);
                if (entry->name)
                    free(entry->name);
                if (prev) {
                    prev->next = entry->next;
                    free(entry);
                } else {
                    if (entry->next == NULL) {
                        entry->valid = 0;
                        entry->name = NULL;
                    } else {
                        entry = entry->next;
                        memcpy(&(table->table[i]), entry,
                               sizeof(virHashEntry));
                        free(entry);
                        entry = NULL;
                    }
                }
                table->nbElems--;
            }
            prev = entry;
            if (entry) {
                entry = entry->next;
            } else {
                entry = NULL;
            }
        }
    }
    return (count);
}

/**
 * virHashSearch:
 * @table: the hash table to search
 * @iter: an iterator to identify the desired element
 * @data: extra opaque information passed to the iter
 *
 * Iterates over the hash table calling the 'iter' callback
 * for each element. The first element for which the iter
 * returns non-zero will be returned by this function.
 * The elements are processed in a undefined order
 */
void *virHashSearch(virHashTablePtr table, virHashSearcher iter, const void *data) {
    int i;

    if (table == NULL || iter == NULL)
        return (NULL);

    for (i = 0 ; i < table->size ; i++) {
        virHashEntryPtr entry = table->table + i;
        while (entry) {
            if (entry->valid) {
                if (iter(entry->payload, entry->name, data))
                    return entry->payload;
            }
            entry = entry->next;
        }
    }
    return (NULL);
}

601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623
/************************************************************************
 *									*
 *			Domain and Connections allocations		*
 *									*
 ************************************************************************/

/**
 * virHashError:
 * @conn: the connection if available
 * @error: the error noumber
 * @info: extra information string
 *
 * Handle an error at the connection level
 */
static void
virHashError(virConnectPtr conn, virErrorNumber error, const char *info)
{
    const char *errmsg;

    if (error == VIR_ERR_OK)
        return;

    errmsg = __virErrorMsg(error, info);
624
    __virRaiseError(conn, NULL, NULL, VIR_FROM_NONE, error, VIR_ERR_ERROR,
625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642
                    errmsg, info, NULL, 0, 0, errmsg, info);
}


/**
 * virDomainFreeName:
 * @domain: a domain object
 *
 * Destroy the domain object, this is just used by the domain hash callback.
 *
 * Returns 0 in case of success and -1 in case of failure.
 */
static int
virDomainFreeName(virDomainPtr domain, const char *name ATTRIBUTE_UNUSED)
{
    return (virDomainFree(domain));
}

643 644 645 646 647 648 649 650 651 652 653 654 655 656
/**
 * virNetworkFreeName:
 * @network: a network object
 *
 * Destroy the network object, this is just used by the network hash callback.
 *
 * Returns 0 in case of success and -1 in case of failure.
 */
static int
virNetworkFreeName(virNetworkPtr network, const char *name ATTRIBUTE_UNUSED)
{
    return (virNetworkFree(network));
}

657 658 659 660 661 662 663 664 665 666 667
/**
 * virGetConnect:
 *
 * Allocates a new hypervisor connection structure
 *
 * Returns a new pointer or NULL in case of error.
 */
virConnectPtr
virGetConnect(void) {
    virConnectPtr ret;

668
    ret = calloc(1, sizeof(*ret));
669
    if (ret == NULL) {
670
        virHashError(NULL, VIR_ERR_NO_MEMORY, _("allocating connection"));
671 672 673
        goto failed;
    }
    ret->magic = VIR_CONNECT_MAGIC;
674 675 676 677
    ret->driver = NULL;
    ret->networkDriver = NULL;
    ret->privateData = NULL;
    ret->networkPrivateData = NULL;
678 679 680
    ret->domains = virHashCreate(20);
    if (ret->domains == NULL)
        goto failed;
681 682 683
    ret->networks = virHashCreate(20);
    if (ret->networks == NULL)
        goto failed;
684

685 686 687
    pthread_mutex_init(&ret->lock, NULL);

    ret->refs = 1;
688 689 690 691
    return(ret);

failed:
    if (ret != NULL) {
692 693 694 695 696 697
        if (ret->domains != NULL)
            virHashFree(ret->domains, (virHashDeallocator) virDomainFreeName);
        if (ret->networks != NULL)
            virHashFree(ret->networks, (virHashDeallocator) virNetworkFreeName);

        pthread_mutex_destroy(&ret->lock);
698 699 700 701 702 703
        free(ret);
    }
    return(NULL);
}

/**
704 705
 * virReleaseConnect:
 * @conn: the hypervisor connection to release
706
 *
707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731
 * Unconditionally release all memory associated with a connection.
 * The conn.lock mutex must be held prior to calling this, and will
 * be released prior to this returning. The connection obj must not
 * be used once this method returns.
 */
static void
virReleaseConnect(virConnectPtr conn) {
    DEBUG("release connection %p %s", conn, conn->name);
    if (conn->domains != NULL)
        virHashFree(conn->domains, (virHashDeallocator) virDomainFreeName);
    if (conn->networks != NULL)
        virHashFree(conn->networks, (virHashDeallocator) virNetworkFreeName);
    virResetError(&conn->err);
    free(conn->name);

    pthread_mutex_unlock(&conn->lock);
    pthread_mutex_destroy(&conn->lock);
    free(conn);
}

/**
 * virUnrefConnect:
 * @conn: the hypervisor connection to unreference
 *
 * Unreference the connection. If the use count drops to zero, the structure is
732 733 734 735
 * actually freed.
 *
 * Returns the reference count or -1 in case of failure.
 */
736 737 738
int
virUnrefConnect(virConnectPtr conn) {
    int refs;
739

740
    if ((!VIR_IS_CONNECT(conn))) {
741 742 743
        virHashError(conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
        return(-1);
    }
744 745 746 747 748 749 750 751
    pthread_mutex_lock(&conn->lock);
    DEBUG("unref connection %p %s %d", conn, conn->name, conn->refs);
    conn->refs--;
    refs = conn->refs;
    if (refs == 0) {
        virReleaseConnect(conn);
        /* Already unlocked mutex */
        return (0);
752
    }
753 754
    pthread_mutex_unlock(&conn->lock);
    return (refs);
755 756 757 758 759
}

/**
 * virGetDomain:
 * @conn: the hypervisor connection
760 761
 * @name: pointer to the domain name
 * @uuid: pointer to the uuid
762 763 764 765
 *
 * Lookup if the domain is already registered for that connection,
 * if yes return a new pointer to it, if no allocate a new structure,
 * and register it in the table. In any case a corresponding call to
766
 * virUnrefDomain() is needed to not leak data.
767 768 769 770
 *
 * Returns a pointer to the domain, or NULL in case of failure
 */
virDomainPtr
771
__virGetDomain(virConnectPtr conn, const char *name, const unsigned char *uuid) {
772 773
    virDomainPtr ret = NULL;

774
    if ((!VIR_IS_CONNECT(conn)) || (name == NULL) || (uuid == NULL)) {
775 776 777
        virHashError(conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
        return(NULL);
    }
778
    pthread_mutex_lock(&conn->lock);
779 780 781 782

    /* TODO search by UUID first as they are better differenciators */

    ret = (virDomainPtr) virHashLookup(conn->domains, name);
783
    /* TODO check the UUID */
784
    if (ret == NULL) {
785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806
        ret = (virDomainPtr) calloc(1, sizeof(*ret));
        if (ret == NULL) {
            virHashError(conn, VIR_ERR_NO_MEMORY, _("allocating domain"));
            goto error;
        }
        ret->name = strdup(name);
        if (ret->name == NULL) {
            virHashError(conn, VIR_ERR_NO_MEMORY, _("allocating domain"));
            goto error;
        }
        ret->magic = VIR_DOMAIN_MAGIC;
        ret->conn = conn;
        ret->id = -1;
        if (uuid != NULL)
            memcpy(&(ret->uuid[0]), uuid, VIR_UUID_BUFLEN);

        if (virHashAddEntry(conn->domains, name, ret) < 0) {
            virHashError(conn, VIR_ERR_INTERNAL_ERROR,
                         _("failed to add domain to connection hash table"));
            goto error;
        }
        conn->refs++;
807
    }
808 809
    ret->refs++;
    pthread_mutex_unlock(&conn->lock);
810 811
    return(ret);

812 813
 error:
    pthread_mutex_unlock(&conn->lock);
814
    if (ret != NULL) {
815 816 817
        if (ret->name != NULL)
            free(ret->name );
        free(ret);
818 819 820 821 822
    }
    return(NULL);
}

/**
823
 * virReleaseDomain:
824 825
 * @domain: the domain to release
 *
826 827 828 829
 * Unconditionally release all memory associated with a domain.
 * The conn.lock mutex must be held prior to calling this, and will
 * be released prior to this returning. The domain obj must not
 * be used once this method returns.
830
 *
831 832
 * It will also unreference the associated connection object,
 * which may also be released if its ref count hits zero.
833
 */
834 835 836 837
static void
virReleaseDomain(virDomainPtr domain) {
    virConnectPtr conn = domain->conn;
    DEBUG("release domain %p %s", domain, domain->name);
838 839

    /* TODO search by UUID first as they are better differenciators */
840
    if (virHashRemoveEntry(conn->domains, domain->name, NULL) < 0)
841
        virHashError(conn, VIR_ERR_INTERNAL_ERROR,
842 843
                     _("domain missing from connection hash table"));

844
    domain->magic = -1;
845
    domain->id = -1;
846
    free(domain->name);
847 848
    free(domain);

849 850 851 852 853 854 855
    DEBUG("unref connection %p %s %d", conn, conn->name, conn->refs);
    conn->refs--;
    if (conn->refs == 0) {
        virReleaseConnect(conn);
        /* Already unlocked mutex */
        return;
    }
856

857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889
    pthread_mutex_unlock(&conn->lock);
}


/**
 * virUnrefDomain:
 * @domain: the domain to unreference
 *
 * Unreference the domain. If the use count drops to zero, the structure is
 * actually freed.
 *
 * Returns the reference count or -1 in case of failure.
 */
int
virUnrefDomain(virDomainPtr domain) {
    int refs;

    if (!VIR_IS_CONNECTED_DOMAIN(domain)) {
        virHashError(domain->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
        return(-1);
    }
    pthread_mutex_lock(&domain->conn->lock);
    DEBUG("unref domain %p %s %d", domain, domain->name, domain->refs);
    domain->refs--;
    refs = domain->refs;
    if (refs == 0) {
        virReleaseDomain(domain);
        /* Already unlocked mutex */
        return (0);
    }

    pthread_mutex_lock(&domain->conn->lock);
    return (refs);
890 891
}

892 893 894 895 896 897 898 899 900
/**
 * virGetNetwork:
 * @conn: the hypervisor connection
 * @name: pointer to the network name
 * @uuid: pointer to the uuid
 *
 * Lookup if the network is already registered for that connection,
 * if yes return a new pointer to it, if no allocate a new structure,
 * and register it in the table. In any case a corresponding call to
901
 * virUnrefNetwork() is needed to not leak data.
902 903 904 905
 *
 * Returns a pointer to the network, or NULL in case of failure
 */
virNetworkPtr
906
__virGetNetwork(virConnectPtr conn, const char *name, const unsigned char *uuid) {
907 908
    virNetworkPtr ret = NULL;

909
    if ((!VIR_IS_CONNECT(conn)) || (name == NULL) || (uuid == NULL)) {
910 911 912
        virHashError(conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
        return(NULL);
    }
913
    pthread_mutex_lock(&conn->lock);
914 915 916 917

    /* TODO search by UUID first as they are better differenciators */

    ret = (virNetworkPtr) virHashLookup(conn->networks, name);
918
    /* TODO check the UUID */
919
    if (ret == NULL) {
920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940
        ret = (virNetworkPtr) calloc(1, sizeof(*ret));
        if (ret == NULL) {
            virHashError(conn, VIR_ERR_NO_MEMORY, _("allocating network"));
            goto error;
        }
        ret->name = strdup(name);
        if (ret->name == NULL) {
            virHashError(conn, VIR_ERR_NO_MEMORY, _("allocating network"));
            goto error;
        }
        ret->magic = VIR_NETWORK_MAGIC;
        ret->conn = conn;
        if (uuid != NULL)
            memcpy(&(ret->uuid[0]), uuid, VIR_UUID_BUFLEN);

        if (virHashAddEntry(conn->networks, name, ret) < 0) {
            virHashError(conn, VIR_ERR_INTERNAL_ERROR,
                         _("failed to add network to connection hash table"));
            goto error;
        }
        conn->refs++;
941
    }
942 943
    ret->refs++;
    pthread_mutex_unlock(&conn->lock);
944 945
    return(ret);

946 947
 error:
    pthread_mutex_unlock(&conn->lock);
948
    if (ret != NULL) {
949 950 951
        if (ret->name != NULL)
            free(ret->name );
        free(ret);
952 953 954 955 956
    }
    return(NULL);
}

/**
957
 * virReleaseNetwork:
958 959
 * @network: the network to release
 *
960 961 962 963
 * Unconditionally release all memory associated with a network.
 * The conn.lock mutex must be held prior to calling this, and will
 * be released prior to this returning. The network obj must not
 * be used once this method returns.
964
 *
965 966
 * It will also unreference the associated connection object,
 * which may also be released if its ref count hits zero.
967
 */
968 969 970 971
static void
virReleaseNetwork(virNetworkPtr network) {
    virConnectPtr conn = network->conn;
    DEBUG("release network %p %s", network, network->name);
972 973

    /* TODO search by UUID first as they are better differenciators */
974
    if (virHashRemoveEntry(conn->networks, network->name, NULL) < 0)
975
        virHashError(conn, VIR_ERR_INTERNAL_ERROR,
976 977
                     _("network missing from connection hash table"));

978
    network->magic = -1;
979
    free(network->name);
980 981
    free(network);

982 983 984 985 986 987 988
    DEBUG("unref connection %p %s %d", conn, conn->name, conn->refs);
    conn->refs--;
    if (conn->refs == 0) {
        virReleaseConnect(conn);
        /* Already unlocked mutex */
        return;
    }
989

990 991
    pthread_mutex_unlock(&conn->lock);
}
992

993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022

/**
 * virUnrefNetwork:
 * @network: the network to unreference
 *
 * Unreference the network. If the use count drops to zero, the structure is
 * actually freed.
 *
 * Returns the reference count or -1 in case of failure.
 */
int
virUnrefNetwork(virNetworkPtr network) {
    int refs;

    if (!VIR_IS_CONNECTED_NETWORK(network)) {
        virHashError(network->conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
        return(-1);
    }
    pthread_mutex_lock(&network->conn->lock);
    DEBUG("unref network %p %s %d", network, network->name, network->refs);
    network->refs--;
    refs = network->refs;
    if (refs == 0) {
        virReleaseNetwork(network);
        /* Already unlocked mutex */
        return (0);
    }

    pthread_mutex_lock(&network->conn->lock);
    return (refs);
1023 1024
}

1025 1026 1027 1028 1029
/*
 * vim: set tabstop=4:
 * vim: set shiftwidth=4:
 * vim: set expandtab:
 */
1030 1031 1032 1033 1034 1035 1036 1037
/*
 * Local variables:
 *  indent-tabs-mode: nil
 *  c-indent-level: 4
 *  c-basic-offset: 4
 *  tab-width: 4
 * End:
 */