mirror of
https://github.com/libguestfs/libguestfs.git
synced 2026-03-21 22:53:37 +00:00
75abec1f70
This brings libguestfs into line with other projects which have a separate include/ directory for the public header. It's also the case that <guestfs.h> has never particularly belonged in the lib/ subdirectory. Some tools add -Ilib/ but they only need <guestfs.h> and not any other headers from that directory, and separating out the public header allows us to clean those up. This is certainly the case for examples, and some language bindings and some tests. In future I'm hopeful we can use this as the basis to tease out other dependencies, as a prelude to separating them out from the repo.
121 lines
3.3 KiB
C
121 lines
3.3 KiB
C
/* libguestfs
|
|
* Copyright (C) 2011 Red Hat Inc.
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with this program; if not, write to the Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*/
|
|
|
|
/* Test aspects of the private data area API. */
|
|
|
|
#include <config.h>
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#include <assert.h>
|
|
#include <errno.h>
|
|
#include <error.h>
|
|
|
|
#include "guestfs.h"
|
|
|
|
#define PREFIX "test_"
|
|
|
|
static size_t close_callback_called = 0;
|
|
|
|
/* This callback deletes all test keys in the handle. */
|
|
static void
|
|
close_callback (guestfs_h *g,
|
|
void *opaque,
|
|
uint64_t event,
|
|
int event_handle,
|
|
int flags,
|
|
const char *buf, size_t buf_len,
|
|
const uint64_t *array, size_t array_len)
|
|
{
|
|
const char *key;
|
|
void *data;
|
|
|
|
close_callback_called++;
|
|
|
|
again:
|
|
data = guestfs_first_private (g, &key);
|
|
while (data != NULL) {
|
|
if (strncmp (key, PREFIX, strlen (PREFIX)) == 0) {
|
|
guestfs_set_private (g, key, NULL);
|
|
goto again;
|
|
}
|
|
data = guestfs_next_private (g, &key);
|
|
}
|
|
}
|
|
|
|
int
|
|
main (int argc, char *argv[])
|
|
{
|
|
guestfs_h *g;
|
|
const char *key;
|
|
void *data;
|
|
size_t count;
|
|
|
|
g = guestfs_create ();
|
|
if (g == NULL)
|
|
error (EXIT_FAILURE, errno, "guestfs_create");
|
|
|
|
if (guestfs_set_event_callback (g, close_callback, GUESTFS_EVENT_CLOSE,
|
|
0, NULL) == -1)
|
|
exit (EXIT_FAILURE);
|
|
|
|
guestfs_set_private (g, PREFIX "a", (void *) 1);
|
|
guestfs_set_private (g, PREFIX "b", (void *) 2);
|
|
guestfs_set_private (g, PREFIX "c", (void *) 3);
|
|
guestfs_set_private (g, PREFIX "a", (void *) 4); /* overwrites previous */
|
|
|
|
/* Check we can fetch keys. */
|
|
assert (guestfs_get_private (g, PREFIX "a") == (void *) 4);
|
|
assert (guestfs_get_private (g, PREFIX "b") == (void *) 2);
|
|
assert (guestfs_get_private (g, PREFIX "c") == (void *) 3);
|
|
assert (guestfs_get_private (g, PREFIX "d") == NULL);
|
|
|
|
/* Check we can count keys by iterating. */
|
|
count = 0;
|
|
data = guestfs_first_private (g, &key);
|
|
while (data != NULL) {
|
|
if (strncmp (key, PREFIX, strlen (PREFIX)) == 0)
|
|
count++;
|
|
data = guestfs_next_private (g, &key);
|
|
}
|
|
assert (count == 3);
|
|
|
|
/* Delete some keys. */
|
|
guestfs_set_private (g, PREFIX "a", NULL);
|
|
guestfs_set_private (g, PREFIX "b", NULL);
|
|
|
|
/* Count them again. */
|
|
count = 0;
|
|
data = guestfs_first_private (g, &key);
|
|
while (data != NULL) {
|
|
if (strncmp (key, PREFIX, strlen (PREFIX)) == 0)
|
|
count++;
|
|
data = guestfs_next_private (g, &key);
|
|
}
|
|
assert (count == 1);
|
|
|
|
/* Closing should implicitly call the close_callback function. */
|
|
guestfs_close (g);
|
|
|
|
assert (close_callback_called == 1);
|
|
|
|
exit (EXIT_SUCCESS);
|
|
}
|