hash.c 27.1 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
 */

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

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
    /*  If the two loops are merged, there would be situations where
148
     * a new entry needs to allocated and data copied into it from
149 150 151 152
     * 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
        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);
229
                free(iter->name);
230 231 232 233 234 235 236 237 238 239
                iter->payload = NULL;
                if (!inside_table)
                    free(iter);
                nbElems--;
                inside_table = 0;
                iter = next;
            }
            inside_table = 0;
        }
        free(table->table);
D
Daniel Veillard 已提交
240 241 242 243 244
    }
    free(table);
}

/**
245
 * virHashAddEntry3:
D
Daniel Veillard 已提交
246 247 248 249 250 251 252 253 254 255
 * @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
256 257
virHashAddEntry(virHashTablePtr table, const char *name, void *userdata)
{
D
Daniel Veillard 已提交
258
    unsigned long key, len = 0;
259 260
    virHashEntryPtr entry;
    virHashEntryPtr insert;
D
Daniel Veillard 已提交
261 262

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

    /*
     * Check for duplicate and insertion location.
     */
268
    key = virHashComputeKey(table, name);
D
Daniel Veillard 已提交
269
    if (table->table[key].valid == 0) {
270
        insert = NULL;
D
Daniel Veillard 已提交
271
    } else {
272 273 274 275 276 277 278 279
        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 已提交
280 281 282
    }

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

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


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

    table->nbElems++;

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

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

/**
308
 * virHashUpdateEntry:
D
Daniel Veillard 已提交
309 310 311 312 313 314 315 316 317 318 319 320
 * @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
321
virHashUpdateEntry(virHashTablePtr table, const char *name,
322 323
                   void *userdata, virHashDeallocator f)
{
D
Daniel Veillard 已提交
324
    unsigned long key;
325 326
    virHashEntryPtr entry;
    virHashEntryPtr insert;
D
Daniel Veillard 已提交
327 328

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

    /*
     * Check for duplicate and insertion location.
     */
334
    key = virHashComputeKey(table, name);
D
Daniel Veillard 已提交
335
    if (table->table[key].valid == 0) {
336
        insert = NULL;
D
Daniel Veillard 已提交
337
    } else {
338 339 340 341 342 343 344 345 346 347 348 349 350 351 352
        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 已提交
353 354 355
    }

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

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


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

/**
377
 * virHashLookup:
D
Daniel Veillard 已提交
378 379 380 381 382 383 384 385
 * @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 *
386 387
virHashLookup(virHashTablePtr table, const char *name)
{
D
Daniel Veillard 已提交
388
    unsigned long key;
389
    virHashEntryPtr entry;
D
Daniel Veillard 已提交
390 391

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

/**
406
 * virHashSize:
D
Daniel Veillard 已提交
407 408 409 410 411 412 413 414
 * @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
415 416
virHashSize(virHashTablePtr table)
{
D
Daniel Veillard 已提交
417
    if (table == NULL)
418 419
        return (-1);
    return (table->nbElems);
D
Daniel Veillard 已提交
420 421 422
}

/**
423
 * virHashRemoveEntry:
D
Daniel Veillard 已提交
424 425 426 427 428 429 430 431 432 433 434
 * @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
435
virHashRemoveEntry(virHashTablePtr table, const char *name,
436 437
                   virHashDeallocator f)
{
D
Daniel Veillard 已提交
438
    unsigned long key;
439 440
    virHashEntryPtr entry;
    virHashEntryPtr prev = NULL;
D
Daniel Veillard 已提交
441 442

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

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

/**
 * 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);
539
                free(entry->name);
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
                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);
}

598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620
/************************************************************************
 *									*
 *			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);
621
    __virRaiseError(conn, NULL, NULL, VIR_FROM_NONE, error, VIR_ERR_ERROR,
622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639
                    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));
}

640 641 642 643 644 645 646 647 648 649 650 651 652 653
/**
 * 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));
}

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

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

682 683 684
    pthread_mutex_init(&ret->lock, NULL);

    ret->refs = 1;
685 686 687 688
    return(ret);

failed:
    if (ret != NULL) {
689 690 691 692 693 694
        if (ret->domains != NULL)
            virHashFree(ret->domains, (virHashDeallocator) virDomainFreeName);
        if (ret->networks != NULL)
            virHashFree(ret->networks, (virHashDeallocator) virNetworkFreeName);

        pthread_mutex_destroy(&ret->lock);
695 696 697 698 699 700
        free(ret);
    }
    return(NULL);
}

/**
701 702
 * virReleaseConnect:
 * @conn: the hypervisor connection to release
703
 *
704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728
 * 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
729 730 731 732
 * actually freed.
 *
 * Returns the reference count or -1 in case of failure.
 */
733 734 735
int
virUnrefConnect(virConnectPtr conn) {
    int refs;
736

737
    if ((!VIR_IS_CONNECT(conn))) {
738 739 740
        virHashError(conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
        return(-1);
    }
741 742 743 744 745 746 747 748
    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);
749
    }
750 751
    pthread_mutex_unlock(&conn->lock);
    return (refs);
752 753 754 755 756
}

/**
 * virGetDomain:
 * @conn: the hypervisor connection
757 758
 * @name: pointer to the domain name
 * @uuid: pointer to the uuid
759 760 761 762
 *
 * 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
763
 * virUnrefDomain() is needed to not leak data.
764 765 766 767
 *
 * Returns a pointer to the domain, or NULL in case of failure
 */
virDomainPtr
768
__virGetDomain(virConnectPtr conn, const char *name, const unsigned char *uuid) {
769 770
    virDomainPtr ret = NULL;

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

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

    ret = (virDomainPtr) virHashLookup(conn->domains, name);
780
    /* TODO check the UUID */
781
    if (ret == NULL) {
782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803
        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++;
804
    }
805 806
    ret->refs++;
    pthread_mutex_unlock(&conn->lock);
807 808
    return(ret);

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

/**
819
 * virReleaseDomain:
820 821
 * @domain: the domain to release
 *
822 823 824 825
 * 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.
826
 *
827 828
 * It will also unreference the associated connection object,
 * which may also be released if its ref count hits zero.
829
 */
830 831 832 833
static void
virReleaseDomain(virDomainPtr domain) {
    virConnectPtr conn = domain->conn;
    DEBUG("release domain %p %s", domain, domain->name);
834 835

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

840
    domain->magic = -1;
841
    domain->id = -1;
842
    free(domain->name);
843 844
    free(domain);

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

853 854 855 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
    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);
    }

D
Daniel P. Berrange 已提交
884
    pthread_mutex_unlock(&domain->conn->lock);
885
    return (refs);
886 887
}

888 889 890 891 892 893 894 895 896
/**
 * 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
897
 * virUnrefNetwork() is needed to not leak data.
898 899 900 901
 *
 * Returns a pointer to the network, or NULL in case of failure
 */
virNetworkPtr
902
__virGetNetwork(virConnectPtr conn, const char *name, const unsigned char *uuid) {
903 904
    virNetworkPtr ret = NULL;

905
    if ((!VIR_IS_CONNECT(conn)) || (name == NULL) || (uuid == NULL)) {
906 907 908
        virHashError(conn, VIR_ERR_INVALID_ARG, __FUNCTION__);
        return(NULL);
    }
909
    pthread_mutex_lock(&conn->lock);
910 911 912 913

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

    ret = (virNetworkPtr) virHashLookup(conn->networks, name);
914
    /* TODO check the UUID */
915
    if (ret == NULL) {
916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936
        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++;
937
    }
938 939
    ret->refs++;
    pthread_mutex_unlock(&conn->lock);
940 941
    return(ret);

942 943
 error:
    pthread_mutex_unlock(&conn->lock);
944
    if (ret != NULL) {
945
        free(ret->name );
946
        free(ret);
947 948 949 950 951
    }
    return(NULL);
}

/**
952
 * virReleaseNetwork:
953 954
 * @network: the network to release
 *
955 956 957 958
 * 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.
959
 *
960 961
 * It will also unreference the associated connection object,
 * which may also be released if its ref count hits zero.
962
 */
963 964 965 966
static void
virReleaseNetwork(virNetworkPtr network) {
    virConnectPtr conn = network->conn;
    DEBUG("release network %p %s", network, network->name);
967 968

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

973
    network->magic = -1;
974
    free(network->name);
975 976
    free(network);

977 978 979 980 981 982 983
    DEBUG("unref connection %p %s %d", conn, conn->name, conn->refs);
    conn->refs--;
    if (conn->refs == 0) {
        virReleaseConnect(conn);
        /* Already unlocked mutex */
        return;
    }
984

985 986
    pthread_mutex_unlock(&conn->lock);
}
987

988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015

/**
 * 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);
    }

D
Daniel P. Berrange 已提交
1016
    pthread_mutex_unlock(&network->conn->lock);
1017
    return (refs);
1018 1019
}

1020 1021 1022 1023 1024
/*
 * vim: set tabstop=4:
 * vim: set shiftwidth=4:
 * vim: set expandtab:
 */
1025 1026 1027 1028 1029 1030 1031 1032
/*
 * Local variables:
 *  indent-tabs-mode: nil
 *  c-indent-level: 4
 *  c-basic-offset: 4
 *  tab-width: 4
 * End:
 */